/
glob.go
86 lines (73 loc) · 1.96 KB
/
glob.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
package topdown
import (
"strings"
"sync"
"github.com/gobwas/glob"
"github.com/open-policy-agent/opa/ast"
"github.com/open-policy-agent/opa/topdown/builtins"
)
var globCacheLock = sync.Mutex{}
var globCache map[string]glob.Glob
func builtinGlobMatch(_ BuiltinContext, operands []*ast.Term, iter func(*ast.Term) error) error {
pattern, err := builtins.StringOperand(operands[0].Value, 1)
if err != nil {
return err
}
var delimiters []rune
switch operands[1].Value.(type) {
case ast.Null:
delimiters = []rune{}
case *ast.Array:
delimiters, err = builtins.RuneSliceOperand(operands[1].Value, 2)
if err != nil {
return err
}
if len(delimiters) == 0 {
delimiters = []rune{'.'}
}
default:
return builtins.NewOperandTypeErr(2, operands[1].Value, "array", "null")
}
match, err := builtins.StringOperand(operands[2].Value, 3)
if err != nil {
return err
}
builder := strings.Builder{}
builder.WriteString(string(pattern))
builder.WriteRune('-')
for _, v := range delimiters {
builder.WriteRune(v)
}
id := builder.String()
m, err := globCompileAndMatch(id, string(pattern), string(match), delimiters)
if err != nil {
return err
}
return iter(ast.BooleanTerm(m))
}
func globCompileAndMatch(id, pattern, match string, delimiters []rune) (bool, error) {
globCacheLock.Lock()
defer globCacheLock.Unlock()
p, ok := globCache[id]
if !ok {
var err error
if p, err = glob.Compile(pattern, delimiters...); err != nil {
return false, err
}
globCache[id] = p
}
out := p.Match(match)
return out, nil
}
func builtinGlobQuoteMeta(_ BuiltinContext, operands []*ast.Term, iter func(*ast.Term) error) error {
pattern, err := builtins.StringOperand(operands[0].Value, 1)
if err != nil {
return err
}
return iter(ast.StringTerm(glob.QuoteMeta(string(pattern))))
}
func init() {
globCache = map[string]glob.Glob{}
RegisterBuiltinFunc(ast.GlobMatch.Name, builtinGlobMatch)
RegisterBuiltinFunc(ast.GlobQuoteMeta.Name, builtinGlobQuoteMeta)
}