feat(wallet): fetch asset info of all assets in balance

We have recently updated BaseWallet to operate with partial tx history.
AssetsTracker is fetching AssetInfo only for assets found in tx history.
Therefore it wouldn't fetch asset info for assets currently in balance.

Author: mkazlauskas

packages/wallet/src/services/AssetsTracker.ts

@@ -5,6 +5,7 @@ import { Logger } from 'ts-log';
 import {
   Observable,
   buffer,
+  combineLatest,
   concat,
   connect,
   debounceTime,
@@ -151,14 +152,19 @@ export const createAssetService =
     ).pipe(map((arr) => arr.flat())); // Concatenate the chunk results
 
 export type AssetService = ReturnType<typeof createAssetService>;
+export type UtxoTotalBalance = {
+  utxo: {
+    total$: BalanceTracker['utxo']['total$'];
+  };
+};
 
 export interface AssetsTrackerProps {
   transactionsTracker: TransactionsTracker;
   assetProvider: TrackedAssetProvider;
   retryBackoffConfig: RetryBackoffConfig;
   logger: Logger;
   assetsCache$: Observable<Assets>;
-  balanceTracker: BalanceTracker;
+  balanceTracker: UtxoTotalBalance;
   maxAssetInfoCacheAge?: Milliseconds;
 }
 
@@ -198,13 +204,14 @@ export const createAssetsTracker = (
     const allAssetIds = new Set<Cardano.AssetId>();
     const sharedHistory$ = history$.pipe(share());
     return concat(
-      sharedHistory$.pipe(
-        map((historyTxs) => uniq(historyTxs.flatMap(uniqueAssetIds))),
+      combineLatest([
+        sharedHistory$.pipe(map((historyTxs) => uniq(historyTxs.flatMap(uniqueAssetIds)))),
+        total$.pipe(map((balance) => [...(balance.assets?.keys() || [])]))
+      ]).pipe(
+        map(([txAssets, balanceAssets]) => uniq([...txAssets, ...balanceAssets])),
         tap((assetIds) =>
           logger.debug(
-            assetIds.length > 0
-              ? `Historical total assets: ${assetIds.length}`
-              : 'Setting assetProvider stats as initialized'
+            assetIds.length > 0 ? `Total assets: ${assetIds.length}` : 'Setting assetProvider stats as initialized'
           )
         ),
         tap((assetIds) => assetIds.length === 0 && assetProvider.setStatInitialized(assetProvider.stats.getAsset$)),

packages/wallet/test/services/AssetsTracker.test.ts

@@ -3,9 +3,9 @@ import { AssetId, createTestScheduler, generateRandomHexString, logger } from '@
 import {
   AssetService,
   AssetsTrackerProps,
-  BalanceTracker,
   TrackedAssetProvider,
   TransactionsTracker,
+  UtxoTotalBalance,
   createAssetService,
   createAssetsTracker
 } from '../../src/services';
@@ -55,7 +55,7 @@ const assetInfo = {
 describe('createAssetsTracker', () => {
   let assetService: AssetService;
   let assetProvider: TrackedAssetProvider;
-  let balanceTracker: BalanceTracker;
+  let balanceTracker: UtxoTotalBalance;
   let assetsCache$: Observable<Map<Cardano.AssetId, Asset.AssetInfo>>;
   const retryBackoffConfig: RetryBackoffConfig = { initialInterval: 2 };
 
@@ -86,14 +86,8 @@ describe('createAssetsTracker', () => {
     } as unknown as TrackedAssetProvider;
 
     balanceTracker = {
-      rewardAccounts: {
-        deposit$: of(0n),
-        rewards$: of(0n)
-      },
       utxo: {
-        available$: of({ coins: 0n }),
-        total$: of({ coins: 0n }),
-        unspendable$: of({ coins: 0n })
+        total$: of({ coins: 0n })
       }
     };
 
@@ -144,6 +138,52 @@ describe('createAssetsTracker', () => {
     });
   });
 
+  it('fetches asset info for assets in balance but not in transaction history', () => {
+    createTestScheduler().run(({ cold, expectObservable, flush }) => {
+      const transactionsTracker: Partial<TransactionsTracker> = {
+        history$: cold('a', {
+          a: [createTxWithValues([{ assets: new Map([[AssetId.TSLA, 1n]]) }])]
+        })
+      };
+
+      balanceTracker = {
+        utxo: {
+          total$: cold('a', {
+            a: {
+              assets: new Map([
+                [AssetId.TSLA, 1n],
+                [AssetId.PXL, 2n]
+              ]),
+              coins: 1n
+            }
+          })
+        }
+      };
+
+      const target$ = createAssetsTracker(
+        {
+          assetProvider,
+          assetsCache$,
+          balanceTracker,
+          logger,
+          retryBackoffConfig,
+          transactionsTracker
+        } as unknown as AssetsTrackerProps,
+        {
+          assetService
+        }
+      );
+      expectObservable(target$).toBe('a', {
+        a: new Map([
+          [AssetId.TSLA, assetInfo.TSLA],
+          [AssetId.PXL, assetInfo.PXL]
+        ])
+      });
+      flush();
+      expect(assetService).toHaveBeenCalledTimes(1);
+    });
+  });
+
   it('re-fetches asset info when there is a cip68 reference nft in some tx history output', () => {
     createTestScheduler().run(({ cold, expectObservable, flush }) => {
       const transactionsTracker: Partial<TransactionsTracker> = {