action for fetching active token balances wip

src/logic/tokens/store/actions/fetchTokenBalances.js

@@ -1,10 +1,11 @@
 // @flow
-import type { Dispatch as ReduxDispatch } from 'redux'
+import type { Dispatch as ReduxDispatch, GetState } from 'redux'
 import { Map, List } from 'immutable'
 import { type Token } from '~/logic/tokens/store/model/token'
 import { type GlobalState } from '~/store/index'
 import { calculateBalanceOf } from './fetchTokens'
 import { addTokens } from './saveTokens'
+import { activeTokensSelector } from '~/logic/tokens/store/selectors'
 
 const fetchTokenBalances = (safeAddress: string, tokens: List<Token>) => async (
   dispatch: ReduxDispatch<GlobalState>,
@@ -24,4 +25,18 @@ const fetchTokenBalances = (safeAddress: string, tokens: List<Token>) => async (
   }
 }
 
+export const fetchActiveTokenBalances = (safeAddress: string) => async (
+  dispatch: ReduxDispatch<GlobalState>,
+  getState: GetState<GlobalState>,
+) => {
+  try {
+    const state = getState()
+    const activeTokens = activeTokensSelector(state)
+    dispatch(fetchTokenBalances(safeAddress, activeTokens))
+  } catch (err) {
+    // eslint-disable-next-line
+    console.error('Error while loading active tokens from storage:', err)
+  }
+}
+
 export default fetchTokenBalances

src/routes/safe/container/index.jsx

@@ -35,18 +35,17 @@ class SafeView extends React.PureComponent<Props> {
 
   componentDidUpdate(prevProps) {
     const {
-      safe, fetchTokenBalances, loadActiveTokens, activeTokens,
+      fetchTokenBalances, loadActiveTokens, activeTokens, safeUrl,
     } = this.props
 
     if (prevProps.safe) {
       return
     }
 
-    if (safe) {
-      const safeAddress = safe.get('address')
-      loadActiveTokens(safeAddress)
+    if (safeUrl) {
+      loadActiveTokens(safeUrl)
       if (activeTokens.size) {
-        fetchTokenBalances(safeAddress, activeTokens)
+        fetchTokenBalances(safeUrl, activeTokens)
       }
     }
   }