wabbajack/Compression.BSA/TES3Reader.cs

using System;
using System.Collections.Generic;
using System.IO;
using System.Text;
using System.Threading.Tasks;
using Wabbajack.Common;
using Wabbajack.Common.Serialization.Json;
#nullable disable

namespace Compression.BSA
{
    public class TES3Reader : IBSAReader
    {
        public static string TES3_MAGIC = Encoding.ASCII.GetString(new byte[] {0, 1, 0, 0});
        private uint _versionNumber;
        private uint _hashTableOffset;
        private uint _fileCount;
        private TES3FileEntry[] _files;
        internal long _dataOffset;
        public IStreamFactory _streamFactory;

        public static async ValueTask<TES3Reader> Load(IStreamFactory factory)
        {
            await using var fs = await factory.GetStream();
            using var br = new BinaryReader(fs);
            var rdr = new TES3Reader
            {
                _streamFactory = factory,
                _versionNumber = br.ReadUInt32(),
                _hashTableOffset = br.ReadUInt32(),
                _fileCount = br.ReadUInt32()
            };

            rdr._files = new TES3FileEntry[rdr._fileCount];
            for (int i = 0; i < rdr._fileCount; i++)
            {
                var file = new TES3FileEntry {
                    Index = i,
                    Archive = rdr, 
                    Size = br.ReadUInt32(), 
                    Offset = br.ReadUInt32()
                    
                };
                rdr._files[i] = file;
            }

            for (int i = 0; i < rdr._fileCount; i++)
            {
                rdr._files[i].NameOffset = br.ReadUInt32();
            }

            var origPos = br.BaseStream.Position;
            for (int i = 0; i < rdr._fileCount; i++)
            {
                br.BaseStream.Position = origPos + rdr._files[i].NameOffset;
                rdr._files[i].Path = new RelativePath(br.ReadStringTerm(VersionType.TES3));
            }

            br.BaseStream.Position = rdr._hashTableOffset + 12;
            for (int i = 0; i < rdr._fileCount; i++)
            {
                rdr._files[i].Hash1 = br.ReadUInt32();
                rdr._files[i].Hash2 = br.ReadUInt32();
            }

            rdr._dataOffset = br.BaseStream.Position;
            return rdr;
        }

        public async ValueTask DisposeAsync()
        {
        }

        public IEnumerable<IFile> Files => _files;
        public ArchiveStateObject State
        {
            get
            {
                return new TES3ArchiveState
                {
                    FileCount = _fileCount,
                    DataOffset = _dataOffset,
                    HashOffset = _hashTableOffset,
                    VersionNumber = _versionNumber,
                    
                };
            }
        }

        public void Dump(Action<string> print)
        {
            throw new NotImplementedException();
        }
    }

    [JsonName("TES3Archive")]
    public class TES3ArchiveState : ArchiveStateObject
    {
        public uint FileCount { get; set; }
        public long DataOffset { get; set; }
        public uint HashOffset { get; set; }
        public uint VersionNumber { get; set; }

        public override async Task<IBSABuilder> MakeBuilder(long size)
        {
            return new TES3Builder(this);
        }
    }

    public class TES3FileEntry : IFile
    {
        public RelativePath Path { get; set;  }
        public uint Size { get; set; }
        public FileStateObject State =>
            new TES3FileState
            {
                Index = Index,
                Path = Path,
                Size = Size,
                Offset = Offset,
                NameOffset = NameOffset,
                Hash1 = Hash1,
                Hash2 = Hash2
            };

        public async ValueTask CopyDataTo(Stream output)
        {
            await using var fs = await Archive._streamFactory.GetStream();
            fs.Position = Archive._dataOffset + Offset;
            await fs.CopyToLimitAsync(output, (int)Size);
        }

        public async ValueTask<IStreamFactory> GetStreamFactory()
        {
            var ms = new MemoryStream();
            await CopyDataTo(ms);
            ms.Position = 0;
            return new MemoryStreamFactory(ms);
        }


        
        public void Dump(Action<string> print)
        {
            throw new NotImplementedException();
        }


        public uint Offset { get; set; }
        public uint NameOffset { get; set; }
        public uint Hash1 { get; set; }
        public uint Hash2 { get; set; }
        public TES3Reader Archive { get; set; }
        public int Index { get; set; }

    }
    
    
    [JsonName("TES3FileState")]
    public class TES3FileState : FileStateObject
    {
        public uint Offset { get; set; }
        public uint NameOffset { get; set; }
        public uint Hash1 { get; set; }
        public uint Hash2 { get; set; }
        public uint Size { get; set; }
    }


}