Use custom contract abstraction
This commit is contained in:
28
src/contract.ts
Normal file
28
src/contract.ts
Normal file
@@ -0,0 +1,28 @@
|
||||
import * as Web3 from 'web3';
|
||||
import * as _ from 'lodash';
|
||||
import promisify = require('es6-promisify');
|
||||
|
||||
export class Contract<A extends Web3.ContractInstance> implements Web3.ContractInstance {
|
||||
public address: string;
|
||||
public abi: Web3.ContractAbi;
|
||||
private contract: A;
|
||||
[name: string]: any;
|
||||
constructor(web3ContractInstance: A) {
|
||||
this.contract = web3ContractInstance;
|
||||
this.address = web3ContractInstance.address;
|
||||
this.abi = web3ContractInstance.abi;
|
||||
const functionsAbi = _.filter(this.abi, abiPart => abiPart.type === 'function');
|
||||
_.forEach(functionsAbi, (functionAbi: Web3.MethodAbi) => {
|
||||
const cbStyleFunction = web3ContractInstance[functionAbi.name];
|
||||
this[functionAbi.name] = promisify(cbStyleFunction, web3ContractInstance);
|
||||
if (functionAbi.constant) {
|
||||
const cbStyleCallFunction = web3ContractInstance[functionAbi.name].call;
|
||||
this[functionAbi.name].call = promisify(cbStyleCallFunction, web3ContractInstance);
|
||||
} else {
|
||||
const cbStyleEstimateGasFunction = web3ContractInstance[functionAbi.name].estimateGas;
|
||||
this[functionAbi.name].estimateGas =
|
||||
promisify(cbStyleEstimateGasFunction, web3ContractInstance);
|
||||
}
|
||||
});
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user