From 14f7a86c1c7cf7f7ade1c7982fc8dda5056d1683 Mon Sep 17 00:00:00 2001
From: Mauro Morales <mauro.morales@spectrocloud.com>
Date: Fri, 3 May 2024 18:01:32 +0200
Subject: [PATCH] Add overwrites (#109)

Relates to kairos-io/kairos#2492

Signed-off-by: Mauro Morales <mauro.morales@spectrocloud.com>
---
 collector/collector.go      | 11 +++++-
 collector/collector_test.go | 77 +++++++++++++++++++++++++++++++++++++
 collector/options.go        |  8 ++++
 3 files changed, 95 insertions(+), 1 deletion(-)

diff --git a/collector/collector.go b/collector/collector.go
index a807c00..40b4594 100644
--- a/collector/collector.go
+++ b/collector/collector.go
@@ -235,7 +235,16 @@ func Scan(o *Options, filter func(d []byte) ([]byte, error)) (*Config, error) {
 		}
 	}
 
-	return configs.Merge()
+	mergedConfig, err := configs.Merge()
+	if err != nil {
+		return mergedConfig, err
+	}
+
+	if o.Overwrites != "" {
+		yaml.Unmarshal([]byte(o.Overwrites), &mergedConfig) //nolint:errcheck
+	}
+
+	return mergedConfig, nil
 }
 
 func allFiles(dir []string) []string {
diff --git a/collector/collector_test.go b/collector/collector_test.go
index d49e0b1..9dbe9fd 100644
--- a/collector/collector_test.go
+++ b/collector/collector_test.go
@@ -661,6 +661,83 @@ stages:
 			})
 		})
 
+		Context("With Overwrittes", func() {
+			var tmpDir1 string
+			var err error
+
+			BeforeEach(func() {
+				tmpDir1, err = os.MkdirTemp("", "config1")
+				Expect(err).ToNot(HaveOccurred())
+				err := os.WriteFile(path.Join(tmpDir1, "local_config_1.yaml"), []byte(`#cloud-config
+install:
+  auto: false
+foo: bar
+stages:
+  initramfs:
+    - users:
+        kairos:
+          groups:
+            - sudo
+          passwd: kairos
+`), os.ModePerm)
+				Expect(err).ToNot(HaveOccurred())
+			})
+
+			AfterEach(func() {
+				err = os.RemoveAll(tmpDir1)
+				Expect(err).ToNot(HaveOccurred())
+			})
+
+			It("replaces completely the keys given by the overwrite", func() {
+				o := &Options{}
+				overwriteYaml := `#cloud-config
+install:
+  auto: true
+options:
+  device: /dev/sda
+stages:
+  initramfs:
+    - users:
+        kairos:
+          groups:
+            - sudo
+          passwd: kairos
+        foobar:
+          groups:
+            - sudo
+          passwd: barbaz
+`
+				err = o.Apply(
+					Directories(tmpDir1),
+					Overwrites(overwriteYaml),
+				)
+				Expect(err).ToNot(HaveOccurred())
+
+				c, err := Scan(o, FilterKeysTestMerge)
+				Expect(err).ToNot(HaveOccurred())
+
+				Expect(c.String()).To(Equal(`#cloud-config
+
+foo: bar
+install:
+    auto: true
+options:
+    device: /dev/sda
+stages:
+    initramfs:
+        - users:
+            foobar:
+                groups:
+                    - sudo
+                passwd: barbaz
+            kairos:
+                groups:
+                    - sudo
+                passwd: kairos
+`))
+			})
+		})
+
 		Context("Deep merge maps within arrays", func() {
 			var cmdLinePath, tmpDir1 string
 			var err error
diff --git a/collector/options.go b/collector/options.go
index b665934..9c050cd 100644
--- a/collector/options.go
+++ b/collector/options.go
@@ -12,6 +12,7 @@ type Options struct {
 	NoLogs           bool
 	StrictValidation bool
 	Readers          []io.Reader
+	Overwrites       string
 }
 
 type Option func(o *Options) error
@@ -72,3 +73,10 @@ func Readers(r ...io.Reader) Option {
 		return nil
 	}
 }
+
+func Overwrites(m string) Option {
+	return func(o *Options) error {
+		o.Overwrites = m
+		return nil
+	}
+}