1 files changed, 38 insertions, 0 deletions
diff --git a/src/device_trezor/trezor/tools/build_protob.py b/src/device_trezor/trezor/tools/build_protob.py
new file mode 100644
index 000000000..2611f3296
--- /dev/null
+++ b/src/device_trezor/trezor/tools/build_protob.py
@@ -0,0 +1,38 @@
+#!/usr/bin/env python
+import os
+import subprocess
+import sys
+
+CWD = os.path.dirname(os.path.realpath(__file__))
+ROOT_DIR = os.path.abspath(os.path.join(CWD, "..", "..", "..", ".."))
+TREZOR_COMMON = os.path.join(ROOT_DIR, "external", "trezor-common")
+TREZOR_MESSAGES = os.path.join(CWD, "..", "messages")
+
+# check for existence of the submodule directory
+common_defs = os.path.join(TREZOR_COMMON, "defs")
+if not os.path.exists(common_defs):
+    raise ValueError(
+        "trezor-common submodule seems to be missing.\n"
+        + 'Use "git submodule update --init --recursive" to retrieve it.'
+    )
+
+# regenerate messages
+try:
+    selected = [
+        "messages.proto",
+        "messages-common.proto",
+        "messages-management.proto",
+        "messages-monero.proto",
+    ]
+    proto_srcs = [os.path.join(TREZOR_COMMON, "protob", x) for x in selected]
+    exec_args = [
+        sys.executable,
+        os.path.join(CWD, "pb2cpp.py"),
+        "-o",
+        TREZOR_MESSAGES,
+    ] + proto_srcs
+
+    subprocess.check_call(exec_args)
+
+except Exception as e:
+    raise