132 lines
3.8 KiB
Go
132 lines
3.8 KiB
Go
package cli
|
|
|
|
import (
|
|
"strconv"
|
|
|
|
"github.com/CosmWasm/wasmd/x/wasm/internal/types"
|
|
"github.com/cosmos/cosmos-sdk/client"
|
|
"github.com/cosmos/cosmos-sdk/client/flags"
|
|
"github.com/cosmos/cosmos-sdk/client/tx"
|
|
sdk "github.com/cosmos/cosmos-sdk/types"
|
|
sdkerrors "github.com/cosmos/cosmos-sdk/types/errors"
|
|
"github.com/spf13/cobra"
|
|
)
|
|
|
|
// MigrateContractCmd will migrate a contract to a new code version
|
|
func MigrateContractCmd() *cobra.Command {
|
|
cmd := &cobra.Command{
|
|
Use: "migrate [contract_addr_bech32] [new_code_id_int64] [json_encoded_migration_args]",
|
|
Short: "Migrate a wasm contract to a new code version",
|
|
Args: cobra.ExactArgs(3),
|
|
RunE: func(cmd *cobra.Command, args []string) error {
|
|
clientCtx := client.GetClientContextFromCmd(cmd)
|
|
clientCtx, err := client.ReadTxCommandFlags(clientCtx, cmd.Flags())
|
|
|
|
msg, err := parseMigrateContractArgs(args, clientCtx)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if err := msg.ValidateBasic(); err != nil {
|
|
return nil
|
|
}
|
|
return tx.GenerateOrBroadcastTxCLI(clientCtx, cmd.Flags(), &msg)
|
|
},
|
|
}
|
|
flags.AddTxFlagsToCmd(cmd)
|
|
return cmd
|
|
}
|
|
|
|
func parseMigrateContractArgs(args []string, cliCtx client.Context) (types.MsgMigrateContract, error) {
|
|
contractAddr, err := sdk.AccAddressFromBech32(args[0])
|
|
if err != nil {
|
|
return types.MsgMigrateContract{}, sdkerrors.Wrap(err, "contract")
|
|
}
|
|
|
|
// get the id of the code to instantiate
|
|
codeID, err := strconv.ParseUint(args[1], 10, 64)
|
|
if err != nil {
|
|
return types.MsgMigrateContract{}, sdkerrors.Wrap(err, "code id")
|
|
}
|
|
|
|
migrateMsg := args[2]
|
|
|
|
msg := types.MsgMigrateContract{
|
|
Sender: cliCtx.GetFromAddress(),
|
|
Contract: contractAddr,
|
|
CodeID: codeID,
|
|
MigrateMsg: []byte(migrateMsg),
|
|
}
|
|
return msg, nil
|
|
}
|
|
|
|
// UpdateContractAdminCmd sets an new admin for a contract
|
|
func UpdateContractAdminCmd() *cobra.Command {
|
|
cmd := &cobra.Command{
|
|
Use: "set-contract-admin [contract_addr_bech32] [new_admin_addr_bech32]",
|
|
Short: "Set new admin for a contract",
|
|
Args: cobra.ExactArgs(2),
|
|
RunE: func(cmd *cobra.Command, args []string) error {
|
|
clientCtx := client.GetClientContextFromCmd(cmd)
|
|
clientCtx, err := client.ReadTxCommandFlags(clientCtx, cmd.Flags())
|
|
|
|
msg, err := parseUpdateContractAdminArgs(args, clientCtx)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if err := msg.ValidateBasic(); err != nil {
|
|
return err
|
|
}
|
|
return tx.GenerateOrBroadcastTxCLI(clientCtx, cmd.Flags(), &msg)
|
|
},
|
|
}
|
|
flags.AddTxFlagsToCmd(cmd)
|
|
return cmd
|
|
}
|
|
|
|
func parseUpdateContractAdminArgs(args []string, cliCtx client.Context) (types.MsgUpdateAdmin, error) {
|
|
contractAddr, err := sdk.AccAddressFromBech32(args[0])
|
|
if err != nil {
|
|
return types.MsgUpdateAdmin{}, sdkerrors.Wrap(err, "contract")
|
|
}
|
|
newAdmin, err := sdk.AccAddressFromBech32(args[1])
|
|
if err != nil {
|
|
return types.MsgUpdateAdmin{}, sdkerrors.Wrap(err, "new admin")
|
|
}
|
|
|
|
msg := types.MsgUpdateAdmin{
|
|
Sender: cliCtx.GetFromAddress(),
|
|
Contract: contractAddr,
|
|
NewAdmin: newAdmin,
|
|
}
|
|
return msg, nil
|
|
}
|
|
|
|
// ClearContractAdminCmd clears an admin for a contract
|
|
func ClearContractAdminCmd() *cobra.Command {
|
|
cmd := &cobra.Command{
|
|
Use: "clear-contract-admin [contract_addr_bech32]",
|
|
Short: "Clears admin for a contract to prevent further migrations",
|
|
Args: cobra.ExactArgs(1),
|
|
RunE: func(cmd *cobra.Command, args []string) error {
|
|
clientCtx := client.GetClientContextFromCmd(cmd)
|
|
clientCtx, err := client.ReadTxCommandFlags(clientCtx, cmd.Flags())
|
|
|
|
contractAddr, err := sdk.AccAddressFromBech32(args[0])
|
|
if err != nil {
|
|
return sdkerrors.Wrap(err, "contract")
|
|
}
|
|
|
|
msg := types.MsgClearAdmin{
|
|
Sender: clientCtx.GetFromAddress(),
|
|
Contract: contractAddr,
|
|
}
|
|
if err := msg.ValidateBasic(); err != nil {
|
|
return err
|
|
}
|
|
return tx.GenerateOrBroadcastTxCLI(clientCtx, cmd.Flags(), &msg)
|
|
},
|
|
}
|
|
flags.AddTxFlagsToCmd(cmd)
|
|
return cmd
|
|
}
|