From 72659a46d4f9bd571fe87985f2720db848092e2d Mon Sep 17 00:00:00 2001 From: Steve Myers Date: Wed, 10 Jan 2024 14:43:57 -0600 Subject: [PATCH] example(wallet): update esplora examples to use full_scan and sync requests --- crates/bdk/src/wallet/error.rs | 2 +- .../wallet_esplora_async/Cargo.toml | 2 + .../wallet_esplora_async/src/main.rs | 121 ++++++++++++------ .../wallet_esplora_blocking/.gitignore | 2 +- .../wallet_esplora_blocking/Cargo.toml | 2 + .../wallet_esplora_blocking/src/main.rs | 110 ++++++++++------ 6 files changed, 156 insertions(+), 83 deletions(-) diff --git a/crates/bdk/src/wallet/error.rs b/crates/bdk/src/wallet/error.rs index a90083fa12..b4a77b5855 100644 --- a/crates/bdk/src/wallet/error.rs +++ b/crates/bdk/src/wallet/error.rs @@ -246,7 +246,7 @@ impl

From for CreateTxError

{ impl std::error::Error for CreateTxError

{} #[derive(Debug)] -/// Error returned by [`Wallet::build_fee_bump`] +/// Error returned by [`crate::Wallet::build_fee_bump`] pub enum BuildFeeBumpError { /// Happens when trying to spend an UTXO that is not in the internal database UnknownUtxo(OutPoint), diff --git a/example-crates/wallet_esplora_async/Cargo.toml b/example-crates/wallet_esplora_async/Cargo.toml index c588a87aa6..0d07558add 100644 --- a/example-crates/wallet_esplora_async/Cargo.toml +++ b/example-crates/wallet_esplora_async/Cargo.toml @@ -11,3 +11,5 @@ bdk_esplora = { path = "../../crates/esplora", features = ["async-https"] } bdk_file_store = { path = "../../crates/file_store" } tokio = { version = "1", features = ["rt", "rt-multi-thread", "macros"] } anyhow = "1" +env_logger = { version = "0.10.1", default-features = false } +log = "0.4.20" diff --git a/example-crates/wallet_esplora_async/src/main.rs b/example-crates/wallet_esplora_async/src/main.rs index 690cd87e24..a50e8dff47 100644 --- a/example-crates/wallet_esplora_async/src/main.rs +++ b/example-crates/wallet_esplora_async/src/main.rs @@ -1,4 +1,6 @@ -use std::{io::Write, str::FromStr}; +use env_logger::Env; +use std::env; +use std::str::FromStr; use bdk::{ bitcoin::{Address, Network}, @@ -7,6 +9,7 @@ use bdk::{ }; use bdk_esplora::{esplora_client, EsploraAsyncExt}; use bdk_file_store::Store; +use log::info; const DB_MAGIC: &str = "bdk_wallet_esplora_async_example"; const SEND_AMOUNT: u64 = 5000; @@ -15,7 +18,12 @@ const PARALLEL_REQUESTS: usize = 5; #[tokio::main] async fn main() -> Result<(), anyhow::Error> { - let db_path = std::env::temp_dir().join("bdk-esplora-async-example"); + env_logger::Builder::from_env(Env::default().default_filter_or("info")).init(); + + let args: Vec = env::args().collect(); + let cmd = args.get(1); + + let db_path = "bdk_wallet_esplora_async_example.dat"; let db = Store::::open_or_create_new(DB_MAGIC.as_bytes(), db_path)?; let external_descriptor = "wpkh(tprv8ZgxMBicQKsPdy6LMhUtFHAgpocR8GC6QmwMSFpZs7h6Eziw3SpThFfczTDh5rW2krkqffa11UpX3XkeTTB2FvzZKWXqPY54Y6Rq4AQ5R8L/84'/1'/0'/0/*)"; let internal_descriptor = "wpkh(tprv8ZgxMBicQKsPdy6LMhUtFHAgpocR8GC6QmwMSFpZs7h6Eziw3SpThFfczTDh5rW2krkqffa11UpX3XkeTTB2FvzZKWXqPY54Y6Rq4AQ5R8L/84'/1'/0'/1/*)"; @@ -24,54 +32,85 @@ async fn main() -> Result<(), anyhow::Error> { external_descriptor, Some(internal_descriptor), db, - Network::Testnet, + Network::Signet, )?; let address = wallet.try_get_address(AddressIndex::New)?; - println!("Generated Address: {}", address); + info!("Generated Address: {}", address); let balance = wallet.get_balance(); - println!("Wallet balance before syncing: {} sats", balance.total()); - - print!("Syncing..."); - let client = - esplora_client::Builder::new("https://blockstream.info/testnet/api").build_async()?; - - let prev_tip = wallet.latest_checkpoint(); - let keychain_spks = wallet - .all_unbounded_spk_iters() - .into_iter() - .map(|(k, k_spks)| { - let mut once = Some(()); - let mut stdout = std::io::stdout(); - let k_spks = k_spks - .inspect(move |(spk_i, _)| match once.take() { - Some(_) => print!("\nScanning keychain [{:?}]", k), - None => print!(" {:<3}", spk_i), - }) - .inspect(move |_| stdout.flush().expect("must flush")); - (k, k_spks) - }) - .collect(); - let (update_graph, last_active_indices) = client - .full_scan(keychain_spks, STOP_GAP, PARALLEL_REQUESTS) - .await?; - let missing_heights = update_graph.missing_heights(wallet.local_chain()); - let chain_update = client.update_local_chain(prev_tip, missing_heights).await?; - let update = Update { - last_active_indices, - graph: update_graph, - chain: Some(chain_update), - }; + info!("Wallet balance: {} sats", balance.total()); + + let client = esplora_client::Builder::new("http://signet.bitcoindevkit.net").build_async()?; + + let (update, cmd) = match cmd.map(|c| c.as_str()) { + Some(cmd) if cmd == "fullscan" => { + info!("Start full scan..."); + // 1. get data required to do a wallet full_scan + let request = wallet.full_scan_request(); + // 2. full scan to discover wallet transactions + let (graph_update, last_active_indices) = client + .full_scan(request.spks_by_keychain, STOP_GAP, PARALLEL_REQUESTS) + .await?; + // 3. from wallet transactions determine missing blockchain heights + let missing_heights = graph_update.missing_heights(wallet.local_chain()); + // 4. get blockchain update from original request checkpoint and missing heights + let chain_update = client + .update_local_chain(request.checkpoint, missing_heights) + .await?; + // 5. create wallet update + Ok(( + Update { + last_active_indices, + graph: graph_update, + chain: Some(chain_update), + }, + cmd, + )) + } + Some(cmd) if cmd == "sync" => { + info!("Start sync..."); + // 1. get data required to do a wallet sync, if also syncing previously used addresses set unused_spks_only = false + let request = wallet.sync_request(true); + // 2. sync unused wallet spks (addresses), unconfirmed tx, and utxos + let graph_update = client + .sync( + request.spks, + request.txids, + request.outpoints, + PARALLEL_REQUESTS, + ) + .await?; + // 3. from wallet transactions determine missing blockchain heights + let missing_heights = graph_update.missing_heights(wallet.local_chain()); + // 4. get blockchain update from original request checkpoint and missing heights + let chain_update = client + .update_local_chain(request.checkpoint, missing_heights) + .await?; + // 5. create wallet update + Ok(( + Update { + graph: graph_update, + chain: Some(chain_update), + ..Update::default() + }, + cmd, + )) + } + _ => Err(()), + } + .expect("Specify if you want to do a wallet 'fullscan' or a 'sync'."); + + // 6. apply update to wallet wallet.apply_update(update)?; + // 7. commit wallet update to database wallet.commit()?; - println!(); let balance = wallet.get_balance(); - println!("Wallet balance after syncing: {} sats", balance.total()); + info!("Wallet balance after {}: {} sats", cmd, balance.total()); if balance.total() < SEND_AMOUNT { - println!( + info!( "Please send at least {} sats to the receiving address", SEND_AMOUNT ); @@ -79,7 +118,7 @@ async fn main() -> Result<(), anyhow::Error> { } let faucet_address = Address::from_str("mkHS9ne12qx9pS9VojpwU5xtRd4T7X7ZUt")? - .require_network(Network::Testnet)?; + .require_network(Network::Signet)?; let mut tx_builder = wallet.build_tx(); tx_builder @@ -92,7 +131,7 @@ async fn main() -> Result<(), anyhow::Error> { let tx = psbt.extract_tx(); client.broadcast(&tx).await?; - println!("Tx broadcasted! Txid: {}", tx.txid()); + info!("Tx broadcasted! Txid: {}", tx.txid()); Ok(()) } diff --git a/example-crates/wallet_esplora_blocking/.gitignore b/example-crates/wallet_esplora_blocking/.gitignore index 630a732ef2..b5ca8a0b45 100644 --- a/example-crates/wallet_esplora_blocking/.gitignore +++ b/example-crates/wallet_esplora_blocking/.gitignore @@ -1 +1 @@ -bdk_wallet_esplora_async_example.dat +bdk_wallet_esplora_blocking_example.dat diff --git a/example-crates/wallet_esplora_blocking/Cargo.toml b/example-crates/wallet_esplora_blocking/Cargo.toml index 0679bd8f38..7d647f9c41 100644 --- a/example-crates/wallet_esplora_blocking/Cargo.toml +++ b/example-crates/wallet_esplora_blocking/Cargo.toml @@ -11,3 +11,5 @@ bdk = { path = "../../crates/bdk" } bdk_esplora = { path = "../../crates/esplora", features = ["blocking"] } bdk_file_store = { path = "../../crates/file_store" } anyhow = "1" +env_logger = { version = "0.10.1", default-features = false } +log = "0.4.20" diff --git a/example-crates/wallet_esplora_blocking/src/main.rs b/example-crates/wallet_esplora_blocking/src/main.rs index 73bfdd5598..2a9c0c9588 100644 --- a/example-crates/wallet_esplora_blocking/src/main.rs +++ b/example-crates/wallet_esplora_blocking/src/main.rs @@ -1,9 +1,12 @@ -const DB_MAGIC: &str = "bdk_wallet_esplora_example"; +const DB_MAGIC: &str = "bdk_wallet_esplora_blocking_example"; const SEND_AMOUNT: u64 = 1000; const STOP_GAP: usize = 5; const PARALLEL_REQUESTS: usize = 1; -use std::{io::Write, str::FromStr}; +use env_logger::Env; +use log::info; +use std::env; +use std::str::FromStr; use bdk::{ bitcoin::{Address, Network}, @@ -14,7 +17,12 @@ use bdk_esplora::{esplora_client, EsploraExt}; use bdk_file_store::Store; fn main() -> Result<(), anyhow::Error> { - let db_path = std::env::temp_dir().join("bdk-esplora-example"); + env_logger::Builder::from_env(Env::default().default_filter_or("info")).init(); + + let args: Vec = env::args().collect(); + let cmd = args.get(1); + + let db_path = "bdk_wallet_esplora_blocking_example.dat"; let db = Store::::open_or_create_new(DB_MAGIC.as_bytes(), db_path)?; let external_descriptor = "wpkh(tprv8ZgxMBicQKsPdy6LMhUtFHAgpocR8GC6QmwMSFpZs7h6Eziw3SpThFfczTDh5rW2krkqffa11UpX3XkeTTB2FvzZKWXqPY54Y6Rq4AQ5R8L/84'/1'/0'/0/*)"; let internal_descriptor = "wpkh(tprv8ZgxMBicQKsPdy6LMhUtFHAgpocR8GC6QmwMSFpZs7h6Eziw3SpThFfczTDh5rW2krkqffa11UpX3XkeTTB2FvzZKWXqPY54Y6Rq4AQ5R8L/84'/1'/0'/1/*)"; @@ -23,55 +31,77 @@ fn main() -> Result<(), anyhow::Error> { external_descriptor, Some(internal_descriptor), db, - Network::Testnet, + Network::Signet, )?; let address = wallet.try_get_address(AddressIndex::New)?; - println!("Generated Address: {}", address); + info!("Generated Address: {}", address); let balance = wallet.get_balance(); - println!("Wallet balance before syncing: {} sats", balance.total()); + info!("Wallet balance: {} sats", balance.total()); - print!("Syncing..."); let client = - esplora_client::Builder::new("https://blockstream.info/testnet/api").build_blocking()?; - - let prev_tip = wallet.latest_checkpoint(); - let keychain_spks = wallet - .all_unbounded_spk_iters() - .into_iter() - .map(|(k, k_spks)| { - let mut once = Some(()); - let mut stdout = std::io::stdout(); - let k_spks = k_spks - .inspect(move |(spk_i, _)| match once.take() { - Some(_) => print!("\nScanning keychain [{:?}]", k), - None => print!(" {:<3}", spk_i), - }) - .inspect(move |_| stdout.flush().expect("must flush")); - (k, k_spks) - }) - .collect(); - - let (update_graph, last_active_indices) = - client.full_scan(keychain_spks, STOP_GAP, PARALLEL_REQUESTS)?; - let missing_heights = update_graph.missing_heights(wallet.local_chain()); - let chain_update = client.update_local_chain(prev_tip, missing_heights)?; - let update = Update { - last_active_indices, - graph: update_graph, - chain: Some(chain_update), - }; + esplora_client::Builder::new("http://signet.bitcoindevkit.net").build_blocking()?; + + let (update, cmd) = match cmd.map(|c| c.as_str()) { + Some(cmd) if cmd == "fullscan" => { + info!("Start full scan..."); + // 1. get data required to do a wallet full_scan + let request = wallet.full_scan_request(); + // 2. full scan to discover wallet transactions + let (graph_update, last_active_indices) = + client.full_scan(request.spks_by_keychain, STOP_GAP, PARALLEL_REQUESTS)?; + // 3. from wallet transactions determine missing blockchain heights + let missing_heights = graph_update.missing_heights(wallet.local_chain()); + // 4. get blockchain update from original request checkpoint and missing heights + let chain_update = client.update_local_chain(request.checkpoint, missing_heights)?; + // 5. create wallet update + Ok(( + Update { + last_active_indices, + graph: graph_update, + chain: Some(chain_update), + }, + cmd, + )) + } + Some(cmd) if cmd == "sync" => { + info!("Start sync..."); + // 1. get data required to do a wallet sync, if also syncing previously used addresses set unused_spks_only = false + let request = wallet.sync_request(true); + // 2. sync unused wallet spks (addresses), unconfirmed tx, and utxos + let graph_update = client.sync( + request.spks, + request.txids, + request.outpoints, + PARALLEL_REQUESTS, + )?; + // 3. from wallet transactions determine missing blockchain heights + let missing_heights = graph_update.missing_heights(wallet.local_chain()); + // 4. get blockchain update from original request checkpoint and missing heights + let chain_update = client.update_local_chain(request.checkpoint, missing_heights)?; + // 5. create wallet update + Ok(( + Update { + graph: graph_update, + chain: Some(chain_update), + ..Update::default() + }, + cmd, + )) + } + _ => Err(()), + } + .expect("Specify if you want to do a wallet 'fullscan' or a 'sync'."); wallet.apply_update(update)?; wallet.commit()?; - println!(); let balance = wallet.get_balance(); - println!("Wallet balance after syncing: {} sats", balance.total()); + info!("Wallet balance after {}: {} sats", cmd, balance.total()); if balance.total() < SEND_AMOUNT { - println!( + info!( "Please send at least {} sats to the receiving address", SEND_AMOUNT ); @@ -79,7 +109,7 @@ fn main() -> Result<(), anyhow::Error> { } let faucet_address = Address::from_str("mkHS9ne12qx9pS9VojpwU5xtRd4T7X7ZUt")? - .require_network(Network::Testnet)?; + .require_network(Network::Signet)?; let mut tx_builder = wallet.build_tx(); tx_builder @@ -92,7 +122,7 @@ fn main() -> Result<(), anyhow::Error> { let tx = psbt.extract_tx(); client.broadcast(&tx)?; - println!("Tx broadcasted! Txid: {}", tx.txid()); + info!("Tx broadcasted! Txid: {}", tx.txid()); Ok(()) }