feat(suite-native): only download token definitions for relevant networks

Author: vytick

suite-native/discovery/package.json

@@ -26,6 +26,7 @@
         "@suite-native/device": "workspace:*",
         "@suite-native/device-mutex": "workspace:*",
         "@suite-native/feature-flags": "workspace:*",
+        "@suite-native/tokens": "workspace:*",
         "@trezor/connect": "workspace:*",
         "@trezor/utils": "workspace:*",
         "proxy-memoize": "2.0.2",

suite-native/discovery/src/discoveryConfigSlice.ts

@@ -1,5 +1,5 @@
 import { createSlice, PayloadAction } from '@reduxjs/toolkit';
-import { pipe } from '@mobily/ts-belt';
+import { A, F, pipe } from '@mobily/ts-belt';
 import { memoize, memoizeWithArgs } from 'proxy-memoize';
 
 import {
@@ -21,6 +21,10 @@ import {
     FeatureFlagsRootState,
     selectIsFeatureFlagEnabled,
 } from '@suite-native/feature-flags';
+import {
+    selectNetworkSymbolsOfAccountsWithTokensAllowed,
+    TokensRootState,
+} from '@suite-native/tokens';
 
 type DiscoveryInfo = {
     startTimestamp: number;
@@ -193,6 +197,15 @@ export const selectDeviceEnabledDiscoveryNetworkSymbols = memoizeWithArgs(
     { size: 2 },
 );
 
+export const selectTokenDefinitionsEnabledNetworks = memoize(
+    (state: TokensRootState & DiscoveryConfigSliceRootState) => {
+        const enabledNetworkSymbols = selectEnabledDiscoveryNetworkSymbols(state);
+        const accountNetworkSymbols = selectNetworkSymbolsOfAccountsWithTokensAllowed(state);
+
+        return F.toMutable(A.uniq([...enabledNetworkSymbols, ...accountNetworkSymbols]));
+    },
+);
+
 export const {
     toggleAreTestnetsEnabled,
     setDiscoveryInfo,

suite-native/discovery/src/discoveryMiddleware.ts

@@ -1,3 +1,5 @@
+import { createMiddlewareWithExtraDeps } from '@suite-common/redux-utils';
+import { periodicCheckTokenDefinitionsThunk } from '@suite-common/token-definitions';
 import {
     deviceActions,
     discoveryActions,
@@ -6,14 +8,15 @@ import {
     authorizeDeviceThunk,
     accountsActions,
 } from '@suite-common/wallet-core';
-import { createMiddlewareWithExtraDeps } from '@suite-common/redux-utils';
 import { isFirmwareVersionSupported } from '@suite-native/device';
 
 import { startDescriptorPreloadedDiscoveryThunk, discoveryCheckThunk } from './discoveryThunks';
 import {
     selectAreTestnetsEnabled,
     selectIsCoinEnablingInitFinished,
     toggleAreTestnetsEnabled,
+    setEnabledDiscoveryNetworkSymbols,
+    toggleEnabledDiscoveryNetworkSymbol,
 } from './discoveryConfigSlice';
 
 export const prepareDiscoveryMiddleware = createMiddlewareWithExtraDeps(
@@ -75,6 +78,15 @@ export const prepareDiscoveryMiddleware = createMiddlewareWithExtraDeps(
             dispatch(discoveryCheckThunk());
         }
 
+        // if we changed enabled networks, check for token definitions right away
+        if (
+            (toggleEnabledDiscoveryNetworkSymbol.match(action) ||
+                setEnabledDiscoveryNetworkSymbols.match(action)) &&
+            isCoinEnablingInitFinished
+        ) {
+            dispatch(periodicCheckTokenDefinitionsThunk());
+        }
+
         return action;
     },
 );

suite-native/discovery/tsconfig.json

@@ -29,6 +29,7 @@
         { "path": "../device" },
         { "path": "../device-mutex" },
         { "path": "../feature-flags" },
+        { "path": "../tokens" },
         { "path": "../../packages/connect" },
         { "path": "../../packages/utils" }
     ]

suite-native/module-accounts-import/src/accountsImportThunks.ts

@@ -11,7 +11,10 @@ import { getXpubOrDescriptorInfo } from '@trezor/utxo-lib';
 import { getAccountIdentity, shouldUseIdentities } from '@suite-common/wallet-utils';
 import { Timestamp, TokenAddress } from '@suite-common/wallet-types';
 import { FiatCurrencyCode } from '@suite-common/suite-config';
-import { selectFilterKnownTokens } from '@suite-common/token-definitions';
+import {
+    periodicCheckTokenDefinitionsThunk,
+    selectFilterKnownTokens,
+} from '@suite-common/token-definitions';
 
 import { paymentTypeToAccountType } from './constants';
 
@@ -71,6 +74,7 @@ export const importAccountThunk = createThunk(
                 }),
             );
         }
+        dispatch(periodicCheckTokenDefinitionsThunk());
     },
 );
 

suite-native/state/package.json

@@ -31,7 +31,6 @@
         "@suite-native/module-send": "workspace:*",
         "@suite-native/settings": "workspace:*",
         "@suite-native/storage": "workspace:*",
-        "@suite-native/tokens": "workspace:*",
         "@trezor/transport-native": "workspace:*",
         "@trezor/utils": "workspace:*",
         "expo-device": "6.0.2",

suite-native/state/src/extraDependencies.ts

@@ -13,8 +13,10 @@ import {
 } from '@suite-native/settings';
 import { mergeDeepObject } from '@trezor/utils';
 import { NativeUsbTransport } from '@trezor/transport-native';
-import { selectEnabledDiscoveryNetworkSymbols } from '@suite-native/discovery';
-import { NETWORK_SYMBOLS_WITH_TOKENS } from '@suite-native/tokens';
+import {
+    selectEnabledDiscoveryNetworkSymbols,
+    selectTokenDefinitionsEnabledNetworks,
+} from '@suite-native/discovery';
 
 const deviceType = Device.isDevice ? 'device' : 'emulator';
 
@@ -34,7 +36,7 @@ export const extraDependencies: ExtraDependencies = mergeDeepObject(extraDepende
         // otherwise disableAccountsThunk might erase accounts not supported by current device
         selectEnabledNetworks: selectEnabledDiscoveryNetworkSymbols,
         // todo: this is temporary solution to make token definitions work on native in portfolio tracker
-        selectTokenDefinitionsEnabledNetworks: () => NETWORK_SYMBOLS_WITH_TOKENS,
+        selectTokenDefinitionsEnabledNetworks,
         selectBitcoinAmountUnit: selectBitcoinUnits,
         selectAreSatsAmountUnit,
         selectLocalCurrency: selectFiatCurrencyCode,

suite-native/state/tsconfig.json

@@ -34,7 +34,6 @@
         { "path": "../module-send" },
         { "path": "../settings" },
         { "path": "../storage" },
-        { "path": "../tokens" },
         {
             "path": "../../packages/transport-native"
         },

suite-native/tokens/src/tokensSelectors.ts

@@ -10,6 +10,7 @@ import {
     AccountsRootState,
     DeviceRootState,
     selectAccountByKey,
+    selectAccounts,
     selectAccountTransactions,
     selectVisibleDeviceAccountsByNetworkSymbol,
     TransactionsRootState,
@@ -275,3 +276,14 @@ export const selectAccountHasAnyKnownToken = (state: TokensRootState, accountKey
 
     return anyOfTokensIsKnown;
 };
+
+export const selectNetworkSymbolsOfAccountsWithTokensAllowed = (state: TokensRootState) =>
+    selectAccounts(state)
+        .filter(a => isCoinWithTokens(a.symbol))
+        .reduce((acc, account) => {
+            if (!acc.includes(account.symbol)) {
+                acc.push(account.symbol);
+            }
+
+            return acc;
+        }, new Array<NetworkSymbol>());

yarn.lock

@@ -9734,6 +9734,7 @@ __metadata:
     "@suite-native/device": "workspace:*"
     "@suite-native/device-mutex": "workspace:*"
     "@suite-native/feature-flags": "workspace:*"
+    "@suite-native/tokens": "workspace:*"
     "@trezor/connect": "workspace:*"
     "@trezor/utils": "workspace:*"
     proxy-memoize: "npm:2.0.2"
@@ -10507,7 +10508,6 @@ __metadata:
     "@suite-native/module-send": "workspace:*"
     "@suite-native/settings": "workspace:*"
     "@suite-native/storage": "workspace:*"
-    "@suite-native/tokens": "workspace:*"
     "@trezor/transport-native": "workspace:*"
     "@trezor/utils": "workspace:*"
     expo-device: "npm:6.0.2"