PKHeX/PKHeX.WinForms/Subforms/Save Editors/Gen7/SAV_Trainer7GG.cs

using System;
using System.Collections.Generic;
using System.IO;
using System.Linq;
using System.Windows.Forms;
using PKHeX.Core;

namespace PKHeX.WinForms;

public partial class SAV_Trainer7GG : Form
{
    private readonly SaveFile Origin;
    private readonly SAV7b SAV;
    private readonly GoParkStorage Park;

    public SAV_Trainer7GG(SaveFile sav)
    {
        InitializeComponent();
        WinFormsUtil.TranslateInterface(this, Main.CurrentLanguage);
        SAV = (SAV7b)(Origin = sav).Clone();
        Park = new GoParkStorage(SAV);
        UpdateGoSummary(0);

        if (Main.Unicode)
        {
            TB_OTName.Font = TB_RivalName.Font = FontUtil.GetPKXFont();
        }

        B_MaxCash.Click += (sender, e) => MT_Money.Text = "9,999,999";

        GetComboBoxes();
        LoadTrainerInfo();
    }

    // Drag & Drop Events
    private void Main_DragEnter(object? sender, DragEventArgs? e)
    {
        if (e is null)
            return;
        if (e.AllowedEffect == (DragDropEffects.Copy | DragDropEffects.Link)) // external file
            e.Effect = DragDropEffects.Copy;
        else if (e.Data != null) // within
            e.Effect = DragDropEffects.Move;
    }

    private void Main_DragDrop(object? sender, DragEventArgs? e)
    {
        if (e?.Data?.GetData(DataFormats.FileDrop) is not string[] {Length: not 0} files)
            return;
        ImportGP1From(files[0]);
        e.Effect = DragDropEffects.Copy;
    }

    private void GetComboBoxes()
    {
        CB_Gender.Items.Clear();
        CB_Gender.Items.AddRange(Main.GenderSymbols.Take(2).ToArray()); // m/f depending on unicode selection
        CB_Language.InitializeBinding();
        CB_Language.DataSource = GameInfo.LanguageDataSource(SAV.Generation);
        CB_Game.InitializeBinding();
        CB_Game.DataSource = new BindingSource(GameInfo.VersionDataSource.Where(z => GameVersion.Gen7b.Contains(z.Value)).ToList(), null);
    }

    private void LoadTrainerInfo()
    {
        // Get Data
        TB_OTName.Text = SAV.OT;
        TB_RivalName.Text = SAV.Blocks.Misc.Rival;
        CB_Language.SelectedValue = SAV.Language;
        MT_Money.Text = SAV.Blocks.Misc.Money.ToString();

        CB_Game.SelectedValue = SAV.Game;
        CB_Gender.SelectedIndex = SAV.Gender;
        trainerID1.LoadIDValues(SAV, SAV.Generation);

        // Load Play Time
        MT_Hours.Text = SAV.PlayedHours.ToString();
        MT_Minutes.Text = SAV.PlayedMinutes.ToString();
        MT_Seconds.Text = SAV.PlayedSeconds.ToString();
    }

    private void Save()
    {
        SaveTrainerInfo();
    }

    private void SaveTrainerInfo()
    {
        SAV.Game = WinFormsUtil.GetIndex(CB_Game);
        SAV.Gender = (byte)CB_Gender.SelectedIndex;

        SAV.Money = Util.ToUInt32(MT_Money.Text);
        SAV.Language = WinFormsUtil.GetIndex(CB_Language);

        SAV.OT = TB_OTName.Text;
        SAV.Blocks.Misc.Rival = TB_RivalName.Text;

        // Save PlayTime
        SAV.PlayedHours = ushort.Parse(MT_Hours.Text);
        SAV.PlayedMinutes = ushort.Parse(MT_Minutes.Text)%60;
        SAV.PlayedSeconds = ushort.Parse(MT_Seconds.Text)%60;
    }

    private void ClickString(object sender, MouseEventArgs e)
    {
        if (ModifierKeys != Keys.Control)
            return;

        TextBox tb = sender as TextBox ?? TB_OTName;

        // Special Character Form
        var d = new TrashEditor(tb, SAV);
        d.ShowDialog();
        tb.Text = d.FinalString;
    }

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

    private void B_Save_Click(object sender, EventArgs e)
    {
        Save();
        Origin.CopyChangesFrom(SAV);
        Close();
    }

    private void Change255(object sender, EventArgs e)
    {
        MaskedTextBox box = (MaskedTextBox)sender;
        if (box.Text.Length == 0) box.Text = "0";
        if (Util.ToInt32(box.Text) > 255) box.Text = "255";
    }

    private void B_ExportGoSummary_Click(object sender, EventArgs e)
    {
        var summary = Park.DumpAll(GameInfo.Strings.Species).ToArray();
        if (summary.Length == 0)
        {
            WinFormsUtil.Alert("No entities present in Go Park to dump.");
            return;
        }
        WinFormsUtil.SetClipboardText(string.Join(Environment.NewLine, summary));
        System.Media.SystemSounds.Asterisk.Play();
    }

    private void B_ExportGoFiles_Click(object sender, EventArgs e)
    {
        var gofiles = Park.GetAllEntities().Where(z => z.Species != 0).ToArray();
        if (gofiles.Length == 0)
        {
            WinFormsUtil.Alert("No entities present in Go Park to dump.");
            return;
        }
        using var fbd = new FolderBrowserDialog();
        if (fbd.ShowDialog() != DialogResult.OK)
            return;

        var folder = fbd.SelectedPath;
        foreach (var gpk in gofiles)
            File.WriteAllBytes(Path.Combine(folder, Util.CleanFileName(gpk.FileName)), gpk.Data);
        WinFormsUtil.Alert($"Dumped {gofiles.Length} files to {folder}");
    }

    private void B_Import_Click(object sender, EventArgs e)
    {
        using var sfd = new OpenFileDialog
        {
            Filter = GoFilter,
            FilterIndex = 0,
            RestoreDirectory = true,
        };

        // Export
        if (sfd.ShowDialog() != DialogResult.OK)
            return;

        string path = sfd.FileName;
        ImportGP1From(path);
    }

    private void ImportGP1From(string path)
    {
        int index = (int)NUD_GoIndex.Value;
        const int max = GoParkStorage.Count - 1;
        index = Math.Clamp(index, 0, max);
        ImportGP1From(path, index);
    }

    private void ImportGP1From(string path, int index)
    {
        var data = File.ReadAllBytes(path);
        if (data.Length != GP1.SIZE)
        {
            WinFormsUtil.Error(MessageStrings.MsgFileLoadIncompatible);
            return;
        }
        var gp1 = new GP1();
        data.CopyTo(gp1.Data, 0);
        Park[index] = gp1;
        UpdateGoSummary((int)NUD_GoIndex.Value);
    }

    private const string GoFilter = "Go Park Entity |*.gp1|All Files|*.*";

    private void B_Export_Click(object sender, EventArgs e)
    {
        int index = (int)NUD_GoIndex.Value;
        const int max = GoParkStorage.Count - 1;
        index = Math.Clamp(index, 0, max);
        var data = Park[index];

        using var sfd = new SaveFileDialog
        {
            FileName = data.FileName,
            Filter = GoFilter,
            FilterIndex = 0,
            RestoreDirectory = true,
        };

        if (sfd.ShowDialog() != DialogResult.OK)
            return;

        File.WriteAllBytes(sfd.FileName, data.Data);
    }

    private void B_ImportGoFiles_Click(object sender, EventArgs e)
    {
        using var fbd = new FolderBrowserDialog();
        if (fbd.ShowDialog() != DialogResult.OK)
            return;

        IEnumerable<string> files = Directory.GetFiles(fbd.SelectedPath);
        files = files.Where(z => Path.GetExtension(z) == ".gp1" && new FileInfo(z).Length == GP1.SIZE);

        int ctr = (int)NUD_GoIndex.Value;
        foreach (var f in files)
        {
            while (true)
            {
                if (ctr >= GoParkStorage.Count)
                    return;
                if (Park[ctr].Species != 0)
                    ctr++;
                else
                    break;
            }
            var data = File.ReadAllBytes(f);
            Park[ctr] = new GP1(data);
            ctr++;
        }
        UpdateGoSummary((int) NUD_GoIndex.Value);
        System.Media.SystemSounds.Asterisk.Play();
    }

    private void NUD_GoIndex_ValueChanged(object sender, EventArgs e) => UpdateGoSummary((int)NUD_GoIndex.Value);

    private void UpdateGoSummary(int index)
    {
        const int max = GoParkStorage.Count - 1;
        index = Math.Clamp(index, 0, max);
        int area = index / GoParkStorage.SlotsPerArea;
        int slot = index % GoParkStorage.SlotsPerArea;

        var data = Park[index];
        var prefix = $"Area: {area + 1:00}, Slot: {slot + 1:00}{Environment.NewLine}";
        var dump = data.Species == 0 ? "Empty" : data.Dump(GameInfo.Strings.Species, index);
        L_GoSlotSummary.Text = prefix + dump;
    }

    private void B_DeleteAll_Click(object sender, EventArgs e)
    {
        if (WinFormsUtil.Prompt(MessageBoxButtons.YesNo, "Delete all slots?") != DialogResult.Yes)
            return;

        Park.DeleteAll();
        UpdateGoSummary((int) NUD_GoIndex.Value);
        System.Media.SystemSounds.Asterisk.Play();
    }

    private void B_DeleteGo_Click(object sender, EventArgs e)
    {
        int index = (int)NUD_GoIndex.Value;
        const int max = GoParkStorage.Count - 1;
        index = Math.Clamp(index, 0, max);
        Park[index] = new GP1();
        UpdateGoSummary((int)NUD_GoIndex.Value);
        System.Media.SystemSounds.Asterisk.Play();
    }

    private void B_AllTrainerTitles_Click(object sender, EventArgs e)
    {
        SAV.Blocks.EventWork.UnlockAllTitleFlags();
        System.Media.SystemSounds.Asterisk.Play();
    }
}