vaporvee/acecore
1
1
Fork 0
Code Issues Packages Projects Releases Wiki Activity

finished plugin system base

Browse Source
This commit is contained in:
Yannik
2024-04-15 18:17:12 +02:00
parent 31e96c80f4
commit 077054109f
51 changed files with 2218 additions and 2160 deletions
Download Patch File Download Diff File Expand all files Collapse all files

62
plugin_src/stickymessages/data.go Normal file
View File

@@ -0,0 +1,62 @@
package main
import (
"github.com/sirupsen/logrus"
)
func addSticky(guildID string, channelID string, messageContent string, messageID string) {
_, err := db.Exec("INSERT INTO sticky (guild_id, channel_id, message_id, message_content) VALUES ($1, $2, $3, $4)", guildID, channelID, messageID, messageContent)
if err != nil {
logrus.Error(err)
}
}
func hasSticky(guildID string, channelID string) bool {
var exists bool
err := db.QueryRow("SELECT EXISTS (SELECT 1 FROM sticky WHERE guild_id = $1 AND channel_id = $2)", guildID, channelID).Scan(&exists)
if err != nil {
logrus.Error(err)
}
return exists
}
func getStickyMessageID(guildID string, channelID string) string {
var messageID string
exists := hasSticky(guildID, channelID)
if exists {
err := db.QueryRow("SELECT message_id FROM sticky WHERE guild_id = $1 AND channel_id = $2", guildID, channelID).Scan(&messageID)
if err != nil {
logrus.Error(err)
}
}
return messageID
}
func getStickyMessageContent(guildID string, channelID string) string {
var messageID string
exists := hasSticky(guildID, channelID)
if exists {
err := db.QueryRow("SELECT message_content FROM sticky WHERE guild_id = $1 AND channel_id = $2", guildID, channelID).Scan(&messageID)
if err != nil {
logrus.Error(err)
}
}
return messageID
}
func updateStickyMessageID(guildID string, channelID string, messageID string) {
exists := hasSticky(guildID, channelID)
if exists {
_, err := db.Exec("UPDATE sticky SET message_id = $1 WHERE guild_id = $2 AND channel_id = $3", messageID, guildID, channelID)
if err != nil {
logrus.Error(err)
}
}
}
func removeSticky(guildID string, channelID string) {
_, err := db.Exec("DELETE FROM sticky WHERE guild_id = $1 AND channel_id = $2", guildID, channelID)
if err != nil {
logrus.Error(err)
}
}

33
plugin_src/stickymessages/listener.go Normal file
View File

@@ -0,0 +1,33 @@
package main
import (
"github.com/disgoorg/disgo/discord"
"github.com/disgoorg/disgo/events"
"github.com/disgoorg/snowflake/v2"
"github.com/sirupsen/logrus"
"github.com/vaporvee/acecore/custom"
)
func messageCreate(e *events.MessageCreate) {
if len(e.Message.Embeds) == 0 || e.Message.Embeds[0].Footer == nil || e.Message.Embeds[0].Footer.Text != "📌 Sticky message" {
if hasSticky(e.Message.GuildID.String(), e.Message.ChannelID.String()) {
stickymessageID := getStickyMessageID(e.Message.GuildID.String(), e.Message.ChannelID.String())
err := e.Client().Rest().DeleteMessage(e.ChannelID, snowflake.MustParse(stickymessageID))
stickyMessage, _ := e.Client().Rest().CreateMessage(e.ChannelID, discord.MessageCreate{
Embeds: []discord.Embed{
{
Footer: &discord.EmbedFooter{
Text: "📌 Sticky message",
},
Color: custom.GetColor("primary"),
Description: getStickyMessageContent(e.Message.GuildID.String(), e.Message.ChannelID.String()),
},
},
})
if err != nil {
logrus.Error(err)
}
updateStickyMessageID(e.Message.GuildID.String(), e.Message.ChannelID.String(), stickyMessage.ID.String())
}
}
}

145
plugin_src/stickymessages/main.go Normal file
View File

@@ -0,0 +1,145 @@
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/custom"
"github.com/vaporvee/acecore/shared"
)
var db *sql.DB
var dbCreateQuery string = `CREATE TABLE IF NOT EXISTS sticky (
message_id TEXT NOT NULL,
channel_id TEXT NOT NULL,
message_content TEXT NOT NULL,
guild_id TEXT NOT NULL,
PRIMARY KEY (channel_id, guild_id)
);
`
var Plugin = &shared.Plugin{
Name: "Sticky Messages",
Init: func(d *sql.DB) error {
db = d
_, err := d.Exec(dbCreateQuery)
if err != nil {
return err
}
shared.BotConfigs = append(shared.BotConfigs, bot.WithEventListenerFunc(messageCreate))
return nil
},
Commands: []shared.Command{
{
Definition: discord.SlashCommandCreate{
Name: "sticky",
Description: "Stick or unstick messages to the bottom of the current channel",
DefaultMemberPermissions: json.NewNullablePtr(discord.PermissionManageMessages),
Contexts: []discord.InteractionContextType{
discord.InteractionContextTypeGuild,
discord.InteractionContextTypePrivateChannel},
IntegrationTypes: []discord.ApplicationIntegrationType{
discord.ApplicationIntegrationTypeGuildInstall},
Options: []discord.ApplicationCommandOption{
&discord.ApplicationCommandOptionString{
Name: "message",
Description: "The message you want to stick to the bottom of this channel",
Required: false,
},
},
},
Interact: func(e *events.ApplicationCommandInteractionCreate) {
if len(e.SlashCommandInteractionData().Options) == 0 {
if hasSticky(e.GuildID().String(), e.Channel().ID().String()) {
err := e.Client().Rest().DeleteMessage(e.Channel().ID(), snowflake.MustParse(getStickyMessageID(e.GuildID().String(), e.Channel().ID().String())))
if err != nil {
logrus.Error(err)
}
removeSticky(e.GuildID().String(), e.Channel().ID().String())
err = e.CreateMessage(discord.NewMessageCreateBuilder().
SetContent("The sticky message was removed from this channel!").SetEphemeral(true).
Build())
if err != nil {
logrus.Error(err)
}
} else {
err := e.CreateMessage(discord.NewMessageCreateBuilder().
SetContent("This channel has no sticky message!").SetEphemeral(true).
Build())
if err != nil {
logrus.Error(err)
}
}
} else {
inputStickyMessage(e)
}
},
},
{
Definition: discord.MessageCommandCreate{
Name: "Stick to channel",
DefaultMemberPermissions: json.NewNullablePtr(discord.PermissionManageMessages),
Contexts: []discord.InteractionContextType{
discord.InteractionContextTypeGuild,
discord.InteractionContextTypePrivateChannel},
IntegrationTypes: []discord.ApplicationIntegrationType{
discord.ApplicationIntegrationTypeGuildInstall},
},
Interact: func(e *events.ApplicationCommandInteractionCreate) {
inputStickyMessage(e)
},
},
},
}
func inputStickyMessage(e *events.ApplicationCommandInteractionCreate) {
var messageText string
if e.ApplicationCommandInteraction.Data.Type() == discord.ApplicationCommandTypeMessage {
messageText = e.MessageCommandInteractionData().TargetMessage().Content //TODO add more data then just content
} else {
messageText = e.SlashCommandInteractionData().String("message")
}
if messageText == "" {
err := e.CreateMessage(discord.NewMessageCreateBuilder().
SetContent("Can't add empty sticky messages!").SetEphemeral(true).
Build())
if err != nil {
logrus.Error(err)
}
} else {
message, err := e.Client().Rest().CreateMessage(e.Channel().ID(), discord.MessageCreate{Embeds: []discord.Embed{
{Description: messageText, Footer: &discord.EmbedFooter{Text: "📌 Sticky message"}, Color: custom.GetColor("primary")}}})
if err != nil {
logrus.Error(err)
}
if hasSticky(e.GuildID().String(), e.Channel().ID().String()) {
err = e.Client().Rest().DeleteMessage(e.Channel().ID(), snowflake.MustParse(getStickyMessageID(e.GuildID().String(), e.Channel().ID().String())))
if err != nil {
logrus.Error(err, getStickyMessageID(e.GuildID().String(), e.Channel().ID().String()))
}
removeSticky(e.GuildID().String(), e.Channel().ID().String())
addSticky(e.GuildID().String(), e.Channel().ID().String(), messageText, message.ID.String())
err = e.CreateMessage(discord.NewMessageCreateBuilder().
SetContent("Sticky message in this channel was updated!").SetEphemeral(true).
Build())
if err != nil {
logrus.Error(err)
}
} else {
addSticky(e.GuildID().String(), e.Channel().ID().String(), messageText, message.ID.String())
err := e.CreateMessage(discord.NewMessageCreateBuilder().
SetContent("Message sticked to the channel!").SetEphemeral(true).
Build())
if err != nil {
logrus.Error(err)
}
}
}
}