op-node: super experimental interop genesis CLI

op-node/cmd/interop/interop.go

@@ -0,0 +1,184 @@
+package interop
+
+import (
+	"encoding/json"
+	"fmt"
+	"os"
+	"path/filepath"
+	"strings"
+	"time"
+
+	"github.com/urfave/cli/v2"
+
+	"github.com/ethereum-optimism/optimism/op-chain-ops/devkeys"
+	"github.com/ethereum-optimism/optimism/op-chain-ops/foundry"
+	"github.com/ethereum-optimism/optimism/op-chain-ops/interopgen"
+	op_service "github.com/ethereum-optimism/optimism/op-service"
+	"github.com/ethereum-optimism/optimism/op-service/cliapp"
+	oplog "github.com/ethereum-optimism/optimism/op-service/log"
+)
+
+var EnvPrefix = "OP_INTEROP"
+
+var (
+	l1ChainIDFlag = &cli.Uint64Flag{
+		Name:    "l1.chainid",
+		Value:   900100,
+		EnvVars: op_service.PrefixEnvVar(EnvPrefix, "L1_CHAINID"),
+	}
+	l2ChainIDsFlag = &cli.Uint64SliceFlag{
+		Name:    "l2.chainids",
+		Value:   cli.NewUint64Slice(900200, 900201),
+		EnvVars: op_service.PrefixEnvVar(EnvPrefix, "L2_CHAINIDS"),
+	}
+	timestampFlag = &cli.Uint64Flag{
+		Name:    "timestamp",
+		Value:   0,
+		EnvVars: op_service.PrefixEnvVar(EnvPrefix, "TIMESTAMP"),
+		Usage:   "Will use current timestamp, plus 5 seconds, if not set",
+	}
+	mnemonicFlag = &cli.StringFlag{
+		Name:    "mnemonic",
+		Value:   devkeys.TestMnemonic,
+		EnvVars: op_service.PrefixEnvVar(EnvPrefix, "MNEMONIC"),
+	}
+	artifactsDirFlag = &cli.StringFlag{
+		Name:    "artifacts-dir",
+		Value:   "packages/contracts-bedrock/forge-artifacts",
+		EnvVars: op_service.PrefixEnvVar(EnvPrefix, "ARTIFACTS_DIR"),
+	}
+	foundryDirFlag = &cli.StringFlag{
+		Name:    "foundry-dir",
+		Value:   "packages/contracts-bedrock",
+		EnvVars: op_service.PrefixEnvVar(EnvPrefix, "FOUNDRY_DIR"),
+		Usage:   "Optional, for source-map info during genesis generation",
+	}
+	outDirFlag = &cli.StringFlag{
+		Name:    "out-dir",
+		Value:   ".interop-devnet",
+		EnvVars: op_service.PrefixEnvVar(EnvPrefix, "OUT_DIR"),
+	}
+)
+
+var InteropDevSetup = &cli.Command{
+	Name:  "dev-setup",
+	Usage: "Generate devnet genesis configs with one L1 and multiple L2s",
+	Flags: cliapp.ProtectFlags(append([]cli.Flag{
+		l1ChainIDFlag,
+		l2ChainIDsFlag,
+		timestampFlag,
+		mnemonicFlag,
+		artifactsDirFlag,
+		foundryDirFlag,
+	}, oplog.CLIFlags(EnvPrefix)...)),
+	Action: func(cliCtx *cli.Context) error {
+		logCfg := oplog.ReadCLIConfig(cliCtx)
+		logger := oplog.NewLogger(cliCtx.App.Writer, logCfg)
+
+		recipe := &interopgen.InteropDevRecipe{
+			L1ChainID:        cliCtx.Uint64(l1ChainIDFlag.Name),
+			L2ChainIDs:       cliCtx.Uint64Slice(l2ChainIDsFlag.Name),
+			GenesisTimestamp: cliCtx.Uint64(timestampFlag.Name),
+		}
+		if recipe.GenesisTimestamp == 0 {
+			recipe.GenesisTimestamp = uint64(time.Now().Unix() + 5)
+		}
+		mnemonic := strings.TrimSpace(cliCtx.String(mnemonicFlag.Name))
+		if mnemonic == devkeys.TestMnemonic {
+			logger.Warn("Using default test mnemonic!")
+		}
+		keys, err := devkeys.NewMnemonicDevKeys(mnemonic)
+		if err != nil {
+			return fmt.Errorf("failed to setup dev keys from mnemonic: %w", err)
+		}
+		worldCfg, err := recipe.Build(keys)
+		if err != nil {
+			return fmt.Errorf("failed to build deploy configs from interop recipe: %w", err)
+		}
+		if err := worldCfg.Check(logger); err != nil {
+			return fmt.Errorf("invalid deploy configs: %w", err)
+		}
+		artifactsDir := cliCtx.String(artifactsDirFlag.Name)
+		af := foundry.OpenArtifactsDir(artifactsDir)
+		var srcFs *foundry.SourceMapFS
+		if cliCtx.IsSet(foundryDirFlag.Name) {
+			srcDir := cliCtx.String(foundryDirFlag.Name)
+			srcFs = foundry.NewSourceMapFS(os.DirFS(srcDir))
+		}
+		worldDeployment, worldOutput, err := interopgen.Deploy(logger, af, srcFs, worldCfg)
+		if err != nil {
+			return fmt.Errorf("failed to deploy interop dev setup: %w", err)
+		}
+		outDir := cliCtx.String(outDirFlag.Name)
+		if err := os.MkdirAll(outDir, 0644); err != nil {
+			return fmt.Errorf("failed to create output dir: %w", err)
+		}
+		// Write deployments
+		{
+			deploymentsDir := filepath.Join(outDir, "deployments")
+			if err := os.MkdirAll(deploymentsDir, 0644); err != nil {
+				return fmt.Errorf("failed to create deployments output dir: %w", err)
+			}
+			if err := writeJson(filepath.Join(deploymentsDir, "l1.json"), worldDeployment.L1); err != nil {
+				return fmt.Errorf("failed to write L1 deployment data: %w", err)
+			}
+			if err := writeJson(filepath.Join(deploymentsDir, "superchain.json"), worldDeployment.Superchain); err != nil {
+				return fmt.Errorf("failed to write Superchain deployment data: %w", err)
+			}
+			l2DeploymentsDir := filepath.Join(deploymentsDir, "l2")
+			if err := os.MkdirAll(l2DeploymentsDir, 0644); err != nil {
+				return fmt.Errorf("failed to create deployments output dir: %w", err)
+			}
+			for id, dep := range worldDeployment.L2s {
+				if err := writeJson(filepath.Join(l2DeploymentsDir, id+".json"), dep); err != nil {
+					return fmt.Errorf("failed to write L2 %s deployment data: %w", id, err)
+				}
+			}
+		}
+		// write configs
+		{
+			configsDir := filepath.Join(outDir, "configs")
+			if err := os.MkdirAll(configsDir, 0644); err != nil {
+				return fmt.Errorf("failed to create deployments output dir: %w", err)
+			}
+			if err := writeJson(filepath.Join(configsDir, "l1-genesis.json"), worldOutput.L1); err != nil {
+				return fmt.Errorf("failed to write L1 deployment data: %w", err)
+			}
+			l2ConfigsDir := filepath.Join(configsDir, "l2")
+			if err := os.MkdirAll(l2ConfigsDir, 0644); err != nil {
+				return fmt.Errorf("failed to create deployments output dir: %w", err)
+			}
+			for id, dep := range worldOutput.L2s {
+				if err := writeJson(filepath.Join(l2ConfigsDir, id+"-genesis.json"), dep.Genesis); err != nil {
+					return fmt.Errorf("failed to write L2 %s genesis config: %w", id, err)
+				}
+				if err := writeJson(filepath.Join(l2ConfigsDir, id+"-rollup.json"), dep.Genesis); err != nil {
+					return fmt.Errorf("failed to write L2 %s rollup config: %w", id, err)
+				}
+			}
+		}
+		return nil
+	},
+}
+
+func writeJson(path string, content any) error {
+	f, err := os.OpenFile(path, os.O_WRONLY|os.O_TRUNC|os.O_CREATE, 0644)
+	if err != nil {
+		return fmt.Errorf("failed to open %q: %w", path, err)
+	}
+	defer f.Close()
+	enc := json.NewEncoder(f)
+	enc.SetEscapeHTML(false)
+	if err := enc.Encode(content); err != nil {
+		return fmt.Errorf("failed to write JSON content: %w", err)
+	}
+	return nil
+}
+
+var InteropCmd = &cli.Command{
+	Name:  "interop",
+	Usage: "Experimental tools for OP-Stack interop networks.",
+	Subcommands: cli.Commands{
+		InteropDevSetup,
+	},
+}

op-node/cmd/main.go

@@ -3,6 +3,7 @@ package main
 import (
 	"context"
 	"fmt"
+	"github.com/ethereum-optimism/optimism/op-node/cmd/interop"
 	"os"
 
 	"github.com/urfave/cli/v2"
@@ -62,6 +63,7 @@ func main() {
 			Name:        "networks",
 			Subcommands: networks.Subcommands,
 		},
+		interop.InteropCmd,
 	}
 
 	ctx := opio.WithInterruptBlocker(context.Background())