import * as Bluebird from 'bluebird';
import * as _ from 'lodash';

import {Artifacts} from '../util/artifacts';
import {constants} from '../util/constants';
import {ContractInstance, Token} from '../util/types';

import {tokenInfo} from './config/token_info';
const {
    DummyToken,
    EtherToken,
    ZRXToken,
    TokenRegistry,
} = new Artifacts(artifacts);

module.exports = (deployer: any, network: string) => {
    const tokens = network === 'live' ? tokenInfo.live : tokenInfo.development;
    deployer.then(() => {
        return TokenRegistry.deployed();
    }).then((tokenRegistry: ContractInstance) => {
        if (network !== 'live') {
            const totalSupply = Math.pow(10, 18) * 1000000000;
            return Bluebird.each(tokens.map((token: Token) => DummyToken.new(
                token.name,
                token.symbol,
                token.decimals,
                totalSupply,
        )), _.noop).then((dummyTokens: ContractInstance[]) => {
                const weth = {
                    address: EtherToken.address,
                    name: 'Ether Token',
                    symbol: 'WETH',
                    url: '',
                    decimals: 18,
                    ipfsHash: constants.NULL_BYTES,
                    swarmHash: constants.NULL_BYTES,
                };
                return Bluebird.each(dummyTokens.map((tokenContract: ContractInstance, i: number) => {
                    const token = tokens[i];
                    return tokenRegistry.addToken(
                        tokenContract.address,
                        token.name,
                        token.symbol,
                        token.decimals,
                        token.ipfsHash,
                        token.swarmHash,
                    );
                }).concat(tokenRegistry.addToken(
                    weth.address,
                    weth.name,
                    weth.symbol,
                    weth.decimals,
                    weth.ipfsHash,
                    weth.swarmHash,
            )), _.noop);
            });
        } else {
            const zrx = {
                address: ZRXToken.address,
                name: '0x Protocol Token',
                symbol: 'ZRX',
                url: 'https://www.0xproject.com/',
                decimals: 18,
                ipfsHash: constants.NULL_BYTES,
                swarmHash: constants.NULL_BYTES,
            };
            return Bluebird.each(tokens.map((token: Token) => {
                return tokenRegistry.addToken(
                    token.address,
                    token.name,
                    token.symbol,
                    token.decimals,
                    token.ipfsHash,
                    token.swarmHash,
                );
            }).concat(tokenRegistry.addToken(
                zrx.address,
                zrx.name,
                zrx.symbol,
                zrx.decimals,
                zrx.ipfsHash,
                zrx.swarmHash,
        )), _.noop);
        }
    });
};