openethereum/js/src/ui/Balance/balance.js

// Copyright 2015-2017 Parity Technologies (UK) Ltd.
// This file is part of Parity.

// Parity is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.

// Parity is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU General Public License for more details.

// You should have received a copy of the GNU General Public License
// along with Parity.  If not, see <http://www.gnu.org/licenses/>.

import BigNumber from 'bignumber.js';
import React, { Component, PropTypes } from 'react';
import { FormattedMessage } from 'react-intl';
import { connect } from 'react-redux';

import TokenImage from '~/ui/TokenImage';

import styles from './balance.css';

export class Balance extends Component {
  static contextTypes = {
    api: PropTypes.object
  };

  static propTypes = {
    balance: PropTypes.object.isRequired,
    tokens: PropTypes.object.isRequired,
    address: PropTypes.string,
    className: PropTypes.string,
    showOnlyEth: PropTypes.bool,
    showZeroValues: PropTypes.bool
  };

  static defaultProps = {
    showOnlyEth: false,
    showZeroValues: false
  };

  render () {
    const { balance, className, showOnlyEth, tokens } = this.props;

    if (Object.keys(balance).length === 0) {
      return null;
    }

    let body = Object.keys(balance)
      .sort((tokenIdA, tokenIdB) => {
        const tokenA = tokens[tokenIdA];
        const tokenB = tokens[tokenIdB];

        if (tokenA.native) {
          return -1;
        }

        if (tokenB.native) {
          return 1;
        }

        return (tokenA.name || tokenA.tag || '').localeCompare(tokenB.name || tokenB.tag || '');
      })
      .map((tokenId) => {
        const token = tokens[tokenId];
        const balanceValue = balance[tokenId];

        const isEthToken = token.native;
        const isFullToken = !showOnlyEth || isEthToken;
        const hasBalance = (balanceValue instanceof BigNumber) && balanceValue.gt(0);

        if (!hasBalance && !isEthToken) {
          return null;
        }

        const bnf = new BigNumber(token.format || 1);
        let decimals = 0;

        if (bnf.gte(1000)) {
          decimals = 3;
        } else if (bnf.gte(100)) {
          decimals = 2;
        } else if (bnf.gte(10)) {
          decimals = 1;
        }

        const rawValue = new BigNumber(balanceValue).div(bnf);
        const value = rawValue.toFormat(decimals);

        const classNames = [styles.balance];
        let details = null;

        if (isFullToken) {
          classNames.push(styles.full);
          details = [
            <div
              className={ styles.value }
              key='value'
            >
              <span title={ `${rawValue.toFormat()} ${token.tag}` }>
                { value }
              </span>
            </div>,
            <div
              className={ styles.tag }
              key='tag'
            >
              { token.tag }
            </div>
          ];
        }

        return (
          <div
            className={ classNames.join(' ') }
            key={ tokenId }
          >
            <TokenImage token={ token } />
            { details }
          </div>
        );
      })
      .filter((node) => node);

    if (!body.length) {
      body = (
        <div className={ styles.empty }>
          <FormattedMessage
            id='ui.balance.none'
            defaultMessage='No balances associated with this account'
          />
        </div>
      );
    }

    return (
      <div
        className={
          [
            styles.balances,
            showOnlyEth
              ? ''
              : styles.full,
            className
          ].join(' ')
        }
      >
        { body }
      </div>
    );
  }
}

function mapStateToProps (state, props) {
  const { balances, tokens } = state;
  const { address } = props;

  return {
    balance: balances[address] || props.balance || {},
    tokens
  };
}

export default connect(mapStateToProps)(Balance);