Merge pull request #86 from comit-network/wallet-struct

Wrap bdk::wallet inside a struct

daemon/src/maker.rs

@@ -1,9 +1,8 @@
 use crate::seed::Seed;
+use crate::wallet::Wallet;
 use anyhow::Result;
 use bdk::bitcoin::secp256k1::{schnorrsig, SECP256K1};
 use bdk::bitcoin::{Amount, Network};
-use bdk::blockchain::{ElectrumBlockchain, NoopProgress};
-use bdk::KeychainKind;
 use clap::Clap;
 use model::cfd::{Cfd, Order};
 use rocket::fairing::AdHoc;
@@ -21,6 +20,7 @@ mod seed;
 mod send_wire_message_actor;
 mod setup_contract_actor;
 mod to_sse_event;
+mod wallet;
 mod wire;
 
 #[derive(Database)]
@@ -64,23 +64,14 @@ async fn main() -> Result<()> {
 
     let seed = Seed::initialize(&data_dir.join("maker_seed"), opts.generate_seed).await?;
 
-    let client = bdk::electrum_client::Client::new(&opts.electrum).unwrap();
-
-    // TODO: Replace with sqlite once https://github.com/bitcoindevkit/bdk/pull/376 is merged.
-    let db = bdk::sled::open(data_dir.join("maker_wallet_db"))?;
-    let wallet_db = db.open_tree("wallet")?;
-
     let ext_priv_key = seed.derive_extended_priv_key(Network::Testnet)?;
 
-    let wallet = bdk::Wallet::new(
+    let wallet = Wallet::new(
-        bdk::template::Bip84(ext_priv_key, KeychainKind::External),
+        &opts.electrum,
-        Some(bdk::template::Bip84(ext_priv_key, KeychainKind::Internal)),
+        &data_dir.join("maker_wallet_db"),
-        ext_priv_key.network,
+        ext_priv_key,
-        wallet_db,
-        ElectrumBlockchain::from(client),
     )
-    .unwrap();
+    .await?;
-    wallet.sync(NoopProgress, None).unwrap(); // TODO: Use LogProgress once we have logging.
 
     let oracle = schnorrsig::KeyPair::new(SECP256K1, &mut rand::thread_rng()); // TODO: Fetch oracle public key from oracle.
 

daemon/src/maker_cfd_actor.rs

@@ -3,12 +3,11 @@ use std::time::SystemTime;
 use crate::db::{insert_cfd, insert_order, load_all_cfds, load_order_by_id};
 use crate::model::cfd::{Cfd, CfdState, CfdStateCommon, FinalizedCfd, Order, OrderId};
 use crate::model::{TakerId, Usd};
+use crate::wallet::Wallet;
 use crate::wire::SetupMsg;
 use crate::{maker_cfd_actor, maker_inc_connections_actor, setup_contract_actor};
 use bdk::bitcoin::secp256k1::schnorrsig;
 use bdk::bitcoin::{self};
-use bdk::database::BatchDatabase;
-use cfd_protocol::WalletExt;
 use futures::Future;
 use tokio::sync::{mpsc, watch};
 
@@ -32,9 +31,9 @@ pub enum Command {
     CfdSetupCompleted(FinalizedCfd),
 }
 
-pub fn new<B, D>(
+pub fn new(
     db: sqlx::SqlitePool,
-    wallet: bdk::Wallet<B, D>,
+    wallet: Wallet,
     oracle_pk: schnorrsig::PublicKey,
     takers: mpsc::UnboundedSender<maker_inc_connections_actor::Command>,
     cfd_feed_actor_inbox: watch::Sender<Vec<Cfd>>,
@@ -42,10 +41,7 @@ pub fn new<B, D>(
 ) -> (
     impl Future<Output = ()>,
     mpsc::UnboundedSender<maker_cfd_actor::Command>,
-)
+) {
-where
-    D: BatchDatabase,
-{
     let (sender, mut receiver) = mpsc::unbounded_channel();
     let mut current_contract_setup = None;
 

daemon/src/taker.rs

@@ -1,8 +1,7 @@
+use crate::wallet::Wallet;
 use anyhow::Result;
 use bdk::bitcoin::secp256k1::{schnorrsig, SECP256K1};
 use bdk::bitcoin::{Amount, Network};
-use bdk::blockchain::{ElectrumBlockchain, NoopProgress};
-use bdk::KeychainKind;
 use clap::Clap;
 use model::cfd::{Cfd, Order};
 use rocket::fairing::AdHoc;
@@ -24,6 +23,7 @@ mod setup_contract_actor;
 mod taker_cfd_actor;
 mod taker_inc_message_actor;
 mod to_sse_event;
+mod wallet;
 mod wire;
 
 const CONNECTION_RETRY_INTERVAL: Duration = Duration::from_secs(5);
@@ -69,23 +69,14 @@ async fn main() -> Result<()> {
 
     let seed = Seed::initialize(&data_dir.join("taker_seed"), opts.generate_seed).await?;
 
-    let client = bdk::electrum_client::Client::new(&opts.electrum).unwrap();
-
-    // TODO: Replace with sqlite once https://github.com/bitcoindevkit/bdk/pull/376 is merged.
-    let db = bdk::sled::open(data_dir.join("taker_wallet_db"))?;
-    let wallet_db = db.open_tree("wallet")?;
-
     let ext_priv_key = seed.derive_extended_priv_key(Network::Testnet)?;
 
-    let wallet = bdk::Wallet::new(
+    let wallet = Wallet::new(
-        bdk::template::Bip84(ext_priv_key, KeychainKind::External),
+        &opts.electrum,
-        Some(bdk::template::Bip84(ext_priv_key, KeychainKind::Internal)),
+        &data_dir.join("taker_wallet_db"),
-        ext_priv_key.network,
+        ext_priv_key,
-        wallet_db,
-        ElectrumBlockchain::from(client),
     )
-    .unwrap();
+    .await?;
-    wallet.sync(NoopProgress, None).unwrap(); // TODO: Use LogProgress once we have logging.
 
     let oracle = schnorrsig::KeyPair::new(SECP256K1, &mut rand::thread_rng()); // TODO: Fetch oracle public key from oracle.
 

daemon/src/taker_cfd_actor.rs

@@ -3,12 +3,11 @@ use crate::db::{
 };
 use crate::model::cfd::{Cfd, CfdState, CfdStateCommon, FinalizedCfd, Order, OrderId};
 use crate::model::Usd;
+use crate::wallet::Wallet;
 use crate::wire::SetupMsg;
 use crate::{setup_contract_actor, wire};
 use bdk::bitcoin::secp256k1::schnorrsig;
 use bdk::bitcoin::{self};
-use bdk::database::BatchDatabase;
-use cfd_protocol::WalletExt;
 use core::panic;
 use futures::Future;
 use std::time::SystemTime;
@@ -24,17 +23,14 @@ pub enum Command {
     CfdSetupCompleted(FinalizedCfd),
 }
 
-pub fn new<B, D>(
+pub fn new(
     db: sqlx::SqlitePool,
-    wallet: bdk::Wallet<B, D>,
+    wallet: Wallet,
     oracle_pk: schnorrsig::PublicKey,
     cfd_feed_actor_inbox: watch::Sender<Vec<Cfd>>,
     order_feed_actor_inbox: watch::Sender<Option<Order>>,
     out_msg_maker_inbox: mpsc::UnboundedSender<wire::TakerToMaker>,
-) -> (impl Future<Output = ()>, mpsc::UnboundedSender<Command>)
+) -> (impl Future<Output = ()>, mpsc::UnboundedSender<Command>) {
-where
-    D: BatchDatabase,
-{
     let (sender, mut receiver) = mpsc::unbounded_channel();
     let mut current_contract_setup = None;
 

daemon/src/wallet.rs

@@ -0,0 +1,49 @@
+use anyhow::{Context, Result};
+use bdk::bitcoin::util::bip32::ExtendedPrivKey;
+use bdk::bitcoin::{Amount, PublicKey};
+use bdk::blockchain::{ElectrumBlockchain, NoopProgress};
+use bdk::KeychainKind;
+use cfd_protocol::{PartyParams, WalletExt};
+use std::path::Path;
+
+const SLED_TREE_NAME: &str = "wallet";
+
+pub struct Wallet<B = ElectrumBlockchain, D = bdk::sled::Tree> {
+    wallet: bdk::Wallet<B, D>,
+}
+
+impl Wallet {
+    pub async fn new(
+        electrum_rpc_url: &str,
+        wallet_dir: &Path,
+        ext_priv_key: ExtendedPrivKey,
+    ) -> Result<Self> {
+        let client = bdk::electrum_client::Client::new(electrum_rpc_url)
+            .context("Failed to initialize Electrum RPC client")?;
+
+        // TODO: Replace with sqlite once https://github.com/bitcoindevkit/bdk/pull/376 is merged.
+        let db = bdk::sled::open(wallet_dir)?.open_tree(SLED_TREE_NAME)?;
+
+        let wallet = bdk::Wallet::new(
+            bdk::template::Bip84(ext_priv_key, KeychainKind::External),
+            Some(bdk::template::Bip84(ext_priv_key, KeychainKind::Internal)),
+            ext_priv_key.network,
+            db,
+            ElectrumBlockchain::from(client),
+        )?;
+
+        wallet
+            .sync(NoopProgress, None)
+            .context("Failed to sync the wallet")?; // TODO: Use LogProgress once we have logging.
+
+        Ok(Self { wallet })
+    }
+
+    pub fn build_party_params(
+        &self,
+        amount: Amount,
+        identity_pk: PublicKey,
+    ) -> Result<PartyParams> {
+        self.wallet.build_party_params(amount, identity_pk)
+    }
+}