From 03723b3e99bcbf61008e635a59918d737d39325f Mon Sep 17 00:00:00 2001
From: Simon Knott <info@simonknott.de>
Date: Fri, 6 Sep 2024 15:42:50 +0200
Subject: [PATCH] add pageId onto context.pages

---
 packages/trace-viewer/src/entries.ts         | 1 +
 packages/trace-viewer/src/traceModernizer.ts | 1 +
 packages/trace-viewer/src/ui/modelUtil.ts    | 2 +-
 3 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/packages/trace-viewer/src/entries.ts b/packages/trace-viewer/src/entries.ts
index b032884493..117134748b 100644
--- a/packages/trace-viewer/src/entries.ts
+++ b/packages/trace-viewer/src/entries.ts
@@ -41,6 +41,7 @@ export type ContextEntry = {
 };
 
 export type PageEntry = {
+  pageId: string,
   screencastFrames: {
     sha1: string,
     pageId: string,
diff --git a/packages/trace-viewer/src/traceModernizer.ts b/packages/trace-viewer/src/traceModernizer.ts
index b15ad527b0..7c6ff19ce6 100644
--- a/packages/trace-viewer/src/traceModernizer.ts
+++ b/packages/trace-viewer/src/traceModernizer.ts
@@ -58,6 +58,7 @@ export class TraceModernizer {
     let pageEntry = this._pageEntries.get(pageId);
     if (!pageEntry) {
       pageEntry = {
+        pageId,
         screencastFrames: [],
       };
       this._pageEntries.set(pageId, pageEntry);
diff --git a/packages/trace-viewer/src/ui/modelUtil.ts b/packages/trace-viewer/src/ui/modelUtil.ts
index 90a382ddb1..0c22d954f4 100644
--- a/packages/trace-viewer/src/ui/modelUtil.ts
+++ b/packages/trace-viewer/src/ui/modelUtil.ts
@@ -149,7 +149,7 @@ function indexModel(context: ContextEntry) {
   for (let i = 0; i < context.actions.length; ++i) {
     const action = context.actions[i] as any;
     action[contextSymbol] = context;
-    action[pageSymbol] = context.pages.find(page => page.screencastFrames.find(frame => frame.pageId === action.pageId));
+    action[pageSymbol] = context.pages.find(page => page.pageId === action.pageId);
   }
   let lastNonRouteAction = undefined;
   for (let i = context.actions.length - 1; i >= 0; i--) {