76 lines
1.6 KiB
Go
76 lines
1.6 KiB
Go
|
package config
|
||
|
|
|
||
|
|
import "github.com/mitchellh/copystructure"
|
||
|
|
|
||
|
|
// PluginConfig is used to configure a plugin explicitly
|
||
|
|
type PluginConfig struct {
|
||
|
|
Name string `hcl:",key"`
|
||
|
|
Args []string `hcl:"args"`
|
||
|
|
Config map[string]interface{} `hcl:"config"`
|
||
|
|
}
|
||
|
|
|
||
|
|
func (p *PluginConfig) Merge(o *PluginConfig) *PluginConfig {
|
||
|
|
m := *p
|
||
|
|
|
||
|
|
if len(o.Name) != 0 {
|
||
|
|
m.Name = o.Name
|
||
|
|
}
|
||
|
|
if len(o.Args) != 0 {
|
||
|
|
m.Args = o.Args
|
||
|
|
}
|
||
|
|
if len(o.Config) != 0 {
|
||
|
|
m.Config = o.Config
|
||
|
|
}
|
||
|
|
|
||
|
|
return m.Copy()
|
||
|
|
}
|
||
|
|
|
||
|
|
func (p *PluginConfig) Copy() *PluginConfig {
|
||
|
|
c := *p
|
||
|
|
if i, err := copystructure.Copy(p.Config); err != nil {
|
||
|
|
panic(err.Error())
|
||
|
|
} else {
|
||
|
|
c.Config = i.(map[string]interface{})
|
||
|
|
}
|
||
|
|
return &c
|
||
|
|
}
|
||
|
|
|
||
|
|
// PluginConfigSetMerge merges to sets of plugin configs. For plugins with the
|
||
|
|
// same name, the configs are merged.
|
||
|
|
func PluginConfigSetMerge(first, second []*PluginConfig) []*PluginConfig {
|
||
|
|
findex := make(map[string]*PluginConfig, len(first))
|
||
|
|
for _, p := range first {
|
||
|
|
findex[p.Name] = p
|
||
|
|
}
|
||
|
|
|
||
|
|
sindex := make(map[string]*PluginConfig, len(second))
|
||
|
|
for _, p := range second {
|
||
|
|
sindex[p.Name] = p
|
||
|
|
}
|
||
|
|
|
||
|
|
var out []*PluginConfig
|
||
|
|
|
||
|
|
// Go through the first set and merge any value that exist in both
|
||
|
|
for pluginName, original := range findex {
|
||
|
|
second, ok := sindex[pluginName]
|
||
|
|
if !ok {
|
||
|
|
out = append(out, original.Copy())
|
||
|
|
continue
|
||
|
|
}
|
||
|
|
|
||
|
|
out = append(out, original.Merge(second))
|
||
|
|
}
|
||
|
|
|
||
|
|
// Go through the second set and add any value that didn't exist in both
|
||
|
|
for pluginName, plugin := range sindex {
|
||
|
|
_, ok := findex[pluginName]
|
||
|
|
if ok {
|
||
|
|
continue
|
||
|
|
}
|
||
|
|
|
||
|
|
out = append(out, plugin)
|
||
|
|
}
|
||
|
|
|
||
|
|
return out
|
||
|
|
}
|