From d92b1d1bf848d69f8e10ab519e7586bc1949e4a5 Mon Sep 17 00:00:00 2001
From: Leonard Lyubich <leonard@nspcc.ru>
Date: Wed, 16 Feb 2022 18:56:16 +0300
Subject: [PATCH] [#1175] CLI: Implement command to lock the objects

Implement `lock` command and add it to `object` section. The command
accepts container argument (string `cid.ID`) and list of locked objects
(string `oid.ID` list). From the provided input `LOCK` object is
constructed and stored using NeoFS API protocol.

Signed-off-by: Leonard Lyubich <leonard@nspcc.ru>
---
 cmd/neofs-cli/modules/lock.go   | 65 +++++++++++++++++++++++++++++++++
 cmd/neofs-cli/modules/object.go |  2 +
 2 files changed, 67 insertions(+)
 create mode 100644 cmd/neofs-cli/modules/lock.go

diff --git a/cmd/neofs-cli/modules/lock.go b/cmd/neofs-cli/modules/lock.go
new file mode 100644
index 00000000..b6ab40de
--- /dev/null
+++ b/cmd/neofs-cli/modules/lock.go
@@ -0,0 +1,65 @@
+package cmd
+
+import (
+	"fmt"
+
+	internalclient "github.com/nspcc-dev/neofs-node/cmd/neofs-cli/internal/client"
+	objectcore "github.com/nspcc-dev/neofs-node/pkg/core/object"
+	cid "github.com/nspcc-dev/neofs-sdk-go/container/id"
+	"github.com/nspcc-dev/neofs-sdk-go/object"
+	oid "github.com/nspcc-dev/neofs-sdk-go/object/id"
+	"github.com/spf13/cobra"
+)
+
+// object lock command.
+var cmdObjectLock = &cobra.Command{
+	Use:   "lock CONTAINER OBJECT...",
+	Short: "Lock object in container",
+	Long:  "Lock object in container",
+	Args:  cobra.MinimumNArgs(2),
+	Run: func(cmd *cobra.Command, args []string) {
+		var cnr cid.ID
+
+		err := cnr.Parse(args[0])
+		exitOnErr(cmd, errf("Incorrect container arg: %v", err))
+
+		argsList := args[1:]
+
+		lockList := make([]oid.ID, len(argsList))
+
+		for i := range argsList {
+			err = lockList[i].Parse(argsList[i])
+			exitOnErr(cmd, errf(fmt.Sprintf("Incorrect object arg #%d: %%v", i+1), err))
+		}
+
+		key, err := getKey()
+		exitOnErr(cmd, errf("can't fetch private key: %w", err))
+
+		idOwner, err := getOwnerID(key)
+		exitOnErr(cmd, err)
+
+		var lock object.Lock
+		lock.WriteMembers(lockList)
+
+		obj := object.NewRaw()
+		obj.SetContainerID(&cnr)
+		obj.SetOwnerID(idOwner)
+		obj.SetType(object.TypeLock)
+		obj.SetPayload(lock.Marshal())
+
+		var prm internalclient.PutObjectPrm
+
+		prepareSessionPrmWithOwner(cmd, objectcore.AddressOf(obj), key, idOwner, &prm)
+		prepareObjectPrm(cmd, &prm)
+		prm.SetHeader(obj.Object())
+
+		_, err = internalclient.PutObject(prm)
+		exitOnErr(cmd, errf("Store lock object in NeoFS: %w", err))
+
+		cmd.Println("Objects successfully locked.")
+	},
+}
+
+func initCommandObjectLock() {
+	initCommonFlags(cmdObjectLock)
+}
diff --git a/cmd/neofs-cli/modules/object.go b/cmd/neofs-cli/modules/object.go
index 67b0cea7..9fe893fa 100644
--- a/cmd/neofs-cli/modules/object.go
+++ b/cmd/neofs-cli/modules/object.go
@@ -246,6 +246,7 @@ func init() {
 		objectHeadCmd,
 		objectHashCmd,
 		objectRangeCmd,
+		cmdObjectLock,
 	}
 
 	rootCmd.AddCommand(objectCmd)
@@ -276,6 +277,7 @@ func init() {
 	initObjectHeadCmd()
 	initObjectHashCmd()
 	initObjectRangeCmd()
+	initCommandObjectLock()
 }
 
 type clientKeySession interface {