wabbajack/Wabbajack.VFS/IndexedVirtualFile.cs

using System.Collections.Generic;
using System.IO;
using System.IO.Compression;
using System.Text;
using Wabbajack.DTOs.Texture;
using Wabbajack.Hashing.xxHash64;
using Wabbajack.Paths;

namespace Wabbajack.VFS;

public class IndexedVirtualFile
{
    public IPath Name { get; set; }
    public Hash Hash { get; set; }

    public ImageState? ImageState { get; set; }
    public long Size { get; set; }
    public List<IndexedVirtualFile> Children { get; set; } = new();

    private void Write(BinaryWriter bw)
    {
        bw.Write(Name.ToString()!);
        bw.Write((ulong) Hash);

        if (ImageState == null)
        {
            bw.Write(false);
        }
        else
        {
            bw.Write(true);
            WriteImageState(bw, ImageState);
        }

        bw.Write(Size);
        bw.Write(Children.Count);
        foreach (var file in Children)
            file.Write(bw);
    }

    private void WriteImageState(BinaryWriter bw, ImageState state)
    {
        bw.Write((ushort) state.Width);
        bw.Write((ushort) state.Height);
        bw.Write((byte) state.Format);
        state.PerceptualHash.Write(bw);
    }

    public static ImageState ReadImageState(BinaryReader br)
    {
        return new ImageState
        {
            Width = br.ReadUInt16(),
            Height = br.ReadUInt16(),
            Format = (DXGI_FORMAT) br.ReadByte(),
            PerceptualHash = PHash.Read(br)
        };
    }


    public void Write(Stream s)
    {
        using var cs = new GZipStream(s, CompressionLevel.Optimal, true);
        using var bw = new BinaryWriter(cs, Encoding.UTF8, true);
        Write(bw);
    }

    private static IndexedVirtualFile Read(BinaryReader br)
    {
        var ivf = new IndexedVirtualFile
        {
            Name = (RelativePath) br.ReadString(),
            Hash = Hash.FromULong(br.ReadUInt64())
        };

        if (br.ReadBoolean())
            ivf.ImageState = ReadImageState(br);

        ivf.Size = br.ReadInt64();

        var lst = new List<IndexedVirtualFile>();
        ivf.Children = lst;
        var count = br.ReadInt32();
        for (var x = 0; x < count; x++) lst.Add(Read(br));

        return ivf;
    }

    public static IndexedVirtualFile Read(Stream s)
    {
        using var cs = new GZipStream(s, CompressionMode.Decompress, true);
        using var br = new BinaryReader(cs);
        return Read(br);
    }
}