Add more context

This commit is contained in:
Daniel Mason 2025-01-16 00:32:32 +13:00
parent a8b2007f05
commit 464fa00efa
Signed by: idanoo
GPG key ID: 387387CDBC02F132
2 changed files with 125 additions and 38 deletions

View file

@ -1,6 +1,7 @@
package bot package bot
import ( import (
"fmt"
"log/slog" "log/slog"
"strings" "strings"
@ -20,34 +21,8 @@ var (
} }
commandHandlers = map[string]func(s *discordgo.Session, i *discordgo.InteractionCreate){ commandHandlers = map[string]func(s *discordgo.Session, i *discordgo.InteractionCreate){
"show-top-emojis": func(s *discordgo.Session, i *discordgo.InteractionCreate) { "show-top-emojis": showTopEmojis,
top, err := b.Db.GetTopEmojisForGuild(i.GuildID, 5) "show-top-users": showTopUsers,
if err != nil {
slog.Error("Error getting top emojis", "err", err)
return
}
s.InteractionRespond(i.Interaction, &discordgo.InteractionResponse{
Type: discordgo.InteractionResponseChannelMessageWithSource,
Data: &discordgo.InteractionResponseData{
Content: "Top Emojis:\n" + strings.Join(top, "\n"),
},
})
},
"show-top-users": func(s *discordgo.Session, i *discordgo.InteractionCreate) {
top, err := b.Db.GetTopUsersForGuild(i.GuildID, 5)
if err != nil {
slog.Error("Error getting top users", "err", err)
return
}
s.InteractionRespond(i.Interaction, &discordgo.InteractionResponse{
Type: discordgo.InteractionResponseChannelMessageWithSource,
Data: &discordgo.InteractionResponseData{
Content: "Top Users:\n" + strings.Join(top, "\n"),
},
})
},
} }
) )
@ -73,3 +48,67 @@ func (bot *Bot) DeregisterCommands() {
} }
} }
} }
// showTopEmojis - Show top emojis with users
func showTopEmojis(s *discordgo.Session, i *discordgo.InteractionCreate) {
top, err := b.Db.GetTopEmojisForGuild(i.GuildID, 5)
if err != nil {
slog.Error("Error getting top emojis", "err", err)
return
}
msg := "Most used emojis:\n"
for k, v := range top {
topUsers, err := b.Db.GetTopUsersForGuildEmoji(i.GuildID, k, 3)
if err != nil {
slog.Error("Error getting top users for guild emoji", "err", err)
continue
}
users := []string{}
msg += fmt.Sprintf("%s: %d", k, v)
for sk, sv := range topUsers {
users = append(users, fmt.Sprintf("<@%s>: %d", sk, sv))
}
msg += " (" + strings.Join(users, ", ") + ")\n"
}
s.InteractionRespond(i.Interaction, &discordgo.InteractionResponse{
Type: discordgo.InteractionResponseChannelMessageWithSource,
Data: &discordgo.InteractionResponseData{
Content: msg,
},
})
}
// showTopUsers - Show top users with emojis
func showTopUsers(s *discordgo.Session, i *discordgo.InteractionCreate) {
top, err := b.Db.GetTopUsersForGuild(i.GuildID, 5)
if err != nil {
slog.Error("Error getting top users", "err", err)
return
}
msg := "Users who use the most emojis:\n"
for k, v := range top {
topUsers, err := b.Db.GetTopEmojisForGuildUser(i.GuildID, k, 3)
if err != nil {
slog.Error("Error getting top emojis for guild user", "err", err)
continue
}
users := []string{}
msg += fmt.Sprintf("<@%s>: %d", k, v)
for sk, sv := range topUsers {
users = append(users, fmt.Sprintf("%s: %d", sk, sv))
}
msg += " (" + strings.Join(users, ", ") + ")\n"
}
s.InteractionRespond(i.Interaction, &discordgo.InteractionResponse{
Type: discordgo.InteractionResponseChannelMessageWithSource,
Data: &discordgo.InteractionResponseData{
Content: msg,
},
})
}

View file

@ -1,7 +1,5 @@
package db package db
import "fmt"
// LogEmojiUsage - Log usage // LogEmojiUsage - Log usage
func (db *Database) LogEmojiUsage(guildID, channelID, userID, emojiID string) error { func (db *Database) LogEmojiUsage(guildID, channelID, userID, emojiID string) error {
_, err := db.db.Exec( _, err := db.db.Exec(
@ -13,8 +11,8 @@ func (db *Database) LogEmojiUsage(guildID, channelID, userID, emojiID string) er
} }
// GetTopUsersForGuild - Report usage // GetTopUsersForGuild - Report usage
func (db *Database) GetTopUsersForGuild(guildID string, num int) ([]string, error) { func (db *Database) GetTopUsersForGuild(guildID string, num int) (map[string]int64, error) {
var data []string data := make(map[string]int64)
row, err := db.db.Query( row, err := db.db.Query(
"SELECT user_id, count(*) FROM `emoji_usage` WHERE `guild_id` = ? GROUP BY user_id ORDER BY count(*) DESC LIMIT ?", "SELECT user_id, count(*) FROM `emoji_usage` WHERE `guild_id` = ? GROUP BY user_id ORDER BY count(*) DESC LIMIT ?",
guildID, guildID,
@ -30,18 +28,19 @@ func (db *Database) GetTopUsersForGuild(guildID string, num int) ([]string, erro
var user string var user string
var count int64 var count int64
row.Scan(&user, &count) row.Scan(&user, &count)
data = append(data, fmt.Sprintf("<@%s>: %d", user, count)) data[user] = count
} }
return data, nil return data, nil
} }
// GetTopEmojisForGuild - Report usage // GetTopUsersForGuildEmoji - Report usage
func (db *Database) GetTopEmojisForGuild(guildID string, num int) ([]string, error) { func (db *Database) GetTopUsersForGuildEmoji(guildID string, emojiID string, num int) (map[string]int64, error) {
var data []string data := make(map[string]int64)
row, err := db.db.Query( row, err := db.db.Query(
"SELECT emoji_id, count(*) FROM `emoji_usage` WHERE `guild_id` = ? GROUP BY emoji_id ORDER BY count(*) DESC LIMIT ?", "SELECT user_id, count(*) FROM `emoji_usage` WHERE `guild_id` = ? AND `emoji_id` = ? GROUP BY emoji_id, user_id ORDER BY count(*) DESC LIMIT ?",
guildID, guildID,
emojiID,
num, num,
) )
@ -54,7 +53,56 @@ func (db *Database) GetTopEmojisForGuild(guildID string, num int) ([]string, err
var user string var user string
var count int64 var count int64
row.Scan(&user, &count) row.Scan(&user, &count)
data = append(data, fmt.Sprintf("%s: %d", user, count)) data[user] = count
}
return data, nil
}
// GetTopEmojisForGuild - Report usage
func (db *Database) GetTopEmojisForGuild(guildID string, num int) (map[string]int64, error) {
data := make(map[string]int64)
row, err := db.db.Query(
"SELECT emoji_id, count(*) FROM `emoji_usage` WHERE `guild_id` = ? GROUP BY emoji_id ORDER BY count(*) DESC LIMIT ?",
guildID,
num,
)
if err != nil {
return data, err
}
defer row.Close()
for row.Next() {
var emoji string
var count int64
row.Scan(&emoji, &count)
data[emoji] = count
}
return data, nil
}
// GetTopEmojisForGuildUser - Report usage
func (db *Database) GetTopEmojisForGuildUser(guildID string, userID string, num int) (map[string]int64, error) {
data := make(map[string]int64)
row, err := db.db.Query(
"SELECT emoji_id, count(*) FROM `emoji_usage` WHERE `guild_id` = ? AND `user_id` = ? GROUP BY emoji_id ORDER BY count(*) DESC LIMIT ?",
guildID,
userID,
num,
)
if err != nil {
return data, err
}
defer row.Close()
for row.Next() {
var emoji string
var count int64
row.Scan(&emoji, &count)
data[emoji] = count
} }
return data, nil return data, nil