115 lines
3.1 KiB
Go
115 lines
3.1 KiB
Go
// nolint
|
|
package cli
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
|
|
"github.com/spf13/cobra"
|
|
"github.com/spf13/viper"
|
|
|
|
"github.com/cosmos/cosmos-sdk/client/context"
|
|
"github.com/cosmos/cosmos-sdk/client/utils"
|
|
"github.com/cosmos/cosmos-sdk/codec"
|
|
sdk "github.com/cosmos/cosmos-sdk/types"
|
|
distr "github.com/cosmos/cosmos-sdk/x/distribution"
|
|
)
|
|
|
|
var (
|
|
flagOnlyFromValidator = "only-from-validator"
|
|
flagIsValidator = "is-validator"
|
|
)
|
|
|
|
// command to withdraw rewards
|
|
func GetCmdWithdrawDelegationRewardsAll(cdc *codec.Codec) *cobra.Command {
|
|
cmd := &cobra.Command{
|
|
Use: "withdraw-rewards",
|
|
Short: "withdraw rewards for either: all-delegations, a delegation, or a validator",
|
|
Args: cobra.NoArgs,
|
|
RunE: func(cmd *cobra.Command, args []string) error {
|
|
|
|
onlyFromVal := viper.GetString(flagOnlyFromValidator)
|
|
isVal := viper.GetBool(flagIsValidator)
|
|
|
|
if onlyFromVal != "" && isVal {
|
|
return fmt.Errorf("cannot use --%v, and --%v flags together",
|
|
flagOnlyFromValidator, flagIsValidator)
|
|
}
|
|
|
|
txCtx := authctx.NewTxContextFromCLI().WithCodec(cdc)
|
|
cliCtx := context.NewCLIContext().
|
|
WithCodec(cdc).
|
|
WithLogger(os.Stdout).
|
|
WithAccountDecoder(authcmd.GetAccountDecoder(cdc))
|
|
|
|
var msg sdk.Msg
|
|
switch {
|
|
case isVal:
|
|
addr, err := cliCtx.GetFromAddress()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
valAddr := sdk.ValAddress{addr.Bytes()}
|
|
msg := distr.NewMsgWithdrawValidatorRewardsAll(valAddr)
|
|
case onlyFromVal != "":
|
|
delAddr, err := cliCtx.GetFromAddress()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
valAddr, err := sdk.ValAddressFromBech32(onlyFromVal)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
msg := distr.NewMsgWithdrawDelegationReward(delAddr, valAddr)
|
|
default:
|
|
delAddr, err := cliCtx.GetFromAddress()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
msg := distr.NewMsgWithdrawDelegationRewardsAll(delAddr)
|
|
}
|
|
|
|
// build and sign the transaction, then broadcast to Tendermint
|
|
return utils.SendTx(txCtx, cliCtx, []sdk.Msg{msg})
|
|
},
|
|
}
|
|
cmd.Flags().String(flagOnlyFromValidator, "", "only withdraw from this validator address (in bech)")
|
|
cmd.Flags().Bool(flagIsValidator, false, "also withdraw validator's commission")
|
|
return cmd
|
|
}
|
|
|
|
// GetCmdDelegate implements the delegate command.
|
|
func GetCmdSetWithdrawAddr(cdc *codec.Codec) *cobra.Command {
|
|
cmd := &cobra.Command{
|
|
Use: "set-withdraw-addr [withdraw-addr]",
|
|
Short: "change the default withdraw address for rewards associated with an address",
|
|
Args: cobra.ExactArgs(1),
|
|
RunE: func(cmd *cobra.Command, args []string) error {
|
|
|
|
txCtx := authctx.NewTxContextFromCLI().WithCodec(cdc)
|
|
cliCtx := context.NewCLIContext().
|
|
WithCodec(cdc).
|
|
WithLogger(os.Stdout).
|
|
WithAccountDecoder(authcmd.GetAccountDecoder(cdc))
|
|
|
|
delAddr, err := cliCtx.GetFromAddress()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
withdrawAddr, err := sdk.AccAddressFromBech32(args[0])
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
msg := distr.NewMsgSetWithdrawAddress(delAddr, withdrawAddr)
|
|
|
|
// build and sign the transaction, then broadcast to Tendermint
|
|
return utils.SendTx(txCtx, cliCtx, []sdk.Msg{msg})
|
|
},
|
|
}
|
|
return cmd
|
|
}
|