Fix zstd dictionary loading

NX-Editor · Feb 23, 2024 · e3afae3 · e3afae3
1 parent bfa56a9
commit e3afae3
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/src/TotkZstd.cs b/src/TotkZstd.cs
@@ -48,7 +48,7 @@ public static void Reload(string zsDicPath)
             _decompressors[id] = decompressor;
 
             Compressor compressor = new();
-            decompressor.LoadDictionary(fileData);
+            compressor.LoadDictionary(fileData);
             _compressors[id] = compressor;
         }
     }
@@ -65,14 +65,13 @@ public void TransformSource(IEditorFile handle)
     {
         Span<byte> buffer = handle.Source.AsSpan();
         if (buffer.Length < 5 || buffer.Read<uint>() != ZSTD_MAGIC) {
-            handle.Source = buffer.ToArray();
             return;
         }
 
         int id = GetDictionaryId(buffer);
-        _cache[handle.Id] = id;
 
         if (id > -1 && _decompressors.TryGetValue(id, out Decompressor? decompressor)) {
+            _cache[handle.Id] = id;
             handle.Source = decompressor.Unwrap(buffer).ToArray();
             return;
         }