From 3afedd515f3615e18875ae02e1abac001ff17a41 Mon Sep 17 00:00:00 2001
From: SillyStar-Github <90102598+SillyStar-Github@users.noreply.github.com>
Date: Sat, 9 Aug 2025 04:09:50 -0400
Subject: [PATCH] td + back button fix

---
 .../Patches/GameObjects/BaseMenu_Patch.cs     | 10 +++++++-
 .../GameObjects/TowerUpgradeMenu_Patch.cs     | 23 ++++++++++++++++---
 2 files changed, 29 insertions(+), 4 deletions(-)

diff --git a/MelonLoader/Patches/GameObjects/BaseMenu_Patch.cs b/MelonLoader/Patches/GameObjects/BaseMenu_Patch.cs
index 00dc08b..00b21a0 100644
--- a/MelonLoader/Patches/GameObjects/BaseMenu_Patch.cs
+++ b/MelonLoader/Patches/GameObjects/BaseMenu_Patch.cs
@@ -106,7 +106,15 @@ namespace PvZ_Fusion_Translator.Patches.GameObjects
                         StringStore.TranslateTextTransform(pageScaryPotTransform.Find("LastPage"));
                     }
 
-                    foreach(TextMeshProUGUI txt in mainTransform.GetComponentsInChildren<TextMeshProUGUI>())
+                    Transform pageEndlessLevelTransform = levelsTransform.transform.Find("PageEndlessLevel");
+                    if (pageEndlessLevelTransform != null)
+                    {
+                        StringStore.TranslateTextTransform(pageEndlessLevelTransform.Find("BackToIndex"));
+                        StringStore.TranslateTextTransform(pageEndlessLevelTransform.Find("Nextpage"));
+                        StringStore.TranslateTextTransform(pageEndlessLevelTransform.Find("LastPage"));
+                    }
+
+                    foreach (TextMeshProUGUI txt in mainTransform.GetComponentsInChildren<TextMeshProUGUI>())
                     {
                         FileLoader.DumpUntranslatedStrings(txt.text);
                         txt.text = StringStore.TranslateText(txt.text);
diff --git a/MelonLoader/Patches/GameObjects/TowerUpgradeMenu_Patch.cs b/MelonLoader/Patches/GameObjects/TowerUpgradeMenu_Patch.cs
index 3ad6c1c..4b863b3 100644
--- a/MelonLoader/Patches/GameObjects/TowerUpgradeMenu_Patch.cs
+++ b/MelonLoader/Patches/GameObjects/TowerUpgradeMenu_Patch.cs
@@ -9,11 +9,28 @@ namespace PvZ_Fusion_Translator.Patches.GameObjects
     [HarmonyPatch(typeof(TowerUpgradeMenu))]
     public static class TowerUpgradeMenu_Patch
     {
-        [HarmonyPatch(nameof(TowerUpgradeMenu.Update))]
+        [HarmonyPatch(nameof(TowerUpgradeMenu.Awake))]
         [HarmonyPostfix]
-        private static void Update(TowerUpgradeMenu __instance)
+        private static void Awake(TowerUpgradeMenu __instance)
         {
-            foreach(TextMeshProUGUI text in __instance.GetComponentsInChildren<TextMeshProUGUI>())
+            __instance.des.text = StringStore.TranslateText(__instance.des.text);
+            __instance.des2.text = StringStore.TranslateText(__instance.des2.text);
+
+            foreach (TextMeshProUGUI text in __instance.GetComponentsInChildren<TextMeshProUGUI>())
+            {
+                text.text = StringStore.TranslateText(text.text);
+                text.font = FontStore.LoadTMPFont(Utils.Language.ToString());
+            }
+        }
+
+        [HarmonyPatch(nameof(TowerUpgradeMenu.SetText))]
+        [HarmonyPostfix]
+        private static void SetText(TowerUpgradeMenu __instance)
+        {
+            __instance.des.text = StringStore.TranslateText(__instance.des.text);
+            __instance.des2.text = StringStore.TranslateText(__instance.des2.text);
+
+            foreach (TextMeshProUGUI text in __instance.GetComponentsInChildren<TextMeshProUGUI>())
             {
                 text.text = StringStore.TranslateText(text.text);
                 text.font = FontStore.LoadTMPFont(Utils.Language.ToString());