package object import ( "context" "fmt" "strconv" "time" objectV2 "github.com/nspcc-dev/neofs-api-go/v2/object" internalclient "github.com/nspcc-dev/neofs-node/cmd/neofs-cli/internal/client" "github.com/nspcc-dev/neofs-node/cmd/neofs-cli/internal/common" "github.com/nspcc-dev/neofs-node/cmd/neofs-cli/internal/commonflags" "github.com/nspcc-dev/neofs-node/cmd/neofs-cli/internal/key" sessionCli "github.com/nspcc-dev/neofs-node/cmd/neofs-cli/modules/session" cid "github.com/nspcc-dev/neofs-sdk-go/container/id" objectSDK "github.com/nspcc-dev/neofs-sdk-go/object" oid "github.com/nspcc-dev/neofs-sdk-go/object/id" "github.com/nspcc-dev/neofs-sdk-go/user" "github.com/spf13/cobra" ) const lockExpiresOnFlag = "expires-on" // object lock command. var objectLockCmd = &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.DecodeString(args[0]) common.ExitOnErr(cmd, "Incorrect container arg: %v", err) argsList := args[1:] lockList := make([]oid.ID, len(argsList)) for i := range argsList { err = lockList[i].DecodeString(argsList[i]) common.ExitOnErr(cmd, fmt.Sprintf("Incorrect object arg #%d: %%v", i+1), err) } key := key.GetOrGenerate(cmd) var idOwner user.ID user.IDFromKey(&idOwner, key.PublicKey) var lock objectSDK.Lock lock.WriteMembers(lockList) exp, relative, err := common.ParseEpoch(cmd, lockExpiresOnFlag) common.ExitOnErr(cmd, "Parsing expiration epoch: %w", err) if relative { ctx, cancel := context.WithTimeout(context.Background(), time.Second*30) defer cancel() endpoint, _ := cmd.Flags().GetString(commonflags.RPC) currEpoch, err := internalclient.GetCurrentEpoch(ctx, endpoint) common.ExitOnErr(cmd, "Request current epoch: %w", err) exp += currEpoch } var expirationAttr objectSDK.Attribute expirationAttr.SetKey(objectV2.SysAttributeExpEpoch) expirationAttr.SetValue(strconv.FormatUint(exp, 10)) obj := objectSDK.New() obj.SetContainerID(cnr) obj.SetOwnerID(&idOwner) obj.SetType(objectSDK.TypeLock) obj.SetAttributes(expirationAttr) obj.SetPayload(lock.Marshal()) var prm internalclient.PutObjectPrm sessionCli.Prepare(cmd, cnr, nil, key, &prm) Prepare(cmd, &prm) prm.SetHeader(obj) res, err := internalclient.PutObject(prm) common.ExitOnErr(cmd, "Store lock object in NeoFS: %w", err) cmd.Printf("Lock object ID: %s\n", res.ID()) cmd.Println("Objects successfully locked.") }, } func initCommandObjectLock() { commonflags.Init(objectLockCmd) commonflags.InitSession(objectLockCmd) objectLockCmd.Flags().StringP(lockExpiresOnFlag, "e", "", "Lock expiration epoch") _ = objectLockCmd.MarkFlagRequired(lockExpiresOnFlag) }