MyCrypto/common/selectors/wallet.js

42 lines
981 B
JavaScript

// @flow
import type { State } from 'reducers';
import { BaseWallet } from 'libs/wallet';
import { getNetworkConfig } from 'selectors/config';
import Big from 'big.js';
import type { Token } from 'config/data';
export function getWalletInst(state: State): ?BaseWallet {
return state.wallet.inst;
}
export type TokenBalance = {
symbol: string,
balance: Big,
custom: boolean
};
type MergedToken = Token & {
custom: boolean
};
export function getTokens(state: State): MergedToken[] {
const tokens: MergedToken[] = (getNetworkConfig(state).tokens: any);
return tokens.concat(
state.customTokens.map(token => ({ ...token, custom: true }))
);
}
export function getTokenBalances(state: State): TokenBalance[] {
const tokens = getTokens(state);
if (!tokens) {
return [];
}
return tokens.map(t => ({
symbol: t.symbol,
balance: state.wallet.tokens[t.symbol]
? state.wallet.tokens[t.symbol]
: new Big(0),
custom: t.custom
}));
}