fix: connection to the website closed

src/content/index.js

@@ -138,19 +138,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;
-        default:
+      } 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);
+        }
       }
     });
   });