PKHeX/PKHeX.WinForms/Subforms/PKM Editors/MemoryAmie.cs

using System;
using System.Windows.Forms;
using PKHeX.Core;

namespace PKHeX.WinForms;

public partial class MemoryAmie : Form
{
    private readonly TextMarkup TextArgs;
    private readonly MemoryStrings MemStrings;

    public MemoryAmie(PKM pk)
    {
        InitializeComponent();
        WinFormsUtil.TranslateInterface(this, Main.CurrentLanguage);
        Entity = pk;
        MemStrings = new MemoryStrings(GameInfo.Strings);
        PrevCountries = [CB_Country0, CB_Country1, CB_Country2, CB_Country3, CB_Country4];
        PrevRegions = [CB_Region0, CB_Region1, CB_Region2, CB_Region3, CB_Region4];
        string[] arguments = L_Arguments.Text.Split(" ; ");

        TextArgs = new TextMarkup(arguments);

        if (Entity is IGeoTrack)
        {
            foreach (var cb in PrevRegions)
                cb.InitializeBinding();
            foreach (var cb in PrevCountries)
            {
                cb.InitializeBinding();
                Main.SetCountrySubRegion(cb, "countries");
            }
        }
        else
        {
            tabControl1.TabPages.Remove(Tab_Residence);
        }
        if (Entity is PK9)
            tabControl1.TabPages.Remove(Tab_Other); // No Fullness/Enjoyment stored.

        GetLangStrings();
        LoadFields();
    }

    private bool init;
    private readonly ComboBox[] PrevCountries;
    private readonly ComboBox[] PrevRegions;
    private readonly PKM Entity;

    // Load/Save Actions
    private void LoadFields()
    {
        // Load the region/country values.
        if (Entity is IGeoTrack g)
        {
            CB_Country0.SelectedValue = (int)g.Geo1_Country;
            CB_Country1.SelectedValue = (int)g.Geo2_Country;
            CB_Country2.SelectedValue = (int)g.Geo3_Country;
            CB_Country3.SelectedValue = (int)g.Geo4_Country;
            CB_Country4.SelectedValue = (int)g.Geo5_Country;
            CB_Region0.SelectedValue = (int)g.Geo1_Region;
            CB_Region1.SelectedValue = (int)g.Geo2_Region;
            CB_Region2.SelectedValue = (int)g.Geo3_Region;
            CB_Region3.SelectedValue = (int)g.Geo4_Region;
            CB_Region4.SelectedValue = (int)g.Geo5_Region;
        }

        // Load the Fullness, and Enjoyment
        M_Fullness.Text = Entity.Fullness.ToString();
        M_Enjoyment.Text = Entity.Enjoyment.ToString();

        M_OT_Friendship.Text = Entity.OriginalTrainerFriendship.ToString();
        M_CT_Friendship.Text = Entity.HandlingTrainerFriendship.ToString();

        if (Entity is IAffection a)
        {
            M_OT_Affection.Text = a.OriginalTrainerAffection.ToString();
            M_CT_Affection.Text = a.HandlingTrainerAffection.ToString();
        }

        if (Entity is ISociability s)
            MT_Sociability.Text = Math.Min(byte.MaxValue, s.Sociability).ToString();

        if (Entity is ITrainerMemories m)
        {
            // Load the OT Memories
            CB_OTQual.SelectedIndex = m.OriginalTrainerMemoryIntensity;
            CB_OTMemory.SelectedValue = (int)m.OriginalTrainerMemory;
            CB_OTVar.SelectedValue = (int)m.OriginalTrainerMemoryVariable;
            CB_OTFeel.SelectedIndex = m.OriginalTrainerMemoryFeeling;

            // Load the HT Memories
            CB_CTQual.SelectedIndex = m.HandlingTrainerMemoryIntensity;
            CB_CTMemory.SelectedValue = (int)m.HandlingTrainerMemory;
            CB_CTVar.SelectedValue = (int)m.HandlingTrainerMemoryVariable;
            CB_CTFeel.SelectedIndex = m.HandlingTrainerMemoryFeeling;
        }

        CB_Handler.Items.Clear();
        CB_Handler.Items.Add($"{Entity.OriginalTrainerName} ({TextArgs.OT})"); // OTNAME : OT

        if (!string.IsNullOrEmpty(Entity.HandlingTrainerName))
            CB_Handler.Items.Add(Entity.HandlingTrainerName);
        else
            Entity.CurrentHandler = 0;

        tabControl1.SelectedIndex = CB_Handler.SelectedIndex = Entity.CurrentHandler;

        GB_M_OT.Enabled = GB_M_CT.Enabled = GB_Residence.Enabled =
            BTN_Save.Enabled = M_Fullness.Enabled = M_Enjoyment.Enabled =
                L_Sociability.Enabled = MT_Sociability.Enabled =
                    L_Fullness.Enabled = L_Enjoyment.Enabled = Entity is not { IsEgg: true, IsUntraded: true, HandlingTrainerFriendship: 0 };

        if (!Entity.IsEgg)
        {
            bool enable;
            if (Entity.Generation < 6)
            {
                // Previous Generation Mon
                GB_M_OT.Text = $"{TextArgs.PastGen} {Entity.OriginalTrainerName}: {TextArgs.OT}"; // Past Gen OT : OTNAME
                GB_M_CT.Text = $"{TextArgs.MemoriesWith} {Entity.HandlingTrainerName}"; // Memories with : HTNAME
                enable = false;
                // Reset to no memory -- don't reset affection as OR/AS can raise it (+20 * n) via Contests
                CB_OTQual.SelectedIndex = CB_OTFeel.SelectedIndex = 0;
                CB_OTVar.SelectedValue = CB_OTMemory.SelectedValue = 0;
            }
            else
            {
                enable = true;
                GB_M_OT.Text = $"{TextArgs.MemoriesWith} {Entity.OriginalTrainerName} ({TextArgs.OT})"; // Memories with : OTNAME
                GB_M_CT.Text = $"{TextArgs.MemoriesWith} {Entity.HandlingTrainerName}"; // Memories with : HTNAME
                if (Entity.HandlingTrainerName.Length == 0)
                {
                    CB_Country1.Enabled = CB_Country2.Enabled = CB_Country3.Enabled = CB_Country4.Enabled =
                        CB_Region1.Enabled = CB_Region2.Enabled = CB_Region3.Enabled = CB_Region4.Enabled =
                            GB_M_CT.Enabled = false;
                    GB_M_CT.Text = $"{TextArgs.NeverLeft} {TextArgs.OT} - {TextArgs.Disabled}"; // Never Left : OT : Disabled
                }
                else
                {
                    GB_M_CT.Text = $"{TextArgs.MemoriesWith} {Entity.HandlingTrainerName}";
                }
            }
            RTB_OT.Visible = CB_OTQual.Enabled = CB_OTMemory.Enabled = CB_OTFeel.Enabled = CB_OTVar.Enabled = enable;
            M_OT_Affection.Enabled = true;
        }
        else
        {
            GB_M_OT.Text = GB_M_CT.Text = $"N/A: {GameInfo.Strings.EggName}";
        }

        init = true;

        // Manually load the Memory Parse
        RTB_CT.Text = GetMemoryString(CB_CTMemory, CB_CTVar, CB_CTQual, CB_CTFeel, Entity.HandlingTrainerName);
        RTB_OT.Text = GetMemoryString(CB_OTMemory, CB_OTVar, CB_OTQual, CB_OTFeel, Entity.OriginalTrainerName);

        // Affection no longer stored in Gen8+, so only show in Gen6/7.
        L_OT_Affection.Visible = L_CT_Affection.Visible = M_OT_Affection.Visible = M_CT_Affection.Visible = Entity.Format <= 7;
        L_Sociability.Visible = MT_Sociability.Visible = Entity is ISociability;
    }

    private void SaveFields()
    {
        // Save Region & Country Data
        if (Entity is IGeoTrack g)
        {
            g.Geo1_Region = (byte)WinFormsUtil.GetIndex(CB_Region0);
            g.Geo2_Region = (byte)WinFormsUtil.GetIndex(CB_Region1);
            g.Geo3_Region = (byte)WinFormsUtil.GetIndex(CB_Region2);
            g.Geo4_Region = (byte)WinFormsUtil.GetIndex(CB_Region3);
            g.Geo5_Region = (byte)WinFormsUtil.GetIndex(CB_Region4);
            g.Geo1_Country = (byte)WinFormsUtil.GetIndex(CB_Country0);
            g.Geo2_Country = (byte)WinFormsUtil.GetIndex(CB_Country1);
            g.Geo3_Country = (byte)WinFormsUtil.GetIndex(CB_Country2);
            g.Geo4_Country = (byte)WinFormsUtil.GetIndex(CB_Country3);
            g.Geo5_Country = (byte)WinFormsUtil.GetIndex(CB_Country4);
        }

        // Save 0-255 stats
        Entity.HandlingTrainerFriendship = (byte)Util.ToInt32(M_CT_Friendship.Text);
        Entity.OriginalTrainerFriendship = (byte)Util.ToInt32(M_OT_Friendship.Text);

        if (Entity is IAffection a)
        {
            a.OriginalTrainerAffection = (byte)Util.ToInt32(M_OT_Affection.Text);
            a.HandlingTrainerAffection = (byte)Util.ToInt32(M_CT_Affection.Text);
        }
        Entity.Fullness = (byte)Util.ToInt32(M_Fullness.Text);
        Entity.Enjoyment = (byte)Util.ToInt32(M_Enjoyment.Text);

        // Save Memories
        if (Entity is ITrainerMemories m)
        {
            m.OriginalTrainerMemory = (byte)WinFormsUtil.GetIndex(CB_OTMemory);
            m.OriginalTrainerMemoryVariable = CB_OTVar.Enabled ? (ushort)WinFormsUtil.GetIndex(CB_OTVar) : (ushort)0;
            m.OriginalTrainerMemoryIntensity = CB_OTFeel.Enabled ? (byte)CB_OTQual.SelectedIndex : (byte)0;
            m.OriginalTrainerMemoryFeeling = CB_OTFeel.Enabled ? (byte)CB_OTFeel.SelectedIndex : (byte)0;

            m.HandlingTrainerMemory = (byte)WinFormsUtil.GetIndex(CB_CTMemory);
            m.HandlingTrainerMemoryVariable = CB_CTVar.Enabled ? (ushort)WinFormsUtil.GetIndex(CB_CTVar) : (ushort)0;
            m.HandlingTrainerMemoryIntensity = CB_CTFeel.Enabled ? (byte)CB_CTQual.SelectedIndex : (byte)0;
            m.HandlingTrainerMemoryFeeling = CB_CTFeel.Enabled ? (byte)CB_CTFeel.SelectedIndex : (byte)0;
        }

        if (Entity is G8PKM pk8)
            pk8.Sociability = (byte)Util.ToInt32(MT_Sociability.Text);
    }

    // Event Actions
    private void B_Save_Click(object sender, EventArgs e)
    {
        SaveFields();
        Close();
    }

    private void B_Cancel_Click(object sender, EventArgs e)
    {
        Close();
    }

    private void GetLangStrings()
    {
        var strings = MemStrings;
        CB_OTMemory.InitializeBinding();
        CB_CTMemory.InitializeBinding();
        CB_OTMemory.DataSource = new BindingSource(strings.Memory, string.Empty);
        CB_CTMemory.DataSource = new BindingSource(strings.Memory, string.Empty);

        // Quality Chooser
        AddIntensity(this, strings.Species[0].Text); // None
        foreach (var q in strings.GetMemoryQualities()[1..])
            AddIntensity(this, q);

        // Feeling Chooser
        foreach (var q in strings.GetMemoryFeelings(Entity.Generation))
            CB_OTFeel.Items.Add(q);
        foreach (var q in strings.GetMemoryFeelings(Entity.Format))
            CB_CTFeel.Items.Add(q);

        // Same for each game.
        static void AddIntensity(MemoryAmie f, string line)
        {
            f.CB_CTQual.Items.Add(line);
            f.CB_OTQual.Items.Add(line);
        }
    }

    private void UpdateMemoryDisplay(object sender)
    {
        if (sender == CB_OTMemory)
        {
            int memoryGen = Entity.Generation;
            if (memoryGen == 0)
                memoryGen = Entity.Format;

            var memory = (byte)WinFormsUtil.GetIndex((ComboBox)sender);
            var memIndex = Memories.GetMemoryArgType(memory, memoryGen);
            var args = MemStrings.GetArgumentStrings(memIndex, memoryGen);
            CB_OTVar.InitializeBinding();
            CB_OTVar.DataSource = new BindingSource(args, string.Empty);
            LOTV.Text = TextArgs.GetMemoryCategory(memIndex, memoryGen);
            LOTV.Visible = CB_OTVar.Visible = CB_OTVar.Enabled = args.Count > 1;
        }
        else
        {
            int memoryGen = Entity.Format;
            var memory = (byte)WinFormsUtil.GetIndex((ComboBox)sender);
            var memIndex = Memories.GetMemoryArgType(memory, memoryGen);
            var argvals = MemStrings.GetArgumentStrings(memIndex, memoryGen);
            CB_CTVar.InitializeBinding();
            CB_CTVar.DataSource = new BindingSource(argvals, string.Empty);
            LCTV.Text = TextArgs.GetMemoryCategory(memIndex, memoryGen);
            LCTV.Visible = CB_CTVar.Visible = CB_CTVar.Enabled = argvals.Count > 1;
        }
    }

    private string GetMemoryString(ComboBox m, Control arg, Control q, Control f, string tr)
    {
        var messages = GameInfo.Strings.memories;
        string result;
        bool enabled;
        int mem = WinFormsUtil.GetIndex(m);
        if (mem == 0)
        {
            string nn = Entity.Nickname;
            result = string.Format(messages[0], nn);
            enabled = false;
        }
        else
        {
            var msg = (uint)mem < messages.Length ? messages[mem] : $"{mem}";
            string nn = Entity.Nickname;
            string a = arg.Text;
            result = string.Format(msg, nn, tr, a, f.Text, q.Text);
            enabled = true;
        }

        // Show labels if the memory allows for them.
        if (q == CB_CTQual)
            L_CT_Quality.Visible = L_CT_Feeling.Visible = enabled;
        else
            L_OT_Quality.Visible = L_OT_Feeling.Visible = enabled;

        // Show Quality and Feeling.
        q.Visible = q.Enabled = f.Visible = f.Enabled = enabled;

        return result;
    }

    private void ChangeMemory(object sender, EventArgs e)
    {
        ComboBox m = (ComboBox)sender;
        if (m == CB_CTMemory || m == CB_OTMemory)
            UpdateMemoryDisplay(m);

        if (!init)
            return;
        RTB_OT.Text = GetMemoryString(CB_OTMemory, CB_OTVar, CB_OTQual, CB_OTFeel, Entity.OriginalTrainerName);
        RTB_CT.Text = GetMemoryString(CB_CTMemory, CB_CTVar, CB_CTQual, CB_CTFeel, Entity.HandlingTrainerName);
    }

    private void ChangeCountryIndex(object sender, EventArgs e)
    {
        if (sender is not ComboBox cb)
            return;
        int index = Array.IndexOf(PrevCountries, cb);
        int val = WinFormsUtil.GetIndex(cb);
        if (val > 0)
        {
            Main.SetCountrySubRegion(PrevRegions[index], $"sr_{val:000}");
            PrevRegions[index].Enabled = true;
        }
        else
        {
            PrevRegions[index].DataSource = new[] { new ComboItem("", 0) };
            PrevRegions[index].Enabled = false;
            PrevRegions[index].SelectedValue = 0;
        }
    }

    private void ChangeCountryText(object sender, EventArgs e)
    {
        if (sender is not ComboBox cb || !string.IsNullOrWhiteSpace(cb.Text))
            return;
        cb.SelectedValue = 0;
        ChangeCountryIndex(sender, e);
    }

    private void Update255_MTB(object sender, EventArgs e)
    {
        if (sender is not MaskedTextBox tb)
            return;
        if (Util.ToInt32(tb.Text) > byte.MaxValue)
            tb.Text = "255";
    }

    private void ClickResetLocation(object sender, EventArgs e)
    {
        if (sender is not Label l)
            return;
        Label[] labels = [L_Geo0, L_Geo1, L_Geo2, L_Geo3, L_Geo4];
        int index = Array.IndexOf(labels, l);
        if (index < 0)
            return;
        PrevCountries[index].SelectedValue = 0;

        PrevRegions[index].InitializeBinding();
        PrevRegions[index].DataSource = new[] { new ComboItem("", 0) };
        PrevRegions[index].SelectedValue = 0;
    }

    private void B_ClearAll_Click(object sender, EventArgs e)
    {
        for (int i = 0; i < 5; i++)
            PrevCountries[i].SelectedValue = 0;
    }

    private sealed class TextMarkup
    {
        public string Disabled { get; } = nameof(Disabled);
        public string NeverLeft { get; } = "Never left";
        public string OT { get; } = "OT";
        public string PastGen { get; } = "Past Gen";
        public string MemoriesWith { get; } = "Memories with";

        private string Species { get; } = "Species:";
        private string Area { get; } = "Area:";
        private string Item { get; } = "Item:";
        private string Move { get; } = "Move:";
        private string Location { get; } = "Location:";

        public TextMarkup(string[] args)
        {
            Array.Resize(ref args, 10);
            if (!string.IsNullOrWhiteSpace(args[0])) Disabled = args[0];
            if (!string.IsNullOrWhiteSpace(args[1])) NeverLeft = args[1];
            if (!string.IsNullOrWhiteSpace(args[2])) OT = args[2];
            if (!string.IsNullOrWhiteSpace(args[3])) PastGen = args[3];
            if (!string.IsNullOrWhiteSpace(args[4])) MemoriesWith = args[4];

            // Pokémon ; Area ; Item(s) ; Move ; Location
            if (!string.IsNullOrWhiteSpace(args[5])) Species = args[5] + ":";
            if (!string.IsNullOrWhiteSpace(args[6])) Area = args[6] + ":";
            if (!string.IsNullOrWhiteSpace(args[7])) Item = args[7] + ":";
            if (!string.IsNullOrWhiteSpace(args[8])) Move = args[8] + ":";
            if (!string.IsNullOrWhiteSpace(args[9])) Location = args[9] + ":";
        }

        public string GetMemoryCategory(MemoryArgType type, int memoryGen) => type switch
        {
            MemoryArgType.GeneralLocation => Area,
            MemoryArgType.SpecificLocation when memoryGen <= 7 => Location,
            MemoryArgType.Species => Species,
            MemoryArgType.Move => Move,
            MemoryArgType.Item => Item,
            _ => string.Empty,
        };
    }
}