tmp resetDevice as LoadDevice

trezor · Aug 29, 2024 · cb1857f · cb1857f
1 parent de4ae05
commit cb1857f
Showing 1 changed file with 31 additions and 5 deletions.
diff --git a/packages/connect/src/api/resetDevice.ts b/packages/connect/src/api/resetDevice.ts
@@ -4,7 +4,7 @@ import { AbstractMethod } from '../core/AbstractMethod';
 import { UI } from '../events';
 import { getFirmwareRange } from './common/paramsValidator';
 import { PROTO } from '../constants';
-import { Assert } from '@trezor/schema-utils';
+// import { Assert } from '@trezor/schema-utils';
 
 export default class ResetDevice extends AbstractMethod<'resetDevice', PROTO.ResetDevice> {
     init() {
@@ -15,7 +15,7 @@ export default class ResetDevice extends AbstractMethod<'resetDevice', PROTO.Res
 
         const { payload } = this;
         // validate bundle type
-        Assert(PROTO.ResetDevice, payload);
+        // Assert(PROTO.ResetDevice, payload);
 
         this.params = {
             display_random: payload.display_random,
@@ -43,9 +43,35 @@ export default class ResetDevice extends AbstractMethod<'resetDevice', PROTO.Res
     }
 
     async run() {
-        const cmd = this.device.getCommands();
-        const response = await cmd.typedCall('ResetDevice', 'Success', this.params);
+        // const cmd = this.device.getCommands();
+        // const response = await cmd.typedCall('ResetDevice', 'Success', this.params);
+        console.warn('----> running reset function!');
+        const response = await this.device.transport.call({
+            session: this.device.activitySessionID!,
+            name: 'LoadDevice',
+            data: {
+                pin: '',
+                label: 'THP device',
+                passphrase_protection: true,
+                mnemonics: ['all all all all all all all all all all all all'],
+                skip_checksum: true,
+            },
+            protocol: this.device.protocol,
+            protocolState: this.device.transportState,
+        }).promise;
 
-        return response.message;
+        if (response.success && response.payload.type === 'ButtonRequest') {
+            return this.device.transport.call({
+                session: this.device.activitySessionID!,
+                name: 'ButtonAck',
+                data: {},
+                protocol: this.device.protocol,
+                protocolState: this.device.transportState,
+            }).promise;
+        }
+
+        console.warn('----> end reset function!', response);
+
+        return response as any;
     }
 }