Added support for decoding Brotli compression used in new formats (#7)

aoemods · Nov 9, 2023 · 2bb952f · 2bb952f
1 parent c21a0d5
commit 2bb952f
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 1 deletion.
diff --git a/AOEMods.Essence/SGA/Core/FileStorageType.cs b/AOEMods.Essence/SGA/Core/FileStorageType.cs
@@ -18,5 +18,15 @@ public enum FileStorageType : byte
     /// <summary>
     /// Stored compressed.
     /// </summary>
-    BufferCompress
+    BufferCompress,
+
+    /// <summary>
+    /// Stored compressed.
+    /// </summary>
+    StreamCompressBrotli,
+
+    /// <summary>
+    /// Stored compressed.
+    /// </summary>
+    BufferCompressBrotli
 }
diff --git a/AOEMods.Essence/SGA/Graph/ArchiveFileNode.cs b/AOEMods.Essence/SGA/Graph/ArchiveFileNode.cs
@@ -1,4 +1,5 @@
 using AOEMods.Essence.SGA.Core;
+using System.IO;
 using System.IO.Compression;
 using System.Text;
 
@@ -67,6 +68,15 @@ public IEnumerable<byte> GetData()
                     MemoryStream decoded = new((int)dataUncompressedLength);
                     deflateStream.CopyTo(decoded);
 
+                    return decoded.ToArray();
+                }
+            case FileStorageType.StreamCompressBrotli:
+            case FileStorageType.BufferCompressBrotli:
+                {
+                    using var brotliStream = new BrotliStream(dataStream, CompressionMode.Decompress, leaveOpen: true);
+                    MemoryStream decoded = new((int)dataUncompressedLength);
+                    brotliStream.CopyTo(decoded);
+
                     return decoded.ToArray();
                 }
             default: