浏览代码

fix: connection to the website closed

Ahmad Kholid 3 年之前
父节点
当前提交
2e58312319
共有 1 个文件被更改,包括 18 次插入19 次删除
  1. 18 19
      src/content/index.js

+ 18 - 19
src/content/index.js

@@ -139,26 +139,25 @@ function messageListener({ data, source }) {
     return new Promise((resolve, reject) => {
       if (data.isBlock) {
         executeBlock(data).then(resolve).catch(reject);
-        return;
-      }
-
-      switch (data.type) {
-        case 'condition-builder':
-          handleTestCondition(data.data)
-            .then((result) => resolve(result))
-            .catch((error) => reject(error));
-          break;
-        case 'content-script-exists':
-          resolve(true);
-          break;
-        case 'automa-element-selector': {
-          const selectorInstance = elementSelectorInstance();
-
-          resolve(selectorInstance);
-          break;
+      } else {
+        switch (data.type) {
+          case 'condition-builder':
+            handleTestCondition(data.data)
+              .then((result) => resolve(result))
+              .catch((error) => reject(error));
+            break;
+          case 'content-script-exists':
+            resolve(true);
+            break;
+          case 'automa-element-selector': {
+            const selectorInstance = elementSelectorInstance();
+
+            resolve(selectorInstance);
+            break;
+          }
+          default:
+            resolve(null);
         }
-        default:
-          resolve(null);
       }
     });
   });