package discord

import (
	"bytes"
	"fmt"
	"strings"

	"git.canopymc.net/Etzelia/stonebot/assets"
	"git.canopymc.net/Etzelia/stonebot/database"

	"github.com/bwmarrin/discordgo"
	"go.jolheiser.com/beaver"
)

func New(token, prefix string, db *database.Database) (*discordgo.Session, error) {
	bot, err := discordgo.New("Bot " + token)
	if err != nil {
		return nil, err
	}
	bot.Identify.Intents = discordgo.IntentsAllWithoutPrivileged

	bot.AddHandler(CommandHandler(prefix, db))
	bot.AddHandler(MessageHandler(prefix, db))

	return bot, nil
}

func CommandHandler(prefix string, db *database.Database) func(session *discordgo.Session, message *discordgo.MessageCreate) {
	return func(session *discordgo.Session, message *discordgo.MessageCreate) {
		// Ignore non-prefix
		if !strings.HasPrefix(message.Content, prefix) {
			return
		}

		// Ignore bots
		if message.Author.Bot {
			return
		}

		args := strings.Fields(message.Content[1:])
		switch strings.ToLower(args[0]) {
		case "rocks", "stones":
			rocks, err := db.Rocks()
			if err != nil {
				beaver.Errorf("could not count rocks: %v", err)
				return
			}
			if _, err := session.ChannelMessageSend(message.ChannelID, fmt.Sprintf("%d ROCKS", rocks)); err != nil {
				beaver.Errorf("could not send message: %v", err)
				return
			}
		}
	}
}

func MessageHandler(prefix string, db *database.Database) func(session *discordgo.Session, message *discordgo.MessageCreate) {
	return func(session *discordgo.Session, message *discordgo.MessageCreate) {
		// Ignore prefix
		if strings.HasPrefix(message.Content, prefix) {
			return
		}

		// Ignore bots
		if message.Author.Bot {
			return
		}

		rocks := strings.Count(message.Content, "rock")
		rocks += strings.Count(message.Content, "stone")
		if err := db.Rock(rocks); err != nil {
			beaver.Errorf("could not add rocks: %v", err)
		}

		if rocks > 0 {
			if _, err := session.ChannelMessageSendComplex(message.ChannelID, &discordgo.MessageSend{
				Files: []*discordgo.File{
					{
						Name:        "rocks.gif",
						ContentType: "multipart/form-data",
						Reader:      bytes.NewReader(assets.RocksGif),
					},
				},
			}); err != nil {
				beaver.Errorf("could not send rocks gif: %v", err)
			}
		}
	}
}