Improve Reddit chat and media reliability

extensions/reddit-session-exporter/service-worker.js

@@ -26,27 +26,40 @@ chrome.webRequest.onBeforeSendHeaders.addListener(
     });
     await chrome.action.setBadgeText({ text: 'OK' });
     await chrome.action.setBadgeBackgroundColor({ color: '#198754' });
+    await chrome.action.setTitle({ title: 'Reddit Chat token captured. Click to export.' });
   },
   { urls: MATRIX_URLS },
   ['requestHeaders', 'extraHeaders'],
 );
 
-chrome.cookies.onChanged.addListener(async ({ cookie }) => {
-  if (!['reddit_session', 'token_v2'].includes(cookie.name)) return;
-  await chrome.storage.local.remove('matrixCapture');
-  await chrome.action.setBadgeText({ text: '' });
-  await chrome.action.setTitle({
-    title: 'Reddit account changed. Open Reddit Chat before exporting.',
-  });
-});
-
 async function getRedditIdentity() {
-  const response = await fetch('https://www.reddit.com/api/me.json', {
-    credentials: 'include',
-    cache: 'no-store',
+  const tabs = await chrome.tabs.query({
+    url: ['https://reddit.com/*', 'https://www.reddit.com/*'],
   });
-  if (!response.ok) throw new Error('Reddit cookies did not authenticate. Sign in and try again.');
-  const payload = await response.json();
+  const tab = tabs.find((candidate) => candidate.active) ?? tabs[0];
+  if (tab?.id === undefined) {
+    throw new Error('Keep a signed-in www.reddit.com tab open while exporting.');
+  }
+
+  const [{ result } = {}] = await chrome.scripting.executeScript({
+    target: { tabId: tab.id },
+    world: 'MAIN',
+    func: async () => {
+      try {
+        const response = await fetch('https://www.reddit.com/api/me.json', {
+          credentials: 'include',
+          cache: 'no-store',
+        });
+        if (!response.ok) return { error: 'Reddit cookies did not authenticate. Sign in and try again.' };
+        return { payload: await response.json() };
+      } catch {
+        return { error: 'Could not read the current Reddit account. Reload Reddit and try again.' };
+      }
+    },
+  });
+
+  if (result?.error) throw new Error(result.error);
+  const payload = result?.payload;
   const account = payload?.data ?? payload;
   if (!account?.name || !account?.id) {
     throw new Error('Reddit did not return an authenticated account identity.');
@@ -93,6 +106,8 @@ chrome.action.onClicked.addListener(async () => {
       ...createSessionExportDownload(cookies, { ...matrixCapture, userId: identity.user_id }),
       saveAs: true,
     });
+    await chrome.action.setBadgeText({ text: 'OK' });
+    await chrome.action.setTitle({ title: 'RedditBot session export created.' });
   } catch (error) {
     await showError(error instanceof Error ? error.message : 'Could not create the session dump.');
   }