mirror of
https://github.com/idanoo/GoDiscMoji
synced 2025-07-02 02:42:15 +00:00
Add more context
This commit is contained in:
parent
a8b2007f05
commit
464fa00efa
2 changed files with 125 additions and 38 deletions
|
@ -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,
|
||||||
|
},
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
|
@ -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
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue