Add type declarations for variable type

So that completion, hover and semantic-tokens work for variable `type`. Related to hashicorp/terraform-ls#490
aztfmod · Jun 7, 2021 · 09c4e3d · 09c4e3d
1 parent 283182c
commit 09c4e3d
Show file tree

Hide file tree

Showing 5 changed files with 213 additions and 0 deletions.
diff --git a/decoder/expression_candidates.go b/decoder/expression_candidates.go
@@ -287,6 +287,28 @@ func (d *Decoder) constraintToCandidates(constraint schema.ExprConstraint, outer
 				},
 			})
 		}
+	case schema.TypeDeclarationExpr:
+		for _, t := range []string{
+			"bool",
+			"number",
+			"string",
+			"list()",
+			"set()",
+			"tuple()",
+			"map()",
+			"object()",
+		} {
+			candidates = append(candidates, lang.Candidate{
+				Label:  t,
+				Detail: t,
+				Kind:   lang.AttributeCandidateKind,
+				TextEdit: lang.TextEdit{
+					NewText: t,
+					Snippet: t,
+					Range:   editRng,
+				},
+			})
+		}
 	}
 
 	return candidates

diff --git a/decoder/expression_candidates_test.go b/decoder/expression_candidates_test.go
@@ -1281,6 +1281,166 @@ func TestDecoder_CandidateAtPos_expressions(t *testing.T) {
 			hcl.Pos{Line: 1, Column: 8, Byte: 7},
 			lang.ZeroCandidates(),
 		},
+		{
+			"type declaration",
+			map[string]*schema.AttributeSchema{
+				"attr": {
+					Expr: schema.ExprConstraints{
+						schema.TypeDeclarationExpr{},
+					},
+				},
+			},
+			`attr = 
+`,
+			hcl.Pos{Line: 1, Column: 8, Byte: 7},
+			lang.CompleteCandidates([]lang.Candidate{
+				{
+					Label:  "bool",
+					Detail: "bool",
+					TextEdit: lang.TextEdit{
+						Range: hcl.Range{
+							Filename: "test.tf",
+							Start: hcl.Pos{
+								Line:   1,
+								Column: 8,
+								Byte:   7,
+							},
+							End: hcl.Pos{
+								Line:   1,
+								Column: 8,
+								Byte:   7,
+							},
+						}, NewText: "bool", Snippet: "bool"},
+					Kind: lang.AttributeCandidateKind,
+				},
+				{
+					Label:  "number",
+					Detail: "number",
+					TextEdit: lang.TextEdit{Range: hcl.Range{
+						Filename: "test.tf",
+						Start: hcl.Pos{
+							Line:   1,
+							Column: 8,
+							Byte:   7,
+						},
+						End: hcl.Pos{
+							Line:   1,
+							Column: 8,
+							Byte:   7,
+						},
+					}, NewText: "number", Snippet: "number"},
+					Kind: lang.AttributeCandidateKind,
+				},
+				{
+					Label:  "string",
+					Detail: "string",
+					TextEdit: lang.TextEdit{Range: hcl.Range{
+						Filename: "test.tf",
+						Start: hcl.Pos{
+							Line:   1,
+							Column: 8,
+							Byte:   7,
+						},
+						End: hcl.Pos{
+							Line:   1,
+							Column: 8,
+							Byte:   7,
+						},
+					}, NewText: "string", Snippet: "string"},
+					Kind: lang.AttributeCandidateKind,
+				},
+				{
+					Label:  "list()",
+					Detail: "list()",
+					TextEdit: lang.TextEdit{Range: hcl.Range{
+						Filename: "test.tf",
+						Start: hcl.Pos{
+							Line:   1,
+							Column: 8,
+							Byte:   7,
+						},
+						End: hcl.Pos{
+							Line:   1,
+							Column: 8,
+							Byte:   7,
+						},
+					}, NewText: "list()", Snippet: "list()"},
+					Kind: lang.AttributeCandidateKind,
+				},
+				{
+					Label:  "set()",
+					Detail: "set()",
+					TextEdit: lang.TextEdit{Range: hcl.Range{
+						Filename: "test.tf",
+						Start: hcl.Pos{
+							Line:   1,
+							Column: 8,
+							Byte:   7,
+						},
+						End: hcl.Pos{
+							Line:   1,
+							Column: 8,
+							Byte:   7,
+						},
+					}, NewText: "set()", Snippet: "set()"},
+					Kind: lang.AttributeCandidateKind,
+				},
+				{
+					Label:  "tuple()",
+					Detail: "tuple()",
+					TextEdit: lang.TextEdit{Range: hcl.Range{
+						Filename: "test.tf",
+						Start: hcl.Pos{
+							Line:   1,
+							Column: 8,
+							Byte:   7,
+						},
+						End: hcl.Pos{
+							Line:   1,
+							Column: 8,
+							Byte:   7,
+						},
+					}, NewText: "tuple()", Snippet: "tuple()"},
+					Kind: lang.AttributeCandidateKind,
+				},
+				{
+					Label:  "map()",
+					Detail: "map()",
+					TextEdit: lang.TextEdit{Range: hcl.Range{
+						Filename: "test.tf",
+						Start: hcl.Pos{
+							Line:   1,
+							Column: 8,
+							Byte:   7,
+						},
+						End: hcl.Pos{
+							Line:   1,
+							Column: 8,
+							Byte:   7,
+						},
+					}, NewText: "map()", Snippet: "map()"},
+					Kind: lang.AttributeCandidateKind,
+				},
+				{
+					Label:  "object()",
+					Detail: "object()",
+					TextEdit: lang.TextEdit{Range: hcl.Range{
+						Filename: "test.tf",
+						Start: hcl.Pos{
+							Line:   1,
+							Column: 8,
+							Byte:   7,
+						},
+						End: hcl.Pos{
+							Line:   1,
+							Column: 8,
+							Byte:   7,
+						},
+					}, NewText: "object()", Snippet: "object()"},
+					Kind: lang.AttributeCandidateKind,
+				},
+			}),
+		},
 	}
 
 	for i, tc := range testCases {

diff --git a/decoder/expression_constraints.go b/decoder/expression_constraints.go
@@ -219,3 +219,12 @@ func (ec ExprConstraints) LiteralValueOfObjectConsExpr(expr *hclsyntax.ObjectCon
 
 	return schema.LiteralValue{}, false
 }
+
+func (ec ExprConstraints) TypeDeclarationExpr() (schema.TypeDeclarationExpr, bool) {
+	for _, c := range ec {
+		if td, ok := c.(schema.TypeDeclarationExpr); ok {
+			return td, ok
+		}
+	}
+	return schema.TypeDeclarationExpr{}, false
+}
diff --git a/decoder/hover.go b/decoder/hover.go
@@ -246,6 +246,14 @@ func (d *Decoder) hoverDataForExpr(expr hcl.Expression, constraints ExprConstrai
 				Range:   expr.Range(),
 			}, nil
 		}
+
+		_, ok = constraints.TypeDeclarationExpr()
+		if ok {
+			return &lang.HoverData{
+				Content: lang.Markdown("Type declaration"),
+				Range:   expr.Range(),
+			}, nil
+		}
 	case *hclsyntax.TemplateExpr:
 		if e.IsStringLiteral() {
 			data, err := d.hoverDataForExpr(e.Parts[0], constraints, nestingLvl, pos)

diff --git a/schema/expressions.go b/schema/expressions.go
@@ -384,3 +384,17 @@ func LiteralTypeOnly(t cty.Type) ExprConstraints {
 		LiteralTypeExpr{Type: t},
 	}
 }
+
+type TypeDeclarationExpr struct{}
+
+func (TypeDeclarationExpr) isExprConstraintImpl() exprConstrSigil {
+	return exprConstrSigil{}
+}
+
+func (td TypeDeclarationExpr) FriendlyName() string {
+	return "type"
+}
+
+func (td TypeDeclarationExpr) Copy() ExprConstraint {
+	return TypeDeclarationExpr{}
+}