fix: couple fixes and tweaks

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

Author: tmigone

packages/hardhat-graph-protocol/package.json

@@ -18,7 +18,8 @@
   "scripts": {
     "build": "tsc",
     "lint": "eslint '**/*.{js,ts}' --fix",
-    "test": "mocha --exit --recursive 'test/**/*.test.ts'"
+    "test": "mocha --exit --recursive 'test/**/*.test.ts'",
+    "prepublishOnly": "npm run build"
   },
   "files": [
     "dist/",

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

@@ -4,8 +4,7 @@ import { GraphPluginError } from './sdk/utils/error'
 import { logDebug } from './logger'
 import { normalizePath } from './sdk/utils/path'
 
-import type { GraphDeployment } from './deployments'
-import type { GraphRuntimeEnvironmentOptions } from './type-extensions'
+import type { GraphDeployment, GraphRuntimeEnvironmentOptions } from './types'
 import type { HardhatRuntimeEnvironment } from 'hardhat/types'
 
 export function getAddressBookPath(

packages/hardhat-graph-protocol/src/deployments.ts renamed to packages/hardhat-graph-protocol/src/deployment-list.ts

@@ -1,22 +1,13 @@
 import type { GraphHorizonAddressBook, GraphHorizonContracts } from './sdk/deployments/horizon'
 
 // List of supported Graph deployments
-const GraphDeploymentsList = [
+export const GraphDeploymentsList = [
   'horizon',
 ] as const
 
-export type GraphDeployment = (typeof GraphDeploymentsList)[number]
-
 export type GraphDeploymentRuntimeEnvironmentMap = {
   horizon: {
     contracts: GraphHorizonContracts
     addressBook: GraphHorizonAddressBook
   }
 }
-
-export function isGraphDeployment(deployment: unknown): deployment is GraphDeployment {
-  return (
-    typeof deployment === 'string'
-    && GraphDeploymentsList.includes(deployment as GraphDeployment)
-  )
-}

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

@@ -1,14 +1,12 @@
 import path from 'path'
 
-import { assertGraphRuntimeEnvironment } from './type-extensions'
 import { getAddressBookPath } from './config'
 import { GraphHorizonAddressBook } from './sdk/deployments/horizon'
 import { HardhatEthersProvider } from '@nomicfoundation/hardhat-ethers/internal/hardhat-ethers-provider'
-import { isGraphDeployment } from './deployments'
 import { logDebug } from './logger'
 
+import { assertGraphRuntimeEnvironment, type GraphRuntimeEnvironmentOptions, isGraphDeployment } from './types'
 import type { HardhatConfig, HardhatRuntimeEnvironment, HardhatUserConfig } from 'hardhat/types'
-import type { GraphRuntimeEnvironmentOptions } from './type-extensions'
 
 export const greExtendConfig = (config: HardhatConfig, userConfig: Readonly<HardhatUserConfig>) => {
   const userPath = userConfig.paths?.graph

packages/hardhat-graph-protocol/src/sdk/address-book.ts

@@ -78,15 +78,14 @@ export abstract class AddressBook<
     if (!fs.existsSync(this.file)) throw new Error(`Address book path provided does not exist!`)
 
     // Load address book and validate its shape
+    // If it's empty, initialize it with an empty object
     const fileContents = JSON.parse(fs.readFileSync(this.file, 'utf8') || '{}')
+    if (!fileContents[this.chainId]) {
+      fileContents[this.chainId] = {} as Record<ContractName, AddressBookEntry>
+    }
     this.assertAddressBookJson(fileContents)
     this.addressBook = fileContents
     this._parseAddressBook()
-
-    // If the address book is empty for this chain id, initialize it with an empty object
-    if (!this.addressBook[this.chainId]) {
-      this.addressBook[this.chainId] = {} as Record<ContractName, AddressBookEntry>
-    }
   }
 
   /**
@@ -165,6 +164,9 @@ export abstract class AddressBook<
     signerOrProvider?: Signer | Provider,
   ): ContractList<ContractName> {
     const contracts = {} as ContractList<ContractName>
+    if (this.listEntries().length == 0) {
+      throw Error('No valid contracts found in address book')
+    }
     for (const contractName of this.listEntries()) {
       const artifactPath = typeof artifactsPath === 'object' && !Array.isArray(artifactsPath)
         ? artifactsPath[contractName]

packages/hardhat-graph-protocol/src/type-extensions.ts

@@ -1,102 +1,44 @@
 // To extend one of Hardhat's types, you need to import the module where it has been defined, and redeclare it.
 import 'hardhat/types/config'
 import 'hardhat/types/runtime'
-
-import type { GraphDeployment, GraphDeploymentRuntimeEnvironmentMap } from './deployments'
-import type { HardhatEthersProvider } from '@nomicfoundation/hardhat-ethers/internal/hardhat-ethers-provider'
-
-export type GraphRuntimeEnvironmentOptions = {
-  deployments?: {
-    [deployment in GraphDeployment]?: string | {
-      addressBook: string
-    }
-  }
-}
-
-export type GraphRuntimeEnvironment = {
-  [deployment in keyof GraphDeploymentRuntimeEnvironmentMap]?: GraphDeploymentRuntimeEnvironmentMap[deployment]
-} & {
-  provider: HardhatEthersProvider
-  chainId: number
-}
-
-export function assertGraphRuntimeEnvironment(
-  obj: unknown,
-): obj is GraphRuntimeEnvironment {
-  if (typeof obj !== 'object' || obj === null) return false
-
-  const deployments = obj as Partial<GraphDeploymentRuntimeEnvironmentMap>
-
-  for (const deployment in deployments) {
-    const environment = deployments[deployment as keyof GraphDeploymentRuntimeEnvironmentMap]
-    if (!environment || typeof environment !== 'object') {
-      return false
-    }
-  }
-
-  if (typeof (obj as GraphRuntimeEnvironment).provider !== 'object') {
-    return false
-  }
-
-  if (typeof (obj as GraphRuntimeEnvironment).chainId !== 'function') {
-    return false
-  }
-
-  return true
-}
+import type { GraphDeployments, GraphRuntimeEnvironment, GraphRuntimeEnvironmentOptions } from './types'
 
 declare module 'hardhat/types/runtime' {
-  export interface HardhatRuntimeEnvironment {
+  interface HardhatRuntimeEnvironment {
     graph: (opts?: GraphRuntimeEnvironmentOptions) => GraphRuntimeEnvironment
   }
 }
 
 declare module 'hardhat/types/config' {
-  export interface HardhatConfig {
+  interface HardhatConfig {
     graph: GraphRuntimeEnvironmentOptions
   }
 
-  export interface HardhatUserConfig {
+  interface HardhatUserConfig {
     graph: GraphRuntimeEnvironmentOptions
   }
 
-  export interface HardhatNetworkConfig {
-    deployments?: {
-      [deployment in GraphDeployment]?: string | {
-        addressBook: string
-      }
-    }
+  interface HardhatNetworkConfig {
+    deployments?: GraphDeployments
   }
 
-  export interface HardhatNetworkUserConfig {
-    deployments?: {
-      [deployment in GraphDeployment]?: string | {
-        addressBook: string
-      }
-    }
+  interface HardhatNetworkUserConfig {
+    deployments?: GraphDeployments
   }
 
-  export interface HttpNetworkConfig {
-    deployments?: {
-      [deployment in GraphDeployment]?: string | {
-        addressBook: string
-      }
-    }
+  interface HttpNetworkConfig {
+    deployments?: GraphDeployments
   }
 
-  export interface HttpNetworkUserConfig {
-    deployments?: {
-      [deployment in GraphDeployment]?: string | {
-        addressBook: string
-      }
-    }
+  interface HttpNetworkUserConfig {
+    deployments?: GraphDeployments
   }
 
-  export interface ProjectPathsConfig {
+  interface ProjectPathsConfig {
     graph?: string
   }
 
-  export interface ProjectPathsUserConfig {
+  interface ProjectPathsUserConfig {
     graph?: string
   }
 }

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

@@ -0,0 +1,57 @@
+import { type GraphDeploymentRuntimeEnvironmentMap, GraphDeploymentsList } from './deployment-list'
+import type { HardhatEthersProvider } from '@nomicfoundation/hardhat-ethers/internal/hardhat-ethers-provider'
+
+export type GraphDeployment = (typeof GraphDeploymentsList)[number]
+
+export type GraphDeployments = {
+  [deployment in GraphDeployment]?: string | {
+    addressBook: string
+  }
+}
+
+export type GraphRuntimeEnvironmentOptions = {
+  deployments?: {
+    [deployment in GraphDeployment]?: string | {
+      addressBook: string
+    }
+  }
+}
+
+export type GraphRuntimeEnvironment = {
+  [deployment in keyof GraphDeploymentRuntimeEnvironmentMap]?: GraphDeploymentRuntimeEnvironmentMap[deployment]
+} & {
+  provider: HardhatEthersProvider
+  chainId: number
+}
+
+export function isGraphDeployment(deployment: unknown): deployment is GraphDeployment {
+  return (
+    typeof deployment === 'string'
+    && GraphDeploymentsList.includes(deployment as GraphDeployment)
+  )
+}
+
+export function assertGraphRuntimeEnvironment(
+  obj: unknown,
+): obj is GraphRuntimeEnvironment {
+  if (typeof obj !== 'object' || obj === null) return false
+
+  const deployments = obj as Partial<GraphDeploymentRuntimeEnvironmentMap>
+
+  for (const deployment in deployments) {
+    const environment = deployments[deployment as keyof GraphDeploymentRuntimeEnvironmentMap]
+    if (!environment || typeof environment !== 'object') {
+      return false
+    }
+  }
+
+  if (typeof (obj as GraphRuntimeEnvironment).provider !== 'object') {
+    return false
+  }
+
+  if (typeof (obj as GraphRuntimeEnvironment).chainId !== 'function') {
+    return false
+  }
+
+  return true
+}

packages/hardhat-graph-protocol/tsconfig.json

@@ -4,6 +4,7 @@
     "module": "commonjs",
     "declaration": true,
     "declarationMap": true,
+    "sourceMap": true,
     "esModuleInterop": true,
     "forceConsistentCasingInFileNames": true,
     "strict": true,

packages/horizon/hardhat.config.ts

@@ -8,8 +8,6 @@ import 'hardhat-graph-protocol'
 
 import type { HardhatUserConfig } from 'hardhat/config'
 
-import 'tasks/test'
-
 const config: HardhatUserConfig = {
   solidity: {
     version: '0.8.27',

packages/horizon/package.json

@@ -55,7 +55,7 @@
     "solidity-coverage": "^0.8.0",
     "ts-node": ">=8.0.0",
     "typechain": "^8.3.0",
-    "typescript": "^5.3.3"
+    "typescript": "^5.6.3"
   },
   "lint-staged": {
     "contracts/**/*.sol": [