using HarmonyLib;
using System;
using TMPro;
using PvZ_Fusion_Translator__BepInEx_.AssetStore;
using static PvZ_Fusion_Translator__BepInEx_.Patches.Managers.TravelMgr_Patch;
using UnityEngine;
using System.Linq;

namespace PvZ_Fusion_Translator__BepInEx_.Patches.GameObjects
{
    [HarmonyPatch(typeof(TravelStore))]
    public static class TravelStore_Patch
    {
        [HarmonyPatch(nameof(TravelStore.Update))]
        [HarmonyPostfix]
        private static void Update(TravelStore __instance)
        {
            foreach(TextMeshProUGUI text in __instance.introduces)
            {
                if(__instance.currentSelect != null)
                {
                    var buffList = translatedTravelBuffs[buffLinks[(BuffType)__instance.currentSelect.theBuffType]];
                    if(__instance.currentSelect.theBuffNumber < buffList.Count)
                    {
                        text.text = buffList[__instance.currentSelect.theBuffNumber];
                    }
                    else
                    {
                        text.text = StringStore.TranslateText(text.text);
                    }
                }
            }
            foreach (var textMesh in __instance.points)
            {
                textMesh.text = StringStore.TranslateText(textMesh.text);
            }
        }

        [HarmonyPatch(nameof(TravelStore.RefreshBuff))]
        [HarmonyPostfix]
        private static void RefreshBuff(TravelStore __instance)
        {
            foreach (TravelBuff buff in __instance.travelBuffs)
            {
                TextMeshProUGUI cost = buff.transform.GetChild(1).GetComponent<TextMeshProUGUI>();
                cost.text = StringStore.TranslateText(cost.text);
                cost.font = FontStore.LoadTMPFont(Utils.Language.ToString());
                //FileLoader.DumpUntranslatedStrings(cost.text);
            }
            foreach (TextMeshProUGUI text in __instance.introduces)
            {
                if (__instance.currentSelect != null)
                {
                    text.text = translatedTravelBuffs[buffLinks[(BuffType)__instance.currentSelect.theBuffType]][__instance.currentSelect.theBuffNumber];
                }
            }
            foreach (var textMesh in __instance.points)
            { 
                textMesh.text = StringStore.TranslateText(textMesh.text); 
            }
        }
    }
}