fix: nested field selection path bug, add test

danielgtaylor · Mar 16, 2023 · dc54b9a · dc54b9a
1 parent b15ce8a
commit dc54b9a
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/get.go b/get.go
@@ -503,7 +503,7 @@ func (d *Document) getFields(input any) (any, Error) {
 			d.parseUntilNoReset(1, '|')
 			continue
 		}
-		if r == ':' {
+		if r == ':' && open <= 1 {
 			key = d.buf.String()
 			d.buf.Reset()
 			d.skipWhitespace()

diff --git a/get_test.go b/get_test.go
@@ -256,6 +256,12 @@ var getExamples = []struct {
 		Query: `{body: body.{id, created}, one: headers.one}`,
 		Go:    map[string]any{"body": []any{map[string]any{"id": "a", "created": "2022"}}, "one": 1.0},
 	},
+	{
+		Name:  "Field expression nested multi path",
+		Input: `{"body": [{"id": "a", "created": "2022", "link": "..."}], "headers": {"one": 1, "two": 2}}`,
+		Query: `{body: body.{id, created: created}, one: headers.one}`,
+		Go:    map[string]any{"body": []any{map[string]any{"id": "a", "created": "2022"}}, "one": 1.0},
+	},
 	{
 		Name:  "Field expression with pipe",
 		Input: `{"foo": "bar", "link": {"id": 1, "verified": true, "tags": ["a", "b"]}}`,