Support folder completions for the contexts attribute

CHANGELOG.md

@@ -10,6 +10,7 @@ All notable changes to the Docker Language Server will be documented in this fil
   - textDocument/completion
     - provide local file and folder name suggestions ([#414](https://github.com/docker/docker-language-server/issues/414))
       - `context` attribute in a `target` block
+      - `contexts` attribute in a `target` block
       - `dockerfile` attribute in a `target` block
 
 ### Fixed

internal/bake/hcl/completion.go

@@ -70,11 +70,22 @@ func Completion(ctx context.Context, params *protocol.CompletionParams, manager
 			}
 
 			if attribute, ok := attributes["dockerfile"]; ok && isInsideRange(attribute.Expr.Range(), params.Position) {
-				return fileStructureCompletionItems(bakeDocument, attribute, params.Position.Character, false)
+				return fileStructureCompletionItems(bakeDocument, attribute.Expr, params.Position.Character, false)
 			}
 
 			if attribute, ok := attributes["context"]; ok && isInsideRange(attribute.Expr.Range(), params.Position) {
-				return fileStructureCompletionItems(bakeDocument, attribute, params.Position.Character, true)
+				return fileStructureCompletionItems(bakeDocument, attribute.Expr, params.Position.Character, true)
+			}
+
+			if attribute, ok := attributes["contexts"]; ok && isInsideRange(attribute.Expr.Range(), params.Position) {
+				if expr, ok := attribute.Expr.(*hclsyntax.ObjectConsExpr); ok {
+					for _, item := range expr.Items {
+						if isInsideRange(item.ValueExpr.Range(), params.Position) {
+							return fileStructureCompletionItems(bakeDocument, item.ValueExpr, params.Position.Character, true)
+						}
+					}
+				}
+				return nil, nil
 			}
 
 			dockerfileURI, dockerfilePath, err := bakeDocument.DockerfileForTarget(b)
@@ -166,9 +177,9 @@ func Completion(ctx context.Context, params *protocol.CompletionParams, manager
 	return list, nil
 }
 
-func fileStructureCompletionItems(bakeDocument document.BakeHCLDocument, attribute *hclsyntax.Attribute, character protocol.UInteger, hideFiles bool) (*protocol.CompletionList, error) {
-	if expr, ok := attribute.Expr.(*hclsyntax.TemplateExpr); ok && len(expr.Parts) == 1 {
-		if literal, ok := expr.Parts[0].(*hclsyntax.LiteralValueExpr); ok {
+func fileStructureCompletionItems(bakeDocument document.BakeHCLDocument, expression hclsyntax.Expression, character protocol.UInteger, hideFiles bool) (*protocol.CompletionList, error) {
+	if templateExpr, ok := expression.(*hclsyntax.TemplateExpr); ok && len(templateExpr.Parts) == 1 {
+		if literal, ok := templateExpr.Parts[0].(*hclsyntax.LiteralValueExpr); ok {
 			path, err := bakeDocument.DocumentPath()
 			if err == nil {
 				value, _ := literal.Value(&hcl.EvalContext{})

internal/bake/hcl/completion_test.go

@@ -443,6 +443,31 @@ target t1 {
 			line:      2,
 			character: 12,
 		},
+		{
+			name: "contexts attribute in a target block",
+			content: `
+target t1 {
+	contexts = {
+		src = "%v"
+	}
+}`,
+			hideFiles: true,
+			line:      3,
+			character: 9,
+		},
+		{
+			name: "contexts attribute in a target block with multiple contexts",
+			content: `
+target t1 {
+	contexts = {
+		src = "../path/to/source"
+		src2 = "%v"
+	}
+}`,
+			hideFiles: true,
+			line:      4,
+			character: 10,
+		},
 	}
 
 	setups := []struct {