feat: checkpointing/add create-bls-key cli

cmd/babylond/cmd/create_bls_key.go

@@ -0,0 +1,69 @@
+package cmd
+
+import (
+	"errors"
+	"fmt"
+	"github.com/babylonchain/babylon/app"
+	appparams "github.com/babylonchain/babylon/app/params"
+	"github.com/babylonchain/babylon/crypto/bls12381"
+	"github.com/babylonchain/babylon/privval"
+	"github.com/cosmos/cosmos-sdk/client/flags"
+	sdk "github.com/cosmos/cosmos-sdk/types"
+	"github.com/spf13/cobra"
+	tmconfig "github.com/tendermint/tendermint/config"
+	tmos "github.com/tendermint/tendermint/libs/os"
+	"path/filepath"
+	"strings"
+)
+
+func createBlsKeyCmd() *cobra.Command {
+	bech32PrefixAccAddr := appparams.Bech32PrefixAccAddr
+
+	cmd := &cobra.Command{
+		Use:   "create-bls-key [account-address]",
+		Args:  cobra.ExactArgs(1),
+		Short: "Create a pair of BLS keys for a validator",
+		Long: strings.TrimSpace(
+			fmt.Sprintf(`create-bls will create a pair of BLS keys that are used to
+send BLS signatures for checkpointing.
+
+BLS keys are stored along with other validator keys in priv_validator_key.json,
+which should exist before running the command (via babylond init or babylond testnet).
+
+Example:
+$ babylond create-bls-key %s1f5tnl46mk4dfp4nx3n2vnrvyw2h2ydz6ykhk3r --home ./
+`,
+				bech32PrefixAccAddr,
+			),
+		),
+
+		RunE: func(cmd *cobra.Command, args []string) error {
+			homeDir, _ := cmd.Flags().GetString(flags.FlagHome)
+
+			addr, err := sdk.AccAddressFromBech32(args[0])
+			if err != nil {
+				return err
+			}
+
+			return CreateBlsKey(homeDir, addr)
+		},
+	}
+
+	cmd.Flags().String(flags.FlagHome, app.DefaultNodeHome, "The node home directory")
+
+	return cmd
+}
+
+func CreateBlsKey(home string, addr sdk.AccAddress) error {
+	nodeCfg := tmconfig.DefaultConfig()
+	keyPath := filepath.Join(home, nodeCfg.PrivValidatorKeyFile())
+	statePath := filepath.Join(home, nodeCfg.PrivValidatorStateFile())
+	if !tmos.FileExists(keyPath) {
+		return errors.New("validator key file does not exist")
+	}
+	pv := privval.LoadWrappedFilePV(keyPath, statePath)
+	wrappedPV := privval.NewWrappedFilePV(pv.GetValPrivKey(), bls12381.GenPrivKey(), keyPath, statePath)
+	wrappedPV.SetAccAddress(addr)
+
+	return nil
+}

cmd/babylond/cmd/root.go

@@ -140,6 +140,7 @@ func initRootCmd(rootCmd *cobra.Command, encodingConfig params.EncodingConfig) {
 		AddGenesisAccountCmd(app.DefaultNodeHome),
 		tmcli.NewCompletionCmd(rootCmd, true),
 		testnetCmd(app.ModuleBasics, banktypes.GenesisBalancesIterator{}),
+		createBlsKeyCmd(),
 		debug.Cmd(),
 		config.Cmd(),
 	)