Update message types (#282)

* update message types * lint fix * fix test * linto fixo
MetaMask · Jan 22, 2024 · 5092082 · 5092082
1 parent 9a73980
commit 5092082
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 10 deletions.
diff --git a/src/wallet.test.ts b/src/wallet.test.ts
@@ -2,7 +2,7 @@ import { providerFromEngine } from '@metamask/eth-json-rpc-provider';
 import { JsonRpcEngine } from '@metamask/json-rpc-engine';
 import pify from 'pify';
 
-import type { TransactionParams, MessageParams } from '.';
+import type { TransactionParams, MessageParams, TypedMessageV1Params } from '.';
 import { createWalletMiddleware } from '.';
 
 const testAddresses = [
@@ -237,8 +237,8 @@ describe('wallet', () => {
     it('should sign with a valid address', async () => {
       const { engine } = createTestSetup();
       const getAccounts = async () => testAddresses.slice();
-      const witnessedMsgParams: MessageParams[] = [];
-      const processTypedMessage = async (msgParams: MessageParams) => {
+      const witnessedMsgParams: TypedMessageV1Params[] = [];
+      const processTypedMessage = async (msgParams: TypedMessageV1Params) => {
         witnessedMsgParams.push(msgParams);
         return testMsgSig;
       };
@@ -266,14 +266,15 @@ describe('wallet', () => {
         from: testAddresses[0],
         data: message,
         signatureMethod: 'eth_signTypedData',
+        version: 'V1',
       });
     });
 
     it('should throw with invalid address', async () => {
       const { engine } = createTestSetup();
       const getAccounts = async () => testAddresses.slice();
-      const witnessedMsgParams: MessageParams[] = [];
-      const processTypedMessage = async (msgParams: MessageParams) => {
+      const witnessedMsgParams: TypedMessageV1Params[] = [];
+      const processTypedMessage = async (msgParams: TypedMessageV1Params) => {
         witnessedMsgParams.push(msgParams);
         return testMsgSig;
       };
@@ -299,8 +300,8 @@ describe('wallet', () => {
     it('should throw with unknown address', async () => {
       const { engine } = createTestSetup();
       const getAccounts = async () => testAddresses.slice();
-      const witnessedMsgParams: MessageParams[] = [];
-      const processTypedMessage = async (msgParams: MessageParams) => {
+      const witnessedMsgParams: TypedMessageV1Params[] = [];
+      const processTypedMessage = async (msgParams: TypedMessageV1Params) => {
         witnessedMsgParams.push(msgParams);
         return testMsgSig;
       };

diff --git a/src/wallet.ts b/src/wallet.ts
@@ -39,6 +39,10 @@ export type TypedMessageParams = MessageParams & {
   version: string;
 };
 
+export type TypedMessageV1Params = Omit<TypedMessageParams, 'data'> & {
+  data: Record<string, unknown>[];
+};
+
 export interface WalletMiddlewareOptions {
   getAccounts: (req: JsonRpcRequest) => Promise<string[]>;
   processDecryptMessage?: (
@@ -66,7 +70,7 @@ export interface WalletMiddlewareOptions {
     req: JsonRpcRequest,
   ) => Promise<string>;
   processTypedMessage?: (
-    msgParams: MessageParams,
+    msgParams: TypedMessageV1Params,
     req: JsonRpcRequest,
     version: string,
   ) => Promise<string>;
@@ -234,16 +238,21 @@ WalletMiddlewareOptions): JsonRpcMiddleware<any, Block> {
       throw rpcErrors.invalidInput();
     }
 
-    const params = req.params as [string, string, Record<string, string>?];
+    const params = req.params as [
+      Record<string, unknown>[],
+      string,
+      Record<string, string>?,
+    ];
     const message = params[0];
     const address = await validateAndNormalizeKeyholder(params[1], req);
     const version = 'V1';
     const extraParams = params[2] || {};
-    const msgParams: MessageParams = {
+    const msgParams: TypedMessageV1Params = {
       ...extraParams,
       from: address,
       data: message,
       signatureMethod: 'eth_signTypedData',
+      version,
     };
 
     res.result = await processTypedMessage(msgParams, req, version);