Implement extension guessing for unknown files (#150)

Crauzer · Mar 11, 2023 · 16d33c8 · 16d33c8
1 parent 6c6cac6
commit 16d33c8
Show file tree

Hide file tree

Showing 2 changed files with 24 additions and 2 deletions.
diff --git a/Obsidian/Data/Wad/WadTabModel.cs b/Obsidian/Data/Wad/WadTabModel.cs
@@ -59,7 +59,11 @@ public void Rebuild()
 
  foreach (var (_, chunk) in this.Wad.Chunks)
  {
- string path = this.Hashtable.GetChunkPath(chunk);
+ string path = this.Hashtable.TryGetChunkPath(chunk, out path) switch
+ {
+ true => path,
+ false => this.Hashtable.GuessChunkPath(chunk, this.Wad),
+ };
  string[] pathComponents = path.Split('/');
 
  if (pathComponents.Length is 1)

diff --git a/Obsidian/Services/HashtableService.cs b/Obsidian/Services/HashtableService.cs
@@ -1,4 +1,6 @@
-using LeagueToolkit.Core.Wad;
+using CommunityToolkit.HighPerformance;
+using LeagueToolkit.Core.Wad;
+using LeagueToolkit.Utils;
 using Obsidian.Data;
 using Octokit;
 using Serilog;
@@ -236,4 +238,20 @@ public string GetChunkPath(WadChunk chunk)
 
  return string.Format("{0:x16}", chunk.PathHash);
  }
+
+ public bool TryGetChunkPath(WadChunk chunk, out string path) =>
+ this.Hashes.TryGetValue(chunk.PathHash, out path);
+
+ public string GuessChunkPath(WadChunk chunk, WadFile wad)
+ {
+ using Stream stream = wad.LoadChunkDecompressed(chunk).AsStream();
+
+ string extension = LeagueFile.GetExtension(LeagueFile.GetFileType(stream));
+
+ return string.IsNullOrEmpty(extension) switch
+ {
+ true => string.Format("{0:x16}", chunk.PathHash),
+ false => string.Format("{0:x16}.{1}", chunk.PathHash, extension),
+ };
+ }
 }