export-stake command works

cmd/gaia/app/app.go

@@ -148,6 +148,11 @@ func (app *GaiaApp) initChainer(ctx sdk.Context, req abci.RequestInitChain) abci
 	return abci.ResponseInitChain{}
 }
 
+// custom logic for export
+func (app *GaiaApp) ExportStake() stake.GenesisState {
+	return app.stakeKeeper.WriteGenesis(app.NewContext(true, abci.Header{}))
+}
+
 //__________________________________________________________
 
 // State to Unmarshal

cmd/gaia/cmd/gaiad/main.go

@@ -13,6 +13,8 @@ import (
 
 	"github.com/cosmos/cosmos-sdk/cmd/gaia/app"
 	"github.com/cosmos/cosmos-sdk/server"
+	stake "github.com/cosmos/cosmos-sdk/x/stake"
+	stakecmd "github.com/cosmos/cosmos-sdk/x/stake/commands"
 )
 
 // rootCmd is the entry point for this binary
@@ -38,8 +40,24 @@ func generateApp(rootDir string, logger log.Logger) (abci.Application, error) {
 func main() {
 	server.AddCommands(rootCmd, app.DefaultGenAppState, generateApp, context)
 
-	// prepare and add flags
 	rootDir := os.ExpandEnv("$HOME/.gaiad")
+
+	export := func() (stake.GenesisState, error) {
+		dataDir := filepath.Join(rootDir, "data")
+		db, err := dbm.NewGoLevelDB("gaia", dataDir)
+		if err != nil {
+			return stake.GenesisState{}, err
+		}
+		app := app.NewGaiaApp(log.NewNopLogger(), db)
+		if err != nil {
+			return stake.GenesisState{}, err
+		}
+		return app.ExportStake(), nil
+	}
+
+	rootCmd.AddCommand(stakecmd.GetCmdExport(export, app.MakeCodec()))
+
+	// prepare and add flags
 	executor := cli.PrepareBaseCmd(rootCmd, "GA", rootDir)
 	executor.Execute()
 }

x/stake/commands/export.go

@@ -5,19 +5,20 @@ import (
 
 	"github.com/spf13/cobra"
 
-	sdk "github.com/cosmos/cosmos-sdk/types"
 	"github.com/cosmos/cosmos-sdk/wire" // XXX fix
 	"github.com/cosmos/cosmos-sdk/x/stake"
 )
 
 // get the command to export state
-func GetCmdExport(load func() (stake.Keeper, sdk.Context), cdc *wire.Codec) *cobra.Command {
+func GetCmdExport(export func() (stake.GenesisState, error), cdc *wire.Codec) *cobra.Command {
 	cmd := &cobra.Command{
-		Use:   "export",
+		Use:   "export-stake",
 		Short: "Export stake module state as JSON to stdout",
 		RunE: func(cmd *cobra.Command, args []string) error {
-			keeper, ctx := load()
-			genesis := keeper.WriteGenesis(ctx)
+			genesis, err := export()
+			if err != nil {
+				return err
+			}
 			output, err := wire.MarshalJSONIndent(cdc, genesis)
 			if err != nil {
 				return err

x/stake/handler.go

@@ -64,8 +64,8 @@ func (k Keeper) InitGenesis(ctx sdk.Context, data GenesisState) {
 func (k Keeper) WriteGenesis(ctx sdk.Context) GenesisState {
 	pool := k.GetPool(ctx)
 	params := k.GetParams(ctx)
-	candidates := k.GetCandidates(ctx, -1)
-	bonds := k.GetBonds(ctx, -1)
+	candidates := k.GetCandidates(ctx, 32767)
+	bonds := k.GetBonds(ctx, 32767)
 	return GenesisState{
 		pool,
 		params,

x/stake/keeper.go

@@ -74,15 +74,15 @@ func (k Keeper) GetCandidate(ctx sdk.Context, addr sdk.Address) (candidate Candi
 	return candidate, true
 }
 
-// Get the set of all candidates, retrieve a maxRetrieve number of records, -1 maxRetrieve = no limit
+// Get the set of all candidates, retrieve a maxRetrieve number of records
 func (k Keeper) GetCandidates(ctx sdk.Context, maxRetrieve int16) (candidates Candidates) {
 	store := ctx.KVStore(k.storeKey)
 	iterator := store.Iterator(subspace(CandidatesKey))
 
 	candidates = make([]Candidate, maxRetrieve)
 	i := 0
 	for ; ; i++ {
-		if !iterator.Valid() || (maxRetrieve >= 0 && i > int(maxRetrieve-1)) {
+		if !iterator.Valid() || i > int(maxRetrieve-1) {
 			iterator.Close()
 			break
 		}
@@ -371,15 +371,15 @@ func (k Keeper) GetDelegatorBond(ctx sdk.Context,
 	return bond, true
 }
 
-// load all bonds, -1 maxRetrieve = no limit
+// load all bonds
 func (k Keeper) GetBonds(ctx sdk.Context, maxRetrieve int16) (bonds []DelegatorBond) {
 	store := ctx.KVStore(k.storeKey)
 	iterator := store.Iterator(subspace(DelegatorBondKeyPrefix))
 
 	bonds = make([]DelegatorBond, maxRetrieve)
 	i := 0
 	for ; ; i++ {
-		if !iterator.Valid() || (maxRetrieve >= 0 && i > int(maxRetrieve-1)) {
+		if !iterator.Valid() || i > int(maxRetrieve-1) {
 			iterator.Close()
 			break
 		}