This repository has been archived by the owner on Jan 6, 2022. It is now read-only.
-
Notifications
You must be signed in to change notification settings - Fork 13
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
* go mod init = * genaccounts * genwasm * Cosmos app * Encoding * Prefix * gosum * Init gitignore * root * Make types public * add imports * Organize imports * Lint * Rename to cosmoscmd * Simplification * Remove blank line
- Loading branch information
Showing
9 changed files
with
1,624 additions
and
0 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1 @@ | ||
.idea/ |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,40 @@ | ||
package cosmoscmd | ||
|
||
import ( | ||
abci "github.com/tendermint/tendermint/abci/types" | ||
|
||
"github.com/cosmos/cosmos-sdk/codec" | ||
"github.com/cosmos/cosmos-sdk/server/types" | ||
sdk "github.com/cosmos/cosmos-sdk/types" | ||
) | ||
|
||
// CosmosApp implements the common methods for a Cosmos SDK-based application | ||
// specific blockchain. | ||
type CosmosApp interface { | ||
// The assigned name of the app. | ||
Name() string | ||
|
||
// The application types codec. | ||
// NOTE: This should be sealed before being returned. | ||
LegacyAmino() *codec.LegacyAmino | ||
|
||
// Application updates every begin block. | ||
BeginBlocker(ctx sdk.Context, req abci.RequestBeginBlock) abci.ResponseBeginBlock | ||
|
||
// Application updates every end block. | ||
EndBlocker(ctx sdk.Context, req abci.RequestEndBlock) abci.ResponseEndBlock | ||
|
||
// Application update at chain (i.e app) initialization. | ||
InitChainer(ctx sdk.Context, req abci.RequestInitChain) abci.ResponseInitChain | ||
|
||
// Loads the app at a given height. | ||
LoadHeight(height int64) error | ||
|
||
// Exports the state of the application for a genesis file. | ||
ExportAppStateAndValidators( | ||
forZeroHeight bool, jailAllowedAddrs []string, | ||
) (types.ExportedApp, error) | ||
|
||
// All the registered module account addreses. | ||
ModuleAccountAddrs() map[string]bool | ||
} |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,44 @@ | ||
package cosmoscmd | ||
|
||
import ( | ||
"github.com/cosmos/cosmos-sdk/client" | ||
"github.com/cosmos/cosmos-sdk/codec" | ||
"github.com/cosmos/cosmos-sdk/codec/types" | ||
"github.com/cosmos/cosmos-sdk/std" | ||
"github.com/cosmos/cosmos-sdk/types/module" | ||
"github.com/cosmos/cosmos-sdk/x/auth/tx" | ||
) | ||
|
||
// EncodingConfig specifies the concrete encoding types to use for a given app. | ||
// This is provided for compatibility between protobuf and amino implementations. | ||
type EncodingConfig struct { | ||
InterfaceRegistry types.InterfaceRegistry | ||
Marshaler codec.Marshaler | ||
TxConfig client.TxConfig | ||
Amino *codec.LegacyAmino | ||
} | ||
|
||
// makeEncodingConfig creates an EncodingConfig for an amino based test configuration. | ||
func makeEncodingConfig() EncodingConfig { | ||
amino := codec.NewLegacyAmino() | ||
interfaceRegistry := types.NewInterfaceRegistry() | ||
marshaler := codec.NewProtoCodec(interfaceRegistry) | ||
txCfg := tx.NewTxConfig(marshaler, tx.DefaultSignModes) | ||
|
||
return EncodingConfig{ | ||
InterfaceRegistry: interfaceRegistry, | ||
Marshaler: marshaler, | ||
TxConfig: txCfg, | ||
Amino: amino, | ||
} | ||
} | ||
|
||
// MakeEncodingConfig creates an EncodingConfig for testing | ||
func MakeEncodingConfig(moduleBasics module.BasicManager) EncodingConfig { | ||
encodingConfig := makeEncodingConfig() | ||
std.RegisterLegacyAminoCodec(encodingConfig.Amino) | ||
std.RegisterInterfaces(encodingConfig.InterfaceRegistry) | ||
moduleBasics.RegisterLegacyAminoCodec(encodingConfig.Amino) | ||
moduleBasics.RegisterInterfaces(encodingConfig.InterfaceRegistry) | ||
return encodingConfig | ||
} |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,192 @@ | ||
package cosmoscmd | ||
|
||
import ( | ||
"bufio" | ||
"encoding/json" | ||
"errors" | ||
"fmt" | ||
|
||
"github.com/spf13/cobra" | ||
|
||
"github.com/cosmos/cosmos-sdk/client" | ||
"github.com/cosmos/cosmos-sdk/client/flags" | ||
"github.com/cosmos/cosmos-sdk/codec" | ||
"github.com/cosmos/cosmos-sdk/crypto/keyring" | ||
"github.com/cosmos/cosmos-sdk/server" | ||
sdk "github.com/cosmos/cosmos-sdk/types" | ||
authtypes "github.com/cosmos/cosmos-sdk/x/auth/types" | ||
authvesting "github.com/cosmos/cosmos-sdk/x/auth/vesting/types" | ||
banktypes "github.com/cosmos/cosmos-sdk/x/bank/types" | ||
"github.com/cosmos/cosmos-sdk/x/genutil" | ||
genutiltypes "github.com/cosmos/cosmos-sdk/x/genutil/types" | ||
) | ||
|
||
const ( | ||
flagVestingStart = "vesting-start-time" | ||
flagVestingEnd = "vesting-end-time" | ||
flagVestingAmt = "vesting-amount" | ||
) | ||
|
||
// AddGenesisAccountCmd returns add-genesis-account cobra Command. | ||
func AddGenesisAccountCmd(defaultNodeHome string) *cobra.Command { | ||
cmd := &cobra.Command{ | ||
Use: "add-genesis-account [address_or_key_name] [coin][,[coin]]", | ||
Short: "Add a genesis account to genesis.json", | ||
Long: `Add a genesis account to genesis.json. The provided account must specify | ||
the account address or key name and a list of initial coins. If a key name is given, | ||
the address will be looked up in the local Keybase. The list of initial tokens must | ||
contain valid denominations. Accounts may optionally be supplied with vesting parameters. | ||
`, | ||
Args: cobra.ExactArgs(2), | ||
RunE: func(cmd *cobra.Command, args []string) error { | ||
clientCtx := client.GetClientContextFromCmd(cmd) | ||
depCdc := clientCtx.JSONMarshaler | ||
cdc := depCdc.(codec.Marshaler) | ||
|
||
serverCtx := server.GetServerContextFromCmd(cmd) | ||
config := serverCtx.Config | ||
|
||
config.SetRoot(clientCtx.HomeDir) | ||
|
||
coins, err := sdk.ParseCoinsNormalized(args[1]) | ||
if err != nil { | ||
return fmt.Errorf("failed to parse coins: %w", err) | ||
} | ||
|
||
addr, err := sdk.AccAddressFromBech32(args[0]) | ||
if err != nil { | ||
inBuf := bufio.NewReader(cmd.InOrStdin()) | ||
keyringBackend, err := cmd.Flags().GetString(flags.FlagKeyringBackend) | ||
if err != nil { | ||
return err | ||
} | ||
|
||
// attempt to lookup address from Keybase if no address was provided | ||
kb, err := keyring.New(sdk.KeyringServiceName(), keyringBackend, clientCtx.HomeDir, inBuf) | ||
if err != nil { | ||
return err | ||
} | ||
|
||
info, err := kb.Key(args[0]) | ||
if err != nil { | ||
return fmt.Errorf("failed to get address from Keybase: %w", err) | ||
} | ||
|
||
addr = info.GetAddress() | ||
} | ||
|
||
vestingStart, err := cmd.Flags().GetInt64(flagVestingStart) | ||
if err != nil { | ||
return err | ||
} | ||
vestingEnd, err := cmd.Flags().GetInt64(flagVestingEnd) | ||
if err != nil { | ||
return err | ||
} | ||
vestingAmtStr, err := cmd.Flags().GetString(flagVestingAmt) | ||
if err != nil { | ||
return err | ||
} | ||
|
||
vestingAmt, err := sdk.ParseCoinsNormalized(vestingAmtStr) | ||
if err != nil { | ||
return fmt.Errorf("failed to parse vesting amount: %w", err) | ||
} | ||
|
||
// create concrete account type based on input parameters | ||
var genAccount authtypes.GenesisAccount | ||
|
||
balances := banktypes.Balance{Address: addr.String(), Coins: coins.Sort()} | ||
baseAccount := authtypes.NewBaseAccount(addr, nil, 0, 0) | ||
|
||
if !vestingAmt.IsZero() { | ||
baseVestingAccount := authvesting.NewBaseVestingAccount(baseAccount, vestingAmt.Sort(), vestingEnd) | ||
|
||
if (balances.Coins.IsZero() && !baseVestingAccount.OriginalVesting.IsZero()) || | ||
baseVestingAccount.OriginalVesting.IsAnyGT(balances.Coins) { | ||
return errors.New("vesting amount cannot be greater than total amount") | ||
} | ||
|
||
switch { | ||
case vestingStart != 0 && vestingEnd != 0: | ||
genAccount = authvesting.NewContinuousVestingAccountRaw(baseVestingAccount, vestingStart) | ||
|
||
case vestingEnd != 0: | ||
genAccount = authvesting.NewDelayedVestingAccountRaw(baseVestingAccount) | ||
|
||
default: | ||
return errors.New("invalid vesting parameters; must supply start and end time or end time") | ||
} | ||
} else { | ||
genAccount = baseAccount | ||
} | ||
|
||
if err := genAccount.Validate(); err != nil { | ||
return fmt.Errorf("failed to validate new genesis account: %w", err) | ||
} | ||
|
||
genFile := config.GenesisFile() | ||
appState, genDoc, err := genutiltypes.GenesisStateFromGenFile(genFile) | ||
if err != nil { | ||
return fmt.Errorf("failed to unmarshal genesis state: %w", err) | ||
} | ||
|
||
authGenState := authtypes.GetGenesisStateFromAppState(cdc, appState) | ||
|
||
accs, err := authtypes.UnpackAccounts(authGenState.Accounts) | ||
if err != nil { | ||
return fmt.Errorf("failed to get accounts from any: %w", err) | ||
} | ||
|
||
if accs.Contains(addr) { | ||
return fmt.Errorf("cannot add account at existing address %s", addr) | ||
} | ||
|
||
// Add the new account to the set of genesis accounts and sanitize the | ||
// accounts afterwards. | ||
accs = append(accs, genAccount) | ||
accs = authtypes.SanitizeGenesisAccounts(accs) | ||
|
||
genAccs, err := authtypes.PackAccounts(accs) | ||
if err != nil { | ||
return fmt.Errorf("failed to convert accounts into any's: %w", err) | ||
} | ||
authGenState.Accounts = genAccs | ||
|
||
authGenStateBz, err := cdc.MarshalJSON(&authGenState) | ||
if err != nil { | ||
return fmt.Errorf("failed to marshal auth genesis state: %w", err) | ||
} | ||
|
||
appState[authtypes.ModuleName] = authGenStateBz | ||
|
||
bankGenState := banktypes.GetGenesisStateFromAppState(depCdc, appState) | ||
bankGenState.Balances = append(bankGenState.Balances, balances) | ||
bankGenState.Balances = banktypes.SanitizeGenesisBalances(bankGenState.Balances) | ||
|
||
bankGenStateBz, err := cdc.MarshalJSON(bankGenState) | ||
if err != nil { | ||
return fmt.Errorf("failed to marshal bank genesis state: %w", err) | ||
} | ||
|
||
appState[banktypes.ModuleName] = bankGenStateBz | ||
|
||
appStateJSON, err := json.Marshal(appState) | ||
if err != nil { | ||
return fmt.Errorf("failed to marshal application genesis state: %w", err) | ||
} | ||
|
||
genDoc.AppState = appStateJSON | ||
return genutil.ExportGenesisFile(genDoc, genFile) | ||
}, | ||
} | ||
|
||
cmd.Flags().String(flags.FlagKeyringBackend, flags.DefaultKeyringBackend, "Select keyring's backend (os|file|kwallet|pass|test)") | ||
cmd.Flags().String(flags.FlagHome, defaultNodeHome, "The application home directory") | ||
cmd.Flags().String(flagVestingAmt, "", "amount of coins for vesting accounts") | ||
cmd.Flags().Int64(flagVestingStart, 0, "schedule start time (unix epoch) for vesting accounts") | ||
cmd.Flags().Int64(flagVestingEnd, 0, "schedule end time (unix epoch) for vesting accounts") | ||
flags.AddQueryFlagsToCmd(cmd) | ||
|
||
return cmd | ||
} |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,19 @@ | ||
package cosmoscmd | ||
|
||
import sdk "github.com/cosmos/cosmos-sdk/types" | ||
|
||
func SetPrefixes(accountAddressPrefix string) { | ||
// Set prefixes | ||
accountPubKeyPrefix := accountAddressPrefix + "pub" | ||
validatorAddressPrefix := accountAddressPrefix + "valoper" | ||
validatorPubKeyPrefix := accountAddressPrefix + "valoperpub" | ||
consNodeAddressPrefix := accountAddressPrefix + "valcons" | ||
consNodePubKeyPrefix := accountAddressPrefix + "valconspub" | ||
|
||
// Set and seal config | ||
config := sdk.GetConfig() | ||
config.SetBech32PrefixForAccount(accountAddressPrefix, accountPubKeyPrefix) | ||
config.SetBech32PrefixForValidator(validatorAddressPrefix, validatorPubKeyPrefix) | ||
config.SetBech32PrefixForConsensusNode(consNodeAddressPrefix, consNodePubKeyPrefix) | ||
config.Seal() | ||
} |
Oops, something went wrong.