fix(connect): fix types, re-enable assert in typedCall

trezor · Dec 20, 2023 · d75bbaf · d75bbaf
1 parent 9df30f9
commit d75bbaf
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 3 deletions.
diff --git a/packages/connect/src/device/DeviceCommands.ts b/packages/connect/src/device/DeviceCommands.ts
@@ -22,7 +22,7 @@ import { initLog } from '../utils/debug';
 import type { Device } from './Device';
 import type { CoinInfo, BitcoinNetworkInfo, Network } from '../types';
 import type { HDNodeResponse } from '../types/api/getPublicKey';
-// import { Assert } from '@trezor/schema-utils';
+import { Assert } from '@trezor/schema-utils';
 
 type MessageType = Messages.MessageType;
 type MessageKey = keyof MessageType;
@@ -422,7 +422,7 @@ export class DeviceCommands {
         }
         // Assert message type
         // msg is allowed to be undefined for some calls, in that case the schema is an empty object
-        // Assert(Messages.MessageType.properties[type], msg ?? {});
+        Assert(Messages.MessageType.properties[type], msg ?? {});
         const response = await this._commonCall(type, msg);
         try {
             assertType(response, resType);

diff --git a/packages/connect/src/types/api/bitcoin/index.ts b/packages/connect/src/types/api/bitcoin/index.ts
@@ -10,7 +10,7 @@ import { Static, Type } from '@trezor/schema-utils';
 export type SignMessage = Static<typeof SignMessage>;
 export const SignMessage = Type.Object({
     path: DerivationPath,
-    coin: Type.String(),
+    coin: Type.Optional(Type.String()),
     message: Type.String(),
     hex: Type.Optional(Type.Boolean()),
     no_script_type: Type.Optional(Type.Boolean()),

diff --git a/packages/connect/src/types/params.ts b/packages/connect/src/types/params.ts
@@ -68,6 +68,7 @@ export const GetAddress = Type.Object({
     address: Type.Optional(Type.String()),
     showOnTrezor: Type.Optional(Type.Boolean()),
     chunkify: Type.Optional(Type.Boolean()),
+    useEventListener: Type.Optional(Type.Boolean()),
 });
 
 export interface Address {