feat(suite-common): fetch stake data based on device account symbols

trezor · Nov 20, 2024 · be23fc9 · be23fc9
1 parent 58b854c
commit be23fc9
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 3 deletions.
diff --git a/suite-common/wallet-core/src/accounts/accountsReducer.ts b/suite-common/wallet-core/src/accounts/accountsReducer.ts
@@ -1,5 +1,5 @@
 import { isAnyOf } from '@reduxjs/toolkit';
-import { A, G, pipe } from '@mobily/ts-belt';
+import { A, F, G, pipe } from '@mobily/ts-belt';
 
 import {
     createReducerWithExtraDeps,
@@ -325,6 +325,18 @@ export const selectVisibleNonEmptyDeviceAccountsByNetworkSymbol = createMemoized
         ),
 );
 
+export const selectAllNetworkSymbolsOfVisibleAccounts = createMemoizedSelector(
+    [selectAccounts],
+    accounts =>
+        pipe(
+            accounts,
+            A.filter(account => account.visible),
+            A.map(account => account.symbol),
+            A.uniq,
+            F.toMutable,
+        ),
+);
+
 export const selectNonEmptyDeviceAccounts = createMemoizedSelector(
     [selectDeviceAccounts],
     accounts =>

diff --git a/suite-common/wallet-core/src/stake/stakeThunks.ts b/suite-common/wallet-core/src/stake/stakeThunks.ts
@@ -7,6 +7,7 @@ import { SupportedNetworkSymbol } from '@suite-common/wallet-types';
 import { selectEverstakeData } from './stakeSelectors';
 import { EVERSTAKE_ENDPOINT_TYPES, EverstakeEndpointType, ValidatorsQueue } from './stakeTypes';
 import { EVERSTAKE_ENDPOINT_PREFIX } from './stakeConstants';
+import { selectAllNetworkSymbolsOfVisibleAccounts } from '../accounts/accountsReducer';
 
 const STAKE_MODULE = '@common/wallet-core/stake';
 
@@ -60,11 +61,15 @@ export const fetchEverstakeData = createThunk<
 export const initStakeDataThunk = createThunk(
     `${STAKE_MODULE}/initStakeDataThunk`,
     (_, { getState, dispatch, extra }) => {
+        //because fetch only happens every 5 minutes we fetch according all devices in case a device is changed within those 5 minutes
+        const accountsNetworks = selectAllNetworkSymbolsOfVisibleAccounts(getState());
+        //also join with enabled networks in case account was not yet discovered, but network is already enabled
         const enabledNetworks = extra.selectors.selectEnabledNetworks(getState());
+        const networks = [...new Set([...accountsNetworks, ...enabledNetworks])];
 
-        const ethereumBasedNetworksWithStaking = getStakingSymbols(enabledNetworks);
+        const networksWithStaking = getStakingSymbols(networks);
 
-        const promises = ethereumBasedNetworksWithStaking.flatMap(networkSymbol =>
+        const promises = networksWithStaking.flatMap(networkSymbol =>
             Object.values(EverstakeEndpointType).map(endpointType => {
                 const data = selectEverstakeData(getState(), networkSymbol, endpointType);