Add support for Kustomize components

Fix #753

Signed-off-by: Kristian Klausen <[email protected]>

Author: klausenbusk

api/v1beta2/kustomization_types.go

@@ -153,6 +153,10 @@ type KustomizationSpec struct {
 	// +kubebuilder:validation:Enum=none;client;server
 	// +optional
 	Validation string `json:"validation,omitempty"`
+
+	// Components specifies relative paths to specifications of other Components
+	// +optional
+	Components []string `json:"components,omitempty"`
 }
 
 // Decryption defines how decryption is handled for Kubernetes manifests.

api/v1beta2/zz_generated.deepcopy.go

@@ -585,6 +585,12 @@ spec:
             description: KustomizationSpec defines the configuration to calculate
               the desired state from a Source using Kustomize.
             properties:
+              components:
+                description: Components specifies relative paths to specifications
+                  of other Components
+                items:
+                  type: string
+                type: array
               decryption:
                 description: Decrypt Kubernetes secrets before applying them on the
                   cluster.

config/crd/bases/kustomize.toolkit.fluxcd.io_kustomizations.yaml

@@ -79,6 +79,24 @@ func (kg *KustomizeGenerator) WriteFile(dirPath string) (string, error) {
 		})
 	}
 
+	for _, m := range kg.kustomization.Spec.Components {
+		// From: https://github.com/kubernetes-sigs/kustomize/blob/84bd402cc0662c5df3f109c4f80c22611243c5f9/api/internal/git/repospec.go#L231-L239
+		// with "file://" removed/
+		for _, p := range []string{
+			// Order matters here.
+			"git::", "gh:", "ssh://", "https://", "http://",
+			"git@", "github.com:", "github.com/"} {
+			if len(p) < len(m) && strings.ToLower(m[:len(p)]) == p {
+				return "", fmt.Errorf("component path '%s' must be local", m)
+			}
+		}
+
+		if filepath.IsAbs(m) || filepath.IsAbs(strings.TrimPrefix(strings.ToLower(m), "file://")) {
+			return "", fmt.Errorf("component path '%s' must be relative", m)
+		}
+		kus.Components = append(kus.Components, m)
+	}
+
 	for _, m := range kg.kustomization.Spec.PatchesStrategicMerge {
 		kus.PatchesStrategicMerge = append(kus.PatchesStrategicMerge, kustypes.PatchStrategicMerge(m.Raw))
 	}

internal/generator/generator.go

@@ -93,3 +93,52 @@ func TestGenerator_WriteFile(t *testing.T) {
 		})
 	}
 }
+
+func TestGenerator_Components(t *testing.T) {
+	tests := []struct {
+		name               string
+		dir                string
+		fluxComponents     []string
+		expectedComponents []string
+	}{
+		{
+			name:               "test kustomization.yaml with components and Flux Kustomization without components",
+			dir:                "components",
+			fluxComponents:     []string{},
+			expectedComponents: []string{"componentA"},
+		},
+		{
+			name:               "test kustomization.yaml without components and Flux Kustomization with components",
+			dir:                "zero-components",
+			fluxComponents:     []string{"componentB", "componentC"},
+			expectedComponents: []string{"componentB", "componentC"},
+		},
+		{
+			name:               "test kustomization.yaml with components and Flux Kustomization with components",
+			dir:                "components",
+			fluxComponents:     []string{"componentB", "componentC"},
+			expectedComponents: []string{"componentA", "componentB", "componentC"},
+		},
+	}
+
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			g := NewWithT(t)
+			tmpDir := t.TempDir()
+			g.Expect(copy.Copy("./testdata/components", tmpDir)).To(Succeed())
+			ks := v1beta2.Kustomization{
+				Spec: v1beta2.KustomizationSpec{
+					Components: tt.fluxComponents,
+				},
+			}
+			kfile, err := NewGenerator(filepath.Join(tmpDir, tt.dir), &ks).WriteFile(filepath.Join(tmpDir, tt.dir))
+			g.Expect(err).ToNot(HaveOccurred())
+
+			kfileYAML, err := os.ReadFile(kfile)
+			g.Expect(err).ToNot(HaveOccurred())
+			var k kustypes.Kustomization
+			g.Expect(k.Unmarshal(kfileYAML)).To(Succeed())
+			g.Expect(k.Components).Should(Equal(tt.expectedComponents))
+		})
+	}
+}

internal/generator/generator_test.go

@@ -0,0 +1,7 @@
+---
+apiVersion: kustomize.config.k8s.io/v1beta1
+kind: Kustomization
+resources:
+  - configmap.yaml
+components:
+  - componentA

internal/generator/testdata/components/components/kustomization.yaml

@@ -0,0 +1,5 @@
+---
+apiVersion: kustomize.config.k8s.io/v1beta1
+kind: Kustomization
+resources:
+  - configmap.yaml