refactor: extract swap/src/monero/wallet.rs to new monero-wallet crate

Closes #798

Author: nabijaczleweli

Cargo.lock

@@ -9,6 +9,7 @@ members = [
   "monero-rpc-pool",
   "monero-sys",
   "monero-tests",
+  "monero-wallet",
   "src-tauri",
   "swap",
   "swap-asb",

Cargo.toml

@@ -0,0 +1,28 @@
+[package]
+name = "monero-wallet"
+version = "0.1.0"
+edition = "2021"
+
+[dependencies]
+# Error handling
+anyhow = { workspace = true }
+
+# Other stuff
+monero = { workspace = true }
+monero-sys = { path = "../monero-sys" }
+swap-core = { path = "../swap-core" }
+throttle = { path = "../throttle" }
+time = "0.3"
+uuid = { workspace = true }
+
+# Tokio
+tokio = { workspace = true, features = ["process", "fs", "net", "parking_lot", "rt"] }
+
+# Tracing
+tracing = { workspace = true }
+tracing-appender = "0.2"
+tracing-subscriber = { workspace = true }
+
+# monero-oxide
+monero-oxide-rpc = { git = "https://github.com/monero-oxide/monero-oxide.git", package = "monero-rpc" }
+monero-simple-request-rpc = { git = "https://github.com/monero-oxide/monero-oxide.git" }

monero-wallet/Cargo.toml

@@ -16,12 +16,31 @@ use throttle::{throttle, Throttle};
 use tokio::sync::RwLock;
 use uuid::Uuid;
 
-use crate::cli::api::{
-    request::{GetMoneroBalanceResponse, GetMoneroHistoryResponse, GetMoneroSyncProgressResponse},
-    tauri_bindings::{MoneroWalletUpdate, TauriEmitter, TauriEvent, TauriHandle},
-};
-
-use super::{BlockHeight, TxHash, WatchRequest};
+use swap_core::monero::primitives::{Amount, BlockHeight, PrivateViewKey, TxHash, WatchRequest};
+
+pub type TauriHandle = Arc<Box<dyn MoneroTauriHandle>>;
+pub trait MoneroTauriHandle: Send + Sync {
+    /// tauri_handle.emit_unified_event(TauriEvent::MoneroWalletUpdate(
+    ///     MoneroWalletUpdate::BalanceChange(GetMoneroBalanceResponse {
+    ///         total_balance, unlocked_balance,
+    ///     })
+    /// ));
+    fn balance_change(&self, total_balance: Amount, unlocked_balance: Amount);
+
+    /// tauri_handle.emit_unified_event(TauriEvent::MoneroWalletUpdate(
+    ///     MoneroWalletUpdate::HistoryUpdate(
+    ///         GetMoneroHistoryResponse { transactions }
+    ///     ),
+    /// ));
+    fn history_update(&self, transactions: Vec<monero_sys::TransactionInfo>);
+
+    /// tauri_handle.emit_unified_event(TauriEvent::MoneroWalletUpdate(
+    ///     MoneroWalletUpdate::SyncProgress(GetMoneroSyncProgressResponse {
+    ///         current_block, target_block, progress_percentage,
+    ///     }),
+    /// ));
+    fn sync_progress(&self, current_block: u64, target_block: u64, progress_percentage: f32);
+}
 
 /// Entrance point to the Monero blockchain.
 /// You can use this struct to open specific wallets and monitor the blockchain.
@@ -84,13 +103,7 @@ impl TauriWalletListener {
                         }
                     };
 
-                    let response = GetMoneroBalanceResponse {
-                        total_balance: total_balance.into(),
-                        unlocked_balance: unlocked_balance.into(),
-                    };
-                    tauri.emit_unified_event(TauriEvent::MoneroWalletUpdate(
-                        MoneroWalletUpdate::BalanceChange(response),
-                    ));
+                    tauri.balance_change(total_balance.into(), unlocked_balance.into());
                 });
             }
         };
@@ -111,11 +124,8 @@ impl TauriWalletListener {
                             return;
                         }
                     };
-                    let response = GetMoneroHistoryResponse { transactions };
 
-                    tauri.emit_unified_event(TauriEvent::MoneroWalletUpdate(
-                        MoneroWalletUpdate::HistoryUpdate(response),
-                    ));
+                    tauri.history_update(transactions);
                 });
             }
         };
@@ -138,16 +148,11 @@ impl TauriWalletListener {
                     };
 
                     let progress_percentage = sync_progress.percentage();
-
-                    let response = GetMoneroSyncProgressResponse {
-                        current_block: sync_progress.current_block,
-                        target_block: sync_progress.target_block,
-                        progress_percentage: progress_percentage,
-                    };
-
-                    tauri.emit_unified_event(TauriEvent::MoneroWalletUpdate(
-                        MoneroWalletUpdate::SyncProgress(response),
-                    ));
+                    tauri.sync_progress(
+                        sync_progress.current_block,
+                        sync_progress.target_block,
+                        progress_percentage,
+                    );
                 });
             }
         };
@@ -223,15 +228,17 @@ impl Wallets {
     ///
     /// The main wallet will be kept alive and synced, other wallets are
     /// opened and closed on demand.
-    pub async fn new(
+    pub async fn new<T: Into<TauriHandle>>(
         wallet_dir: PathBuf,
         main_wallet_name: String,
         daemon: Daemon,
         network: Network,
         regtest: bool,
-        tauri_handle: Option<TauriHandle>,
+        tauri_handle: Option<T>,
         wallet_database: Option<Arc<monero_sys::Database>>,
     ) -> Result<Self> {
+        let tauri_handle = tauri_handle.map(|th| th.into());
+
         let main_wallet = Wallet::open_or_create(
             wallet_dir.join(&main_wallet_name).display().to_string(),
             daemon.clone(),
@@ -290,15 +297,16 @@ impl Wallets {
 
     /// Create a new `Wallets` instance with an existing wallet as the main wallet.
     /// This is used when we want to use a user-selected wallet instead of creating a new one.
-    pub async fn new_with_existing_wallet(
+    pub async fn new_with_existing_wallet<T: Into<TauriHandle>>(
         wallet_dir: PathBuf,
         daemon: Daemon,
         network: Network,
         regtest: bool,
-        tauri_handle: Option<TauriHandle>,
+        tauri_handle: Option<T>,
         existing_wallet: Wallet,
         wallet_database: Option<Arc<monero_sys::Database>>,
     ) -> Result<Self> {
+        let tauri_handle = tauri_handle.map(|th| th.into());
         // TODO: This code is duplicated in [`Wallets::new`]. Unify it.
 
         if regtest {
@@ -367,7 +375,7 @@ impl Wallets {
         &self,
         swap_id: Uuid,
         spend_key: monero::PrivateKey,
-        view_key: super::PrivateViewKey,
+        view_key: PrivateViewKey,
         tx_lock_id: TxHash,
     ) -> Result<Arc<Wallet>> {
         // Derive wallet address from the keys

swap/src/monero/wallet.rs monero-wallet/src/lib.rsswap/src/monero/wallet.rs renamed to monero-wallet/src/lib.rs

@@ -17,8 +17,9 @@ bdk = { workspace = true }
 bdk_wallet = { workspace = true }
 bitcoin = { workspace = true }
 
-# Bitcoin Wallet
+# Wallets
 bitcoin-wallet = { path = "../bitcoin-wallet" }
+monero-wallet = { path = "../monero-wallet" }
 
 # Tor
 arti-client = { workspace = true, features = ["static-sqlite", "tokio", "rustls", "onion-service-service"] }
@@ -71,7 +72,6 @@ swap-p2p = { path = "../swap-p2p" }
 swap-proptest = { path = "../swap-proptest" }
 swap-serde = { path = "../swap-serde" }
 tauri = { version = "2.0", features = ["config-json5"], optional = true, default-features = false }
-throttle = { path = "../throttle" }
 time = "0.3"
 url = { workspace = true }
 uuid = { workspace = true, features = ["serde"] }
@@ -97,10 +97,6 @@ serde_with = { version = "1", features = ["macros"] }
 strum = { workspace = true, features = ["derive"] }
 typeshare = { workspace = true }
 
-# monero-oxide
-monero-oxide-rpc = { git = "https://github.com/monero-oxide/monero-oxide.git", package = "monero-rpc" }
-monero-simple-request-rpc = { git = "https://github.com/monero-oxide/monero-oxide.git" }
-
 # Database
 sqlx = { version = "0.8", features = ["sqlite", "runtime-tokio-rustls"] }
 

swap/Cargo.toml

@@ -505,6 +505,41 @@ impl bitcoin_wallet::BitcoinTauriBackgroundTask
     }
 }
 
+struct MoneroTauriHandle(TauriHandle);
+
+impl Into<monero_wallet::TauriHandle> for TauriHandle {
+    fn into(self) -> monero_wallet::TauriHandle {
+        Arc::new(Box::new(MoneroTauriHandle(self)))
+    }
+}
+
+impl monero_wallet::MoneroTauriHandle for MoneroTauriHandle {
+    fn balance_change(&self, total_balance: monero::Amount, unlocked_balance: monero::Amount) {
+        self.0.emit_unified_event(TauriEvent::MoneroWalletUpdate(
+            MoneroWalletUpdate::BalanceChange(GetMoneroBalanceResponse {
+                total_balance,
+                unlocked_balance,
+            }),
+        ))
+    }
+
+    fn history_update(&self, transactions: Vec<monero_sys::TransactionInfo>) {
+        self.0.emit_unified_event(TauriEvent::MoneroWalletUpdate(
+            MoneroWalletUpdate::HistoryUpdate(GetMoneroHistoryResponse { transactions }),
+        ))
+    }
+
+    fn sync_progress(&self, current_block: u64, target_block: u64, progress_percentage: f32) {
+        self.0.emit_unified_event(TauriEvent::MoneroWalletUpdate(
+            MoneroWalletUpdate::SyncProgress(GetMoneroSyncProgressResponse {
+                current_block,
+                target_block,
+                progress_percentage,
+            }),
+        ))
+    }
+}
+
 impl Display for ApprovalRequest {
     fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
         match self.request {

swap/src/cli/api/tauri_bindings.rs

@@ -223,6 +223,7 @@ mod crates {
     pub const OUR_CRATES: &[&str] = &[
         // Library crates
         "bitcoin_wallet",
+        "monero_wallet",
         "swap_p2p",
         "swap_env",
         "swap_core",

swap/src/common/tracing_util.rs

@@ -1,4 +1,4 @@
-pub mod wallet;
+pub use monero_wallet as wallet;
 pub mod wallet_rpc;
 
 pub use ::monero::network::Network;