feat: add tally command

pkg/reporters/discord/discord.go

@@ -96,6 +96,7 @@ func (reporter *Reporter) Init() error {
 		"proposals_unmute": reporter.GetDeleteMuteCommand(),
 		"proposals_mutes":  reporter.GetMutesCommand(),
 		"params":           reporter.GetParamsCommand(),
+		"tally":            reporter.GetTallyCommand(),
 	}
 
 	go reporter.InitCommands()
@@ -236,24 +237,31 @@ func (reporter *Reporter) BotRespond(s *discordgo.Session, i *discordgo.Interact
 	chunks := utils.SplitStringIntoChunks(text, 2000)
 	firstChunk, rest := chunks[0], chunks[1:]
 
+	reporter.BotSendInteraction(s, i, firstChunk)
+
+	for _, chunk := range rest {
+		reporter.BotSendFollowup(s, i, chunk)
+	}
+}
+
+func (reporter *Reporter) BotSendInteraction(s *discordgo.Session, i *discordgo.InteractionCreate, text string) {
 	if err := s.InteractionRespond(i.Interaction, &discordgo.InteractionResponse{
 		Type: discordgo.InteractionResponseChannelMessageWithSource,
 		Data: &discordgo.InteractionResponseData{
-			Content: firstChunk,
+			Content: text,
 		},
 	}); err != nil {
 		reporter.Logger.Error().Err(err).Msg("Error sending response")
 	}
+}
 
-	for index, chunk := range rest {
-		if _, err := s.FollowupMessageCreate(i.Interaction, false, &discordgo.WebhookParams{
-			Content: chunk,
-		}); err != nil {
-			reporter.Logger.Error().
-				Int("chunk", index).
-				Err(err).
-				Msg("Error sending followup message")
-		}
+func (reporter *Reporter) BotSendFollowup(s *discordgo.Session, i *discordgo.InteractionCreate, text string) {
+	if _, err := s.FollowupMessageCreate(i.Interaction, false, &discordgo.WebhookParams{
+		Content: text,
+	}); err != nil {
+		reporter.Logger.Error().
+			Err(err).
+			Msg("Error sending followup message")
 	}
 }
 

pkg/reporters/discord/tally.go

@@ -0,0 +1,38 @@
+package discord
+
+import (
+	"context"
+	"fmt"
+	"main/pkg/utils"
+
+	"github.com/bwmarrin/discordgo"
+)
+
+func (reporter *Reporter) GetTallyCommand() *Command {
+	return &Command{
+		Info: &discordgo.ApplicationCommand{
+			Name:        "tally",
+			Description: "Get active proposals' tallies",
+		},
+		Handler: func(s *discordgo.Session, i *discordgo.InteractionCreate) {
+			reporter.BotSendInteraction(s, i, "Calculating tally for proposals. This might take a while...")
+
+			tallies, err := reporter.DataManager.GetTallies(context.Background())
+			if err != nil {
+				reporter.BotRespond(s, i, fmt.Sprintf("Error getting tallies info: %s", err))
+				return
+			}
+
+			template, err := reporter.TemplatesManager.Render("tally", tallies)
+			if err != nil {
+				reporter.Logger.Error().Err(err).Str("template", "tally").Msg("Error rendering template")
+				return
+			}
+
+			chunks := utils.SplitStringIntoChunks(template, 2000)
+			for _, chunk := range chunks {
+				reporter.BotSendFollowup(s, i, chunk)
+			}
+		},
+	}
+}

templates/discord/help.html

@@ -7,6 +7,7 @@
 - </proposals_unmute:{{ .Commands.proposals_unmute.Info.ID }}> - unmute notifications for a specific chain or proposal
 - </proposals_mutes:{{ .Commands.proposals_mutes.Info.ID }}> - displays the active mutes list
 - </params:{{ .Commands.params.Info.ID }}> - list chains params
+- </tally:{{ .Commands.tally.Info.ID }}> - list active proposals' tallies
 - </help:{{ .Commands.help.Info.ID }}> - display this message
 
 Created by [🐹 Quokka Stake](<https://quokkastake.io>) with ❤️.

templates/discord/tally.html

@@ -0,0 +1,18 @@
+{{- if not . }}
+**No active proposals.**
+{{- end }}
+{{- range $chainName, $tallyInfos := . }}
+{{- if $tallyInfos.TallyInfos }}
+**Proposals on chain {{ $tallyInfos.Chain.GetName }}:**
+{{ range $chainIndex, $tallyInfo := $tallyInfos.TallyInfos }}
+{{- $proposalLink := $tallyInfos.Chain.GetProposalLink .Proposal }}
+Proposal #{{ .Proposal.ID }}: {{ SerializeLink $proposalLink }}
+Ends in: {{ .Proposal.GetTimeLeft }}
+- Not voted: {{ .GetNotVoted }}
+- Voted: {{ .GetQuorum }}
+{{- range $tallyOptionIndex, $tallyOption := .Tally }}
+- Voted "{{ $tallyOption.Option }}": {{ $tallyInfo.Tally.GetVoted $tallyOption }}
+{{- end }}
+{{ end }}
+{{- end }}
+{{ end }}

templates/telegram/help.html

@@ -6,6 +6,7 @@
 - /proposals_unmute [<chain> <proposal ID>] - unmute notifications for a specific chain/proposal
 - /proposals_mutes - display the active proposals mutes list
 - /params - list chains params
+- /tally - list active proposals' tallies
 - /help - display this command
 
 Created by <a href="https://quokkastake.io">🐹 Quokka Stake</a> with ❤️.