Update demo repair

NeKzor · Apr 15, 2024 · 46b7e53 · 46b7e53
1 parent 16620d2
commit 46b7e53
Showing 1 changed file with 21 additions and 4 deletions.
diff --git a/src/server/demo.ts b/src/server/demo.ts
@@ -545,11 +545,28 @@ export const getPlayerInfo = (demo: SourceDemo): PlayerInfoData => {
 
 // Imported from: https://github.com/NeKzor/sdp/blob/master/examples/repair.ts
 export const repairDemo = (buffer: Uint8Array): Uint8Array => {
-  const parser = SourceDemoParser.default();
+  const parser = SourceDemoParser.default()
+    .setOptions({ packets: true, dataTables: true });
 
-  const demo = parser
-    .setOptions({ packets: true })
-    .parse(buffer);
+  const demo = SourceDemo.default();
+
+  try {
+    const buf = parser.prepare(buffer);
+    demo.readHeader(buf)
+      .readMessages(buf);
+  } catch (err) {
+    console.error(err);
+  }
+  try {
+    demo.readDataTables();
+  } catch (err) {
+    console.error(err);
+  }
+  try {
+    demo.readPackets();
+  } catch (err) {
+    console.error(err);
+  }
 
   const tryFixup = () => {
     const dt = demo.findMessage(Messages.DataTable)?.dataTable;