-
Notifications
You must be signed in to change notification settings - Fork 0
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
Showing
8 changed files
with
333 additions
and
33 deletions.
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,172 @@ | ||
| package slack | ||
|
|
||
| import ( | ||
| "context" | ||
| "fmt" | ||
| "github.com/kballard/go-shellquote" | ||
| "github.com/slack-go/slack" | ||
| "github.com/slack-go/slack/slackevents" | ||
| "github.com/slack-go/slack/socketmode" | ||
| "github.com/traPtitech/DevOpsBot/pkg/config" | ||
| "github.com/traPtitech/DevOpsBot/pkg/domain" | ||
| "go.uber.org/zap" | ||
| "strings" | ||
| ) | ||
|
|
||
| const slashPrefix = "/" | ||
|
|
||
| type slackBot struct { | ||
| api *slack.Client | ||
| sock *socketmode.Client | ||
| rootCmd domain.Command | ||
| logger *zap.Logger | ||
| } | ||
|
|
||
| func NewBot(rootCmd domain.Command, logger *zap.Logger) (domain.Bot, error) { | ||
| // Prepare socket mode bot | ||
| api := slack.New(config.C.Slack.OAuthToken, slack.OptionAppLevelToken(config.C.Slack.AppToken)) | ||
| sock := socketmode.New(api) | ||
|
|
||
| return &slackBot{ | ||
| api: api, | ||
| sock: sock, | ||
| rootCmd: rootCmd, | ||
| logger: logger, | ||
| }, nil | ||
| } | ||
|
|
||
| func (s *slackBot) Start(ctx context.Context) error { | ||
| // Join channel | ||
| _, _, _, err := s.api.JoinConversation(config.C.Slack.ChannelID) | ||
| if err != nil { | ||
| return err | ||
| } | ||
|
|
||
| go func() { | ||
| for e := range s.sock.Events { | ||
| err := s.handle(e) | ||
| if err != nil { | ||
| s.logger.Error("failed to process event", zap.Error(err)) | ||
| } | ||
| } | ||
| }() | ||
| return s.sock.RunContext(ctx) | ||
| } | ||
|
|
||
| func (s *slackBot) handle(e socketmode.Event) error { | ||
| switch e.Type { | ||
| case socketmode.EventTypeConnecting: | ||
| s.logger.Info("Connecting to Slack with Socket Mode...") | ||
|
|
||
| case socketmode.EventTypeConnectionError: | ||
| s.logger.Info("Connection failed. Retrying later...") | ||
|
|
||
| case socketmode.EventTypeConnected: | ||
| s.logger.Info("Connected to Slack with Socket Mode.") | ||
|
|
||
| case socketmode.EventTypeEventsAPI: | ||
| eventsE, ok := e.Data.(slackevents.EventsAPIEvent) | ||
| if !ok { | ||
| return fmt.Errorf("failed to parse events api type") | ||
| } | ||
|
|
||
| // Acknowledge the event | ||
| s.sock.Ack(*e.Request) | ||
|
|
||
| // Process the event | ||
| err := s.handleEventsAPI(&eventsE) | ||
| if err != nil { | ||
| return fmt.Errorf("failed to process events api event: %w", err) | ||
| } | ||
|
|
||
| case socketmode.EventTypeSlashCommand: | ||
| slashE, ok := e.Data.(slack.SlashCommand) | ||
| if !ok { | ||
| return fmt.Errorf("failed to parse slash command type") | ||
| } | ||
|
|
||
| // Acknowledge the event | ||
| s.sock.Ack(*e.Request, map[string]any{ | ||
| "response_type": "in_channel", | ||
| }) | ||
|
|
||
| // Process the event | ||
| err := s.handleSlashEvent(&slashE) | ||
| if err != nil { | ||
| return fmt.Errorf("failed to process slash event: %w", err) | ||
| } | ||
| } | ||
|
|
||
| return nil | ||
| } | ||
|
|
||
| func (s *slackBot) handleEventsAPI(e *slackevents.EventsAPIEvent) error { | ||
| switch ev := e.InnerEvent.Data.(type) { | ||
| case *slackevents.MessageEvent: | ||
| // Validate command execution context | ||
| if ev.BotID != "" { | ||
| return nil // Ignore bots | ||
| } | ||
| if ev.Channel != config.C.Slack.ChannelID { | ||
| return nil // Ignore messages not from the specified channel | ||
| } | ||
| if !strings.HasPrefix(ev.Text, config.C.Prefix) { | ||
| return nil // Command prefix does not match | ||
| } | ||
|
|
||
| // Execute | ||
| messageRef := slack.ItemRef{ | ||
| Channel: ev.Channel, | ||
| Timestamp: ev.TimeStamp, | ||
| } | ||
| commandText := strings.Trim(ev.Text, config.C.Prefix) | ||
| return s.executeCommand(commandText, messageRef, ev.User) | ||
| default: | ||
| return nil | ||
| } | ||
| } | ||
|
|
||
| func (s *slackBot) handleSlashEvent(e *slack.SlashCommand) error { | ||
| // Validate command execution context | ||
| if e.ChannelID != config.C.Slack.ChannelID { | ||
| return nil // Ignore messages not from the specified channel | ||
| } | ||
|
|
||
| // Prepare a new message to add reaction to | ||
| commandText := fmt.Sprintf("%s %s", e.Command, e.Text) | ||
| responseText := fmt.Sprintf("%s <@%s|%s> used slash command: %s", | ||
| e.UserName, e.UserID, e.UserName, | ||
| commandText) | ||
| _, ts, err := s.sock.PostMessage(e.ChannelID, slack.MsgOptionText(responseText, false)) | ||
| if err != nil { | ||
| return fmt.Errorf("failed to post message in response to slash command: %w", err) | ||
| } | ||
|
|
||
| // Execute | ||
| messageRef := slack.ItemRef{ | ||
| Channel: e.ChannelID, | ||
| Timestamp: ts, | ||
| } | ||
| commandText = strings.TrimPrefix(commandText, slashPrefix) | ||
| return s.executeCommand(commandText, messageRef, e.UserID) | ||
| } | ||
|
|
||
| func (s *slackBot) executeCommand(commandText string, messageRef slack.ItemRef, executorID string) error { | ||
| // Prepare command args | ||
| ctx := &slackContext{ | ||
| Context: context.Background(), | ||
| api: s.api, | ||
| logger: s.logger, | ||
| message: messageRef, | ||
| executorID: executorID, | ||
| args: nil, | ||
| } | ||
| args, err := shellquote.Split(commandText) | ||
| if err != nil { | ||
| return ctx.ReplyBad(fmt.Sprintf("failed to parse arguments: %v", err)) | ||
| } | ||
| ctx.args = args | ||
|
|
||
| // Execute | ||
| return s.rootCmd.Execute(ctx) | ||
| } |
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,96 @@ | ||
| package slack | ||
|
|
||
| import ( | ||
| "context" | ||
| "github.com/slack-go/slack" | ||
| "github.com/traPtitech/DevOpsBot/pkg/config" | ||
| "github.com/traPtitech/DevOpsBot/pkg/domain" | ||
| "github.com/traPtitech/DevOpsBot/pkg/utils" | ||
| "go.uber.org/zap" | ||
| "strings" | ||
| ) | ||
|
|
||
| type slackContext struct { | ||
| context.Context | ||
| api *slack.Client | ||
| logger *zap.Logger | ||
|
|
||
| message slack.ItemRef | ||
| executorID string | ||
| args []string | ||
| } | ||
|
|
||
| func (ctx *slackContext) Executor() string { | ||
| return ctx.executorID | ||
| } | ||
|
|
||
| func (ctx *slackContext) Args() []string { | ||
| return ctx.args | ||
| } | ||
|
|
||
| func (ctx *slackContext) ShiftArgs() domain.Context { | ||
| newCtx := *ctx | ||
| newCtx.args = newCtx.args[1:] | ||
| return &newCtx | ||
| } | ||
|
|
||
| func (ctx *slackContext) L() *zap.Logger { | ||
| return ctx.logger.With( | ||
| zap.String("executor", ctx.Executor()), | ||
| //zap.String("command", s.p.Message.PlainText), | ||
| //zap.Time("datetime", s.p.EventTime), | ||
| ) | ||
| } | ||
|
|
||
| func (ctx *slackContext) sendSlackMessage(channelID string, text string) error { | ||
| api := ctx.api | ||
| return utils.WithRetry(ctx, 10, func(ctx context.Context) error { | ||
| _, _, err := api.PostMessage(channelID, slack.MsgOptionText(text, false)) | ||
| return err | ||
| }) | ||
| } | ||
|
|
||
| func (ctx *slackContext) pushSlackReaction(message slack.ItemRef, stampID string) error { | ||
| api := ctx.api | ||
| return utils.WithRetry(ctx, 10, func(ctx context.Context) error { | ||
| return api.AddReaction(stampID, message) | ||
| }) | ||
| } | ||
|
|
||
| func (ctx *slackContext) reply(message ...string) error { | ||
| return ctx.sendSlackMessage(ctx.message.Channel, strings.Join(message, "\n")) | ||
| } | ||
|
|
||
| func (ctx *slackContext) replyWithStamp(stamp string, message ...string) error { | ||
| err := ctx.pushSlackReaction(ctx.message, stamp) | ||
| if err != nil { | ||
| return err | ||
| } | ||
| if len(message) > 0 { | ||
| err = ctx.reply(message...) | ||
| if err != nil { | ||
| return err | ||
| } | ||
| } | ||
| return nil | ||
| } | ||
|
|
||
| func (ctx *slackContext) ReplyBad(message ...string) error { | ||
| return ctx.replyWithStamp(config.C.Stamps.BadCommand, message...) | ||
| } | ||
|
|
||
| func (ctx *slackContext) ReplyForbid(message ...string) error { | ||
| return ctx.replyWithStamp(config.C.Stamps.Forbid, message...) | ||
| } | ||
|
|
||
| func (ctx *slackContext) ReplySuccess(message ...string) error { | ||
| return ctx.replyWithStamp(config.C.Stamps.Success, message...) | ||
| } | ||
|
|
||
| func (ctx *slackContext) ReplyFailure(message ...string) error { | ||
| return ctx.replyWithStamp(config.C.Stamps.Failure, message...) | ||
| } | ||
|
|
||
| func (ctx *slackContext) ReplyRunning(message ...string) error { | ||
| return ctx.replyWithStamp(config.C.Stamps.Running, message...) | ||
| } |
Oops, something went wrong.