frostfs-node/pkg/morph/client/container/wrapper/eacl.go
Alex Vanin 3038c6cc5c [#37] Implement eACL related morph wrapper functions
Signed-off-by: Alex Vanin <alexey@nspcc.ru>
2020-10-02 11:25:36 +03:00

68 lines
1.7 KiB
Go

package wrapper
import (
"github.com/nspcc-dev/neofs-api-go/pkg/acl/eacl"
"github.com/nspcc-dev/neofs-api-go/pkg/container"
v2ACL "github.com/nspcc-dev/neofs-api-go/v2/acl"
msgACL "github.com/nspcc-dev/neofs-api-go/v2/acl/grpc"
client "github.com/nspcc-dev/neofs-node/pkg/morph/client/container"
"github.com/pkg/errors"
)
// GetEACL reads the extended ACL table from NeoFS system
// through Container contract call.
func (w *Wrapper) GetEACL(cid *container.ID) (*eacl.Table, []byte, error) {
if cid == nil {
return nil, nil, errNilArgument
}
args := client.EACLArgs{}
if v2 := cid.ToV2(); v2 == nil {
return nil, nil, errUnsupported // use other major version if there any
} else {
args.SetCID(v2.GetValue())
}
rpcAnswer, err := w.client.EACL(args)
if err != nil {
return nil, nil, err
}
grpcMsg := new(msgACL.EACLTable)
err = grpcMsg.Unmarshal(rpcAnswer.EACL())
if err != nil {
// use other major version if there any
return nil, nil, err
}
v2table := v2ACL.TableFromGRPCMessage(grpcMsg)
return eacl.NewTableFromV2(v2table), rpcAnswer.Signature(), nil
}
// PutEACL saves the extended ACL table in NeoFS system
// through Container contract call.
//
// Returns any error encountered that caused the saving to interrupt.
func (w *Wrapper) PutEACL(table *eacl.Table, signature []byte) error {
if table == nil || len(signature) == 0 {
return errNilArgument
}
args := client.SetEACLArgs{}
args.SetSignature(signature)
if v2 := table.ToV2(); v2 == nil {
return errUnsupported // use other major version if there any
} else {
data, err := v2.StableMarshal(nil)
if err != nil {
return errors.Wrap(err, "can't marshal eacl table")
}
args.SetEACL(data)
}
return w.client.SetEACL(args)
}