96 lines
2.6 KiB
TypeScript
96 lines
2.6 KiB
TypeScript
import Big, { BigNumber } from 'bignumber.js';
|
|
import { Token } from 'config/data';
|
|
import { TransactionWithoutGas } from 'libs/messages';
|
|
import { Wei } from 'libs/units';
|
|
import { INode } from '../INode';
|
|
import RPCClient from './client';
|
|
import RPCRequests from './requests';
|
|
|
|
export default class RpcNode implements INode {
|
|
public client: RPCClient;
|
|
public requests: RPCRequests;
|
|
|
|
constructor(endpoint: string) {
|
|
this.client = new RPCClient(endpoint);
|
|
this.requests = new RPCRequests();
|
|
}
|
|
|
|
public getBalance(address: string): Promise<Wei> {
|
|
return this.client
|
|
.call(this.requests.getBalance(address))
|
|
.then(response => {
|
|
if (response.error) {
|
|
throw new Error(response.error.message);
|
|
}
|
|
return new Wei(String(response.result));
|
|
});
|
|
}
|
|
|
|
public estimateGas(transaction: TransactionWithoutGas): Promise<BigNumber> {
|
|
return this.client
|
|
.call(this.requests.estimateGas(transaction))
|
|
.then(response => {
|
|
if (response.error) {
|
|
throw new Error(response.error.message);
|
|
}
|
|
return new Big(String(response.result));
|
|
});
|
|
}
|
|
|
|
public getTokenBalance(address: string, token: Token): Promise<BigNumber> {
|
|
return this.client
|
|
.call(this.requests.getTokenBalance(address, token))
|
|
.then(response => {
|
|
if (response.error) {
|
|
// TODO - Error handling
|
|
return new Big(0);
|
|
}
|
|
return new Big(String(response.result)).div(
|
|
new Big(10).pow(token.decimal)
|
|
);
|
|
});
|
|
}
|
|
|
|
public getTokenBalances(
|
|
address: string,
|
|
tokens: Token[]
|
|
): Promise<BigNumber[]> {
|
|
return this.client
|
|
.batch(tokens.map(t => this.requests.getTokenBalance(address, t)))
|
|
.then(response => {
|
|
return response.map((item, idx) => {
|
|
// FIXME wrap in maybe-like
|
|
if (item.error) {
|
|
return new Big(0);
|
|
}
|
|
return new Big(String(item.result)).div(
|
|
new Big(10).pow(tokens[idx].decimal)
|
|
);
|
|
});
|
|
});
|
|
// TODO - Error handling
|
|
}
|
|
|
|
public getTransactionCount(address: string): Promise<string> {
|
|
return this.client
|
|
.call(this.requests.getTransactionCount(address))
|
|
.then(response => {
|
|
if (response.error) {
|
|
throw new Error(response.error.message);
|
|
}
|
|
return response.result;
|
|
});
|
|
}
|
|
|
|
public sendRawTx(signedTx: string): Promise<string> {
|
|
return this.client
|
|
.call(this.requests.sendRawTx(signedTx))
|
|
.then(response => {
|
|
if (response.error) {
|
|
throw new Error(response.error.message);
|
|
}
|
|
return response.result;
|
|
});
|
|
}
|
|
}
|