@rpascene/shared 0.30.8

package/src/extractor/cs_postmessage.ts

@@ -0,0 +1,101 @@
+// 发送消息的封装，利用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)
+}