finished plugin system base
This commit is contained in:
60
plugin_src/autojoinroles/data.go
Normal file
60
plugin_src/autojoinroles/data.go
Normal file
@@ -0,0 +1,60 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"github.com/sirupsen/logrus"
|
||||
)
|
||||
|
||||
func setAutoJoinRole(guildID string, option string, roleID string) bool {
|
||||
var role_exists bool
|
||||
var autojoinroles_exists bool
|
||||
err := db.QueryRow("SELECT EXISTS (SELECT 1 FROM autojoinroles WHERE guild_id = $1)", guildID).Scan(&autojoinroles_exists)
|
||||
if err != nil {
|
||||
logrus.Error(err)
|
||||
}
|
||||
err = db.QueryRow("SELECT EXISTS (SELECT 1 FROM autojoinroles WHERE guild_id = $1 AND "+option+"_role IS NOT NULL AND "+option+"_role != '')", guildID).Scan(&role_exists)
|
||||
if err != nil {
|
||||
logrus.Error(err)
|
||||
}
|
||||
if autojoinroles_exists {
|
||||
_, err = db.Exec("UPDATE autojoinroles SET "+option+"_role = $1 WHERE guild_id = $2", roleID, guildID)
|
||||
if err != nil {
|
||||
logrus.Error(err)
|
||||
}
|
||||
} else {
|
||||
_, err = db.Exec("INSERT INTO autojoinroles (guild_id, "+option+"_role) VALUES ($1, $2)", guildID, roleID)
|
||||
if err != nil {
|
||||
logrus.Error(err)
|
||||
}
|
||||
}
|
||||
return role_exists
|
||||
}
|
||||
|
||||
func purgeUnusedAutoJoinRoles(guildID string) {
|
||||
_, err := db.Exec("DELETE FROM autojoinroles WHERE guild_id = $1 AND user_role = '' OR user_role IS NULL AND bot_role = '' OR bot_role IS NULL", guildID)
|
||||
if err != nil {
|
||||
logrus.Error(err)
|
||||
}
|
||||
}
|
||||
|
||||
func getAutoJoinRole(guildID string, isBot bool) string {
|
||||
var isBotString string
|
||||
var role string
|
||||
if isBot {
|
||||
isBotString = "bot"
|
||||
} else {
|
||||
isBotString = "user"
|
||||
}
|
||||
var exists bool
|
||||
err := db.QueryRow("SELECT EXISTS (SELECT 1 FROM autojoinroles WHERE guild_id = $1)", guildID).Scan(&exists)
|
||||
if err != nil {
|
||||
logrus.Error(err)
|
||||
return role
|
||||
}
|
||||
if exists {
|
||||
err = db.QueryRow("SELECT "+isBotString+"_role FROM autojoinroles WHERE guild_id = $1", guildID).Scan(&role)
|
||||
if err != nil {
|
||||
logrus.Error(err, guildID)
|
||||
}
|
||||
}
|
||||
return role
|
||||
}
|
||||
17
plugin_src/autojoinroles/listener.go
Normal file
17
plugin_src/autojoinroles/listener.go
Normal file
@@ -0,0 +1,17 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"github.com/disgoorg/disgo/events"
|
||||
"github.com/disgoorg/snowflake/v2"
|
||||
"github.com/sirupsen/logrus"
|
||||
)
|
||||
|
||||
func guildMemberJoin(e *events.GuildMemberJoin) {
|
||||
role := getAutoJoinRole(e.GuildID.String(), e.Member.User.Bot)
|
||||
if role != "" {
|
||||
err := e.Client().Rest().AddMemberRole(e.GuildID, e.Member.User.ID, snowflake.MustParse(role))
|
||||
if err != nil {
|
||||
logrus.Error(err)
|
||||
}
|
||||
}
|
||||
}
|
||||
127
plugin_src/autojoinroles/main.go
Normal file
127
plugin_src/autojoinroles/main.go
Normal file
@@ -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
|
||||
}
|
||||
Reference in New Issue
Block a user