Precompute native token asset types to increase chances of decoding.

Author: murisi

crates/apps_lib/src/client/rpc.rs

@@ -488,8 +488,12 @@ async fn query_shielded_balance(
     {
         let mut shielded = context.shielded_mut().await;
         let _ = shielded.load().await;
+        // Precompute asset types to increase chances of success in decoding
+        let token_map = context.wallet().await.get_addresses();
+        let mut tokens: BTreeSet<_> = token_map.values().collect();
+        tokens.insert(&token);
         let _ = shielded
-            .precompute_asset_types(context.client(), vec![&token])
+            .precompute_asset_types(context.client(), tokens)
             .await;
         // Save the update state so that future fetches can be short-circuited
         let _ = shielded.save().await;

crates/shielded_token/src/masp/shielded_wallet.rs

@@ -437,7 +437,7 @@ pub trait ShieldedApi<U: ShieldedUtils + MaybeSend + MaybeSync>:
     async fn precompute_asset_types<C: Client + Sync>(
         &mut self,
         client: &C,
-        tokens: Vec<&Address>,
+        tokens: BTreeSet<&Address>,
     ) -> Result<(), eyre::Error> {
         // To facilitate lookups of human-readable token names
         for token in tokens {

crates/shielded_token/src/vp.rs

@@ -423,7 +423,7 @@ where
 
         // Note the balance changes they imply
         let ChangedBalances {
-            unepoched_tokens: tokens,
+            unepoched_tokens,
             decoder,
             pre,
             post,
@@ -438,7 +438,7 @@ where
                 keys_changed,
             )?;
         Ok(ChangedBalances {
-            unepoched_tokens: tokens,
+            unepoched_tokens,
             decoder,
             pre,
             post,