From 23dc8bf7821af48662abee9629413e00b42ec6d9 Mon Sep 17 00:00:00 2001
From: f321x <f321x@tutamail.com>
Date: Tue, 25 Jun 2024 13:54:40 +0000
Subject: [PATCH] begin coordinator wallet implementation

---
 taptrade-cli-demo/coordinator/.env            |  1 +
 .../src/{sqlite_db => database}/mod.rs        |  5 +--
 taptrade-cli-demo/coordinator/src/main.rs     |  8 ++---
 .../coordinator/src/wallet/mod.rs             | 31 ++++++++++++++++++-
 4 files changed, 38 insertions(+), 7 deletions(-)
 rename taptrade-cli-demo/coordinator/src/{sqlite_db => database}/mod.rs (94%)

diff --git a/taptrade-cli-demo/coordinator/.env b/taptrade-cli-demo/coordinator/.env
index 44a0324..5ae95b0 100644
--- a/taptrade-cli-demo/coordinator/.env
+++ b/taptrade-cli-demo/coordinator/.env
@@ -1,2 +1,3 @@
 ELECTRUM_BACKEND="127.0.0.1:50001"
 DATABASE_PATH="./db/trades.db"
+WALLET_XPRV="tprv8ZgxMBicQKsPdHuCSjhQuSZP1h6ZTeiRqREYS5guGPdtL7D1uNLpnJmb2oJep99Esq1NbNZKVJBNnD2ZhuXSK7G5eFmmcx73gsoa65e2U32"
diff --git a/taptrade-cli-demo/coordinator/src/sqlite_db/mod.rs b/taptrade-cli-demo/coordinator/src/database/mod.rs
similarity index 94%
rename from taptrade-cli-demo/coordinator/src/sqlite_db/mod.rs
rename to taptrade-cli-demo/coordinator/src/database/mod.rs
index 84e96b3..d5ee904 100644
--- a/taptrade-cli-demo/coordinator/src/sqlite_db/mod.rs
+++ b/taptrade-cli-demo/coordinator/src/database/mod.rs
@@ -4,11 +4,12 @@ use super::*;
 use sqlx::{sqlite::SqlitePoolOptions, Pool, Sqlite};
 
 #[derive(Clone, Debug)]
-pub struct SqliteDB {
+pub struct CoordinatorDB {
 	pub db_pool: Arc<Pool<Sqlite>>,
 }
 
-impl SqliteDB {
+// is our implementation secure against sql injections?
+impl CoordinatorDB {
 	// will either create a new db or load existing one. Will create according tables in new db
 	pub async fn init() -> Result<Self> {
 		let db_pool = SqlitePoolOptions::new()
diff --git a/taptrade-cli-demo/coordinator/src/main.rs b/taptrade-cli-demo/coordinator/src/main.rs
index 8dc783a..b0bf2bf 100755
--- a/taptrade-cli-demo/coordinator/src/main.rs
+++ b/taptrade-cli-demo/coordinator/src/main.rs
@@ -1,13 +1,13 @@
 mod communication;
 mod coordinator;
-mod sqlite_db;
+mod database;
 mod wallet;
 
 use anyhow::{anyhow, Result};
 use bdk::{database::MemoryDatabase, Wallet};
 use communication::{api::*, api_server};
+use database::CoordinatorDB;
 use dotenv::dotenv;
-use sqlite_db::SqliteDB;
 use std::{env, sync::Arc};
 use wallet::CoordinatorWallet;
 
@@ -16,8 +16,8 @@ use wallet::CoordinatorWallet;
 async fn main() -> Result<()> {
 	dotenv().ok();
 	// Initialize the database pool
-	let coordinator_db = SqliteDB::init().await?;
-	let wallet = CoordinatorWallet::init().await?;
+	let coordinator_db = CoordinatorDB::init().await?;
+	// let wallet = CoordinatorWallet::init().await?;
 
 	api_server(coordinator_db).await?;
 	Ok(())
diff --git a/taptrade-cli-demo/coordinator/src/wallet/mod.rs b/taptrade-cli-demo/coordinator/src/wallet/mod.rs
index 94d9ae7..749f371 100644
--- a/taptrade-cli-demo/coordinator/src/wallet/mod.rs
+++ b/taptrade-cli-demo/coordinator/src/wallet/mod.rs
@@ -1,7 +1,36 @@
 use super::*;
+use anyhow::Context;
+use bdk::{
+	bitcoin::{self, bip32::ExtendedPrivKey},
+	blockchain::ElectrumBlockchain,
+	electrum_client::Client,
+	template::Bip86,
+	KeychainKind, SyncOptions, Wallet,
+};
+use std::str::FromStr;
 
 pub struct CoordinatorWallet {
 	pub wallet: Wallet<MemoryDatabase>,
 }
 
-impl CoordinatorWallet {}
+impl CoordinatorWallet {
+	pub async fn init() -> Result<Self> {
+		let wallet_xprv = ExtendedPrivKey::from_str(
+			&env::var("WALLET_XPRV").context("loading WALLET_XPRV from .env failed")?,
+		)?;
+		let backend = ElectrumBlockchain::from(Client::new(
+			&env::var("ELECTRUM_BACKEND")
+				.context("Parsing ELECTRUM_BACKEND from .env failed, is it set?")?,
+		)?);
+		let wallet = Wallet::new(
+			Bip86(wallet_xprv, KeychainKind::External),
+			Some(Bip86(wallet_xprv, KeychainKind::Internal)),
+			bitcoin::Network::Testnet,
+			MemoryDatabase::default(), // non-permanent storage
+		)?;
+
+		wallet.sync(&backend, SyncOptions::default())?;
+		dbg!("Balance: {} SAT", wallet.get_balance()?);
+		Ok(TradingWallet { wallet, backend })
+	}
+}