forked from TrueCloudLab/frostfs-node
69 lines
1.8 KiB
Go
69 lines
1.8 KiB
Go
package object
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
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"
|
|
"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"
|
|
)
|
|
|
|
// 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 object.Lock
|
|
lock.WriteMembers(lockList)
|
|
|
|
obj := object.New()
|
|
obj.SetContainerID(cnr)
|
|
obj.SetOwnerID(&idOwner)
|
|
obj.SetType(object.TypeLock)
|
|
obj.SetPayload(lock.Marshal())
|
|
|
|
var prm internalclient.PutObjectPrm
|
|
|
|
sessionCli.Prepare(cmd, cnr, nil, key, &prm)
|
|
Prepare(cmd, &prm)
|
|
prm.SetHeader(obj)
|
|
|
|
_, err = internalclient.PutObject(prm)
|
|
common.ExitOnErr(cmd, "Store lock object in NeoFS: %w", err)
|
|
|
|
cmd.Println("Objects successfully locked.")
|
|
},
|
|
}
|
|
|
|
func initCommandObjectLock() {
|
|
commonflags.Init(objectLockCmd)
|
|
commonflags.InitSession(objectLockCmd)
|
|
}
|