fix: wrong rootId value in special iframes

packages/rrweb/src/record/iframe-manager.ts

@@ -1,5 +1,5 @@
 import type { Mirror, serializedNodeWithId } from 'rrweb-snapshot';
-import { genId } from 'rrweb-snapshot';
+import { genId, NodeType } from 'rrweb-snapshot';
 import type { CrossOriginIframeMessageEvent } from '../types';
 import CrossOriginIframeMirror from './cross-origin-iframe-mirror';
 import { EventType, IncrementalSource } from '@rrweb/types';
@@ -14,6 +14,10 @@ export class IframeManager {
   > = new WeakMap();
   public crossOriginIframeMirror = new CrossOriginIframeMirror(genId);
   public crossOriginIframeStyleMirror: CrossOriginIframeMirror;
+  public crossOriginIframeRootIdMap: WeakMap<
+    HTMLIFrameElement,
+    number
+  > = new WeakMap();
   private mirror: Mirror;
   private mutationCb: mutationCallBack;
   private wrappedEmit: (e: eventWithTime, isCheckout?: boolean) => void;
@@ -121,6 +125,9 @@ export class IframeManager {
          * Replaces the original id of the iframe with a new set of unique ids
          */
         this.replaceIdOnNode(e.data.node, iframeEl);
+        const rootId = e.data.node.id;
+        this.crossOriginIframeRootIdMap.set(iframeEl, rootId);
+        this.patchRootIdOnNode(e.data.node, rootId);
         return {
           timestamp: e.timestamp,
           type: EventType.IncrementalSnapshot,
@@ -171,6 +178,8 @@ export class IframeManager {
                 'previousId',
               ]);
               this.replaceIdOnNode(n.node, iframeEl);
+              const rootId = this.crossOriginIframeRootIdMap.get(iframeEl);
+              rootId && this.patchRootIdOnNode(n.node, rootId);
             });
             e.data.removes.forEach((n) => {
               this.replaceIds(n, iframeEl, ['parentId', 'id']);
@@ -273,11 +282,20 @@ export class IframeManager {
     node: serializedNodeWithId,
     iframeEl: HTMLIFrameElement,
   ) {
-    this.replaceIds(node, iframeEl, ['id']);
+    this.replaceIds(node, iframeEl, ['id', 'rootId']);
     if ('childNodes' in node) {
       node.childNodes.forEach((child) => {
         this.replaceIdOnNode(child, iframeEl);
       });
     }
   }
+
+  private patchRootIdOnNode(node: serializedNodeWithId, rootId: number) {
+    if (node.type !== NodeType.Document && !node.rootId) node.rootId = rootId;
+    if ('childNodes' in node) {
+      node.childNodes.forEach((child) => {
+        this.patchRootIdOnNode(child, rootId);
+      });
+    }
+  }
 }

packages/rrweb/test/benchmark/dom-mutation.test.ts

@@ -1,7 +1,8 @@
 import * as fs from 'fs';
 import * as path from 'path';
 import type { Page } from 'puppeteer';
-import type { eventWithTime, recordOptions } from '../../src/types';
+import type { eventWithTime } from '@rrweb/types';
+import type { recordOptions } from '../../src/types';
 import { startServer, launchPuppeteer, ISuite, getServerURL } from '../utils';
 
 const suites: Array<

packages/rrweb/test/benchmark/replay-fast-forward.test.ts

@@ -1,6 +1,7 @@
 import * as fs from 'fs';
 import * as path from 'path';
-import type { eventWithTime, recordOptions } from '../../src/types';
+import type { eventWithTime } from '@rrweb/types';
+import type { recordOptions } from '../../src/types';
 import { launchPuppeteer, ISuite } from '../utils';
 
 const suites: Array<{