From feac279cede5cdbf081b1e19722b963cfedc67e4 Mon Sep 17 00:00:00 2001
From: Alex Vanin <alexey@nspcc.ru>
Date: Thu, 15 Oct 2020 18:00:53 +0300
Subject: [PATCH] [#25] Add container.get command in CLI

Signed-off-by: Alex Vanin <alexey@nspcc.ru>
---
 cmd/neofs-cli/modules/container.go | 112 +++++++++++++++++++++++++++++
 1 file changed, 112 insertions(+)

diff --git a/cmd/neofs-cli/modules/container.go b/cmd/neofs-cli/modules/container.go
index bde003bf3..ca28679dc 100644
--- a/cmd/neofs-cli/modules/container.go
+++ b/cmd/neofs-cli/modules/container.go
@@ -19,6 +19,7 @@ import (
 	"github.com/nspcc-dev/neofs-api-go/pkg/object"
 	"github.com/nspcc-dev/neofs-api-go/pkg/owner"
 	v2container "github.com/nspcc-dev/neofs-api-go/v2/container"
+	grpccontainer "github.com/nspcc-dev/neofs-api-go/v2/container/grpc"
 	"github.com/nspcc-dev/neofs-node/pkg/policy"
 	"github.com/spf13/cobra"
 )
@@ -39,6 +40,9 @@ var (
 	containerAwait      bool
 
 	containerID string
+
+	containerPathFrom string
+	containerPathTo   string
 )
 
 // containerCmd represents the container command
@@ -242,12 +246,74 @@ var listContainerObjectsCmd = &cobra.Command{
 	},
 }
 
+var getContainerInfoCmd = &cobra.Command{
+	Use:   "get",
+	Short: "Get container field info",
+	Long:  `Get container field info`,
+	RunE: func(cmd *cobra.Command, args []string) error {
+		var (
+			cnr *container.Container
+
+			ctx = context.Background()
+		)
+
+		if containerPathFrom != "" {
+			data, err := ioutil.ReadFile(containerPathFrom)
+			if err != nil {
+				return fmt.Errorf("can't read file: %w", err)
+			}
+
+			// todo: make more user friendly way to parse raw data
+			msg := new(grpccontainer.Container)
+			if msg.Unmarshal(data) != nil {
+				return errors.New("can't unmarshal container")
+			}
+
+			v2cnr := v2container.ContainerFromGRPCMessage(msg)
+
+			cnr = container.NewContainerFromV2(v2cnr)
+		} else {
+			cli, err := getSDKClient()
+			if err != nil {
+				return err
+			}
+
+			id, err := parseContainerID(containerID)
+			if err != nil {
+				return err
+			}
+
+			cnr, err = cli.GetContainer(ctx, id)
+			if err != nil {
+				return fmt.Errorf("rpc error: %w", err)
+			}
+		}
+
+		prettyPrintContainer(cnr)
+
+		if containerPathTo != "" {
+			data, err := cnr.ToV2().StableMarshal(nil)
+			if err != nil {
+				return errors.New("can't marshal container")
+			}
+
+			err = ioutil.WriteFile(containerPathTo, data, 0644)
+			if err != nil {
+				return fmt.Errorf("can't write container to file: %w", err)
+			}
+		}
+
+		return nil
+	},
+}
+
 func init() {
 	rootCmd.AddCommand(containerCmd)
 	containerCmd.AddCommand(listContainersCmd)
 	containerCmd.AddCommand(createContainerCmd)
 	containerCmd.AddCommand(deleteContainerCmd)
 	containerCmd.AddCommand(listContainerObjectsCmd)
+	containerCmd.AddCommand(getContainerInfoCmd)
 
 	// Here you will define your flags and configuration settings.
 
@@ -278,6 +344,11 @@ func init() {
 
 	// container list-object
 	listContainerObjectsCmd.Flags().StringVar(&containerID, "cid", "", "container ID")
+
+	// container get
+	getContainerInfoCmd.Flags().StringVar(&containerID, "cid", "", "container ID")
+	getContainerInfoCmd.Flags().StringVar(&containerPathTo, "to", "", "path to dump binary encoded container")
+	getContainerInfoCmd.Flags().StringVar(&containerPathFrom, "from", "", "path to file with binary encoded container")
 }
 
 func prettyPrintContainerList(list []*container.ID) {
@@ -378,3 +449,44 @@ func parseContainerID(cid string) (*container.ID, error) {
 
 	return id, nil
 }
+
+func prettyPrintContainer(cnr *container.Container) {
+	if cnr == nil {
+		return
+	}
+
+	id := container.CalculateID(cnr)
+	fmt.Println("container ID:", id)
+
+	version := cnr.GetVersion()
+	fmt.Printf("version: %d.%d\n", version.GetMajor(), version.GetMinor())
+
+	// todo: return pkg structures instead of v2 structures
+	ownerID := owner.NewIDFromV2(cnr.GetOwnerID())
+	fmt.Println("owner ID:", ownerID)
+
+	basicACL := cnr.GetBasicACL()
+	fmt.Printf("basic ACL: %s", strconv.FormatUint(uint64(basicACL), 16))
+	switch basicACL {
+	case acl.PublicBasicRule:
+		fmt.Println(" (public)")
+	case acl.PrivateBasicRule:
+		fmt.Println(" (private)")
+	case acl.ReadOnlyBasicRule:
+		fmt.Println(" (readonly)")
+	default:
+		fmt.Println()
+	}
+
+	for _, attribute := range cnr.GetAttributes() {
+		fmt.Printf("attribute: `%s:%s`\n", attribute.GetKey(), attribute.GetValue())
+	}
+
+	nonce, err := uuid.FromBytes(cnr.GetNonce())
+	if err == nil {
+		fmt.Println("nonce:", nonce)
+	}
+
+	fmt.Println("placement policy:")
+	fmt.Println(strings.Join(policy.Encode(cnr.GetPlacementPolicy()), "\n"))
+}