using System.Globalization;
using System.Numerics;

namespace LightlessSync.UI
{
    internal static class UIColors
    {
        private static readonly Dictionary<string, string> DefaultHexColors = new(StringComparer.OrdinalIgnoreCase)
        {
            { "LightlessPurple", "#ad8af5" },
            { "LightlessBlue", "#a6c2ff" },
            { "LightlessYellow", "#ffe97a" },
            { "PairBlue", "#88a2db" },
            { "DimRed", "#d44444" },
        };

        private static readonly Dictionary<string, Vector4> CustomColors = new(StringComparer.OrdinalIgnoreCase);

        public static Vector4 Get(string name)
        {
            if (CustomColors.TryGetValue(name, out var customColor))
                return customColor;

            if (!DefaultHexColors.TryGetValue(name, out var hex))
                throw new ArgumentException($"Color '{name}' not found in UIColors.");

            return HexToRgba(hex);
        }

        public static void Set(string name, Vector4 color)
        {
            if (!DefaultHexColors.ContainsKey(name))
                throw new ArgumentException($"Color '{name}' not found in UIColors.");

            CustomColors[name] = color;
        }

        public static void Reset(string name)
        {
            CustomColors.Remove(name);
        }

        public static void ResetAll()
        {
            CustomColors.Clear();
        }

        public static Vector4 GetDefault(string name)
        {
            if (!DefaultHexColors.TryGetValue(name, out var hex))
                throw new ArgumentException($"Color '{name}' not found in UIColors.");

            return HexToRgba(hex);
        }

        public static bool IsCustom(string name)
        {
            return CustomColors.ContainsKey(name);
        }

        public static IEnumerable<string> GetColorNames()
        {
            return DefaultHexColors.Keys;
        }

        public static Vector4 HexToRgba(string hexColor)
        {
            hexColor = hexColor.TrimStart('#');
            int r = int.Parse(hexColor.Substring(0, 2), NumberStyles.HexNumber);
            int g = int.Parse(hexColor.Substring(2, 2), NumberStyles.HexNumber);
            int b = int.Parse(hexColor.Substring(4, 2), NumberStyles.HexNumber);
            int a = hexColor.Length == 8 ? int.Parse(hexColor.Substring(6, 2), NumberStyles.HexNumber) : 255;
            return new Vector4(r / 255f, g / 255f, b / 255f, a / 255f);
        }

        public static string RgbaToHex(Vector4 color)
        {
            int r = (int)(color.X * 255);
            int g = (int)(color.Y * 255);
            int b = (int)(color.Z * 255);
            int a = (int)(color.W * 255);
            return $"#{r:X2}{g:X2}{b:X2}{(a != 255 ? a.ToString("X2") : "")}";
        }
    }
}