Browse Source

fix: connection to the website closed

Ahmad Kholid 3 years ago
parent
commit
2e58312319
1 changed files with 18 additions and 19 deletions
  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);
       }
     });
   });