diff --git a/helm/helm.go b/helm/helm.go new file mode 100644 index 0000000..972708a --- /dev/null +++ b/helm/helm.go @@ -0,0 +1,117 @@ +package helm + +import ( + "os" + "path/filepath" + + "helm.sh/helm/v3/pkg/chart" + "helm.sh/helm/v3/pkg/chartutil" + "helm.sh/helm/v3/pkg/engine" +) + +type LangOpts struct { + Strict bool + LintMode bool + EnableDNS bool +} + +var defaultLangOpts = LangOpts{ + Strict: true, + LintMode: false, + EnableDNS: true, +} + +type Templater struct { + engine engine.Engine + chart chart.Chart +} + +func (t *Templater) Render(values map[string]interface{}) (map[string]string, error) { + t.chart.Values = values + return t.engine.Render(&t.chart, chartutil.Values(values)) +} + +func NewTemplaterWithLangOpts( + templates map[string][]byte, + langOpts *LangOpts, +) *Templater { + if langOpts == nil { + langOpts = &defaultLangOpts + } + + if templates == nil { + templates = map[string][]byte{ + "": []byte{}, + } + } + + ht := &Templater{ + engine: engine.Engine{ + Strict: langOpts.Strict, + LintMode: langOpts.LintMode, + EnableDNS: langOpts.EnableDNS, + }, + chart: chart.Chart{ + Raw: make([]*chart.File, 0, 0), + Metadata: &chart.Metadata{}, + Lock: &chart.Lock{}, + Templates: make([]*chart.File, len(templates), len(templates)), + Values: nil, + Schema: make([]byte, 0, 0), + Files: make([]*chart.File, 0, 0), + }, + } + + var i int + for name, data := range templates { + ht.chart.Templates[i] = &chart.File{ + Name: name, + Data: data, + } + i++ + } + return ht +} + +func NewTemplater(templates map[string][]byte) *Templater { + return NewTemplaterWithLangOpts(templates, nil) +} + +func NewTemplaterWithFiles( + tplFiles []string, + langOpts *LangOpts, +) (*Templater, error) { + tpls := map[string][]byte{} + var err error + for _, tf := range tplFiles { + tpls[filepath.Base(tf)], err = os.ReadFile(tf) + if err != nil { + return nil, err + } + } + return NewTemplaterWithLangOpts(tpls, langOpts), nil +} + +func NewTemplaterWithStr( + tplStr string, + langOpts *LangOpts, +) *Templater { + return NewTemplaterWithLangOpts( + map[string][]byte{ + "": []byte(tplStr), + }, + langOpts, + ) +} + +func NewTemplaterWithBytes( + tplBytes []byte, + langOpts *LangOpts, +) *Templater { + return NewTemplaterWithLangOpts( + map[string][]byte{ + "": tplBytes, + }, + langOpts, + ) +} diff --git a/helmCompat/helm.go b/helmCompat/helm.go deleted file mode 100644 index c7ba77f..0000000 --- a/helmCompat/helm.go +++ /dev/null @@ -1,25 +0,0 @@ -package helmCompat - -import ( - "helm.sh/helm/v3/pkg/chart" - "helm.sh/helm/v3/pkg/chartutil" - "helm.sh/helm/v3/pkg/engine" -) - -type HelmCompatLayer struct { - engine engine.Engine - chart chart.Chart - values chartutil.Values -} - -func NewHelmCompatLayer() *HelmCompatLayer { - return &HelmCompatLayer{} -} - -func NewHelmCompatLayerWithFiles(tplFiles []string) *HelmCompatLayer { - return NewHelmCompatLayer() -} - -func NewHelmCompatLayerWithStr(tplStr string) *HelmCompatLayer { - return NewHelmCompatLayer() -} diff --git a/main.go b/main.go index f8da464..7d74542 100644 --- a/main.go +++ b/main.go @@ -17,7 +17,7 @@ import ( sprig "github.com/Masterminds/sprig/v3" yaml "gopkg.in/yaml.v3" - helm "gitlab.com/CRThaze/spry/helmCompat" + "gitlab.com/CRThaze/spry/helm" ) type StrSet map[string]struct{} @@ -256,7 +256,7 @@ func mergeValues() (res map[string]interface{}) { func main() { var err error - var hc *helm.HelmCompatLayer + var ht *helm.Templater tpl := template.New("base").Funcs(sprig.FuncMap()) values := mergeValues() @@ -282,7 +282,7 @@ func main() { checkErr(err) if flags.helmCompat { - hc = helm.NewHelmCompatLayerWithStr(string(data)) + ht = helm.NewTemplaterWithBytes(data, nil) fmt.Fprintf(os.Stderr, "WARN: Helm Compatibility only partially implemented.\n") tpl, err = tpl.Parse(string(data)) } else { @@ -296,7 +296,7 @@ func main() { checkErr(err) if flags.helmCompat { - hc = helm.NewHelmCompatLayerWithStr(string(data)) + ht = helm.NewTemplaterWithBytes(data, nil) fmt.Fprintf(os.Stderr, "WARN: Helm Compatibility only partially implemented.\n") tpl, err = tpl.Parse(string(data)) } else { @@ -308,7 +308,8 @@ func main() { tplFiles := expandFiles(flag.Args()) if flags.helmCompat { fmt.Fprintf(os.Stderr, "WARN: Helm Compatibility only partially implemented.\n") - hc = helm.NewHelmCompatLayerWithFiles(tplFiles) + ht, err = helm.NewTemplaterWithFiles(tplFiles, nil) + checkErr(err) tpl, err = tpl.ParseFiles(tplFiles...) checkErr(err) for i, file := range tplFiles { @@ -329,7 +330,7 @@ func main() { } return } - if hc != nil { + if ht != nil { checkErr(tpl.Execute(os.Stdout, values)) } else { checkErr(tpl.Execute(os.Stdout, values))