PVZFusionTranslation/Source/Patches/GameObjects/ButtonObjects/OptLanguageBtn_Patch.cs

415 lines
14 KiB
C#

using HarmonyLib;
using PvZ_Fusion_Translator.AssetStore;
using Object = UnityEngine.Object;
namespace PvZ_Fusion_Translator.Patches.GameObjects.ButtonObjects;
public sealed class OptionButtonData
{
public OptionBtn Button { get; init; }
public Vector3 Position { get; init; }
///<remarks><see langword="null"/> if it is the "Next" button.</remarks>
public Utils.LanguageEnum? Language { get; set; }
///<remarks><see langword="null"/> if it is the "Next" button.</remarks>
public Utils.ToggleEnum? Toggle { get; set; }
public bool IsNextButton { get; set; }
public bool Shifted;
}
internal static class OptLanguageBtn_Patch
{
public static readonly Dictionary<int, OptionButtonData> LanguageBtnDict = new(), ToggleBtnDict = new();
private static bool s_buttonsCreated, s_togglesCreated;
// ReSharper disable once NotAccessedField.Local
private static OptionBtn s_cachedTemplateButton, s_cachedTemplateToggleButton;
private const float
StartX = 4.3241f + 2.56f,
StartY = 2.7769f,
YSpacing = 1.2505f,
ToggleStartX = 4.3241f,
ToggleStartY = -0.9746f;
private static List<Utils.LanguageEnum> s_availableLanguages;
private static List<Utils.ToggleEnum> s_availableToggles;
private const int LanguagesPerPage = 5;
private static int s_currentPage;
// 5 language buttons + 1 next button
public static readonly OptionBtn[] ButtonSlots = new OptionBtn[6], ToggleSlots = new OptionBtn[3];
public static void CreateLanguageButtons(OptionBtn templateButton)
{
if (s_buttonsCreated)
return;
s_buttonsCreated = true;
s_cachedTemplateButton = templateButton;
s_availableLanguages = Enum.GetValues(typeof(Utils.LanguageEnum))
.Cast<Utils.LanguageEnum>()
.Where(lang => lang != Utils.LanguageEnum.LangEnd)
.ToList();
for (var i = 0; i < 6; i++)
{
var newButton = Object.Instantiate(templateButton, templateButton.transform.parent);
newButton.tag = "LangOpt";
foreach (var child in newButton.GetComponentsInChildren<Transform>())
child.tag = "LangOpt";
newButton.optionType = 80 + i;
var yPos = StartY - i * YSpacing;
Vector3 pos = new(StartX, yPos);
newButton.transform.position = pos;
ButtonSlots[i] = newButton;
LanguageBtnDict[newButton.GetInstanceID()] = new()
{
Button = newButton,
Position = pos,
};
}
UpdatePage();
}
public static void CreateToggleButtons(OptionBtn templateButton)
{
if (s_togglesCreated)
return;
s_togglesCreated = true;
s_cachedTemplateToggleButton = templateButton;
s_availableToggles = Enum.GetValues<Utils.ToggleEnum>()
.Where(lang => lang != Utils.ToggleEnum.ToggleEnd)
.ToList();
for (var i = 0; i < 3; i++)
{
var newButton = Object.Instantiate(templateButton, templateButton.transform.parent);
newButton.tag = "LangOpt";
foreach (var child in newButton.GetComponentsInChildren<Transform>())
child.tag = "LangOpt";
newButton.optionType = 100 + i;
var yPos = ToggleStartY - i * YSpacing;
Vector3 pos = new(ToggleStartX, yPos);
newButton.transform.position = pos;
ToggleSlots[i] = newButton;
ToggleBtnDict[newButton.GetInstanceID()] = new()
{
Button = newButton,
Position = pos,
};
}
UpdatePage();
}
private static void UpdatePage()
{
var startIndex = s_currentPage * LanguagesPerPage;
var langCount = s_availableLanguages.Count;
var toggleCount = s_togglesCreated ? s_availableToggles.Count : 0;
for (var i = 0; i < 5; i++)
{
var btn = ButtonSlots[i];
var data = LanguageBtnDict[btn.GetInstanceID()];
if (startIndex + i < langCount)
{
var lang = s_availableLanguages[startIndex + i];
data.Language = lang;
data.IsNextButton = false;
btn.gameObject.SetActive(true);
var languageName = Utils.LanguageNames.TryGetValue(lang, out var name) ? name : lang.ToString();
UpdateButtonText(btn, languageName, FontStore.LoadTMPFont(lang.ToString()));
}
else
{
data.Language = null;
data.IsNextButton = false;
btn.gameObject.SetActive(false);
}
}
if (s_togglesCreated)
for (var i = 0; i < 3; i++)
{
var btn = ToggleSlots[i];
var data = ToggleBtnDict[btn.GetInstanceID()];
if (i < toggleCount)
{
var toggle = s_availableToggles[i];
data.Toggle = toggle;
data.IsNextButton = false;
btn.gameObject.SetActive(true);
var buttonText = Utils.ToggleNames.TryGetValue(toggle, out var name)
? name
: toggle.ToString();
UpdateButtonText(
btn,
StringStore.TranslateText(buttonText),
FontStore.LoadTMPFont(Utils.Language.ToString())
);
string sourceMsg;
switch (i)
{
case 0:
sourceMsg =
$"<size=10>Texture Source:\n{(Core.Instance.ConfigDefaultTextures.Value ? "Default" : "Custom")}";
FlashMessage(btn, sourceMsg, 0.1f, false);
break;
case 1:
sourceMsg =
$"<size=10>Audio Source:\n{(Core.Instance.ConfigDefaultAudio.Value ? "Default" : "Custom")}";
FlashMessage(btn, sourceMsg, 0.1f, false);
break;
case 2:
sourceMsg =
$"<size=10>Translation Source:\n{(Core.Instance.ConfigUseLocal.Value ? "Local" : "Online")}";
FlashMessage(btn, sourceMsg, 0.1f, false);
break;
}
}
else
{
data.Language = null;
data.IsNextButton = false;
btn.gameObject.SetActive(false);
}
}
// "Next" button always visible at index 5
var nextBtn = ButtonSlots[5];
var nextData = LanguageBtnDict[nextBtn.GetInstanceID()];
nextData.Language = null;
nextData.IsNextButton = true;
nextBtn.gameObject.SetActive(true);
UpdateButtonText(nextBtn, "Next");
}
public static void UpdateButtonText(OptionBtn button, string languageName, TMP_FontAsset fontAsset = null)
{
var defaultAsset = FontStore.LoadTMPFont("English");
for (var i = 0; i < 3; i++)
{
var textTransform = button.transform.GetChild(i);
if (textTransform == null)
continue;
if (i == 0)
textTransform.gameObject.SetActive(false);
var text = textTransform.GetComponent<TextMeshProUGUI>();
if (text == null)
continue;
if (i == 0)
{
text.text = "";
text.color = Color.red;
}
else
text.text = languageName;
text.fontSize = 16;
text.font = fontAsset != null ? fontAsset : defaultAsset;
text.autoSizeTextContainer = false;
}
}
public static void FlashMessage(OptionBtn button, string message, float yShift = 0.0f, bool useTimer = true)
{
var child = button.transform.GetChild(0);
if (child == null)
return;
var text = child.GetComponent<TextMeshProUGUI>();
var originalMsg = "";
if (text != null)
{
originalMsg = text.text;
text.text = message;
var isToggle = ToggleBtnDict.TryGetValue(button.GetInstanceID(), out _);
if (isToggle)
{
text.transform.position = new(
text.transform.position.x,
text.transform.position.y + (ToggleBtnDict[button.GetInstanceID()].Shifted ? 0.0f : yShift),
text.transform.position.z
);
ToggleBtnDict[button.GetInstanceID()].Shifted = true;
}
else
{
text.transform.position = new(
text.transform.position.x,
text.transform.position.y + (LanguageBtnDict[button.GetInstanceID()].Shifted ? 0.0f : yShift),
text.transform.position.z
);
LanguageBtnDict[button.GetInstanceID()].Shifted = true;
}
}
child.gameObject.SetActive(true);
if (useTimer)
Task.Delay(1000)
.ContinueWith(
_ =>
{
if (button == null || child == null || !child.gameObject.activeSelf)
return;
var buttonData = ToggleBtnDict[button.GetInstanceID()];
if (buttonData.Toggle == null)
return;
var sourceMsg = buttonData.Toggle switch
{
Utils.ToggleEnum.Textures =>
$"<size=10>Texture Source:\n{(Core.Instance.ConfigDefaultAudio.Value ? "Default" : "Custom")}",
Utils.ToggleEnum.Audio =>
$"<size=10>Audio Source:\n{(Core.Instance.ConfigDefaultAudio.Value ? "Default" : "Custom")}",
_ => originalMsg,
};
text.text = sourceMsg;
}
);
}
private static void ToggleCustomAssets(string type)
{
switch (type)
{
case "Textures":
Core.Instance.ConfigDefaultTextures.Value = !Core.Instance.ConfigDefaultTextures.Value;
TextureStore.Reload();
Core.Coroutine(TextureStore.ReplaceTexturesCoroutine());
break;
case "Audio":
Core.Instance.ConfigDefaultAudio.Value = !Core.Instance.ConfigDefaultAudio.Value;
break;
case "SwapLocal":
Utils.WarnLocalData();
break;
}
}
[HarmonyPatch(typeof(OptionBtn))]
public static class OptLangBtn_Patch
{
public static void Awake(OptionBtn __instance)
{
if (s_buttonsCreated && s_cachedTemplateButton != null)
return;
s_buttonsCreated = false;
s_togglesCreated = false;
CreateLanguageButtons(__instance);
CreateToggleButtons(__instance);
}
[HarmonyPatch("OnMouseUpAsButton"), HarmonyPrefix]
private static void OnMouseUpAsButton(OptionBtn __instance)
{
if (!ModFeatures.IsMultiLanguage)
return;
if (LanguageBtnDict.TryGetValue(__instance.GetInstanceID(), out var btnData))
{
if (btnData.IsNextButton)
{
s_currentPage = (s_currentPage + 1) %
(int)Math.Ceiling((float)s_availableLanguages.Count / LanguagesPerPage);
UpdatePage();
return;
}
if (btnData.Language is { } selectedLang)
{
var currentLang = Utils.Language.ToString();
if (currentLang == selectedLang.ToString())
FlashMessage(btnData.Button, "Already Selected!");
else
{
Utils.ChangeLanguage(selectedLang.ToString());
Debug.Log($"Language changed to: {selectedLang}");
FlashMessage(btnData.Button, "Language Changed!");
}
}
}
if (!ToggleBtnDict.TryGetValue(__instance.GetInstanceID(), out var toggleData))
return;
var toggleType = toggleData.Toggle.ToString();
ToggleCustomAssets(toggleType);
switch (toggleType)
{
case "Textures":
FlashMessage(toggleData.Button, "<size=10>Toggled custom textures!\n(Restart Required)", 0.1f);
break;
case "Audio":
FlashMessage(toggleData.Button, "<size=10>Toggled custom audio!", 0.1f);
break;
}
}
[HarmonyPatch(nameof(OptionBtn.Update)), HarmonyPostfix]
private static void Update(OptionBtn __instance)
{
if (!ModFeatures.IsMultiLanguage)
return;
if (LanguageBtnDict.TryGetValue(__instance.GetInstanceID(), out var btnData))
btnData.Button.transform.position = btnData.Position;
if (ToggleBtnDict.TryGetValue(__instance.GetInstanceID(), out var toggleData))
toggleData.Button.transform.position = toggleData.Position;
}
}
}