1#[cfg(all(any(target_os = "android", target_os = "ios"), feature = "tls-native-roots"))]
296compile_error!("feature `tls-native-roots` can't be used on Android or iOS, use `tls-webpki-roots` instead");
297
298pub extern crate ark;
299
300pub extern crate bip39;
301pub extern crate lightning_invoice;
302pub extern crate lnurl as lnurllib;
303
304#[macro_use] extern crate anyhow;
305#[macro_use] extern crate async_trait;
306#[macro_use] extern crate serde;
307
308pub mod actions;
309pub mod chain;
310pub mod exit;
311pub mod movement;
312pub mod onchain;
313pub mod payment_request;
314pub mod persist;
315pub mod round;
316pub mod subsystem;
317pub mod vtxo;
318
319pub mod lock_manager;
320
321mod arkoor;
322mod board;
323mod config;
324mod daemon;
325mod fees;
326mod lightning;
327mod mailbox;
328mod notification;
329mod offboard;
330#[cfg(feature = "socks5-proxy")]
331mod proxy;
332mod psbtext;
333mod utils;
334
335pub use self::arkoor::{ArkoorCreateResult, ArkoorAddressError};
336pub use self::config::{BarkNetwork, Config};
337pub use self::daemon::DaemonHandle;
338pub use self::fees::FeeEstimate;
339pub use self::notification::{WalletNotification, NotificationStream};
340pub use self::vtxo::WalletVtxo;
341
342use std::borrow::Cow;
343use std::collections::HashSet;
344use std::sync::Arc;
345use std::time::Duration;
346
347use anyhow::{bail, Context};
348use bip39::Mnemonic;
349use bitcoin::{Amount, Network, OutPoint};
350use bitcoin::bip32::{self, ChildNumber, Fingerprint};
351use bitcoin::secp256k1::{self, Keypair, PublicKey};
352use log::{trace, info, warn, error};
353
354use ark::{ArkInfo, ProtocolEncoding, Vtxo, VtxoId, VtxoPolicy, VtxoRequest};
355use ark::address::VtxoDelivery;
356use ark::fees::{validate_and_subtract_fee_min_dust, VtxoFeeInfo};
357use ark::vtxo::{Full, PubkeyVtxoPolicy, VtxoRef};
358use ark::vtxo::policy::signing::VtxoSigner;
359use bitcoin_ext::{BlockHeight, P2TR_DUST};
360use server_rpc::{protos, ServerConnection};
361use server_rpc::client::{ConnectError, CreateEndpointError};
362use crate::chain::{ChainSource, ChainSourceSpec};
363use crate::exit::Exit;
364use crate::lock_manager::LockManager;
365use crate::movement::{Movement, MovementId, PaymentMethod};
366use crate::movement::manager::MovementManager;
367use crate::movement::update::MovementUpdate;
368use crate::notification::NotificationDispatch;
369use crate::onchain::{ExitUnilaterally, PreparePsbt, SignPsbt, Utxo};
370use crate::onchain::DaemonizableOnchainWallet;
371use crate::persist::BarkPersister;
372use crate::persist::models::{RoundStateId, StoredRoundState, Unlocked};
373#[cfg(feature = "socks5-proxy")]
374use crate::proxy::proxy_for_url;
375use crate::round::{RoundParticipation, RoundStatus};
376use crate::subsystem::{ArkoorMovement, RoundMovement};
377use crate::vtxo::{FilterVtxos, RefreshStrategy, VtxoFilter, VtxoStateKind};
378
379#[cfg(all(feature = "wasm-web", feature = "socks5-proxy"))]
380compile_error!("features `wasm-web` does not support feature `socks5-proxy");
381
382#[cfg(all(feature = "wasm-web", feature = "bitcoind-rpc"))]
383compile_error!("`wasm-web` does not support the `bitcoind-rpc` feature");
384
385const HEALTHY_STREAM_DURATION: Duration = Duration::from_secs(59);
389
390const BARK_PURPOSE_INDEX: u32 = 350;
392const VTXO_KEYS_INDEX: u32 = 0;
394const MAILBOX_KEY_INDEX: u32 = 1;
396const RECOVERY_MAILBOX_KEY_INDEX: u32 = 2;
398const MISSING_SERVER_TRANSPORT_HELP: &str =
399 "This build of bark-wallet does not include an Ark server transport backend. Enable feature `bark-wallet/native` or `bark-wallet/wasm-web` to use server-backed wallet functionality.";
400
401lazy_static::lazy_static! {
402 static ref SECP: secp256k1::Secp256k1<secp256k1::All> = secp256k1::Secp256k1::new();
404}
405
406fn log_server_pubkey_changed_error(expected: PublicKey, got: PublicKey) {
412 error!(
413 "
414Server public key has changed!
415
416The Ark server's public key is different from the one stored when this
417wallet was created. This typically happens when:
418
419 - The server operator has rotated their keys
420 - You are connecting to a different server
421 - The server has been replaced
422
423For safety, this wallet will not connect to the server until you
424resolve this. You can recover your funds on-chain by doing an emergency exit.
425
426This will exit your VTXOs to on-chain Bitcoin without needing the server's cooperation.
427
428Expected: {expected}
429Got: {got}")
430}
431
432fn log_server_mailbox_pubkey_changed_error(expected: PublicKey, got: PublicKey) {
434 error!(
435 "
436Server mailbox public key has changed!
437
438The Ark server's mailbox public key is different from the one stored when this
439wallet was created. This typically happens when:
440
441 - The server operator has rotated their keys
442 - You are connecting to a different server
443 - The server has been replaced
444
445For safety, this wallet will not connect to the server until you resolve this.
446
447Unlike a server pubkey change, your VTXOs are not at risk - the mailbox pubkey
448only affects address receive semantics. Any Ark addresses you previously
449shared will stop receiving new payments; you will need to share new addresses
450after reconnecting.
451
452Expected: {expected}
453Got: {got}")
454}
455
456#[derive(Debug, Clone)]
458pub struct LightningReceiveBalance {
459 pub total: Amount,
461 pub claimable: Amount,
463}
464
465#[derive(Debug, Clone)]
467pub struct Balance {
468 pub spendable: Amount,
470 pub pending_lightning_send: Amount,
472 pub claimable_lightning_receive: Amount,
474 pub pending_in_round: Amount,
476 pub pending_exit: Option<Amount>,
479 pub pending_board: Amount,
481}
482
483pub struct UtxoInfo {
484 pub outpoint: OutPoint,
485 pub amount: Amount,
486 pub confirmation_height: Option<u32>,
487}
488
489impl From<Utxo> for UtxoInfo {
490 fn from(value: Utxo) -> Self {
491 match value {
492 Utxo::Local(o) => UtxoInfo {
493 outpoint: o.outpoint,
494 amount: o.amount,
495 confirmation_height: o.confirmation_height,
496 },
497 Utxo::Exit(e) => UtxoInfo {
498 outpoint: e.vtxo.point(),
499 amount: e.vtxo.amount(),
500 confirmation_height: Some(e.height),
501 },
502 }
503 }
504}
505
506pub struct OffchainBalance {
509 pub available: Amount,
511 pub pending_in_round: Amount,
513 pub pending_exit: Amount,
516}
517
518#[derive(Debug, Clone, PartialEq, Eq, Serialize, Deserialize)]
520pub struct WalletProperties {
521 pub network: Network,
525
526 pub fingerprint: Fingerprint,
530
531 pub server_pubkey: Option<PublicKey>,
538
539 pub server_mailbox_pubkey: Option<PublicKey>,
547}
548
549pub struct WalletSeed {
555 master: bip32::Xpriv,
556 vtxo: bip32::Xpriv,
557}
558
559impl WalletSeed {
560 fn new(network: Network, seed: &[u8; 64]) -> Self {
561 let bark_path = [ChildNumber::from_hardened_idx(BARK_PURPOSE_INDEX).unwrap()];
562 let master = bip32::Xpriv::new_master(network, seed)
563 .expect("invalid seed")
564 .derive_priv(&SECP, &bark_path)
565 .expect("purpose is valid");
566
567 let vtxo_path = [ChildNumber::from_hardened_idx(VTXO_KEYS_INDEX).unwrap()];
568 let vtxo = master.derive_priv(&SECP, &vtxo_path)
569 .expect("vtxo path is valid");
570
571 Self { master, vtxo }
572 }
573
574 fn fingerprint(&self) -> Fingerprint {
575 self.master.fingerprint(&SECP)
576 }
577
578 fn derive_vtxo_keypair(&self, idx: u32) -> Keypair {
579 self.vtxo.derive_priv(&SECP, &[idx.into()]).unwrap().to_keypair(&SECP)
580 }
581
582 fn to_mailbox_keypair(&self) -> Keypair {
583 let mailbox_path = [ChildNumber::from_hardened_idx(MAILBOX_KEY_INDEX).unwrap()];
584 self.master.derive_priv(&SECP, &mailbox_path).unwrap().to_keypair(&SECP)
585 }
586
587 fn to_recovery_mailbox_keypair(&self) -> Keypair {
588 let mailbox_path = [ChildNumber::from_hardened_idx(RECOVERY_MAILBOX_KEY_INDEX).unwrap()];
589 self.master.derive_priv(&SECP, &mailbox_path).unwrap().to_keypair(&SECP)
590 }
591}
592
593struct WalletInner {
594 chain: Arc<ChainSource>,
596
597 exit: Exit,
599
600 movements: Arc<MovementManager>,
602
603 notifications: NotificationDispatch,
605
606 config: Config,
608
609 db: Arc<dyn BarkPersister>,
611
612 lock_manager: Box<dyn LockManager>,
616
617 seed: WalletSeed,
619
620 server: tokio::sync::OnceCell<ServerConnection>,
627
628 daemon: parking_lot::Mutex<Option<DaemonHandle>>,
630}
631
632#[derive(Clone)]
771pub struct Wallet {
772 inner: Arc<WalletInner>,
773}
774
775impl Wallet {
776 pub async fn require_chainsource_version(&self) -> anyhow::Result<()> {
780 self.inner.chain.require_version().await
781 }
782
783 pub async fn network(&self) -> anyhow::Result<Network> {
784 Ok(self.properties().await?.network)
785 }
786
787 pub fn chain(&self) -> &Arc<ChainSource> {
789 &self.inner.chain
790 }
791
792 pub fn exit_mgr(&self) -> &Exit {
794 &self.inner.exit
795 }
796
797 pub fn movements_mgr(&self) -> &MovementManager {
799 &self.inner.movements
800 }
801
802 pub async fn peek_next_keypair(&self) -> anyhow::Result<(Keypair, u32)> {
805 let last_revealed = self.inner.db.get_last_vtxo_key_index().await?;
806
807 let index = last_revealed.map(|i| i + 1).unwrap_or(u32::MIN);
808 let keypair = self.inner.seed.derive_vtxo_keypair(index);
809
810 Ok((keypair, index))
811 }
812
813 pub async fn derive_store_next_keypair(&self) -> anyhow::Result<(Keypair, u32)> {
816 let (keypair, index) = self.peek_next_keypair().await?;
817 self.inner.db.store_vtxo_key(index, keypair.public_key()).await?;
818 Ok((keypair, index))
819 }
820
821 #[deprecated(note = "use peek_keypair instead")]
822 pub async fn peak_keypair(&self, index: u32) -> anyhow::Result<Keypair> {
823 self.peek_keypair(index).await
824 }
825
826 pub async fn peek_keypair(&self, index: u32) -> anyhow::Result<Keypair> {
840 let keypair = self.inner.seed.derive_vtxo_keypair(index);
841 if self.inner.db.get_public_key_idx(&keypair.public_key()).await?.is_some() {
842 Ok(keypair)
843 } else {
844 bail!("VTXO key {} does not exist, please derive it first", index)
845 }
846 }
847
848
849 pub async fn pubkey_keypair(&self, public_key: &PublicKey) -> anyhow::Result<Option<(u32, Keypair)>> {
861 if let Some(index) = self.inner.db.get_public_key_idx(&public_key).await? {
862 Ok(Some((index, self.inner.seed.derive_vtxo_keypair(index))))
863 } else {
864 Ok(None)
865 }
866 }
867
868 pub async fn get_vtxo_key(&self, vtxo: impl VtxoRef) -> anyhow::Result<Keypair> {
879 let bare_vtxo = match vtxo.as_bare_vtxo() {
880 Some(bare) => bare,
881 None => Cow::Owned(self.get_vtxo_by_id(vtxo.vtxo_id()).await?.vtxo),
882 };
883 let pubkey = self.find_signable_clause(&bare_vtxo).await
884 .context("VTXO is not signable by wallet")?
885 .pubkey();
886 let idx = self.inner.db.get_public_key_idx(&pubkey).await?
887 .context("VTXO key not found")?;
888 Ok(self.inner.seed.derive_vtxo_keypair(idx))
889 }
890
891 #[deprecated(note = "use peek_address instead")]
892 pub async fn peak_address(&self, index: u32) -> anyhow::Result<ark::Address> {
893 self.peek_address(index).await
894 }
895
896 pub async fn peek_address(&self, index: u32) -> anyhow::Result<ark::Address> {
900 let properties = self.properties().await?;
901 let network = properties.network;
902 let keypair = self.peek_keypair(index).await?;
903 let mailbox = self.mailbox_identifier();
904
905
906 let (server_pubkey, mailbox_pubkey) =
907 if let (Some(spk), Some(mpk)) = (properties.server_pubkey, properties.server_mailbox_pubkey) {
908 (spk, mpk)
909 } else {
910 let (_, ark_info) = self.require_server().await?;
911 (ark_info.server_pubkey, ark_info.mailbox_pubkey)
912 };
913
914 Ok(ark::Address::builder()
915 .testnet(network != bitcoin::Network::Bitcoin)
916 .server_pubkey(server_pubkey)
917 .pubkey_policy(keypair.public_key())
918 .mailbox(mailbox_pubkey, mailbox, &keypair)
919 .expect("Failed to assign mailbox")
920 .into_address().unwrap())
921 }
922
923 pub async fn new_address_with_index(&self) -> anyhow::Result<(ark::Address, u32)> {
927 let (_, index) = self.derive_store_next_keypair().await?;
928 let addr = self.peek_address(index).await?;
929 Ok((addr, index))
930 }
931
932 pub async fn new_address(&self) -> anyhow::Result<ark::Address> {
934 let (addr, _) = self.new_address_with_index().await?;
935 Ok(addr)
936 }
937
938 pub async fn create(
947 mnemonic: &Mnemonic,
948 network: Network,
949 config: Config,
950 db: Arc<dyn BarkPersister>,
951 lock_manager: Box<dyn LockManager>,
952 force: bool,
953 ) -> anyhow::Result<Wallet> {
954 trace!("Config: {:?}", config);
955
956 let wallet_fingerprint = WalletSeed::new(network, &mnemonic.to_seed("")).fingerprint();
957
958 let create_guard = lock_manager.lock(
963 &format!("{}.create", wallet_fingerprint),
964 Duration::from_secs(5),
965 ).await.context("wallet initialization already in progress")?;
966
967 if let Some(existing) = db.read_properties().await? {
968 trace!("Existing config: {:?}", existing);
969 bail!("cannot overwrite already existing config")
970 }
971
972 let (server_pubkey, mailbox_pubkey) = if !force {
974 match Self::connect_to_server(&config, network).await {
975 Ok(conn) => {
976 let ark_info = conn.ark_info().await;
977 (Some(ark_info.server_pubkey), Some(ark_info.mailbox_pubkey))
978 }
979 Err(err) => {
980 bail!("Failed to connect to provided server (if you are sure use the --force flag): {:#}", err);
981 }
982 }
983 } else {
984 (None, None)
985 };
986
987 let properties = WalletProperties {
988 network,
989 fingerprint: wallet_fingerprint,
990 server_pubkey,
991 server_mailbox_pubkey: mailbox_pubkey,
992 };
993
994 db.init_wallet(&properties).await.context("cannot init wallet in the database")?;
996 info!("Created wallet with fingerprint: {}", wallet_fingerprint);
997 if let Some(pk) = server_pubkey {
998 info!("Stored server pubkey: {}", pk);
999 }
1000
1001 drop(create_guard);
1004
1005 let wallet = Wallet::open(&mnemonic, db, config, lock_manager).await.context("failed to open wallet")?;
1007 wallet.require_chainsource_version().await?;
1008
1009 Ok(wallet)
1010 }
1011
1012 pub async fn create_with_onchain(
1020 mnemonic: &Mnemonic,
1021 network: Network,
1022 config: Config,
1023 db: Arc<dyn BarkPersister>,
1024 lock_manager: Box<dyn LockManager>,
1025 onchain: &dyn ExitUnilaterally,
1026 force: bool,
1027 ) -> anyhow::Result<Wallet> {
1028 let wallet = Wallet::create(mnemonic, network, config, db, lock_manager, force).await?;
1029 wallet.inner.exit.load(onchain).await?;
1030 Ok(wallet)
1031 }
1032
1033 pub async fn open(
1038 mnemonic: &Mnemonic,
1039 db: Arc<dyn BarkPersister>,
1040 config: Config,
1041 lock_manager: Box<dyn LockManager>,
1042 ) -> anyhow::Result<Wallet> {
1043 let properties = db.read_properties().await?.context("Wallet is not initialised")?;
1044
1045 let seed = {
1046 let seed = mnemonic.to_seed("");
1047 WalletSeed::new(properties.network, &seed)
1048 };
1049
1050 if properties.fingerprint != seed.fingerprint() {
1051 bail!("incorrect mnemonic")
1052 }
1053
1054 let chain_source = if let Some(ref url) = config.esplora_address {
1055 ChainSourceSpec::Esplora {
1056 url: url.clone(),
1057 }
1058 } else if let Some(ref url) = config.bitcoind_address {
1059 let auth = if let Some(ref c) = config.bitcoind_cookiefile {
1060 bitcoin_ext::rpc::Auth::CookieFile(c.clone())
1061 } else {
1062 bitcoin_ext::rpc::Auth::UserPass(
1063 config.bitcoind_user.clone().context("need bitcoind auth config")?,
1064 config.bitcoind_pass.clone().context("need bitcoind auth config")?,
1065 )
1066 };
1067 ChainSourceSpec::Bitcoind { url: url.clone(), auth }
1068 } else {
1069 bail!("Need to either provide esplora or bitcoind info");
1070 };
1071
1072 #[cfg(feature = "socks5-proxy")]
1073 let chain_proxy = proxy_for_url(&config.socks5_proxy, chain_source.url())?;
1074 let chain_source_client = ChainSource::new(
1075 chain_source, properties.network, config.fallback_fee_rate,
1076 #[cfg(feature = "socks5-proxy")] chain_proxy.as_deref(),
1077 ).await?;
1078 let chain = Arc::new(chain_source_client);
1079
1080 let server = tokio::sync::OnceCell::new();
1081
1082 let notifications = NotificationDispatch::new();
1083 let movements = Arc::new(MovementManager::new(db.clone(), notifications.clone()));
1084 let exit = Exit::new(db.clone(), chain.clone(), movements.clone()).await?;
1085
1086 Ok(Wallet { inner: Arc::new(WalletInner {
1087 config, db, lock_manager, seed, exit, movements, notifications, server, chain,
1088 daemon: parking_lot::Mutex::new(None),
1089 })})
1090 }
1091
1092 pub async fn open_with_onchain(
1095 mnemonic: &Mnemonic,
1096 db: Arc<dyn BarkPersister>,
1097 onchain: &dyn ExitUnilaterally,
1098 cfg: Config,
1099 lock_manager: Box<dyn LockManager>,
1100 ) -> anyhow::Result<Wallet> {
1101 let wallet = Wallet::open(mnemonic, db, cfg, lock_manager).await?;
1102 wallet.inner.exit.load(onchain).await?;
1103 Ok(wallet)
1104 }
1105
1106 pub async fn open_with_daemon(
1109 mnemonic: &Mnemonic,
1110 db: Arc<dyn BarkPersister>,
1111 cfg: Config,
1112 onchain: Option<Arc<tokio::sync::RwLock<dyn DaemonizableOnchainWallet>>>,
1113 lock_manager: Box<dyn LockManager>,
1114 ) -> anyhow::Result<Wallet> {
1115 let wallet = Wallet::open(mnemonic, db, cfg, lock_manager).await?;
1116 if let Some(onchain) = onchain.as_ref() {
1117 let mut onchain = onchain.write().await;
1118 wallet.inner.exit.load(&mut *onchain).await?;
1119 }
1120
1121 wallet.start_daemon(onchain)?;
1122
1123 Ok(wallet)
1124 }
1125
1126 pub fn config(&self) -> &Config {
1128 &self.inner.config
1129 }
1130
1131 pub async fn properties(&self) -> anyhow::Result<WalletProperties> {
1133 let properties = self.inner.db.read_properties().await?.context("Wallet is not initialised")?;
1134 Ok(properties)
1135 }
1136
1137 pub fn fingerprint(&self) -> Fingerprint {
1139 self.inner.seed.fingerprint()
1140 }
1141
1142 async fn connect_to_server(
1143 config: &Config,
1144 network: Network,
1145 ) -> anyhow::Result<ServerConnection> {
1146 let mut builder = ServerConnection::builder()
1147 .address(&config.server_address)
1148 .network(network);
1149
1150 #[cfg(feature = "socks5-proxy")]
1151 if let Some(proxy) = proxy_for_url(&config.socks5_proxy, &config.server_address)? {
1152 builder = builder.proxy(&proxy)
1153 }
1154
1155 if let Some(ref token) = config.server_access_token {
1156 builder = builder.access_token(token);
1157 }
1158
1159 builder.connect().await.map_err(wrap_server_connect_error)
1160 .context("Failed to connect to Ark server")
1161 }
1162
1163 async fn require_server(&self) -> anyhow::Result<(ServerConnection, ArkInfo)> {
1164 let conn = self.inner.server.get_or_try_init(|| async {
1168 let network = self.properties().await?.network;
1169 Self::connect_to_server(&self.inner.config, network).await
1170 .context("You should be connected to Ark server to perform this action")
1171 }).await?.clone();
1172
1173 let ark_info = conn.ark_info().await;
1174 self.check_and_store_server_keys(&ark_info).await?;
1175
1176 Ok((conn, ark_info))
1177 }
1178
1179 pub async fn refresh_server(&self) -> anyhow::Result<()> {
1180 let srv = self.inner.server.get_or_try_init(|| async {
1186 let properties = self.properties().await?;
1187 Self::connect_to_server(&self.inner.config, properties.network).await
1188 .map_err(anyhow::Error::from)
1189 }).await?;
1190
1191 srv.check_connection().await?;
1192 let ark_info = srv.ark_info().await;
1193 ark_info.fees.validate().context("invalid fee schedule")?;
1194 self.check_and_store_server_keys(&ark_info).await?;
1195
1196 Ok(())
1197 }
1198
1199 async fn check_and_store_server_keys(&self, ark_info: &ArkInfo) -> anyhow::Result<()> {
1206 let properties = self.properties().await?;
1207
1208 if let Some(stored_pubkey) = properties.server_pubkey {
1209 if stored_pubkey != ark_info.server_pubkey {
1210 log_server_pubkey_changed_error(stored_pubkey, ark_info.server_pubkey);
1211 bail!("Server public key has changed. You should exit all your VTXOs!");
1212 }
1213 } else {
1214 self.inner.db.set_server_pubkey(ark_info.server_pubkey).await?;
1215 info!("Stored server pubkey for existing wallet: {}", ark_info.server_pubkey);
1216 }
1217
1218 if let Some(stored_mailbox_pubkey) = properties.server_mailbox_pubkey {
1219 if stored_mailbox_pubkey != ark_info.mailbox_pubkey {
1220 log_server_mailbox_pubkey_changed_error(stored_mailbox_pubkey, ark_info.mailbox_pubkey);
1221 bail!("Server mailbox public key has changed.");
1222 }
1223 } else {
1224 self.inner.db.set_server_mailbox_pubkey(ark_info.mailbox_pubkey).await?;
1225 info!("Stored server mailbox pubkey for existing wallet: {}", ark_info.mailbox_pubkey);
1226 }
1227
1228 Ok(())
1229 }
1230
1231 pub async fn ark_info(&self) -> anyhow::Result<Option<ArkInfo>> {
1233 match self.inner.server.get() {
1234 Some(srv) => Ok(Some(srv.ark_info().await)),
1235 None => Ok(None),
1236 }
1237 }
1238
1239 pub async fn require_ark_info(&self) -> anyhow::Result<ArkInfo> {
1245 let (_, ark_info) = self.require_server().await?;
1246 Ok(ark_info)
1247 }
1248
1249 pub async fn balance(&self) -> anyhow::Result<Balance> {
1253 let vtxos = self.vtxos().await?;
1254
1255 let spendable = {
1256 let mut v = vtxos.iter().collect();
1257 VtxoStateKind::Spendable.filter_vtxos(&mut v).await?;
1258 v.into_iter().map(|v| v.amount()).sum::<Amount>()
1259 };
1260
1261 let pending_lightning_send = self.pending_lightning_send_vtxos().await?.iter()
1262 .map(|v| v.amount())
1263 .sum::<Amount>();
1264
1265 let claimable_lightning_receive = self.claimable_lightning_receive_balance().await?;
1266
1267 let pending_board = self.pending_board_vtxos().await?.iter()
1268 .map(|v| v.amount())
1269 .sum::<Amount>();
1270
1271 let pending_in_round = self.pending_round_balance().await?;
1272
1273 let pending_exit = self.exit_mgr().try_pending_total();
1274
1275 Ok(Balance {
1276 spendable,
1277 pending_in_round,
1278 pending_lightning_send,
1279 claimable_lightning_receive,
1280 pending_exit,
1281 pending_board,
1282 })
1283 }
1284
1285 pub async fn validate_vtxo(&self, vtxo: &Vtxo<Full>) -> anyhow::Result<()> {
1287 let tx = self.inner.chain.get_tx(&vtxo.chain_anchor().txid).await
1288 .context("could not fetch chain tx")?;
1289
1290 let tx = tx.with_context(|| {
1291 format!("vtxo chain anchor not found for vtxo: {}", vtxo.chain_anchor().txid)
1292 })?;
1293
1294 vtxo.validate(&tx)?;
1295
1296 Ok(())
1297 }
1298
1299 pub async fn import_vtxo(&self, vtxo: &Vtxo<Full>) -> anyhow::Result<()> {
1309 if self.inner.db.get_wallet_vtxo(vtxo.id()).await?.is_some() {
1310 info!("VTXO {} already exists in wallet, skipping import", vtxo.id());
1311 return Ok(());
1312 }
1313
1314 self.validate_vtxo(vtxo).await.context("VTXO validation failed")?;
1315
1316 if self.find_signable_clause(vtxo).await.is_none() {
1317 bail!("VTXO {} is not owned by this wallet (no signable clause found)", vtxo.id());
1318 }
1319
1320 let current_height = self.inner.chain.tip().await?;
1321 if vtxo.expiry_height() <= current_height {
1322 bail!("Vtxo {} has expired", vtxo.id());
1323 }
1324
1325 self.store_spendable_vtxos([vtxo]).await.context("failed to store imported VTXO")?;
1326
1327 info!("Successfully imported VTXO {}", vtxo.id());
1328 Ok(())
1329 }
1330
1331 pub async fn get_vtxo_by_id(&self, vtxo_id: VtxoId) -> anyhow::Result<WalletVtxo> {
1333 let vtxo = self.inner.db.get_wallet_vtxo(vtxo_id).await
1334 .with_context(|| format!("Error when querying vtxo {} in database", vtxo_id))?
1335 .with_context(|| format!("The VTXO with id {} cannot be found", vtxo_id))?;
1336 Ok(vtxo)
1337 }
1338
1339 pub async fn get_full_vtxo(&self, vtxo_id: VtxoId) -> anyhow::Result<Vtxo<Full>> {
1347 self.inner.db.get_full_vtxo(vtxo_id).await
1348 .with_context(|| format!("Error when querying full vtxo {} in database", vtxo_id))?
1349 .with_context(|| format!("The VTXO with id {} cannot be found", vtxo_id))
1350 }
1351
1352 pub async fn get_full_vtxos<V: VtxoRef>(
1354 &self,
1355 vtxos: impl IntoIterator<Item = V>,
1356 ) -> anyhow::Result<Vec<Vtxo<Full>>> {
1357 let ids = vtxos.into_iter().map(|v| v.vtxo_id()).collect::<Vec<_>>();
1358 self.inner.db.get_full_vtxos(&ids).await
1359 .with_context(||
1360 format!("Error when querying full vtxos in database with IDs: {:?}", ids)
1361 )
1362 }
1363
1364 #[deprecated(since="0.1.0-beta.5", note = "Use Wallet::history instead")]
1366 pub async fn movements(&self) -> anyhow::Result<Vec<Movement>> {
1367 self.history().await
1368 }
1369
1370 pub async fn history(&self) -> anyhow::Result<Vec<Movement>> {
1372 Ok(self.inner.db.get_all_movements().await?)
1373 }
1374
1375 pub async fn update_history_metadata(
1395 &self,
1396 movement_id: MovementId,
1397 patch: &serde_json::Value,
1398 ) -> anyhow::Result<()> {
1399 self.inner.movements.patch_metadata(movement_id, patch).await?;
1400 Ok(())
1401 }
1402
1403 pub async fn history_by_payment_method(
1405 &self,
1406 payment_method: &PaymentMethod,
1407 ) -> anyhow::Result<Vec<Movement>> {
1408 let mut ret = self.inner.db.get_movements_by_payment_method(payment_method).await?;
1409 ret.sort_by_key(|m| m.id);
1410 Ok(ret)
1411 }
1412
1413 pub async fn all_vtxos(&self) -> anyhow::Result<Vec<WalletVtxo>> {
1415 Ok(self.inner.db.get_all_vtxos().await?)
1416 }
1417
1418 pub async fn vtxos(&self) -> anyhow::Result<Vec<WalletVtxo>> {
1420 Ok(self.inner.db.get_vtxos_by_state(&VtxoStateKind::UNSPENT_STATES).await?)
1421 }
1422
1423 pub async fn vtxos_with(&self, filter: &impl FilterVtxos) -> anyhow::Result<Vec<WalletVtxo>> {
1425 let mut vtxos = self.vtxos().await?;
1426 filter.filter_vtxos(&mut vtxos).await.context("error filtering vtxos")?;
1427 Ok(vtxos)
1428 }
1429
1430 pub async fn spendable_vtxos(&self) -> anyhow::Result<Vec<WalletVtxo>> {
1432 Ok(self.vtxos_with(&VtxoStateKind::Spendable).await?)
1433 }
1434
1435 pub async fn spendable_vtxos_with(
1437 &self,
1438 filter: &impl FilterVtxos,
1439 ) -> anyhow::Result<Vec<WalletVtxo>> {
1440 let mut vtxos = self.spendable_vtxos().await?;
1441 filter.filter_vtxos(&mut vtxos).await.context("error filtering vtxos")?;
1442 Ok(vtxos)
1443 }
1444
1445 pub async fn get_expiring_vtxos(
1447 &self,
1448 threshold: BlockHeight,
1449 ) -> anyhow::Result<Vec<WalletVtxo>> {
1450 let expiry = self.inner.chain.tip().await? + threshold;
1451 let filter = VtxoFilter::new(&self).expires_before(expiry);
1452 Ok(self.spendable_vtxos_with(&filter).await?)
1453 }
1454
1455 pub async fn maintenance(&self) -> anyhow::Result<()> {
1461 info!("Starting wallet maintenance in interactive mode");
1462 self.sync().await;
1463
1464 let rounds = self.progress_pending_rounds(None).await;
1465 if let Err(e) = rounds.as_ref() {
1466 warn!("Error progressing pending rounds: {:#}", e);
1467 }
1468 let refresh = self.maintenance_refresh().await;
1469 if let Err(e) = refresh.as_ref() {
1470 warn!("Error refreshing VTXOs: {:#}", e);
1471 }
1472 if rounds.is_err() || refresh.is_err() {
1473 bail!("Maintenance encountered errors.\nprogress_rounds: {:#?}\nrefresh: {:#?}", rounds, refresh);
1474 }
1475 Ok(())
1476 }
1477
1478 pub async fn maintenance_delegated(&self) -> anyhow::Result<()> {
1484 info!("Starting wallet maintenance in delegated mode");
1485 self.sync().await;
1486 let rounds = self.progress_pending_rounds(None).await;
1487 if let Err(e) = rounds.as_ref() {
1488 warn!("Error progressing pending rounds: {:#}", e);
1489 }
1490 let refresh = self.maybe_schedule_maintenance_refresh_delegated().await;
1491 if let Err(e) = refresh.as_ref() {
1492 warn!("Error refreshing VTXOs: {:#}", e);
1493 }
1494 if rounds.is_err() || refresh.is_err() {
1495 bail!("Delegated maintenance encountered errors.\nprogress_rounds: {:#?}\nrefresh: {:#?}", rounds, refresh);
1496 }
1497 Ok(())
1498 }
1499
1500 pub async fn maintenance_with_onchain<W: PreparePsbt + SignPsbt + ExitUnilaterally>(
1508 &self,
1509 onchain: &mut W,
1510 ) -> anyhow::Result<()> {
1511 info!("Starting wallet maintenance in interactive mode with onchain wallet");
1512
1513 let maintenance = self.maintenance().await;
1515
1516 let exit_sync = self.sync_exits(onchain).await;
1518 if let Err(e) = exit_sync.as_ref() {
1519 warn!("Error syncing exits: {:#}", e);
1520 }
1521 let exit_progress = self.exit_mgr().progress_exits(&self, onchain, None).await;
1522 if let Err(e) = exit_progress.as_ref() {
1523 warn!("Error progressing exits: {:#}", e);
1524 }
1525 if maintenance.is_err() || exit_sync.is_err() || exit_progress.is_err() {
1526 bail!("Maintenance encountered errors.\nmaintenance: {:#?}\nexit_sync: {:#?}\nexit_progress: {:#?}", maintenance, exit_sync, exit_progress);
1527 }
1528 Ok(())
1529 }
1530
1531 pub async fn maintenance_with_onchain_delegated<W: PreparePsbt + SignPsbt + ExitUnilaterally>(
1538 &self,
1539 onchain: &mut W,
1540 ) -> anyhow::Result<()> {
1541 info!("Starting wallet maintenance in delegated mode with onchain wallet");
1542
1543 let maintenance = self.maintenance_delegated().await;
1545
1546 let exit_sync = self.sync_exits(onchain).await;
1548 if let Err(e) = exit_sync.as_ref() {
1549 warn!("Error syncing exits: {:#}", e);
1550 }
1551 let exit_progress = self.exit_mgr().progress_exits(&self, onchain, None).await;
1552 if let Err(e) = exit_progress.as_ref() {
1553 warn!("Error progressing exits: {:#}", e);
1554 }
1555 if maintenance.is_err() || exit_sync.is_err() || exit_progress.is_err() {
1556 bail!("Delegated maintenance encountered errors.\nmaintenance: {:#?}\nexit_sync: {:#?}\nexit_progress: {:#?}", maintenance, exit_sync, exit_progress);
1557 }
1558 Ok(())
1559 }
1560
1561 pub async fn maybe_schedule_maintenance_refresh(&self) -> anyhow::Result<Option<RoundStateId>> {
1569 let vtxos = self.get_vtxos_to_refresh().await?;
1570 if vtxos.len() == 0 {
1571 return Ok(None);
1572 }
1573
1574 let participation = match self.build_refresh_participation(vtxos).await? {
1575 Some(participation) => participation,
1576 None => return Ok(None),
1577 };
1578
1579 info!("Scheduling maintenance refresh ({} vtxos)", participation.inputs.len());
1580 let state = self.join_next_round(participation, Some(RoundMovement::Refresh)).await?;
1581 Ok(Some(state.id()))
1582 }
1583
1584 pub async fn maybe_schedule_maintenance_refresh_delegated(
1592 &self,
1593 ) -> anyhow::Result<Option<RoundStateId>> {
1594 let vtxos = self.get_vtxos_to_refresh().await?;
1595 if vtxos.len() == 0 {
1596 return Ok(None);
1597 }
1598
1599 let participation = match self.build_refresh_participation(vtxos).await? {
1600 Some(participation) => participation,
1601 None => return Ok(None),
1602 };
1603
1604 info!("Scheduling delegated maintenance refresh ({} vtxos)", participation.inputs.len());
1605 let state = self.join_next_round_delegated(participation, Some(RoundMovement::Refresh)).await?;
1606 Ok(Some(state.id()))
1607 }
1608
1609 pub async fn maintenance_refresh(&self) -> anyhow::Result<Option<RoundStatus>> {
1617 let vtxos = self.get_vtxos_to_refresh().await?;
1618 if vtxos.len() == 0 {
1619 return Ok(None);
1620 }
1621
1622 info!("Performing maintenance refresh");
1623 self.refresh_vtxos(vtxos).await
1624 }
1625
1626 pub async fn sync(&self) {
1632 futures::join!(
1633 async {
1634 if let Err(e) = self.inner.chain.update_fee_rates(self.inner.config.fallback_fee_rate).await {
1637 warn!("Error updating fee rates: {:#}", e);
1638 }
1639 },
1640 async {
1641 if let Err(e) = self.sync_mailbox().await {
1642 warn!("Error in mailbox sync: {:#}", e);
1643 }
1644 },
1645 async {
1646 if let Err(e) = self.sync_pending_rounds().await {
1647 warn!("Error while trying to progress rounds awaiting confirmations: {:#}", e);
1648 }
1649 },
1650 async {
1651 if let Err(e) = self.sync_pending_lightning_send_vtxos().await {
1652 warn!("Error syncing pending lightning payments: {:#}", e);
1653 }
1654 },
1655 async {
1656 if let Err(e) = self.try_claim_all_lightning_receives(false).await {
1657 warn!("Error claiming pending lightning receives: {:#}", e);
1658 }
1659 },
1660 async {
1661 if let Err(e) = self.sync_pending_boards().await {
1662 warn!("Error syncing pending boards: {:#}", e);
1663 }
1664 },
1665 async {
1666 if let Err(e) = self.sync_pending_offboards().await {
1667 warn!("Error syncing pending offboards: {:#}", e);
1668 }
1669 }
1670 );
1671 }
1672
1673 pub async fn sync_exits(
1679 &self,
1680 onchain: &mut dyn ExitUnilaterally,
1681 ) -> anyhow::Result<()> {
1682 self.exit_mgr().sync(&self, onchain).await?;
1683 Ok(())
1684 }
1685
1686 pub async fn dangerous_drop_vtxo(&self, vtxo_id: VtxoId) -> anyhow::Result<()> {
1689 warn!("Drop vtxo {} from the database", vtxo_id);
1690 self.inner.db.remove_vtxo(vtxo_id).await?;
1691 Ok(())
1692 }
1693
1694 pub async fn dangerous_drop_all_vtxos(&self) -> anyhow::Result<()> {
1697 warn!("Dropping all vtxos from the db...");
1698 for vtxo in self.vtxos().await? {
1699 self.inner.db.remove_vtxo(vtxo.id()).await?;
1700 }
1701
1702 self.exit_mgr().dangerous_clear_exit().await?;
1703 Ok(())
1704 }
1705
1706 async fn has_counterparty_risk(&self, vtxo: &Vtxo<Full>) -> anyhow::Result<bool> {
1714 for past_pks in vtxo.past_arkoor_pubkeys() {
1715 let mut owns_any = false;
1716 for past_pk in past_pks {
1717 if self.inner.db.get_public_key_idx(&past_pk).await?.is_some() {
1718 owns_any = true;
1719 break;
1720 }
1721 }
1722 if !owns_any {
1723 return Ok(true);
1724 }
1725 }
1726
1727 let my_clause = self.find_signable_clause(vtxo).await;
1728 Ok(!my_clause.is_some())
1729 }
1730
1731 async fn add_should_refresh_vtxos(
1737 &self,
1738 participation: &mut RoundParticipation,
1739 ) -> anyhow::Result<()> {
1740 let tip = self.inner.chain.tip().await?;
1743 let mut vtxos_to_refresh = self.spendable_vtxos_with(
1744 &RefreshStrategy::should_refresh(self, tip, self.inner.chain.fee_rates().await.fast),
1745 ).await?;
1746 if vtxos_to_refresh.is_empty() {
1747 return Ok(());
1748 }
1749
1750 let excluded_ids = participation.inputs.iter().map(|v| v.vtxo_id())
1751 .collect::<HashSet<_>>();
1752 let mut total_amount = Amount::ZERO;
1753 for i in (0..vtxos_to_refresh.len()).rev() {
1754 let vtxo = &vtxos_to_refresh[i];
1755 if excluded_ids.contains(&vtxo.id()) {
1756 vtxos_to_refresh.swap_remove(i);
1757 continue;
1758 }
1759 total_amount += vtxo.amount();
1760 }
1761 if vtxos_to_refresh.is_empty() {
1762 return Ok(());
1764 }
1765
1766 let (_, ark_info) = self.require_server().await?;
1769 let fee = ark_info.fees.refresh.calculate_no_base_fee(
1770 vtxos_to_refresh.iter().map(|wv| VtxoFeeInfo::from_vtxo_and_tip(&wv.vtxo, tip)),
1771 ).context("fee overflowed")?;
1772
1773 let output_amount = match validate_and_subtract_fee_min_dust(total_amount, fee) {
1775 Ok(amount) => amount,
1776 Err(e) => {
1777 trace!("Cannot add should-refresh VTXOs: {}", e);
1778 return Ok(());
1779 },
1780 };
1781 info!(
1782 "Adding {} extra VTXOs to round participation total = {}, fee = {}, output = {}",
1783 vtxos_to_refresh.len(), total_amount, fee, output_amount,
1784 );
1785 let (user_keypair, _) = self.derive_store_next_keypair().await?;
1786 let req = VtxoRequest {
1787 policy: VtxoPolicy::new_pubkey(user_keypair.public_key()),
1788 amount: output_amount,
1789 };
1790 let extra_ids = vtxos_to_refresh.into_iter().map(|wv| wv.id()).collect::<Vec<_>>();
1791 let extra_full = self.inner.db.get_full_vtxos(&extra_ids).await
1792 .context("failed to hydrate refresh candidates")?;
1793 participation.inputs.reserve(extra_full.len());
1794 participation.inputs.extend(extra_full);
1795 participation.outputs.push(req);
1796
1797 Ok(())
1798 }
1799
1800 pub async fn build_refresh_participation<V: VtxoRef>(
1801 &self,
1802 vtxos: impl IntoIterator<Item = V>,
1803 ) -> anyhow::Result<Option<RoundParticipation>> {
1804 let (vtxos, total_amount) = {
1805 let iter = vtxos.into_iter();
1806 let size_hint = iter.size_hint();
1807 let mut vtxos = Vec::<Vtxo<Full>>::with_capacity(size_hint.1.unwrap_or(size_hint.0));
1808 let mut amount = Amount::ZERO;
1809 for vref in iter {
1810 let id = vref.vtxo_id();
1815 if vtxos.iter().any(|v| v.id() == id) {
1816 bail!("duplicate VTXO id: {}", id);
1817 }
1818 let vtxo = if let Some(vtxo) = vref.into_full_vtxo() {
1819 vtxo
1820 } else {
1821 self.inner.db.get_full_vtxo(id).await?
1824 .with_context(|| format!("vtxo with id {} not found", id))?
1825 };
1826 amount += vtxo.amount();
1827 vtxos.push(vtxo);
1828 }
1829 (vtxos, amount)
1830 };
1831
1832 if vtxos.is_empty() {
1833 info!("Skipping refresh since no VTXOs are provided.");
1834 return Ok(None);
1835 }
1836 ensure!(total_amount >= P2TR_DUST,
1837 "vtxo amount must be at least {} to participate in a round",
1838 P2TR_DUST,
1839 );
1840
1841 let (_, ark_info) = self.require_server().await?;
1843 let current_height = self.inner.chain.tip().await?;
1844 let vtxo_fee_infos = vtxos.iter()
1845 .map(|v| VtxoFeeInfo::from_vtxo_and_tip(v, current_height));
1846 let fee = ark_info.fees.refresh.calculate(vtxo_fee_infos).context("fee overflowed")?;
1847 let output_amount = validate_and_subtract_fee_min_dust(total_amount, fee)?;
1848
1849 info!("Refreshing {} VTXOs (total amount = {}, fee = {}, output = {}).",
1850 vtxos.len(), total_amount, fee, output_amount,
1851 );
1852 let (user_keypair, _) = self.derive_store_next_keypair().await?;
1853 let req = VtxoRequest {
1854 policy: VtxoPolicy::Pubkey(PubkeyVtxoPolicy { user_pubkey: user_keypair.public_key() }),
1855 amount: output_amount,
1856 };
1857
1858 Ok(Some(RoundParticipation {
1859 inputs: vtxos,
1860 outputs: vec![req],
1861 unblinded_mailbox_id: None,
1862 }))
1863 }
1864
1865 pub async fn refresh_vtxos<V: VtxoRef>(
1870 &self,
1871 vtxos: impl IntoIterator<Item = V>,
1872 ) -> anyhow::Result<Option<RoundStatus>> {
1873 let mut participation = match self.build_refresh_participation(vtxos).await? {
1874 Some(participation) => participation,
1875 None => return Ok(None),
1876 };
1877
1878 if let Err(e) = self.add_should_refresh_vtxos(&mut participation).await {
1879 warn!("Error trying to add additional VTXOs that should be refreshed: {:#}", e);
1880 }
1881
1882 Ok(Some(self.participate_round(participation, Some(RoundMovement::Refresh)).await?))
1883 }
1884
1885 pub async fn refresh_vtxos_delegated<V: VtxoRef>(
1891 &self,
1892 vtxos: impl IntoIterator<Item = V>,
1893 ) -> anyhow::Result<Option<StoredRoundState<Unlocked>>> {
1894 let mut part = match self.build_refresh_participation(vtxos).await? {
1895 Some(participation) => participation,
1896 None => return Ok(None),
1897 };
1898
1899 if let Err(e) = self.add_should_refresh_vtxos(&mut part).await {
1900 warn!("Error trying to add additional VTXOs that should be refreshed: {:#}", e);
1901 }
1902
1903 Ok(Some(self.join_next_round_delegated(part, Some(RoundMovement::Refresh)).await?))
1904 }
1905
1906 pub async fn get_vtxos_to_refresh(&self) -> anyhow::Result<Vec<WalletVtxo>> {
1909 let vtxos = self.spendable_vtxos_with(&RefreshStrategy::should_refresh_if_must(
1910 self,
1911 self.inner.chain.tip().await?,
1912 self.inner.chain.fee_rates().await.fast,
1913 )).await?;
1914 Ok(vtxos)
1915 }
1916
1917 pub async fn get_first_expiring_vtxo_blockheight(
1919 &self,
1920 ) -> anyhow::Result<Option<BlockHeight>> {
1921 Ok(self.spendable_vtxos().await?.iter().map(|v| v.expiry_height()).min())
1922 }
1923
1924 pub async fn get_next_required_refresh_blockheight(
1927 &self,
1928 ) -> anyhow::Result<Option<BlockHeight>> {
1929 let first_expiry = self.get_first_expiring_vtxo_blockheight().await?;
1930 Ok(first_expiry.map(|h| h.saturating_sub(self.inner.config.vtxo_refresh_expiry_threshold)))
1931 }
1932
1933 async fn select_vtxos_to_cover(
1939 &self,
1940 amount: Amount,
1941 ) -> anyhow::Result<Vec<WalletVtxo>> {
1942 let mut vtxos = self.spendable_vtxos().await?;
1943 self.sort_vtxos_for_selection(&mut vtxos);
1944
1945 let (last, _total_amount) = self.select_vtxos_inner(amount, &vtxos)?;
1946 vtxos.truncate(last+1);
1947 Ok(vtxos)
1948 }
1949
1950 async fn select_vtxos_to_cover_with_fee<F>(
1956 &self,
1957 amount: Amount,
1958 calc_fee: F,
1959 ) -> anyhow::Result<(Vec<WalletVtxo>, Amount)>
1960 where
1961 F: for<'a> Fn(
1962 Amount, std::iter::Copied<std::slice::Iter<'a, VtxoFeeInfo>>,
1963 ) -> anyhow::Result<Amount>,
1964 {
1965 let tip = self.inner.chain.tip().await?;
1966 let mut vtxos = self.spendable_vtxos().await?;
1967 self.sort_vtxos_for_selection(&mut vtxos);
1968
1969 let fee_info = vtxos.iter()
1970 .map(|v| VtxoFeeInfo::from_vtxo_and_tip(v, tip))
1971 .collect::<Vec<_>>();
1972
1973 const MAX_ITERATIONS: usize = 100;
1976 let mut fee = Amount::ZERO;
1977 for _ in 0..MAX_ITERATIONS {
1978 let required = amount.checked_add(fee)
1979 .context("Amount + fee overflow")?;
1980
1981 let (last, vtxo_amount) = self.select_vtxos_inner(required, &vtxos)
1982 .context("Could not find enough suitable VTXOs to cover payment + fees")?;
1983 fee = calc_fee(amount, fee_info[..=last].iter().copied())?;
1984
1985 if amount + fee <= vtxo_amount {
1986 trace!("Selected vtxos to cover amount + fee: amount = {}, fee = {}, total inputs = {}",
1987 amount, fee, vtxo_amount,
1988 );
1989 vtxos.truncate(last+1);
1990 return Ok((vtxos, fee));
1991 }
1992 trace!("VTXO sum of {} did not exceed amount {} and fee {}, iterating again",
1993 vtxo_amount, amount, fee,
1994 );
1995 }
1996 bail!("Fee calculation did not converge after maximum iterations")
1997 }
1998
1999 fn sort_vtxos_for_selection(&self, vtxos: &mut Vec<WalletVtxo>) {
2001 vtxos.sort_by_key(|v| v.expiry_height());
2002 }
2003
2004 fn select_vtxos_inner(
2010 &self,
2011 amount: Amount,
2012 vtxos: &Vec<WalletVtxo>,
2013 ) -> anyhow::Result<(usize, Amount)> {
2014 let mut total_amount = Amount::ZERO;
2016 for (i, vtxo) in vtxos.iter().enumerate() {
2017 total_amount += vtxo.amount();
2018
2019 if total_amount >= amount {
2020 return Ok((i, total_amount))
2021 }
2022 }
2023
2024 bail!("Insufficient money available. Needed {} but {} is available",
2025 amount, total_amount,
2026 );
2027 }
2028
2029 pub fn start_daemon(
2035 &self,
2036 onchain: Option<Arc<tokio::sync::RwLock<dyn DaemonizableOnchainWallet>>>,
2037 ) -> anyhow::Result<()> {
2038 let mut daemon = self.inner.daemon.lock();
2039 if daemon.is_some() {
2040 warn!("Called Wallet::start_daemon while daemon was already running.");
2041 return Ok(());
2042 }
2043
2044 let handle = crate::daemon::start_daemon(self.clone(), onchain);
2047 let _ = daemon.insert(handle);
2048
2049 Ok(())
2050 }
2051
2052 #[deprecated(since = "0.1.4", note = "use start_daemon instead")]
2054 pub fn run_daemon(
2055 &self,
2056 onchain: Option<Arc<tokio::sync::RwLock<dyn DaemonizableOnchainWallet>>>,
2057 ) -> anyhow::Result<()> {
2058 self.start_daemon(onchain)
2059 }
2060
2061 pub fn stop_daemon(&self) {
2063 let mut daemon = self.inner.daemon.lock();
2064 if let Some(handle) = daemon.take() {
2065 handle.stop();
2066 }
2067 }
2068
2069 pub async fn register_vtxo_transactions_with_server(
2073 &self,
2074 vtxos: &[impl AsRef<Vtxo<Full>>],
2075 ) -> anyhow::Result<()> {
2076 if vtxos.is_empty() {
2077 return Ok(());
2078 }
2079
2080 let (mut srv, _) = self.require_server().await?;
2081 srv.client.register_vtxo_transactions(protos::RegisterVtxoTransactionsRequest {
2082 vtxos: vtxos.iter().map(|v| v.as_ref().serialize()).collect(),
2083 }).await.context("failed to register vtxo transactions")?;
2084
2085 Ok(())
2086 }
2087}
2088
2089fn wrap_server_connect_error(err: ConnectError) -> anyhow::Error {
2090 match err {
2091 ConnectError::CreateEndpoint(CreateEndpointError::NoTransportBackend) => {
2092 anyhow!(MISSING_SERVER_TRANSPORT_HELP)
2093 },
2094 other => anyhow::Error::from(other),
2095 }
2096}
2097
2098impl std::ops::Drop for WalletInner {
2099 fn drop(&mut self) {
2100 if let Some(handle) = self.daemon.lock().take() {
2101 handle.stop();
2102 }
2103 }
2104}
2105
2106#[cfg(test)]
2107mod tests {
2108 use server_rpc::client::CreateEndpointError;
2109
2110 use super::{wrap_server_connect_error, MISSING_SERVER_TRANSPORT_HELP};
2111
2112 #[test]
2113 fn no_transport_connect_error_is_reworded_for_wallet_users() {
2114 let err = wrap_server_connect_error(CreateEndpointError::NoTransportBackend.into());
2115 assert!(err.to_string().contains(MISSING_SERVER_TRANSPORT_HELP));
2116 assert!(err.to_string().contains("feature `bark-wallet/native` or `bark-wallet/wasm-web`"));
2117 }
2118}