add to client + control

Author: felipemadero

client/client.go

@@ -37,6 +37,7 @@ type Client interface {
 	AddPermissionlessValidator(ctx context.Context, validatorSpec []*rpcpb.PermissionlessStakerSpec) (*rpcpb.AddPermissionlessValidatorResponse, error)
 	AddPermissionlessDelegator(ctx context.Context, validatorSpec []*rpcpb.PermissionlessStakerSpec) (*rpcpb.AddPermissionlessDelegatorResponse, error)
 	RemoveSubnetValidator(ctx context.Context, validatorSpec []*rpcpb.RemoveSubnetValidatorSpec) (*rpcpb.RemoveSubnetValidatorResponse, error)
+	AddSubnetValidators(ctx context.Context, validatorSpec []*rpcpb.SubnetValidatorsSpec) (*rpcpb.AddSubnetValidatorsResponse, error)
 	Health(ctx context.Context) (*rpcpb.HealthResponse, error)
 	WaitForHealthy(ctx context.Context) (*rpcpb.WaitForHealthyResponse, error)
 	URIs(ctx context.Context) ([]string, error)
@@ -203,6 +204,15 @@ func (c *client) RemoveSubnetValidator(ctx context.Context, validatorSpec []*rpc
 	return c.controlc.RemoveSubnetValidator(ctx, req)
 }
 
+func (c *client) AddSubnetValidators(ctx context.Context, validatorSpec []*rpcpb.SubnetValidatorsSpec) (*rpcpb.AddSubnetValidatorsResponse, error) {
+	req := &rpcpb.AddSubnetValidatorsRequest{
+		ValidatorsSpec: validatorSpec,
+	}
+
+	c.log.Info("add subnet validators")
+	return c.controlc.AddSubnetValidators(ctx, req)
+}
+
 func (c *client) Health(ctx context.Context) (*rpcpb.HealthResponse, error) {
 	c.log.Info("health")
 	return c.controlc.Health(ctx, &rpcpb.HealthRequest{})

cmd/control/control.go

@@ -63,6 +63,7 @@ func NewCommand() *cobra.Command {
 		newTransformElasticSubnetsCommand(),
 		newAddPermissionlessValidatorCommand(),
 		newAddPermissionlessDelegatorCommand(),
+		newAddSubnetValidatorsCommand(),
 		newRemoveSubnetValidatorCommand(),
 		newHealthCommand(),
 		newWaitForHealthyCommand(),
@@ -444,6 +445,16 @@ func newAddPermissionlessValidatorCommand() *cobra.Command {
 	return cmd
 }
 
+func newAddSubnetValidatorsCommand() *cobra.Command {
+	cmd := &cobra.Command{
+		Use:   "add-subnet-validators validatorsSpec [options]",
+		Short: "Adds subnet validators",
+		RunE:  addSubnetValidatorsFunc,
+		Args:  cobra.ExactArgs(1),
+	}
+	return cmd
+}
+
 func newRemoveSubnetValidatorCommand() *cobra.Command {
 	cmd := &cobra.Command{
 		Use:   "remove-subnet-validator removeValidatorSpec [options]",
@@ -538,6 +549,34 @@ func addPermissionlessValidatorFunc(_ *cobra.Command, args []string) error {
 	return nil
 }
 
+func addSubnetValidatorsFunc(_ *cobra.Command, args []string) error {
+	cli, err := newClient()
+	if err != nil {
+		return err
+	}
+	defer cli.Close()
+
+	validatorSpecsStr := args[0]
+
+	validatorSpecs := []*rpcpb.SubnetValidatorsSpec{}
+	if err := json.Unmarshal([]byte(validatorSpecsStr), &validatorSpecs); err != nil {
+		return err
+	}
+
+	ctx := getAsyncContext()
+
+	info, err := cli.AddSubnetValidators(
+		ctx,
+		validatorSpecs,
+	)
+	if err != nil {
+		return err
+	}
+
+	ux.Print(log, logging.Green.Wrap("add-subnet-validators response: %+v"), info)
+	return nil
+}
+
 func removeSubnetValidatorFunc(_ *cobra.Command, args []string) error {
 	cli, err := newClient()
 	if err != nil {