idanoo/GoDiscMoji
1
0
Fork 0
mirror of https://github.com/idanoo/GoDiscMoji synced 2025-07-05 04:02:14 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Handle removing emojis

Browse source
This commit is contained in:
Daniel Mason 2025-02-12 17:02:42 +13:00
parent 08d83ef87f
commit 9103c4710c
Signed by: idanoo
GPG key ID: 387387CDBC02F132
3 changed files with 56 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

29
src/internal/bot/main.go
View file

@ -54,7 +54,9 @@ func (bot *Bot) Start() error {
})
// Add handlers
bot.DiscordSession.AddHandler(bot.HandleReaction)
bot.DiscordSession.AddHandler(bot.HandleAddReaction)
bot.DiscordSession.AddHandler(bot.HandleRemoveReaction)
bot.DiscordSession.AddHandler(bot.HandleRemoveAllReaction)
// Load session
err = discord.Open()
@ -80,14 +82,35 @@ func (bot *Bot) Start() error {
}
// HandleReaction - Simply log it
func (bot *Bot) HandleReaction(discord *discordgo.Session, reaction *discordgo.MessageReactionAdd) {
func (bot *Bot) HandleAddReaction(discord *discordgo.Session, reaction *discordgo.MessageReactionAdd) {
// Ignore Dyno user
if reaction.UserID == dynoUserID {
return
}
err := bot.Db.LogEmojiUsage(reaction.GuildID, reaction.ChannelID, reaction.UserID, reaction.Emoji.Name)
err := bot.Db.LogEmojiUsage(reaction.GuildID, reaction.ChannelID, reaction.MessageID, reaction.UserID, reaction.Emoji.Name)
if err != nil {
slog.Error("Failed to log emoji usage", "err", err)
}
}
// HandleRemoveReaction - Remove for user/message/emoji
func (bot *Bot) HandleRemoveReaction(discord *discordgo.Session, reaction *discordgo.MessageReactionRemove) {
// Ignore Dyno user
if reaction.UserID == dynoUserID {
return
}
err := bot.Db.DeleteEmojiUsage(reaction.GuildID, reaction.ChannelID, reaction.MessageID, reaction.UserID, reaction.Emoji.Name)
if err != nil {
slog.Error("Failed to delete single emoji usage", "err", err)
}
}
// HandleRemoveAllReaction - Remove all for message
func (bot *Bot) HandleRemoveAllReaction(discord *discordgo.Session, reaction *discordgo.MessageReactionRemoveAll) {
err := bot.Db.DeleteEmojiAll(reaction.GuildID, reaction.ChannelID, reaction.MessageID)
if err != nil {
slog.Error("Failed to delete all emoji usage for message", "err", err)
}
}

8
src/internal/db/database.go
View file

@ -41,9 +41,10 @@ func (db *Database) runMigrations() (*Database, error) {
"`id` INTEGER PRIMARY KEY AUTOINCREMENT, " +
"`guild_id` TEXT, " +
"`channel_id` TEXT, " +
"`message_id` TEXT, " +
"`user_id` TEXT, " +
"`emoji_id` TEXT, " +
"`timestamp` DATETIME, `viewed` INT DEFAULT 0" +
"`timestamp` DATETIME" +
")")
if err != nil {
return db, err
@ -54,6 +55,11 @@ func (db *Database) runMigrations() (*Database, error) {
return db, err
}
_, err = db.db.Exec("CREATE INDEX IF NOT EXISTS `idx_emoji_usage_message_id_user_id_emoji_id` ON `emoji_usage` (`message_id`, `user_id`, `guild_id`, `emoji_id`)")
if err != nil {
return db, err
}
_, err = db.db.Exec("CREATE INDEX IF NOT EXISTS `idx_emoji_usage_emoji_id,guild_id` ON `emoji_usage` (`guild_id`, `emoji_id`)")
return db, err
}

26
src/internal/db/emoji_usage.go
View file

@ -6,10 +6,30 @@ type EmojiMap struct {
}
// LogEmojiUsage - Log usage
func (db *Database) LogEmojiUsage(guildID, channelID, userID, emojiID string) error {
func (db *Database) LogEmojiUsage(guildID, channelID, messageID, userID, emojiID string) error {
_, err := db.db.Exec(
"INSERT INTO `emoji_usage` (`guild_id`, `channel_id`, `user_id`, `emoji_id`, `timestamp`) VALUES (?,?,?,?, datetime())",
guildID, channelID, userID, emojiID,
"INSERT INTO `emoji_usage` (`guild_id`, `channel_id`, `message_id`, `user_id`, `emoji_id`, `timestamp`) VALUES (?,?,?,?,?, datetime())",
guildID, channelID, messageID, userID, emojiID,
)
return err
}
// DeleteEmojiUsage - Delete for guild/channel/message/user
func (db *Database) DeleteEmojiUsage(guildID, channelID, messageID, userID, emojiID string) error {
_, err := db.db.Exec(
"DELETE FROM `emoji_usage` WHERE `guild_id` = ? AND `channel_id` = ? AND `message_id` = ? AND `user_id` = ? AND `emoji_id` = ?",
guildID, channelID, messageID, userID, emojiID,
)
return err
}
// DeleteEmojiAll - Delete for whole message
func (db *Database) DeleteEmojiAll(guildID, channelID, messageID string) error {
_, err := db.db.Exec(
"DELETE FROM `emoji_usage` WHERE `guild_id` = ? AND `channel_id` = ? AND `message_id` = ?",
guildID, channelID, messageID,
)
return err