DeforaNetworks/corteza
3
0
Fork 0
Code Activity

Improve codegen templating tool

Browse Source
This commit is contained in:
Denis Arh
2022-02-06 16:11:49 +01:00
parent d103d60a3d
commit d09b037e84
2 changed files with 46 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
codegen/tool/main.go
View File

@@ -51,19 +51,26 @@ func main() {
cli.HandleError(fmt.Errorf("failed to decode input from standard input: %v", err))
}
if tpl, err = LoadTemplates(BaseTemplate(), tplRootPath); err != nil {
if tpl, err = loadTemplates(baseTemplate(), tplRootPath); err != nil {
cli.HandleError(fmt.Errorf("failed to load templates: %v", err))
}
for _, j := range tasks {
print(fmt.Sprintf("generating %s (from %s) ...", j.Output, j.Template))
switch j.Syntax {
case "go":
print(fmt.Sprintf("generating %s (from %s) ...", j.Output, j.Template))
if err = GoTemplate(j.Output, tpl.Lookup(j.Template), j.Payload); err != nil {
cli.HandleError(fmt.Errorf("failed to write template: %v", err))
}
err = writeFormattedGo(j.Output, tpl.Lookup(j.Template), j.Payload)
default:
err = write(j.Output, tpl.Lookup(j.Template), j.Payload)
}
if err != nil {
cli.HandleError(fmt.Errorf("failed to write template: %v", err))
} else {
print("done\n")
}
}
}

47
codegen/tool/templating.go
View File

@@ -14,12 +14,12 @@ import (
"github.com/Masterminds/sprig"
)
func BaseTemplate() *template.Template {
func baseTemplate() *template.Template {
return template.New("").
Funcs(sprig.TxtFuncMap())
}
func LoadTemplates(rTpl *template.Template, rootDir string) (*template.Template, error) {
func loadTemplates(rTpl *template.Template, rootDir string) (*template.Template, error) {
cleanRoot := filepath.Clean(rootDir)
pfx := len(cleanRoot) + 1
@@ -44,24 +44,45 @@ func LoadTemplates(rTpl *template.Template, rootDir string) (*template.Template,
})
}
func GoTemplate(dst string, tpl *template.Template, payload interface{}) (err error) {
var output io.WriteCloser
buf := bytes.Buffer{}
func writeFormattedGo(dst string, tpl *template.Template, payload interface{}) error {
return write(dst, tpl, payload, func(in io.ReadWriter) (out io.ReadWriter, err error) {
var (
bb []byte
)
if bb, err = ioutil.ReadAll(in); err != nil {
return
}
if bb, err = format.Source(bb); err != nil {
// output error and return unformatted source
_, _ = fmt.Fprintf(os.Stderr, "%s fmt warn: %v\n", dst, err)
return in, err
}
return bytes.NewBuffer(bb), nil
})
}
func write(dst string, tpl *template.Template, payload interface{}, pp ...func(io.ReadWriter) (io.ReadWriter, error)) (err error) {
var (
output io.WriteCloser
buf io.ReadWriter
)
if tpl == nil {
return fmt.Errorf("could not find template for %s", dst)
}
if err := tpl.Execute(&buf, payload); err != nil {
buf = &bytes.Buffer{}
if err := tpl.Execute(buf, payload); err != nil {
return err
}
fmtsrc, err := format.Source(buf.Bytes())
if err != nil {
_, _ = fmt.Fprintf(os.Stderr, "%s fmt warn: %v\n", dst, err)
err = nil
fmtsrc = buf.Bytes()
for _, proc := range pp {
if buf, err = proc(buf); err != nil {
return
}
}
if dst == "" || dst == "-" {
@@ -74,7 +95,7 @@ func GoTemplate(dst string, tpl *template.Template, payload interface{}) (err er
defer output.Close()
}
if _, err = output.Write(fmtsrc); err != nil {
if _, err = io.Copy(output, buf); err != nil {
return err
}