Recover cashu wallets on init

Author: benthecarman

orange-sdk/src/trusted_wallet/cashu/cashu_store.rs

@@ -21,6 +21,8 @@ use cdk::wallet::{
 use std::collections::hash_map::DefaultHasher;
 use std::hash::{Hash, Hasher};
 
+use crate::trusted_wallet::TrustedError;
+
 // Constants for organizing data in the KV store
 const CASHU_PRIMARY_KEY: &str = "cashu_wallet";
 
@@ -35,6 +37,7 @@ const KEYSET_COUNTERS_KEY: &str = "keyset_counters";
 const TRANSACTIONS_KEY: &str = "transactions";
 const KEYSETS_TABLE_KEY: &str = "keysets_table";
 const KEYSET_U32_MAPPING_KEY: &str = "keyset_u32_mapping";
+const HAS_RECOVERED_KEY: &str = "has_recovered";
 
 /// Error type for database operations
 #[derive(Debug)]
@@ -866,3 +869,26 @@ impl WalletDatabase for CashuKvDatabase {
 		Ok(())
 	}
 }
+
+pub(super) fn read_has_recovered(
+	store: &Arc<dyn KVStore + Send + Sync>,
+) -> Result<bool, TrustedError> {
+	match store.read(CASHU_PRIMARY_KEY, "", HAS_RECOVERED_KEY) {
+		Ok(data) => {
+			if data.is_empty() {
+				return Ok(false);
+			}
+			Ok(data[0] != 0)
+		},
+		Err(e) if e.kind() == io::ErrorKind::NotFound => Ok(false),
+		Err(e) => Err(TrustedError::IOError(e)),
+	}
+}
+
+pub(super) fn write_has_recovered(
+	store: &Arc<dyn KVStore + Send + Sync>, has_recovered: bool,
+) -> Result<(), TrustedError> {
+	let data = vec![if has_recovered { 1 } else { 0 }];
+
+	store.write(CASHU_PRIMARY_KEY, "", HAS_RECOVERED_KEY, &data).map_err(TrustedError::IOError)
+}

orange-sdk/src/trusted_wallet/cashu/mod.rs

@@ -38,7 +38,7 @@ use tokio::runtime::Runtime;
 /// Cashu KV store implementation
 pub mod cashu_store;
 
-use cashu_store::CashuKvDatabase;
+use cashu_store::{CashuKvDatabase, read_has_recovered, write_has_recovered};
 
 /// Configuration for the Cashu wallet
 #[derive(Debug, Clone)]
@@ -448,7 +448,7 @@ impl Cashu {
 			},
 		};
 
-		let db = Arc::new(CashuKvDatabase::new(store).map_err(|e| {
+		let db = Arc::new(CashuKvDatabase::new(Arc::clone(&store)).map_err(|e| {
 			InitFailure::TrustedFailure(TrustedError::Other(format!(
 				"Failed to create Cashu database: {e}"
 			)))
@@ -520,6 +520,26 @@ impl Cashu {
 			}
 		}
 
+		// spawn background task to recover funds if first time initializing
+		let has_recovered = read_has_recovered(&store)?;
+		if !has_recovered {
+			let w = Arc::clone(&cashu_wallet);
+			let l = Arc::clone(&logger);
+			runtime.spawn(async move {
+				match w.restore().await {
+					Err(e) => log_error!(l, "Failed to restore cashu mint: {e}"),
+					Ok(amt) => {
+						if amt > cdk::Amount::ZERO {
+							log_info!(l, "Restored cashu mint: {}, amt: {amt}", w.mint_url);
+						}
+						if let Err(e) = write_has_recovered(&store, true) {
+							log_error!(l, "Failed to write has_recovered flag: {e:?}");
+						}
+					},
+				}
+			});
+		}
+
 		Ok(Cashu {
 			cashu_wallet,
 			unit: cashu_config.unit,