78 lines
2.1 KiB
Go
78 lines
2.1 KiB
Go
package init
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
|
|
"github.com/spf13/cobra"
|
|
"github.com/spf13/viper"
|
|
"github.com/tendermint/tendermint/libs/cli"
|
|
"github.com/tendermint/tendermint/libs/common"
|
|
|
|
"github.com/cosmos/cosmos-sdk/cmd/gaia/app"
|
|
"github.com/cosmos/cosmos-sdk/codec"
|
|
"github.com/cosmos/cosmos-sdk/server"
|
|
sdk "github.com/cosmos/cosmos-sdk/types"
|
|
"github.com/cosmos/cosmos-sdk/x/auth"
|
|
)
|
|
|
|
// AddGenesisAccountCmd returns add-genesis-account cobra Command
|
|
func AddGenesisAccountCmd(ctx *server.Context, cdc *codec.Codec) *cobra.Command {
|
|
cmd := &cobra.Command{
|
|
Use: "add-genesis-account [address] [coin][,[coin]]",
|
|
Short: "Add genesis account to genesis.json",
|
|
Args: cobra.ExactArgs(2),
|
|
RunE: func(_ *cobra.Command, args []string) error {
|
|
config := ctx.Config
|
|
config.SetRoot(viper.GetString(cli.HomeFlag))
|
|
|
|
addr, err := sdk.AccAddressFromBech32(args[0])
|
|
if err != nil {
|
|
return err
|
|
}
|
|
coins, err := sdk.ParseCoins(args[1])
|
|
if err != nil {
|
|
return err
|
|
}
|
|
coins.Sort()
|
|
|
|
genFile := config.GenesisFile()
|
|
if !common.FileExists(genFile) {
|
|
return fmt.Errorf("%s does not exist, run `gaiad init` first", genFile)
|
|
}
|
|
genDoc, err := loadGenesisDoc(cdc, genFile)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
var appState app.GenesisState
|
|
if err = cdc.UnmarshalJSON(genDoc.AppState, &appState); err != nil {
|
|
return err
|
|
}
|
|
|
|
appStateJSON, err := addGenesisAccount(cdc, appState, addr, coins)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return ExportGenesisFile(genFile, genDoc.ChainID, nil, appStateJSON)
|
|
},
|
|
}
|
|
|
|
cmd.Flags().String(cli.HomeFlag, app.DefaultNodeHome, "node's home directory")
|
|
return cmd
|
|
}
|
|
|
|
func addGenesisAccount(cdc *codec.Codec, appState app.GenesisState, addr sdk.AccAddress, coins sdk.Coins) (json.RawMessage, error) {
|
|
for _, stateAcc := range appState.Accounts {
|
|
if stateAcc.Address.Equals(addr) {
|
|
return nil, fmt.Errorf("the application state already contains account %v", addr)
|
|
}
|
|
}
|
|
|
|
acc := auth.NewBaseAccountWithAddress(addr)
|
|
acc.Coins = coins
|
|
appState.Accounts = append(appState.Accounts, app.NewGenesisAccount(&acc))
|
|
return cdc.MarshalJSON(appState)
|
|
}
|