feat(transport): load protobuf definitions

trezor · Dec 12, 2024 · fa7ff3b · fa7ff3b
1 parent ea65a43
commit fa7ff3b
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
@@ -7,7 +7,7 @@ import {
     TransportProtocolState,
     thp as protocolThp,
 } from '@trezor/protocol';
-import { MessageFromTrezor as ProtobufMessageType } from '@trezor/protobuf';
+import { MessageFromTrezor as ProtobufMessageType, loadDefinitions } from '@trezor/protobuf';
 
 import {
     Session,
@@ -355,6 +355,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);
     }