From 35c6a25382762fd6d525057dd458864215a91856 Mon Sep 17 00:00:00 2001
From: Mike Bierlee <m.bierlee@lostmoment.com>
Date: Fri, 7 Oct 2022 00:51:17 +0300
Subject: [PATCH] Add setter to easily assign values to path in config

---
 source/mirage/config.d | 100 ++++++++++++++++++++++++++++++++++++++++-
 1 file changed, 98 insertions(+), 2 deletions(-)

diff --git a/source/mirage/config.d b/source/mirage/config.d
index 953d100..88dc78c 100644
--- a/source/mirage/config.d
+++ b/source/mirage/config.d
@@ -349,7 +349,19 @@ class ConfigDictionary {
         return new ConfigDictionary(node);
     }
 
-    string createExceptionPath(ConfigPath path) {
+    /** 
+     * Assign a value at the given path.
+     * 
+     * Params:
+     *   configPath = Path where to assign the value to. If the path does not exist, it will be created.
+     *   value = Value to set at path.
+     */
+    void set(string configPath, string value) {
+        auto path = new ConfigPath(configPath);
+        rootNode = insertAtPath(rootNode, path, value);
+    }
+
+    private string createExceptionPath(ConfigPath path) {
         return "'" ~ path.path ~ "' (at '" ~ path.getCurrentPath() ~ "')";
     }
 
@@ -530,6 +542,28 @@ class ConfigDictionary {
 
         return result;
     }
+
+    private ConfigNode insertAtPath(ConfigNode node, ConfigPath path, const string value) {
+        auto nextSegment = path.getNextSegment();
+        if (nextSegment is null) {
+            return new ValueNode(value);
+        }
+
+        auto propertySegment = cast(PropertyPathSegment) nextSegment;
+        if (propertySegment is null) {
+            throw new Exception("Not yet implemented: cannot set array"); // todo
+        }
+
+        auto objectNode = node is null ? new ObjectNode() : cast(ObjectNode) node;
+        if (objectNode is null) {
+            throw new Exception("Not yet implemented: cannot set array"); // todo
+        }
+
+        auto childNodePtr = propertySegment.propertyName in objectNode.children;
+        ConfigNode childNode = childNodePtr !is null ? *childNodePtr : new ObjectNode();
+        objectNode.children[propertySegment.propertyName] = insertAtPath(childNode, path, value);
+        return objectNode;
+    }
 }
 
 /** 
@@ -1004,5 +1038,67 @@ version (unittest) {
         assert(config.get("escape room") == "\\\\");
     }
 
-    //TODO: Test null nodes should gracefully fail
+    @("Set value at path when path does not exist")
+    unittest {
+        auto config = new ConfigDictionary();
+        config.set("do.re.mi.fa.so", "buh");
+
+        assert(config.get("do.re.mi.fa.so") == "buh");
+    }
+
+    @("Set value at path when object at path exists")
+    unittest {
+        auto config = new ConfigDictionary(
+            new ObjectNode(
+                [
+                "one": new ObjectNode([
+                    "two": new ObjectNode([
+                            "mouseSound": "meep"
+                        ])
+                ])
+            ])
+        );
+
+        config.set("one.two.catSound", "meow");
+
+        assert(config.get("one.two.mouseSound") == "meep");
+        assert(config.get("one.two.catSound") == "meow");
+    }
+
+    @("Overwrite value at path")
+    unittest {
+        auto config = new ConfigDictionary(
+            new ObjectNode(
+                [
+                "one": new ObjectNode([
+                    "two": new ObjectNode([
+                            "mouseSound": "meep"
+                        ])
+                ])
+            ])
+        );
+
+        config.set("one.two.mouseSound", "yo");
+
+        assert(config.get("one.two.mouseSound") == "yo");
+    }
+
+    @("Diverge path completely")
+    unittest {
+        auto config = new ConfigDictionary(
+            new ObjectNode(
+                [
+                "one": new ObjectNode([
+                    "two": new ObjectNode([
+                            "mouseSound": "meep"
+                        ])
+                ])
+            ])
+        );
+
+        config.set("I.am", "baboon");
+
+        assert(config.get("one.two.mouseSound") == "meep");
+        assert(config.get("I.am") == "baboon");
+    }
 }