added autojoinroles without permissioncheck

2024-03-04 14:37:14 +01:00
parent eb6a2dbc5c
commit 031d8ee81d
5 changed files with 69 additions and 33 deletions
--- a/handlers.go
+++ b/handlers.go
@@ -0,0 +1,133 @@
+package main
+
+import (
+	"fmt"
+	"log"
+	"os"
+	"slices"
+	"strings"
+
+	"github.com/bwmarrin/discordgo"
+)
+
+type Command struct {
+	Definition        discordgo.ApplicationCommand
+	Interact          func(s *discordgo.Session, i *discordgo.InteractionCreate)
+	ComponentInteract func(s *discordgo.Session, i *discordgo.InteractionCreate)
+	ComponentIDs      []string
+	Autocomplete      func(s *discordgo.Session, i *discordgo.InteractionCreate)
+	ModalSubmit       func(s *discordgo.Session, i *discordgo.InteractionCreate)
+	ModalIDs          []string
+}
+
+var commands []Command = []Command{form_command, tag_command, short_get_tag_command, dadjoke_command, ping_command, ask_command, sticky_command, cat_command, autojoinroles_command}
+
+func ready(s *discordgo.Session, event *discordgo.Ready) {
+	fmt.Print("\nStarting up...")
+	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="+command.Definition.Name) {
+				cmd, err := s.ApplicationCommandCreate(s.State.User.ID, guild.ID, &command.Definition)
+				fmt.Printf("\nAdded command \"%s\"", cmd.Name)
+				if err != nil {
+					fmt.Println("error creating command,", err)
+					continue
+				}
+			}
+		}
+	}
+	fmt.Print("\nSuccessfully started the Bot!")
+}
+
+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:
+			var hasID bool = false
+			if command.ModalSubmit != nil {
+				for _, modalID := range command.ModalIDs {
+					if strings.HasPrefix(i.ModalSubmitData().CustomID, modalID) {
+						hasID = true
+					}
+				}
+				if hasID {
+					command.ModalSubmit(s, i)
+				}
+			}
+		case discordgo.InteractionMessageComponent:
+			if command.ComponentInteract != nil {
+				if command.Definition.Name == "form" {
+					command.ComponentIDs = getFormButtonIDs()
+				} // FIXME: Makes it dynamic i don't know why it isn't otherwise
+				if slices.Contains(command.ComponentIDs, i.MessageComponentData().CustomID) {
+					command.ComponentInteract(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)
+				}
+			}
+		}
+	}
+}
+
+func messageCreate(s *discordgo.Session, m *discordgo.MessageCreate) {
+	if len(m.Embeds) == 0 || m.Embeds[0].Footer == nil || m.Embeds[0].Footer.Text != "📌 Sticky message" {
+		if hasSticky(m.GuildID, m.ChannelID) {
+			s.ChannelMessageDelete(m.ChannelID, getStickyMessageID(m.GuildID, m.ChannelID))
+			stickyMessage, _ := s.ChannelMessageSendEmbed(m.ChannelID, &discordgo.MessageEmbed{
+				Type: discordgo.EmbedTypeArticle,
+				Footer: &discordgo.MessageEmbedFooter{
+					Text: "📌 Sticky message",
+				},
+				Color:       hexToDecimal(color["primary"]),
+				Description: getStickyMessageContent(m.GuildID, m.ChannelID),
+			})
+			updateStickyMessageID(m.GuildID, m.ChannelID, stickyMessage.ID)
+		}
+	}
+}
+
+func guildMemberJoin(s *discordgo.Session, m *discordgo.GuildMemberAdd) {
+	err := s.GuildMemberRoleAdd(m.GuildID, m.User.ID, getAutoJoinRole(m.GuildID, m.User.Bot))
+	if err != nil {
+		log.Println(err)
+	}
+}