consumable icons

packages/game/JSExport/GameInterface.cs

@@ -11,6 +11,10 @@ using CUE4Parse.UE4.Objects.UObject;
 using Microsoft.CodeAnalysis.CSharp.Syntax;
 using Microsoft.JavaScript.NodeApi;
 using SkiaSharp;
+using CUE4Parse.UE4.AssetRegistry.Objects;
+using CUE4Parse.UE4.Readers;
+using CUE4Parse.UE4.AssetRegistry.Readers;
+using CUE4Parse.UE4.AssetRegistry;
 
 namespace BlitzKit.Game.JSExport;
 
@@ -24,7 +28,7 @@ public partial class GameInterface
 
   public GameInterface(string directory, string map, string temp)
   {
-    provider = new(directory, map);
+    provider = new(directory, map, temp);
 
     files = [.. provider.Files.Keys];
 
@@ -38,10 +42,6 @@ public partial class GameInterface
         .Where(nation => !nation.EndsWith(".uasset") && nation != "TankStub")
         .Distinct(),
     ];
-
-    string oodlePath = $"{temp}/oodle.so";
-    OodleHelper.DownloadOodleDll(ref oodlePath!);
-    OodleHelper.Initialize(oodlePath);
   }
 
   public HashSet<string> Files => files;
@@ -123,6 +123,13 @@ public partial class GameInterface
     return [];
   }
 
+  public byte[] ConsumableIcon(string tag)
+  {
+    var pda = provider.Discovered<UPrimaryDataAsset>(tag);
+    var icon = pda.Get<FSoftObjectPath>("Icon");
+    return provider.Image(icon);
+  }
+
   private static readonly string equipmentPDAPrefix =
     "Blitz/Content/TanksStuff/Equipment/PDA_Equipment";
   private static readonly string equipmentPDASuffix = ".uasset";

packages/game/Models/BlitzFileProvider.cs

@@ -10,12 +10,21 @@ using CUE4Parse.UE4.Objects.UObject;
 using CUE4Parse.UE4.Versions;
 using CUE4Parse.Utils;
 using SkiaSharp;
+using CUE4Parse.UE4.Readers;
+using CUE4Parse.UE4.AssetRegistry;
+using System.Text.RegularExpressions;
+using CUE4Parse.UE4.Assets.Exports;
 
 namespace BlitzKit.Game.Models;
 
-public class BlitzFileProvider : DefaultFileProvider
+public partial class BlitzFileProvider : DefaultFileProvider
 {
-  public BlitzFileProvider(string directory, string map)
+  [GeneratedRegex(@"\(GameplayTags=\(\(TagName=""(.+)""\)\)\)")]
+  private static partial Regex GameplayTagsRegex();
+
+  private readonly Dictionary<string, string> discovered = [];
+
+  public BlitzFileProvider(string directory, string map, string temp)
     : base(
       directory: directory,
       searchOption: SearchOption.AllDirectories,
@@ -27,6 +36,38 @@ public class BlitzFileProvider : DefaultFileProvider
 
     Initialize();
     Mount();
+
+    string oodlePath = $"{temp}/oodle.so";
+    OodleHelper.DownloadOodleDll(ref oodlePath!);
+    OodleHelper.Initialize(oodlePath);
+
+    if (TryGetGameFile("Blitz/AssetRegistry.bin", out var file))
+    {
+      var bytes = file.Read();
+      var archive = new FByteArchive(file.Path, bytes, Versions);
+      var state = new FAssetRegistryState(archive);
+
+      foreach (var asset in state.PreallocatedAssetDataBuffers)
+      {
+        if (asset.TagsAndValues.TryGetValue("DiscoveryTags", out var tags))
+        {
+          var match = GameplayTagsRegex().Match(tags);
+
+          if (!match.Success)
+          {
+            continue;
+          }
+
+          var tag = match.Groups[1].Value;
+          discovered[tag] = asset.ObjectPath;
+        }
+      }
+    }
+  }
+
+  public T Discovered<T>(string path) where T : UObject
+  {
+    return LoadPackageObject<T>(discovered[path]);
   }
 
   public UPrimaryDataAsset PDA(string path)