finished plugin system base

2024-04-15 18:17:12 +02:00
parent 31e96c80f4
commit 077054109f
51 changed files with 2218 additions and 2160 deletions
--- a/plugin_src/autojoinroles/main.go
+++ b/plugin_src/autojoinroles/main.go
@@ -0,0 +1,127 @@
+package main
+
+import (
+	"database/sql"
+
+	"github.com/disgoorg/disgo/bot"
+	"github.com/disgoorg/disgo/discord"
+	"github.com/disgoorg/disgo/events"
+	"github.com/disgoorg/json"
+	"github.com/disgoorg/snowflake/v2"
+	"github.com/sirupsen/logrus"
+	"github.com/vaporvee/acecore/shared"
+)
+
+var db *sql.DB
+
+var dbCreateQuery string = ` 
+CREATE TABLE IF NOT EXISTS autojoinroles (
+	guild_id TEXT NOT NULL,
+	bot_role TEXT,
+	user_role TEXT,
+	PRIMARY KEY (guild_id)
+);
+`
+
+var Plugin = &shared.Plugin{
+	Name: "Auto Join Roles",
+	Init: func(d *sql.DB) error {
+		db = d
+		_, err := d.Exec(dbCreateQuery)
+		if err != nil {
+			return err
+		}
+		shared.BotConfigs = append(shared.BotConfigs, bot.WithEventListenerFunc(guildMemberJoin))
+		return nil
+	},
+	Commands: []shared.Command{
+		{
+			Definition: discord.SlashCommandCreate{
+				Name:                     "autojoinroles",
+				Description:              "Give users a role when they join",
+				DefaultMemberPermissions: json.NewNullablePtr(discord.PermissionManageRoles),
+				Contexts: []discord.InteractionContextType{
+					discord.InteractionContextTypeGuild,
+					discord.InteractionContextTypePrivateChannel},
+				IntegrationTypes: []discord.ApplicationIntegrationType{
+					discord.ApplicationIntegrationTypeGuildInstall},
+				Options: []discord.ApplicationCommandOption{
+					&discord.ApplicationCommandOptionSubCommand{
+						Name:        "bot",
+						Description: "Give bots a role when they join (Leave empty to remove current)",
+						Options: []discord.ApplicationCommandOption{
+							&discord.ApplicationCommandOptionRole{
+								Name:        "role",
+								Description: "The role bots should get when they join the server",
+							},
+						},
+					},
+					&discord.ApplicationCommandOptionSubCommand{
+						Name:        "user",
+						Description: "Give users a role when they join (Leave empty to remove current)",
+						Options: []discord.ApplicationCommandOption{
+							&discord.ApplicationCommandOptionRole{
+								Name:        "role",
+								Description: "The role users should get when they join the server",
+							}},
+					},
+				},
+			},
+			Interact: func(e *events.ApplicationCommandInteractionCreate) {
+				var role string
+				option := *e.SlashCommandInteractionData().SubCommandName
+				var content string
+				if len(e.SlashCommandInteractionData().Options) == 1 {
+					var givenRole discord.Role = e.SlashCommandInteractionData().Role("role")
+					role = givenRole.ID.String()
+					botrole, err := getHighestRole(e.GuildID().String(), e.Client())
+					if err != nil {
+						logrus.Error(err)
+					}
+					if givenRole.Position >= botrole.Position {
+						content = "<@&" + role + "> is not below the Bot's current highest role(<@&" + botrole.ID.String() + ">). That makes it unable to manage it."
+					} else {
+						if setAutoJoinRole(e.GuildID().String(), option, role) {
+							content = "Updated auto join role for " + option + "s as <@&" + role + ">"
+						} else {
+							content = "Setup auto join role for " + option + "s as <@&" + role + ">"
+						}
+					}
+				} else if setAutoJoinRole(e.GuildID().String(), option, role) {
+					content = "Deleted auto join role for " + option + "s"
+				}
+				if content == "" {
+					content = "No auto join role set for " + option + "s to delete."
+				}
+				err := e.CreateMessage(discord.NewMessageCreateBuilder().SetContent(content).SetEphemeral(true).Build())
+				if err != nil {
+					logrus.Error(err)
+				}
+				purgeUnusedAutoJoinRoles(e.GuildID().String())
+			},
+		},
+	},
+}
+
+func getHighestRole(guildID string, c bot.Client) (*discord.Role, error) {
+	botmember, err := c.Rest().GetMember(snowflake.MustParse(guildID), c.ApplicationID())
+	if err != nil {
+		return nil, err
+	}
+	roles, err := c.Rest().GetRoles(snowflake.MustParse(guildID))
+	if err != nil {
+		return nil, err
+	}
+	var highestRole *discord.Role
+	for _, roleID := range botmember.RoleIDs {
+		for _, role := range roles {
+			if role.ID == roleID {
+				if highestRole == nil || role.Position > highestRole.Position {
+					highestRole = &role
+				}
+				break
+			}
+		}
+	}
+	return highestRole, nil
+}