From 4afeae167ee47f5666e77e0f44a97379baeed0aa Mon Sep 17 00:00:00 2001 From: Niranjana Binoy <43930900+NiranjanaBinoy@users.noreply.github.com> Date: Thu, 28 Mar 2024 12:20:38 -0400 Subject: [PATCH] updating to pendingjsonrpcresponse --- .../handlers/log-web3-shim-usage.ts | 64 ++++++++----------- 1 file changed, 27 insertions(+), 37 deletions(-) diff --git a/app/scripts/lib/rpc-method-middleware/handlers/log-web3-shim-usage.ts b/app/scripts/lib/rpc-method-middleware/handlers/log-web3-shim-usage.ts index d24e6fe55bbf..2c0afe62a481 100644 --- a/app/scripts/lib/rpc-method-middleware/handlers/log-web3-shim-usage.ts +++ b/app/scripts/lib/rpc-method-middleware/handlers/log-web3-shim-usage.ts @@ -4,52 +4,48 @@ import type { } from '@metamask/json-rpc-engine'; import type { JsonRpcRequest, - JsonRpcResponse, JsonRpcParams, - Json, + PendingJsonRpcResponse, } from '@metamask/utils'; import { MESSAGE_TYPE } from '../../../../../shared/constants/app'; import { - HandlerWrapperType, - getWeb3ShimUsageStateType, - setWeb3ShimUsageRecordedType, -} from './handlers-helpers'; + HandlerWrapper, + GetWeb3ShimUsageState, + SetWeb3ShimUsageRecorded, +} from './handlers-helper'; -type logWeb3ShimUsageOptionsType = { - getWeb3ShimUsageState: getWeb3ShimUsageStateType; - setWeb3ShimUsageRecorded: setWeb3ShimUsageRecordedType; +type LogWeb3ShimUsageOptions = { + getWeb3ShimUsageState: GetWeb3ShimUsageState; + setWeb3ShimUsageRecorded: SetWeb3ShimUsageRecorded; }; - -type logWeb3ShimUsageType< - Params extends JsonRpcParams = JsonRpcParams, - Result extends Json = Json, -> = { - implementation: ( - req: JsonRpcRequest, - res: JsonRpcResponse, - _next: JsonRpcEngineNextCallback, - end: JsonRpcEngineEndCallback, - { - getWeb3ShimUsageState, - setWeb3ShimUsageRecorded, - }: logWeb3ShimUsageOptionsType, - ) => void; -} & HandlerWrapperType; - +type LogWeb3ShimUsageConstraint = + { + implementation: ( + req: JsonRpcRequest, + res: PendingJsonRpcResponse, + _next: JsonRpcEngineNextCallback, + end: JsonRpcEngineEndCallback, + { + getWeb3ShimUsageState, + setWeb3ShimUsageRecorded, + }: LogWeb3ShimUsageOptions, + ) => void; + } & HandlerWrapper; /** * This RPC method is called by the inpage provider whenever it detects the * accessing of a non-existent property on our window.web3 shim. We use this * to alert the user that they are using a legacy dapp, and will have to take * further steps to be able to use it. */ -const logWeb3ShimUsage: logWeb3ShimUsageType = { +const logWeb3ShimUsage = { methodNames: [MESSAGE_TYPE.LOG_WEB3_SHIM_USAGE], implementation: logWeb3ShimUsageHandler, hookNames: { getWeb3ShimUsageState: true, setWeb3ShimUsageRecorded: true, }, -}; +} satisfies LogWeb3ShimUsageConstraint; + export default logWeb3ShimUsage; /** @@ -63,18 +59,12 @@ export default logWeb3ShimUsage; * @param options.setWeb3ShimUsageRecorded - A function that records web3 shim * usage for a particular origin. */ -function logWeb3ShimUsageHandler< - Params extends JsonRpcParams = JsonRpcParams, - Result extends Json = Json, ->( +function logWeb3ShimUsageHandler( req: JsonRpcRequest, - res: JsonRpcResponse, + res: PendingJsonRpcResponse, _next: JsonRpcEngineNextCallback, end: JsonRpcEngineEndCallback, - { - getWeb3ShimUsageState, - setWeb3ShimUsageRecorded, - }: logWeb3ShimUsageOptionsType, + { getWeb3ShimUsageState, setWeb3ShimUsageRecorded }: LogWeb3ShimUsageOptions, ) { const { origin } = req; if (getWeb3ShimUsageState(origin) === undefined) {