Add utxo context

aspectron · Jan 22, 2025 · e13cc86 · e13cc86
1 parent a3e0452
commit e13cc86
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/wallet/core/src/account/mod.rs b/wallet/core/src/account/mod.rs
@@ -484,7 +484,8 @@ pub trait Account: AnySync + Send + Sync + 'static {
             match result {
                 Ok(pskt) => {
                     let change = self.change_address()?;
-                    let transaction = pskt_to_pending_transaction(pskt, self.wallet().network_id()?, change)?;
+                    let transaction =
+                        pskt_to_pending_transaction(pskt, self.wallet().network_id()?, change, self.utxo_context().clone().into())?;
                     ids.push(transaction.try_submit(&self.wallet().rpc_api()).await?);
                 }
                 Err(e) => {

diff --git a/wallet/core/src/account/pskb.rs b/wallet/core/src/account/pskb.rs
@@ -290,6 +290,7 @@ pub fn pskt_to_pending_transaction(
     finalized_pskt: PSKT<Finalizer>,
     network_id: NetworkId,
     change_address: Address,
+    source_utxo_context: Option<UtxoContext>,
 ) -> Result<PendingTransaction, Error> {
     let mass = 10;
     let (signed_tx, _) = match finalized_pskt.clone().extractor() {
@@ -339,7 +340,7 @@ pub fn pskt_to_pending_transaction(
         change_address,
         utxo_iterator,
         priority_utxo_entries: None,
-        source_utxo_context: None,
+        source_utxo_context,
         destination_utxo_context: None,
         fee_rate: None,
         final_transaction_priority_fee: fee_u.into(),
@@ -495,6 +496,7 @@ pub async fn commit_reveal_batch_bundle(
             pskt_finalizer.clone(),
             network_id,
             account.clone().as_derivation_capable()?.change_address()?,
+            account.utxo_context().clone().into(),
         )
         .map_err(|_| Error::CommitTransactionIdExtractionError)?
         .id();