Migrate genesis cmd

.pending/features/sdk/4409-migration-scrip

@@ -0,0 +1 @@
+#4409 Implement an export tool to migrate exported state from v0.34.x to v0.36

contrib/export/v0.33.x-to-v0.34.0.py → contrib/migrate/v0.33.x-to-v0.34.0.py

@@ -5,8 +5,9 @@
 
 def process_raw_genesis(genesis, parsed_args):
     # update genesis with breaking changes
-    genesis['consensus_params']['block'] = genesis['consensus_params']['block_size']
-    del genesis['consensus_params']['block_size']
+    if 'block_size' in genesis['consensus_params']:
+        genesis['consensus_params']['block'] = genesis['consensus_params']['block_size']
+        del genesis['consensus_params']['block_size']
 
     genesis['app_state']['crisis'] = {
         'constant_fee': {

docs/clients/README.md

@@ -16,3 +16,11 @@ Light-clients enable users to interact with your application without having to d
 
 - [Command-Line interface for SDK-based blockchain](./cli.md)
 - [Service provider doc](./service-providers.md)
+
+## Genesis upgrade
+
+If you need to upgrade your node you could export the genesis and migrate it to the new version through this script:
+
+```bash
+<appbinary> migrate v0.36 genesis_0_34.json [--time "2019-04-22T17:00:11Z"] [--chain-id test] > ~/.gaiad/genesis.json 
+```

x/genutil/client/cli/migrate.go

@@ -0,0 +1,87 @@
+package cli
+
+import (
+	"fmt"
+	"github.com/cosmos/cosmos-sdk/server"
+	"strings"
+	"time"
+
+	"github.com/spf13/cobra"
+	"github.com/tendermint/tendermint/types"
+
+	"github.com/cosmos/cosmos-sdk/codec"
+	extypes "github.com/cosmos/cosmos-sdk/x/genutil"
+	"github.com/cosmos/cosmos-sdk/x/genutil/legacy/v036"
+)
+
+var migrationMap = extypes.MigrationMap{
+	"v0.36": v036.Migrate,
+}
+
+const (
+	flagGenesisTime = "genesis-time"
+	flagChainId     = "chain-id"
+)
+
+func MigrateGenesisCmd(_ *server.Context, cdc *codec.Codec) *cobra.Command {
+	cmd := &cobra.Command{
+		Use:   "migrate [target-version] [genesis-file]",
+		Short: "Migrate genesis to a specified target version",
+		Long: strings.TrimSpace(`Migrate the source genesis into the target version and print to STDOUT.
+
+Example:
+[binary] migrate v0.36 /path/to/genesis.json --chain-id=cosmoshub-3 --genesis-time=2019-04-22T17:00:00Z
+`),
+		Args: cobra.ExactArgs(2),
+		RunE: func(cmd *cobra.Command, args []string) error {
+			target := args[0]
+			importGenesis := args[1]
+
+			genDoc, err := types.GenesisDocFromFile(importGenesis)
+			if err != nil {
+				return err
+			}
+
+			var initialState extypes.AppMap
+			cdc.MustUnmarshalJSON(genDoc.AppState, &initialState)
+
+			if migrationMap[target] == nil {
+				return fmt.Errorf("unknown migration function version: %s", target)
+			}
+
+			newGenState := migrationMap[target](initialState, cdc)
+			genDoc.AppState = cdc.MustMarshalJSON(newGenState)
+
+			genesisTime := cmd.Flag(flagGenesisTime).Value.String()
+			if genesisTime != "" {
+				var t time.Time
+
+				err := t.UnmarshalText([]byte(genesisTime))
+				if err != nil {
+					return err
+				}
+
+				genDoc.GenesisTime = t
+			}
+
+			chainId := cmd.Flag(flagChainId).Value.String()
+			if chainId != "" {
+				genDoc.ChainID = chainId
+			}
+
+			out, err := cdc.MarshalJSONIndent(genDoc, "", "  ")
+			if err != nil {
+				return err
+			}
+
+			fmt.Println(string(out))
+			return nil
+
+		},
+	}
+
+	cmd.Flags().String(flagGenesisTime, "", "Override genesis_time with this flag")
+	cmd.Flags().String(flagChainId, "", "Override chain_id with this flag")
+
+	return cmd
+}

x/genutil/legacy/v036/migrate.go

@@ -0,0 +1,26 @@
+package v036
+
+import (
+	"github.com/cosmos/cosmos-sdk/codec"
+	"github.com/cosmos/cosmos-sdk/x/genutil"
+	v034gov "github.com/cosmos/cosmos-sdk/x/gov/legacy/v034"
+	v036gov "github.com/cosmos/cosmos-sdk/x/gov/legacy/v036"
+)
+
+// Migrate migrates exported state from v0.34 to a v0.36 genesis state.
+func Migrate(appState genutil.AppMap, cdc *codec.Codec) genutil.AppMap {
+	v034Codec := codec.New()
+	codec.RegisterCrypto(v034Codec)
+	v036Codec := codec.New()
+	codec.RegisterCrypto(v036Codec)
+
+	if appState[v034gov.ModuleName] != nil {
+		var govState v034gov.GenesisState
+		v034gov.RegisterCodec(v034Codec)
+		v034Codec.MustUnmarshalJSON(appState[v034gov.ModuleName], &govState)
+		v036gov.RegisterCodec(v036Codec)
+		delete(appState, v034gov.ModuleName) // Drop old key, in case it changed name
+		appState[v036gov.ModuleName] = v036Codec.MustMarshalJSON(v036gov.MigrateGovernance(govState))
+	}
+	return appState
+}

x/genutil/types.go

@@ -0,0 +1,15 @@
+package genutil
+
+import (
+	"encoding/json"
+	"github.com/cosmos/cosmos-sdk/codec"
+)
+
+type (
+	// AppMap map modules names with their json raw representation
+	AppMap map[string]json.RawMessage
+	// MigrationCallback converts a genesis map from the previous version to the targeted one
+	MigrationCallback func(AppMap, *codec.Codec) AppMap
+	// MigrationMap defines a mapping from a version to a MigrationCallback
+	MigrationMap map[string]MigrationCallback
+)