feat: rename cli

src/BlitzKit.CLI/Functions/Mangler.cs

@@ -0,0 +1,26 @@
+using BlitzKit.CLI.Models;
+
+namespace BlitzKit.CLI.Functions
+{
+  public static class Mangler
+  {
+    public static void Mangle()
+    {
+      BlitzProvider provider = new();
+
+      foreach (var nation in provider.RootDirectory.GetDirectory("Blitz/Content/Tanks").Directories)
+      {
+        if (nation.Key == "TankStub")
+          continue;
+
+        foreach (var tank in nation.Value.Directories)
+        {
+          if (!tank.Value.HasFile($"DA_{tank.Key}_Attributes.uasset"))
+          {
+            Console.WriteLine($"{nation.Key}/{tank.Key}");
+          }
+        }
+      }
+    }
+  }
+}

src/BlitzKit.CLI/Functions/Unpacker.cs

@@ -0,0 +1,201 @@
+using System.Text.Json;
+using BlitzKit.CLI.Models;
+using BlitzKit.CLI.Utils;
+using CUE4Parse.FileProvider.Objects;
+using CUE4Parse.UE4.Pak;
+using CUE4Parse.UE4.Readers;
+using DotNet.Globbing;
+
+namespace BlitzKit.CLI.Functions
+{
+  public static class Unpacker
+  {
+    private const string CONTENT_GROUP = "p14y73c7_p5e_b10gg3RS_big_play";
+    private const string VFS_PATH = "../../temp/vfs/";
+
+    // this may be different from the one provided in args so check this and not args[1]
+    // we don't wanna delete my local installation of the game when debugging haha!
+    private const string TEMP_DEPOT_DIR = "../../temp/depot/";
+    private const string WG_DLC_DOMAIN = "http://dl-wotblitz-gc.wargaming.net";
+    private const string BUILD_MANIFEST_OS = "Windows";
+
+    public static async Task Unpack(string[] args)
+    {
+      if (args.Length < 2)
+      {
+        throw new ArgumentException("Missing required initial paks directory");
+      }
+
+      UnpackDepot(args[1]);
+      await UnpackDLC();
+    }
+
+    private static async Task UnpackDLC()
+    {
+      string defaultDlcPath = Path.Combine(VFS_PATH, "Blitz/Config/DefaultDlc.ini");
+      IniParser iniParser = new(defaultDlcPath);
+      string? contentBuildId = iniParser.GetString("DefaultContentBuildId");
+
+      if (string.IsNullOrEmpty(contentBuildId))
+      {
+        throw new Exception("Failed to read content build id from DefaultDlc.ini");
+      }
+
+      Console.WriteLine($"Exploring DLC \"{contentBuildId}\" within group \"{CONTENT_GROUP}\"");
+
+      DlcManifest manifest = await FetchManifest(CONTENT_GROUP, contentBuildId);
+
+      List<PakFile> nonHdPaks = [];
+      List<PakFile> hdPaks = [];
+
+      Console.WriteLine("Filtering HD and non-HD containers");
+
+      foreach (var pakFile in manifest.PakFiles)
+      {
+        long chunkId = pakFile.ChunkId;
+        manifest.ChunkTags.TryGetValue(chunkId.ToString(), out var tags);
+
+        if (tags == null)
+        {
+          throw new Exception($"Failed to find tags for chunk {chunkId}");
+        }
+
+        bool isHd = tags.GameplayTags.Any(tag => tag.TagName == "Dlc.HdTextures");
+
+        if (isHd)
+        {
+          hdPaks.Add(pakFile);
+        }
+        else
+        {
+          nonHdPaks.Add(pakFile);
+        }
+      }
+
+      Console.WriteLine(
+        $"Found {nonHdPaks.Count} non-HD containers and {hdPaks.Count} HD containers"
+      );
+
+      Console.WriteLine($"Exergizing non-HD containers");
+
+      foreach (var pak in nonHdPaks)
+      {
+        await ExergizeManifestPakFile(pak);
+      }
+
+      Console.WriteLine($"Exergizing HD containers");
+
+      foreach (var pak in hdPaks)
+      {
+        await ExergizeManifestPakFile(pak);
+      }
+    }
+
+    private static async Task ExergizeManifestPakFile(PakFile pak)
+    {
+      Console.WriteLine($"Exergizing DLC container \"{pak.FileName}\"");
+
+      string url = $"{WG_DLC_DOMAIN}/dlc/{CONTENT_GROUP}/dlc/{pak.RelativeUrl}";
+
+      using HttpClient httpClient = new();
+      Stream httpStream = await httpClient.GetStreamAsync(url);
+      MemoryStream memoryStream = new();
+
+      await httpStream.CopyToAsync(memoryStream);
+      memoryStream.Seek(0, SeekOrigin.Begin);
+
+      FStreamArchive archive = new(url, memoryStream);
+      PakFileReader reader = new(archive);
+
+      Exergize(reader);
+
+      memoryStream.Dispose();
+      httpStream.Dispose();
+      archive.Dispose();
+    }
+
+    public static async Task<DlcManifest> FetchManifest(string group, string build)
+    {
+      string url =
+        $"{WG_DLC_DOMAIN}/dlc/{group}/dlc/BuildManifest-{BUILD_MANIFEST_OS}-{build}.json";
+      using HttpClient client = new();
+      string jsonString = await client.GetStringAsync(url);
+      DlcManifest manifest =
+        JsonSerializer.Deserialize<DlcManifest>(jsonString)
+        ?? throw new Exception("Failed to deserialize manifest");
+
+      return manifest;
+    }
+
+    private static void UnpackDepot(string workingDirectory)
+    {
+      string paksDirectory = Path.Combine(workingDirectory, "Blitz/Content/Paks");
+      string[] containers = Directory.GetFiles(paksDirectory, "*.pak");
+
+      foreach (string container in containers)
+      {
+        Console.WriteLine($"Exergizing disk container \"{container}\"");
+
+        PakFileReader reader = new(container);
+        Exergize(reader);
+      }
+
+      if (Directory.Exists(TEMP_DEPOT_DIR))
+      {
+        Console.WriteLine("Deleting temp depots directory");
+        Directory.Delete(TEMP_DEPOT_DIR, true);
+      }
+    }
+
+    private static void Exergize(PakFileReader reader)
+    {
+      reader.Mount();
+
+      foreach (var file in reader.Files)
+      {
+        GameFile gameFile = file.Value;
+        bool exergizable = ShouldExergize(gameFile.Path);
+
+        if (!exergizable)
+        {
+          Console.WriteLine($"Exergize 🔴 {gameFile.Path}");
+          continue;
+        }
+
+        string writePath = Path.Combine(VFS_PATH, gameFile.Path);
+
+        if (File.Exists(writePath))
+        {
+          Console.WriteLine($"Exergize 🟡 {gameFile.Path}");
+        }
+        else
+        {
+          Console.WriteLine($"Exergize 🟢 {gameFile.Path}");
+        }
+
+        string? directory = Path.GetDirectoryName(writePath);
+
+        if (directory != null && !Directory.Exists(directory))
+        {
+          Directory.CreateDirectory(directory);
+        }
+
+        File.WriteAllBytes(writePath, gameFile.Read());
+      }
+    }
+
+    private static readonly string[] ExergizableGlobsRaw =
+    [
+      "Blitz/Config/DefaultDlc.ini",
+      "Blitz/Content/Tanks/**",
+      "Blitz/Content/TankSets/*",
+    ];
+    private static readonly Glob[] ExergizableGlobs = [.. ExergizableGlobsRaw.Select(Glob.Parse)];
+
+    private static bool ShouldExergize(string path)
+    {
+      // return true;
+      return ExergizableGlobs.Any(glob => glob.IsMatch(path));
+    }
+  }
+}