wire rollback command

Author: julienrbrt

evolve/template/constants.go

@@ -15,5 +15,9 @@ const (
 	EvNodePackage = "github.com/evstack/ev-node"
 	EvNodeVersion = "v1.0.0-beta.2.0.20250819153439-29b8ddf5a753"
 
+	GoDataStorePackageFork = "github.com/celestiaorg/go-datastore"
+	GoDataStoreVersionFork = "v0.0.0-20250801131506-48a63ae531e4"
+	GoDataStorePackage     = "github.com/ipfs/go-datastore"
+
 	EvNodeDaCmd = "github.com/evstack/ev-node/da/cmd/local-da"
 )

evolve/template/generator.go

@@ -30,6 +30,7 @@ func NewEvolveGenerator(chain *chain.Chain, withCometMigration bool) (*genny.Gen
 
 	g.RunFn(commandsStartModify(appPath, binaryName, chain.Version))
 	g.RunFn(commandsGenesisModify(appPath, binaryName))
+	g.RunFn(commandsRollbackModify(appPath, binaryName))
 	if withCometMigration {
 		g.RunFn(migrateFromCometModify(appPath))
 	}

evolve/template/init.go

@@ -128,6 +128,35 @@ func commandsGenesisModify(appPath, binaryName string) genny.RunFn {
 	}
 }
 
+// commandsRollbackModify modifies the application rollback command to use evolve.
+func commandsRollbackModify(appPath, binaryName string) genny.RunFn {
+	return func(r *genny.Runner) error {
+		cmdPath := filepath.Join(appPath, "cmd", binaryName, "cmd/commands.go")
+		f, err := r.Disk.Find(cmdPath)
+		if err != nil {
+			return err
+		}
+
+		// use ast to modify the function that initializes genesisCmd
+		content, err := xast.ModifyFunction(f.String(), "initRootCmd",
+			xast.AppendFuncCode(`
+				// override rollback command
+				evNodeRollbackCmd := abciserver.NewRollbackCmd(newApp, app.DefaultNodeHome)
+				if currentRollbackCmd, _, err := rootCmd.Find([]string{evNodeRollbackCmd.Name()}); err == nil{
+					rootCmd.RemoveCommand(currentRollbackCmd)
+				}
+				rootCmd.AddCommand(evNodeRollbackCmd)
+		    `,
+			),
+		)
+		if err != nil {
+			return err
+		}
+
+		return r.File(genny.NewFileS(cmdPath, content))
+	}
+}
+
 // updateDependencies makes sure the correct dependencies are added to the go.mod files.
 // ev-abci expects evolve v1 to be used.
 func updateDependencies(appPath string) error {
@@ -144,6 +173,9 @@ func updateDependencies(appPath string) error {
 		return errors.Errorf("failed to add local-da tool: %w", err)
 	}
 
+	// add required replaces
+	gomod.AddReplace(GoDataStorePackage, "", GoDataStorePackageFork, GoDataStoreVersionFork)
+
 	// save go.mod
 	data, err := gomod.Format()
 	if err != nil {