aboutsummaryrefslogtreecommitdiffstats
path: root/packages/contracts/test/utils/token_registry_wrapper.ts
blob: 0abf20e03e732bb72db40dcf61203f45b4c6a133 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
import { Web3Wrapper } from '@0xproject/web3-wrapper';
import { Provider } from 'ethereum-types';

import { TokenRegistryContract } from '../../generated_contract_wrappers/token_registry';

import { Token } from './types';

import { constants } from './constants';

export class TokenRegWrapper {
    private _tokenReg: TokenRegistryContract;
    private _web3Wrapper: Web3Wrapper;
    constructor(tokenRegContract: TokenRegistryContract, provider: Provider) {
        this._tokenReg = tokenRegContract;
        this._web3Wrapper = new Web3Wrapper(provider);
    }
    public async addTokenAsync(token: Token, from: string): Promise<string> {
        const txHash = await this._tokenReg.addToken.sendTransactionAsync(
            token.address as string,
            token.name,
            token.symbol,
            token.decimals,
            token.ipfsHash,
            token.swarmHash,
            { from },
        );
        await this._web3Wrapper.awaitTransactionSuccessAsync(txHash, constants.AWAIT_TRANSACTION_MINED_MS);
        return txHash;
    }
    public async getTokenMetaDataAsync(tokenAddress: string): Promise<Token> {
        const data = await this._tokenReg.getTokenMetaData.callAsync(tokenAddress);
        const token: Token = {
            address: data[0],
            name: data[1],
            symbol: data[2],
            decimals: data[3],
            ipfsHash: data[4],
            swarmHash: data[5],
        };
        return token;
    }
    public async getTokenByNameAsync(tokenName: string): Promise<Token> {
        const data = await this._tokenReg.getTokenByName.callAsync(tokenName);
        const token: Token = {
            address: data[0],
            name: data[1],
            symbol: data[2],
            decimals: data[3],
            ipfsHash: data[4],
            swarmHash: data[5],
        };
        return token;
    }
    public async getTokenBySymbolAsync(tokenSymbol: string): Promise<Token> {
        const data = await this._tokenReg.getTokenBySymbol.callAsync(tokenSymbol);
        const token: Token = {
            address: data[0],
            name: data[1],
            symbol: data[2],
            decimals: data[3],
            ipfsHash: data[4],
            swarmHash: data[5],
        };
        return token;
    }
}