mirror of
https://github.com/mudler/luet.git
synced 2025-09-08 18:49:39 +00:00
♻️ Use sprig instead of helm
This alleviates luet build dependency graph while it simplifies template logic too.
This commit is contained in:
164
pkg/api/core/template/string.go
Normal file
164
pkg/api/core/template/string.go
Normal file
@@ -0,0 +1,164 @@
|
||||
// Copyright © 2022 Ettore Di Giacinto <mudler@mocaccino.org>
|
||||
//
|
||||
// This program is free software; you can redistribute it and/or modify
|
||||
// it under the terms of the GNU General Public License as published by
|
||||
// the Free Software Foundation; either version 2 of the License, or
|
||||
// (at your option) any later version.
|
||||
//
|
||||
// This program is distributed in the hope that it will be useful,
|
||||
// but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||
// GNU General Public License for more details.
|
||||
//
|
||||
// You should have received a copy of the GNU General Public License along
|
||||
// with this program; if not, see <http://www.gnu.org/licenses/>.
|
||||
|
||||
package template
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"io/ioutil"
|
||||
"path/filepath"
|
||||
"strings"
|
||||
"text/template"
|
||||
|
||||
fileHelper "github.com/mudler/luet/pkg/helpers/file"
|
||||
"github.com/pkg/errors"
|
||||
"gopkg.in/yaml.v2"
|
||||
|
||||
"github.com/Masterminds/sprig/v3"
|
||||
"github.com/imdario/mergo"
|
||||
)
|
||||
|
||||
// String templates a string with the interface
|
||||
func String(t string, i interface{}) (string, error) {
|
||||
b := bytes.NewBuffer([]byte{})
|
||||
tmpl, err := template.New("").Funcs(sprig.TxtFuncMap()).Parse(t)
|
||||
if err != nil {
|
||||
return "", err
|
||||
}
|
||||
|
||||
err = tmpl.Option("missingkey=zero").Execute(b, i)
|
||||
|
||||
return b.String(), err
|
||||
}
|
||||
|
||||
// Render renders the template string like helm
|
||||
func Render(strings []string, values, d map[string]interface{}) (string, error) {
|
||||
|
||||
// We slurp all the files into one here. This is not elegant, but still works.
|
||||
// As a reminder, the files passed here have on the head the templates in the 'templates/' folder
|
||||
// of each luet tree, and it have at the bottom the package buildpsec to be templated.
|
||||
// TODO: Replace by correctly populating the files so that the helm render engine templates it
|
||||
// correctly
|
||||
toTemplate := ""
|
||||
for _, f := range strings {
|
||||
toTemplate += f
|
||||
}
|
||||
|
||||
input := map[string]interface{}{"Values": d}
|
||||
|
||||
if err := mergo.Merge(&input, map[string]interface{}{"Values": values}); err != nil {
|
||||
return "", err
|
||||
}
|
||||
|
||||
return String(toTemplate, input)
|
||||
}
|
||||
|
||||
// ReadFilesInDir reads a list of paths and reads all yaml file inside. It returns a
|
||||
// slice of strings with the raw content of the yaml
|
||||
func FilesInDir(path []string) (res []string, err error) {
|
||||
for _, t := range path {
|
||||
var rel string
|
||||
rel, err = fileHelper.Rel2Abs(t)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
if !fileHelper.Exists(rel) {
|
||||
continue
|
||||
}
|
||||
var files []string
|
||||
files, err = fileHelper.ListDir(rel)
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
|
||||
for _, f := range files {
|
||||
if strings.ToLower(filepath.Ext(f)) == ".yaml" {
|
||||
res = append(res, f)
|
||||
}
|
||||
}
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
// ReadFiles reads all the given files and returns a slice of []*chart.File
|
||||
// containing the raw content and the file name for each file
|
||||
func ReadFiles(s ...string) (res []string) {
|
||||
for _, c := range s {
|
||||
raw, err := ioutil.ReadFile(c)
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
res = append(res, string(raw))
|
||||
}
|
||||
|
||||
return
|
||||
}
|
||||
|
||||
type templatedata map[string]interface{}
|
||||
|
||||
// UnMarshalValues unmarshal values files and joins them into a unique templatedata
|
||||
// the join happens from right to left, so any rightmost value file overwrites the content of the ones before it.
|
||||
func UnMarshalValues(values []string) (templatedata, error) {
|
||||
dst := templatedata{}
|
||||
if len(values) > 0 {
|
||||
for _, bv := range reverse(values) {
|
||||
current := templatedata{}
|
||||
|
||||
defBuild, err := ioutil.ReadFile(bv)
|
||||
if err != nil {
|
||||
return nil, errors.Wrap(err, "rendering file "+bv)
|
||||
}
|
||||
err = yaml.Unmarshal(defBuild, ¤t)
|
||||
if err != nil {
|
||||
return nil, errors.Wrap(err, "rendering file "+bv)
|
||||
}
|
||||
if err := mergo.Merge(&dst, current); err != nil {
|
||||
return nil, errors.Wrap(err, "merging values file "+bv)
|
||||
}
|
||||
}
|
||||
}
|
||||
return dst, nil
|
||||
}
|
||||
|
||||
func reverse(s []string) []string {
|
||||
for i, j := 0, len(s)-1; i < j; i, j = i+1, j-1 {
|
||||
s[i], s[j] = s[j], s[i]
|
||||
}
|
||||
return s
|
||||
}
|
||||
|
||||
// RenderWithValues render a group of files with values files
|
||||
func RenderWithValues(rawFiles []string, valuesFile string, defaultFile ...string) (string, error) {
|
||||
if !fileHelper.Exists(valuesFile) {
|
||||
return "", errors.New("file does not exist: " + valuesFile)
|
||||
}
|
||||
val, err := ioutil.ReadFile(valuesFile)
|
||||
if err != nil {
|
||||
return "", errors.Wrap(err, "reading file: "+valuesFile)
|
||||
}
|
||||
|
||||
var values templatedata
|
||||
if err = yaml.Unmarshal(val, &values); err != nil {
|
||||
return "", errors.Wrap(err, "unmarshalling values")
|
||||
}
|
||||
|
||||
dst, err := UnMarshalValues(defaultFile)
|
||||
if err != nil {
|
||||
return "", errors.Wrap(err, "unmarshalling values")
|
||||
}
|
||||
|
||||
return Render(ReadFiles(rawFiles...), values, dst)
|
||||
}
|
Reference in New Issue
Block a user