From 5fed81d58c5d4b28658638c02c0ff7b36825abd5 Mon Sep 17 00:00:00 2001
From: Archangel <jesse@archangel.wtf>
Date: Tue, 26 Aug 2025 19:19:34 +0200
Subject: [PATCH] WIP: Add new postbuild script

---
 .../SPTarkov.Server.Assets/Bootstrap.ps1      |  3 +
 .../SPTarkov.Server.Assets.csproj             |  4 +
 .../SPTarkov.Server.Assets/build/PostBuild.cs | 86 +++++++++++++++++++
 3 files changed, 93 insertions(+)
 create mode 100644 Libraries/SPTarkov.Server.Assets/Bootstrap.ps1
 create mode 100644 Libraries/SPTarkov.Server.Assets/build/PostBuild.cs

diff --git a/Libraries/SPTarkov.Server.Assets/Bootstrap.ps1 b/Libraries/SPTarkov.Server.Assets/Bootstrap.ps1
new file mode 100644
index 00000000..1d1a9ce8
--- /dev/null
+++ b/Libraries/SPTarkov.Server.Assets/Bootstrap.ps1
@@ -0,0 +1,3 @@
+# This is required as dotnet run app.cs doesn't work yet from the same directory as the csproj
+cd build
+dotnet run PostBuild.cs
diff --git a/Libraries/SPTarkov.Server.Assets/SPTarkov.Server.Assets.csproj b/Libraries/SPTarkov.Server.Assets/SPTarkov.Server.Assets.csproj
index c2716512..58470343 100644
--- a/Libraries/SPTarkov.Server.Assets/SPTarkov.Server.Assets.csproj
+++ b/Libraries/SPTarkov.Server.Assets/SPTarkov.Server.Assets.csproj
@@ -16,6 +16,7 @@
     <GenerateDocumentationFile>true</GenerateDocumentationFile>
   </PropertyGroup>
   <ItemGroup>
+    <None Include="build\PostBuild.cs" />
     <None Include="build\SPTarkov.Server.Assets.targets" Pack="true" PackagePath="build\" />
     <Content Include="SPT_Data\**">
       <CopyToOutputDirectory>Always</CopyToOutputDirectory>
@@ -27,6 +28,9 @@
   <Target Name="PostBuildHashFile" AfterTargets="Build" Condition="'$(Configuration)' == 'Release'">
     <Exec Command="pwsh -NoProfile -ExecutionPolicy Bypass -File &quot;$(ProjectDir)PostBuild.ps1&quot;" />
   </Target>
+  <ItemGroup>
+    <Compile Remove="build\PostBuild.cs" />
+  </ItemGroup>
   <ItemGroup>
     <None Include="..\..\LICENSE" Pack="true" Visible="false" PackagePath="" />
   </ItemGroup>
diff --git a/Libraries/SPTarkov.Server.Assets/build/PostBuild.cs b/Libraries/SPTarkov.Server.Assets/build/PostBuild.cs
new file mode 100644
index 00000000..00302745
--- /dev/null
+++ b/Libraries/SPTarkov.Server.Assets/build/PostBuild.cs
@@ -0,0 +1,86 @@
+using System.Formats.Tar;
+using System.IO.Compression;
+using System.Security.Cryptography;
+using System.Text;
+using System.Text.Json;
+using System.Text.Json.Serialization.Metadata;
+
+string scriptDir = Path.GetFullPath(Path.Combine(Directory.GetCurrentDirectory(), ".."));
+string sptDataPath = Path.Combine(scriptDir, "SPT_Data");
+string outputFile = Path.Combine(sptDataPath, "checks.dat");
+
+GenerateHashes();
+
+void GenerateHashes()
+{
+    // Get all files recursively, excluding the 'images' directory
+    string imagesPath = Path.Combine(sptDataPath, "images");
+    var files = Directory
+        .GetFiles(sptDataPath, "*", SearchOption.AllDirectories)
+        .Where(file => !file.StartsWith(imagesPath, StringComparison.OrdinalIgnoreCase))
+        .OrderBy(file => file)
+        .ToArray();
+
+    var hashes = new List<FileHash>();
+
+    using (var md5 = MD5.Create())
+    {
+        foreach (string file in files)
+        {
+            byte[] fileBytes = File.ReadAllBytes(file);
+            byte[] hashBytes = md5.ComputeHash(fileBytes);
+
+            string hashString = BitConverter.ToString(hashBytes).Replace("-", "");
+
+            string relativePath = file.Substring(sptDataPath.Length + 1).Replace('\\', '/');
+
+            hashes.Add(new FileHash { Path = relativePath, Hash = hashString });
+        }
+    }
+
+    string jsonString = JsonSerializer.Serialize(
+        hashes,
+        new JsonSerializerOptions { TypeInfoResolver = new DefaultJsonTypeInfoResolver() }
+    );
+
+    byte[] jsonBytes = Encoding.UTF8.GetBytes(jsonString);
+    string base64String = Convert.ToBase64String(jsonBytes);
+
+    File.WriteAllText(outputFile, base64String, Encoding.ASCII);
+
+    Console.WriteLine($"Hashed {hashes.Count} files");
+}
+
+class FileHash
+{
+    public string? Path { get; set; }
+    public string? Hash { get; set; }
+}
+
+class TarGz
+{
+    public static void ExtractTarGz(string tarGzPath, string destinationDirectory, bool deleteTarGzFile = false)
+    {
+        string tempTarPath = Path.GetTempFileName();
+
+        // Yes it's disgusting I know
+        using (FileStream gzipStream = File.OpenRead(tarGzPath))
+        using (FileStream tarFileStream = File.Create(tempTarPath))
+        using (GZipStream decompressionStream = new GZipStream(gzipStream, CompressionMode.Decompress))
+        {
+            decompressionStream.CopyTo(tarFileStream);
+        }
+
+        using (FileStream tarStream = File.OpenRead(tempTarPath))
+        {
+            TarFile.ExtractToDirectory(tarStream, destinationDirectory, overwriteFiles: true);
+        }
+
+        File.Delete(tempTarPath);
+
+        if (deleteTarGzFile)
+        {
+            File.Delete(tarGzPath);
+        }
+    }
+}