Define AbiType

2 files changed, 10 insertions, 2 deletions

diff --git a/src/contract.ts b/src/contract.ts
index b9ab496b2..1aacc65dc 100644
--- a/src/contract.ts
+++ b/src/contract.ts
@@ -2,6 +2,7 @@ import * as Web3 from 'web3';
 import * as _ from 'lodash';
 import promisify = require('es6-promisify');
 import {SchemaValidator, schemas} from '0x-json-schemas';
+import {AbiType} from './types';
 
 export class Contract implements Web3.ContractInstance {
     public address: string;
@@ -22,7 +23,7 @@ export class Contract implements Web3.ContractInstance {
         this.validator = new SchemaValidator();
     }
     private populateFunctions(): void {
-        const functionsAbi = _.filter(this.abi, abiPart => abiPart.type === Web3.AbiType.Function);
+        const functionsAbi = _.filter(this.abi, abiPart => abiPart.type === AbiType.Function);
         _.forEach(functionsAbi, (functionAbi: Web3.MethodAbi) => {
             if (functionAbi.constant) {
                 const cbStyleCallFunction = this.contract[functionAbi.name].call;
@@ -40,7 +41,7 @@ export class Contract implements Web3.ContractInstance {
         });
     }
     private populateEvents(): void {
-        const eventsAbi = _.filter(this.abi, abiPart => abiPart.type === Web3.AbiType.Event);
+        const eventsAbi = _.filter(this.abi, abiPart => abiPart.type === AbiType.Event);
         _.forEach(eventsAbi, (eventAbi: Web3.EventAbi) => {
             this[eventAbi.name] = this.contract[eventAbi.name];
         });
diff --git a/src/types.ts b/src/types.ts
index e47f57e5e..ddf449c16 100644
--- a/src/types.ts
+++ b/src/types.ts
@@ -390,3 +390,10 @@ export interface ZeroExConfig {
 }
 
 export type TransactionReceipt = Web3.TransactionReceipt;
+
+export enum AbiType {
+    Function = 'function',
+    Constructor = 'constructor',
+    Event = 'event',
+    Fallback = 'fallback',
+}