110 lines
2 KiB
Go
110 lines
2 KiB
Go
package templates
|
|
|
|
import (
|
|
"bytes"
|
|
"fmt"
|
|
"io/fs"
|
|
"path"
|
|
|
|
"github.com/evcc-io/evcc/templates/definition"
|
|
"golang.org/x/exp/slices"
|
|
"gopkg.in/yaml.v3"
|
|
)
|
|
|
|
var (
|
|
templates = make(map[Class][]Template)
|
|
configDefaults = ConfigDefaults{}
|
|
)
|
|
|
|
type Class string
|
|
|
|
const (
|
|
Charger Class = "charger"
|
|
Meter Class = "meter"
|
|
Vehicle Class = "vehicle"
|
|
)
|
|
|
|
func init() {
|
|
configDefaults.LoadDefaults()
|
|
|
|
loadTemplates(Charger)
|
|
loadTemplates(Meter)
|
|
loadTemplates(Vehicle)
|
|
}
|
|
|
|
func FromBytes(b []byte) (Template, error) {
|
|
// panic if template definition contains unknown fields
|
|
dec := yaml.NewDecoder(bytes.NewReader(b))
|
|
dec.KnownFields(true)
|
|
|
|
var definition TemplateDefinition
|
|
if err := dec.Decode(&definition); err != nil {
|
|
return Template{}, err
|
|
}
|
|
|
|
tmpl := Template{
|
|
TemplateDefinition: definition,
|
|
ConfigDefaults: configDefaults,
|
|
}
|
|
|
|
err := tmpl.ResolvePresets()
|
|
if err == nil {
|
|
err = tmpl.ResolveGroup()
|
|
}
|
|
if err == nil {
|
|
err = tmpl.UpdateParamsWithDefaults()
|
|
}
|
|
if err == nil {
|
|
err = tmpl.Validate()
|
|
}
|
|
|
|
return tmpl, err
|
|
}
|
|
|
|
func loadTemplates(class Class) {
|
|
if templates[class] != nil {
|
|
return
|
|
}
|
|
|
|
err := fs.WalkDir(definition.YamlTemplates, ".", func(filepath string, d fs.DirEntry, err error) error {
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if d.IsDir() {
|
|
return nil
|
|
}
|
|
|
|
b, err := fs.ReadFile(definition.YamlTemplates, filepath)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
tmpl, err := FromBytes(b)
|
|
if err != nil {
|
|
return fmt.Errorf("processing template '%s' failed: %w", filepath, err)
|
|
}
|
|
|
|
path := Class(path.Dir(filepath))
|
|
templates[path] = append(templates[path], tmpl)
|
|
|
|
return nil
|
|
})
|
|
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
}
|
|
|
|
func ByClass(class Class) []Template {
|
|
return templates[class]
|
|
}
|
|
|
|
func ByName(class Class, name string) (Template, error) {
|
|
for _, tmpl := range templates[class] {
|
|
if tmpl.Template == name || slices.Contains(tmpl.Covers, name) {
|
|
return tmpl, nil
|
|
}
|
|
}
|
|
|
|
return Template{}, fmt.Errorf("template not found: %s", name)
|
|
}
|