Merge pull request #14 from oslabs-beta/feature/ellie

updated reconnection logic to further correct loss of port connection…

Author: elliesimens

src/extension/background.js

@@ -348,14 +348,18 @@ chrome.runtime.onConnect.addListener(async (port) => {
         portsArr.splice(i, 1);
         setTimeout(async () => {
           try {
-            const response = await chrome.runtime.sendMessage({
-              action: 'attemptReconnect',
-            });
-            if (response && response.success) {
+            const newPort = chrome.runtime.connect({ name: 'reconnected' }); // Attempt to reconnect
+            if (newPort) {
+              portsArr.push(newPort); // Add the new port to the array
+              newPort.onMessage.addListener((msg) => {
+                console.log('Message received on reconnected port:', msg);
+              });
               console.log('Port successfully reconnected');
+            } else {
+              console.warn('Failed to reconnect port');
             }
           } catch (error) {
-            console.warn('Port reconnection failed:', error);
+            console.warn('Port reconnection attempt failed:', error);
           }
         }, 1000);
         break;

src/extension/contentScript.ts

@@ -2,6 +2,21 @@
 // in Web Metrics tab of Reactime app.
 import { onTTFB, onLCP, onFID, onFCP, onCLS, onINP } from 'web-vitals';
 
+function establishConnection() {
+  const port = chrome.runtime.connect({ name: 'content-script' }); // Reconnect
+  port.onMessage.addListener((msg) => {
+    console.log('Received message from background:', msg);
+  });
+
+  port.onDisconnect.addListener(() => {
+    console.warn('Port disconnected, attempting to reconnect...');
+    setTimeout(establishConnection, 1000); // Retry after 1 second
+  });
+}
+
+// Initially establish connection
+establishConnection();
+
 // Reactime application starts off with this file, and will send
 // first message to background.js for initial tabs object set up.
 // A "tabs object" holds the information of the current tab,