[pairing] update types

prax-wallet · Aug 23, 2024 · 3f788eb · 3f788eb
1 parent 8e03a5d
commit 3f788eb
Show file tree

Hide file tree

Showing 4 changed files with 22 additions and 10 deletions.
diff --git a/apps/extension/src/entry/popup-root.tsx b/apps/extension/src/entry/popup-root.tsx
@@ -21,6 +21,7 @@ chrome.runtime.onMessage.addListener(
         if (isTxApprovalRequest(req)) {
           void txApprovalSelector(useStore.getState()).acceptRequest(req, responder);
         } else if (isOriginApprovalRequest(req)) {
+          req.request.origin;
           originApprovalSelector(useStore.getState()).acceptRequest(req, responder);
         } else {
           throw new Error('Unknown popup request');

diff --git a/apps/extension/src/hooks/popup-ready.ts b/apps/extension/src/hooks/popup-ready.ts
@@ -1,5 +1,5 @@
 import { useEffect, useRef } from 'react';
-import { PopupType, PopupReadyResponse } from '../message/popup';
+import { PopupResponse, PopupType, Ready } from '../message/popup';
 
 type IsReady = boolean | undefined;
 
@@ -18,7 +18,7 @@ export const usePopupReady = (isReady: IsReady = undefined) => {
         data: {
           popupId,
         },
-      } as PopupReadyResponse);
+      } as PopupResponse<Ready>);
     }
   }, [popupId, isReady]);
 };
diff --git a/apps/extension/src/message/popup.ts b/apps/extension/src/message/popup.ts
@@ -14,10 +14,9 @@ export enum PopupType {
   Ready = 'PopupReady',
 }
 
-export type PopupMessage = TxApproval | OriginApproval;
+export type PopupMessage = TxApproval | OriginApproval | Ready;
 export type PopupRequest<T extends PopupMessage = PopupMessage> = InternalRequest<T>;
 export type PopupResponse<T extends PopupMessage = PopupMessage> = InternalResponse<T>;
-export type PopupReadyResponse<T extends Ready = Ready> = InternalResponse<T>;
 
 export type OriginApproval = InternalMessage<
   PopupType.OriginApproval,
@@ -53,7 +52,10 @@ export const isPopupRequest = (req: unknown): req is PopupRequest =>
   req.type in PopupType;
 
 export const isOriginApprovalRequest = (req: unknown): req is InternalRequest<OriginApproval> =>
-  isPopupRequest(req) && req.type === PopupType.OriginApproval && 'origin' in req.request;
+  isPopupRequest(req) && req.type === PopupType.OriginApproval;
 
 export const isTxApprovalRequest = (req: unknown): req is InternalRequest<TxApproval> =>
-  isPopupRequest(req) && req.type === PopupType.TxApproval && 'authorizeRequest' in req.request;
+  isPopupRequest(req) && req.type === PopupType.TxApproval;
+
+export const isPopupReadyResponse = (req: unknown): req is InternalResponse<Ready> =>
+  isPopupRequest(req) && req.type === PopupType.Ready;
diff --git a/apps/extension/src/popup.ts b/apps/extension/src/popup.ts
@@ -1,6 +1,11 @@
 import { sessionExtStorage } from './storage/session';
-import { PopupMessage, PopupRequest, PopupType, PopupReadyResponse } from './message/popup';
-// import { PopupMessage, PopupRequest, PopupType } from './message/popup';
+import {
+  isPopupReadyResponse,
+  PopupMessage,
+  PopupRequest,
+  PopupResponse,
+  PopupType,
+} from './message/popup';
 import { PopupPath } from './routes/popup/paths';
 import type { InternalRequest, InternalResponse } from '@penumbra-zone/types/internal-msg/shared';
 import { Code, ConnectError } from '@connectrpc/connect';
@@ -101,8 +106,12 @@ const popupReady = async (popupId: string): Promise<void> => {
       reject(new Error('Popup ready timed out'));
     }, POPUP_READY_TIMEOUT);
 
-    const handlePopupReady = (res: PopupReadyResponse): void => {
-      if (res.type === PopupType.Ready && res.data.popupId === popupId) {
+    const handlePopupReady = (res: PopupResponse): void => {
+      if (!isPopupReadyResponse(res)) {
+        return;
+      }
+
+      if ('data' in res && res.data.popupId === popupId) {
         chrome.runtime.onMessage.removeListener(handlePopupReady);
         resolve();
       }