160 lines
4 KiB
Go
160 lines
4 KiB
Go
package main
|
|
|
|
import (
|
|
"flag"
|
|
"fmt"
|
|
"log"
|
|
"os"
|
|
"os/signal"
|
|
"strings"
|
|
|
|
"github.com/bwmarrin/discordgo"
|
|
)
|
|
|
|
// Bot parameters
|
|
var (
|
|
GuildID = flag.String("guild", "", "Test guild ID")
|
|
BotToken = flag.String("token", "", "Bot access token")
|
|
AppID = flag.String("app", "", "Application ID")
|
|
Cleanup = flag.Bool("cleanup", true, "Cleanup of commands")
|
|
ResultsChannel = flag.String("results", "", "Channel where send survey results to")
|
|
)
|
|
|
|
var s *discordgo.Session
|
|
|
|
func init() {
|
|
flag.Parse()
|
|
}
|
|
|
|
func init() {
|
|
var err error
|
|
s, err = discordgo.New("Bot " + *BotToken)
|
|
if err != nil {
|
|
log.Fatalf("Invalid bot parameters: %v", err)
|
|
}
|
|
}
|
|
|
|
var (
|
|
commands = []discordgo.ApplicationCommand{
|
|
{
|
|
Name: "modals-survey",
|
|
Description: "Take a survey about modals",
|
|
},
|
|
}
|
|
commandsHandlers = map[string]func(s *discordgo.Session, i *discordgo.InteractionCreate){
|
|
"modals-survey": func(s *discordgo.Session, i *discordgo.InteractionCreate) {
|
|
err := s.InteractionRespond(i.Interaction, &discordgo.InteractionResponse{
|
|
Type: discordgo.InteractionResponseModal,
|
|
Data: &discordgo.InteractionResponseData{
|
|
CustomID: "modals_survey_" + i.Interaction.Member.User.ID,
|
|
Title: "Modals survey",
|
|
Components: []discordgo.MessageComponent{
|
|
discordgo.ActionsRow{
|
|
Components: []discordgo.MessageComponent{
|
|
discordgo.TextInput{
|
|
CustomID: "opinion",
|
|
Label: "What is your opinion on them?",
|
|
Style: discordgo.TextInputShort,
|
|
Placeholder: "Don't be shy, share your opinion with us",
|
|
Required: true,
|
|
MaxLength: 300,
|
|
MinLength: 10,
|
|
},
|
|
},
|
|
},
|
|
discordgo.ActionsRow{
|
|
Components: []discordgo.MessageComponent{
|
|
discordgo.TextInput{
|
|
CustomID: "suggestions",
|
|
Label: "What would you suggest to improve them?",
|
|
Style: discordgo.TextInputParagraph,
|
|
Required: false,
|
|
MaxLength: 2000,
|
|
},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
})
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
},
|
|
}
|
|
)
|
|
|
|
func main() {
|
|
s.AddHandler(func(s *discordgo.Session, r *discordgo.Ready) {
|
|
log.Println("Bot is up!")
|
|
})
|
|
|
|
s.AddHandler(func(s *discordgo.Session, i *discordgo.InteractionCreate) {
|
|
switch i.Type {
|
|
case discordgo.InteractionApplicationCommand:
|
|
if h, ok := commandsHandlers[i.ApplicationCommandData().Name]; ok {
|
|
h(s, i)
|
|
}
|
|
case discordgo.InteractionModalSubmit:
|
|
err := s.InteractionRespond(i.Interaction, &discordgo.InteractionResponse{
|
|
Type: discordgo.InteractionResponseChannelMessageWithSource,
|
|
Data: &discordgo.InteractionResponseData{
|
|
Content: "Thank you for taking your time to fill this survey",
|
|
Flags: discordgo.MessageFlagsEphemeral,
|
|
},
|
|
})
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
data := i.ModalSubmitData()
|
|
|
|
if !strings.HasPrefix(data.CustomID, "modals_survey") {
|
|
return
|
|
}
|
|
|
|
userid := strings.Split(data.CustomID, "_")[2]
|
|
_, err = s.ChannelMessageSend(*ResultsChannel, fmt.Sprintf(
|
|
"Feedback received. From <@%s>\n\n**Opinion**:\n%s\n\n**Suggestions**:\n%s",
|
|
userid,
|
|
data.Components[0].(*discordgo.ActionsRow).Components[0].(*discordgo.TextInput).Value,
|
|
data.Components[1].(*discordgo.ActionsRow).Components[0].(*discordgo.TextInput).Value,
|
|
))
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
}
|
|
})
|
|
|
|
cmdIDs := make(map[string]string, len(commands))
|
|
|
|
for _, cmd := range commands {
|
|
rcmd, err := s.ApplicationCommandCreate(*AppID, *GuildID, &cmd)
|
|
if err != nil {
|
|
log.Fatalf("Cannot create slash command %q: %v", cmd.Name, err)
|
|
}
|
|
|
|
cmdIDs[rcmd.ID] = rcmd.Name
|
|
}
|
|
|
|
err := s.Open()
|
|
if err != nil {
|
|
log.Fatalf("Cannot open the session: %v", err)
|
|
}
|
|
defer s.Close()
|
|
|
|
stop := make(chan os.Signal, 1)
|
|
signal.Notify(stop, os.Interrupt)
|
|
<-stop
|
|
log.Println("Graceful shutdown")
|
|
|
|
if !*Cleanup {
|
|
return
|
|
}
|
|
|
|
for id, name := range cmdIDs {
|
|
err := s.ApplicationCommandDelete(*AppID, *GuildID, id)
|
|
if err != nil {
|
|
log.Fatalf("Cannot delete slash command %q: %v", name, err)
|
|
}
|
|
}
|
|
|
|
}
|