forked from TrueCloudLab/frostfs-node
[#1090] ape: Move ape request and resource implementations to common package
Signed-off-by: Airat Arifullin <a.arifullin@yadro.com>
This commit is contained in:
parent
661faa639e
commit
f4dcb418f2
5 changed files with 144 additions and 114 deletions
44
pkg/ape/converter/converter.go
Normal file
44
pkg/ape/converter/converter.go
Normal file
|
@ -0,0 +1,44 @@
|
||||||
|
package converter
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
|
||||||
|
"git.frostfs.info/TrueCloudLab/frostfs-sdk-go/container/acl"
|
||||||
|
nativeschema "git.frostfs.info/TrueCloudLab/policy-engine/schema/native"
|
||||||
|
)
|
||||||
|
|
||||||
|
func SchemaRoleFromACLRole(role acl.Role) (string, error) {
|
||||||
|
switch role {
|
||||||
|
case acl.RoleOwner:
|
||||||
|
return nativeschema.PropertyValueContainerRoleOwner, nil
|
||||||
|
case acl.RoleContainer:
|
||||||
|
return nativeschema.PropertyValueContainerRoleContainer, nil
|
||||||
|
case acl.RoleInnerRing:
|
||||||
|
return nativeschema.PropertyValueContainerRoleIR, nil
|
||||||
|
case acl.RoleOthers:
|
||||||
|
return nativeschema.PropertyValueContainerRoleOthers, nil
|
||||||
|
default:
|
||||||
|
return "", fmt.Errorf("failed to convert %s", role.String())
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func SchemaMethodFromACLOperation(op acl.Op) (string, error) {
|
||||||
|
switch op {
|
||||||
|
case acl.OpObjectGet:
|
||||||
|
return nativeschema.MethodGetObject, nil
|
||||||
|
case acl.OpObjectHead:
|
||||||
|
return nativeschema.MethodHeadObject, nil
|
||||||
|
case acl.OpObjectPut:
|
||||||
|
return nativeschema.MethodPutObject, nil
|
||||||
|
case acl.OpObjectDelete:
|
||||||
|
return nativeschema.MethodDeleteObject, nil
|
||||||
|
case acl.OpObjectSearch:
|
||||||
|
return nativeschema.MethodSearchObject, nil
|
||||||
|
case acl.OpObjectRange:
|
||||||
|
return nativeschema.MethodRangeObject, nil
|
||||||
|
case acl.OpObjectHash:
|
||||||
|
return nativeschema.MethodHashObject, nil
|
||||||
|
default:
|
||||||
|
return "", fmt.Errorf("operation cannot be converted: %d", op)
|
||||||
|
}
|
||||||
|
}
|
55
pkg/ape/request/request.go
Normal file
55
pkg/ape/request/request.go
Normal file
|
@ -0,0 +1,55 @@
|
||||||
|
package ape
|
||||||
|
|
||||||
|
import (
|
||||||
|
aperesource "git.frostfs.info/TrueCloudLab/policy-engine/pkg/resource"
|
||||||
|
)
|
||||||
|
|
||||||
|
type Request struct {
|
||||||
|
operation string
|
||||||
|
resource Resource
|
||||||
|
properties map[string]string
|
||||||
|
}
|
||||||
|
|
||||||
|
func NewRequest(operation string, resource Resource, properties map[string]string) Request {
|
||||||
|
return Request{
|
||||||
|
operation: operation,
|
||||||
|
resource: resource,
|
||||||
|
properties: properties,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
var _ aperesource.Request = Request{}
|
||||||
|
|
||||||
|
func (r Request) Operation() string {
|
||||||
|
return r.operation
|
||||||
|
}
|
||||||
|
|
||||||
|
func (r Request) Property(key string) string {
|
||||||
|
return r.properties[key]
|
||||||
|
}
|
||||||
|
|
||||||
|
func (r Request) Resource() aperesource.Resource {
|
||||||
|
return r.resource
|
||||||
|
}
|
||||||
|
|
||||||
|
type Resource struct {
|
||||||
|
name string
|
||||||
|
properties map[string]string
|
||||||
|
}
|
||||||
|
|
||||||
|
var _ aperesource.Resource = Resource{}
|
||||||
|
|
||||||
|
func NewResource(name string, properties map[string]string) Resource {
|
||||||
|
return Resource{
|
||||||
|
name: name,
|
||||||
|
properties: properties,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (r Resource) Name() string {
|
||||||
|
return r.name
|
||||||
|
}
|
||||||
|
|
||||||
|
func (r Resource) Property(key string) string {
|
||||||
|
return r.properties[key]
|
||||||
|
}
|
|
@ -15,6 +15,7 @@ import (
|
||||||
"git.frostfs.info/TrueCloudLab/frostfs-api-go/v2/refs"
|
"git.frostfs.info/TrueCloudLab/frostfs-api-go/v2/refs"
|
||||||
session "git.frostfs.info/TrueCloudLab/frostfs-api-go/v2/session"
|
session "git.frostfs.info/TrueCloudLab/frostfs-api-go/v2/session"
|
||||||
"git.frostfs.info/TrueCloudLab/frostfs-contract/frostfsid/client"
|
"git.frostfs.info/TrueCloudLab/frostfs-contract/frostfsid/client"
|
||||||
|
aperequest "git.frostfs.info/TrueCloudLab/frostfs-node/pkg/ape/request"
|
||||||
containercore "git.frostfs.info/TrueCloudLab/frostfs-node/pkg/core/container"
|
containercore "git.frostfs.info/TrueCloudLab/frostfs-node/pkg/core/container"
|
||||||
"git.frostfs.info/TrueCloudLab/frostfs-node/pkg/core/netmap"
|
"git.frostfs.info/TrueCloudLab/frostfs-node/pkg/core/netmap"
|
||||||
"git.frostfs.info/TrueCloudLab/frostfs-observability/tracing"
|
"git.frostfs.info/TrueCloudLab/frostfs-observability/tracing"
|
||||||
|
@ -26,7 +27,6 @@ import (
|
||||||
"git.frostfs.info/TrueCloudLab/frostfs-sdk-go/user"
|
"git.frostfs.info/TrueCloudLab/frostfs-sdk-go/user"
|
||||||
apechain "git.frostfs.info/TrueCloudLab/policy-engine/pkg/chain"
|
apechain "git.frostfs.info/TrueCloudLab/policy-engine/pkg/chain"
|
||||||
policyengine "git.frostfs.info/TrueCloudLab/policy-engine/pkg/engine"
|
policyengine "git.frostfs.info/TrueCloudLab/policy-engine/pkg/engine"
|
||||||
"git.frostfs.info/TrueCloudLab/policy-engine/pkg/resource"
|
|
||||||
nativeschema "git.frostfs.info/TrueCloudLab/policy-engine/schema/native"
|
nativeschema "git.frostfs.info/TrueCloudLab/policy-engine/schema/native"
|
||||||
"github.com/nspcc-dev/neo-go/pkg/crypto/keys"
|
"github.com/nspcc-dev/neo-go/pkg/crypto/keys"
|
||||||
"github.com/nspcc-dev/neo-go/pkg/util"
|
"github.com/nspcc-dev/neo-go/pkg/util"
|
||||||
|
@ -148,14 +148,14 @@ func (ac *apeChecker) List(ctx context.Context, req *container.ListRequest) (*co
|
||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
|
|
||||||
request := &apeRequest{
|
request := aperequest.NewRequest(
|
||||||
resource: &apeResource{
|
nativeschema.MethodListContainers,
|
||||||
name: resourceName(namespace, ""),
|
aperequest.NewResource(
|
||||||
props: make(map[string]string),
|
resourceName(namespace, ""),
|
||||||
},
|
make(map[string]string),
|
||||||
op: nativeschema.MethodListContainers,
|
),
|
||||||
props: reqProps,
|
reqProps,
|
||||||
}
|
)
|
||||||
|
|
||||||
s, found, err := ac.router.IsAllowed(apechain.Ingress,
|
s, found, err := ac.router.IsAllowed(apechain.Ingress,
|
||||||
policyengine.NewRequestTargetWithNamespace(namespace),
|
policyengine.NewRequestTargetWithNamespace(namespace),
|
||||||
|
@ -193,14 +193,14 @@ func (ac *apeChecker) Put(ctx context.Context, req *container.PutRequest) (*cont
|
||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
|
|
||||||
request := &apeRequest{
|
request := aperequest.NewRequest(
|
||||||
resource: &apeResource{
|
nativeschema.MethodPutContainer,
|
||||||
name: resourceName(namespace, ""),
|
aperequest.NewResource(
|
||||||
props: make(map[string]string),
|
resourceName(namespace, ""),
|
||||||
},
|
make(map[string]string),
|
||||||
op: nativeschema.MethodPutContainer,
|
),
|
||||||
props: reqProps,
|
reqProps,
|
||||||
}
|
)
|
||||||
|
|
||||||
s, found, err := ac.router.IsAllowed(apechain.Ingress,
|
s, found, err := ac.router.IsAllowed(apechain.Ingress,
|
||||||
policyengine.NewRequestTargetWithNamespace(namespace),
|
policyengine.NewRequestTargetWithNamespace(namespace),
|
||||||
|
@ -288,14 +288,14 @@ func (ac *apeChecker) validateContainerBoundedOperation(containerID *refs.Contai
|
||||||
namespace = cntNamespace
|
namespace = cntNamespace
|
||||||
}
|
}
|
||||||
|
|
||||||
request := &apeRequest{
|
request := aperequest.NewRequest(
|
||||||
resource: &apeResource{
|
op,
|
||||||
name: resourceName(namespace, id.EncodeToString()),
|
aperequest.NewResource(
|
||||||
props: ac.getContainerProps(cont),
|
resourceName(namespace, id.EncodeToString()),
|
||||||
},
|
ac.getContainerProps(cont),
|
||||||
op: op,
|
),
|
||||||
props: reqProps,
|
reqProps,
|
||||||
}
|
)
|
||||||
|
|
||||||
s, found, err := ac.router.IsAllowed(apechain.Ingress,
|
s, found, err := ac.router.IsAllowed(apechain.Ingress,
|
||||||
policyengine.NewRequestTarget(namespace, id.EncodeToString()),
|
policyengine.NewRequestTarget(namespace, id.EncodeToString()),
|
||||||
|
@ -329,40 +329,6 @@ func getContainerID(reqContID *refs.ContainerID) (cid.ID, error) {
|
||||||
return id, nil
|
return id, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
type apeRequest struct {
|
|
||||||
resource *apeResource
|
|
||||||
op string
|
|
||||||
props map[string]string
|
|
||||||
}
|
|
||||||
|
|
||||||
// Operation implements resource.Request.
|
|
||||||
func (r *apeRequest) Operation() string {
|
|
||||||
return r.op
|
|
||||||
}
|
|
||||||
|
|
||||||
// Property implements resource.Request.
|
|
||||||
func (r *apeRequest) Property(key string) string {
|
|
||||||
return r.props[key]
|
|
||||||
}
|
|
||||||
|
|
||||||
// Resource implements resource.Request.
|
|
||||||
func (r *apeRequest) Resource() resource.Resource {
|
|
||||||
return r.resource
|
|
||||||
}
|
|
||||||
|
|
||||||
type apeResource struct {
|
|
||||||
name string
|
|
||||||
props map[string]string
|
|
||||||
}
|
|
||||||
|
|
||||||
func (r *apeResource) Name() string {
|
|
||||||
return r.name
|
|
||||||
}
|
|
||||||
|
|
||||||
func (r *apeResource) Property(key string) string {
|
|
||||||
return r.props[key]
|
|
||||||
}
|
|
||||||
|
|
||||||
func resourceName(namespace string, container string) string {
|
func resourceName(namespace string, container string) string {
|
||||||
if namespace == "" && container == "" {
|
if namespace == "" && container == "" {
|
||||||
return nativeschema.ResourceFormatRootContainers
|
return nativeschema.ResourceFormatRootContainers
|
||||||
|
|
|
@ -6,51 +6,16 @@ import (
|
||||||
"strconv"
|
"strconv"
|
||||||
|
|
||||||
objectV2 "git.frostfs.info/TrueCloudLab/frostfs-api-go/v2/object"
|
objectV2 "git.frostfs.info/TrueCloudLab/frostfs-api-go/v2/object"
|
||||||
|
aperequest "git.frostfs.info/TrueCloudLab/frostfs-node/pkg/ape/request"
|
||||||
"git.frostfs.info/TrueCloudLab/frostfs-sdk-go/container/acl"
|
"git.frostfs.info/TrueCloudLab/frostfs-sdk-go/container/acl"
|
||||||
cid "git.frostfs.info/TrueCloudLab/frostfs-sdk-go/container/id"
|
cid "git.frostfs.info/TrueCloudLab/frostfs-sdk-go/container/id"
|
||||||
objectSDK "git.frostfs.info/TrueCloudLab/frostfs-sdk-go/object"
|
objectSDK "git.frostfs.info/TrueCloudLab/frostfs-sdk-go/object"
|
||||||
oid "git.frostfs.info/TrueCloudLab/frostfs-sdk-go/object/id"
|
oid "git.frostfs.info/TrueCloudLab/frostfs-sdk-go/object/id"
|
||||||
"git.frostfs.info/TrueCloudLab/frostfs-sdk-go/user"
|
"git.frostfs.info/TrueCloudLab/frostfs-sdk-go/user"
|
||||||
aperesource "git.frostfs.info/TrueCloudLab/policy-engine/pkg/resource"
|
|
||||||
nativeschema "git.frostfs.info/TrueCloudLab/policy-engine/schema/native"
|
nativeschema "git.frostfs.info/TrueCloudLab/policy-engine/schema/native"
|
||||||
)
|
)
|
||||||
|
|
||||||
type request struct {
|
var defaultRequest = aperequest.Request{}
|
||||||
operation string
|
|
||||||
resource resource
|
|
||||||
properties map[string]string
|
|
||||||
}
|
|
||||||
|
|
||||||
var defaultRequest = request{}
|
|
||||||
|
|
||||||
var _ aperesource.Request = request{}
|
|
||||||
|
|
||||||
type resource struct {
|
|
||||||
name string
|
|
||||||
properties map[string]string
|
|
||||||
}
|
|
||||||
|
|
||||||
var _ aperesource.Resource = resource{}
|
|
||||||
|
|
||||||
func (r resource) Name() string {
|
|
||||||
return r.name
|
|
||||||
}
|
|
||||||
|
|
||||||
func (r resource) Property(key string) string {
|
|
||||||
return r.properties[key]
|
|
||||||
}
|
|
||||||
|
|
||||||
func (r request) Operation() string {
|
|
||||||
return r.operation
|
|
||||||
}
|
|
||||||
|
|
||||||
func (r request) Property(key string) string {
|
|
||||||
return r.properties[key]
|
|
||||||
}
|
|
||||||
|
|
||||||
func (r request) Resource() aperesource.Resource {
|
|
||||||
return r.resource
|
|
||||||
}
|
|
||||||
|
|
||||||
func nativeSchemaRole(role acl.Role) string {
|
func nativeSchemaRole(role acl.Role) string {
|
||||||
switch role {
|
switch role {
|
||||||
|
@ -125,7 +90,7 @@ func objectProperties(cnr cid.ID, oid *oid.ID, cnrOwner user.ID, header *objectV
|
||||||
// newAPERequest creates an APE request to be passed to a chain router. It collects resource properties from
|
// newAPERequest creates an APE request to be passed to a chain router. It collects resource properties from
|
||||||
// header provided by headerProvider. If it cannot be found in headerProvider, then properties are
|
// header provided by headerProvider. If it cannot be found in headerProvider, then properties are
|
||||||
// initialized from header given in prm (if it is set). Otherwise, just CID and OID are set to properties.
|
// initialized from header given in prm (if it is set). Otherwise, just CID and OID are set to properties.
|
||||||
func (c *checkerImpl) newAPERequest(ctx context.Context, prm Prm) (request, error) {
|
func (c *checkerImpl) newAPERequest(ctx context.Context, prm Prm) (aperequest.Request, error) {
|
||||||
switch prm.Method {
|
switch prm.Method {
|
||||||
case nativeschema.MethodGetObject,
|
case nativeschema.MethodGetObject,
|
||||||
nativeschema.MethodHeadObject,
|
nativeschema.MethodHeadObject,
|
||||||
|
@ -150,15 +115,15 @@ func (c *checkerImpl) newAPERequest(ctx context.Context, prm Prm) (request, erro
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
return request{
|
return aperequest.NewRequest(
|
||||||
operation: prm.Method,
|
prm.Method,
|
||||||
resource: resource{
|
aperequest.NewResource(
|
||||||
name: resourceName(prm.Container, prm.Object, prm.Namespace),
|
resourceName(prm.Container, prm.Object, prm.Namespace),
|
||||||
properties: objectProperties(prm.Container, prm.Object, prm.ContainerOwner, header),
|
objectProperties(prm.Container, prm.Object, prm.ContainerOwner, header),
|
||||||
},
|
),
|
||||||
properties: map[string]string{
|
map[string]string{
|
||||||
nativeschema.PropertyKeyActorPublicKey: prm.SenderKey,
|
nativeschema.PropertyKeyActorPublicKey: prm.SenderKey,
|
||||||
nativeschema.PropertyKeyActorRole: prm.Role,
|
nativeschema.PropertyKeyActorRole: prm.Role,
|
||||||
},
|
},
|
||||||
}, nil
|
), nil
|
||||||
}
|
}
|
||||||
|
|
|
@ -6,6 +6,7 @@ import (
|
||||||
"testing"
|
"testing"
|
||||||
|
|
||||||
objectV2 "git.frostfs.info/TrueCloudLab/frostfs-api-go/v2/object"
|
objectV2 "git.frostfs.info/TrueCloudLab/frostfs-api-go/v2/object"
|
||||||
|
aperequest "git.frostfs.info/TrueCloudLab/frostfs-node/pkg/ape/request"
|
||||||
checksumtest "git.frostfs.info/TrueCloudLab/frostfs-sdk-go/checksum/test"
|
checksumtest "git.frostfs.info/TrueCloudLab/frostfs-sdk-go/checksum/test"
|
||||||
objectSDK "git.frostfs.info/TrueCloudLab/frostfs-sdk-go/object"
|
objectSDK "git.frostfs.info/TrueCloudLab/frostfs-sdk-go/object"
|
||||||
"git.frostfs.info/TrueCloudLab/frostfs-sdk-go/user"
|
"git.frostfs.info/TrueCloudLab/frostfs-sdk-go/user"
|
||||||
|
@ -256,22 +257,21 @@ func TestNewAPERequest(t *testing.T) {
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
expectedRequest := request{
|
expectedRequest := aperequest.NewRequest(
|
||||||
operation: method,
|
method,
|
||||||
resource: resource{
|
aperequest.NewResource(
|
||||||
name: resourceName(cnr, obj, prm.Namespace),
|
resourceName(cnr, obj, prm.Namespace),
|
||||||
properties: objectProperties(cnr, obj, testCnrOwner, func() *objectV2.Header {
|
objectProperties(cnr, obj, testCnrOwner, func() *objectV2.Header {
|
||||||
if headerObjSDK != nil {
|
if headerObjSDK != nil {
|
||||||
return headerObjSDK.ToV2().GetHeader()
|
return headerObjSDK.ToV2().GetHeader()
|
||||||
}
|
}
|
||||||
return prm.Header
|
return prm.Header
|
||||||
}()),
|
}())),
|
||||||
},
|
map[string]string{
|
||||||
properties: map[string]string{
|
|
||||||
nativeschema.PropertyKeyActorPublicKey: prm.SenderKey,
|
nativeschema.PropertyKeyActorPublicKey: prm.SenderKey,
|
||||||
nativeschema.PropertyKeyActorRole: prm.Role,
|
nativeschema.PropertyKeyActorRole: prm.Role,
|
||||||
},
|
},
|
||||||
}
|
)
|
||||||
|
|
||||||
require.Equal(t, expectedRequest, r)
|
require.Equal(t, expectedRequest, r)
|
||||||
})
|
})
|
||||||
|
|
Loading…
Reference in a new issue