MyCrypto/common/selectors/wallet.js

// @flow
import type { State } from 'reducers';
import { BaseWallet } from 'libs/wallet';
import { getNetworkConfig } from 'selectors/config';
import Big from 'bignumber.js';
import type { Token } from 'config/data';
import type { BroadcastStatusTransaction } from 'libs/transaction';

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
  }));
}

export function getTxFromState(
  state: State,
  signedTx: string
): ?BroadcastStatusTransaction {
  const transactions = state.wallet.transactions;
  return getTxFromBroadcastStatusTransactions(transactions, signedTx);
}

export function getTxFromBroadcastStatusTransactions(
  transactions: Array<BroadcastStatusTransaction>,
  signedTx: string
): ?BroadcastStatusTransaction {
  return transactions.find(transaction => {
    return transaction.signedTx === signedTx;
  });
}