PKHeX/PKHeX.WinForms/Util/DevUtil.cs

#if DEBUG
using System;
using System.Collections.Generic;
using System.IO;
using System.Linq;
using System.Windows.Forms;
using PKHeX.Core;
using PKHeX.WinForms.Controls;

namespace PKHeX.WinForms
{
    public static class DevUtil
    {
        public static void AddControl(ToolStripDropDownItem t)
        {
            t.DropDownItems.Add(GetTranslationUpdater());
        }

        private static readonly string[] Languages = ["ja", "fr", "it", "de", "es", "ko", "zh", "zh2"];
        private const string DefaultLanguage = GameLanguage.DefaultLanguage;

        public static bool IsUpdatingTranslations { get; private set; }

        /// <summary>
        /// Call this to update all translatable resources (Program Messages, Legality Text, Program GUI)
        /// </summary>
        private static void UpdateAll()
        {
            if (DialogResult.Yes != WinFormsUtil.Prompt(MessageBoxButtons.YesNo, "Update translation files with current values?"))
                return;
            IsUpdatingTranslations = true;
            DumpStringsLegality();
            DumpStringsMessage();
            UpdateTranslations();
            IsUpdatingTranslations = false;
        }

        private static ToolStripMenuItem GetTranslationUpdater()
        {
            var ti = new ToolStripMenuItem
            {
                ShortcutKeys = Keys.Control | Keys.Alt | Keys.D,
                Visible = false,
            };
            ti.Click += (s, e) => UpdateAll();
            return ti;
        }

        private static void UpdateTranslations()
        {
            var assembly = System.Reflection.Assembly.GetExecutingAssembly();
            var types = assembly.GetTypes();
            // add mode
            WinFormsTranslator.SetRemovalMode(false);
            WinFormsTranslator.LoadSettings<PKHeXSettings>(DefaultLanguage);
            WinFormsTranslator.LoadAllForms(types, LoadBanlist); // populate with every possible control
            WinFormsTranslator.TranslateControls(GetExtraControls());
            WinFormsTranslator.UpdateAll(DefaultLanguage, Languages); // propagate to others
            WinFormsTranslator.DumpAll(Banlist); // dump current to file

            // de-populate
            WinFormsTranslator.SetRemovalMode(); // remove used keys, don't add any
            WinFormsTranslator.LoadSettings<PKHeXSettings>(DefaultLanguage, false);
            WinFormsTranslator.LoadAllForms(types, LoadBanlist);
            WinFormsTranslator.TranslateControls(GetExtraControls());
            WinFormsTranslator.RemoveAll(DefaultLanguage, PurgeBanlist); // remove all lines from above generated files that still remain

            // Move translated files from the debug exe loc to their project location
            var files = Directory.GetFiles(Application.StartupPath);
            var dir = GetResourcePath("PKHeX.WinForms", "Resources", "text");
            foreach (var f in files)
            {
                var fn = Path.GetFileName(f);
                if (!fn.EndsWith(".txt"))
                    continue;
                if (!fn.StartsWith("lang_"))
                    continue;

                var loc = Path.Combine(dir, fn);
                if (File.Exists(loc))
                    File.Delete(loc);
                File.Move(f, loc, true);
            }

            Application.Exit();
        }

        private static IEnumerable<Control> GetExtraControls()
        {
            var slotGroupLabels = Enum.GetNames(typeof(StorageSlotType));
            foreach (var name in slotGroupLabels)
                yield return new Label { Name = $"{nameof(Main)}.L_{name}", Text = name };
        }

        private static readonly string[] LoadBanlist =
        [
            nameof(SplashScreen),
            nameof(PokePreview),
        ];

        private static readonly string[] Banlist =
        [
            ..LoadBanlist,
            "Gender=", // editor gender labels
            "BTN_Shinytize", // ☆
            "Hidden_", // Hidden controls
            "CAL_", // calendar controls now expose Text, don't care.
            $"{nameof(Main)}.L_SizeH", // height rating
            $"{nameof(Main)}.L_SizeW", // weight rating
            $"{nameof(Main)}.L_SizeS", // scale rating
            $"{nameof(Main)}.B_Box", // << and >> arrows
            $"{nameof(Main)}.L_Characteristic=", // Characterstic (dynamic)
            $"{nameof(Main)}.L_Potential", // ★☆☆☆ IV judge evaluation
            $"{nameof(SAV_HoneyTree)}.L_Tree0", // dynamic, don't bother
            $"{nameof(SAV_Misc3)}.BTN_Symbol", // symbols should stay as their current character
            $"{nameof(SAV_GameSelect)}.L_Prompt", // prompt text (dynamic)
            $"{nameof(SAV_BlockDump8)}.L_BlockName", // Block name (dynamic)
            $"{nameof(SAV_PokedexResearchEditorLA)}.L_", // Dynamic label
        ];

        private static readonly string[] PurgeBanlist =
        [
            nameof(SuperTrainingEditor),
            nameof(ErrorWindow),
            nameof(SettingsEditor),
        ];

        private static void DumpStringsMessage() => DumpStrings(typeof(MessageStrings), false, "PKHeX.Core", "Resources", "text", "program");
        private static void DumpStringsLegality() => DumpStrings(typeof(LegalityCheckStrings), true, "PKHeX.Core", "Resources", "legality", "checks");

        private static void DumpStrings(Type t, bool sorted, params string[] rel)
        {
            var dir = GetResourcePath(rel);
            DumpStrings(t, sorted, DefaultLanguage, dir);
            foreach (var lang in Languages)
                DumpStrings(t, sorted, lang, dir);
        }

        private static void DumpStrings(Type t, bool sorted, string lang, string dir)
        {
            LocalizationUtil.SetLocalization(t, lang);
            var entries = LocalizationUtil.GetLocalization(t);
            IEnumerable<string> export = entries.OrderBy(GetName); // sorted lines

            if (!sorted)
                export = entries;

            var location = GetFileLocationInText(t.Name, dir, lang);
            File.WriteAllLines(location, export);
            LocalizationUtil.SetLocalization(t, DefaultLanguage);

            static string GetName(string line)
            {
                var index = line.IndexOf('=');
                if (index == -1)
                    return line;
                return line[..index];
            }
        }

        private static string GetFileLocationInText(string fileType, string dir, string lang)
        {
            var fn = $"{fileType}_{lang}.txt";
            return Path.Combine(dir, fn);
        }

        private static string GetResourcePath(params string[] subdir)
        {
            // Starting from the executable path, crawl upwards until we get to the repository/sln root
            const string repo = "PKHeX";
            var path = Application.StartupPath;
            while (true)
            {
                var parent = Directory.GetParent(path) ?? throw new DirectoryNotFoundException(path);
                path = parent.FullName;
                if (path.EndsWith(repo))
                    return Path.Combine(path, Path.Combine(subdir));
            }
        }
    }
}
#endif