This commit is contained in:
Daniel Mason 2025-02-11 21:22:37 +13:00
parent 1da9d79d70
commit 5a5071cb44
Signed by: idanoo
GPG key ID: 387387CDBC02F132
3 changed files with 30 additions and 11 deletions

View file

@ -93,15 +93,15 @@ func showTopEmojis(s *discordgo.Session, i *discordgo.InteractionCreate) {
} }
msg := "Most used emojis:\n" msg := "Most used emojis:\n"
for k, v := range top { for _, v := range top {
topUsers, err := b.Db.GetTopUsersForGuildEmoji(i.GuildID, k, 3) topUsers, err := b.Db.GetTopUsersForGuildEmoji(i.GuildID, v.EmojiID, 3)
if err != nil { if err != nil {
slog.Error("Error getting top users for guild emoji", "err", err) slog.Error("Error getting top users for guild emoji", "err", err)
continue continue
} }
users := []string{} users := []string{}
msg += fmt.Sprintf("%s: %d", k, v) msg += fmt.Sprintf("%s: %d", v.EmojiID, v.Count)
for sk, sv := range topUsers { for sk, sv := range topUsers {
users = append(users, fmt.Sprintf("<@%s>: %d", sk, sv)) users = append(users, fmt.Sprintf("<@%s>: %d", sk, sv))
} }
@ -147,8 +147,8 @@ func showTopUsers(s *discordgo.Session, i *discordgo.InteractionCreate) {
users := []string{} users := []string{}
msg += fmt.Sprintf("<@%s>: %d", k, v) msg += fmt.Sprintf("<@%s>: %d", k, v)
for sk, sv := range topUsers { for _, sv := range topUsers {
users = append(users, fmt.Sprintf("%s: %d", sk, sv)) users = append(users, fmt.Sprintf("%s: %d", sv.EmojiID, sv.Count))
} }
msg += " (" + strings.Join(users, ", ") + ")\n" msg += " (" + strings.Join(users, ", ") + ")\n"
} }

View file

@ -0,0 +1,10 @@
package bot
func orderMap(m map[string]int64) []string {
keys := make([]string, 0, len(m))
for k := range m {
keys = append(keys, k)
}
return keys
}

View file

@ -1,5 +1,10 @@
package db package db
type EmojiMap struct {
EmojiID string
Count int64
}
// 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(
@ -60,8 +65,8 @@ func (db *Database) GetTopUsersForGuildEmoji(guildID string, emojiID string, num
} }
// GetTopEmojisForGuild - Report usage // GetTopEmojisForGuild - Report usage
func (db *Database) GetTopEmojisForGuild(guildID string, num int64) (map[string]int64, error) { func (db *Database) GetTopEmojisForGuild(guildID string, num int64) (map[int]EmojiMap, error) {
data := make(map[string]int64) data := make(map[int]EmojiMap)
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 emoji_id, count(*) FROM `emoji_usage` WHERE `guild_id` = ? GROUP BY emoji_id ORDER BY count(*) DESC LIMIT ?",
guildID, guildID,
@ -73,19 +78,21 @@ func (db *Database) GetTopEmojisForGuild(guildID string, num int64) (map[string]
} }
defer row.Close() defer row.Close()
i := 0
for row.Next() { for row.Next() {
var emoji string var emoji string
var count int64 var count int64
row.Scan(&emoji, &count) row.Scan(&emoji, &count)
data[emoji] = count data[i] = EmojiMap{EmojiID: emoji, Count: count}
i++
} }
return data, nil return data, nil
} }
// GetTopEmojisForGuildUser - Report usage // GetTopEmojisForGuildUser - Report usage
func (db *Database) GetTopEmojisForGuildUser(guildID string, userID string, num int) (map[string]int64, error) { func (db *Database) GetTopEmojisForGuildUser(guildID string, userID string, num int) (map[int]EmojiMap, error) {
data := make(map[string]int64) data := make(map[int]EmojiMap)
row, err := db.db.Query( 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 ?", "SELECT emoji_id, count(*) FROM `emoji_usage` WHERE `guild_id` = ? AND `user_id` = ? GROUP BY emoji_id ORDER BY count(*) DESC LIMIT ?",
guildID, guildID,
@ -98,11 +105,13 @@ func (db *Database) GetTopEmojisForGuildUser(guildID string, userID string, num
} }
defer row.Close() defer row.Close()
i := 0
for row.Next() { for row.Next() {
var emoji string var emoji string
var count int64 var count int64
row.Scan(&emoji, &count) row.Scan(&emoji, &count)
data[emoji] = count data[i] = EmojiMap{EmojiID: emoji, Count: count}
i++
} }
return data, nil return data, nil