helmfile · mumoshu · Jan 4, 2023 · Dec 28, 2022 · Dec 28, 2022 · Dec 28, 2022
@@ -61,6 +61,7 @@ jobs:
            plugin-secrets-version: 3.15.0
            plugin-diff-version: 3.5.0
            extra-helmfile-flags:
+           v1mode:
          - helm-version: v3.9.4
            kustomize-version: v4.5.7
            # We assume that the helm-secrets plugin is supposed to
@@ -71,23 +72,34 @@ jobs:
            plugin-secrets-version: 4.0.0
            plugin-diff-version: 3.6.0
            extra-helmfile-flags:
+           v1mode:
          - helm-version: v3.10.3
            kustomize-version: v4.4.1
            plugin-secrets-version: 3.15.0
            plugin-diff-version: 3.5.0
            extra-helmfile-flags:
+           v1mode:
          - helm-version: v3.10.3
            kustomize-version: v4.5.7
            plugin-secrets-version: 4.0.0
            plugin-diff-version: 3.6.0
            extra-helmfile-flags:
+           v1mode:
+         # Helmfile v1
+         - helm-version: v3.10.3
+           kustomize-version: v4.5.7
+           plugin-secrets-version: 4.0.0
+           plugin-diff-version: 3.6.0
+           extra-helmfile-flags:
+           v1mode: "true"
          # In case you need to test some optional helmfile features,
          # enable it via extra-helmfile-flags below.
          - helm-version: v3.10.3
            kustomize-version: v4.5.7
            plugin-secrets-version: 4.0.0
            plugin-diff-version: 3.6.0
            extra-helmfile-flags: "--enable-live-output"
+           v1mode:
     steps:
     - uses: actions/checkout@v2
     - name: Cache libraries
@@ -131,6 +143,7 @@ jobs:
         HELMFILE_HELM3: 1
         TERM: xterm
         EXTRA_HELMFILE_FLAGS: ${{ matrix.extra-helmfile-flags }}
+        HELMFILE_V1MODE: ${{ matrix.v1mode }}
       run: make integration
   e2e_tests:
     needs: tests

@@ -15,8 +15,8 @@ import (
 	"github.com/helmfile/helmfile/pkg/exectest"
 	ffs "github.com/helmfile/helmfile/pkg/filesystem"
 	"github.com/helmfile/helmfile/pkg/helmexec"
+	"github.com/helmfile/helmfile/pkg/runtime"
 	"github.com/helmfile/helmfile/pkg/testhelper"
-	"github.com/helmfile/helmfile/pkg/yaml"
 )
 
 func TestTemplate(t *testing.T) {
@@ -309,20 +309,21 @@ releases:
 }
 
 func TestTemplate_StrictParsing(t *testing.T) {
-	v := yaml.GoccyGoYaml
-	yaml.GoccyGoYaml = true
-	t.Cleanup(func() {
-		yaml.GoccyGoYaml = v
-	})
-
 	type testcase struct {
-		ns    string
-		error string
+		goccyGoYaml bool
+		ns          string
+		error       string
 	}
 
 	check := func(t *testing.T, tc testcase) {
 		t.Helper()
 
+		v := runtime.GoccyGoYaml
+		runtime.GoccyGoYaml = tc.goccyGoYaml
+		t.Cleanup(func() {
+			runtime.GoccyGoYaml = v
+		})
+
 		var helm = &exectest.Helm{
 			FailOnUnexpectedList: true,
 			FailOnUnexpectedDiff: true,
@@ -381,8 +382,9 @@ releases:
 		})
 	}
 
-	t.Run("fail due to known field", func(t *testing.T) {
+	t.Run("fail due to unknown field with goccy/go-yaml", func(t *testing.T) {
 		check(t, testcase{
+			goccyGoYaml: true,
 			error: `in ./helmfile.yaml: failed to read helmfile.yaml: reading document at index 1: [4:3] unknown field "foobar"
        2 | releases:
        3 | - name: app1
@@ -391,6 +393,14 @@ releases:
        5 |   chart: incubator/raw`,
 		})
 	})
+
+	t.Run("fail due to unknown field with gopkg.in/yaml.v2", func(t *testing.T) {
+		check(t, testcase{
+			goccyGoYaml: false,
+			error: `in ./helmfile.yaml: failed to read helmfile.yaml: reading document at index 1: yaml: unmarshal errors:
+  line 4: field foobar not found in type state.ReleaseSpec`,
+		})
+	})
 }
 
 func TestTemplate_CyclicInheritance(t *testing.T) {

@@ -9,7 +9,7 @@ import (
 	"path/filepath"
 	"reflect"
 	"regexp"
-	"runtime"
+	goruntime "runtime"
 	"strings"
 	"sync"
 	"testing"
@@ -25,6 +25,7 @@ import (
 	ffs "github.com/helmfile/helmfile/pkg/filesystem"
 	"github.com/helmfile/helmfile/pkg/helmexec"
 	"github.com/helmfile/helmfile/pkg/remote"
+	"github.com/helmfile/helmfile/pkg/runtime"
 	"github.com/helmfile/helmfile/pkg/state"
 	"github.com/helmfile/helmfile/pkg/testhelper"
 	"github.com/helmfile/helmfile/pkg/testutil"
@@ -4338,7 +4339,15 @@ myrelease4	testNamespace	true   	true     	id:myrelease1             	mychart1
 	assert.Equal(t, expected, out)
 }
 
-func TestSetValuesTemplate(t *testing.T) {
+func testSetValuesTemplate(t *testing.T, goccyGoYaml bool) {
+	t.Helper()
+
+	v := runtime.GoccyGoYaml
+	runtime.GoccyGoYaml = goccyGoYaml
+	t.Cleanup(func() {
+		runtime.GoccyGoYaml = v
+	})
+
 	files := map[string]string{
 		"/path/to/helmfile.yaml": `
 releases:
@@ -4357,7 +4366,7 @@ releases:
 `,
 	}
 	expectedValues := []interface{}{
-		map[interface{}]interface{}{"val1": "zipkin"},
+		map[string]interface{}{"val1": "zipkin"},
 		map[string]interface{}{"val2": "val2"}}
 	expectedSetValues := []state.SetValue{
 		{Name: "name-zipkin", Value: "val-zipkin"},
@@ -4402,7 +4411,17 @@ releases:
 	}
 }
 
+func TestSetValuesTemplate(t *testing.T) {
+	t.Run("with goccy/go-yaml", func(t *testing.T) {
+		testSetValuesTemplate(t, true)
+	})
+
+	t.Run("with gopkg.in/yaml.v2", func(t *testing.T) {
+		testSetValuesTemplate(t, false)
+	})
+}
+
 func location() string {
-	_, fn, line, _ := runtime.Caller(1)
+	_, fn, line, _ := goruntime.Caller(1)
 	return fmt.Sprintf("%s:%d", filepath.Base(fn), line)
 }
@@ -10,4 +10,6 @@ const (
 	Helm3                         = "HELMFILE_HELM3"
 	UpgradeNoticeDisabled         = "HELMFILE_UPGRADE_NOTICE_DISABLED"
 	V1Mode                        = "HELMFILE_V1MODE"
+	GoccyGoYaml                   = "HELMFILE_GOCCY_GOYAML"
+	CacheHome                     = "HELMFILE_CACHE_HOME"
 )
@@ -27,6 +27,10 @@ func init() {
 }
 
 func CacheDir() string {
+	if h := os.Getenv(envvar.CacheHome); h != "" {
+		return h
+	}
+
 	dir, err := os.UserCacheDir()
 	if err != nil {
 		// fall back to relative path with hidden directory

@@ -13,13 +13,23 @@ import (
 var (
 	V1Mode bool
 
+	// GoccyGoYaml is set to true in order to let Helmfile use
+	// goccy/go-yaml instead of gopkg.in/yaml.v2.
+	// It's false by default in Helmfile v0.x and true by default for Helmfile v1.x.
+	GoccyGoYaml bool
+
 	// We set this via ldflags at build-time so that we can use the
 	// value specified at the build time as the runtime default.
 	v1Mode string
 )
 
 func Info() string {
-	return fmt.Sprintf("V1 mode = %v", V1Mode)
+	yamlLib := "gopkg.in/yaml.v2"
+	if GoccyGoYaml {
+		yamlLib = "goccy/go-yaml"
+	}
+
+	return fmt.Sprintf("V1 mode = %v\nYAML library = %v", V1Mode, yamlLib)
 }
 
 func init() {
@@ -34,4 +44,14 @@ func init() {
 	default:
 		V1Mode, _ = strconv.ParseBool(v1Mode)
 	}
+
+	// You can switch the YAML library at runtime via an envvar:
+	switch os.Getenv(envvar.GoccyGoYaml) {
+	case "true":
+		GoccyGoYaml = true
+	case "false":
+		GoccyGoYaml = false
+	default:
+		GoccyGoYaml = V1Mode
+	}
 }
@@ -3,6 +3,7 @@ package state
 import (
 	"fmt"
 
+	"github.com/helmfile/helmfile/pkg/maputil"
 	"github.com/helmfile/helmfile/pkg/tmpl"
 	"github.com/helmfile/helmfile/pkg/yaml"
 )
@@ -107,13 +108,18 @@ func (r ReleaseSpec) ExecuteTemplateExpressions(renderer *tmpl.FileRenderer) (*R
 					return nil, fmt.Errorf("failed executing template expressions in release \"%s\".values[%d] = \"%v\": %v", r.Name, i, string(serialized), err)
 				}
 
-				var deserialized map[interface{}]interface{}
+				var deserialized map[string]interface{}
 
 				if err := yaml.Unmarshal(s.Bytes(), &deserialized); err != nil {
 					return nil, fmt.Errorf("failed executing template expressions in release \"%s\".values[%d] = \"%v\": %v", r.Name, i, ts, err)
 				}
 
-				result.ValuesTemplate[i] = deserialized
+				m, err := maputil.CastKeysToStrings(deserialized)
+				if err != nil {
+					return nil, fmt.Errorf("failed executing template expressions in release \"%s\".values[%d] = \"%v\": %v", r.Name, i, ts, err)
+				}
+
+				result.ValuesTemplate[i] = m
 			}
 		}
 
@@ -130,6 +136,12 @@ func (r ReleaseSpec) ExecuteTemplateExpressions(renderer *tmpl.FileRenderer) (*R
 				return nil, fmt.Errorf("failed executing template expressions in release \"%s\".values[%d] = \"%s\": %v", r.Name, i, ts, err)
 			}
 			result.Values[i] = s.String()
+		case map[interface{}]interface{}:
+			m, err := maputil.CastKeysToStrings(ts)
+			if err != nil {
+				return nil, fmt.Errorf("failed executing template expressions in release \"%s\".values[%d] = \"%s\": %v", r.Name, i, ts, err)
+			}
+			result.Values[i] = m
 		}
 	}
 

@@ -121,7 +121,7 @@ func TestHelmState_executeTemplates(t *testing.T) {
 				Verify:    nil,
 				Name:      "app",
 				Namespace: "dev",
-				Values:    []interface{}{map[interface{}]interface{}{"key": "app-val0"}},
+				Values:    []interface{}{map[string]interface{}{"key": "app-val0"}},
 			},
 		},
 	}

@@ -17,6 +17,7 @@ import (
 
 	"github.com/helmfile/helmfile/pkg/envvar"
 	"github.com/helmfile/helmfile/pkg/helmexec"
+	"github.com/helmfile/helmfile/pkg/maputil"
 	"github.com/helmfile/helmfile/pkg/yaml"
 )
 
@@ -286,11 +287,17 @@ func ToYaml(v interface{}) (string, error) {
 }
 
 func FromYaml(str string) (Values, error) {
-	m := Values{}
+	m := map[string]interface{}{}
 
 	if err := yaml.Unmarshal([]byte(str), &m); err != nil {
 		return nil, fmt.Errorf("%s, offending yaml: %s", err, str)
 	}
+
+	m, err := maputil.CastKeysToStrings(m)
+	if err != nil {
+		return nil, fmt.Errorf("%s, offending yaml: %s", err, str)
+	}
+
 	return m, nil
 }