Genesis port script v0.34.0

scripts/export/export.go

@@ -0,0 +1,93 @@
+package export
+
+import (
+	"fmt"
+	"os"
+	"path/filepath"
+	"strings"
+	"time"
+
+	app "github.com/cosmos/cosmos-sdk/cmd/gaia/app"
+	"github.com/cosmos/cosmos-sdk/codec"
+	tmtypes "github.com/tendermint/tendermint/types"
+)
+
+// GenesisFile defines the Gaia genesis format
+type GenesisFile struct {
+	GenesisTime     string                   `json:"genesis_time"`
+	ChainID         string                   `json:"chain_id"`
+	ConsensusParams *tmtypes.ConsensusParams `json:"consensus_params"`
+	AppHash         string                   `json:"app_hash"`
+	AppState        app.GenesisState         `json:"app_state"`
+}
+
+// NewGenesisFile builds a default GenesisDoc and creates a GenesisFile from it
+func NewGenesisFile(cdc *codec.Codec, path string) (GenesisFile, error) {
+
+	genDoc, err := importGenesis(path)
+	if err != nil {
+		return GenesisFile{}, err
+	}
+
+	var appState app.GenesisState
+	if genDoc.AppState == nil {
+		appState = app.GenesisState{}
+	} else {
+		if err = cdc.UnmarshalJSON(genDoc.AppState, &appState); err != nil {
+			return GenesisFile{}, err
+		}
+	}
+
+	return GenesisFile{
+		GenesisTime:     genDoc.GenesisTime.String(),
+		ChainID:         genDoc.ChainID,
+		ConsensusParams: genDoc.ConsensusParams,
+		AppHash:         genDoc.AppHash.String(),
+		AppState:        appState,
+	}, nil
+}
+
+// ValidateInputs validates each of the parameters used by
+func ValidateInputs(path, chainID, genesisTime string) error {
+	if chainID = strings.Trim(chainID, " "); chainID == "" {
+		return fmt.Errorf("chain-id cannot be blank")
+	}
+	_, err := time.Parse(time.RFC3339, genesisTime)
+	if err != nil {
+		return err
+	}
+
+	if ext := filepath.Ext(path); ext != ".json" {
+		return fmt.Errorf("%s is not a JSON file", path)
+	}
+
+	if _, err = os.Stat(path); err != nil {
+		return err
+	}
+	return nil
+}
+
+// importGenesis imports genesis from JSON and completes missing fields
+func importGenesis(path string) (genDoc *tmtypes.GenesisDoc, err error) {
+	genDoc, err = tmtypes.GenesisDocFromFile(path)
+	if err != nil {
+		return
+	}
+
+	err = genDoc.ValidateAndComplete()
+	if err != nil {
+		return
+	}
+	return
+}
+
+func defaultGenesisDoc(chainID string) (tmtypes.GenesisDoc, error) {
+	genDoc := tmtypes.GenesisDoc{
+		ChainID: chainID,
+	}
+	err := (&genDoc).ValidateAndComplete()
+	if err != nil {
+		return genDoc, err
+	}
+	return genDoc, nil
+}

scripts/export/export_test.go

@@ -0,0 +1,72 @@
+package export
+
+import (
+	"encoding/json"
+	"io/ioutil"
+	"os"
+	"testing"
+
+	app "github.com/cosmos/cosmos-sdk/cmd/gaia/app"
+	"github.com/stretchr/testify/require"
+	tmtypes "github.com/tendermint/tendermint/types"
+)
+
+var (
+	path        = "./genesis.json"
+	chainID     = "cosmos-zone"
+	genesisTime = "2019-02-11T12:00:00Z"
+)
+
+func TestNewGenesisFile(t *testing.T) {
+	cdc := app.MakeCodec()
+	genDoc, err := defaultGenesisDoc(chainID)
+	require.NoError(t, err)
+
+	output, err := cdc.MarshalJSONIndent(genDoc, "", " ")
+	require.NoError(t, err)
+
+	err = ioutil.WriteFile(path, output, 0644)
+	require.NoError(t, err)
+
+	genesisFile, err := NewGenesisFile(cdc, path)
+	require.NoError(t, err)
+	require.NotEqual(t, GenesisFile{}, genesisFile)
+	os.Remove(path)
+}
+
+func TestDefaultGenesisDoc(t *testing.T) {
+	expectedGenDoc := tmtypes.GenesisDoc{ChainID: chainID}
+	genDoc, err := defaultGenesisDoc(chainID)
+	require.NoError(t, err)
+	require.NotEqual(t, expectedGenDoc, genDoc)
+
+	genDoc, err = defaultGenesisDoc("")
+	require.Error(t, err)
+}
+
+func TestImportGenesis(t *testing.T) {
+	genesis := tmtypes.GenesisDoc{ChainID: chainID}
+
+	output, err := json.Marshal(genesis)
+	require.NoError(t, err)
+
+	err = ioutil.WriteFile(path, output, 0644)
+	require.NoError(t, err)
+
+	genDoc, err := importGenesis(path)
+	require.NoError(t, err)
+	require.NotEqual(t, genesis, genDoc)
+	os.Remove(path)
+
+	// should fail with invalid genesis
+	genesis = tmtypes.GenesisDoc{}
+	output, err = json.Marshal(genesis)
+	require.NoError(t, err)
+
+	err = ioutil.WriteFile(path, output, 0644)
+	require.NoError(t, err)
+
+	genDoc, err = importGenesis(path)
+	require.Error(t, err)
+	os.Remove(path)
+}

scripts/export/v0_33_0/main.go

@@ -0,0 +1,59 @@
+package main
+
+import (
+	"fmt"
+	"os"
+	"strings"
+
+	app "github.com/cosmos/cosmos-sdk/cmd/gaia/app"
+	"github.com/cosmos/cosmos-sdk/scripts/export"
+)
+
+// Command: go run main.go [path_to_old_genesis.json] [chain-id] [genesis-start-time] > [path_to_new_genesis.json]
+func main() {
+	cdc := app.MakeCodec()
+
+	args := os.Args[1:]
+	if len(args) != 3 {
+		panic(fmt.Errorf("please provide path, chain-id and genesis time"))
+	}
+
+	pathToGenesis := args[0]
+	chainID := args[1]
+	genesisTime := args[2]
+
+	err := export.ValidateInputs(pathToGenesis, chainID, genesisTime)
+	if err != nil {
+		panic(err)
+	}
+
+	genesis, err := export.NewGenesisFile(cdc, pathToGenesis)
+	if err != nil {
+		panic(err)
+	}
+
+	genesis.ChainID = strings.Trim(chainID, " ")
+	genesis.GenesisTime = genesisTime
+
+	// proposal #1 updates
+	genesis.AppState.MintData.Params.BlocksPerYear = 4855015
+
+	// proposal #2 updates
+	genesis.ConsensusParams.Block.MaxGas = 200000
+	genesis.ConsensusParams.Block.MaxBytes = 2000000
+
+	// enable transfers
+	genesis.AppState.BankData.SendEnabled = true
+	genesis.AppState.DistrData.WithdrawAddrEnabled = true
+
+	err = app.GaiaValidateGenesisState(genesis.AppState)
+	if err != nil {
+		panic(err)
+	}
+
+	genesisJSON, err := cdc.MarshalJSONIndent(genesis, "", "  ")
+	if err != nil {
+		panic(err)
+	}
+	fmt.Println(string(genesisJSON))
+}