@godscene/shared 1.7.11

package/src/extractor/cs_postmessage.ts

@@ -0,0 +1,136 @@
+// 发送消息的封装，利用secret实现发送后等待结果的接口
+const TYPE = 'RPA_IDE_CS_MSG';
+// 发送消息，并等待回复结果
+export const postWindowMessage: (
+  targetWin: any,
+  sourceWin: any,
+  payload: any,
+  targetOrigin?: any,
+  timeout?: any,
+) => Promise<any> = (
+  targetWin: any,
+  sourceWin: any,
+  payload: any,
+  targetOrigin: any = '*',
+  timeout: any = 60000,
+) => {
+  return new Promise((resolve, reject) => {
+    if (!targetWin || !targetWin.postMessage) {
+      reject(new Error('E350: csPostMessage: targetWin is not a window'));
+    }
+
+    if (
+      !sourceWin ||
+      !sourceWin.addEventListener ||
+      !sourceWin.removeEventListener
+    ) {
+      reject(new Error('E351: csPostMessage: sourceWin is not a window'));
+    }
+
+    const secret = Math.random();
+    const type = TYPE;
+
+    // Note: create a listener with a corresponding secret every time
+    const onMsg = (e: any) => {
+      if (
+        e.data &&
+        e.data.type === TYPE &&
+        !e.data.isRequest &&
+        e.data.secret === secret
+      ) {
+        sourceWin.removeEventListener('message', onMsg);
+        const { payload, error } = e.data;
+
+        if (error) return reject(new Error(error));
+        if (payload !== undefined) return resolve(payload);
+
+        reject(new Error('E352: csPostMessage: No payload nor error found'));
+      }
+    };
+
+    sourceWin.addEventListener('message', onMsg);
+
+    // Note:
+    // * `type` to make sure we check our own msg only
+    // * `secret` is for 1 to 1 relationship between a msg and a listener
+    // * `payload` is the real data you want to send
+    // * `isRequest` is to mark that it's not an answer to some previous request
+
+    targetWin.postMessage(
+      {
+        type,
+        secret,
+        payload,
+        isRequest: true,
+      },
+      targetOrigin,
+    );
+
+    setTimeout(() => {
+      reject(new Error(`E353: csPostMessage: timeout ${timeout} ms`)); //Why 5000?
+    }, timeout);
+  });
+};
+// 监听消息，并自动回复
+export const onWindowMessage = (win: any, fn: any) => {
+  if (!win || !win.addEventListener || !win.removeEventListener) {
+    throw new Error('csOnMessage: not a window');
+  }
+
+  const onMsg = (e: any) => {
+    // Note: only respond to msg with `isRequest` as true
+    if (
+      e &&
+      e.data &&
+      e.data.type === TYPE &&
+      e.data.isRequest &&
+      e.data.secret
+    ) {
+      const tpl = {
+        type: TYPE,
+        secret: e.data.secret,
+      };
+
+      // Note: wrapped with a new Promise to catch any exception during the execution of fn
+      new Promise((resolve, reject) => {
+        let ret;
+
+        try {
+          ret = fn(e.data.payload, {
+            source: e.source,
+          });
+        } catch (err) {
+          reject(err);
+        }
+
+        // Note: only resolve if returned value is not undefined. With this, we can have multiple
+        // listeners added to onMessage, and each one takes care of what it really cares
+        if (ret !== undefined) {
+          resolve(ret);
+        }
+      }).then(
+        (res) => {
+          e.source.postMessage(
+            {
+              ...tpl,
+              payload: res,
+            },
+            '*',
+          );
+        },
+        (err) => {
+          e.source.postMessage(
+            {
+              ...tpl,
+              error: err.message,
+            },
+            '*',
+          );
+        },
+      );
+    }
+  };
+
+  win.addEventListener('message', onMsg);
+  return () => win.removeEventListener('message', onMsg);
+};