From 0690a54299c08adfdc0589799e632e7f567f951c Mon Sep 17 00:00:00 2001
From: Leonard Lyubich <leonard@nspcc.ru>
Date: Fri, 15 Jan 2021 14:55:06 +0300
Subject: [PATCH] [#315] cmd/cli: Add set-status command to control section

Add set-status command that uses SetNetmapStatus rpc of Control API.

Signed-off-by: Leonard Lyubich <leonard@nspcc.ru>
---
 cmd/neofs-cli/modules/control.go | 83 +++++++++++++++++++++++++++++++-
 1 file changed, 81 insertions(+), 2 deletions(-)

diff --git a/cmd/neofs-cli/modules/control.go b/cmd/neofs-cli/modules/control.go
index 94232c379..f72c56792 100644
--- a/cmd/neofs-cli/modules/control.go
+++ b/cmd/neofs-cli/modules/control.go
@@ -2,11 +2,13 @@ package cmd
 
 import (
 	"context"
+	"fmt"
 
 	"github.com/nspcc-dev/neofs-api-go/util/signature"
 	"github.com/nspcc-dev/neofs-api-go/v2/client"
 	"github.com/nspcc-dev/neofs-node/pkg/services/control"
 	controlSvc "github.com/nspcc-dev/neofs-node/pkg/services/control/server"
+	"github.com/pkg/errors"
 	"github.com/spf13/cobra"
 )
 
@@ -23,12 +25,38 @@ var healthCheckCmd = &cobra.Command{
 	RunE:  healthCheck,
 }
 
+var setNetmapStatusCmd = &cobra.Command{
+	Use:   "set-status",
+	Short: "Set status of the storage node in NeoFS network map",
+	Long:  "Set status of the storage node in NeoFS network map",
+	RunE:  setNetmapStatus,
+}
+
+const (
+	netmapStatusFlag = "status"
+
+	netmapStatusOnline  = "online"
+	netmapStatusOffline = "offline"
+)
+
+var netmapStatus string
+
 func init() {
 	rootCmd.AddCommand(controlCmd)
 
-	controlCmd.AddCommand(healthCheckCmd)
+	controlCmd.AddCommand(
+		healthCheckCmd,
+		setNetmapStatusCmd,
+	)
 
-	controlCmd.AddCommand(snapshotCmd)
+	setNetmapStatusCmd.Flags().StringVarP(&netmapStatus, netmapStatusFlag, "", "",
+		fmt.Sprintf("new netmap status keyword ('%s', '%s')",
+			netmapStatusOnline,
+			netmapStatusOffline,
+		),
+	)
+
+	_ = setNetmapStatusCmd.MarkFlagRequired(netmapStatusFlag)
 }
 
 func getControlServiceClient() (control.ControlServiceClient, error) {
@@ -89,3 +117,54 @@ func healthCheck(cmd *cobra.Command, _ []string) error {
 
 	return nil
 }
+
+func setNetmapStatus(cmd *cobra.Command, _ []string) error {
+	key, err := getKey()
+	if err != nil {
+		return err
+	}
+
+	var status control.NetmapStatus
+
+	switch netmapStatus {
+	default:
+		return errors.Errorf("unsupported status %s", netmapStatus)
+	case netmapStatusOnline:
+		status = control.NetmapStatus_ONLINE
+	case netmapStatusOffline:
+		status = control.NetmapStatus_OFFLINE
+	}
+
+	req := new(control.SetNetmapStatusRequest)
+
+	body := new(control.SetNetmapStatusRequest_Body)
+	req.SetBody(body)
+
+	body.SetStatus(status)
+
+	if err := controlSvc.SignMessage(key, req); err != nil {
+		return err
+	}
+
+	cli, err := getControlServiceClient()
+	if err != nil {
+		return err
+	}
+
+	resp, err := cli.SetNetmapStatus(context.Background(), req)
+	if err != nil {
+		return err
+	}
+
+	sign := resp.GetSignature()
+
+	if err := signature.VerifyDataWithSource(resp, func() ([]byte, []byte) {
+		return sign.GetKey(), sign.GetSign()
+	}); err != nil {
+		return err
+	}
+
+	cmd.Println("Network status update request successfully sent.")
+
+	return nil
+}