rke/templates/templates.go

package templates

import (
	"bytes"
	"text/template"
)

var VersionedTemplate = map[string]map[string]string{
	"calico": map[string]string{
		"v1.13.1-rancher1-1": CalicoTemplateV113,
		"default":            CalicoTemplateV112,
	},
	"canal": map[string]string{
		"v1.13.1-rancher1-1": CanalTemplateV113,
		"default":            CanalTemplateV112,
	},
}

func CompileTemplateFromMap(tmplt string, configMap interface{}) (string, error) {
	out := new(bytes.Buffer)
	t := template.Must(template.New("compiled_template").Parse(tmplt))
	if err := t.Execute(out, configMap); err != nil {
		return "", err
	}
	return out.String(), nil
}

func GetVersionedTemplates(templateName string, k8sVersion string) string {
	versionedTemplate := VersionedTemplate[templateName]
	if _, ok := versionedTemplate[k8sVersion]; ok {
		return versionedTemplate[k8sVersion]
	}
	return versionedTemplate["default"]
}