/
wrappers.go
53 lines (44 loc) · 1.55 KB
/
wrappers.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
package layers
import "github.com/go-go-golems/glazed/pkg/cmds/parameters"
type WhitelistParameterLayer struct {
ParameterLayer
WhitelistedParameters map[string]interface{}
}
var _ ParameterLayer = (*WhitelistParameterLayer)(nil)
func NewWhitelistParameterLayer(layer ParameterLayer, whitelistedParameters map[string]interface{}) *WhitelistParameterLayer {
return &WhitelistParameterLayer{
ParameterLayer: layer,
WhitelistedParameters: whitelistedParameters,
}
}
func (l *WhitelistParameterLayer) GetParameterDefinitions() *parameters.ParameterDefinitions {
pds := l.ParameterLayer.GetParameterDefinitions()
ret := parameters.NewParameterDefinitions()
pds.ForEach(func(pd *parameters.ParameterDefinition) {
if _, ok := l.WhitelistedParameters[pd.Name]; ok {
ret.Set(pd.Name, pd)
}
})
return ret
}
type BlacklistParameterLayer struct {
ParameterLayer
BlacklistedParameters map[string]interface{}
}
var _ ParameterLayer = (*BlacklistParameterLayer)(nil)
func NewBlacklistParameterLayer(layer ParameterLayer, blacklistedParameters map[string]interface{}) *BlacklistParameterLayer {
return &BlacklistParameterLayer{
ParameterLayer: layer,
BlacklistedParameters: blacklistedParameters,
}
}
func (l *BlacklistParameterLayer) GetParameterDefinitions() *parameters.ParameterDefinitions {
pds := l.ParameterLayer.GetParameterDefinitions()
ret := parameters.NewParameterDefinitions()
pds.ForEach(func(pd *parameters.ParameterDefinition) {
if _, ok := l.BlacklistedParameters[pd.Name]; !ok {
ret.Set(pd.Name, pd)
}
})
return ret
}