init: Move package.json generation to Scaffold class

Author: evaporei

src/old-scaffold.js

@@ -1,49 +1,10 @@
 const fs = require('fs-extra')
 const path = require('path')
 const prettier = require('prettier')
-const pkginfo = require('pkginfo')(module)
 
-const { getSubgraphBasename } = require('./command-helpers/subgraph')
 const { step } = require('./command-helpers/spinner')
 const Scaffold = require('./scaffold')
 
-const graphCliVersion = process.env.GRAPH_CLI_TESTS
-  // JSON.stringify should remove this key, we will install the local
-  // graph-cli for the tests using `npm link` instead of fetching from npm.
-  ? undefined
-  // For scaffolding real subgraphs
-  : `${module.exports.version}`
-
-// package.json
-
-const generatePackageJson = ({ subgraphName, node }) =>
-  prettier.format(
-    JSON.stringify({
-      name: getSubgraphBasename(subgraphName),
-      license: 'UNLICENSED',
-      scripts: {
-        codegen: 'graph codegen',
-        build: 'graph build',
-        deploy:
-          `graph deploy ` +
-          `--node ${node} ` +
-          subgraphName,
-        'create-local': `graph create --node http://localhost:8020/ ${subgraphName}`,
-        'remove-local': `graph remove --node http://localhost:8020/ ${subgraphName}`,
-        'deploy-local':
-          `graph deploy ` +
-          `--node http://localhost:8020/ ` +
-          `--ipfs http://localhost:5001 ` +
-          subgraphName,
-      },
-      dependencies: {
-        '@graphprotocol/graph-cli': graphCliVersion,
-        '@graphprotocol/graph-ts': `0.24.1`,
-      },
-    }),
-    { parser: 'json' },
-  )
-
 const tsConfig = prettier.format(
   JSON.stringify({
     extends: '@graphprotocol/graph-ts/types/tsconfig.base.json',
@@ -67,8 +28,6 @@ const generateScaffold = async (
 ) => {
   step(spinner, 'Generate subgraph')
 
-  let packageJson = generatePackageJson({ subgraphName, node })
-
   let scaffold = new Scaffold({
     protocol: protocolInstance,
     abi,
@@ -77,8 +36,10 @@ const generateScaffold = async (
     network,
     contractName,
     subgraphName,
+    node,
   })
 
+  let packageJson = scaffold.generatePackageJson()
   let manifest = scaffold.generateManifest()
   let schema = scaffold.generateSchema()
   let mapping = scaffold.generateMapping()

src/scaffold/index.js

@@ -1,10 +1,20 @@
 const prettier = require('prettier')
+const pkginfo = require('pkginfo')(module)
+
+const GRAPH_CLI_VERSION = process.env.GRAPH_CLI_TESTS
+  // JSON.stringify should remove this key, we will install the local
+  // graph-cli for the tests using `npm link` instead of fetching from npm.
+  ? undefined
+  // For scaffolding real subgraphs
+  : `${module.exports.version}`
+
 const {
   abiEvents,
   generateEventType,
   generateExampleEntityType,
 } = require('./schema')
 const { generateEventIndexingHandlers } = require('./mapping')
+const { getSubgraphBasename } = require('../command-helpers/subgraph')
 
 module.exports = class Scaffold {
   constructor(options = {}) {
@@ -15,6 +25,36 @@ module.exports = class Scaffold {
     this.network = options.network
     this.contractName = options.contractName
     this.subgraphName = options.subgraphName
+    this.node = options.node
+  }
+
+  generatePackageJson() {
+    return prettier.format(
+      JSON.stringify({
+        name: getSubgraphBasename(this.subgraphName),
+        license: 'UNLICENSED',
+        scripts: {
+          codegen: 'graph codegen',
+          build: 'graph build',
+          deploy:
+            `graph deploy ` +
+            `--node ${this.node} ` +
+            this.subgraphName,
+          'create-local': `graph create --node http://localhost:8020/ ${this.subgraphName}`,
+          'remove-local': `graph remove --node http://localhost:8020/ ${this.subgraphName}`,
+          'deploy-local':
+            `graph deploy ` +
+            `--node http://localhost:8020/ ` +
+            `--ipfs http://localhost:5001 ` +
+            this.subgraphName,
+        },
+        dependencies: {
+          '@graphprotocol/graph-cli': GRAPH_CLI_VERSION,
+          '@graphprotocol/graph-ts': `0.24.1`,
+        },
+      }),
+      { parser: 'json' },
+    )
   }
 
   generateManifest() {