ModernUO/Projects/UOContent.Tests/Tests/WorldSaves/ManagedArchiveTests.cs

using System;
using System.Formats.Tar;
using System.IO;
using System.IO.Compression;
using Server.Compression;
using Xunit;

namespace Server.Tests;

[Collection("Sequential UOContent Tests")]
public class ManagedArchiveTests : IDisposable
{
    private readonly string _testDir;

    public ManagedArchiveTests()
    {
        _testDir = Path.Combine(Path.GetTempPath(), $"modernuo-test-{Guid.NewGuid():N}");
        Directory.CreateDirectory(_testDir);
    }

    public void Dispose()
    {
        if (Directory.Exists(_testDir))
        {
            Directory.Delete(_testDir, true);
        }
    }

    private string CreateTestDirectory(string name, params (string fileName, string content)[] files)
    {
        var dir = Path.Combine(_testDir, name);
        Directory.CreateDirectory(dir);

        foreach (var (fileName, content) in files)
        {
            var filePath = Path.Combine(dir, fileName);
            var fileDir = Path.GetDirectoryName(filePath);
            if (fileDir != null)
            {
                Directory.CreateDirectory(fileDir);
            }

            File.WriteAllText(filePath, content);
        }

        return dir;
    }

    [Fact]
    public void CreateTarZstd_MultipleDirectories_IncludesAll()
    {
        // Arrange
        var dir1 = CreateTestDirectory("backup1",
            ("items.bin", "items data"),
            ("items.idx", "items index")
        );
        var dir2 = CreateTestDirectory("backup2",
            ("mobiles.bin", "mobiles data"),
            ("mobiles.idx", "mobiles index")
        );

        var archivePath = Path.Combine(_testDir, "multi.tar.zst");

        // Act
        var entryCount = ManagedArchive.CreateTarZstd(
            [dir1, dir2], archivePath, _testDir, compressionLevel: 1
        );

        // Assert
        Assert.Equal(4, entryCount);
    }

    [Fact]
    public void ExtractTarZstd_RecreatesFiles()
    {
        // Arrange
        var sourceDir = CreateTestDirectory("source",
            ("data.txt", "Test data content"),
            ("config/settings.json", "{\"key\":\"value\"}")
        );

        var archivePath = Path.Combine(_testDir, "extract-test.tar.zst");
        ManagedArchive.CreateTarZstd([sourceDir], archivePath, _testDir, compressionLevel: 1);

        var extractDir = Path.Combine(_testDir, "extracted");

        // Act
        var success = ManagedArchive.ExtractTarZstd(archivePath, extractDir);

        // Assert
        Assert.True(success);
        Assert.True(Directory.Exists(extractDir));

        // Verify extracted content matches
        var extractedDataPath = Path.Combine(extractDir, "source", "data.txt");
        Assert.True(File.Exists(extractedDataPath));
        Assert.Equal("Test data content", File.ReadAllText(extractedDataPath));

        var extractedConfigPath = Path.Combine(extractDir, "source", "config", "settings.json");
        Assert.True(File.Exists(extractedConfigPath));
        Assert.Equal("{\"key\":\"value\"}", File.ReadAllText(extractedConfigPath));
    }

    [Fact]
    public void CountEntries_MatchesCreateCount()
    {
        // Arrange
        var sourceDir = CreateTestDirectory("source",
            ("a.txt", "a"),
            ("b.txt", "b"),
            ("c/d.txt", "d"),
            ("c/e.txt", "e")
        );

        var archivePath = Path.Combine(_testDir, "count-test.tar.zst");
        var createCount = ManagedArchive.CreateTarZstd(
            [sourceDir], archivePath, _testDir, compressionLevel: 1
        );

        // Act
        var verifyCount = ManagedArchive.CountEntries(archivePath);

        // Assert
        Assert.Equal(createCount, verifyCount);
    }

    [Fact]
    public void CreateTarZstd_EmptyDirectory_ReturnsZero()
    {
        // Arrange
        var sourceDir = Path.Combine(_testDir, "empty");
        Directory.CreateDirectory(sourceDir);

        var archivePath = Path.Combine(_testDir, "empty.tar.zst");

        // Act
        var entryCount = ManagedArchive.CreateTarZstd(
            [sourceDir], archivePath, _testDir, compressionLevel: 1
        );

        // Assert
        Assert.Equal(0, entryCount);
    }

    [Fact]
    public void CreateTarZstd_NonexistentSource_SkipsGracefully()
    {
        // Arrange
        var archivePath = Path.Combine(_testDir, "nonexistent.tar.zst");

        // Act
        var entryCount = ManagedArchive.CreateTarZstd(
            [Path.Combine(_testDir, "does-not-exist")], archivePath, _testDir, compressionLevel: 1
        );

        // Assert
        Assert.Equal(0, entryCount);
    }

    [Fact]
    public void CreateTarZstd_TempFileCleanedUp()
    {
        // Arrange
        var sourceDir = CreateTestDirectory("source", ("file.txt", "content"));
        var archivePath = Path.Combine(_testDir, "cleanup-test.tar.zst");

        // Act
        ManagedArchive.CreateTarZstd([sourceDir], archivePath, _testDir, compressionLevel: 1);

        // Assert - no .tmp file should remain
        Assert.False(File.Exists(archivePath + ".tmp"));
        Assert.True(File.Exists(archivePath));
    }

    [Fact]
    public void ExtractTarZstd_InvalidFile_ReturnsFalse()
    {
        // Arrange
        var badFile = Path.Combine(_testDir, "bad.tar.zst");
        File.WriteAllText(badFile, "this is not a valid archive");
        var extractDir = Path.Combine(_testDir, "extract-bad");

        // Act
        var result = ManagedArchive.ExtractTarZstd(badFile, extractDir);

        // Assert
        Assert.False(result);
    }

    [Fact]
    public void CountEntries_InvalidFile_ReturnsNegative()
    {
        // Arrange
        var badFile = Path.Combine(_testDir, "bad.tar.zst");
        File.WriteAllText(badFile, "not a valid archive");

        // Act
        var count = ManagedArchive.CountEntries(badFile);

        // Assert
        Assert.Equal(-1, count);
    }

    [Fact]
    public void ExtractTarGz_RoundTrips()
    {
        // Arrange — create a .tar.gz using built-in .NET APIs
        var sourceDir = CreateTestDirectory("gz-source",
            ("data.txt", "GZip test data"),
            ("sub/nested.txt", "Nested content")
        );

        var archivePath = Path.Combine(_testDir, "test.tar.gz");

        using (var fileStream = new FileStream(archivePath, FileMode.Create))
        using (var gzStream = new GZipStream(fileStream, System.IO.Compression.CompressionLevel.Fastest))
        using (var tarWriter = new TarWriter(gzStream, TarEntryFormat.Pax, leaveOpen: true))
        {
            foreach (var file in new DirectoryInfo(sourceDir).EnumerateFiles("*", SearchOption.AllDirectories))
            {
                var entryName = Path.GetRelativePath(sourceDir, file.FullName).Replace('\\', '/');
                tarWriter.WriteEntry(file.FullName, entryName);
            }
        }

        var extractDir = Path.Combine(_testDir, "gz-extracted");

        // Act
        var success = ManagedArchive.ExtractTarGz(archivePath, extractDir);

        // Assert
        Assert.True(success);
        Assert.Equal("GZip test data", File.ReadAllText(Path.Combine(extractDir, "data.txt")));
        Assert.Equal("Nested content", File.ReadAllText(Path.Combine(extractDir, "sub", "nested.txt")));
    }

    [Fact]
    public void ExtractTar_PlainTarRoundTrips()
    {
        // Arrange — create a plain .tar using built-in .NET APIs
        var sourceDir = CreateTestDirectory("tar-source",
            ("plain.txt", "Plain tar test data")
        );

        var archivePath = Path.Combine(_testDir, "test.tar");

        using (var fileStream = new FileStream(archivePath, FileMode.Create))
        using (var tarWriter = new TarWriter(fileStream, TarEntryFormat.Pax, leaveOpen: true))
        {
            foreach (var file in new DirectoryInfo(sourceDir).EnumerateFiles("*", SearchOption.AllDirectories))
            {
                var entryName = Path.GetRelativePath(sourceDir, file.FullName).Replace('\\', '/');
                tarWriter.WriteEntry(file.FullName, entryName);
            }
        }

        var extractDir = Path.Combine(_testDir, "tar-extracted");

        // Act
        var success = ManagedArchive.ExtractTar(archivePath, extractDir);

        // Assert
        Assert.True(success);
        Assert.Equal("Plain tar test data", File.ReadAllText(Path.Combine(extractDir, "plain.txt")));
    }

    [Theory]
    [InlineData("../escaped.txt")]
    [InlineData("sub/../../escaped.txt")]
    public void ExtractTar_EntryOutsideOutputDirectory_IsRejected(string entryName)
    {
        var archivePath = Path.Combine(_testDir, "traversal.tar");

        using (var fileStream = new FileStream(archivePath, FileMode.Create))
        using (var tarWriter = new TarWriter(fileStream, TarEntryFormat.Pax, leaveOpen: true))
        {
            var entry = new PaxTarEntry(TarEntryType.RegularFile, entryName)
            {
                DataStream = new MemoryStream("escaped"u8.ToArray())
            };
            tarWriter.WriteEntry(entry);
        }

        var extractDir = Path.Combine(_testDir, "out", "extract");

        Assert.False(ManagedArchive.ExtractTar(archivePath, extractDir));
        Assert.False(File.Exists(Path.Combine(_testDir, "out", "escaped.txt")));
        Assert.False(File.Exists(Path.Combine(_testDir, "escaped.txt")));
    }

    [Fact]
    public void ExtractTar_RootedEntry_IsRejected()
    {
        var archivePath = Path.Combine(_testDir, "rooted.tar");
        var target = Path.Combine(_testDir, "rooted-escaped.txt");

        using (var fileStream = new FileStream(archivePath, FileMode.Create))
        using (var tarWriter = new TarWriter(fileStream, TarEntryFormat.Pax, leaveOpen: true))
        {
            var entry = new PaxTarEntry(TarEntryType.RegularFile, target)
            {
                DataStream = new MemoryStream("escaped"u8.ToArray())
            };
            tarWriter.WriteEntry(entry);
        }

        Assert.False(ManagedArchive.ExtractTar(archivePath, Path.Combine(_testDir, "rooted-extract")));
        Assert.False(File.Exists(target));
    }

    [Fact]
    public void ExtractTarGz_InvalidFile_ReturnsFalse()
    {
        // Arrange
        var badFile = Path.Combine(_testDir, "bad.tar.gz");
        File.WriteAllText(badFile, "not a valid gzip archive");
        var extractDir = Path.Combine(_testDir, "extract-bad-gz");

        // Act & Assert
        Assert.False(ManagedArchive.ExtractTarGz(badFile, extractDir));
    }
}