fixed commands in DMs and added global commands back

This commit is contained in:
2024-03-20 16:28:28 +01:00
parent 8774281d82
commit 71318359e3
6 changed files with 26 additions and 32 deletions

View File

@@ -29,4 +29,5 @@ var cmd_ask Command = Command{
logrus.Error("Failed to respond with embed: ", err) logrus.Error("Failed to respond with embed: ", err)
} }
}, },
AllowDM: true,
} }

View File

@@ -30,6 +30,7 @@ var cmd_cat Command = Command{
logrus.Error(err) logrus.Error(err)
} }
}, },
AllowDM: true,
} }
type CatImage struct { type CatImage struct {

View File

@@ -16,4 +16,5 @@ var cmd_dadjoke Command = Command{
logrus.Error(err) logrus.Error(err)
} }
}, },
AllowDM: true,
} }

View File

@@ -46,4 +46,5 @@ var cmd_ping Command = Command{
logrus.Error(err) logrus.Error(err)
} }
}, },
AllowDM: true,
} }

View File

@@ -19,6 +19,7 @@ type Command struct {
ModalIDs []string ModalIDs []string
DynamicComponentIDs func() []string DynamicComponentIDs func() []string
DynamicModalIDs func() []string DynamicModalIDs func() []string
AllowDM bool
} }
var commands []Command = []Command{cmd_form, cmd_tag, cmd_tag_short, cmd_dadjoke, cmd_ping, cmd_ask, cmd_sticky, cmd_cat, cmd_autojoinroles, cmd_autopublish, context_sticky, context_tag} var commands []Command = []Command{cmd_form, cmd_tag, cmd_tag_short, cmd_dadjoke, cmd_ping, cmd_ask, cmd_sticky, cmd_cat, cmd_autojoinroles, cmd_autopublish, context_sticky, context_tag}
@@ -28,40 +29,33 @@ func ready(s *discordgo.Session, event *discordgo.Ready) {
findAndDeleteUnusedMessages() findAndDeleteUnusedMessages()
removeOldCommandFromAllGuilds(s) removeOldCommandFromAllGuilds(s)
var existingCommandNames []string var existingCommandNames []string
for _, guild := range event.Guilds { existingCommands, err := s.ApplicationCommands(s.State.User.ID, "")
existingCommands, err := s.ApplicationCommands(s.State.User.ID, guild.ID) if err != nil {
logrus.Errorf("error fetching existing global commands: %v", err)
} else {
for _, existingCommand := range existingCommands { for _, existingCommand := range existingCommands {
existingCommandNames = append(existingCommandNames, existingCommand.Name) existingCommandNames = append(existingCommandNames, existingCommand.Name)
} }
if err != nil {
logrus.Errorf("error fetching existing commands for guild %s: %v", guild.Name, err)
continue
} }
for _, command := range commands { for _, command := range commands {
if !slices.Contains(existingCommandNames, command.Definition.Name) || slices.Contains(os.Args, "--update="+command.Definition.Name) { if !slices.Contains(existingCommandNames, command.Definition.Name) || slices.Contains(os.Args, "--update="+command.Definition.Name) || slices.Contains(os.Args, "--update=all") {
cmd, err := s.ApplicationCommandCreate(s.State.User.ID, guild.ID, &command.Definition) cmd, err := s.ApplicationCommandCreate(s.State.User.ID, "", &command.Definition)
logrus.Infof("Added command '%s'", cmd.Name)
if err != nil { if err != nil {
logrus.Error("error creating command,", err) logrus.Errorf("error creating global command '%s': %v", cmd.Name, err)
continue } else {
} logrus.Infof("Added global command '%s'", cmd.Name)
} }
} }
} }
logrus.Info("Successfully started the Bot!") logrus.Info("Successfully started the Bot!")
} }
func guildCreate(s *discordgo.Session, event *discordgo.GuildCreate) {
for _, command := range commands {
_, err := s.ApplicationCommandCreate(s.State.User.ID, event.Guild.ID, &command.Definition)
if err != nil {
logrus.Errorf("error creating command for guild %s: %v", event.Guild.Name, err)
}
}
}
func interactionCreate(s *discordgo.Session, i *discordgo.InteractionCreate) { func interactionCreate(s *discordgo.Session, i *discordgo.InteractionCreate) {
for _, command := range commands { for _, command := range commands {
if i.ApplicationCommandData().Name == command.Definition.Name && !command.AllowDM && i.Interaction.GuildID == "" {
respond(i.Interaction, "This interaction is not available in DMs.", true)
return
}
switch i.Type { switch i.Type {
case discordgo.InteractionApplicationCommand: case discordgo.InteractionApplicationCommand:
if command.Interact != nil && i.ApplicationCommandData().Name == command.Definition.Name { if command.Interact != nil && i.ApplicationCommandData().Name == command.Definition.Name {
@@ -103,12 +97,9 @@ func interactionCreate(s *discordgo.Session, i *discordgo.InteractionCreate) {
} }
func removeOldCommandFromAllGuilds(s *discordgo.Session) { func removeOldCommandFromAllGuilds(s *discordgo.Session) {
for _, guild := range s.State.Guilds { existingCommands, err := s.ApplicationCommands(s.State.User.ID, "")
existingCommands, err := s.ApplicationCommands(s.State.User.ID, guild.ID)
if err != nil { if err != nil {
logrus.Errorf("error fetching existing commands for guild %s: %v\n", guild.Name, err) logrus.Errorf("error fetching existing commands: %v\n", err)
continue
}
var commandIDs []string var commandIDs []string
for _, command := range commands { for _, command := range commands {
commandIDs = append(commandIDs, command.Definition.Name) commandIDs = append(commandIDs, command.Definition.Name)
@@ -116,9 +107,9 @@ func removeOldCommandFromAllGuilds(s *discordgo.Session) {
for _, existingCommand := range existingCommands { for _, existingCommand := range existingCommands {
if !slices.Contains(commandIDs, existingCommand.Name) { if !slices.Contains(commandIDs, existingCommand.Name) {
logrus.Infof("Deleting command '%s'", existingCommand.Name) logrus.Infof("Deleting command '%s'", existingCommand.Name)
err := s.ApplicationCommandDelete(s.State.User.ID, guild.ID, existingCommand.ID) err := s.ApplicationCommandDelete(s.State.User.ID, "", existingCommand.ID)
if err != nil { if err != nil {
logrus.Errorf("error deleting command %s for guild %s: %v", existingCommand.Name, guild.Name, err) logrus.Errorf("error deleting command %s: %v", existingCommand.Name, err)
} }
} }
} }

View File

@@ -42,7 +42,6 @@ func main() {
} }
bot.Identify.Intents = discordgo.IntentsGuildMessages | discordgo.IntentsGuilds | discordgo.IntentMessageContent | discordgo.IntentGuildMembers bot.Identify.Intents = discordgo.IntentsGuildMessages | discordgo.IntentsGuilds | discordgo.IntentMessageContent | discordgo.IntentGuildMembers
bot.AddHandler(ready) bot.AddHandler(ready)
bot.AddHandler(guildCreate)
bot.AddHandler(interactionCreate) bot.AddHandler(interactionCreate)
bot.AddHandler(messageCreate) bot.AddHandler(messageCreate)
bot.AddHandler(messageDelete) bot.AddHandler(messageDelete)