From b2301041a89a365a5df8efcafe35f5e9c1a2f202 Mon Sep 17 00:00:00 2001
From: Reza Behzadan <rbehzadan@gmail.com>
Date: Wed, 21 Feb 2024 00:47:21 +0330
Subject: [PATCH] Add GetInheritedValue()

---
 dict/dict.go      | 74 +++++++++++++++++++++++++++++++++++++++++++++--
 dict/dict_test.go |  4 +--
 main.go           |  2 +-
 3 files changed, 75 insertions(+), 5 deletions(-)

diff --git a/dict/dict.go b/dict/dict.go
index 25196dc..f2279e1 100644
--- a/dict/dict.go
+++ b/dict/dict.go
@@ -29,7 +29,7 @@ func (d Dict) Print() {
 	fmt.Println(pp)
 }
 
-func (d Dict) GetValue(path string) Dict {
+func (d Dict) GetInheritedValueAll(path string) Dict {
 	var value Dict
 	current := make(Dict)
 	g := make(Dict)
@@ -42,7 +42,6 @@ func (d Dict) GetValue(path string) Dict {
 			current.SoftMergeWith(g)
 		}
 		if !current.HasKey(key) {
-			value = nil
 			break
 		}
 		v := current[key]
@@ -57,6 +56,77 @@ func (d Dict) GetValue(path string) Dict {
 		}
 		if IsDict(v) {
 			current = v.(Dict)
+		} else {
+			break
+		}
+	}
+	return value
+}
+
+func (d Dict) GetInheritedValue(path string) Dict {
+	var value Dict
+	current := make(Dict)
+	g := make(Dict)
+	current.CopyFrom(d)
+	keys := strings.Split(path, ".")
+	lastIndex := len(keys) - 1
+	for i, key := range keys {
+		if current.HasKey("globals") {
+			g.MergeWith(current["globals"].(Dict))
+			current.SoftMergeWith(g)
+		}
+		if !current.HasKey(key) {
+			break
+		}
+		v := current[key]
+		if i == lastIndex {
+			if vv, ok := v.(Dict); ok {
+				if g.HasKey(key) {
+					if gv, ok := g[key].(Dict); ok {
+						vv.SoftMergeWith(gv)
+						value = vv
+					}
+				} else {
+					vv.SoftMergeWith(g)
+					value = vv
+				}
+			} else {
+				value = Dict{"value": v}
+			}
+			break
+		}
+		if IsDict(v) {
+			current = v.(Dict)
+		} else {
+			break
+		}
+	}
+	return value
+}
+
+func (d Dict) GetDirectValue(path string) Dict {
+	var value Dict
+	current := make(Dict)
+	current.CopyFrom(d)
+	keys := strings.Split(path, ".")
+	lastIndex := len(keys) - 1
+	for i, key := range keys {
+		if !current.HasKey(key) {
+			break
+		}
+		v := current[key]
+		if i == lastIndex {
+			if vv, ok := v.(Dict); ok {
+				value = vv
+			} else {
+				value = Dict{"value": v}
+			}
+			break
+		}
+		if IsDict(v) {
+			current = v.(Dict)
+		} else {
+			break
 		}
 	}
 	return value
diff --git a/dict/dict_test.go b/dict/dict_test.go
index c83c559..c41ab60 100644
--- a/dict/dict_test.go
+++ b/dict/dict_test.go
@@ -54,13 +54,13 @@ func TestIsDict(t *testing.T) {
 	}
 }
 
-func TestGeValue(t *testing.T) {
+func TestGetInheritedValue(t *testing.T) {
 	testCases := []dictStrToDictTestCase{}
 	loadYamlFile("test_data/getvalue_cases.yaml", &testCases)
 	for _, testCase := range testCases {
 		copy := make(Dict)
 		copy.CopyFrom(testCase.Arg1)
-		result := testCase.Arg1.GetValue(testCase.Arg2)
+		result := testCase.Arg1.GetInheritedValue(testCase.Arg2)
 		if !reflect.DeepEqual(testCase.Result, result) {
 			t.Errorf("Expected dict to be %v, got %v", testCase.Result, result)
 		}
diff --git a/main.go b/main.go
index d750e03..18f163a 100644
--- a/main.go
+++ b/main.go
@@ -14,6 +14,6 @@ func load(fn string) dict.Dict {
 
 func main() {
 	d := load("dict/test_data/dict4.yaml")
-	v := d.GetValue("nginx.sites.www")
+	v := d.GetInheritedValue("nginx.sites.www.rate_limit")
 	v.Print()
 }