From 29776c93acdf81e4ffade9331c2aabb159c8ddf7 Mon Sep 17 00:00:00 2001
From: James Muscat <jamesremuscat@gmail.com>
Date: Sat, 25 Feb 2023 13:15:13 +0000
Subject: [PATCH] Export `Replay` and add visitor-pattern `forEachFrame`.

---
 src/replay.js | 51 +++++++++++++++++++++++++++++++++++++++++++++++++--
 1 file changed, 49 insertions(+), 2 deletions(-)

diff --git a/src/replay.js b/src/replay.js
index d348d2e..93a5d8a 100644
--- a/src/replay.js
+++ b/src/replay.js
@@ -3,15 +3,17 @@ import { diff, patch } from './diffs.js';
 
 export const REPLAY_FRAME_REGEX = /^([0-9]{5,11})(i?).json$/;
 
-class Replay {
+export class Replay {
   constructor(zipFile) {
     this._file = zipFile;
     this._keyframes = {};
     this._iframes = {};
+
+    this.forEachFrame = this.forEachFrame.bind(this);
   }
 
   async _init() {
-    const entries = await this._file.getEntries();
+    const entries = await this.listEntries();
 
     await Promise.all(
       entries.map(
@@ -57,6 +59,10 @@ class Replay {
     this.manifest.startTime = firstFrame;
   }
 
+  async listEntries() {
+    return await this._file.getEntries();
+  }
+
   async readEntry(e) {
     const text = await e.getData(new TextWriter());
     return JSON.parse(text);
@@ -81,18 +87,59 @@ class Replay {
       myState = await applyIframe(myState, ifrState);
     }
 
+    if (!myState.manifest) {
+      myState.manifest = this.manifest;
+    }
+
+    if (!myState.lastUpdated) {
+      myState.lastUpdated = time;
+    }
+
     return myState;
   }
 
   async getStateAtRelative(relTime) {
     return await this.getStateAt(this.manifest.startTime + relTime);
   }
+
+  async forEachFrame(callback) {
+    const keyframes = Object.keys(this._keyframes).map(k => parseInt(k, 10)).sort();
+    const iframes = Object.keys(this._iframes).map(k => parseInt(k, 10)).sort();
+
+    let prevFrame = null;
+    while (keyframes.length > 0) {
+      const k = keyframes.shift();
+
+      const kf = await this.readEntry(this._keyframes[k]);
+
+      prevFrame = kf;
+
+      if (!prevFrame.manifest) {
+        prevFrame.manifest = this.manifest;
+      }
+
+      if (!prevFrame.lastUpdated) {
+        prevFrame.lastUpdated = k;
+      }
+
+      callback(kf, k);
+
+      while (iframes.length > 0 && (keyframes.length === 0 || iframes[0] < keyframes[0])) {
+        const i = iframes.shift();
+        const ifr = await this.readEntry(this._iframes[i]);
+        prevFrame = await applyIframe(prevFrame, ifr);
+        prevFrame.lastUpdated = i;
+        callback(prevFrame, i);
+      }
+    }
+  }
 }
 
 const applyIframe = (base, iframe) => {
   return new Promise(
     (resolve) => {
       const result = {
+        ...base,
         cars: patch(iframe['cars'], base['cars']),
         session: patch(iframe['session'], base['session']),
         messages: iframe['messages'].concat(base['messages']).slice(0, 100),