using System;
using System.Collections.Generic;
using System.Linq;
using MinecraftClient.Protocol.Handlers;

namespace MinecraftClient.Inventory
{
    public class EnchantmentMapping
    {
        #pragma warning disable format // @formatter:off
        // 1.14 - 1.15.2
        private static Dictionary<short, Enchantment> enchantmentMappings114 = new Dictionary<short, Enchantment>()
        {
            //id    type 
            { 0,    Enchantment.Protection },
            { 1,    Enchantment.FireProtection },
            { 2,    Enchantment.FeatherFalling },
            { 3,    Enchantment.BlastProtection },
            { 4,    Enchantment.ProjectileProtection },
            { 5,    Enchantment.Respiration },
            { 6,    Enchantment.AquaAffinity },
            { 7,    Enchantment.Thorns },
            { 8,    Enchantment.DepthStrieder },
            { 9,    Enchantment.FrostWalker },
            { 10,   Enchantment.BindingCurse },
            { 11,   Enchantment.Sharpness },
            { 12,   Enchantment.Smite },
            { 13,   Enchantment.BaneOfArthropods },
            { 14,   Enchantment.Knockback },
            { 15,   Enchantment.FireAspect },
            { 16,   Enchantment.Looting },
            { 17,   Enchantment.Sweeping },
            { 18,   Enchantment.Efficency },
            { 19,   Enchantment.SilkTouch },
            { 20,   Enchantment.Unbreaking },
            { 21,   Enchantment.Fortune },
            { 22,   Enchantment.Power },
            { 23,   Enchantment.Punch },
            { 24,   Enchantment.Flame },
            { 25,   Enchantment.Infinity },
            { 26,   Enchantment.LuckOfTheSea },
            { 27,   Enchantment.Lure },
            { 28,   Enchantment.Loyality },
            { 29,   Enchantment.Impaling },
            { 30,   Enchantment.Riptide },
            { 31,   Enchantment.Channeling },
            { 32,   Enchantment.Mending },
            { 33,   Enchantment.VanishingCurse }
        };

        // 1.16 - 1.18
        private static Dictionary<short, Enchantment> enchantmentMappings116 = new Dictionary<short, Enchantment>()
        {
            //id    type 
            { 0,    Enchantment.Protection },
            { 1,    Enchantment.FireProtection },
            { 2,    Enchantment.FeatherFalling },
            { 3,    Enchantment.BlastProtection },
            { 4,    Enchantment.ProjectileProtection },
            { 5,    Enchantment.Respiration },
            { 6,    Enchantment.AquaAffinity },
            { 7,    Enchantment.Thorns },
            { 8,    Enchantment.DepthStrieder },
            { 9,    Enchantment.FrostWalker },
            { 10,   Enchantment.BindingCurse },
            { 11,   Enchantment.SoulSpeed },
            { 12,   Enchantment.Sharpness },
            { 13,   Enchantment.Smite },
            { 14,   Enchantment.BaneOfArthropods },
            { 15,   Enchantment.Knockback },
            { 16,   Enchantment.FireAspect },
            { 17,   Enchantment.Looting },
            { 18,   Enchantment.Sweeping },
            { 19,   Enchantment.Efficency },
            { 20,   Enchantment.SilkTouch },
            { 21,   Enchantment.Unbreaking },
            { 22,   Enchantment.Fortune },
            { 23,   Enchantment.Power },
            { 24,   Enchantment.Punch },
            { 25,   Enchantment.Flame },
            { 26,   Enchantment.Infinity },
            { 27,   Enchantment.LuckOfTheSea },
            { 28,   Enchantment.Lure },
            { 29,   Enchantment.Loyality },
            { 30,   Enchantment.Impaling },
            { 31,   Enchantment.Riptide },
            { 32,   Enchantment.Channeling },
            { 33,   Enchantment.Multishot },
            { 34,   Enchantment.QuickCharge },
            { 35,   Enchantment.Piercing },
            { 36,   Enchantment.Mending },
            { 37,   Enchantment.VanishingCurse }
        };

        // 1.19+
        private static Dictionary<short, Enchantment> enchantmentMappings = new Dictionary<short, Enchantment>()
        {
            //id    type 
            { 0,    Enchantment.Protection },
            { 1,    Enchantment.FireProtection },
            { 2,    Enchantment.FeatherFalling },
            { 3,    Enchantment.BlastProtection },
            { 4,    Enchantment.ProjectileProtection },
            { 5,    Enchantment.Respiration },
            { 6,    Enchantment.AquaAffinity },
            { 7,    Enchantment.Thorns },
            { 8,    Enchantment.DepthStrieder },
            { 9,    Enchantment.FrostWalker },
            { 10,   Enchantment.BindingCurse },
            { 11,   Enchantment.SoulSpeed },
            { 12,   Enchantment.SwiftSneak },
            { 13,   Enchantment.Sharpness },
            { 14,   Enchantment.Smite },
            { 15,   Enchantment.BaneOfArthropods },
            { 16,   Enchantment.Knockback },
            { 17,   Enchantment.FireAspect },
            { 18,   Enchantment.Looting },
            { 19,   Enchantment.Sweeping },
            { 20,   Enchantment.Efficency },
            { 21,   Enchantment.SilkTouch },
            { 22,   Enchantment.Unbreaking },
            { 23,   Enchantment.Fortune },
            { 24,   Enchantment.Power },
            { 25,   Enchantment.Punch },
            { 26,   Enchantment.Flame },
            { 27,   Enchantment.Infinity },
            { 28,   Enchantment.LuckOfTheSea },
            { 29,   Enchantment.Lure },
            { 30,   Enchantment.Loyality },
            { 31,   Enchantment.Impaling },
            { 32,   Enchantment.Riptide },
            { 33,   Enchantment.Channeling },
            { 34,   Enchantment.Multishot },
            { 35,   Enchantment.QuickCharge },
            { 36,   Enchantment.Piercing },
            { 37,   Enchantment.Mending },
            { 38,   Enchantment.VanishingCurse }
        };
#pragma warning restore format // @formatter:on

        public static Enchantment GetEnchantmentById(int protocolVersion, short id)
        {
            if (protocolVersion < Protocol18Handler.MC_1_14_Version)
                throw new Exception("Enchantments mappings are not implemented bellow 1.14");

            Dictionary<short, Enchantment> map = enchantmentMappings;

            if (protocolVersion >= Protocol18Handler.MC_1_14_Version && protocolVersion < Protocol18Handler.MC_1_16_Version)
                map = enchantmentMappings114;
            else if (protocolVersion >= Protocol18Handler.MC_1_16_Version && protocolVersion < Protocol18Handler.MC_1_19_Version)
                map = enchantmentMappings116;

            if (!map.ContainsKey(id))
                throw new Exception("Got an Unknown Enchantment ID '" + id + "', please update the Mappings!");

            return map[id];
        }

        public static string GetEnchantmentName(Enchantment enchantment)
        {
            string? trans = Protocol.ChatParser.TranslateString("enchantment.minecraft." + enchantment.ToString().ToUnderscoreCase());
            if (string.IsNullOrEmpty(trans))
                return "Unknown Enchantment with ID: " + ((short)enchantment) + " (Probably not named in the code yet)";
            else
                return trans;
        }

        public static string ConvertLevelToRomanNumbers(int num)
        {
            string result = string.Empty;
            Dictionary<string, int> romanNumbers = new Dictionary<string, int>
            {
                {"M", 1000 },
                {"CM", 900},
                {"D", 500},
                {"CD", 400},
                {"C", 100},
                {"XC", 90},
                {"L", 50},
                {"XL", 40},
                {"X", 10},
                {"IX", 9},
                {"V", 5},
                {"IV", 4},
                {"I", 1}
            };

            foreach (var pair in romanNumbers)
            {
                result += string.Join(string.Empty, Enumerable.Repeat(pair.Key, num / pair.Value));
                num %= pair.Value;
            }

            return result;
        }
    }
}