more cli related

This commit is contained in:
rigelrozanski 2018-09-19 00:10:33 -04:00
parent e3cf4f606e
commit 73c5fdf1f9
4 changed files with 25 additions and 12 deletions

View File

@ -101,6 +101,20 @@ func main() {
stakeCmd,
)
//Add distribution commands
distrCmd := &cobra.Command{
Use: "distribution",
Short: "withdraw rewards for delegation and validation",
}
distrCmd.AddCommand(
client.PostCommands(
stakecmd.GetCmdWithdrawDelegationRewardsAll(cdc),
stakecmd.GetCmdSetWithdrawAddr(cdc),
)...)
rootCmd.AddCommand(
distrCmd,
)
//Add stake commands
govCmd := &cobra.Command{
Use: "gov",

View File

@ -80,7 +80,7 @@ func GetCmdWithdrawDelegationRewardsAll(cdc *wire.Codec) *cobra.Command {
}
// GetCmdDelegate implements the delegate command.
func GetCmdWithdrawDelegationRewardsAll(cdc *wire.Codec) *cobra.Command {
func GetCmdSetWithdrawAddr(cdc *wire.Codec) *cobra.Command {
cmd := &cobra.Command{
Use: "set-withdraw-addr [withdraw-addr]",
Short: "change the default withdraw address for rewards associated with an address",
@ -103,7 +103,7 @@ func GetCmdWithdrawDelegationRewardsAll(cdc *wire.Codec) *cobra.Command {
return err
}
msg := distr.NewMsgModifyWithdrawAddress(delAddr, withdrawAddr)
msg := distr.NewMsgSetWithdrawAddress(delAddr, withdrawAddr)
// build and sign the transaction, then broadcast to Tendermint
return utils.SendTx(txCtx, cliCtx, []sdk.Msg{msg})

View File

@ -4,7 +4,6 @@ import (
"github.com/cosmos/cosmos-sdk/codec"
)
// XXX TODO
// Register concrete types on codec codec
func RegisterCodec(cdc *codec.Codec) {
cdc.RegisterConcrete(MsgWithdrawDelegationRewardsAll{}, "cosmos-sdk/MsgWithdrawDelegationRewardsAll", nil)

View File

@ -9,34 +9,34 @@ import (
const MsgType = "distr"
// Verify interface at compile time
var _, _ sdk.Msg = &MsgModifyWithdrawAddress{}, &MsgWithdrawDelegatorRewardsAll{}
var _, _ sdk.Msg = &MsgSetWithdrawAddress{}, &MsgWithdrawDelegatorRewardsAll{}
var _, _ sdk.Msg = &MsgWithdrawDelegationReward{}, &MsgWithdrawValidatorRewardsAll{}
//______________________________________________________________________
// msg struct for changing the withdraw address for a delegator (or validator self-delegation)
type MsgModifyWithdrawAddress struct {
type MsgSetWithdrawAddress struct {
DelegatorAddr sdk.AccAddress `json:"delegator_addr"`
WithdrawAddr sdk.AccAddress `json:"delegator_addr"`
}
func NewMsgModifyWithdrawAddress(delAddr, withdrawAddr sdk.AccAddress) MsgModifyWithdrawAddress {
return MsgModifyWithdrawAddress{
func NewMsgSetWithdrawAddress(delAddr, withdrawAddr sdk.AccAddress) MsgSetWithdrawAddress {
return MsgSetWithdrawAddress{
DelegatorAddr: delAddr,
WithdrawAddr: withdrawAddr,
}
}
func (msg MsgModifyWithdrawAddress) Type() string { return MsgType }
func (msg MsgModifyWithdrawAddress) Name() string { return "withdraw_delegation_rewards_all" }
func (msg MsgSetWithdrawAddress) Type() string { return MsgType }
func (msg MsgSetWithdrawAddress) Name() string { return "withdraw_delegation_rewards_all" }
// Return address that must sign over msg.GetSignBytes()
func (msg MsgModifyWithdrawAddress) GetSigners() []sdk.AccAddress {
func (msg MsgSetWithdrawAddress) GetSigners() []sdk.AccAddress {
return []sdk.AccAddress{sdk.AccAddress(msg.DelegatorAddr)}
}
// get the bytes for the message signer to sign on
func (msg MsgModifyWithdrawAddress) GetSignBytes() []byte {
func (msg MsgSetWithdrawAddress) GetSignBytes() []byte {
b, err := MsgCdc.MarshalJSON(msg)
if err != nil {
panic(err)
@ -45,7 +45,7 @@ func (msg MsgModifyWithdrawAddress) GetSignBytes() []byte {
}
// quick validity check
func (msg MsgModifyWithdrawAddress) ValidateBasic() sdk.Error {
func (msg MsgSetWithdrawAddress) ValidateBasic() sdk.Error {
if msg.DelegatorAddr == nil {
return ErrNilDelegatorAddr(DefaultCodespace)
}