package object import ( "encoding/json" "fmt" "github.com/nspcc-dev/neofs-api-go/pkg" cid "github.com/nspcc-dev/neofs-api-go/pkg/container/id" "github.com/nspcc-dev/neofs-api-go/pkg/owner" v2object "github.com/nspcc-dev/neofs-api-go/v2/object" ) // SearchMatchType indicates match operation on specified header. type SearchMatchType uint32 const ( MatchUnknown SearchMatchType = iota MatchStringEqual MatchStringNotEqual MatchNotPresent ) func (m SearchMatchType) ToV2() v2object.MatchType { switch m { case MatchStringEqual: return v2object.MatchStringEqual case MatchStringNotEqual: return v2object.MatchStringNotEqual case MatchNotPresent: return v2object.MatchNotPresent default: return v2object.MatchUnknown } } func SearchMatchFromV2(t v2object.MatchType) (m SearchMatchType) { switch t { case v2object.MatchStringEqual: m = MatchStringEqual case v2object.MatchStringNotEqual: m = MatchStringNotEqual case v2object.MatchNotPresent: m = MatchNotPresent default: m = MatchUnknown } return m } type SearchFilter struct { header filterKey value fmt.Stringer op SearchMatchType } type staticStringer string type filterKey struct { typ filterKeyType str string } // enumeration of reserved filter keys. type filterKeyType int type boolStringer bool type SearchFilters []SearchFilter const ( _ filterKeyType = iota fKeyVersion fKeyObjectID fKeyContainerID fKeyOwnerID fKeyCreationEpoch fKeyPayloadLength fKeyPayloadHash fKeyType fKeyHomomorphicHash fKeyParent fKeySplitID fKeyPropRoot fKeyPropPhy ) func (k filterKey) String() string { switch k.typ { default: return k.str case fKeyVersion: return v2object.FilterHeaderVersion case fKeyObjectID: return v2object.FilterHeaderObjectID case fKeyContainerID: return v2object.FilterHeaderContainerID case fKeyOwnerID: return v2object.FilterHeaderOwnerID case fKeyCreationEpoch: return v2object.FilterHeaderCreationEpoch case fKeyPayloadLength: return v2object.FilterHeaderPayloadLength case fKeyPayloadHash: return v2object.FilterHeaderPayloadHash case fKeyType: return v2object.FilterHeaderObjectType case fKeyHomomorphicHash: return v2object.FilterHeaderHomomorphicHash case fKeyParent: return v2object.FilterHeaderParent case fKeySplitID: return v2object.FilterHeaderSplitID case fKeyPropRoot: return v2object.FilterPropertyRoot case fKeyPropPhy: return v2object.FilterPropertyPhy } } func (s staticStringer) String() string { return string(s) } func (s boolStringer) String() string { if s { return v2object.BooleanPropertyValueTrue } return v2object.BooleanPropertyValueFalse } func (f *SearchFilter) Header() string { return f.header.String() } func (f *SearchFilter) Value() string { return f.value.String() } func (f *SearchFilter) Operation() SearchMatchType { return f.op } func NewSearchFilters() SearchFilters { return SearchFilters{} } func NewSearchFiltersFromV2(v2 []*v2object.SearchFilter) SearchFilters { filters := make(SearchFilters, 0, len(v2)) for i := range v2 { if v2[i] == nil { continue } filters.AddFilter( v2[i].GetKey(), v2[i].GetValue(), SearchMatchFromV2(v2[i].GetMatchType()), ) } return filters } func (f *SearchFilters) addFilter(op SearchMatchType, keyTyp filterKeyType, key string, val fmt.Stringer) { if *f == nil { *f = make(SearchFilters, 0, 1) } *f = append(*f, SearchFilter{ header: filterKey{ typ: keyTyp, str: key, }, value: val, op: op, }) } func (f *SearchFilters) AddFilter(header, value string, op SearchMatchType) { f.addFilter(op, 0, header, staticStringer(value)) } func (f *SearchFilters) addReservedFilter(op SearchMatchType, keyTyp filterKeyType, val fmt.Stringer) { f.addFilter(op, keyTyp, "", val) } // addFlagFilters adds filters that works like flags: they don't need to have // specific match type or value. They processed by NeoFS nodes by the fact // of presence in search query. E.g.: PHY, ROOT. func (f *SearchFilters) addFlagFilter(keyTyp filterKeyType) { f.addFilter(MatchUnknown, keyTyp, "", staticStringer("")) } func (f *SearchFilters) AddObjectVersionFilter(op SearchMatchType, v *pkg.Version) { f.addReservedFilter(op, fKeyVersion, v) } func (f *SearchFilters) AddObjectContainerIDFilter(m SearchMatchType, id *cid.ID) { f.addReservedFilter(m, fKeyContainerID, id) } func (f *SearchFilters) AddObjectOwnerIDFilter(m SearchMatchType, id *owner.ID) { f.addReservedFilter(m, fKeyOwnerID, id) } func (f SearchFilters) ToV2() []*v2object.SearchFilter { result := make([]*v2object.SearchFilter, 0, len(f)) for i := range f { v2 := new(v2object.SearchFilter) v2.SetKey(f[i].header.String()) v2.SetValue(f[i].value.String()) v2.SetMatchType(f[i].op.ToV2()) result = append(result, v2) } return result } func (f *SearchFilters) addRootFilter() { f.addFlagFilter(fKeyPropRoot) } func (f *SearchFilters) AddRootFilter() { f.addRootFilter() } func (f *SearchFilters) addPhyFilter() { f.addFlagFilter(fKeyPropPhy) } func (f *SearchFilters) AddPhyFilter() { f.addPhyFilter() } // AddParentIDFilter adds filter by parent identifier. func (f *SearchFilters) AddParentIDFilter(m SearchMatchType, id *ID) { f.addReservedFilter(m, fKeyParent, id) } // AddObjectIDFilter adds filter by object identifier. func (f *SearchFilters) AddObjectIDFilter(m SearchMatchType, id *ID) { f.addReservedFilter(m, fKeyObjectID, id) } func (f *SearchFilters) AddSplitIDFilter(m SearchMatchType, id *SplitID) { f.addReservedFilter(m, fKeySplitID, id) } // AddTypeFilter adds filter by object type. func (f *SearchFilters) AddTypeFilter(m SearchMatchType, typ Type) { f.addReservedFilter(m, fKeyType, typ) } // MarshalJSON encodes SearchFilters to protobuf JSON format. func (f *SearchFilters) MarshalJSON() ([]byte, error) { return json.Marshal(f.ToV2()) } // UnmarshalJSON decodes SearchFilters from protobuf JSON format. func (f *SearchFilters) UnmarshalJSON(data []byte) error { var fsV2 []*v2object.SearchFilter if err := json.Unmarshal(data, &fsV2); err != nil { return err } *f = NewSearchFiltersFromV2(fsV2) return nil }