PVZFusionTranslation/Source/Patches/Modes/Super Editor/GameLevel/EventNodes/RuntimeNodeUI_Patch.cs

119 lines
5.3 KiB
C#

using HarmonyLib;
using PvZ_Fusion_Translator.AssetStore;
using PvZ_Fusion_Translator.Patches.BaseTextObjects;
namespace PvZ_Fusion_Translator.Patches.Modes.Super_Editor.GameLevel.EventNodes;
[HarmonyPatch(typeof(RuntimeNodeUI))]
public static class RuntimeNodeUI_Patch
{
[HarmonyPatch(nameof(RuntimeNodeUI.Awake)), HarmonyPatch(nameof(RuntimeNodeUI.CreatePorts)),
HarmonyPatch(nameof(RuntimeNodeUI.CreateValueEditUI)), HarmonyPatch(nameof(RuntimeNodeUI.GetPortUI)),
HarmonyPatch(nameof(RuntimeNodeUI.UpdateDisplay)), HarmonyPatch(nameof(RuntimeNodeUI.UpdateSelectionVisual)),
HarmonyPatch(nameof(RuntimeNodeUI.UpdateVisable)), HarmonyPostfix]
public static void Post_UpdateDisplay(RuntimeNodeUI __instance)
{
if (!__instance.isActiveAndEnabled)
return;
if (__instance.Node == null)
return;
var fontAsset = FontStore.LoadTMPFont(Utils.Language.ToString());
__instance.titleText.text = TranslateCodeText(__instance.titleText.text);
__instance.titleText.font = fontAsset;
foreach (var txt in __instance.inputPortsContainer.GetComponentsInChildren<TextMeshProUGUI>())
{
txt.text = TranslateCodeText(txt.text);
txt.font = fontAsset;
}
foreach (var txt in __instance.outputPortsContainer.GetComponentsInChildren<TextMeshProUGUI>())
{
txt.text = TranslateCodeText(txt.text);
txt.font = fontAsset;
}
if (
__instance.GetComponentsInChildren<TMP_Dropdown>().Count == 0 &&
__instance.GetComponentsInChildren<TMP_InputField>().Count == 0 &&
__instance.Node.nodeType != "PlantTypeValueNode" &&
__instance.Node.nodeType != "SinglePlantTypeListNode" &&
__instance.Node.nodeType != "ZombieTypeValueNode"
)
return;
if (__instance.Node != null)
{
if (__instance.Node.nodeType == "PlantTypeValueNode")
foreach (var txt in __instance.GetComponentsInChildren<TextMeshProUGUI>())
if (txt.gameObject.name.Contains("Label"))
if (Utils.CheckForUntranslatedText(txt.text))
{
var castedNode = __instance.Node.TryCast<PlantTypeValueNode>();
if (castedNode == null)
continue;
_ = castedNode.value_PortName;
var plantName = Utils.GetPlantNameFromAlmanac(castedNode.value, true);
txt.text = plantName != "" ? plantName : StringStore.TranslateText(txt.text);
txt.font = fontAsset;
}
if (__instance.Node.nodeType == "SinglePlantTypeListNode")
foreach (var txt in __instance.GetComponentsInChildren<TextMeshProUGUI>())
if (txt.gameObject.name.Contains("Label"))
if (Utils.CheckForUntranslatedText(txt.text))
{
var castedNode = __instance.Node.TryCast<SinglePlantTypeListNode>();
if (castedNode == null)
continue;
_ = castedNode.plantTypeList_PortName;
var plantName = Utils.GetPlantNameFromAlmanac(castedNode.plantType, true);
txt.text = plantName != "" ? plantName : StringStore.TranslateText(txt.text);
txt.font = fontAsset;
}
if (__instance.Node.nodeType == "ZombieTypeValueNode")
foreach (var txt in __instance.GetComponentsInChildren<TextMeshProUGUI>())
if (txt.gameObject.name.Contains("Label"))
if (Utils.CheckForUntranslatedText(txt.text))
{
var castedNode = __instance.Node.TryCast<ZombieTypeValueNode>();
if (castedNode == null)
continue;
_ = castedNode.value_PortName;
var zombieName = Utils.GetZombieNameFromAlmanac(castedNode.value, true);
txt.text = zombieName != "" ? zombieName : StringStore.TranslateText(txt.text);
txt.font = fontAsset;
}
}
if (__instance.GetComponentsInChildren<TMP_Dropdown>().Count > 0)
foreach (var dropdown in __instance.GetComponentsInChildren<TMP_Dropdown>())
TMP_Dropdown_Patch.CreateDropdownList(dropdown);
if (__instance.GetComponentsInChildren<TMP_InputField>().Count <= 0)
return;
foreach (var inputField in __instance.GetComponentsInChildren<TMP_InputField>())
{
TMP_InputField_Patch.OnEnable(inputField);
foreach (var txt in inputField.GetComponentsInChildren<TextMeshProUGUI>(true))
txt.text = TranslateCodeText(txt.text);
}
}
public static string TranslateCodeText(string originalText) =>
StringStore.TranslationString.ContainsKey(originalText + "_code")
? StringStore.TranslateText(originalText + "_code")
: StringStore.TranslateText(originalText);
}