From 96781698babb1682e3d599226ace2d3f60ae8f54 Mon Sep 17 00:00:00 2001
From: Harvey <laozhaochaguan@gmail.com>
Date: Thu, 20 Oct 2022 17:19:02 +0800
Subject: [PATCH] update iframe.helper.js

---
 docs/iframe-inject/iframe.helper.js | 41 +++++++++++++++++++++--------
 docs/iframe-inject/index.html       | 37 +++++++++++++++++++++++++-
 2 files changed, 66 insertions(+), 12 deletions(-)

diff --git a/docs/iframe-inject/iframe.helper.js b/docs/iframe-inject/iframe.helper.js
index 601829c8a..4a198dd80 100644
--- a/docs/iframe-inject/iframe.helper.js
+++ b/docs/iframe-inject/iframe.helper.js
@@ -1,5 +1,5 @@
 class IframeHelper {
-    static postMessage(type, data, id = 0) {
+    static postMessage({ type, data, id = 0 }) {
         window.parent.postMessage(
             {
                 type: type,
@@ -18,12 +18,17 @@ class IframeHelper {
                     if (data.match(/resolve\((.*?)\)/)) {
                         const string = `return new Promise(function(resolve){\n${data}\n})`;
                         const result = await new Function(string)();
-                        IframeHelper.postMessage('response', result, id);
+                        IframeHelper.postMessage({ type: 'response', data: result, id });
                     } else {
-                        IframeHelper.postMessage('response', new Function(data)(), id);
+                        IframeHelper.postMessage({ type: 'response', data: new Function(data)(), id });
                     }
                 } catch (error) {
-                    IframeHelper.postMessage('error', error.message, id);
+                    IframeHelper.postMessage({ type: 'error', data: error.message, id });
+                }
+                break;
+            case 'destroy':
+                if (!IframeHelper.isDestroy) {
+                    IframeHelper.destroy();
                 }
                 break;
             default:
@@ -32,11 +37,16 @@ class IframeHelper {
     }
 
     static inject() {
-        IframeHelper.postMessage('inject');
+        IframeHelper.isInject = true;
+        IframeHelper.isDestroy = false;
+        IframeHelper.postMessage({ type: 'inject' });
         window.addEventListener('message', IframeHelper.onMessage);
     }
 
     static destroy() {
+        IframeHelper.isInject = false;
+        IframeHelper.isDestroy = true;
+        IframeHelper.postMessage({ type: 'destroy' });
         window.removeEventListener('message', IframeHelper.onMessage);
     }
 
@@ -45,6 +55,7 @@ class IframeHelper {
         this.isInject = false;
         this.isDestroy = false;
         this.$iframe = iframe;
+        this.messageCallback = null;
         this.onMessage = this.onMessage.bind(this);
         window.addEventListener('message', this.onMessage);
         this.$iframe.src = url;
@@ -59,6 +70,9 @@ class IframeHelper {
                 break;
             case 'destroy':
                 this.isInject = false;
+                if (!this.isDestroy) {
+                    this.destroy();
+                }
                 break;
             default:
                 break;
@@ -70,12 +84,15 @@ class IframeHelper {
             } else {
                 this.promises[id].resove(data);
             }
-
             delete this.promises[id];
         }
+
+        if (this.messageCallback) {
+            this.messageCallback({ type, data });
+        }
     }
 
-    postMessage(type, data) {
+    postMessage({ type, data }) {
         return new Promise((resove, reject) => {
             (function loop() {
                 if (this.isInject) {
@@ -100,10 +117,12 @@ class IframeHelper {
 
     commit(callback) {
         const callbackString = callback.toString();
-        const bodyString = callbackString
-            .substring(callbackString.indexOf('{') + 1, callbackString.lastIndexOf('}'))
-            .trim();
-        return this.postMessage('commit', bodyString);
+        const bodyString = callbackString.substring(callbackString.indexOf('{') + 1, callbackString.lastIndexOf('}'));
+        return this.postMessage({ type: 'commit', data: bodyString });
+    }
+
+    message(callback) {
+        this.messageCallback = callback;
     }
 
     destroy() {
diff --git a/docs/iframe-inject/index.html b/docs/iframe-inject/index.html
index 5311b3e49..23ee03c8d 100644
--- a/docs/iframe-inject/index.html
+++ b/docs/iframe-inject/index.html
@@ -24,6 +24,17 @@
             padding: 50px;
             box-sizing: border-box;
         }
+
+        .fullscreenWeb {
+            position: fixed;
+            z-index: 9999;
+            width: 100%;
+            height: 100%;
+            left: 0;
+            top: 0;
+            right: 0;
+            bottom: 0;
+        }
     </style>
 </head>
 
@@ -34,17 +45,41 @@
     <script src="./iframe.helper.js"></script>
     <script>
         (async function () {
+            const $iframe = document.getElementById('iframe');
+
             // Init iframe helper
             const helper = new IframeHelper({
-                iframe: document.getElementById('iframe'),
+                iframe: $iframe,
                 url: './inner.html',
             });
 
+            // Listen messages from the iframe
+            helper.message(({ type, data }) => {
+                switch (type) {
+                    case 'fullscreenWeb':
+                        if (data) {
+                            $iframe.classList.add('fullscreenWeb');
+                        } else {
+                            $iframe.classList.remove('fullscreenWeb');
+                        }
+                        break;
+                    default:
+                        break;
+                }
+            });
+
             // Instantiate the player and mount the instance in the window
             await helper.commit(() => {
                 window.art = new Artplayer({
                     container: '.artplayer-app',
                     url: '../assets/sample/video.mp4',
+                    fullscreen: true,
+                    fullscreenWeb: true,
+                });
+
+                // Send message to parent window
+                window.art.on('fullscreenWeb', state => {
+                    IframeHelper.postMessage({ type: 'fullscreenWeb', data: state })
                 });
             });