From 5eef0f46c5458419d8427e03cdc15a7fd63e579d Mon Sep 17 00:00:00 2001
From: Evgenii Stratonikov <evgeniy@nspcc.ru>
Date: Mon, 14 Mar 2022 16:27:22 +0300
Subject: [PATCH] [#1233] neofs-cli: Respect format flags for `SplitInfo`
 output

Signed-off-by: Evgenii Stratonikov <evgeniy@nspcc.ru>
---
 cmd/neofs-cli/modules/object.go | 37 +++++++++++++++++++++++++--------
 1 file changed, 28 insertions(+), 9 deletions(-)

diff --git a/cmd/neofs-cli/modules/object.go b/cmd/neofs-cli/modules/object.go
index 3c8cce178..d2735e3a9 100644
--- a/cmd/neofs-cli/modules/object.go
+++ b/cmd/neofs-cli/modules/object.go
@@ -1,6 +1,7 @@
 package cmd
 
 import (
+	"bytes"
 	"crypto/ecdsa"
 	"encoding/hex"
 	"errors"
@@ -1096,7 +1097,7 @@ func printSplitInfoErr(cmd *cobra.Command, err error) bool {
 	ok := errors.As(err, &errSplitInfo)
 
 	if ok {
-		cmd.Println("Object is complex, split information received.")
+		cmd.PrintErrln("Object is complex, split information received.")
 		printSplitInfo(cmd, errSplitInfo.SplitInfo())
 	}
 
@@ -1104,15 +1105,33 @@ func printSplitInfoErr(cmd *cobra.Command, err error) bool {
 }
 
 func printSplitInfo(cmd *cobra.Command, info *object.SplitInfo) {
-	if splitID := info.SplitID(); splitID != nil {
-		cmd.Println("Split ID:", splitID)
-	}
+	bs, err := marshalSplitInfo(cmd, info)
+	exitOnErr(cmd, fmt.Errorf("can't marshal split info: %w", err))
 
-	if link := info.Link(); link != nil {
-		cmd.Println("Linking object:", link)
-	}
+	cmd.Println(string(bs))
+}
 
-	if last := info.LastPart(); last != nil {
-		cmd.Println("Last object:", last)
+func marshalSplitInfo(cmd *cobra.Command, info *object.SplitInfo) ([]byte, error) {
+	toJSON, _ := cmd.Flags().GetBool("json")
+	toProto, _ := cmd.Flags().GetBool("proto")
+	switch {
+	case toJSON && toProto:
+		return nil, errors.New("'--json' and '--proto' flags are mutually exclusive")
+	case toJSON:
+		return info.MarshalJSON()
+	case toProto:
+		return info.Marshal()
+	default:
+		b := bytes.NewBuffer(nil)
+		if splitID := info.SplitID(); splitID != nil {
+			b.WriteString("Split ID: " + splitID.String() + "\n")
+		}
+		if link := info.Link(); link != nil {
+			b.WriteString("Linking object: " + link.String() + "\n")
+		}
+		if last := info.LastPart(); last != nil {
+			b.WriteString("Last object: " + last.String() + "\n")
+		}
+		return b.Bytes(), nil
 	}
 }