feat: add apply_block_relevant/batch_insert_relevant_uncofirmed to wallet

Author: vladimirfomene

crates/bdk/src/wallet/mod.rs

@@ -33,8 +33,8 @@ use bitcoin::psbt;
 use bitcoin::secp256k1::Secp256k1;
 use bitcoin::sighash::{EcdsaSighashType, TapSighashType};
 use bitcoin::{
-    absolute, Address, Network, OutPoint, Script, ScriptBuf, Sequence, Transaction, TxOut, Txid,
-    Weight, Witness,
+    absolute, Address, Block, Network, OutPoint, Script, ScriptBuf, Sequence, Transaction, TxOut,
+    Txid, Weight, Witness,
 };
 use core::fmt;
 use core::ops::Deref;
@@ -1949,6 +1949,18 @@ impl<D> Wallet<D> {
         Ok(())
     }
 
+    /// Stages wallet changes in memory for later commit.
+    ///
+    /// To commit changes to the persistence backend, call [`commit`].
+    ///
+    /// [`commit`]: Self::commit
+    pub fn stage(&mut self, changeset: ChangeSet)
+    where
+        D: PersistBackend<ChangeSet>,
+    {
+        self.persist.stage(changeset);
+    }
+
     /// Commits all currently [`staged`] changed to the persistence backend returning and error when
     /// this fails.
     ///
@@ -1989,12 +2001,39 @@ impl<D> Wallet<D> {
 
     /// Set lookahead for all keychains.
     pub fn set_lookahead_for_all(&mut self, lookahead: u32) {
-        // Having a lookahead of 0, means if we sync the chain 
+        // Having a lookahead of 0, means if we sync the chain
         // we would not find any update since we don't have any
         // store scripts in our indexer.
         assert_ne!(lookahead, 0, "lookahead must be greater than 0");
         self.indexed_graph.index.set_lookahead_for_all(lookahead);
     }
+
+    /// Insert all the block's relevant transactions into the IndexedTxGraph.
+    pub fn apply_block_relevant(
+        &mut self,
+        block: Block,
+        height: u32,
+    ) -> Result<ChangeSet, CannotConnectError> {
+        let chain_update = CheckPoint::from_header(&block.header, height).into_update(false);
+        let mut changeset = ChangeSet::from(self.chain.apply_update(chain_update)?);
+        changeset.append(ChangeSet::from(
+            self.indexed_graph.apply_block_relevant(block, height),
+        ));
+        Ok(changeset)
+    }
+
+    /// Batch insert unconfirmed transactions into the IndexedTxGraph.
+    /// Filtering out those that are not relevant.
+    ///
+    /// Read more here: [`self.indexed_graph.batch_insert_relevant_unconfirmed()`]
+    pub fn batch_insert_relevant_unconfirmed<'t>(
+        &mut self,
+        unconfirmed_txs: impl IntoIterator<Item = (&'t Transaction, u64)>,
+    ) -> indexed_tx_graph::ChangeSet<ConfirmationTimeAnchor, keychain::ChangeSet<KeychainKind>>
+    {
+        self.indexed_graph
+            .batch_insert_relevant_unconfirmed(unconfirmed_txs)
+    }
 }
 
 impl<D> AsRef<bdk_chain::tx_graph::TxGraph<ConfirmationTimeAnchor>> for Wallet<D> {