cmd: add create dkg command

cmd/cmd.go

@@ -48,6 +48,9 @@ func New() *cobra.Command {
 		newBootnodeCmd(RunBootnode),
 		newCreateClusterCmd(runCreateCluster),
 		newDKGCmd(dkg.Run),
+		newCreateCmd(
+			newCreateDKGCmd(runCreateDKG),
+		),
 	)
 }
 

cmd/create.go

@@ -0,0 +1,31 @@
+// Copyright © 2022 Obol Labs Inc.
+//
+// This program is free software: you can redistribute it and/or modify it
+// under the terms of the GNU General Public License as published by the Free
+// Software Foundation, either version 3 of the License, or (at your option)
+// any later version.
+//
+// This program is distributed in the hope that it will be useful, but WITHOUT
+// ANY WARRANTY; without even the implied warranty of  MERCHANTABILITY or
+// FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for
+// more details.
+//
+// You should have received a copy of the GNU General Public License along with
+// this program.  If not, see <http://www.gnu.org/licenses/>.
+
+package cmd
+
+import "github.com/spf13/cobra"
+
+func newCreateCmd(cmds ...*cobra.Command) *cobra.Command {
+	root := &cobra.Command{
+		Use:   "create",
+		Short: "Create different artifacts required for running a Charon cluster",
+	}
+
+	root.AddCommand(cmds...)
+
+	titledHelp(root)
+
+	return root
+}

cmd/createdkg.go

@@ -0,0 +1,97 @@
+// Copyright © 2022 Obol Labs Inc.
+//
+// This program is free software: you can redistribute it and/or modify it
+// under the terms of the GNU General Public License as published by the Free
+// Software Foundation, either version 3 of the License, or (at your option)
+// any later version.
+//
+// This program is distributed in the hope that it will be useful, but WITHOUT
+// ANY WARRANTY; without even the implied warranty of  MERCHANTABILITY or
+// FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for
+// more details.
+//
+// You should have received a copy of the GNU General Public License along with
+// this program.  If not, see <http://www.gnu.org/licenses/>.
+
+package cmd
+
+import (
+	"context"
+	crand "crypto/rand"
+	"encoding/json"
+	"os"
+	"path"
+
+	"github.com/spf13/cobra"
+	"github.com/spf13/pflag"
+
+	"github.com/obolnetwork/charon/app/errors"
+	"github.com/obolnetwork/charon/cluster"
+)
+
+type createDKGConfig struct {
+	OutputDir         string
+	Name              string
+	NumValidators     int
+	Threshold         int
+	FeeRecipient      string
+	WithdrawalAddress string
+	ForkVersion       string
+	OperatorENRs      []string
+}
+
+func newCreateDKGCmd(runFunc func(context.Context, createDKGConfig) error) *cobra.Command {
+	var config createDKGConfig
+
+	cmd := &cobra.Command{
+		Use:   "dkg",
+		Short: "Configure DKG by creating a cluster definition file",
+		Long:  `Create a cluster definition file that will be used by all participants of a DKG.`,
+		Args:  cobra.NoArgs,
+		RunE: func(cmd *cobra.Command, args []string) error {
+			return runFunc(cmd.Context(), config)
+		},
+	}
+
+	bindCreateDKGFlags(cmd.Flags(), &config)
+
+	return cmd
+}
+
+func bindCreateDKGFlags(flags *pflag.FlagSet, config *createDKGConfig) {
+	flags.StringVar(&config.Name, "name", "", "Optional cosmetic cluster name")
+	flags.StringVar(&config.OutputDir, "output-dir", ".", "The folder to write the output cluster_definition.json file to.")
+	flags.IntVar(&config.NumValidators, "num-validators", 1, "The number of distributed validators the cluster will manage (32ETH staked for each).")
+	flags.IntVarP(&config.Threshold, "threshold", "t", 3, "The threshold required for signature reconstruction. Minimum is n-(ceil(n/3)-1).")
+	flags.StringVar(&config.FeeRecipient, "fee_recipient_address", "", "Optional Ethereum address of the fee recipient")
+	flags.StringVar(&config.WithdrawalAddress, "withdrawal_address", "", "Withdrawal Ethereum address")
+	flags.StringVar(&config.ForkVersion, "fork_version", "", "Optional hex fork version identifying the target network/chain")
+	flags.StringSliceVar(&config.OperatorENRs, "operator_enrs", nil, "Comma-separated list of each operator's Charon ENR address")
+}
+
+func runCreateDKG(_ context.Context, conf createDKGConfig) error {
+	var operators []cluster.Operator
+	for _, opENR := range conf.OperatorENRs {
+		operators = append(operators, cluster.Operator{
+			ENR: opENR,
+		})
+	}
+
+	def := cluster.NewDefinition(conf.Name, conf.NumValidators, conf.Threshold, conf.FeeRecipient, conf.WithdrawalAddress,
+		conf.ForkVersion, operators, crand.Reader)
+
+	b, err := json.Marshal(def)
+	if err != nil {
+		return errors.Wrap(err, "marshal definition")
+	}
+
+	if err := os.MkdirAll(conf.OutputDir, 0o755); err != nil {
+		return errors.Wrap(err, "create output dir")
+	}
+
+	if err := os.WriteFile(path.Join(conf.OutputDir, "cluster_definition.json"), b, 0o400); err != nil {
+		return errors.Wrap(err, "write definition")
+	}
+
+	return nil
+}