88 lines
2.9 KiB
Go
88 lines
2.9 KiB
Go
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"slices"
|
|
"strings"
|
|
|
|
"github.com/bwmarrin/discordgo"
|
|
)
|
|
|
|
type Command struct {
|
|
Definition discordgo.ApplicationCommand
|
|
Interact func(s *discordgo.Session, i *discordgo.InteractionCreate)
|
|
Autocomplete func(s *discordgo.Session, i *discordgo.InteractionCreate)
|
|
ModalSubmit func(s *discordgo.Session, i *discordgo.InteractionCreate)
|
|
ModalID string
|
|
}
|
|
|
|
var commands []Command = []Command{tag_command, short_get_tag_command, dadjoke_command, ping_command, ask_command, sticky_command, cat_command}
|
|
|
|
func ready(s *discordgo.Session, event *discordgo.Ready) {
|
|
removeOldCommandFromAllGuilds(s)
|
|
var existingCommandNames []string
|
|
for _, guild := range event.Guilds {
|
|
existingCommands, err := s.ApplicationCommands(s.State.User.ID, guild.ID)
|
|
for _, existingCommand := range existingCommands {
|
|
existingCommandNames = append(existingCommandNames, existingCommand.Name)
|
|
}
|
|
if err != nil {
|
|
fmt.Printf("error fetching existing commands for guild %s: %v\n", guild.Name, err)
|
|
continue
|
|
}
|
|
for _, command := range commands {
|
|
if !slices.Contains(existingCommandNames, command.Definition.Name) || slices.Contains(os.Args, "--update") {
|
|
cmd, err := s.ApplicationCommandCreate(s.State.User.ID, guild.ID, &command.Definition)
|
|
fmt.Printf("\nDeleted command \"%s\"", cmd.Name)
|
|
if err != nil {
|
|
fmt.Println("error creating command,", err)
|
|
continue
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
func interactionCreate(s *discordgo.Session, i *discordgo.InteractionCreate) {
|
|
for _, command := range commands {
|
|
switch i.Type {
|
|
case discordgo.InteractionApplicationCommand:
|
|
if command.Interact != nil && i.ApplicationCommandData().Name == command.Definition.Name {
|
|
command.Interact(s, i)
|
|
}
|
|
case discordgo.InteractionApplicationCommandAutocomplete:
|
|
if command.Autocomplete != nil && i.ApplicationCommandData().Name == command.Definition.Name {
|
|
command.Autocomplete(s, i)
|
|
}
|
|
case discordgo.InteractionModalSubmit:
|
|
if command.ModalSubmit != nil && strings.HasPrefix(i.ModalSubmitData().CustomID, command.ModalID) {
|
|
command.ModalSubmit(s, i)
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
func removeOldCommandFromAllGuilds(s *discordgo.Session) {
|
|
for _, guild := range s.State.Guilds {
|
|
existingCommands, err := s.ApplicationCommands(s.State.User.ID, guild.ID)
|
|
if err != nil {
|
|
fmt.Printf("error fetching existing commands for guild %s: %v\n", guild.Name, err)
|
|
continue
|
|
}
|
|
var commandIDs []string
|
|
for _, command := range commands {
|
|
commandIDs = append(commandIDs, command.Definition.Name)
|
|
}
|
|
for _, existingCommand := range existingCommands {
|
|
if !slices.Contains(commandIDs, existingCommand.Name) {
|
|
fmt.Printf("\nDeleting command \"%s\"", existingCommand.Name)
|
|
err := s.ApplicationCommandDelete(s.State.User.ID, guild.ID, existingCommand.ID)
|
|
if err != nil {
|
|
fmt.Printf("error deleting command %s for guild %s: %v\n", existingCommand.Name, guild.Name, err)
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|