mirror of
https://github.com/safing/portbase
synced 2025-09-10 15:34:26 +00:00
Add query package with first set of conditions and tests
This commit is contained in:
parent
1bf7e42d8a
commit
6ed50f34fb
16 changed files with 933 additions and 0 deletions
57
database/query/condition-regex.go
Normal file
57
database/query/condition-regex.go
Normal file
|
@ -0,0 +1,57 @@
|
|||
package query
|
||||
|
||||
import (
|
||||
"errors"
|
||||
"fmt"
|
||||
"regexp"
|
||||
)
|
||||
|
||||
type regexCondition struct {
|
||||
key string
|
||||
operator uint8
|
||||
regex *regexp.Regexp
|
||||
}
|
||||
|
||||
func newRegexCondition(key string, operator uint8, value interface{}) *regexCondition {
|
||||
switch v := value.(type) {
|
||||
case string:
|
||||
r, err := regexp.Compile(v)
|
||||
if err != nil {
|
||||
return ®exCondition{
|
||||
key: fmt.Sprintf("could not compile regex \"%s\": %s", v, err),
|
||||
operator: errorPresent,
|
||||
}
|
||||
}
|
||||
return ®exCondition{
|
||||
key: key,
|
||||
operator: operator,
|
||||
regex: r,
|
||||
}
|
||||
default:
|
||||
return ®exCondition{
|
||||
key: fmt.Sprintf("incompatible value %v for string", value),
|
||||
operator: errorPresent,
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func (c *regexCondition) complies(f Fetcher) bool {
|
||||
comp, ok := f.GetString(c.key)
|
||||
if !ok {
|
||||
return false
|
||||
}
|
||||
|
||||
switch c.operator {
|
||||
case Matches:
|
||||
return c.regex.MatchString(comp)
|
||||
default:
|
||||
return false
|
||||
}
|
||||
}
|
||||
|
||||
func (c *regexCondition) check() error {
|
||||
if c.operator == errorPresent {
|
||||
return errors.New(c.key)
|
||||
}
|
||||
return nil
|
||||
}
|
Loading…
Add table
Add a link
Reference in a new issue