mirror of https://github.com/grafana/grafana.git
117 lines
2.9 KiB
Go
117 lines
2.9 KiB
Go
package codegen
|
|
|
|
import (
|
|
"bytes"
|
|
"fmt"
|
|
"go/format"
|
|
"strings"
|
|
|
|
"cuelang.org/go/cue"
|
|
"github.com/grafana/codejen"
|
|
)
|
|
|
|
// K8ResourcesJenny generates resource, metadata and status for each file.
|
|
type K8ResourcesJenny struct {
|
|
}
|
|
|
|
func (jenny *K8ResourcesJenny) JennyName() string {
|
|
return "K8ResourcesJenny"
|
|
}
|
|
|
|
func (jenny *K8ResourcesJenny) Generate(cueFiles ...SchemaForGen) (codejen.Files, error) {
|
|
files := make(codejen.Files, 0)
|
|
for _, val := range cueFiles {
|
|
resource, err := jenny.genResource(val.Name, val.CueFile)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
metadata, err := jenny.genMetadata(val.Name)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
status, err := jenny.genStatus(val.Name)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
files = append(files, resource)
|
|
files = append(files, metadata)
|
|
files = append(files, status)
|
|
}
|
|
|
|
return files, nil
|
|
}
|
|
|
|
func (jenny *K8ResourcesJenny) genResource(pkg string, val cue.Value) (codejen.File, error) {
|
|
version, err := getVersion(val)
|
|
if err != nil {
|
|
return codejen.File{}, err
|
|
}
|
|
|
|
pkgName := strings.ToLower(pkg)
|
|
|
|
buf := new(bytes.Buffer)
|
|
if err := tmpls.Lookup("core_resource.tmpl").Execute(buf, tvars_resource{
|
|
PackageName: pkgName,
|
|
KindName: pkg,
|
|
Version: version,
|
|
}); err != nil {
|
|
return codejen.File{}, fmt.Errorf("failed executing core resource template: %w", err)
|
|
}
|
|
|
|
content, err := format.Source(buf.Bytes())
|
|
if err != nil {
|
|
return codejen.File{}, err
|
|
}
|
|
|
|
return *codejen.NewFile(fmt.Sprintf("pkg/kinds/%s/%s_gen.go", pkgName, pkgName), content, jenny), nil
|
|
}
|
|
|
|
func (jenny *K8ResourcesJenny) genMetadata(pkg string) (codejen.File, error) {
|
|
pkg = strings.ToLower(pkg)
|
|
|
|
buf := new(bytes.Buffer)
|
|
if err := tmpls.Lookup("core_metadata.tmpl").Execute(buf, tvars_metadata{
|
|
PackageName: pkg,
|
|
}); err != nil {
|
|
return codejen.File{}, fmt.Errorf("failed executing core resource template: %w", err)
|
|
}
|
|
|
|
return *codejen.NewFile(fmt.Sprintf("pkg/kinds/%s/%s_metadata_gen.go", pkg, pkg), buf.Bytes(), jenny), nil
|
|
}
|
|
|
|
func (jenny *K8ResourcesJenny) genStatus(pkg string) (codejen.File, error) {
|
|
pkg = strings.ToLower(pkg)
|
|
|
|
buf := new(bytes.Buffer)
|
|
if err := tmpls.Lookup("core_status.tmpl").Execute(buf, tvars_status{
|
|
PackageName: pkg,
|
|
}); err != nil {
|
|
return codejen.File{}, fmt.Errorf("failed executing core resource template: %w", err)
|
|
}
|
|
|
|
return *codejen.NewFile(fmt.Sprintf("pkg/kinds/%s/%s_status_gen.go", pkg, pkg), buf.Bytes(), jenny), nil
|
|
}
|
|
|
|
func getVersion(val cue.Value) (string, error) {
|
|
val = val.LookupPath(cue.ParsePath("lineage.schemas[0].version"))
|
|
versionValues, err := val.List()
|
|
if err != nil {
|
|
return "", fmt.Errorf("missing version in schema: %s", err)
|
|
}
|
|
|
|
version := make([]int64, 0)
|
|
for versionValues.Next() {
|
|
v, err := versionValues.Value().Int64()
|
|
if err != nil {
|
|
return "", fmt.Errorf("version should be a list of two elements: %s", err)
|
|
}
|
|
|
|
version = append(version, v)
|
|
}
|
|
|
|
return fmt.Sprintf("%d-%d", version[0], version[1]), nil
|
|
}
|