PKHeX/PKHeX.Core/Editing/Saves/Slots/SlotEditor.cs

namespace PKHeX.Core;

/// <summary>
/// Facilitates interaction with a <see cref="SaveFile"/> or other data location's slot data.
/// </summary>
public sealed class SlotEditor<T>(SaveFile SAV)
{
    public readonly SlotChangelog Changelog = new(SAV);
    public readonly SlotPublisher<T> Publisher = new();

    private void NotifySlotChanged(ISlotInfo slot, SlotTouchType type, PKM pk) => Publisher.NotifySlotChanged(slot, type, pk);

    /// <summary>
    /// Gets data from a slot.
    /// </summary>
    /// <param name="slot">Slot to retrieve from.</param>
    /// <returns>Operation succeeded or not via enum value.</returns>
    public PKM Get(ISlotInfo slot)
    {
        // Reading from a slot is always allowed.
        var pk = slot.Read(SAV);
        NotifySlotChanged(slot, SlotTouchType.Get, pk);
        return pk;
    }

    /// <summary>
    /// Sets data to a slot.
    /// </summary>
    /// <param name="slot">Slot to be set to.</param>
    /// <param name="pk">Data to set.</param>
    /// <param name="type">Type of slot action</param>
    /// <returns>Operation succeeded or not via enum value.</returns>
    public SlotTouchResult Set(ISlotInfo slot, PKM pk, SlotTouchType type = SlotTouchType.Set)
    {
        if (!slot.CanWriteTo(SAV))
            return SlotTouchResult.FailWrite;

        WriteSlot(slot, pk, type);
        return SlotTouchResult.Success;
    }

    /// <summary>
    /// Deletes a slot.
    /// </summary>
    /// <param name="slot">Slot to be deleted.</param>
    /// <returns>Operation succeeded or not via enum value.</returns>
    public SlotTouchResult Delete(ISlotInfo slot)
    {
        if (!slot.CanWriteTo(SAV))
            return SlotTouchResult.FailDelete;

        if (!DeleteSlot(slot))
            return SlotTouchResult.FailDelete;

        return SlotTouchResult.Success;
    }

    /// <summary>
    /// Swaps two slots.
    /// </summary>
    /// <param name="source">Source slot to be switched with <see cref="dest"/>.</param>
    /// <param name="dest">Destination slot to be switched with <see cref="source"/>.</param>
    /// <returns>Operation succeeded or not via enum value.</returns>
    public SlotTouchResult Swap(ISlotInfo source, ISlotInfo dest)
    {
        if (!source.CanWriteTo(SAV))
            return SlotTouchResult.FailSource;
        if (!dest.CanWriteTo(SAV))
            return SlotTouchResult.FailDestination;

        const PKMImportSetting skip = PKMImportSetting.Skip;
        var s = source.Read(SAV);
        var d = dest.Read(SAV);
        WriteSlot(source, s, SlotTouchType.None, skip);
        WriteSlot(dest, d, SlotTouchType.Swap, skip);

        return SlotTouchResult.Success;
    }

    private bool WriteSlot(ISlotInfo slot, PKM pk, SlotTouchType type = SlotTouchType.Set, PKMImportSetting setDetail = PKMImportSetting.UseDefault)
    {
        Changelog.AddNewChange(slot);
        var result = slot.WriteTo(SAV, pk, setDetail);
        if (result)
            NotifySlotChanged(slot, type, pk);
        return result;
    }

    private bool DeleteSlot(ISlotInfo slot)
    {
        var pk = SAV.BlankPKM;
        return WriteSlot(slot, pk, SlotTouchType.Delete, PKMImportSetting.Skip);
    }

    public void Undo()
    {
        if (!Changelog.CanUndo)
            return;
        var slot = Changelog.Undo();
        NotifySlotChanged(slot, SlotTouchType.Delete, slot.Read(SAV));
    }

    public void Redo()
    {
        if (!Changelog.CanRedo)
            return;
        var slot = Changelog.Redo();
        NotifySlotChanged(slot, SlotTouchType.Delete, slot.Read(SAV));
    }
}