From 01f6d2b76985a92ec19ccd37a8e4005219b94563 Mon Sep 17 00:00:00 2001
From: pythcoiner <pythcoiner@proton.me>
Date: Wed, 30 Oct 2024 11:30:16 +0100
Subject: [PATCH] gui: add support for tapminiscript on bitbox02

---
 gui/src/app/state/psbt.rs |  6 ++++++
 gui/src/hw.rs             | 18 ++++++++++++++----
 2 files changed, 20 insertions(+), 4 deletions(-)

diff --git a/gui/src/app/state/psbt.rs b/gui/src/app/state/psbt.rs
index 0573f6bba..dad7ca034 100644
--- a/gui/src/app/state/psbt.rs
+++ b/gui/src/app/state/psbt.rs
@@ -617,6 +617,12 @@ async fn sign_psbt(
                 psbt_in
                     .partial_sigs
                     .append(&mut pruned_psbt_in.partial_sigs);
+                if let Some(tap_key_sig) = pruned_psbt_in.tap_key_sig {
+                    psbt_in.tap_key_sig = Some(tap_key_sig);
+                }
+                psbt_in
+                    .tap_script_sigs
+                    .append(&mut pruned_psbt_in.tap_script_sigs);
             } else {
                 log::error!(
                     "Not all PSBT inputs are present in the pruned psbt. Pruned psbt: '{}'.",
diff --git a/gui/src/hw.rs b/gui/src/hw.rs
index 379243cad..d74b394df 100644
--- a/gui/src/hw.rs
+++ b/gui/src/hw.rs
@@ -246,6 +246,7 @@ impl HardwareWallets {
                                             let fingerprint =
                                                 bitbox2.get_master_fingerprint().await?;
                                             let mut registered = false;
+                                            let version = bitbox2.get_version().await.ok();
                                             if let Some(wallet) = &wallet {
                                                 let desc = wallet.main_descriptor.to_string();
                                                 bitbox2 = bitbox2.with_policy(&desc)?;
@@ -257,7 +258,7 @@ impl HardwareWallets {
                                                         kind: DeviceKind::BitBox02,
                                                         fingerprint,
                                                         device: bitbox2.into(),
-                                                        version: None,
+                                                        version,
                                                         registered: Some(registered),
                                                         alias: None,
                                                     })
@@ -265,7 +266,7 @@ impl HardwareWallets {
                                                     Ok(HardwareWallet::Unsupported {
                                                         id: id.clone(),
                                                         kind: DeviceKind::BitBox02,
-                                                        version: None,
+                                                        version,
                                                         reason: UnsupportedReason::NotPartOfWallet(
                                                             fingerprint,
                                                         ),
@@ -277,7 +278,7 @@ impl HardwareWallets {
                                                     kind: DeviceKind::BitBox02,
                                                     fingerprint,
                                                     device: bitbox2.into(),
-                                                    version: None,
+                                                    version,
                                                     registered: Some(registered),
                                                     alias: None,
                                                 })
@@ -851,7 +852,7 @@ fn ledger_version_supported(version: &Version) -> bool {
 
 // Kind and minimal version of devices supporting tapminiscript.
 // We cannot use a lazy_static HashMap yet, because DeviceKind does not implement Hash.
-const DEVICES_COMPATIBLE_WITH_TAPMINISCRIPT: [(DeviceKind, Option<Version>); 4] = [
+const DEVICES_COMPATIBLE_WITH_TAPMINISCRIPT: [(DeviceKind, Option<Version>); 5] = [
     (
         DeviceKind::Ledger,
         Some(Version {
@@ -872,6 +873,15 @@ const DEVICES_COMPATIBLE_WITH_TAPMINISCRIPT: [(DeviceKind, Option<Version>); 4]
             prerelease: None,
         }),
     ),
+    (
+        DeviceKind::BitBox02,
+        Some(Version {
+            major: 9,
+            minor: 21,
+            patch: 0,
+            prerelease: None,
+        }),
+    ),
 ];
 
 pub fn is_compatible_with_tapminiscript(