using System.Collections.Generic;
using System.Linq;
using Brigadier.NET;
using MinecraftClient.Inventory;

namespace MinecraftClient.Commands
{
    public class Enchant : Command
    {
        public override string CmdName { get { return "enchant"; } }
        public override string CmdUsage { get { return "enchant <top|middle|bottom>"; } }
        public override string CmdDesc { get { return "cmd.enchant.desc"; } }

        public override void RegisterCommand(McClient handler, CommandDispatcher<CommandSource> dispatcher)
        {
        }

        public override string Run(McClient handler, string command, Dictionary<string, object>? localVars)
        {
            if (!handler.GetInventoryEnabled())
                return Translations.TryGet("error.inventoryhandling_not_enabled");

            if (HasArg(command))
            {
                string slot = GetArg(command).ToLower().Trim();

                int slotId = slot switch
                {
                    "top" => 0,
                    "middle" => 1,
                    "bottom" => 2,
                    _ => -1
                };

                if (slotId == -1)
                    return Translations.TryGet("cmd.enchant.invalid_slot");

                Container? enchantingTable = null;

                foreach (var (id, container) in handler.GetInventories())
                {
                    if (container.Type == ContainerType.Enchantment)
                    {
                        enchantingTable = container;
                        break;
                    }
                }

                if (enchantingTable == null)
                    return Translations.TryGet("cmd.enchant.enchanting_table_not_opened");

                int[] emptySlots = enchantingTable.GetEmpytSlots();

                if (emptySlots.Contains(0))
                    return Translations.TryGet("cmd.enchant.enchanting_no_item");

                if (emptySlots.Contains(1))
                    return Translations.TryGet("cmd.enchant.enchanting_no_lapis");

                Item lapisSlot = enchantingTable.Items[1];

                if (lapisSlot.Type != ItemType.LapisLazuli)
                    return Translations.TryGet("cmd.enchant.enchanting_no_lapis");

                if (lapisSlot.Count < 3)
                    return Translations.TryGet("cmd.enchant.enchanting_no_lapis");

                EnchantmentData? enchantment = handler.GetLastEnchantments();

                if (enchantment == null)
                    return Translations.TryGet("cmd.enchant.no_enchantments");

                short requiredLevel = slotId switch
                {
                    0 => enchantment.TopEnchantmentLevelRequirement,
                    1 => enchantment.MiddleEnchantmentLevelRequirement,
                    2 => enchantment.BottomEnchantmentLevelRequirement,
                    _ => 9999
                };

                if (handler.GetLevel() < requiredLevel)
                    return Translations.TryGet("cmd.enchant.no_levels", handler.GetLevel(), requiredLevel);

                return handler.ClickContainerButton(enchantingTable.ID, slotId) ? Translations.TryGet("cmd.enchant.clicked") : Translations.TryGet("cmd.enchant.not_clicked");
            }

            return GetCmdDescTranslated();
        }
    }
}