forked from TrueCloudLab/frostfs-api-go
158 lines
3.1 KiB
Go
158 lines
3.1 KiB
Go
package eacl
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
v2acl "github.com/nspcc-dev/neofs-api-go/v2/acl"
|
|
)
|
|
|
|
// Filter defines check conditions if request header is matched or not. Matched
|
|
// header means that request should be processed according to EACL action.
|
|
type Filter struct {
|
|
from FilterHeaderType
|
|
key filterKey
|
|
matcher Match
|
|
value fmt.Stringer
|
|
}
|
|
|
|
type staticStringer string
|
|
|
|
type filterKey struct {
|
|
typ filterKeyType
|
|
|
|
str string
|
|
}
|
|
|
|
// enumeration of reserved filter keys.
|
|
type filterKeyType int
|
|
|
|
const (
|
|
_ filterKeyType = iota
|
|
fKeyObjVersion
|
|
fKeyObjID
|
|
fKeyObjContainerID
|
|
fKeyObjOwnerID
|
|
fKeyObjCreationEpoch
|
|
fKeyObjPayloadLength
|
|
fKeyObjPayloadHash
|
|
fKeyObjType
|
|
fKeyObjHomomorphicHash
|
|
)
|
|
|
|
func (s staticStringer) String() string {
|
|
return string(s)
|
|
}
|
|
|
|
func (f Filter) Value() string {
|
|
return f.value.String()
|
|
}
|
|
|
|
func (f Filter) Matcher() Match {
|
|
return f.matcher
|
|
}
|
|
|
|
func (f Filter) Key() string {
|
|
return f.key.String()
|
|
}
|
|
|
|
func (f Filter) From() FilterHeaderType {
|
|
return f.from
|
|
}
|
|
|
|
func (f *Filter) ToV2() *v2acl.HeaderFilter {
|
|
filter := new(v2acl.HeaderFilter)
|
|
filter.SetValue(f.value.String())
|
|
filter.SetKey(f.key.String())
|
|
filter.SetMatchType(f.matcher.ToV2())
|
|
filter.SetHeaderType(f.from.ToV2())
|
|
|
|
return filter
|
|
}
|
|
|
|
func (k filterKey) String() string {
|
|
switch k.typ {
|
|
default:
|
|
return k.str
|
|
case fKeyObjVersion:
|
|
return v2acl.FilterObjectVersion
|
|
case fKeyObjID:
|
|
return v2acl.FilterObjectID
|
|
case fKeyObjContainerID:
|
|
return v2acl.FilterObjectContainerID
|
|
case fKeyObjOwnerID:
|
|
return v2acl.FilterObjectOwnerID
|
|
case fKeyObjCreationEpoch:
|
|
return v2acl.FilterObjectCreationEpoch
|
|
case fKeyObjPayloadLength:
|
|
return v2acl.FilterObjectPayloadLength
|
|
case fKeyObjPayloadHash:
|
|
return v2acl.FilterObjectPayloadHash
|
|
case fKeyObjType:
|
|
return v2acl.FilterObjectType
|
|
case fKeyObjHomomorphicHash:
|
|
return v2acl.FilterObjectHomomorphicHash
|
|
}
|
|
}
|
|
|
|
func NewFilter() *Filter {
|
|
return NewFilterFromV2(new(v2acl.HeaderFilter))
|
|
}
|
|
|
|
func NewFilterFromV2(filter *v2acl.HeaderFilter) *Filter {
|
|
f := new(Filter)
|
|
|
|
if filter == nil {
|
|
return f
|
|
}
|
|
|
|
f.from = FilterHeaderTypeFromV2(filter.GetHeaderType())
|
|
f.matcher = MatchFromV2(filter.GetMatchType())
|
|
f.key.str = filter.GetKey()
|
|
f.value = staticStringer(filter.GetValue())
|
|
|
|
return f
|
|
}
|
|
|
|
// Marshal marshals Filter into a protobuf binary form.
|
|
//
|
|
// Buffer is allocated when the argument is empty.
|
|
// Otherwise, the first buffer is used.
|
|
func (f *Filter) Marshal(b ...[]byte) ([]byte, error) {
|
|
var buf []byte
|
|
if len(b) > 0 {
|
|
buf = b[0]
|
|
}
|
|
|
|
return f.ToV2().
|
|
StableMarshal(buf)
|
|
}
|
|
|
|
// Unmarshal unmarshals protobuf binary representation of Filter.
|
|
func (f *Filter) Unmarshal(data []byte) error {
|
|
fV2 := new(v2acl.HeaderFilter)
|
|
if err := fV2.Unmarshal(data); err != nil {
|
|
return err
|
|
}
|
|
|
|
*f = *NewFilterFromV2(fV2)
|
|
|
|
return nil
|
|
}
|
|
|
|
// MarshalJSON encodes Filter to protobuf JSON format.
|
|
func (f *Filter) MarshalJSON() ([]byte, error) {
|
|
return f.ToV2().
|
|
MarshalJSON()
|
|
}
|
|
|
|
// UnmarshalJSON decodes Filter from protobuf JSON format.
|
|
func (f *Filter) UnmarshalJSON(data []byte) error {
|
|
fV2 := new(v2acl.HeaderFilter)
|
|
if err := fV2.UnmarshalJSON(data); err != nil {
|
|
return err
|
|
}
|
|
|
|
*f = *NewFilterFromV2(fV2)
|
|
|
|
return nil
|
|
}
|