Use custom contract abstraction

1 files changed, 28 insertions, 0 deletions

diff --git a/src/contract.ts b/src/contract.ts
new file mode 100644
index 000000000..b4a54ca09
--- /dev/null
+++ b/src/contract.ts
@@ -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);
+            }
+        });
+    }
+}