feat(transport): load protobuf definitions

trezor · Oct 24, 2024 · a0a7f06 · a0a7f06
1 parent de37729
commit a0a7f06
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/packages/transport/src/transports/abstract.ts b/packages/transport/src/transports/abstract.ts
@@ -13,7 +13,7 @@ import {
     TransportProtocolState,
     thp as protocolThp,
 } from '@trezor/protocol';
-import type { MessageFromTrezor as ProtobufMessageType } from '@trezor/protobuf';
+import { MessageFromTrezor as ProtobufMessageType, loadDefinitions } from '@trezor/protobuf';
 
 import {
     Session,
@@ -447,6 +447,10 @@ export abstract class AbstractTransport extends TransportEmitter {
         this.messages = protobuf.Root.fromJSON(messages);
     }
 
+    public loadMessages(packageName: string, packageLoader: Parameters<typeof loadDefinitions>[2]) {
+        return loadDefinitions(this.messages, packageName, packageLoader);
+    }
+
     protected success<T>(payload: T): Success<T> {
         return success(payload);
     }