2017-12-16 03:37:45 +00:00
|
|
|
package templates
|
|
|
|
|
|
|
|
import (
|
|
|
|
"bytes"
|
2019-03-12 20:40:19 +00:00
|
|
|
"encoding/json"
|
2019-08-13 20:34:49 +00:00
|
|
|
"fmt"
|
2019-08-02 18:28:09 +00:00
|
|
|
"github.com/blang/semver"
|
|
|
|
"github.com/rancher/kontainer-driver-metadata/rke/templates"
|
|
|
|
"github.com/sirupsen/logrus"
|
2017-12-16 03:37:45 +00:00
|
|
|
"text/template"
|
2019-02-26 23:14:01 +00:00
|
|
|
|
2019-06-17 20:52:15 +00:00
|
|
|
"github.com/rancher/norman/types/convert"
|
2019-05-28 18:51:53 +00:00
|
|
|
"github.com/rancher/rke/metadata"
|
2017-12-16 03:37:45 +00:00
|
|
|
)
|
|
|
|
|
2018-02-01 21:28:31 +00:00
|
|
|
func CompileTemplateFromMap(tmplt string, configMap interface{}) (string, error) {
|
2017-12-16 03:37:45 +00:00
|
|
|
out := new(bytes.Buffer)
|
2019-03-12 20:40:19 +00:00
|
|
|
t := template.Must(template.New("compiled_template").Funcs(template.FuncMap{"GetKubednsStubDomains": GetKubednsStubDomains}).Parse(tmplt))
|
2017-12-16 03:37:45 +00:00
|
|
|
if err := t.Execute(out, configMap); err != nil {
|
|
|
|
return "", err
|
|
|
|
}
|
|
|
|
return out.String(), nil
|
|
|
|
}
|
2019-01-08 20:04:42 +00:00
|
|
|
|
2019-08-13 20:34:49 +00:00
|
|
|
func GetVersionedTemplates(templateName string, data map[string]interface{}, k8sVersion string) (string, error) {
|
2019-06-17 20:52:15 +00:00
|
|
|
if template, ok := data[templateName]; ok {
|
2019-08-13 20:34:49 +00:00
|
|
|
return convert.ToString(template), nil
|
2019-06-17 20:52:15 +00:00
|
|
|
}
|
2019-08-02 18:28:09 +00:00
|
|
|
return getTemplate(templateName, k8sVersion)
|
2019-01-08 20:04:42 +00:00
|
|
|
}
|
2019-03-12 20:40:19 +00:00
|
|
|
|
|
|
|
func GetKubednsStubDomains(stubDomains map[string][]string) string {
|
|
|
|
json, _ := json.Marshal(stubDomains)
|
|
|
|
return string(json)
|
|
|
|
}
|
2019-05-28 18:51:53 +00:00
|
|
|
|
2019-08-13 20:34:49 +00:00
|
|
|
func getTemplate(templateName, k8sVersion string) (string, error) {
|
2019-08-02 18:28:09 +00:00
|
|
|
versionData := metadata.K8sVersionToTemplates[templateName]
|
|
|
|
toMatch, err := semver.Make(k8sVersion[1:])
|
|
|
|
if err != nil {
|
2019-08-13 20:34:49 +00:00
|
|
|
return "", fmt.Errorf("k8sVersion not sem-ver %s %v", k8sVersion, err)
|
2019-08-02 18:28:09 +00:00
|
|
|
}
|
|
|
|
for k := range versionData {
|
|
|
|
testRange, err := semver.ParseRange(k)
|
|
|
|
if err != nil {
|
|
|
|
logrus.Errorf("range for %s not sem-ver %v %v", templateName, testRange, err)
|
|
|
|
continue
|
|
|
|
}
|
|
|
|
if testRange(toMatch) {
|
2019-08-13 20:34:49 +00:00
|
|
|
return metadata.K8sVersionToTemplates[templates.TemplateKeys][versionData[k]], nil
|
2019-08-02 18:28:09 +00:00
|
|
|
}
|
|
|
|
}
|
2019-08-13 20:34:49 +00:00
|
|
|
return "", fmt.Errorf("no %s template found for k8sVersion %s", templateName, k8sVersion)
|
2019-05-28 18:51:53 +00:00
|
|
|
}
|