do not merge: sample genesis migration command to run migration to add support for base denoms with slashes

go.mod

@@ -12,6 +12,7 @@ require (
 	github.com/golang/protobuf v1.5.2
 	github.com/gorilla/mux v1.8.0
 	github.com/grpc-ecosystem/grpc-gateway v1.16.0
+	github.com/pkg/errors v0.9.1
 	github.com/rakyll/statik v0.1.7
 	github.com/regen-network/cosmos-proto v0.3.1
 	github.com/spf13/cast v1.5.0
@@ -90,7 +91,6 @@ require (
 	github.com/pelletier/go-toml v1.9.5 // indirect
 	github.com/pelletier/go-toml/v2 v2.0.1 // indirect
 	github.com/petermattis/goid v0.0.0-20180202154549-b0b1615b78e5 // indirect
-	github.com/pkg/errors v0.9.1 // indirect
 	github.com/pmezard/go-difflib v1.0.0 // indirect
 	github.com/prometheus/client_golang v1.12.1 // indirect
 	github.com/prometheus/client_model v0.2.0 // indirect

testing/simapp/migrate.go

@@ -0,0 +1,112 @@
+package simapp
+
+import (
+	"encoding/json"
+	"fmt"
+	"io/ioutil"
+
+	"github.com/cosmos/cosmos-sdk/client"
+	sdk "github.com/cosmos/cosmos-sdk/types"
+	"github.com/cosmos/cosmos-sdk/version"
+	"github.com/cosmos/cosmos-sdk/x/genutil/types"
+	"github.com/pkg/errors"
+	"github.com/spf13/cobra"
+	tmjson "github.com/tendermint/tendermint/libs/json"
+	tmtypes "github.com/tendermint/tendermint/types"
+
+	ibctransfertypes "github.com/cosmos/ibc-go/v3/modules/apps/transfer/types"
+)
+
+// MigrateGenesisCmd returns a command to execute genesis state migration.
+func MigrateGenesisCmd() *cobra.Command {
+	cmd := &cobra.Command{
+		Use:   "migrate [genesis-file]",
+		Short: "Migrate genesis",
+		Long: fmt.Sprintf(`Migrate the source genesis and print to STDOUT.
+
+Example:
+$ %s migrate /path/to/genesis.json
+`, version.AppName),
+		Args: cobra.ExactArgs(1),
+		RunE: func(cmd *cobra.Command, args []string) error {
+			clientCtx := client.GetClientContextFromCmd(cmd)
+
+			var err error
+
+			importGenesis := args[0]
+
+			jsonBlob, err := ioutil.ReadFile(importGenesis)
+
+			if err != nil {
+				return errors.Wrap(err, "failed to read provided genesis file")
+			}
+
+			genDoc, err := tmtypes.GenesisDocFromJSON(jsonBlob)
+			if err != nil {
+				return errors.Wrapf(err, "failed to read genesis document from file %s", importGenesis)
+			}
+
+			var initialState types.AppMap
+			if err := json.Unmarshal(genDoc.AppState, &initialState); err != nil {
+				return errors.Wrap(err, "failed to JSON unmarshal initial genesis state")
+			}
+
+			newGenState, err := migrateGenesisSlashedDenomsUpgrade(initialState, clientCtx, genDoc)
+			if err != nil {
+				return errors.Wrap(err, "failed to migrate")
+			}
+
+			genDoc.AppState, err = json.Marshal(newGenState)
+			if err != nil {
+				return errors.Wrap(err, "failed to JSON marshal migrated genesis state")
+			}
+
+			bz, err := tmjson.Marshal(genDoc)
+			if err != nil {
+				return errors.Wrap(err, "failed to marshal genesis doc")
+			}
+
+			sortedBz, err := sdk.SortJSON(bz)
+			if err != nil {
+				return errors.Wrap(err, "failed to sort JSON genesis doc")
+			}
+
+			fmt.Println(string(sortedBz))
+			return nil
+		},
+	}
+
+	return cmd
+}
+
+// migrateGenesisSlashedDenomsUpgrade corrects any incorrect trace information
+// from previously received coins that had slashes in the base denom.
+func migrateGenesisSlashedDenomsUpgrade(appState types.AppMap, clientCtx client.Context, genDoc *tmtypes.GenesisDoc) (types.AppMap, error) {
+	if appState[ibctransfertypes.ModuleName] != nil {
+		transferGenState := &ibctransfertypes.GenesisState{}
+		clientCtx.Codec.MustUnmarshalJSON(appState[ibctransfertypes.ModuleName], transferGenState)
+
+		substituteTraces := make([]ibctransfertypes.DenomTrace, len(transferGenState.DenomTraces))
+		for i, dt := range transferGenState.DenomTraces {
+			// replace all previous traces with the latest trace if validation passes
+			// note most traces will have same value
+			newTrace := ibctransfertypes.ParseDenomTrace(dt.GetFullDenomPath())
+
+			if err := newTrace.Validate(); err != nil {
+				substituteTraces[i] = dt
+			} else {
+				substituteTraces[i] = newTrace
+			}
+		}
+
+		transferGenState.DenomTraces = substituteTraces
+
+		// delete old genesis state
+		delete(appState, ibctransfertypes.ModuleName)
+
+		// set new ibc transfer genesis state
+		appState[ibctransfertypes.ModuleName] = clientCtx.Codec.MustMarshalJSON(transferGenState)
+	}
+
+	return appState, nil
+}

testing/simapp/simd/cmd/root.go

@@ -144,7 +144,7 @@ func initRootCmd(rootCmd *cobra.Command, encodingConfig params.EncodingConfig) {
 	rootCmd.AddCommand(
 		genutilcli.InitCmd(simapp.ModuleBasics, simapp.DefaultNodeHome),
 		genutilcli.CollectGenTxsCmd(banktypes.GenesisBalancesIterator{}, simapp.DefaultNodeHome),
-		genutilcli.MigrateGenesisCmd(),
+		simapp.MigrateGenesisCmd(),
 		genutilcli.GenTxCmd(simapp.ModuleBasics, encodingConfig.TxConfig, banktypes.GenesisBalancesIterator{}, simapp.DefaultNodeHome),
 		genutilcli.ValidateGenesisCmd(simapp.ModuleBasics),
 		AddGenesisAccountCmd(simapp.DefaultNodeHome),