PKHeX/PKHeX.Core/Items/Bags/PlayerBag3E.cs

using System;
using System.Collections.Generic;
using static PKHeX.Core.InventoryType;

namespace PKHeX.Core;

public sealed class PlayerBag3E : PlayerBag, IPlayerBag3
{
    public override IReadOnlyList<InventoryPouch3> Pouches { get; } = GetPouches(ItemStorage3E.Instance);
    public override ItemStorage3E Info => ItemStorage3E.Instance;

    private static InventoryPouch3[] GetPouches(ItemStorage3E info) =>
    [
        new(0x0C8, 30, 099, info, Items),
        new(0x140, 30, 001, info, KeyItems),
        new(0x1B8, 16, 099, info, Balls),
        new(0x1F8, 64, 099, info, TMHMs),
        new(0x2F8, 46, 999, info, Berries),
        new(0x000, 50, 999, info, PCItems),
    ];

    public PlayerBag3E(SAV3E sav) : this(sav.LargeBlock.Inventory, sav.SmallBlock.SecurityKey) { }
    public PlayerBag3E(ReadOnlySpan<byte> data, uint security)
    {
        UpdateSecurityKey(security);
        Pouches.LoadAll(data);
    }

    public override void CopyTo(SaveFile sav) => CopyTo((SAV3E)sav);
    public void CopyTo(SAV3E sav) => CopyTo(sav.LargeBlock.Inventory);
    public void CopyTo(Span<byte> data) => Pouches.SaveAll(data);

    public override int GetMaxCount(InventoryType type, int itemIndex)
    {
        if (type is TMHMs && ItemConverter.IsItemHM3((ushort)itemIndex))
            return 1;
        return GetMaxCount(type);
    }

    public void UpdateSecurityKey(uint securityKey)
    {
        foreach (var pouch in Pouches)
        {
            if (pouch.Type != PCItems)
                pouch.SecurityKey = securityKey;
        }
    }
}

/// <summary>
/// Encryption interface for player bags that utilize security keys.
/// </summary>
/// <see cref="PlayerBag3E"/>
/// <see cref="PlayerBag3FRLG"/>
public interface IPlayerBag3
{
    /// <summary>
    /// Updates the security key for all pouches that require it.
    /// </summary>
    /// <remarks>
    /// Interior item data is not modified; this only changes the key used for read/write operations.
    /// </remarks>
    /// <param name="securityKey">The new security key to use.</param>
    void UpdateSecurityKey(uint securityKey);
}