Merge pull request #19 from Monadical-SAS/nikita-sidepanel-archiving-sync

feat: start archiving new pages as the user starts to visit them with archiving enabled

Author: nikitalokhmachev-ai

src/ext/bg.ts

@@ -18,6 +18,7 @@ let newRecCollId = null;
 // @ts-expect-error - TS7034 - Variable 'defaultCollId' implicitly has type 'any' in some locations where its type cannot be determined.
 let defaultCollId = null;
 let autorun = false;
+let isRecordingEnabled = false;
 
 const openWinMap = new Map();
 
@@ -159,16 +160,49 @@ function sidepanelHandler(port) {
       }
 
       case "startRecording": {
-        const { collId, autorun } = message;
-        // @ts-expect-error - TS2554 - Expected 2 arguments, but got 3.
-        startRecorder(tabId, { collId, port, autorun }, message.url);
+        isRecordingEnabled = true;
+        defaultCollId = message.collId;
+        autorun = message.autorun;
+
+        // @ts-expect-error - tabs doesn't have type definitions
+        chrome.tabs.query({ active: true, currentWindow: true }, async (tabs) => {
+          for (const tab of tabs) {
+            if (!isValidUrl(tab.url)) continue;
+
+            // @ts-expect-error - TS2554 - Expected 2 arguments, but got 3.
+            await startRecorder(tab.id, { collId: defaultCollId, port: null, autorun }, tab.url);
+          }
+
+          port.postMessage({
+            type: "status",
+            recording: true,
+            autorun,
+            // @ts-expect-error - defaultCollId implicitly has an 'any' type.
+            collId: defaultCollId,
+          });
+        });
+
         break;
       }
 
-      case "stopRecording":
-        // @ts-expect-error - TS7005 - Variable 'tabId' implicitly has an 'any' type.
-        stopRecorder(tabId);
+      case "stopRecording": {
+        isRecordingEnabled = false;
+
+        for (const [tabIdStr, rec] of Object.entries(self.recorders)) {
+          const tabId = parseInt(tabIdStr);
+          stopRecorder(tabId);
+        }
+
+        port.postMessage({
+          type: "status",
+          recording: false,
+          autorun,
+          // @ts-expect-error - defaultCollId implicitly has an 'any' type.
+          collId: defaultCollId,
+        });
+
         break;
+      }
 
       case "toggleBehaviors":
         // @ts-expect-error - TS7005 - Variable 'tabId' implicitly has an 'any' type.
@@ -203,6 +237,21 @@ chrome.debugger.onDetach.addListener((tab, reason) => {
   }
 });
 
+// @ts-expect-error - TS7006 - Parameter 'tab' implicitly has an 'any' type.
+chrome.tabs.onActivated.addListener(async ({ tabId }) => {
+  if (!isRecordingEnabled) return;
+
+  // @ts-expect-error - chrome doesn't have type definitions
+  const tab = await new Promise<chrome.tabs.Tab>((resolve) => chrome.tabs.get(tabId, resolve));
+
+  if (!isValidUrl(tab.url)) return;
+
+  if (!self.recorders[tabId]) {
+    // @ts-expect-error - TS2554 - Expected 2 arguments, but got 3.
+    startRecorder(tabId, { collId: defaultCollId, port: null, autorun }, tab.url);
+  }
+});
+
 // ===========================================================================
 // @ts-expect-error - TS7006 - Parameter 'tab' implicitly has an 'any' type.
 chrome.tabs.onCreated.addListener((tab) => {
@@ -278,14 +327,20 @@ chrome.tabs.onUpdated.addListener((tabId, changeInfo) => {
         return;
       }
     }
-  } else if (changeInfo.url && openWinMap.has(changeInfo.url)) {
-    const collId = openWinMap.get(changeInfo.url);
-    openWinMap.delete(changeInfo.url);
-    if (!tabId || !isValidUrl(changeInfo.url)) {
+  } else if (changeInfo.url) {
+    if (isRecordingEnabled && isValidUrl(changeInfo.url) && !self.recorders[tabId]) {
+      // @ts-expect-error - TS2554 - Expected 2 arguments, but got 3.
+      startRecorder(tabId, { collId: defaultCollId, autorun }, changeInfo.url);
       return;
     }
-    // @ts-expect-error - TS2554 - Expected 2 arguments, but got 3.
-    startRecorder(tabId, { collId, autorun }, changeInfo.url);
+    if (openWinMap.has(changeInfo.url)) {
+      const collId = openWinMap.get(changeInfo.url);
+      openWinMap.delete(changeInfo.url);
+      if (!tabId || !isValidUrl(changeInfo.url)) return;
+
+      // @ts-expect-error - TS2554 - Expected 2 arguments, but got 3.
+      startRecorder(tabId, { collId, autorun }, changeInfo.url);
+    }
   }
 });
 

src/sidepanel.ts

@@ -301,39 +301,32 @@ class ArgoViewer extends LitElement {
       <md-divider></md-divider>
       <div style="padding:1rem; display:flex; align-items:center; justify-content:space-between;">
         ${
-          // @ts-expect-error - TS2339 - Property 'canRecord' does not exist on type 'RecPopup'.
-          this.canRecord
+          // @ts-expect-error - TS2339 - Property 'recording' does not exist on type 'RecPopup'.
+          !this.recording
             ? html`
-                ${
-                  // @ts-expect-error - TS2339 - Property 'recording' does not exist on type 'RecPopup'.
-                  !this.recording
-                    ? html`
-                        <md-filled-button
-                          style="
+                <md-filled-button
+                  style="
                   --md-sys-color-primary-container: #7b1fa2;
                   color: white;
                   border-radius: 9999px;
                 "
-                          ?disabled=${this.actionButtonDisabled}
-                          @click=${this.onStart}
-                        >
-                          <md-icon slot="icon" style="color:white">public</md-icon>
-                          Resume Archiving
-                        </md-filled-button>
-                      `
-                    : html`
-                        <md-outlined-button
-                          style="--md-sys-color-primary: #b00020; --md-sys-color-outline: #b00020; border-radius: 9999px;"
-                          ?disabled=${this.actionButtonDisabled}
-                          @click=${this.onStop}
-                        >
-                          <md-icon slot="icon" style="color:#b00020">pause</md-icon>
-                          Pause Archiving
-                        </md-outlined-button>
-                      `
-                }
+                  ?disabled=${this.actionButtonDisabled}
+                  @click=${this.onStart}
+                >
+                  <md-icon slot="icon" style="color:white">public</md-icon>
+                  Resume Archiving
+                </md-filled-button>
+              `
+            : html`
+                <md-outlined-button
+                  style="--md-sys-color-primary: #b00020; --md-sys-color-outline: #b00020; border-radius: 9999px;"
+                  ?disabled=${this.actionButtonDisabled}
+                  @click=${this.onStop}
+                >
+                  <md-icon slot="icon" style="color:#b00020">pause</md-icon>
+                  Pause Archiving
+                </md-outlined-button>
               `
-            : html`<span></span>`
         }
 
         <md-icon-button aria-label="Settings">