From 20400bc7b9aebc8e7de74952055d4b519db3ef96 Mon Sep 17 00:00:00 2001
From: matejcik <ja@matejcik.cz>
Date: Tue, 5 Dec 2023 21:09:13 +0100
Subject: [PATCH] fix(rust/trezor-client): fix Solana messages

---
 rust/trezor-client/scripts/build_messages    |  1 +
 rust/trezor-client/src/messages/generated.rs | 16 ++++++++++------
 rust/trezor-client/src/protos/mod.rs         |  1 +
 3 files changed, 12 insertions(+), 6 deletions(-)

diff --git a/rust/trezor-client/scripts/build_messages b/rust/trezor-client/scripts/build_messages
index 4a434f8c7..33d736fe3 100755
--- a/rust/trezor-client/scripts/build_messages
+++ b/rust/trezor-client/scripts/build_messages
@@ -25,6 +25,7 @@ FEATURES = {
     "Monero": "monero",
     "NEM": "nem",
     "Ripple": "ripple",
+    "Solana": "solana",
     "Stellar": "stellar",
     "Tezos": "tezos",
     "WebAuthn": "webauthn",
diff --git a/rust/trezor-client/src/messages/generated.rs b/rust/trezor-client/src/messages/generated.rs
index e23118719..5e7608bf7 100644
--- a/rust/trezor-client/src/messages/generated.rs
+++ b/rust/trezor-client/src/messages/generated.rs
@@ -79,12 +79,6 @@ trezor_message_impl! {
     DebugLinkEraseSdCard => MessageType_DebugLinkEraseSdCard,
     DebugLinkWatchLayout => MessageType_DebugLinkWatchLayout,
     DebugLinkResetDebugEvents => MessageType_DebugLinkResetDebugEvents,
-    SolanaGetPublicKey => MessageType_SolanaGetPublicKey,
-    SolanaPublicKey => MessageType_SolanaPublicKey,
-    SolanaGetAddress => MessageType_SolanaGetAddress,
-    SolanaAddress => MessageType_SolanaAddress,
-    SolanaSignTx => MessageType_SolanaSignTx,
-    SolanaTxSignature => MessageType_SolanaTxSignature,
 }
 
 #[cfg(feature = "binance")]
@@ -244,6 +238,16 @@ trezor_message_impl! {
     RippleSignedTx => MessageType_RippleSignedTx,
 }
 
+#[cfg(feature = "solana")]
+trezor_message_impl! {
+    SolanaGetPublicKey => MessageType_SolanaGetPublicKey,
+    SolanaPublicKey => MessageType_SolanaPublicKey,
+    SolanaGetAddress => MessageType_SolanaGetAddress,
+    SolanaAddress => MessageType_SolanaAddress,
+    SolanaSignTx => MessageType_SolanaSignTx,
+    SolanaTxSignature => MessageType_SolanaTxSignature,
+}
+
 #[cfg(feature = "stellar")]
 trezor_message_impl! {
     StellarSignTx => MessageType_StellarSignTx,
diff --git a/rust/trezor-client/src/protos/mod.rs b/rust/trezor-client/src/protos/mod.rs
index 423abfb2a..20bcebe92 100644
--- a/rust/trezor-client/src/protos/mod.rs
+++ b/rust/trezor-client/src/protos/mod.rs
@@ -32,6 +32,7 @@ mod generated {
         "monero" => messages_monero
         "nem" => messages_nem
         "ripple" => messages_ripple
+        "solana" => messages_solana
         "stellar" => messages_stellar
         "tezos" => messages_tezos
         "webauthn" => messages_webauthn