From 2459eb4fed15846ad1f10637e847b54f0d9959aa Mon Sep 17 00:00:00 2001
From: Matthew <ramsden@hey.com>
Date: Sat, 16 Dec 2023 19:24:02 -0600
Subject: [PATCH] feat: add calculate_fee and calculate_fee_rate on wallet

---
 .../org/bitcoindevkit/LiveWalletTest.kt       |  8 ++++-
 bdk-ffi/src/bdk.udl                           | 18 +++++++++-
 bdk-ffi/src/bitcoin.rs                        |  9 +++++
 bdk-ffi/src/error.rs                          | 35 +++++++++++++++++++
 bdk-ffi/src/lib.rs                            |  3 ++
 bdk-ffi/src/types.rs                          | 14 ++++++++
 bdk-ffi/src/wallet.rs                         | 21 ++++++++---
 .../org/bitcoindevkit/LiveWalletTest.kt       |  8 ++++-
 bdk-python/tests/test_live_wallet.py          |  5 +++
 .../BitcoinDevKitTests/LiveWalletTests.swift  |  5 +++
 10 files changed, 119 insertions(+), 7 deletions(-)
 create mode 100644 bdk-ffi/src/error.rs

diff --git a/bdk-android/lib/src/androidTest/kotlin/org/bitcoindevkit/LiveWalletTest.kt b/bdk-android/lib/src/androidTest/kotlin/org/bitcoindevkit/LiveWalletTest.kt
index b9a7d54a..1e71f61f 100644
--- a/bdk-android/lib/src/androidTest/kotlin/org/bitcoindevkit/LiveWalletTest.kt
+++ b/bdk-android/lib/src/androidTest/kotlin/org/bitcoindevkit/LiveWalletTest.kt
@@ -59,8 +59,14 @@ class LiveWalletTest {
         assertTrue(walletDidSign)
 
         val tx: Transaction = psbt.extractTx()
-
         println("Txid is: ${tx.txid()}")
+
+        val txFee = wallet.calculateFee(tx)
+        println("Tx fee is: ${tx.txid()}")
+
+        val feeRate = wallet.calculateFeeRate(tx)
+        println("Tx fee is: ${tx.txid()} sat/vB")
+
         esploraClient.broadcast(tx)
     }
 }
diff --git a/bdk-ffi/src/bdk.udl b/bdk-ffi/src/bdk.udl
index b90f059d..cfc56476 100644
--- a/bdk-ffi/src/bdk.udl
+++ b/bdk-ffi/src/bdk.udl
@@ -82,6 +82,16 @@ enum BdkError {
   "Psbt",
 };
 
+[Error]
+interface CalculateFeeError {
+  MissingTxOut(sequence<OutPoint> out_points);
+  NegativeFee(i64 fee);
+};
+
+interface FeeRate {
+  f32 as_sat_per_vb();
+};
+
 enum ChangeSpendPolicy {
   "ChangeAllowed",
   "OnlyChange",
@@ -111,6 +121,12 @@ interface Wallet {
   SentAndReceivedValues sent_and_received([ByRef] Transaction tx);
 
   sequence<Transaction> transactions();
+
+  [Throws=CalculateFeeError]
+  u64 calculate_fee([ByRef] Transaction tx);
+
+  [Throws=CalculateFeeError]
+  FeeRate calculate_fee_rate([ByRef] Transaction tx);
 };
 
 interface Update {};
@@ -348,4 +364,4 @@ interface PartiallySignedTransaction {
 dictionary OutPoint {
   string txid;
   u32 vout;
-};
+};
\ No newline at end of file
diff --git a/bdk-ffi/src/bitcoin.rs b/bdk-ffi/src/bitcoin.rs
index effa01a9..9061c3f0 100644
--- a/bdk-ffi/src/bitcoin.rs
+++ b/bdk-ffi/src/bitcoin.rs
@@ -300,6 +300,15 @@ impl From<&OutPoint> for BdkOutPoint {
     }
 }
 
+impl From<&BdkOutPoint> for OutPoint {
+    fn from(outpoint: &BdkOutPoint) -> Self {
+        OutPoint {
+            txid: outpoint.txid.to_string(),
+            vout: outpoint.vout,
+        }
+    }
+}
+
 #[derive(Debug, Clone)]
 pub struct TxOut {
     pub value: u64,
diff --git a/bdk-ffi/src/error.rs b/bdk-ffi/src/error.rs
new file mode 100644
index 00000000..a3f7a7c5
--- /dev/null
+++ b/bdk-ffi/src/error.rs
@@ -0,0 +1,35 @@
+use crate::bitcoin::OutPoint;
+
+use bdk::chain::tx_graph::CalculateFeeError as BdkCalculateFeeError;
+
+use std::fmt;
+
+#[derive(Debug)]
+pub enum CalculateFeeError {
+    MissingTxOut { out_points: Vec<OutPoint> },
+    NegativeFee { fee: i64 },
+}
+
+impl fmt::Display for CalculateFeeError {
+    fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
+        match self {
+            CalculateFeeError::MissingTxOut { out_points } => {
+                write!(f, "Missing transaction output: {:?}", out_points)
+            }
+            CalculateFeeError::NegativeFee { fee } => write!(f, "Negative fee value: {}", fee),
+        }
+    }
+}
+
+impl From<BdkCalculateFeeError> for CalculateFeeError {
+    fn from(error: BdkCalculateFeeError) -> Self {
+        match error {
+            BdkCalculateFeeError::MissingTxOut(out_points) => CalculateFeeError::MissingTxOut {
+                out_points: out_points.iter().map(|op| op.into()).collect(),
+            },
+            BdkCalculateFeeError::NegativeFee(fee) => CalculateFeeError::NegativeFee { fee },
+        }
+    }
+}
+
+impl std::error::Error for CalculateFeeError {}
diff --git a/bdk-ffi/src/lib.rs b/bdk-ffi/src/lib.rs
index b4ce7fda..ec86e0f4 100644
--- a/bdk-ffi/src/lib.rs
+++ b/bdk-ffi/src/lib.rs
@@ -1,5 +1,6 @@
 mod bitcoin;
 mod descriptor;
+mod error;
 mod esplora;
 mod keys;
 mod types;
@@ -13,6 +14,7 @@ use crate::bitcoin::Script;
 use crate::bitcoin::Transaction;
 use crate::bitcoin::TxOut;
 use crate::descriptor::Descriptor;
+use crate::error::CalculateFeeError;
 use crate::esplora::EsploraClient;
 use crate::keys::DerivationPath;
 use crate::keys::DescriptorPublicKey;
@@ -21,6 +23,7 @@ use crate::keys::Mnemonic;
 use crate::types::AddressIndex;
 use crate::types::AddressInfo;
 use crate::types::Balance;
+use crate::types::FeeRate;
 use crate::types::LocalUtxo;
 use crate::types::ScriptAmount;
 use crate::wallet::BumpFeeTxBuilder;
diff --git a/bdk-ffi/src/types.rs b/bdk-ffi/src/types.rs
index 1f03ea0f..df8086e1 100644
--- a/bdk-ffi/src/types.rs
+++ b/bdk-ffi/src/types.rs
@@ -7,8 +7,22 @@ use bdk::KeychainKind;
 
 use bdk::LocalUtxo as BdkLocalUtxo;
 
+use bdk::FeeRate as BdkFeeRate;
+
 use std::sync::Arc;
 
+pub struct FeeRate(BdkFeeRate);
+
+impl FeeRate {
+    pub fn new(bdk_fee_rate: BdkFeeRate) -> Self {
+        FeeRate(bdk_fee_rate)
+    }
+
+    pub fn as_sat_per_vb(&self) -> f32 {
+        self.0.as_sat_per_vb()
+    }
+}
+
 pub struct ScriptAmount {
     pub script: Arc<Script>,
     pub amount: u64,
diff --git a/bdk-ffi/src/wallet.rs b/bdk-ffi/src/wallet.rs
index e4dcebb3..3429e51c 100644
--- a/bdk-ffi/src/wallet.rs
+++ b/bdk-ffi/src/wallet.rs
@@ -1,7 +1,8 @@
 use crate::bitcoin::{OutPoint, PartiallySignedTransaction, Transaction};
 use crate::descriptor::Descriptor;
-use crate::types::Balance;
+use crate::error::CalculateFeeError;
 use crate::types::ScriptAmount;
+use crate::types::{Balance, FeeRate};
 use crate::Script;
 use crate::{AddressIndex, AddressInfo, Network};
 
@@ -10,7 +11,7 @@ use bdk::bitcoin::psbt::PartiallySignedTransaction as BdkPartiallySignedTransact
 use bdk::bitcoin::{OutPoint as BdkOutPoint, Sequence, Txid};
 use bdk::wallet::tx_builder::ChangeSpendPolicy;
 use bdk::wallet::Update as BdkUpdate;
-use bdk::{Error as BdkError, FeeRate};
+use bdk::{Error as BdkError, FeeRate as BdkFeeRate};
 use bdk::{SignOptions, Wallet as BdkWallet};
 
 use std::collections::HashSet;
@@ -98,6 +99,18 @@ impl Wallet {
             .map(|tx| Arc::new(tx.tx_node.tx.clone().into()))
             .collect()
     }
+
+    pub fn calculate_fee(&self, tx: &Transaction) -> Result<u64, CalculateFeeError> {
+        let fee_result = self.get_wallet().calculate_fee(&tx.clone().into());
+        fee_result.map_err(|err| err.into())
+    }
+
+    pub fn calculate_fee_rate(&self, tx: &Transaction) -> Result<Arc<FeeRate>, CalculateFeeError> {
+        self.get_wallet()
+            .calculate_fee_rate(&tx.clone().into())
+            .map(|bdk_fee_rate| Arc::new(FeeRate::new(bdk_fee_rate)))
+            .map_err(|err| err.into())
+    }
 }
 
 pub struct SentAndReceivedValues {
@@ -473,7 +486,7 @@ impl TxBuilder {
             tx_builder.manually_selected_only();
         }
         if let Some(sat_per_vb) = self.fee_rate {
-            tx_builder.fee_rate(FeeRate::from_sat_per_vb(sat_per_vb));
+            tx_builder.fee_rate(BdkFeeRate::from_sat_per_vb(sat_per_vb));
         }
         if let Some(fee_amount) = self.fee_absolute {
             tx_builder.fee_absolute(fee_amount);
@@ -551,7 +564,7 @@ impl BumpFeeTxBuilder {
             Txid::from_str(self.txid.as_str()).map_err(|e| BdkError::Generic(e.to_string()))?;
         let mut wallet = wallet.get_wallet();
         let mut tx_builder = wallet.build_fee_bump(txid)?;
-        tx_builder.fee_rate(FeeRate::from_sat_per_vb(self.fee_rate));
+        tx_builder.fee_rate(BdkFeeRate::from_sat_per_vb(self.fee_rate));
         if let Some(allow_shrinking) = &self.allow_shrinking {
             tx_builder.allow_shrinking(allow_shrinking.0.clone())?;
         }
diff --git a/bdk-jvm/lib/src/test/kotlin/org/bitcoindevkit/LiveWalletTest.kt b/bdk-jvm/lib/src/test/kotlin/org/bitcoindevkit/LiveWalletTest.kt
index dda5b9fa..2cea67dd 100644
--- a/bdk-jvm/lib/src/test/kotlin/org/bitcoindevkit/LiveWalletTest.kt
+++ b/bdk-jvm/lib/src/test/kotlin/org/bitcoindevkit/LiveWalletTest.kt
@@ -55,8 +55,14 @@ class LiveWalletTest {
         assertTrue(walletDidSign)
 
         val tx: Transaction = psbt.extractTx()
-
         println("Txid is: ${tx.txid()}")
+
+        val txFee = wallet.calculateFee(tx)
+        println("Tx fee is: ${tx.txid()}")
+
+        val feeRate = wallet.calculateFeeRate(tx)
+        println("Tx fee is: ${tx.txid()} sat/vB")
+
         esploraClient.broadcast(tx)
     }
 }
diff --git a/bdk-python/tests/test_live_wallet.py b/bdk-python/tests/test_live_wallet.py
index c04f8e91..d349a0ef 100644
--- a/bdk-python/tests/test_live_wallet.py
+++ b/bdk-python/tests/test_live_wallet.py
@@ -64,6 +64,11 @@ def test_broadcast_transaction(self):
         walletDidSign = wallet.sign(psbt)
         self.assertTrue(walletDidSign)
         tx = psbt.extract_tx()
+        print(f"Transaction Id: {tx.txid}")
+        fee = wallet.calculate_fee(tx)
+        print(f"Transaction Fee: {fee}")
+        fee_rate = wallet.calculate_fee_rate(tx)
+        print(f"Transaction Fee Rate: {fee_rate} sat/vB")
         
         esploraClient.broadcast(tx)
     
diff --git a/bdk-swift/Tests/BitcoinDevKitTests/LiveWalletTests.swift b/bdk-swift/Tests/BitcoinDevKitTests/LiveWalletTests.swift
index b936ea95..0399d788 100644
--- a/bdk-swift/Tests/BitcoinDevKitTests/LiveWalletTests.swift
+++ b/bdk-swift/Tests/BitcoinDevKitTests/LiveWalletTests.swift
@@ -69,6 +69,11 @@ final class LiveWalletTests: XCTestCase {
         
         let tx: Transaction = psbt.extractTx()
         print(tx.txid())
+        let fee = try wallet.calculateFee(tx: tx)
+        print("Transaction Fee: \(fee)")
+        let feeRate = try wallet.calculateFeeRate(tx: tx)
+        print("Transaction Fee Rate: \(feeRate.asSatPerVb()) sat/vB")
+        
         try esploraClient.broadcast(transaction: tx)
     }
 }