fix(iframe): fix iframe transport

unadlib · Nov 9, 2023 · d487ec2 · d487ec2
1 parent daa0d2c
commit d487ec2
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 10 deletions.
diff --git a/package.json b/package.json
@@ -1,6 +1,6 @@
 {
   "name": "data-transport",
-  "version": "4.3.2",
+  "version": "4.3.3",
   "description": "A simple and responsible transport",
   "main": "lib/index.js",
   "unpkg": "dist/index.umd.js",

diff --git a/src/transports/iframeTransport.ts b/src/transports/iframeTransport.ts
@@ -74,15 +74,19 @@ export abstract class IFrameMainTransport<
       sender,
     });
     if (!skipConnectionCheck) {
-      this.emit({
-        // @ts-ignore
-        name: connectEventName,
-        silent: true,
-      }).then((connected) => {
-        if (connected) {
-          this[beforeEmitResolveKey]!();
-        }
-      });
+      const connect = () => {
+        this.emit({
+          // @ts-ignore
+          name: connectEventName,
+          silent: true,
+          skipBeforeEmit: true,
+        }).then((connected) => {
+          if (connected) {
+            this[beforeEmitResolveKey]!();
+          }
+        });
+      };
+      connect();
       this[beforeEmitKey] = new Promise((resolve) => {
         this[beforeEmitResolveKey] = resolve;
       });
@@ -91,10 +95,12 @@ export abstract class IFrameMainTransport<
         this[beforeEmitResolveKey]!();
         return true;
       });
+      // for iframe reload
       iframe?.addEventListener('load', () => {
         this[beforeEmitKey] = new Promise((resolve) => {
           this[beforeEmitResolveKey] = resolve;
         });
+        connect();
       });
     }
   }