display dxTokenBalance even if tokenBalance == 0 (#388)

gnosis · Oct 8, 2018 · 83efd2e · 83efd2e
1 parent 641ecb8
commit 83efd2e
Showing 1 changed file with 8 additions and 4 deletions.
diff --git a/src/components/MenuWallet/index.tsx b/src/components/MenuWallet/index.tsx
@@ -92,17 +92,21 @@ export class MenuWallet extends React.Component<WalletProps, WalletState> {
                 {Object.keys(tokens).map((addressKey: any) => {
                   if (!addressToSymbolDecimal[addressKey]) return null
                   const { name, decimals } = addressToSymbolDecimal[addressKey]
+                  const tokenBalAvailable = tokens[addressKey].gt(0)
+                  const dxTokenAvailable = dxBalances[addressKey]
+                  const dxTokenBalAvailable = dxTokenAvailable && dxBalances[addressKey].gt(0)
+
                   return (
-                    tokens[addressKey].gt(0) &&
+                    (tokenBalAvailable || dxTokenBalAvailable) &&
                     <tr key={addressKey}>
                       <td>{name || 'Unknown'}</td>
                       <td>{(tokens[addressKey]).div(10 ** decimals).toFixed(FIXED_DECIMALS)}</td>
 
                       {// Conditionally render dxBalances column
                       dxBalancesAvailable &&
-                        <td className={dxBalances[addressKey] && dxBalances[addressKey].gt(0) ? 'withPic' : ''}>
-                          {dxBalances[addressKey] && dxBalances[addressKey].div(10 ** decimals).toFixed(FIXED_DECIMALS)}
-                          {dxBalances[addressKey] && dxBalances[addressKey].gt(0) &&
+                        <td className={dxTokenBalAvailable ? 'withPic' : ''}>
+                          {dxTokenAvailable && dxBalances[addressKey].div(10 ** decimals).toFixed(FIXED_DECIMALS)}
+                          {dxTokenBalAvailable &&
                             <img
                               src={require('assets/claim.svg')}
                               onClick={() => withdrawFromDutchX({ name, address: addressKey })}