mirror of https://github.com/grafana/grafana.git
73 lines
1.8 KiB
Go
73 lines
1.8 KiB
Go
package codegen
|
|
|
|
import (
|
|
"bytes"
|
|
"fmt"
|
|
"path/filepath"
|
|
|
|
"github.com/grafana/codejen"
|
|
"github.com/grafana/kindsys"
|
|
)
|
|
|
|
// CoreKindJenny generates the implementation of [kindsys.Core] for the provided
|
|
// kind declaration.
|
|
//
|
|
// gokindsdir should be the relative path to the parent directory that contains
|
|
// all generated kinds.
|
|
//
|
|
// This generator only has output for core structured kinds.
|
|
func CoreKindJenny(gokindsdir string, cfg *CoreKindJennyConfig) OneToOne {
|
|
if cfg == nil {
|
|
cfg = new(CoreKindJennyConfig)
|
|
}
|
|
if cfg.GenDirName == nil {
|
|
cfg.GenDirName = func(def kindsys.Kind) string {
|
|
return def.Props().Common().MachineName
|
|
}
|
|
}
|
|
|
|
return &coreKindJenny{
|
|
gokindsdir: gokindsdir,
|
|
cfg: cfg,
|
|
}
|
|
}
|
|
|
|
// CoreKindJennyConfig holds configuration options for [CoreKindJenny].
|
|
type CoreKindJennyConfig struct {
|
|
// GenDirName returns the name of the directory in which the file should be
|
|
// generated. Defaults to DefForGen.Lineage().Name() if nil.
|
|
GenDirName func(kindsys.Kind) string
|
|
}
|
|
|
|
type coreKindJenny struct {
|
|
gokindsdir string
|
|
cfg *CoreKindJennyConfig
|
|
}
|
|
|
|
var _ OneToOne = &coreKindJenny{}
|
|
|
|
func (gen *coreKindJenny) JennyName() string {
|
|
return "CoreKindJenny"
|
|
}
|
|
|
|
func (gen *coreKindJenny) Generate(kind kindsys.Kind) (*codejen.File, error) {
|
|
if _, is := kind.(kindsys.Core); !is {
|
|
return nil, nil
|
|
}
|
|
|
|
path := filepath.Join(gen.gokindsdir, gen.cfg.GenDirName(kind), kind.Props().Common().MachineName+"_kind_gen.go")
|
|
buf := new(bytes.Buffer)
|
|
if err := tmpls.Lookup("kind_core.tmpl").Execute(buf, kind); err != nil {
|
|
return nil, fmt.Errorf("failed executing kind_core template for %s: %w", path, err)
|
|
}
|
|
b, err := postprocessGoFile(genGoFile{
|
|
path: path,
|
|
in: buf.Bytes(),
|
|
})
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return codejen.NewFile(path, b, gen), nil
|
|
}
|