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 +}