diff options
Diffstat (limited to 'templater/templater.go')
-rw-r--r-- | templater/templater.go | 97 |
1 files changed, 97 insertions, 0 deletions
diff --git a/templater/templater.go b/templater/templater.go new file mode 100644 index 000000000000..f4be1e6fa60d --- /dev/null +++ b/templater/templater.go @@ -0,0 +1,97 @@ +package templater + +import ( + "fmt" + "io/ioutil" + "strings" + "os" + "path" + "text/template" + "bytes" + + "github.com/tazjin/kontemplate/context" + "github.com/polydawn/meep" +) + +// Error that is caused by non-existent template files being specified +type TemplateNotFoundError struct { + meep.AllTraits + Name string +} + +// Error that is caused during templating, e.g. required value being absent or invalid template format +type TemplatingError struct { + meep.AllTraits +} + +func LoadAndPrepareTemplates(c *context.Context) ([]string, error) { + output := make([]string, 0) + + for _, rs := range c.ResourceSets { + fmt.Fprintf(os.Stderr,"Loading resources for %s\n", rs.Name) + + rp := path.Join(c.BaseDir, rs.Name) + files, err := ioutil.ReadDir(rp) + + if err != nil { + return nil, meep.New( + &TemplateNotFoundError{Name: rs.Name}, + meep.Cause(err), + ) + } + + + for _, file := range files { + if !file.IsDir() && isResourceFile(file) { + p := path.Join(rp, file.Name()) + o, err := templateFile(c, &rs, p) + + if err != nil { + return nil, err + } + + output = append(output, o) + } + } + } + + return output, nil +} + +func templateFile(c *context.Context, rs *context.ResourceSet, filename string) (string, error) { + tpl, err := template.ParseFiles(filename) + + if err != nil { + return "", meep.New( + &TemplateNotFoundError{Name: filename}, + meep.Cause(err), + ) + } + + var b bytes.Buffer + + // Merge global and resourceset-specific values (don't override from global) + for k, v := range c.Global { + if _, ok := rs.Values[k]; !ok { + rs.Values[k] = v + } + } + + err = tpl.Execute(&b, rs.Values) + + if err != nil { + return "", meep.New( + &TemplatingError{}, + meep.Cause(err), + ) + } + + return b.String(), nil +} + +// Checks whether a file is a resource file (i.e. is YAML or JSON) +func isResourceFile(f os.FileInfo) bool { + return strings.HasSuffix(f.Name(), "yaml") || + strings.HasSuffix(f.Name(), "yml") || + strings.HasSuffix(f.Name(), "json") +} |