feat: create address book if doesnt exist when running deploy task

Signed-off-by: Tomás Migone <[email protected]>

Author: tmigone

packages/hardhat-graph-protocol/src/config.ts

@@ -31,7 +31,12 @@ export function getAddressBookPath(
   logDebug(`Address book path: ${normalizedAddressBookPath}`)
 
   if (!fs.existsSync(normalizedAddressBookPath)) {
-    throw new GraphPluginError(`Address book not found: ${normalizedAddressBookPath}`)
+    if (opts.createAddressBook) {
+      logDebug(`Creating address book: ${normalizedAddressBookPath}`)
+      fs.writeFileSync(normalizedAddressBookPath, '{}')
+    } else {
+      throw new GraphPluginError(`Address book not found: ${normalizedAddressBookPath}`)
+    }
   }
 
   return normalizedAddressBookPath

packages/hardhat-graph-protocol/src/gre.ts

@@ -30,9 +30,17 @@ export const greExtendConfig = (config: HardhatConfig, userConfig: Readonly<Hard
 }
 
 export const greExtendEnvironment = (hre: HardhatRuntimeEnvironment) => {
-  hre.graph = lazyFunction(() => (opts: GraphRuntimeEnvironmentOptions = { deployments: {} }) => {
+  hre.graph = lazyFunction(() => (opts?: GraphRuntimeEnvironmentOptions) => {
     logDebug('*** Initializing Graph Runtime Environment (GRE) ***')
     logDebug(`Main network: ${hre.network.name}`)
+
+    if (opts === undefined) {
+      opts = {
+        deployments: {},
+        createAddressBook: false,
+      }
+    }
+
     const chainId = hre.network.config.chainId
     if (chainId === undefined) {
       throw new GraphPluginError('Please define chainId in your Hardhat network configuration')

packages/hardhat-graph-protocol/src/types.ts

@@ -9,6 +9,7 @@ export type GraphDeploymentOptions = {
 
 export type GraphRuntimeEnvironmentOptions = {
   deployments?: GraphDeploymentOptions
+  createAddressBook?: boolean
 }
 
 export type GraphRuntimeEnvironment = GraphDeployments & {

packages/horizon/tasks/deploy.ts

@@ -17,7 +17,7 @@ task('deploy:protocol', 'Deploy a new version of the Graph Protocol Horizon cont
   )
   .addOptionalParam('accountIndex', 'Derivation path index for the account to use', 0, types.int)
   .setAction(async (args, hre: HardhatRuntimeEnvironment) => {
-    const graph = hre.graph()
+    const graph = hre.graph({ createAddressBook: true })
 
     // Load configuration for the deployment
     console.log('\n========== ⚙️ Deployment configuration ==========')

packages/subgraph-service/tasks/deploy.ts

@@ -30,7 +30,7 @@ task('deploy:protocol', 'Deploy a new version of the Graph Protocol Horizon cont
     types.string,
   )
   .setAction(async (args, hre: HardhatRuntimeEnvironment) => {
-    const graph = hre.graph()
+    const graph = hre.graph({ createAddressBook: true })
 
     // Load configuration files for the deployment
     console.log('\n========== ⚙️ Deployment configuration ==========')