cli

x/babylon/client/cli/tx.go

@@ -1,9 +1,13 @@
 package cli
 
 import (
-	"github.com/spf13/cobra"
+	"encoding/hex"
+	"strconv"
 
 	"github.com/cosmos/cosmos-sdk/client"
+	"github.com/cosmos/cosmos-sdk/client/flags"
+	"github.com/cosmos/cosmos-sdk/client/tx"
+	"github.com/spf13/cobra"
 
 	"github.com/babylonlabs-io/babylon-sdk/x/babylon/types"
 )
@@ -20,6 +24,95 @@ func GetTxCmd() *cobra.Command {
 		RunE:                       client.ValidateCmd,
 		SilenceUsage:               true,
 	}
-	txCmd.AddCommand()
+	txCmd.AddCommand(
+		NewInstantiateBabylonContractsCmd(),
+	)
 	return txCmd
 }
+
+// [babylon-contract-code-id] [btc-staking-contract-code-id] [btc-finality-contract-code-id] [btc-network] [babylon-tag] [btc-confirmation-depth] [checkpoint-finalization-timeout] [notify-cosmos-zone] [btc-staking-init-msg-hex] [btc-finality-init-msg-hex] [consumer-name] [consumer-description] [admin]
+func parseInstantiateArgs(args []string, sender string) (*types.MsgInstantiateBabylonContracts, error) {
+	// get the id of the code to instantiate
+	babylonContractCodeID, err := strconv.ParseUint(args[0], 10, 64)
+	if err != nil {
+		return nil, err
+	}
+	btcStakingContractCodeID, err := strconv.ParseUint(args[1], 10, 64)
+	if err != nil {
+		return nil, err
+	}
+	btcFinalityContractCodeID, err := strconv.ParseUint(args[2], 10, 64)
+	if err != nil {
+		return nil, err
+	}
+
+	btcNetwork := args[3]
+	babylonTag := args[4]
+	btcConfirmationDepth, err := strconv.ParseUint(args[5], 10, 32)
+	if err != nil {
+		return nil, err
+	}
+	checkpointFinalizationTimeout, err := strconv.ParseUint(args[6], 10, 32)
+	if err != nil {
+		return nil, err
+	}
+	notifyCosmosZone, err := strconv.ParseBool(args[7])
+	if err != nil {
+		return nil, err
+	}
+	btcStakingInitMsg, err := hex.DecodeString(args[8])
+	if err != nil {
+		return nil, err
+	}
+	btcFinalityInitMsg, err := hex.DecodeString(args[9])
+	if err != nil {
+		return nil, err
+	}
+	consumerName := args[10]
+	consumerDescription := args[11]
+	adminStr := args[12]
+
+	// build and sign the transaction, then broadcast to Tendermint
+	msg := types.MsgInstantiateBabylonContracts{
+		Signer:                        sender,
+		BabylonContractCodeId:         babylonContractCodeID,
+		BtcStakingContractCodeId:      btcStakingContractCodeID,
+		BtcFinalityContractCodeId:     btcFinalityContractCodeID,
+		Network:                       btcNetwork,
+		BabylonTag:                    babylonTag,
+		BtcConfirmationDepth:          uint32(btcConfirmationDepth),
+		CheckpointFinalizationTimeout: uint32(checkpointFinalizationTimeout),
+		NotifyCosmosZone:              notifyCosmosZone,
+		BtcStakingMsg:                 btcStakingInitMsg,
+		BtcFinalityMsg:                btcFinalityInitMsg,
+		ConsumerName:                  consumerName,
+		ConsumerDescription:           consumerDescription,
+		Admin:                         adminStr,
+	}
+	return &msg, nil
+}
+
+func NewInstantiateBabylonContractsCmd() *cobra.Command {
+	cmd := &cobra.Command{
+		Use:     "instantiate-babylon-contracts [babylon-contract-code-id] [btc-staking-contract-code-id] [btc-finality-contract-code-id] [btc-network] [babylon-tag] [btc-confirmation-depth] [checkpoint-finalization-timeout] [notify-cosmos-zone] [btc-staking-init-msg-hex] [btc-finality-init-msg-hex] [consumer-name] [consumer-description] [admin]",
+		Short:   "Instantiate Babylon contracts",
+		Long:    "Instantiate Babylon contracts",
+		Aliases: []string{"i"},
+		Args:    cobra.ExactArgs(11),
+		RunE: func(cmd *cobra.Command, args []string) error {
+			clientCtx, err := client.GetClientTxContext(cmd)
+			if err != nil {
+				return err
+			}
+			msg, err := parseInstantiateArgs(args, clientCtx.GetFromAddress().String())
+			if err != nil {
+				return err
+			}
+			return tx.GenerateOrBroadcastTxCLI(clientCtx, cmd.Flags(), msg)
+		},
+		SilenceUsage: true,
+	}
+
+	flags.AddTxFlagsToCmd(cmd)
+	return cmd
+}