package main

import (
	"fmt"
	"log"
	"os"
	"slices"
	"strings"

	"github.com/bwmarrin/discordgo"
)

type Command struct {
	Definition        discordgo.ApplicationCommand
	Interact          func(s *discordgo.Session, i *discordgo.InteractionCreate)
	ComponentInteract func(s *discordgo.Session, i *discordgo.InteractionCreate)
	ComponentIDs      []string
	Autocomplete      func(s *discordgo.Session, i *discordgo.InteractionCreate)
	ModalSubmit       func(s *discordgo.Session, i *discordgo.InteractionCreate)
	ModalIDs          []string
}

var commands []Command = []Command{form_command, tag_command, short_get_tag_command, dadjoke_command, ping_command, ask_command, sticky_command, cat_command, autojoinroles_command}

func ready(s *discordgo.Session, event *discordgo.Ready) {
	fmt.Print("\nStarting up...")
	removeOldCommandFromAllGuilds(s)
	var existingCommandNames []string
	for _, guild := range event.Guilds {
		existingCommands, err := s.ApplicationCommands(s.State.User.ID, guild.ID)
		for _, existingCommand := range existingCommands {
			existingCommandNames = append(existingCommandNames, existingCommand.Name)
		}
		if err != nil {
			fmt.Printf("error fetching existing commands for guild %s: %v\n", guild.Name, err)
			continue
		}
		for _, command := range commands {
			if !slices.Contains(existingCommandNames, command.Definition.Name) || slices.Contains(os.Args, "--update="+command.Definition.Name) {
				cmd, err := s.ApplicationCommandCreate(s.State.User.ID, guild.ID, &command.Definition)
				fmt.Printf("\nAdded command \"%s\"", cmd.Name)
				if err != nil {
					fmt.Println("error creating command,", err)
					continue
				}
			}
		}
	}
	fmt.Print("\nSuccessfully started the Bot!")
}

func interactionCreate(s *discordgo.Session, i *discordgo.InteractionCreate) {
	for _, command := range commands {
		switch i.Type {
		case discordgo.InteractionApplicationCommand:
			if command.Interact != nil && i.ApplicationCommandData().Name == command.Definition.Name {
				command.Interact(s, i)
			}
		case discordgo.InteractionApplicationCommandAutocomplete:
			if command.Autocomplete != nil && i.ApplicationCommandData().Name == command.Definition.Name {
				command.Autocomplete(s, i)
			}
		case discordgo.InteractionModalSubmit:
			var hasID bool = false
			if command.ModalSubmit != nil {
				for _, modalID := range command.ModalIDs {
					if strings.HasPrefix(i.ModalSubmitData().CustomID, modalID) {
						hasID = true
					}
				}
				if hasID {
					command.ModalSubmit(s, i)
				}
			}
		case discordgo.InteractionMessageComponent:
			if command.ComponentInteract != nil {
				if command.Definition.Name == "form" {
					command.ComponentIDs = getFormButtonIDs()
				} // FIXME: Makes it dynamic i don't know why it isn't otherwise
				if slices.Contains(command.ComponentIDs, i.MessageComponentData().CustomID) {
					command.ComponentInteract(s, i)
				}
			}
		}
	}
}

func removeOldCommandFromAllGuilds(s *discordgo.Session) {
	for _, guild := range s.State.Guilds {
		existingCommands, err := s.ApplicationCommands(s.State.User.ID, guild.ID)
		if err != nil {
			fmt.Printf("error fetching existing commands for guild %s: %v\n", guild.Name, err)
			continue
		}
		var commandIDs []string
		for _, command := range commands {
			commandIDs = append(commandIDs, command.Definition.Name)
		}
		for _, existingCommand := range existingCommands {
			if !slices.Contains(commandIDs, existingCommand.Name) {
				fmt.Printf("\nDeleting command \"%s\"", existingCommand.Name)
				err := s.ApplicationCommandDelete(s.State.User.ID, guild.ID, existingCommand.ID)
				if err != nil {
					fmt.Printf("error deleting command %s for guild %s: %v\n", existingCommand.Name, guild.Name, err)
				}
			}
		}
	}
}

func messageCreate(s *discordgo.Session, m *discordgo.MessageCreate) {
	if len(m.Embeds) == 0 || m.Embeds[0].Footer == nil || m.Embeds[0].Footer.Text != "📌 Sticky message" {
		if hasSticky(m.GuildID, m.ChannelID) {
			s.ChannelMessageDelete(m.ChannelID, getStickyMessageID(m.GuildID, m.ChannelID))
			stickyMessage, _ := s.ChannelMessageSendEmbed(m.ChannelID, &discordgo.MessageEmbed{
				Type: discordgo.EmbedTypeArticle,
				Footer: &discordgo.MessageEmbedFooter{
					Text: "📌 Sticky message",
				},
				Color:       hexToDecimal(color["primary"]),
				Description: getStickyMessageContent(m.GuildID, m.ChannelID),
			})
			updateStickyMessageID(m.GuildID, m.ChannelID, stickyMessage.ID)
		}
	}
}

func guildMemberJoin(s *discordgo.Session, m *discordgo.GuildMemberAdd) {
	err := s.GuildMemberRoleAdd(m.GuildID, m.User.ID, getAutoJoinRole(m.GuildID, m.User.Bot))
	if err != nil {
		log.Println(err)
	}
}