feat[contracts]: ChugSplash tooling to generate complete action bundles from config files

packages/contracts/src/chugsplash/actions.ts

@@ -1,4 +1,4 @@
-/* External Imports */
+/* Imports: External */
 import { fromHexString, toHexString } from '@eth-optimism/core-utils'
 import { ethers } from 'ethers'
 import MerkleTree from 'merkletreejs'

packages/contracts/src/chugsplash/config.ts

@@ -3,6 +3,7 @@ import * as Handlebars from 'handlebars'
 import { ethers } from 'ethers'
 
 type SolidityVariable =
+  | boolean
   | string
   | number
   | Array<SolidityVariable>

packages/contracts/src/chugsplash/hardhat-tools.ts

@@ -0,0 +1,56 @@
+/* Imports: External */
+import { HardhatRuntimeEnvironment } from 'hardhat/types'
+
+/* Imports: Internal */
+import { computeStorageSlots, getStorageLayout } from './storage'
+import { ChugSplashConfig, parseChugSplashConfig } from './config'
+import {
+  ChugSplashAction,
+  ChugSplashActionBundle,
+  getChugSplashActionBundle,
+} from './actions'
+
+/**
+ * Generates a ChugSplash action bundle from a config file.
+ * @param hre Hardhat runtime environment, used to load artifacts + storage layouts.
+ * @param config Config file to convert into a bundle.
+ * @param env Environment variables to inject into the config file.
+ * @returns Action bundle generated from the parsed config file.
+ */
+export const makeActionBundleFromConfig = async (
+  hre: HardhatRuntimeEnvironment,
+  config: ChugSplashConfig,
+  env: {
+    [key: string]: string | number | boolean
+  } = {}
+): Promise<ChugSplashActionBundle> => {
+  // Parse the config to replace any template variables.
+  const parsed = parseChugSplashConfig(config, env)
+
+  const actions: ChugSplashAction[] = []
+  for (const [contractName, contractConfig] of Object.entries(
+    parsed.contracts
+  )) {
+    const artifact = hre.artifacts.readArtifactSync(contractConfig.source)
+    const storageLayout = await getStorageLayout(hre, contractConfig.source)
+
+    // Add a SET_CODE action for each contract first.
+    actions.push({
+      target: contractConfig.address,
+      code: artifact.deployedBytecode,
+    })
+
+    // Add SET_STORAGE actions for each storage slot that we want to modify.
+    const slots = computeStorageSlots(storageLayout, contractConfig.variables)
+    for (const slot of slots) {
+      actions.push({
+        target: contractConfig.address,
+        key: slot.key,
+        value: slot.val,
+      })
+    }
+  }
+
+  // Generate a bundle from the list of actions.
+  return getChugSplashActionBundle(actions)
+}

packages/contracts/src/chugsplash/index.ts

@@ -1,3 +1,4 @@
 export * from './actions'
 export * from './config'
 export * from './storage'
+export * from './hardhat-tools'