Scope approvals to documentId

prax-wallet · Jun 19, 2024 · 5615e4a · 5615e4a
1 parent 77d00c0
commit 5615e4a
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/apps/extension/src/listeners/message-prax-request.ts b/apps/extension/src/listeners/message-prax-request.ts
@@ -15,7 +15,9 @@ chrome.runtime.onMessage.addListener(
         // user made a choice
         if (status === UserChoice.Approved) {
           respond(PraxConnection.Init);
-          void chrome.tabs.sendMessage(sender.tab!.id!, PraxConnection.Init);
+          void chrome.tabs.sendMessage(sender.tab!.id!, PraxConnection.Init, {
+            documentId: sender.documentId, // Ensures tab has not redirected to another url
+          });
         } else {
           respond(PraxConnection.Denied);
         }

diff --git a/apps/extension/src/listeners/tabs-updated-prax-init.ts b/apps/extension/src/listeners/tabs-updated-prax-init.ts
@@ -11,6 +11,7 @@ chrome.tabs.onUpdated.addListener(
         url?.startsWith('https://') &&
         (await originAlreadyApproved(url))
       )
+        // How can we scope this to documentId?
         void chrome.tabs.sendMessage(tabId, PraxConnection.Init);
     })(),
 );