/
envoyplugin.go
121 lines (95 loc) · 2.49 KB
/
envoyplugin.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
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
package envoy
import (
"context"
"strconv"
"strings"
"github.com/open-policy-agent/opa/ast"
"github.com/open-policy-agent/opa/plugins"
"github.com/open-policy-agent/opa/util"
)
// Factory defines the interface OPA uses to instantiate a plugin.
type Factory struct{}
// New returns the object initialized with a valid plugin configuration.
func (Factory) New(m *plugins.Manager, cfg interface{}) plugins.Plugin {
p := &Plugin{
manager: m,
cfg: *cfg.(*PluginConfig),
}
m.UpdatePluginStatus(PluginName, &plugins.Status{State: plugins.StateNotReady})
return p
}
// Validate returns a valid configuration to instantiate the plugin.
func (Factory) Validate(m *plugins.Manager, bs []byte) (interface{}, error) {
cfg := PluginConfig{
DryRun: defaultDryRun,
}
if err := util.Unmarshal(bs, &cfg); err != nil {
return nil, err
}
if err := cfg.ParseQuery(); err != nil {
return nil, err
}
return &cfg, nil
}
func (p *Plugin) Reconfigure(ctx context.Context, config interface{}) {
p.cfg = *config.(*PluginConfig)
}
// PluginConfig represents the plugin configuration.
type PluginConfig struct {
Path string `json:"path"`
DryRun bool `json:"dry-run"`
ParsedQuery ast.Body
}
type Plugin struct {
cfg PluginConfig
manager *plugins.Manager
}
func (p *Plugin) Start(ctx context.Context) error {
p.manager.UpdatePluginStatus(PluginName, &plugins.Status{State: plugins.StateOK})
return nil
}
func (cfg *PluginConfig) ParseQuery() error {
var parsedQuery ast.Body
var err error
if cfg.Path == "" {
cfg.Path = defaultPath
}
path := stringPathToDataRef(cfg.Path)
parsedQuery, err = ast.ParseBody(path.String())
if err != nil {
return err
}
cfg.ParsedQuery = parsedQuery
return nil
}
func (p *Plugin) Stop(ctx context.Context) {
p.manager.UpdatePluginStatus(PluginName, &plugins.Status{State: plugins.StateNotReady})
}
func (p *Plugin) GetConfig() PluginConfig {
return p.cfg
}
func (p *Plugin) ParsedQuery() ast.Body { return p.cfg.ParsedQuery }
func (p *Plugin) Path() string { return p.cfg.Path }
func stringPathToDataRef(s string) (r ast.Ref) {
result := ast.Ref{ast.DefaultRootDocument}
result = append(result, stringPathToRef(s)...)
return result
}
func stringPathToRef(s string) (r ast.Ref) {
if len(s) == 0 {
return r
}
p := strings.Split(s, "/")
for _, x := range p {
if x == "" {
continue
}
i, err := strconv.Atoi(x)
if err != nil {
r = append(r, ast.StringTerm(x))
} else {
r = append(r, ast.IntNumberTerm(i))
}
}
return r
}