136 lines
3.5 KiB
Go
136 lines
3.5 KiB
Go
package state
|
|
|
|
import (
|
|
"fmt"
|
|
"reflect"
|
|
|
|
"github.com/helmfile/helmfile/pkg/tmpl"
|
|
"gopkg.in/yaml.v2"
|
|
)
|
|
|
|
func (st *HelmState) Values() map[string]interface{} {
|
|
if st.RenderedValues == nil {
|
|
panic("[bug] RenderedValues is nil")
|
|
}
|
|
|
|
return st.RenderedValues
|
|
}
|
|
|
|
func (st *HelmState) createReleaseTemplateData(release *ReleaseSpec, vals map[string]interface{}) releaseTemplateData {
|
|
tmplData := releaseTemplateData{
|
|
Environment: st.Env,
|
|
KubeContext: st.OverrideKubeContext,
|
|
Namespace: st.OverrideNamespace,
|
|
Chart: st.OverrideChart,
|
|
Values: vals,
|
|
Release: releaseTemplateDataRelease{
|
|
Name: release.Name,
|
|
Chart: release.Chart,
|
|
Namespace: release.Namespace,
|
|
Labels: release.Labels,
|
|
KubeContext: release.KubeContext,
|
|
},
|
|
}
|
|
tmplData.StateValues = &tmplData.Values
|
|
return tmplData
|
|
}
|
|
|
|
func getBoolRefFromStringTemplate(templateRef string) (*bool, error) {
|
|
var result bool
|
|
if err := yaml.Unmarshal([]byte(templateRef), &result); err != nil {
|
|
return nil, fmt.Errorf("failed deserialising string %s: %v", templateRef, err)
|
|
}
|
|
return &result, nil
|
|
}
|
|
|
|
func updateBoolTemplatedValues(r *ReleaseSpec) error {
|
|
|
|
if r.InstalledTemplate != nil {
|
|
if installed, err := getBoolRefFromStringTemplate(*r.InstalledTemplate); err != nil {
|
|
return fmt.Errorf("installedTemplate: %v", err)
|
|
} else {
|
|
r.InstalledTemplate = nil
|
|
r.Installed = installed
|
|
}
|
|
}
|
|
|
|
if r.WaitTemplate != nil {
|
|
if wait, err := getBoolRefFromStringTemplate(*r.WaitTemplate); err != nil {
|
|
return fmt.Errorf("waitTemplate: %v", err)
|
|
} else {
|
|
r.WaitTemplate = nil
|
|
r.Wait = wait
|
|
}
|
|
}
|
|
|
|
if r.TillerlessTemplate != nil {
|
|
if tillerless, err := getBoolRefFromStringTemplate(*r.TillerlessTemplate); err != nil {
|
|
return fmt.Errorf("tillerlessTemplate: %v", err)
|
|
} else {
|
|
r.TillerlessTemplate = nil
|
|
r.Tillerless = tillerless
|
|
}
|
|
}
|
|
|
|
if r.VerifyTemplate != nil {
|
|
if verify, err := getBoolRefFromStringTemplate(*r.VerifyTemplate); err != nil {
|
|
return fmt.Errorf("verifyTemplate: %v", err)
|
|
} else {
|
|
r.VerifyTemplate = nil
|
|
r.Verify = verify
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (st *HelmState) ExecuteTemplates() (*HelmState, error) {
|
|
r := *st
|
|
|
|
vals := st.Values()
|
|
|
|
for i, rt := range st.Releases {
|
|
release := rt
|
|
if release.KubeContext == "" {
|
|
release.KubeContext = r.HelmDefaults.KubeContext
|
|
}
|
|
if release.Labels == nil {
|
|
release.Labels = map[string]string{}
|
|
}
|
|
for k, v := range st.CommonLabels {
|
|
release.Labels[k] = v
|
|
}
|
|
if len(release.ApiVersions) == 0 {
|
|
release.ApiVersions = st.ApiVersions
|
|
}
|
|
if release.KubeVersion == "" {
|
|
release.KubeVersion = st.KubeVersion
|
|
}
|
|
|
|
successFlag := false
|
|
for it, prev := 0, &release; it < 6; it++ {
|
|
tmplData := st.createReleaseTemplateData(prev, vals)
|
|
renderer := tmpl.NewFileRenderer(st.readFile, st.basePath, tmplData)
|
|
r, err := release.ExecuteTemplateExpressions(renderer)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("failed executing templates in release \"%s\".\"%s\": %v", st.FilePath, release.Name, err)
|
|
}
|
|
if reflect.DeepEqual(prev, r) {
|
|
successFlag = true
|
|
if err := updateBoolTemplatedValues(r); err != nil {
|
|
return nil, fmt.Errorf("failed executing templates in release \"%s\".\"%s\": %v", st.FilePath, release.Name, err)
|
|
}
|
|
st.Releases[i] = *r
|
|
break
|
|
}
|
|
prev = r
|
|
}
|
|
if !successFlag {
|
|
return nil, fmt.Errorf("failed executing templates in release \"%s\".\"%s\": %s", st.FilePath, release.Name,
|
|
"recursive references can't be resolved")
|
|
}
|
|
}
|
|
|
|
return &r, nil
|
|
}
|