exploring bdk-tx with bdk_wallet

Author: thunderbiscuit

Cargo.toml

@@ -19,6 +19,8 @@ bdk_tx = { path = "." }
 bitcoin = { version = "0.32", features = ["rand-std"] }
 bdk_testenv = "0.11.1"
 bdk_bitcoind_rpc = "0.18.0"
+bdk_wallet = "1.2.0"
+bdk_esplora = { version = "0.20.1", features = ["blocking"] }
 
 [features]
 default = ["std"]

tests/synopsis.rs

@@ -1,12 +1,19 @@
+use std::collections::BTreeMap;
+use std::process::exit;
+use std::str::FromStr;
 use bdk_bitcoind_rpc::Emitter;
-use bdk_chain::{bdk_core, Balance};
+use bdk_chain::{bdk_core, Balance, KeychainIndexed};
+use bdk_chain::spk_client::{FullScanRequestBuilder, FullScanResponse, SyncRequestBuilder, SyncResponse};
 use bdk_testenv::{bitcoincore_rpc::RpcApi, TestEnv};
-use bdk_tx::{
-    create_psbt, create_selection, get_candidate_inputs, CreatePsbtParams, CreateSelectionParams,
-    InputGroup, Output, Signer,
-};
-use bitcoin::{key::Secp256k1, Address, Amount, BlockHash, FeeRate};
+use bdk_wallet::{AddressInfo, KeychainKind, LocalOutput, SignOptions};
+use bdk_tx::{create_psbt, create_selection, get_candidate_inputs, CreatePsbtParams, CreateSelectionParams, InputGroup, Output, Signer};
+use bitcoin::{key::Secp256k1, Address, Amount, BlockHash, FeeRate, Network, OutPoint};
 use miniscript::{plan::Assets, Descriptor, DescriptorPublicKey};
+use bdk_esplora::esplora_client;
+use bdk_esplora::esplora_client::Builder;
+use bdk_esplora::EsploraExt;
+use bitcoin::address::NetworkChecked;
+use miniscript::descriptor::KeyMap;
 
 const EXTERNAL: &str = "external";
 const INTERNAL: &str = "internal";
@@ -148,3 +155,112 @@ fn synopsis() -> anyhow::Result<()> {
 
     Ok(())
 }
+
+#[test]
+fn bdk_wallet_simple_tx() -> anyhow::Result<()> {
+
+    const STOP_GAP: usize = 20;
+    const PARALLEL_REQUESTS: usize = 1;
+    let secp = bitcoin::secp256k1::Secp256k1::new();
+
+    let descriptor_private: &str = "tr(tprv8ZgxMBicQKsPdNRGG6HuFapxQCFxsDDf7TDsV8tdUgZDdiiyA6dB2ssN4RSXyp52V3MRBm4KqAps3Txng59rNMUtUEtMPDphKkKDXmamd2T/86'/1'/0'/0/*)#usy7l3tt";
+    let change_descriptor_private: &str = "tr(tprv8ZgxMBicQKsPdNRGG6HuFapxQCFxsDDf7TDsV8tdUgZDdiiyA6dB2ssN4RSXyp52V3MRBm4KqAps3Txng59rNMUtUEtMPDphKkKDXmamd2T/86'/1'/0'/1/*)#dyplzymn";
+
+    let (descriptor, _): (Descriptor<DescriptorPublicKey>, KeyMap) = Descriptor::parse_descriptor(&secp, "tr(tprv8ZgxMBicQKsPdNRGG6HuFapxQCFxsDDf7TDsV8tdUgZDdiiyA6dB2ssN4RSXyp52V3MRBm4KqAps3Txng59rNMUtUEtMPDphKkKDXmamd2T/86'/1'/0'/0/*)#usy7l3tt")?;
+    let (change_descriptor, _): (Descriptor<DescriptorPublicKey>, KeyMap) = Descriptor::parse_descriptor(&secp, "tr(tprv8ZgxMBicQKsPdNRGG6HuFapxQCFxsDDf7TDsV8tdUgZDdiiyA6dB2ssN4RSXyp52V3MRBm4KqAps3Txng59rNMUtUEtMPDphKkKDXmamd2T/86'/1'/0'/1/*)#dyplzymn")?;
+
+    // Create the wallet
+    let mut wallet = bdk_wallet::Wallet::create(descriptor_private, change_descriptor_private)
+        .network(Network::Regtest)
+        .create_wallet_no_persist()?;
+
+    let client: esplora_client::BlockingClient = Builder::new("http://127.0.0.1:3002").build_blocking();
+
+    println!("Syncing wallet...");
+    let full_scan_request: FullScanRequestBuilder<KeychainKind> = wallet.start_full_scan();
+    let update: FullScanResponse<KeychainKind> = client
+        .full_scan(full_scan_request, STOP_GAP, PARALLEL_REQUESTS)?;
+
+    // Apply the update from the full scan to the wallet
+    wallet.apply_update(update)?;
+
+    let balance = wallet.balance();
+    println!("Wallet balance: {} sat", balance.total().to_sat());
+
+    if balance.total().to_sat() < 300000 {
+        println!("Your wallet does not have sufficient balance for the following steps!");
+        // Reveal a new address from your external keychain
+        let address: AddressInfo = wallet.reveal_next_address(KeychainKind::External);
+        println!(
+            "Send coins to {} (address generated at index {})",
+            address.address, address.index
+        );
+        exit(0)
+    }
+
+    let local_outputs: Vec<LocalOutput> = wallet.list_unspent().collect();
+    dbg!(&local_outputs.len());
+    // dbg!(&local_outputs);
+    let outpoints: Vec<KeychainIndexed<KeychainKind, OutPoint>> = local_outputs
+        .into_iter()
+        .map(|o| ((o.keychain, o.derivation_index), o.outpoint.clone()))
+        .collect();
+
+    let mut descriptors_map = BTreeMap::new();
+    descriptors_map.insert(KeychainKind::External, descriptor.clone());
+    descriptors_map.insert(KeychainKind::Internal, change_descriptor.clone());
+
+    let input_candidates_individual = get_candidate_inputs(
+        &wallet.tx_graph(),
+        &wallet.local_chain(),
+        outpoints,
+        descriptors_map,
+        Assets::new(),
+    )?;
+    // dbg!(&input_candidates_0);
+
+    let recipient_address: Address<NetworkChecked> =
+        Address::from_str("bcrt1qe908k9zu8m4jgzdddgg0lkj73yctfqueg7pea9")?
+            .require_network(Network::Regtest)?;
+
+    let input_candidates_groups: Vec<InputGroup> = input_candidates_individual.into_iter().map(|i| i.into()).collect();
+
+    let selection = create_selection(CreateSelectionParams {
+        input_candidates: input_candidates_groups,
+        must_spend: Default::default(),
+        change_descriptor: change_descriptor.at_derivation_index(0)?,
+        target_feerate: FeeRate::from_sat_per_vb(5).unwrap(),
+        long_term_feerate: Some(FeeRate::from_sat_per_vb(1).unwrap()),
+        target_outputs: vec![Output::with_script(
+            recipient_address.script_pubkey(),
+            Amount::from_sat(200_000),
+        )],
+        max_rounds: 100_000,
+    })?;
+
+    dbg!(&selection.inputs.len());
+
+    let (mut psbt, _) = create_psbt(CreatePsbtParams {
+        inputs: selection.inputs,
+        outputs: selection.outputs,
+        ..Default::default()
+    })?;
+    let signed = wallet.sign(&mut psbt, SignOptions::default())?;
+    assert!(signed);
+    let tx = psbt.extract_tx()?;
+
+    client.broadcast(&tx)?;
+    dbg!("tx broadcast: {}", tx.compute_txid());
+
+    println!("Syncing wallet again...");
+    let sync_request = wallet.start_sync_with_revealed_spks();
+    let update_2: SyncResponse = client
+        .sync(sync_request, PARALLEL_REQUESTS)?;
+
+    wallet.apply_update(update_2)?;
+
+    let balance_2 = wallet.balance();
+    println!("Wallet balance: {} sat", balance_2.total().to_sat());
+
+    Ok(())
+}