MyCrypto/common/containers/Tabs/SendTransaction/index.tsx

import { showNotification, TShowNotification } from 'actions/notifications';
import { broadcastTx, TBroadcastTx } from 'actions/wallet';
import Big from 'bignumber.js';
import { BalanceSidebar } from 'components';
// COMPONENTS
import { UnlockHeader } from 'components/ui';
import {
  donationAddressMap,
  NetworkConfig,
  NodeConfig,
  Token
} from 'config/data';
// CONFIG
import { TransactionWithoutGas } from 'libs/messages';
import { RPCNode } from 'libs/nodes';
import {
  BroadcastTransactionStatus,
  CompleteTransaction,
  formatTxInput,
  generateCompleteTransaction,
  getBalanceMinusGasCosts,
  TransactionInput
} from 'libs/transaction';
import { Ether, GWei, UnitKey, Wei } from 'libs/units';
import { isValidETHAddress } from 'libs/validators';
// LIBS
import { IWallet } from 'libs/wallet/IWallet';
import pickBy from 'lodash/pickBy';
import React from 'react';
// REDUX
import { connect } from 'react-redux';
import { AppState } from 'reducers';
import {
  getGasPriceGwei,
  getNetworkConfig,
  getNodeConfig,
  getNodeLib
} from 'selectors/config';
// SELECTORS
import {
  getTokenBalances,
  getTokens,
  getTxFromBroadcastTransactionStatus,
  MergedToken,
  TokenBalance
} from 'selectors/wallet';
import translate from 'translations';
// UTILS
import { formatGasLimit } from 'utils/formatters';
import {
  AddressField,
  AmountField,
  ConfirmationModal,
  CustomMessage,
  DataField,
  Donate,
  GasField
} from './components';
// MISC
import customMessages from './messages';

function getParam(query: { [key: string]: string }, key: string) {
  const keys = Object.keys(query);
  const index = keys.findIndex(k => k.toLowerCase() === key.toLowerCase());
  if (index === -1) {
    return null;
  }
  return query[keys[index]];
}

interface State {
  hasQueryString: boolean;
  readOnly: boolean;
  to: string;
  // amount value
  value: string;
  unit: UnitKey;
  token?: MergedToken | null;
  gasLimit: string;
  data: string;
  gasChanged: boolean;
  transaction: CompleteTransaction | null;
  showTxConfirm: boolean;
  generateDisabled: boolean;
}

interface Props {
  location: {
    query: {
      [key: string]: string;
    };
  };
  wallet: IWallet;
  balance: Ether;
  node: NodeConfig;
  nodeLib: RPCNode;
  network: NetworkConfig;
  tokens: MergedToken[];
  tokenBalances: TokenBalance[];
  gasPrice: Wei;
  transactions: BroadcastTransactionStatus[];
  showNotification: TShowNotification;
  broadcastTx: TBroadcastTx;
}

const initialState: State = {
  hasQueryString: false,
  readOnly: false,
  to: '',
  value: '',
  unit: 'ether',
  token: null,
  gasLimit: '21000',
  data: '',
  gasChanged: false,
  showTxConfirm: false,
  transaction: null,
  generateDisabled: true
};

export class SendTransaction extends React.Component<Props, State> {
  public state: State = initialState;

  public componentDidMount() {
    const queryPresets = pickBy(this.parseQuery());
    if (Object.keys(queryPresets).length) {
      this.setState({ ...queryPresets, hasQueryString: true });
    }
  }

  public componentDidUpdate(prevProps: Props, prevState: State) {
    // TODO listen to gas price changes here
    // TODO debounce the call
    if (
      // if gas has not changed
      !this.state.gasChanged &&
      // if we have valid tx
      this.isValid() &&
      // if any relevant fields changed
      (this.state.to !== prevState.to ||
        this.state.value !== prevState.value ||
        this.state.unit !== prevState.unit ||
        this.state.data !== prevState.data)
    ) {
      if (!isNaN(parseInt(this.state.value, 10))) {
        this.estimateGas();
      }
    }
    if (this.state.generateDisabled === this.isValid()) {
      this.setState({ generateDisabled: !this.isValid() });
    }
    const componentStateTransaction = this.state.transaction;
    if (componentStateTransaction) {
      // lives in redux state
      const currentTxAsSignedTransaction = getTxFromBroadcastTransactionStatus(
        this.props.transactions,
        componentStateTransaction.signedTx
      );
      // if there is a matching tx in redux state
      if (currentTxAsSignedTransaction) {
        // if the broad-casted transaction attempt is successful, clear the form
        if (currentTxAsSignedTransaction.successfullyBroadcast) {
          this.resetTx();
        }
      }
    }
  }

  public render() {
    const unlocked = !!this.props.wallet;
    const {
      to,
      value,
      unit,
      gasLimit,
      data,
      readOnly,
      hasQueryString,
      showTxConfirm,
      transaction
    } = this.state;
    const customMessage = customMessages.find(m => m.to === to);

    return (
      <section className="Tab-content">
        <UnlockHeader title={'NAV_SendEther'} />

        <div className="row">
          {/* Send Form */}
          {unlocked && (
            <main className="col-sm-8">
              <div className="Tab-content-pane">
                {hasQueryString && (
                  <div className="alert alert-info">
                    <p>{translate('WARN_Send_Link')}</p>
                  </div>
                )}

                <AddressField
                  placeholder={donationAddressMap.ETH}
                  value={this.state.to}
                  onChange={readOnly ? null : this.onAddressChange}
                />
                <AmountField
                  value={value}
                  unit={unit}
                  tokens={this.props.tokenBalances
                    .filter(token => !token.balance.eq(0))
                    .map(token => token.symbol)
                    .sort()}
                  onChange={readOnly ? void 0 : this.onAmountChange}
                />
                <GasField
                  value={gasLimit}
                  onChange={readOnly ? void 0 : this.onGasChange}
                />
                {unit === 'ether' && (
                  <DataField
                    value={data}
                    onChange={readOnly ? void 0 : this.onDataChange}
                  />
                )}
                <CustomMessage message={customMessage} />

                <div className="row form-group">
                  <div className="col-xs-12 clearfix">
                    <button
                      disabled={this.state.generateDisabled}
                      className="btn btn-info btn-block"
                      onClick={this.generateTxFromState}
                    >
                      {translate('SEND_generate')}
                    </button>
                  </div>
                </div>

                {transaction && (
                  <div>
                    <div className="row form-group">
                      <div className="col-sm-6">
                        <label>{translate('SEND_raw')}</label>
                        <textarea
                          className="form-control"
                          value={transaction.rawTx}
                          rows={4}
                          readOnly={true}
                        />
                      </div>
                      <div className="col-sm-6">
                        <label>{translate('SEND_signed')}</label>
                        <textarea
                          className="form-control"
                          value={transaction.signedTx}
                          rows={4}
                          readOnly={true}
                        />
                      </div>
                    </div>

                    <div className="form-group">
                      <button
                        className="btn btn-primary btn-block col-sm-11"
                        disabled={!this.state.transaction}
                        onClick={this.openTxModal}
                      >
                        {translate('SEND_trans')}
                      </button>
                    </div>
                  </div>
                )}
              </div>
            </main>
          )}

          {/* Sidebar */}
          {unlocked && (
            <section className="col-sm-4">
              <BalanceSidebar />
            </section>
          )}
        </div>

        {transaction &&
          showTxConfirm && (
            <ConfirmationModal
              wallet={this.props.wallet}
              node={this.props.node}
              signedTx={transaction.signedTx}
              onClose={this.hideConfirmTx}
              onConfirm={this.confirmTx}
            />
          )}
      </section>
    );
  }

  public parseQuery() {
    const query = this.props.location.query;
    const to = getParam(query, 'to');
    const data = getParam(query, 'data');
    // FIXME validate token against presets
    const unit = getParam(query, 'tokenSymbol');
    const value = getParam(query, 'value');
    let gasLimit = getParam(query, 'gas');
    if (gasLimit === null) {
      gasLimit = getParam(query, 'limit');
    }
    const readOnly = getParam(query, 'readOnly') != null;
    return { to, data, value, unit, gasLimit, readOnly };
  }

  public isValid() {
    const { to, value, gasLimit } = this.state;
    return (
      isValidETHAddress(to) &&
      value &&
      Number(value) > 0 &&
      !isNaN(Number(value)) &&
      isFinite(Number(value)) &&
      !isNaN(parseInt(gasLimit, 10)) &&
      isFinite(parseInt(gasLimit, 10))
    );
  }

  public async getFormattedTxFromState(): Promise<TransactionWithoutGas> {
    const { wallet } = this.props;
    const { token, unit, value, to, data } = this.state;
    const transactionInput: TransactionInput = {
      token,
      unit,
      value,
      to,
      data
    };
    return await formatTxInput(wallet, transactionInput);
  }

  public async estimateGas() {
    if (isNaN(parseInt(this.state.value, 10))) {
      return;
    }
    try {
      const cachedFormattedTx = await this.getFormattedTxFromState();
      // Grab a reference to state. If it has changed by the time the estimateGas
      // call comes back, we don't want to replace the gasLimit in state.
      const state = this.state;
      const gasLimit = await this.props.nodeLib.estimateGas(cachedFormattedTx);
      if (this.state === state) {
        this.setState({ gasLimit: formatGasLimit(gasLimit, state.unit) });
      } else {
        // state has changed, so try again from the start (with the hope that state won't change by the next time)
        this.estimateGas();
      }
    } catch (error) {
      this.setState({ generateDisabled: true });
      this.props.showNotification('danger', error.message, 5000);
    }
  }

  // FIXME use mkTx instead or something that could take care of default gas/data and whatnot,
  public onNewTx = (
    address: string,
    amount: string,
    unit: UnitKey,
    data: string = '',
    gasLimit: string = '21000'
  ) => {
    this.setState({
      to: address,
      value: amount,
      unit,
      data,
      gasLimit,
      gasChanged: false
    });
  };

  public onAddressChange = (value: string) => {
    this.setState({
      to: value
    });
  };

  public onDataChange = (value: string) => {
    if (this.state.unit === 'ether') {
      this.setState({ data: value });
    }
  };

  public onGasChange = (value: string) => {
    this.setState({ gasLimit: value, gasChanged: true });
  };

  public handleEverythingAmountChange = (
    value: string,
    unit: string
  ): string => {
    if (unit === 'ether') {
      const { balance, gasPrice } = this.props;
      const { gasLimit } = this.state;
      const weiBalance = balance.toWei();
      const bigGasLimit = new Big(gasLimit);
      value = getBalanceMinusGasCosts(
        bigGasLimit,
        gasPrice,
        weiBalance
      ).toString();
    } else {
      const tokenBalance = this.props.tokenBalances.find(
        tBalance => tBalance.symbol === unit
      );
      if (!tokenBalance) {
        throw new Error(`${unit}: not found in token balances;`);
      }
      value = tokenBalance.balance.toString();
    }
    return value;
  };

  public onAmountChange = (value: string, unit: UnitKey) => {
    if (value === 'everything') {
      value = this.handleEverythingAmountChange(value, unit);
    }
    let transaction = this.state.transaction;
    let generateDisabled = this.state.generateDisabled;
    if (unit && unit !== this.state.unit) {
      value = '';
      transaction = null;
      generateDisabled = true;
    }
    const token = this.props.tokens.find(x => x.symbol === unit);
    this.setState({
      value,
      unit,
      token,
      transaction,
      generateDisabled
    });
  };

  public resetJustTx = async (): Promise<any> =>
    new Promise(resolve =>
      this.setState(
        {
          transaction: null
        },
        resolve
      )
    );

  public generateTxFromState = async () => {
    await this.resetJustTx();
    const { nodeLib, wallet, gasPrice, network } = this.props;

    const { token, unit, value, to, data, gasLimit } = this.state;
    const chainId = network.chainId;
    const transactionInput = {
      token,
      unit,
      value,
      to,
      data
    };
    const bigGasLimit = new Big(gasLimit);
    try {
      const signedTx = await generateCompleteTransaction(
        wallet,
        nodeLib,
        gasPrice,
        bigGasLimit,
        chainId,
        transactionInput
      );
      this.setState({ transaction: signedTx });
    } catch (err) {
      this.props.showNotification('danger', err.message, 5000);
    }
  };

  public openTxModal = () => {
    this.setState({ showTxConfirm: true });
  };

  public hideConfirmTx = () => {
    this.setState({ showTxConfirm: false });
  };

  public resetTx = () => {
    this.setState({
      to: '',
      value: '',
      transaction: null
    });
  };

  public confirmTx = (signedTx: string) => {
    this.props.broadcastTx(signedTx);
  };
}

function mapStateToProps(state: AppState) {
  return {
    wallet: state.wallet.inst,
    balance: state.wallet.balance,
    tokenBalances: getTokenBalances(state),
    node: getNodeConfig(state),
    nodeLib: getNodeLib(state),
    network: getNetworkConfig(state),
    tokens: getTokens(state),
    gasPrice: new GWei(getGasPriceGwei(state)).toWei(),
    transactions: state.wallet.transactions
  };
}

export default connect(mapStateToProps, { showNotification, broadcastTx })(
  SendTransaction
);