using HarmonyLib;
using Il2Cpp;
using PvZ_Fusion_Translator.AssetStore;
using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Text.Encodings.Web;
using System.Text.Json;
using System.Threading.Tasks;
using Unity.Collections;
using UnityEngine.Rendering;
using static PvZ_Fusion_Translator.FileLoader;

namespace PvZ_Fusion_Translator.Patches.Modes.Odyssey
{
    [HarmonyPatch(typeof(TravelMgr))]
    public static class TravelMgr_Patch
    {
        public static readonly Dictionary<string, SortedDictionary<int, string>> templateTravelBuffs = new()
        {
            { "advancedBuffs", new SortedDictionary<int, string>() },
            { "ultimateBuffs", new SortedDictionary<int, string>() },
            { "debuffs", new SortedDictionary<int, string>() },
            { "unlocks", new SortedDictionary<int, string>() },
            { "investmentBuffs", new SortedDictionary<int, string>() }
        };
        
        public static Dictionary<string, SortedDictionary<int, string>> dumpedTravelBuffs = new()
        {
            { "advancedBuffs", new SortedDictionary<int, string>() },
            { "ultimateBuffs", new SortedDictionary<int, string>() },
            { "debuffs", new SortedDictionary<int, string>() },
            { "unlocks", new SortedDictionary<int, string>() },
            { "investmentBuffs", new SortedDictionary<int, string>() }
        };

        public static Dictionary<string, SortedDictionary<int, string>> translatedTravelBuffs = new()
        {
            { "advancedBuffs", new SortedDictionary<int, string>() },
            { "ultimateBuffs", new SortedDictionary<int, string>() },
            { "debuffs", new SortedDictionary<int, string>() },
            { "unlocks", new SortedDictionary<int, string>() },
            { "investmentBuffs", new SortedDictionary<int, string>() }
        };

        public static Dictionary<string, string> travelBuffString = new()
        {

        };

        public static Dictionary<BuffType, string> buffLinks = new()
        {
            { BuffType.AdvancedBuff, "advancedBuffs" },
            { BuffType.UltimateBuff, "ultimateBuffs" },
            { BuffType.Debuff, "debuffs" },
            { BuffType.UnlockPlant, "unlocks" },
            { BuffType.InvestmentBuff, "investmentBuffs" }
        };

        [HarmonyPatch(nameof(TravelMgr.GetText))]
        [HarmonyPostfix]
        public static void GetText(TravelMgr __instance, ref string __result)
        {
            if (travelBuffString.ContainsKey(__result))
            { 
                __result = travelBuffString[__result]; 
            }
            else if(travelBuffString.ContainsKey(RemoveBuffName(__result)))
            {
                __result = travelBuffString[RemoveBuffName(__result)];
            }
            else
            {
                __result = StringStore.TranslateText(__result);
            }
        }

        public static void DumpTravelBuffs()
        {
            TravelMgr.Instance.GetPlantBuffUnlockCount(PlantType.DoomGatling);
            dumpedTravelBuffs = new(templateTravelBuffs);
            travelBuffString.Clear();

            foreach (var pair in TravelDictionary.advancedBuffsText)
            {
                Utils.TryAdd(dumpedTravelBuffs["advancedBuffs"], (int)pair.Key, pair.Value);
                Utils.TryAdd(travelBuffString, pair.Value, pair.Value);
            }

            foreach (var pair in TravelDictionary.ultimateBuffsText)
            {
                Utils.TryAdd(dumpedTravelBuffs["ultimateBuffs"], (int)pair.Key, pair.Value);
                Utils.TryAdd(travelBuffString, pair.Value, pair.Value);
            }

            foreach (var pair in TravelDictionary.debuffData)
            {
                Utils.TryAdd(dumpedTravelBuffs["debuffs"], (int)pair.Key, pair.Value.Item1);
                Utils.TryAdd(travelBuffString, pair.Value.Item1, pair.Value.Item1);
            }

            foreach (var pair in TravelDictionary.unlocksText)
            {
                Utils.TryAdd(dumpedTravelBuffs["unlocks"], (int)pair.Key, pair.Value);
                Utils.TryAdd(travelBuffString, pair.Value, pair.Value);
            }

            foreach (var pair in TravelMgr.InvestBuffsData)
            {
                Utils.TryAdd(dumpedTravelBuffs["investmentBuffs"], (int)pair.Key, pair.Value.GetDescription());
                Utils.TryAdd(travelBuffString, pair.Value.GetDescription(), pair.Value.GetDescription());
            }

            File.WriteAllText(Path.Combine(GetAssetDir(AssetType.Dumps), "travel_buffs.json"), JsonSerializer.Serialize(dumpedTravelBuffs, new JsonSerializerOptions
            {
                WriteIndented = true,
                Encoder = JavaScriptEncoder.UnsafeRelaxedJsonEscaping
            }));

            string translatedTravelBuffsPath = Path.Combine(GetAssetDir(AssetType.Strings, Utils.Language), "travel_buffs.json");

            if (!File.Exists(translatedTravelBuffsPath))
            {
                File.WriteAllText(translatedTravelBuffsPath, JsonSerializer.Serialize(dumpedTravelBuffs, new JsonSerializerOptions
                {
                    WriteIndented = true,
                    Encoder = JavaScriptEncoder.UnsafeRelaxedJsonEscaping
                }));
                translatedTravelBuffs = dumpedTravelBuffs;
            }
        }

        //System.Text.Json.JsonSerializer.Deserialize<Dictionary<string, SortedDictionary<int, string>>>(travelBuffsContent);
        public static void GenerateTranslatedTravelBuffs(string travelBuffsContent)
        {
            if (!Core.isInitialized) return;
            
            DumpTravelBuffs();

            Dictionary<string, SortedDictionary<int, string>> travelBuffs = JsonSerializer.Deserialize<Dictionary<string, SortedDictionary<int, string>>>(travelBuffsContent);
            translatedTravelBuffs = travelBuffs;
            travelBuffString.Clear();

            foreach(var i in travelBuffs)
            {
                var originalBuffSet = dumpedTravelBuffs[i.Key];
                foreach(var j in originalBuffSet)
                {
                    string originalBuff = j.Value;
                    var translatedBuffSet = travelBuffs[i.Key];

                    if (j.Value == originalBuff)
                    {
                        if (translatedBuffSet.ContainsKey(j.Key))
                        {
                            Utils.TryAdd(travelBuffString, originalBuff, translatedBuffSet[j.Key]);
                        }
                    }
                    else if(j.Value == RemoveBuffName(originalBuff))
                    {
                        if (translatedBuffSet.ContainsKey(j.Key))
                        {
                            Utils.TryAdd(travelBuffString, originalBuff, translatedBuffSet[j.Key]);
                        }
                    }

                    /*var translatedBuffSet = travelBuffs[i.Key];
                    if(translatedBuffSet.ContainsKey(j.Key))
                    {
                        string translatedBuff = travelBuffs[i.Key][j.Key];
                        travelBuffString[originalBuff] = translatedBuff;
                    }*/
                }
            }

            string travelBuffsToSave = SerializeWithIndentation(TravelMgr_Patch.translatedTravelBuffs);
			string stringDir = GetAssetDir(AssetType.Strings, Utils.Language);
			File.WriteAllText(Path.Combine(stringDir, "travel_buffs.json"), travelBuffsToSave);
        }

        public static string MatchTravelBuff(string originalText)
        {
            string res = "";

            foreach (var i in dumpedTravelBuffs)
            {
                foreach (var j in dumpedTravelBuffs[i.Key])
                {
                    if (j.Value == originalText || j.Value == RemoveBuffName(originalText))
                    {
                        var translatedBuffSet = translatedTravelBuffs[i.Key];
                        if (translatedBuffSet.ContainsKey(j.Key))
                        {
                            res = translatedBuffSet[j.Key];
                        }
                        break;
                    }
                }

                if (res != "") break;
            }

            return res;
        }

        public static string RemoveBuffName(string buffText)
        {
            string res = buffText;
            int firstColon = res.IndexOf("：");
            if(firstColon > 0)
            {
                res = res.Substring(firstColon + 1);
            }
            return res;
        }

        public static ValueTuple<string, int, string> FindBuffData(string buffText)
        {
            ValueTuple<string, int, string> res = new();
            foreach(var pair1 in translatedTravelBuffs)
            {
                foreach(var pair2 in pair1.Value)
                {
                    if(buffText == pair2.Value)
                    {
                        res.Item1 = pair1.Key;
                        res.Item2 = pair2.Key;
                    }
                }
            }

            foreach(var pair1 in dumpedTravelBuffs)
            {
                foreach(var pair2 in pair1.Value)
                {
                    if(buffText == pair2.Value)
                    {
                        res.Item1 = pair1.Key;
                        res.Item2 = pair2.Key;
                    }
                }
            }

            string finalBuffText = buffText;
            if(travelBuffString.ContainsKey(buffText))
            {
                finalBuffText = travelBuffString[buffText];
            }
            else
            {
                finalBuffText = StringStore.TranslateText(buffText);
            }
            res.Item3 = finalBuffText;

            return res;
        }
    }
}