himbot/command/himbucks.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 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
}