feat(confix): allow customization of migration plan (#21202)

Co-authored-by: Hieu Vu <72878483+hieuvubk@users.noreply.github.com>
(cherry picked from commit 1d7f891)

# Conflicts:
#	tools/confix/migrations.go

Author: GAtom22

tools/confix/CHANGELOG.md

@@ -31,6 +31,10 @@ Ref: https://keepachangelog.com/en/1.0.0/
 
 ## [Unreleased]
 
+### Features
+
+* (confix) [#21202](https://github.com/cosmos/cosmos-sdk/pull/21202) Allow customization of migration `PlanBuilder`.
+
 ## [v0.1.1](https://github.com/cosmos/cosmos-sdk/releases/tag/tools/confix/v0.1.1) - 2023-12-11
 
 * [#18496](https://github.com/cosmos/cosmos-sdk/pull/18496) Remove invalid non SDK config from app.toml migration templates.

tools/confix/migrations.go

@@ -19,8 +19,13 @@ const (
 // MigrationMap defines a mapping from a version to a transformation plan.
 type MigrationMap map[string]func(from *tomledit.Document, to string) transform.Plan
 
+// loadDestConfigFile is the function signature to load the destination version
+// configuration toml file.
+type loadDestConfigFile func(to, planType string) (*tomledit.Document, error)
+
 var Migrations = MigrationMap{
 	"v0.45": NoPlan, // Confix supports only the current supported SDK version. So we do not support v0.44 -> v0.45.
+<<<<<<< HEAD
 	"v0.46": PlanBuilder,
 	"v0.47": PlanBuilder,
 	"v0.50": PlanBuilder,
@@ -33,6 +38,48 @@ func PlanBuilder(from *tomledit.Document, to string) transform.Plan {
 	deletedSections := map[string]bool{}
 
 	target, err := LoadLocalConfig(to)
+=======
+	"v0.46": defaultPlanBuilder,
+	"v0.47": defaultPlanBuilder,
+	"v0.50": defaultPlanBuilder,
+	"v0.52": defaultPlanBuilder,
+	"v2":    V2PlanBuilder,
+	// "v0.xx.x": defaultPlanBuilder, // add specific migration in case of configuration changes in minor versions
+}
+
+type v2KeyChangesMap map[string][]string
+
+// list all the keys which are need to be modified in v2
+var v2KeyChanges = v2KeyChangesMap{
+	"min-retain-blocks": []string{"comet.min-retain-blocks"},
+	"index-events":      []string{"comet.index-events"},
+	"halt-height":       []string{"comet.halt-height"},
+	"halt-time":         []string{"comet.halt-time"},
+	"app-db-backend":    []string{"store.app-db-backend"},
+	"pruning-keep-recent": []string{
+		"store.options.ss-pruning-option.keep-recent",
+		"store.options.sc-pruning-option.keep-recent",
+	},
+	"pruning-interval": []string{
+		"store.options.ss-pruning-option.interval",
+		"store.options.sc-pruning-option.interval",
+	},
+	"iavl-cache-size":       []string{"store.options.iavl-config.cache-size"},
+	"iavl-disable-fastnode": []string{"store.options.iavl-config.skip-fast-storage-upgrade"},
+	// Add other key mappings as needed
+}
+
+func defaultPlanBuilder(from *tomledit.Document, to, planType string) (transform.Plan, *tomledit.Document) {
+	return PlanBuilder(from, to, planType, LoadLocalConfig)
+}
+
+// PlanBuilder is a function that returns a transformation plan for a given diff between two files.
+func PlanBuilder(from *tomledit.Document, to, planType string, loadFn loadDestConfigFile) (transform.Plan, *tomledit.Document) {
+	plan := transform.Plan{}
+	deletedSections := map[string]bool{}
+
+	target, err := loadFn(to, planType)
+>>>>>>> 1d7f891ea (feat(confix): allow customization of migration plan (#21202))
 	if err != nil {
 		panic(fmt.Errorf("failed to parse file: %w. This file should have been valid", err))
 	}