From f5311b5469ef92da902b9198e3065a466e953dd9 Mon Sep 17 00:00:00 2001
From: Niranjana Binoy <43930900+NiranjanaBinoy@users.noreply.github.com>
Date: Tue, 3 Sep 2024 17:24:09 -0400
Subject: [PATCH] updating request type

---
 .../handlers/get-provider-state.ts              | 17 ++++++++---------
 .../lib/rpc-method-middleware/handlers/types.ts |  7 +++++++
 2 files changed, 15 insertions(+), 9 deletions(-)

diff --git a/app/scripts/lib/rpc-method-middleware/handlers/get-provider-state.ts b/app/scripts/lib/rpc-method-middleware/handlers/get-provider-state.ts
index 2f15d61a45cd..613a8d20467f 100644
--- a/app/scripts/lib/rpc-method-middleware/handlers/get-provider-state.ts
+++ b/app/scripts/lib/rpc-method-middleware/handlers/get-provider-state.ts
@@ -2,14 +2,13 @@ import type {
   JsonRpcEngineNextCallback,
   JsonRpcEngineEndCallback,
 } from 'json-rpc-engine';
-import type {
-  JsonRpcRequest,
-  PendingJsonRpcResponse,
-  JsonRpcParams,
-} from '@metamask/utils';
+import type { PendingJsonRpcResponse, JsonRpcParams } from '@metamask/utils';
 import { OriginString } from '@metamask/permission-controller';
 import { MESSAGE_TYPE } from '../../../../../shared/constants/app';
-import { HandlerWrapper } from './types';
+import {
+  HandlerWrapper,
+  HandlerRequestType as ProviderStateHandlerRequest,
+} from './types';
 
 /**
  * @property chainId - The current chain ID.
@@ -31,7 +30,7 @@ type GetProviderState = (
 type GetProviderStateConstraint<Params extends JsonRpcParams = JsonRpcParams> =
   {
     implementation: (
-      _req: JsonRpcRequest<Params>,
+      _req: ProviderStateHandlerRequest<Params>,
       res: PendingJsonRpcResponse<ProviderStateHandlerResult>,
       _next: JsonRpcEngineNextCallback,
       end: JsonRpcEngineEndCallback,
@@ -65,13 +64,13 @@ export default getProviderState;
 async function getProviderStateHandler<
   Params extends JsonRpcParams = JsonRpcParams,
 >(
-  req: JsonRpcRequest<Params>,
+  req: ProviderStateHandlerRequest<Params>,
   res: PendingJsonRpcResponse<ProviderStateHandlerResult>,
   _next: JsonRpcEngineNextCallback,
   end: JsonRpcEngineEndCallback,
   { getProviderState: _getProviderState }: Record<string, GetProviderState>,
 ): Promise<void> {
-  const { origin } = req.origin;
+  const { origin } = req;
   res.result = {
     ...(await _getProviderState(origin)),
   };
diff --git a/app/scripts/lib/rpc-method-middleware/handlers/types.ts b/app/scripts/lib/rpc-method-middleware/handlers/types.ts
index 5b7a2a7494d4..46ceef442ec2 100644
--- a/app/scripts/lib/rpc-method-middleware/handlers/types.ts
+++ b/app/scripts/lib/rpc-method-middleware/handlers/types.ts
@@ -1,6 +1,13 @@
+import { OriginString } from '@metamask/permission-controller';
+import { JsonRpcParams, JsonRpcRequest } from '@metamask/utils';
 import { MessageType } from '../../../../../shared/constants/app';
 
 export type HandlerWrapper = {
   methodNames: [MessageType] | MessageType[];
   hookNames: Record<string, boolean>;
 };
+
+export type HandlerRequestType<Params extends JsonRpcParams = JsonRpcParams> =
+  Required<JsonRpcRequest<Params>> & {
+    origin: OriginString;
+  };