Fix double completion bug (#124)

* Fix double completion bug
woops
Closes #122

* One more layer in test

Author: julienduchesne

pkg/ast/processing/find_field.go

@@ -87,8 +87,8 @@ func extractObjectRangesFromDesugaredObjs(vm *jsonnet.VM, desugaredObjs []*ast.D
 	for len(indexList) > 0 {
 		index := indexList[0]
 		indexList = indexList[1:]
-		partialMatchFields := partialMatchFields && len(indexList) == 0 // Only partial match on the last index. Others are considered complete
-		foundFields := findObjectFieldsInObjects(desugaredObjs, index, partialMatchFields)
+		partialMatchCurrentField := partialMatchFields && len(indexList) == 0 // Only partial match on the last index. Others are considered complete
+		foundFields := findObjectFieldsInObjects(desugaredObjs, index, partialMatchCurrentField)
 		desugaredObjs = nil
 		if len(foundFields) == 0 {
 			return nil, fmt.Errorf("field %s was not found in ast.DesugaredObject", index)
@@ -98,8 +98,8 @@ func extractObjectRangesFromDesugaredObjs(vm *jsonnet.VM, desugaredObjs []*ast.D
 				ranges = append(ranges, FieldToRange(*found))
 
 				// If the field is not PlusSuper (field+: value), we stop there. Other previous values are not relevant
-				// If partialMatchFields is true, we can continue to look for other fields
-				if !found.PlusSuper && !partialMatchFields {
+				// If partialMatchCurrentField is true, we can continue to look for other fields
+				if !found.PlusSuper && !partialMatchCurrentField {
 					break
 				}
 			}

pkg/server/completion_test.go

@@ -566,6 +566,26 @@ func TestCompletion(t *testing.T) {
 				},
 			},
 		},
+		{
+			name:            "autocomplete fix doubled index bug",
+			filename:        "testdata/doubled-index-bug-4.jsonnet",
+			replaceString:   "a: g.hello",
+			replaceByString: "a: g.hello.",
+			expected: protocol.CompletionList{
+				IsIncomplete: false,
+				Items: []protocol.CompletionItem{
+					{
+						Label:      "to",
+						Kind:       protocol.FieldCompletion,
+						Detail:     "g.hello.to",
+						InsertText: "to",
+						LabelDetails: protocol.CompletionItemLabelDetails{
+							Description: "object",
+						},
+					},
+				},
+			},
+		},
 	}
 	for _, tc := range testCases {
 		t.Run(tc.name, func(t *testing.T) {

pkg/server/testdata/doubled-index-bug-1.jsonnet

@@ -0,0 +1,7 @@
+{
+  hello: {
+    to: {
+      the: 'world',
+    },
+  },
+}

pkg/server/testdata/doubled-index-bug-2.jsonnet

@@ -0,0 +1 @@
+{ hello: (import 'doubled-index-bug-1.jsonnet').hello }

pkg/server/testdata/doubled-index-bug-3.jsonnet

@@ -0,0 +1 @@
+import 'doubled-index-bug-2.jsonnet'

pkg/server/testdata/doubled-index-bug-4.jsonnet

@@ -0,0 +1,5 @@
+local g = import 'doubled-index-bug-3.jsonnet';
+{
+  // completing fields of `g.hello` should get use `g.hello.to`, not `g.hello.hello`
+  a: g.hello,
+}