80 lines
1.9 KiB
Go
80 lines
1.9 KiB
Go
package discord
|
|
|
|
import (
|
|
"fmt"
|
|
"log"
|
|
"os"
|
|
"os/signal"
|
|
"ticket-pimp/client/discord/handler"
|
|
"ticket-pimp/internal/controller"
|
|
"ticket-pimp/internal/domain"
|
|
|
|
"github.com/bwmarrin/discordgo"
|
|
)
|
|
|
|
func initBotWith(token string) *discordgo.Session {
|
|
discord, err := discordgo.New("Bot " + token)
|
|
if err != nil {
|
|
log.Fatalf("unable to create discord session: %v", err)
|
|
}
|
|
return discord
|
|
}
|
|
|
|
type DiscordOptions struct {
|
|
AppConfig *domain.Config
|
|
Controller *controller.WorkflowController
|
|
}
|
|
|
|
func Run(conf domain.Config, opts DiscordOptions) error {
|
|
token := conf.Discord.Token
|
|
|
|
session := initBotWith(token)
|
|
|
|
router := handler.InitRouter(*opts.Controller, &conf.Discord)
|
|
|
|
commandHandlers := map[string]func(s *discordgo.Session, i *discordgo.InteractionCreate){}
|
|
for _, handler := range router.Routes {
|
|
commandHandlers[handler.Command.Name] = handler.Handler
|
|
}
|
|
|
|
session.AddHandler(func(s *discordgo.Session, i *discordgo.InteractionCreate) {
|
|
if h, ok := commandHandlers[i.ApplicationCommandData().Name]; ok {
|
|
h(s, i)
|
|
}
|
|
})
|
|
|
|
if err := session.Open(); err != nil {
|
|
return fmt.Errorf("cannot open the session: %v", err)
|
|
}
|
|
|
|
log.Println("Adding commands...")
|
|
var cmds []*discordgo.ApplicationCommand
|
|
var logString []string
|
|
for _, h := range router.Routes {
|
|
cmd, err := session.ApplicationCommandCreate(session.State.User.ID, "", &h.Command)
|
|
if err != nil {
|
|
log.Panicf("Cannot create '%v' command: %v", h.Command.Name, err)
|
|
}
|
|
cmds = append(cmds, cmd)
|
|
logString = append(logString, cmd.Name)
|
|
}
|
|
|
|
log.Println("Following commands added:")
|
|
log.Println(logString)
|
|
defer session.Close()
|
|
stop := make(chan os.Signal, 1)
|
|
signal.Notify(stop, os.Interrupt)
|
|
<-stop
|
|
log.Println("Graceful shutdown")
|
|
|
|
log.Println("Removing commands...")
|
|
for _, h := range cmds {
|
|
err := session.ApplicationCommandDelete(session.State.User.ID, "", h.ID)
|
|
if err != nil {
|
|
log.Panicf("Cannot delete '%v' command: %v", h.Name, err)
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|