PKHeX/PKHeX.Core/Saves/Util/Recognition/SaveHandlerGCI.cs

using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;

namespace PKHeX.Core;

/// <summary>
/// Logic for recognizing .gci save files.
/// </summary>
public sealed class SaveHandlerGCI : ISaveHandler
{
    private const int headerSize = 0x40;
    private const int SIZE_G3BOXGCI  = headerSize + SaveUtil.SIZE_G3BOX; // GCI data
    private const int SIZE_G3COLOGCI = headerSize + SaveUtil.SIZE_G3COLO; // GCI data
    private const int SIZE_G3XDGCI   = headerSize + SaveUtil.SIZE_G3XD; // GCI data

    private static readonly string[] HEADER_COLO  = { "GC6J", "GC6E", "GC6P" }; // NTSC-J, NTSC-U, PAL
    private static readonly string[] HEADER_XD    = { "GXXJ", "GXXE", "GXXP" }; // NTSC-J, NTSC-U, PAL
    private static readonly string[] HEADER_RSBOX = { "GPXJ", "GPXE", "GPXP" }; // NTSC-J, NTSC-U, PAL

    private static bool IsGameMatchHeader(IEnumerable<string> headers, byte[] data) => headers.Contains(Encoding.ASCII.GetString(data, 0, 4));

    public bool IsRecognized(int size) => size is SIZE_G3BOXGCI or SIZE_G3COLOGCI or SIZE_G3XDGCI;

    public SaveHandlerSplitResult? TrySplit(byte[] input)
    {
        switch (input.Length)
        {
            case SIZE_G3COLOGCI when IsGameMatchHeader(HEADER_COLO , input):
            case SIZE_G3XDGCI   when IsGameMatchHeader(HEADER_XD   , input):
            case SIZE_G3BOXGCI  when IsGameMatchHeader(HEADER_RSBOX, input):
                break;
            default:
                return null;
        }

        byte[] header = input.Slice(0, headerSize);
        input = input.SliceEnd(headerSize);

        return new SaveHandlerSplitResult(input, header, Array.Empty<byte>());
    }

    /// <summary>
    /// Checks if the game code is one of the recognizable versions.
    /// </summary>
    /// <param name="gameCode">4 character game code string</param>
    /// <returns>Magic version ID enumeration; <see cref="GameVersion.Unknown"/> if no match.</returns>
    public static GameVersion GetGameCode(string gameCode)
    {
        if (HEADER_COLO.Contains(gameCode))
            return GameVersion.COLO;
        if (HEADER_XD.Contains(gameCode))
            return GameVersion.XD;
        if (HEADER_RSBOX.Contains(gameCode))
            return GameVersion.RSBOX;

        return GameVersion.Unknown;
    }
}