import BigNumber from 'bignumber.js'; import promisify = require('es6-promisify'); import * as _ from 'lodash'; import * as Web3 from 'web3'; import {Contract} from './contract'; import {Artifact, ArtifactContractName, TransactionReceipt, TxData, ZeroExError} from './types'; interface RawLogEntry { logIndex: string|null; transactionIndex: string|null; transactionHash: string; blockHash: string|null; blockNumber: string|null; address: string; data: string; topics: string[]; } const CONTRACT_NAME_TO_NOT_FOUND_ERROR: {[contractName: string]: ZeroExError} = { ZRX: ZeroExError.ZRXContractDoesNotExist, EtherToken: ZeroExError.EtherTokenContractDoesNotExist, Token: ZeroExError.TokenContractDoesNotExist, TokenRegistry: ZeroExError.TokenRegistryContractDoesNotExist, TokenTransferProxy: ZeroExError.TokenTransferProxyContractDoesNotExist, Exchange: ZeroExError.ExchangeContractDoesNotExist, }; export class Web3Wrapper { private web3: Web3; private networkId: number; private defaults: Partial; private jsonRpcRequestId: number; constructor(provider: Web3.Provider, networkId: number, defaults?: Partial) { if (_.isUndefined((provider as any).sendAsync)) { // Web3@1.0 provider doesn't support synchronous http requests, // so it only has an async `send` method, instead of a `send` and `sendAsync` in web3@0.x.x` // We re-assign the send method so that Web3@1.0 providers work with 0x.js (provider as any).sendAsync = (provider as any).send; } this.web3 = new Web3(); this.networkId = networkId; this.web3.setProvider(provider); this.defaults = defaults || {}; this.jsonRpcRequestId = 0; } public getContractDefaults(): Partial { return this.defaults; } public setProvider(provider: Web3.Provider, networkId: number) { this.networkId = networkId; this.web3.setProvider(provider); } public isAddress(address: string): boolean { return this.web3.isAddress(address); } public async isSenderAddressAvailableAsync(senderAddress: string): Promise { const addresses = await this.getAvailableAddressesAsync(); return _.includes(addresses, senderAddress); } public async getNodeVersionAsync(): Promise { const nodeVersion = await promisify(this.web3.version.getNode)(); return nodeVersion; } public async getTransactionReceiptAsync(txHash: string): Promise { const transactionReceipt = await promisify(this.web3.eth.getTransactionReceipt)(txHash); if (!_.isNull(transactionReceipt)) { transactionReceipt.status = this.normalizeTxReceiptStatus(transactionReceipt.status); } return transactionReceipt; } public getCurrentProvider(): Web3.Provider { return this.web3.currentProvider; } public getNetworkId(): number { return this.networkId; } public async getContractInstanceFromArtifactAsync( artifact: Artifact, address?: string, ): Promise { let contractAddress: string; if (_.isUndefined(address)) { const networkId = this.getNetworkId(); if (_.isUndefined(artifact.networks[networkId])) { throw new Error(ZeroExError.ContractNotDeployedOnNetwork); } contractAddress = artifact.networks[networkId].address.toLowerCase(); } else { contractAddress = address; } const doesContractExist = await this.doesContractExistAtAddressAsync(contractAddress); if (!doesContractExist) { throw new Error(CONTRACT_NAME_TO_NOT_FOUND_ERROR[artifact.contract_name]); } const contractInstance = this.getContractInstance( artifact.abi, contractAddress, ); return contractInstance; } public toWei(ethAmount: BigNumber): BigNumber { const balanceWei = this.web3.toWei(ethAmount, 'ether'); return balanceWei; } public async getBalanceInWeiAsync(owner: string): Promise { let balanceInWei = await promisify(this.web3.eth.getBalance)(owner); balanceInWei = new BigNumber(balanceInWei); return balanceInWei; } public async doesContractExistAtAddressAsync(address: string): Promise { const code = await promisify(this.web3.eth.getCode)(address); // Regex matches 0x0, 0x00, 0x in order to accommodate poorly implemented clients const codeIsEmpty = /^0x0{0,40}$/i.test(code); return !codeIsEmpty; } public async signTransactionAsync(address: string, message: string): Promise { const signData = await promisify(this.web3.eth.sign)(address, message); return signData; } public async getBlockNumberAsync(): Promise { const blockNumber = await promisify(this.web3.eth.getBlockNumber)(); return blockNumber; } public async getBlockAsync(blockParam: string|Web3.BlockParam): Promise { const block = await promisify(this.web3.eth.getBlock)(blockParam); return block; } public async getBlockTimestampAsync(blockParam: string|Web3.BlockParam): Promise { const {timestamp} = await this.getBlockAsync(blockParam); return timestamp; } public async getAvailableAddressesAsync(): Promise { const addresses: string[] = await promisify(this.web3.eth.getAccounts)(); return addresses; } public async getLogsAsync(filter: Web3.FilterObject): Promise { let fromBlock = filter.fromBlock; if (_.isNumber(fromBlock)) { fromBlock = this.web3.toHex(fromBlock); } let toBlock = filter.toBlock; if (_.isNumber(toBlock)) { toBlock = this.web3.toHex(toBlock); } const serializedFilter = { ...filter, fromBlock, toBlock, }; const payload = { jsonrpc: '2.0', id: this.jsonRpcRequestId++, method: 'eth_getLogs', params: [serializedFilter], }; const rawLogs = await this.sendRawPayloadAsync(payload); const formattedLogs = _.map(rawLogs, this.formatLog.bind(this)); return formattedLogs; } private getContractInstance(abi: Web3.ContractAbi, address: string): Web3.ContractInstance { const web3ContractInstance = this.web3.eth.contract(abi).at(address); return web3ContractInstance; } private async getNetworkAsync(): Promise { const networkId = await promisify(this.web3.version.getNetwork)(); return networkId; } private async sendRawPayloadAsync(payload: Web3.JSONRPCRequestPayload): Promise { const sendAsync = this.web3.currentProvider.sendAsync.bind(this.web3.currentProvider); const response = await promisify(sendAsync)(payload); const result = response.result; return result; } private normalizeTxReceiptStatus(status: undefined|null|string|0|1): null|0|1 { // Transaction status might have four values // undefined - Testrpc and other old clients // null - New clients on old transactions // number - Parity // hex - Geth if (_.isString(status)) { return this.web3.toDecimal(status) as 0|1; } else if (_.isUndefined(status)) { return null; } else { return status; } } private formatLog(rawLog: RawLogEntry): Web3.LogEntry { const formattedLog = { ...rawLog, logIndex: this.hexToDecimal(rawLog.logIndex), blockNumber: this.hexToDecimal(rawLog.blockNumber), transactionIndex: this.hexToDecimal(rawLog.transactionIndex), }; return formattedLog; } private hexToDecimal(hex: string|null): number|null { if (_.isNull(hex)) { return null; } const decimal = this.web3.toDecimal(hex); return decimal; } }