blob: 69bad2bac87b4383dd52fcae25713a4ac400391a (
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
|
import * as _ from 'lodash';
import {Web3Wrapper} from '../web3_wrapper';
import {Token, TokenRegistryContract, TokenMetadata} from '../types';
import {assert} from '../utils/assert';
import {ContractWrapper} from './contract_wrapper';
import * as TokenRegistryArtifacts from '../artifacts/TokenRegistry.json';
export class TokenRegistryWrapper extends ContractWrapper {
private tokenRegistryContractIfExists: TokenRegistryContract;
constructor(web3Wrapper: Web3Wrapper) {
super(web3Wrapper);
}
public invalidateContractInstance() {
delete this.tokenRegistryContractIfExists;
}
public async getTokensAsync(): Promise<Token[]> {
await this.instantiateTokenRegistryContractIfDoesntExistAsync();
const addresses = await this.tokenRegistryContractIfExists.getTokenAddresses.call();
const tokenMetadataPromises: Array<Promise<TokenMetadata>> = _.map(
addresses,
(address: string) => (this.tokenRegistryContractIfExists.getTokenMetaData.call(address)),
);
const tokensMetadata = await Promise.all(tokenMetadataPromises);
const tokens = _.map(tokensMetadata, metadata => {
return {
address: metadata[0],
name: metadata[1],
symbol: metadata[2],
url: metadata[3],
decimals: metadata[4].toNumber(),
};
});
return tokens;
}
private async instantiateTokenRegistryContractIfDoesntExistAsync() {
if (!_.isUndefined(this.tokenRegistryContractIfExists)) {
return;
}
const contractInstance = await this.instantiateContractIfExistsAsync((TokenRegistryArtifacts as any));
this.tokenRegistryContractIfExists = contractInstance as TokenRegistryContract;
}
}
|