HWI: add get_extended_pubkey_display()

Author: pythcoiner

src/bitbox.rs

@@ -210,6 +210,28 @@ impl<T: Runtime + Sync + Send> HWI for BitBox02<T> {
         Ok(Xpub::from_str(&fg).map_err(|e| HWIError::Device(e.to_string()))?)
     }
 
+    async fn get_extended_pubkey_display(&self, path: &DerivationPath) -> Result<Xpub, HWIError> {
+        let fg = self
+            .client
+            .btc_xpub(
+                if self.network == bitcoin::Network::Bitcoin {
+                    pb::BtcCoin::Btc
+                } else {
+                    pb::BtcCoin::Tbtc
+                },
+                &Keypath::from(path),
+                if self.network == bitcoin::Network::Bitcoin {
+                    pb::btc_pub_request::XPubType::Xpub
+                } else {
+                    pb::btc_pub_request::XPubType::Tpub
+                },
+                true,
+            )
+            .await
+            .map_err(|e| HWIError::Device(e.to_string()))?;
+        Ok(Xpub::from_str(&fg).map_err(|e| HWIError::Device(e.to_string()))?)
+    }
+
     async fn display_address(&self, script: &AddressScript) -> Result<(), HWIError> {
         match script {
             AddressScript::P2TR(path) => {

src/ledger.rs

@@ -97,6 +97,13 @@ impl<T: Transport + Sync + Send> HWI for Ledger<T> {
             .await?)
     }
 
+    async fn get_extended_pubkey_display(&self, path: &DerivationPath) -> Result<Xpub, HWIError> {
+        Ok(self
+            .client
+            .get_extended_pubkey(path, true)
+            .await?)
+    }
+
     async fn display_address(&self, script: &AddressScript) -> Result<(), HWIError> {
         match script {
             AddressScript::P2TR(path) => {

src/lib.rs

@@ -75,6 +75,11 @@ pub trait HWI: Debug {
     async fn get_master_fingerprint(&self) -> Result<Fingerprint, Error>;
     /// Get the xpub with the given derivation path.
     async fn get_extended_pubkey(&self, path: &DerivationPath) -> Result<Xpub, Error>;
+    /// Get the xpub with the given derivation path, will ask user ACK for non "standard"
+    /// derivation path on some devices.
+    async fn get_extended_pubkey_display(&self, path: &DerivationPath) -> Result<Xpub, Error> {
+        self.get_extended_pubkey(path).await
+    }
     /// Register a new wallet policy.
     async fn register_wallet(&self, name: &str, policy: &str) -> Result<Option<[u8; 32]>, Error>;
     /// Returns true if the wallet is registered on the device.