feat: add mute command for discord

pkg/app.go

@@ -79,6 +79,7 @@ func NewApp(configPath string, filesystem fs.FS, version string) *App {
 			version,
 			log,
 			stateManager,
+			mutesManager,
 			stateGenerator,
 			timeZone,
 			tracer,

pkg/reporters/discord/add_mute.go

@@ -0,0 +1,82 @@
+package discord
+
+import (
+	"fmt"
+	mutes "main/pkg/mutes"
+	"time"
+
+	"github.com/bwmarrin/discordgo"
+)
+
+func (reporter *Reporter) GetAddMuteCommand() *Command {
+	return &Command{
+		Info: &discordgo.ApplicationCommand{
+			Name:        "proposals_mute",
+			Description: "Mute proposals' notifications",
+			Options: []*discordgo.ApplicationCommandOption{
+				{
+					Type:        discordgo.ApplicationCommandOptionString,
+					Name:        "duration",
+					Description: "For how long to mute notifications",
+					Required:    true,
+				},
+				{
+					Type:        discordgo.ApplicationCommandOptionString,
+					Name:        "chain",
+					Description: "Chain to mute notifications on",
+					Required:    false,
+				},
+				{
+					Type:        discordgo.ApplicationCommandOptionString,
+					Name:        "proposal",
+					Description: "Proposal to mute notifications on",
+					Required:    false,
+				},
+			},
+		},
+		Handler: func(s *discordgo.Session, i *discordgo.InteractionCreate) {
+			options := i.ApplicationCommandData().Options
+
+			durationString, _ := options[0].Value.(string)
+			var chain string
+			var proposal string
+
+			_, opts := options[0], options[1:]
+
+			for _, opt := range opts {
+				if opt.Name == "chain" {
+					chain, _ = opt.Value.(string)
+				}
+				if opt.Name == "proposal" {
+					proposal, _ = opt.Value.(string)
+				}
+			}
+
+			duration, err := time.ParseDuration(durationString)
+			if err != nil {
+				reporter.BotRespond(s, i, "Invalid mute duration provided: %s")
+				return
+			}
+
+			mute := &mutes.Mute{
+				Chain:      chain,
+				ProposalID: proposal,
+				Expires:    time.Now().Add(duration),
+				Comment: fmt.Sprintf(
+					"Muted using cosmos-proposals-checker for %s",
+					duration,
+				),
+			}
+
+			reporter.MutesManager.AddMute(mute)
+
+			template, err := reporter.TemplatesManager.Render("mute_added", mute)
+			if err != nil {
+				reporter.Logger.Error().Err(err).Str("template", "mute_added").Msg("Error rendering template")
+				return
+			}
+
+			reporter.BotRespond(s, i, template)
+		},
+	}
+}

pkg/reporters/discord/discord.go

@@ -2,6 +2,7 @@ package discord
 
 import (
 	"context"
+	mutes "main/pkg/mutes"
 	"main/pkg/report/entry"
 	statePkg "main/pkg/state"
 	templatesPkg "main/pkg/templates"
@@ -28,6 +29,7 @@ type Reporter struct {
 	Logger           zerolog.Logger
 	Config           *types.Config
 	Manager          *statePkg.Manager
+	MutesManager     *mutes.Manager
 	TemplatesManager templatesPkg.Manager
 	Commands         map[string]*Command
 	Tracer           trace.Tracer
@@ -39,6 +41,7 @@ func NewReporter(
 	version string,
 	logger *zerolog.Logger,
 	manager *statePkg.Manager,
+	mutesManager *mutes.Manager,
 	stateGenerator *statePkg.Generator,
 	timezone *time.Location,
 	tracer trace.Tracer,
@@ -50,6 +53,7 @@ func NewReporter(
 		Config:           config,
 		Logger:           logger.With().Str("component", "discord_reporter").Logger(),
 		Manager:          manager,
+		MutesManager:     mutesManager,
 		StateGenerator:   stateGenerator,
 		TemplatesManager: templatesPkg.NewDiscordTemplatesManager(logger, timezone),
 		Commands:         make(map[string]*Command, 0),
@@ -82,8 +86,9 @@ func (reporter *Reporter) Init() error {
 	reporter.Logger.Info().Err(err).Msg("Discord bot listening")
 
 	reporter.Commands = map[string]*Command{
-		"help":      reporter.GetHelpCommand(),
-		"proposals": reporter.GetProposalsCommand(),
+		"help":           reporter.GetHelpCommand(),
+		"proposals":      reporter.GetProposalsCommand(),
+		"proposals_mute": reporter.GetAddMuteCommand(),
 	}
 
 	go reporter.InitCommands()

templates/discord/help.html

@@ -3,7 +3,7 @@
 Notifies you about the proposals your wallets hasn't voted upon.
 Can understand the following commands:
 - </proposals:{{ .Commands.proposals.Info.ID }}> - displays active proposals and your wallets' votes on them
-- /proposals_mute &lt;duration&gt; &lt;chain&gt; &lt;proposal ID&gt; - mute notifications for a specific proposal
+- </proposals_mute:{{ .Commands.add_mute.Info.ID}}> - mute notifications for a specific proposal
 - /proposals_mutes - display the active proposals mutes list
 - </help:{{ .Commands.help.Info.ID }}> - display this message
 

templates/discord/mute_added.html

@@ -0,0 +1,12 @@
+Created mute with the following params:
+{{- if .Chain }}
+**Chain:** {{ .Chain }}
+{{- else }}
+**Chain:** all chains
+{{- end }}
+{{- if .ProposalID }}
+**Proposal ID:** {{ .ProposalID }}
+{{- else }}
+**Proposal ID:** all proposals
+{{- end }}
+**Expires: **{{ SerializeDate .Expires }}