Merge pull request #6 from pendo-io/x-origin-frame

cross origin frame bug fixes
pendo-io · Sep 27, 2023 · 718539c · 718539c
2 parents 04fa0dd + 0458e37
commit 718539c
Show file tree

Hide file tree

Showing 2 changed files with 23 additions and 0 deletions.
diff --git a/packages/rrweb/src/record/iframe-manager.ts b/packages/rrweb/src/record/iframe-manager.ts
@@ -17,6 +17,7 @@ export class IframeManager {
   private mirror: Mirror;
   private mutationCb: mutationCallBack;
   private wrappedEmit: (e: eventWithTime, isCheckout?: boolean) => void;
+  private takeFullSnapshot: (isCheckout?: boolean) => void;
   private loadListener?: (iframeEl: HTMLIFrameElement) => unknown;
   private stylesheetManager: StylesheetManager;
   private recordCrossOriginIframes: boolean;
@@ -27,9 +28,11 @@ export class IframeManager {
     stylesheetManager: StylesheetManager;
     recordCrossOriginIframes: boolean;
     wrappedEmit: (e: eventWithTime, isCheckout?: boolean) => void;
+    takeFullSnapshot: (isCheckout?: boolean) => void;
   }) {
     this.mutationCb = options.mutationCb;
     this.wrappedEmit = options.wrappedEmit;
+    this.takeFullSnapshot = options.takeFullSnapshot;
     this.stylesheetManager = options.stylesheetManager;
     this.recordCrossOriginIframes = options.recordCrossOriginIframes;
     this.crossOriginIframeStyleMirror = new CrossOriginIframeMirror(
@@ -47,6 +50,16 @@ export class IframeManager {
     this.iframes.set(iframeEl, true);
     if (iframeEl.contentWindow)
       this.crossOriginIframeMap.set(iframeEl.contentWindow, iframeEl);
+
+    if (!iframeEl.contentDocument && iframeEl.contentWindow)
+      iframeEl.contentWindow.postMessage(
+        {
+          type: 'rrweb',
+          origin: window.location.origin,
+          snapshot: true,
+        },
+        '*',
+      );
   }
 
   public addLoadListener(cb: (iframeEl: HTMLIFrameElement) => unknown) {
@@ -94,6 +107,15 @@ export class IframeManager {
     const iframeSourceWindow = message.source;
     if (!iframeSourceWindow) return;
 
+    if (
+      iframeSourceWindow == window.parent &&
+      window != window.parent &&
+      message.data.snapshot
+    ) {
+      this.takeFullSnapshot();
+      return;
+    }
+
     const iframeEl = this.crossOriginIframeMap.get(message.source);
     if (!iframeEl) return;
 

diff --git a/packages/rrweb/src/record/index.ts b/packages/rrweb/src/record/index.ts
@@ -298,6 +298,7 @@ function record<T = eventWithTime>(
     stylesheetManager: stylesheetManager,
     recordCrossOriginIframes,
     wrappedEmit,
+    takeFullSnapshot,
   });
 
   /**