bp: add RgbWallet::wallet_state method

Author: dr-orlovsky

src/contracts.rs

@@ -59,7 +59,13 @@ pub struct ContractStateName {
     pub state_name: StateName,
 }
 
-#[derive(Clone, Eq, PartialEq, Debug, Default)]
+impl ContractStateName {
+    pub fn new(contract_id: ContractId, state_name: StateName) -> Self {
+        ContractStateName { contract_id, state_name }
+    }
+}
+
+#[derive(Clone, Eq, PartialEq, Debug)]
 #[cfg_attr(
     feature = "serde",
     derive(Serialize, Deserialize),
@@ -74,6 +80,10 @@ pub struct WalletState<Seal> {
     pub aggregated: BTreeMap<ContractStateName, StrictVal>,
 }
 
+impl<Seal> Default for WalletState<Seal> {
+    fn default() -> Self { Self { immutable: bmap! {}, owned: bmap! {}, aggregated: bmap! {} } }
+}
+
 /// Collection of RGB smart contracts and contract issuers, which can be cached in memory.
 ///
 /// # Generics

src/popls/bp.rs

@@ -55,8 +55,9 @@ use strict_types::StrictVal;
 
 use crate::contracts::SyncError;
 use crate::{
-    Assignment, CodexId, Consensus, ConsumeError, Contract, ContractState, Contracts, CreateParams,
-    EitherSeal, Identity, Issuer, IssuerError, OwnedState, Pile, SigBlob, Stockpile, WitnessStatus,
+    Assignment, CodexId, Consensus, ConsumeError, Contract, ContractState, ContractStateName,
+    Contracts, CreateParams, EitherSeal, Identity, Issuer, IssuerError, OwnedState, Pile, SigBlob,
+    Stockpile, WalletState, WitnessStatus,
 };
 
 /// Trait abstracting a specific implementation of a bitcoin wallet.
@@ -466,6 +467,29 @@ where
         WitnessOut::new(address.payload, nonce)
     }
 
+    pub fn wallet_state(&self) -> WalletState<TxoSeal> {
+        let mut wallet_state = WalletState::default();
+        for contract_id in self.contracts.contract_ids() {
+            let contract_state = self.contracts.contract_state(contract_id);
+            for (state_name, state) in contract_state.immutable {
+                wallet_state
+                    .immutable
+                    .insert(ContractStateName::new(contract_id, state_name), state);
+            }
+            for (state_name, state) in contract_state.owned {
+                wallet_state
+                    .owned
+                    .insert(ContractStateName::new(contract_id, state_name), state);
+            }
+            for (state_name, state) in contract_state.aggregated {
+                wallet_state
+                    .aggregated
+                    .insert(ContractStateName::new(contract_id, state_name), state);
+            }
+        }
+        wallet_state
+    }
+
     pub fn wallet_contract_state(&self, contract_id: ContractId) -> ContractState<Outpoint> {
         self.contracts
             .contract_state(contract_id)