Merge pull request #124 from weaveworks/add-to-yaml-function

Add toYaml function to the template context.

Author: bigkevmcd

controllers/templates/renderer.go

@@ -251,6 +251,14 @@ func makeTemplateFunctions() template.FuncMap {
 
 		return def
 	}
+	f["toYaml"] = func(v interface{}) string {
+		data, err := syaml.Marshal(v)
+		if err != nil {
+			// Swallow errors inside of a template.
+			return ""
+		}
+		return strings.TrimSuffix(string(data), "\n")
+	}
 
 	return f
 }

controllers/templates/renderer_test.go

@@ -418,6 +418,56 @@ func TestRender(t *testing.T) {
 				},
 			},
 		},
+		{
+			name: "toYaml function",
+			elements: []apiextensionsv1.JSON{
+				{Raw: []byte(`{"env":"testing","depends":[{"name":"testing"}]}`)},
+			},
+			setOptions: []func(*templatesv1.GitOpsSet){
+				func(s *templatesv1.GitOpsSet) {
+					s.ObjectMeta.Annotations = map[string]string{
+						"templates.weave.works/delimiters": "((,))",
+					}
+					s.Spec.Templates = []templatesv1.GitOpsSetTemplate{
+						{
+							Content: runtime.RawExtension{
+								Raw: []byte(`{"apiVersion":"kustomize.toolkit.fluxcd.io/v1beta2","kind":"Kustomization","metadata":{"name":"testing-demo"},"spec":{"dependsOn":"(( .Element.depends | toYaml | nindent 8 ))","interval":"5m","path":"./examples/kustomize/environments/(( .Element.env ))","prune":true,"sourceRef":{"kind":"GitRepository","name":"go-demo-repo"}}}`),
+							},
+						},
+					}
+				},
+			},
+			want: []*unstructured.Unstructured{
+				{
+					Object: map[string]interface{}{
+						"apiVersion": "kustomize.toolkit.fluxcd.io/v1beta2",
+						"kind":       "Kustomization",
+						"metadata": map[string]interface{}{
+							"name":      "testing-demo",
+							"namespace": "demo",
+							"labels": map[string]interface{}{
+								"templates.weave.works/name":      "test-gitops-set",
+								"templates.weave.works/namespace": "demo",
+							},
+						},
+						"spec": map[string]interface{}{
+							"dependsOn": []any{
+								map[string]any{
+									"name": "testing",
+								},
+							},
+							"interval": "5m",
+							"path":     "./examples/kustomize/environments/testing",
+							"prune":    true,
+							"sourceRef": map[string]interface{}{
+								"kind": "GitRepository",
+								"name": "go-demo-repo",
+							},
+						},
+					},
+				},
+			},
+		},
 	}
 
 	for _, tt := range generatorTests {

paul.yaml

@@ -0,0 +1,40 @@
+apiVersion: templates.weave.works/v1alpha1
+kind: GitOpsSet
+metadata:
+  annotations:
+    templates.weave.works/delimiters: "((,))"
+  labels:
+    app.kubernetes.io/name: gitopsset
+  name: gitopsset-sample
+spec:
+  generators:
+  - list:
+      elements:
+      - env: dev
+        team: dev-team
+        depends:
+        - name: testing
+      - env: production
+        team: ops-team
+        depends: []
+      - env: staging
+        team: ops-team
+        depends: []
+  templates:
+    - content:
+        kind: Kustomization
+        apiVersion: kustomize.toolkit.fluxcd.io/v1beta2
+        metadata:
+          name: "(( .Element.env ))-demo"
+          labels:
+            app.kubernetes.io/name: go-demo
+            app.kubernetes.io/instance: "(( .Element.env ))"
+            com.example/team: "(( .Element.team ))"
+        spec:
+          interval: 5m
+          path: "./examples/kustomize/environments/(( .Element.env ))"
+          prune: true
+          sourceRef:
+            kind: GitRepository
+            name: go-demo-repo
+          dependsOn: (( .Element.depends | toYaml | nindent 8))