Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
1 parent
5cb4556
commit cb4ded3
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,202 @@ | ||
| package discord | ||
|
|
||
| import ( | ||
| "context" | ||
| "main/pkg/report/entry" | ||
| statePkg "main/pkg/state" | ||
| templatesPkg "main/pkg/templates" | ||
| types "main/pkg/types" | ||
| "main/pkg/utils" | ||
| "sync" | ||
| "time" | ||
|
|
||
| "go.opentelemetry.io/otel/trace" | ||
|
|
||
| "github.com/bwmarrin/discordgo" | ||
| "github.com/rs/zerolog" | ||
| ) | ||
|
|
||
| type Reporter struct { | ||
| Token string | ||
| Guild string | ||
| Channel string | ||
|
|
||
| Version string | ||
|
|
||
| DiscordSession *discordgo.Session | ||
| Logger zerolog.Logger | ||
| Config *types.Config | ||
| Manager *statePkg.Manager | ||
| TemplatesManager templatesPkg.Manager | ||
| Commands map[string]*Command | ||
| Tracer trace.Tracer | ||
| Timezone *time.Location | ||
| } | ||
|
|
||
| func NewReporter( | ||
| config *types.Config, | ||
| version string, | ||
| logger *zerolog.Logger, | ||
| manager *statePkg.Manager, | ||
| timezone *time.Location, | ||
| tracer trace.Tracer, | ||
| ) *Reporter { | ||
| return &Reporter{ | ||
| Token: config.DiscordConfig.Token, | ||
| Guild: config.DiscordConfig.Guild, | ||
| Channel: config.DiscordConfig.Channel, | ||
| Config: config, | ||
| Logger: logger.With().Str("component", "discord_reporter").Logger(), | ||
| Manager: manager, | ||
| TemplatesManager: templatesPkg.NewDiscordTemplatesManager(logger, timezone), | ||
| Commands: make(map[string]*Command, 0), | ||
| Version: version, | ||
| Timezone: timezone, | ||
| Tracer: tracer, | ||
| } | ||
| } | ||
|
|
||
| func (reporter *Reporter) Init() error { | ||
| if !reporter.Enabled() { | ||
| reporter.Logger.Debug().Msg("Discord credentials not set, not creating Discord reporter") | ||
| return nil | ||
| } | ||
| session, err := discordgo.New("Bot " + reporter.Token) | ||
| if err != nil { | ||
| reporter.Logger.Warn().Err(err).Msg("Error initializing Discord bot") | ||
| return err | ||
| } | ||
|
|
||
| reporter.DiscordSession = session | ||
|
|
||
| // Open a websocket connection to Discord and begin listening. | ||
| err = session.Open() | ||
| if err != nil { | ||
| reporter.Logger.Warn().Err(err).Msg("Error opening Discord websocket session") | ||
| return nil | ||
| } | ||
|
|
||
| reporter.Logger.Info().Err(err).Msg("Discord bot listening") | ||
|
|
||
| reporter.Commands = map[string]*Command{ | ||
| "help": reporter.GetHelpCommand(), | ||
| } | ||
|
|
||
| go reporter.InitCommands() | ||
| return nil | ||
| } | ||
|
|
||
| func (reporter *Reporter) InitCommands() { | ||
| session := reporter.DiscordSession | ||
| var wg sync.WaitGroup | ||
| var mutex sync.Mutex | ||
|
|
||
| session.AddHandler(func(s *discordgo.Session, i *discordgo.InteractionCreate) { | ||
| commandName := i.ApplicationCommandData().Name | ||
|
|
||
| if command, ok := reporter.Commands[commandName]; ok { | ||
| command.Handler(s, i) | ||
| } | ||
| }) | ||
|
|
||
| registeredCommands, err := session.ApplicationCommands(session.State.User.ID, reporter.Guild) | ||
| if err != nil { | ||
| reporter.Logger.Error().Err(err).Msg("Could not fetch registered commands") | ||
| return | ||
| } | ||
|
|
||
| for _, command := range registeredCommands { | ||
| wg.Add(1) | ||
| go func(command *discordgo.ApplicationCommand) { | ||
| defer wg.Done() | ||
|
|
||
| err := session.ApplicationCommandDelete(session.State.User.ID, reporter.Guild, command.ID) | ||
| if err != nil { | ||
| reporter.Logger.Error().Err(err).Str("command", command.Name).Msg("Could not delete command") | ||
| return | ||
| } | ||
| reporter.Logger.Info().Str("command", command.Name).Msg("Deleted command") | ||
| }(command) | ||
| } | ||
|
|
||
| wg.Wait() | ||
|
|
||
| for key, command := range reporter.Commands { | ||
| wg.Add(1) | ||
| go func(key string, command *Command) { | ||
| defer wg.Done() | ||
|
|
||
| cmd, err := session.ApplicationCommandCreate(session.State.User.ID, reporter.Guild, command.Info) | ||
| if err != nil { | ||
| reporter.Logger.Error().Err(err).Str("command", command.Info.Name).Msg("Could not create command") | ||
| return | ||
| } | ||
| reporter.Logger.Info().Str("command", cmd.Name).Msg("Created command") | ||
|
|
||
| mutex.Lock() | ||
| reporter.Commands[key].Info = cmd | ||
| mutex.Unlock() | ||
| }(key, command) | ||
| } | ||
|
|
||
| wg.Wait() | ||
| } | ||
|
|
||
| func (reporter *Reporter) Enabled() bool { | ||
| return reporter.Token != "" && reporter.Guild != "" && reporter.Channel != "" | ||
| } | ||
|
|
||
| func (reporter *Reporter) Name() string { | ||
| return "discord-reporter" | ||
| } | ||
|
|
||
| func (reporter *Reporter) SerializeReportEntry(e entry.ReportEntry) (string, error) { | ||
| return reporter.TemplatesManager.Render(e.Name(), e) | ||
| } | ||
|
|
||
| func (reporter *Reporter) SendReportEntry(reportEntry entry.ReportEntry, ctx context.Context) error { | ||
| _, span := reporter.Tracer.Start(ctx, "Sending Telegram report entry") | ||
| defer span.End() | ||
|
|
||
| serializedEntry, err := reporter.SerializeReportEntry(reportEntry) | ||
| if err != nil { | ||
| reporter.Logger.Err(err).Msg("Could not serialize report entry") | ||
| return err | ||
| } | ||
|
|
||
| _, err = reporter.DiscordSession.ChannelMessageSend( | ||
| reporter.Channel, | ||
| serializedEntry, | ||
| ) | ||
|
|
||
| return err | ||
| } | ||
|
|
||
| func (reporter *Reporter) BotRespond(s *discordgo.Session, i *discordgo.InteractionCreate, text string) { | ||
| chunks := utils.SplitStringIntoChunks(text, 2000) | ||
| firstChunk, rest := chunks[0], chunks[1:] | ||
|
|
||
| if err := s.InteractionRespond(i.Interaction, &discordgo.InteractionResponse{ | ||
| Type: discordgo.InteractionResponseChannelMessageWithSource, | ||
| Data: &discordgo.InteractionResponseData{ | ||
| Content: firstChunk, | ||
| }, | ||
| }); 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) SerializeDate(date time.Time) string { | ||
| return date.Format(time.RFC822) | ||
| } | ||
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,26 @@ | ||
| package discord | ||
|
|
||
| import ( | ||
| "github.com/bwmarrin/discordgo" | ||
| ) | ||
|
|
||
| func (reporter *Reporter) GetHelpCommand() *Command { | ||
| return &Command{ | ||
| Info: &discordgo.ApplicationCommand{ | ||
| Name: "help", | ||
| Description: "Get the bot help", | ||
| }, | ||
| Handler: func(s *discordgo.Session, i *discordgo.InteractionCreate) { | ||
| template, err := reporter.TemplatesManager.Render("help", helpRender{ | ||
| Version: reporter.Version, | ||
| Commands: reporter.Commands, | ||
| }) | ||
| if err != nil { | ||
| reporter.Logger.Error().Err(err).Str("template", "help").Msg("Error rendering template") | ||
| return | ||
| } | ||
|
|
||
| reporter.BotRespond(s, i, template) | ||
| }, | ||
| } | ||
| } | ||
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,15 @@ | ||
| package discord | ||
|
|
||
| import ( | ||
| "github.com/bwmarrin/discordgo" | ||
| ) | ||
|
|
||
| type Command struct { | ||
| Info *discordgo.ApplicationCommand | ||
| Handler func(s *discordgo.Session, i *discordgo.InteractionCreate) | ||
| } | ||
|
|
||
| type helpRender struct { | ||
| Version string | ||
| Commands map[string]*Command | ||
| } |
Oops, something went wrong.