pkNX/pkNX.Containers/FakeContainer.cs

using System.IO;
using System.Threading;
using System.Threading.Tasks;

namespace pkNX.Containers;

public class FakeContainer : IFileContainer
{
    public readonly byte[][] Files;
    private readonly byte[][] Backup;

    public FakeContainer(byte[][] files)
    {
        Files = files;
        Backup = new byte[files.Length][];
        for (int i = 0; i < Backup.Length; i++)
            Backup[i] = (byte[])files[i].Clone();
    }

    public string? FilePath { get; set; } = string.Empty;
    public bool Modified { get; set; }
    public int Count => Files.Length;

    public byte[] this[int index]
    {
        get => Files[index];
        set => Files[index] = value;
    }

    public Task<byte[][]> GetFiles() => Task.FromResult(Files);
    public Task<byte[]> GetFile(int file, int subFile = 0) => Task.FromResult(this[file]);
    public Task SetFile(int file, byte[] value, int subFile = 0) => Task.FromResult(this[file] = value);
    public Task SaveAs(string path, ContainerHandler handler, CancellationToken token) => new(() => SaveAll(path, handler, token), token);

    public void SaveAll(string path, ContainerHandler handler, CancellationToken token)
    {
        handler.Initialize(Files.Length);
        for (int i = 0; i < Files.Length; i++)
        {
            if (token.IsCancellationRequested)
                return;
            File.WriteAllBytes(Path.Combine(path, $"{i}.bin"), Files[i]);
            handler.StepFile(i);
        }
    }

    public void Dump(string path, ContainerHandler handler) => SaveAs(path, handler, CancellationToken.None);

    public void CancelEdits()
    {
        for (int i = 0; i < Files.Length; i++)
            Files[i] = (byte[])Backup[i].Clone();
    }
}