98 lines
2.5 KiB
Go
98 lines
2.5 KiB
Go
package command
|
|
|
|
import (
|
|
"fmt"
|
|
"himbot/lib"
|
|
"strings"
|
|
|
|
"github.com/bwmarrin/discordgo"
|
|
)
|
|
|
|
func BalanceGetCommand(s *discordgo.Session, i *discordgo.InteractionCreate) (string, error) {
|
|
user, err := lib.GetUser(i)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
balance, err := lib.GetBalance(user.ID, i.GuildID)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
return fmt.Sprintf("💸 You have %d Himbucks! 💸", balance), nil
|
|
}
|
|
|
|
func BalanceSyncCommand(s *discordgo.Session, i *discordgo.InteractionCreate) (string, error) {
|
|
syncError := lib.SyncBalance()
|
|
if syncError != nil {
|
|
return "", syncError
|
|
}
|
|
|
|
return fmt.Sprintf("💸 Force-Synchronized Himbucks! 💸"), nil
|
|
}
|
|
|
|
func LeaderboardCommand(s *discordgo.Session, i *discordgo.InteractionCreate) (string, error) {
|
|
entries, err := lib.GetLeaderboard(i.GuildID, 10)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
if len(entries) == 0 {
|
|
return "No himbucks earned yet!", nil
|
|
}
|
|
|
|
var sb strings.Builder
|
|
sb.WriteString("🏆 Himbucks Leaderboard 🏆\n\n")
|
|
for idx, entry := range entries {
|
|
sb.WriteString(fmt.Sprintf("%d. %s: %d himbucks\n", idx+1, entry.Username, entry.Balance))
|
|
}
|
|
|
|
return sb.String(), nil
|
|
}
|
|
|
|
func BalanceSendCommand(s *discordgo.Session, i *discordgo.InteractionCreate) (string, error) {
|
|
options := i.ApplicationCommandData().Options
|
|
|
|
// Discord handles the user mention/tag and provides the correct user ID
|
|
var recipientID string
|
|
var amount int
|
|
|
|
for _, opt := range options {
|
|
switch opt.Name {
|
|
case "user":
|
|
recipientID = opt.UserValue(nil).ID
|
|
case "amount":
|
|
amount = int(opt.IntValue())
|
|
}
|
|
}
|
|
|
|
// Validate amount
|
|
if amount <= 0 {
|
|
return "", fmt.Errorf("amount must be positive")
|
|
}
|
|
|
|
// Get sender's info
|
|
sender, err := lib.GetUser(i)
|
|
if err != nil {
|
|
return "", fmt.Errorf("failed to get sender info: %w", err)
|
|
}
|
|
|
|
// Don't allow sending to self
|
|
if sender.ID == recipientID {
|
|
return "", fmt.Errorf("you cannot send himbucks to yourself")
|
|
}
|
|
|
|
// Get recipient's info
|
|
recipient, err := s.User(recipientID)
|
|
if err != nil {
|
|
return "", fmt.Errorf("failed to get recipient info: %w", err)
|
|
}
|
|
|
|
// Send the himbucks
|
|
err = lib.SendBalance(sender.ID, recipientID, i.GuildID, amount)
|
|
if err != nil {
|
|
return "", fmt.Errorf("failed to send himbucks: %w", err)
|
|
}
|
|
|
|
return fmt.Sprintf("💸 Successfully sent %d Himbucks to %s! 💸", amount, recipient.Username), nil
|
|
}
|