X-Git-Url: https://git.r.bdr.sh/rbdr/junction/blobdiff_plain/d4fb48eb3793a93a8726652692b68592333dd778..bab26a4da2a2a0fdc88ec3af56909b12d822113a:/extension/junction.js

diff --git a/extension/junction.js b/extension/junction.js
index a662f6c..c3cb4e9 100644
--- a/extension/junction.js
+++ b/extension/junction.js
@@ -1,18 +1,19 @@
 const internals = {
-
-  promisesSupported: !!(window.browser),
+  promisesSupported: !!window.browser,
+  injectedScript: false,
   port: null,
+  currentUrl: null,
 
   icons: {
     call: {
-      16: 'icons/action-16.png',
-      32: 'icons/action-32.png'
+      16: "icons/action-16.png",
+      32: "icons/action-32.png",
     },
 
     hangUp: {
-      16: 'icons/hang_up-16.png',
-      32: 'icons/hang_up-32.png'
-    }
+      16: "icons/hang_up-16.png",
+      32: "icons/hang_up-32.png",
+    },
   },
 
   onClick() {
@@ -20,95 +21,101 @@ const internals = {
       return internals.hangUp();
     }
 
-    return internals.joinAudioCall();
+    return internals.initializeContentScript();
   },
 
-  onConnect(port) {
+  joinAudioCall() {
+    internals.port.postMessage({
+      action: "joinAudioCall",
+      data: {
+        currentUrl: internals.currentUrl,
+        tada: internals.getRoot().runtime.getURL("sounds/tada.wav"),
+      },
+    });
+    internals.getRoot().browserAction.enable();
+    internals.setIcon("hangUp");
+  },
 
+  onConnect(port) {
     internals.port = port;
     port.onDisconnect.addListener(internals.onDisconnect);
     port.onMessage.addListener(internals.onMessage);
-    port.postMessage({
-      action: 'joinAudioCall',
-      data: {
-        tada: internals.getRoot().runtime.getURL('sounds/tada.wav')
-      }
-    });
-    internals.getRoot().browserAction.enable();
-    internals.setIcon('hangUp');
+    internals.joinAudioCall();
   },
 
   onMessage(message) {
-
-    if (message.action === 'error') {
-      internals.getRoot().browserAction.setBadgeText({ text: 'x' }, () => {});
+    if (message.action === "error") {
+      internals.getRoot().browserAction.setBadgeText({ text: "x" }, () => {});
     }
   },
 
   onDisconnect() {
-    internals.getRoot().browserAction.setBadgeText({ text: '' }, () => {});
-    internals.setIcon('call');
+    internals.getRoot().browserAction.setBadgeText({ text: "" }, () => {});
+    internals.setIcon("call");
+    internals.currentUrl = null;
     internals.port = null;
     internals.getRoot().browserAction.enable();
   },
 
-  async joinAudioCall() {
-
+  async initializeContentScript() {
     internals.getRoot().browserAction.disable();
     const activeTabs = await internals.getActiveTabs();
 
-      const execution = await internals.getRoot().tabs.executeScript(activeTabs[0].id, {
-        file: '/content_script.js'
-      });
-
-      if (!execution || !execution[0]) {
+    internals.currentUrl = activeTabs[0].url;
+    if (!internals.injectedScript) {
+      const execution = await internals.getRoot().tabs.executeScript(
+        activeTabs[0].id,
+        {
+          file: "/build/content_script.js",
+        },
+        () => {
+          internals.injectedScript = true;
+        },
+      );
+
+      if (execution && !execution[0]) {
         internals.onDisconnect();
       }
-  }
-  ,
+    } else {
+      internals.getRoot().tabs.connect(activeTabs[0].id);
+    }
+  },
 
   hangUp() {
-
     internals.getRoot().browserAction.disable();
     internals.port.postMessage({
-      action: 'hangUp'
+      action: "hangUp",
     });
   },
 
   isInCall() {
-
-    return !!(internals.port);
+    return !!internals.port;
   },
 
   setIcon(iconSet) {
-
     internals.getRoot().browserAction.setIcon({
-      path: internals.icons[iconSet]
+      path: internals.icons[iconSet],
     });
   },
 
   getRoot() {
-
     return window.browser || window.chrome;
   },
 
   // Chrome doesn't yet implement the promise based tabs.query :'(
 
   getActiveTabs() {
-
     const query = {
       currentWindow: true,
-      active: true
+      active: true,
     };
 
     if (internals.promisesSupported) {
       return internals.getRoot().tabs.query(query);
     }
 
-    return new Promise((resolve, reject) => {
-
+    return new Promise((resolve) => {
       internals.getRoot().tabs.query(query, (tabs) => {
-
         return resolve(tabs);
       });
     });