PKHeX/PKHeX.WinForms/Controls/SAV Editor/ContextMenuSAV.cs

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

using static PKHeX.Core.MessageStrings;

namespace PKHeX.WinForms.Controls;

public partial class ContextMenuSAV : UserControl
{
    public ContextMenuSAV()
    {
        InitializeComponent();
        if (Application.IsDarkModeEnabled)
            WinFormsUtil.InvertToolStripIcons(mnuVSD.Items);
    }

    public SaveDataEditor<PictureBox> Editor { private get; set; } = null!;
    public required SlotChangeManager Manager { get; init; }

    public Action<LegalityAnalysis>? RequestEditorLegality;

    public void OmniClick(object sender, EventArgs e, Keys z)
    {
        switch (z)
        {
            case Keys.Control: ClickView(sender, e); break;
            case Keys.Shift: ClickSet(sender, e); break;
            case Keys.Alt: ClickDelete(sender, e); break;
            default:
                return;
        }

        // restart hovering since the mouse event isn't fired
        Manager.MouseEnter(sender, e);
    }

    private void ClickView(object sender, EventArgs e)
    {
        var info = GetSenderInfo(sender);
        if (info.IsEmpty())
        { WinFormsUtil.Asterisk(); return; }

        Manager.Hover.Stop();
        var pk = Editor.Slots.Get(info.Slot);
        Editor.PKMEditor.PopulateFields(pk, false, true);
    }

    private void ClickSet(object sender, EventArgs e)
    {
        var editor = Editor.PKMEditor;
        if (!editor.EditsComplete)
            return;
        PKM pk = editor.PreparePKM();
        var preModify = pk.Clone();

        var info = GetSenderInfo(sender);
        var sav = info.View.SAV;

        if (!CheckDest(info, sav, pk))
            return;

        var errata = sav.EvaluateCompatibility(pk);
        if (errata.Count != 0)
        {
            var msg = string.Join(Environment.NewLine, errata);
            var prompt = WinFormsUtil.Prompt(MessageBoxButtons.YesNo, msg, MsgContinue);
            if (prompt != DialogResult.Yes)
                return;
        }

        editor.NotifyWasExported(preModify);
        Manager.Hover.Stop();
        Editor.Slots.Set(info.Slot, pk);
        Manager.SE.UpdateUndoRedo();
    }

    private void ClickDelete(object sender, EventArgs e)
    {
        var info = GetSenderInfo(sender);
        if (info.IsEmpty())
        { WinFormsUtil.Asterisk(); return; }

        var sav = info.View.SAV;
        var pk = sav.BlankPKM;
        if (!CheckDest(info, sav, pk))
            return;

        Manager.Hover.Stop();
        Editor.Slots.Delete(info.Slot);
        Manager.SE.UpdateUndoRedo();
    }

    private static bool CheckDest(SlotViewInfo<PictureBox> info, SaveFile sav, PKM pk)
    {
        var msg = info.Slot.CanWriteTo(sav, pk);
        if (msg == WriteBlockedMessage.None)
            return true;

        switch (msg)
        {
            case WriteBlockedMessage.InvalidPartyConfiguration:
                WinFormsUtil.Alert(MsgSaveSlotEmpty);
                break;
            case WriteBlockedMessage.IncompatibleFormat:
                break;
            case WriteBlockedMessage.InvalidDestination:
                WinFormsUtil.Alert(MsgSaveSlotLocked);
                break;
            default:
                throw new IndexOutOfRangeException(nameof(msg));
        }
        return false;
    }

    private void ClickShowLegality(object sender, EventArgs e)
    {
        var info = GetSenderInfo(sender);
        var sav = info.View.SAV;
        var pk = info.Slot.Read(sav);
        var type = info.Slot.Type;
        var la = new LegalityAnalysis(pk, sav.Personal, type);
        RequestEditorLegality?.Invoke(la);
    }

    private void MenuOpening(object sender, CancelEventArgs e)
    {
        var info = GetSenderInfo(sender);
        bool canView = !info.IsEmpty() || Main.HaX;
        bool canSet = info.CanWriteTo();
        bool canDelete = canSet && canView;
        bool canLegality = (ModifierKeys == Keys.Control || Main.Settings.Display.SlotLegalityAlwaysVisible) && canView && RequestEditorLegality is not null;

        ToggleItem(mnuView, canView);
        ToggleItem(mnuSet, canSet);
        ToggleItem(mnuDelete, canDelete);
        ToggleItem(mnuLegality, canLegality);

        if (!canView && !canSet && !canDelete)
            e.Cancel = true;
    }

    private static SlotViewInfo<PictureBox> GetSenderInfo(object sender)
    {
        if (!WinFormsUtil.TryGetUnderlying<PictureBox>(sender, out var pb))
            ArgumentNullException.ThrowIfNull(pb);
        if (!WinFormsUtil.TryFindFirstControlOfType<ISlotViewer<PictureBox>>(pb, out var view))
            ArgumentNullException.ThrowIfNull(view);
        var loc = view.GetSlotData(pb);
        return new SlotViewInfo<PictureBox>(loc, view);
    }

    private static void ToggleItem(ToolStripItem item, bool visible)
    {
        item.Visible = visible;
    }
}