abi: Move to protocol except for 'init'

Author: evaporei

src/commands/init.js

@@ -15,6 +15,7 @@ const { withSpinner, step } = require('../command-helpers/spinner')
 const { fixParameters } = require('../command-helpers/gluegun')
 const { chooseNodeUrl } = require('../command-helpers/node')
 const { abiEvents, generateScaffold, writeScaffold } = require('../scaffold')
+// TODO: Use Protocol class to getABI
 const ABI = require('../protocols/ethereum/abi')
 
 const networkChoices = [

src/compiler/index.js

@@ -9,7 +9,6 @@ const toolbox = require('gluegun/toolbox')
 const { step, withSpinner } = require('../command-helpers/spinner')
 const Subgraph = require('../subgraph')
 const Watcher = require('../watcher')
-const ABI = require('../protocols/ethereum/abi')
 const { applyMigrations } = require('../migrations')
 const asc = require('./asc')
 
@@ -53,6 +52,7 @@ class Compiler {
     this.globalsFile = path.join(globalsLib, globalsFile)
 
     this.protocol = this.options.protocol
+    this.ABI = this.protocol.getABI()
 
     process.on('uncaughtException', function(e) {
       toolbox.print.error(`UNCAUGHT EXCEPTION: ${e}`)
@@ -419,7 +419,7 @@ class Compiler {
                   abis.map(abi =>
                     abi.update('file', abiFile => {
                       abiFile = path.resolve(this.sourceDir, abiFile)
-                      let abiData = ABI.load(abi.get('name'), abiFile)
+                      let abiData = this.ABI.load(abi.get('name'), abiFile)
                       return path.relative(
                         this.options.outputDir,
                         this._writeSubgraphFile(
@@ -460,7 +460,7 @@ class Compiler {
                     abis.map(abi =>
                       abi.update('file', abiFile => {
                         abiFile = path.resolve(this.sourceDir, abiFile)
-                        let abiData = ABI.load(abi.get('name'), abiFile)
+                        let abiData = this.ABI.load(abi.get('name'), abiFile)
                         return path.relative(
                           this.options.outputDir,
                           this._writeSubgraphFile(

src/protocols/index.js

@@ -1,6 +1,7 @@
 const EthereumTypeGenerator = require('./ethereum/type-generator')
 const EthereumTemplateCodeGen = require('./ethereum/codegen/template')
 const NearTemplateCodeGen = require('./near/codegen/template')
+const EthereumABI = require('./ethereum/abi')
 
 module.exports = class Protocol {
   static fromDataSources(dataSourcesAndTemplates) {
@@ -45,4 +46,14 @@ module.exports = class Protocol {
         )
     }
   }
+
+  getABI() {
+    switch (this.name) {
+      case 'ethereum':
+      case 'ethereum/contract':
+        return EthereumABI
+      case 'near':
+        return null
+    }
+  }
 }

src/scaffold.js

@@ -6,6 +6,7 @@ const pkginfo = require('pkginfo')(module)
 const { getSubgraphBasename } = require('./command-helpers/subgraph')
 const { step } = require('./command-helpers/spinner')
 const { ascTypeForEthereum, valueTypeForAsc } = require('./codegen/types')
+// TODO: Use Protocol class to getABI
 const ABI = require('./protocols/ethereum/abi')
 const AbiCodeGenerator = require('./protocols/ethereum/codegen/abi')
 const util = require('./codegen/util')