25 lines
738 B
TypeScript
25 lines
738 B
TypeScript
import * as _ from 'lodash';
|
|
import {Token, ZeroExError} from '../../src';
|
|
|
|
const PROTOCOL_TOKEN_SYMBOL = 'ZRX';
|
|
|
|
export class TokenUtils {
|
|
private tokens: Token[];
|
|
constructor(tokens: Token[]) {
|
|
this.tokens = tokens;
|
|
}
|
|
public getProtocolTokenOrThrow(): Token {
|
|
const zrxToken = _.find(this.tokens, {symbol: PROTOCOL_TOKEN_SYMBOL});
|
|
if (_.isUndefined(zrxToken)) {
|
|
throw new Error(ZeroExError.ZRX_NOT_IN_TOKEN_REGISTRY);
|
|
}
|
|
return zrxToken;
|
|
}
|
|
public getNonProtocolTokens(): Token[] {
|
|
const nonProtocolTokens = _.filter(this.tokens, token => {
|
|
return token.symbol !== PROTOCOL_TOKEN_SYMBOL;
|
|
});
|
|
return nonProtocolTokens;
|
|
}
|
|
}
|