From ea36789b51c2b89c11bf17a97001565f5ea03121 Mon Sep 17 00:00:00 2001
From: Leonard Lyubich <leonard@nspcc.ru>
Date: Fri, 20 Nov 2020 11:56:21 +0300
Subject: [PATCH] [#179] cmd/neofs-cli: Implement range command in Object
 section

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

diff --git a/cmd/neofs-cli/modules/object.go b/cmd/neofs-cli/modules/object.go
index 12d5cd030..7ca275708 100644
--- a/cmd/neofs-cli/modules/object.go
+++ b/cmd/neofs-cli/modules/object.go
@@ -22,6 +22,14 @@ import (
 	"github.com/spf13/cobra"
 )
 
+const (
+	getRangeCmdUse = "range"
+
+	getRangeCmdShortDesc = "Get payload range data of an object"
+
+	getRangeCmdLongDesc = "Get payload range data of an object"
+)
+
 var (
 	// objectCmd represents the object command
 	objectCmd = &cobra.Command{
@@ -72,6 +80,13 @@ var (
 		Long:  "Get object hash",
 		RunE:  getObjectHash,
 	}
+
+	objectRangeCmd = &cobra.Command{
+		Use:   getRangeCmdUse,
+		Short: getRangeCmdShortDesc,
+		Long:  getRangeCmdLongDesc,
+		RunE:  getObjectRange,
+	}
 )
 
 const (
@@ -136,6 +151,14 @@ func init() {
 	objectHashCmd.Flags().String("range", "", "Range to take hash from in the form offset1:length1,...")
 	objectHashCmd.Flags().String("type", hashSha256, "Hash type. Either 'sha256' or 'tz'")
 
+	objectCmd.AddCommand(objectRangeCmd)
+	objectRangeCmd.Flags().String("cid", "", "Container ID")
+	_ = objectRangeCmd.MarkFlagRequired("cid")
+	objectRangeCmd.Flags().String("oid", "", "Object ID")
+	_ = objectRangeCmd.MarkFlagRequired("oid")
+	objectRangeCmd.Flags().String("range", "", "Range to take data from in the form offset:length")
+	objectRangeCmd.Flags().String("file", "", "File to write object payload to. Default: stdout.")
+
 	// Here you will define your flags and configuration settings.
 
 	// Cobra supports Persistent Flags which will work for this command
@@ -672,3 +695,63 @@ func getBearerToken(cmd *cobra.Command, flagname string) (*token.BearerToken, er
 
 	return tok, nil
 }
+
+func getObjectRange(cmd *cobra.Command, _ []string) error {
+	objAddr, err := getObjectAddress(cmd)
+	if err != nil {
+		return err
+	}
+
+	ranges, err := getRangeList(cmd)
+	if err != nil {
+		return err
+	} else if len(ranges) != 1 {
+		return errors.New("exactly one range must be specified")
+	}
+
+	var out io.Writer
+
+	filename := cmd.Flag("file").Value.String()
+	if filename == "" {
+		out = os.Stdout
+	} else {
+		f, err := os.OpenFile(filename, os.O_CREATE|os.O_WRONLY, os.ModePerm)
+		if err != nil {
+			return fmt.Errorf("can't open file '%s': %w", filename, err)
+		}
+
+		defer f.Close()
+
+		out = f
+	}
+
+	ctx := context.Background()
+
+	c, sessionToken, err := initSession(ctx)
+	if err != nil {
+		return err
+	}
+
+	bearerToken, err := getBearerToken(cmd, "bearer")
+	if err != nil {
+		return err
+	}
+
+	_, err = c.ObjectPayloadRangeData(ctx,
+		new(client.RangeDataParams).
+			WithAddress(objAddr).
+			WithRange(ranges[0]).
+			WithDataWriter(out),
+		client.WithTTL(getTTL()),
+		client.WithSession(sessionToken),
+		client.WithBearer(bearerToken))
+	if err != nil {
+		return fmt.Errorf("can't get object payload range: %w", err)
+	}
+
+	if filename != "" {
+		cmd.Printf("[%s] Payload successfully saved\n", filename)
+	}
+
+	return nil
+}