notedeck

commit dbe71bbb80c6802db4f791b8505ba6e89221bfd7
parent 6b458431036de4032f0bdee3bf1c304f192181f0
Author: kernelkind <kernelkind@gmail.com>
Date:   Tue,  8 Apr 2025 22:04:58 -0400

add get wallet method

Diffstat:
M
crates/notedeck/src/wallet.rs
 | 
17
++++++++++++++++-

1 file changed, 16 insertions(+), 1 deletion(-)
diff --git a/crates/notedeck/src/wallet.rs b/crates/notedeck/src/wallet.rs
@@ -8,7 +8,7 @@ use poll_promise::Promise;
 use tokenator::TokenSerializable;
 use tokio::sync::RwLock;
 
-use crate::{DataPath, TokenHandler};
+use crate::{Accounts, DataPath, TokenHandler};
 
 #[derive(Debug)]
 pub enum WalletState<'a> {
@@ -22,6 +22,21 @@ pub enum WalletState<'a> {
     },
 }
 
+#[allow(dead_code)]
+pub fn get_wallet_for_mut<'a>(
+    accounts: &'a mut Accounts,
+    global_wallet: &'a mut GlobalWallet,
+    account_pk: &'a [u8; 32],
+) -> Option<&'a mut Wallet> {
+    let cur_acc = accounts.get_account_mut_optimized(account_pk)?;
+
+    if let Some(wallet) = &mut cur_acc.wallet {
+        return Some(wallet);
+    }
+
+    global_wallet.wallet.as_mut()
+}
+
 #[derive(Clone, Eq, PartialEq, Debug)]
 pub enum WalletType {
     Auto,