Add IsSensitive to AttributeSchema

So that sensitive attributes can be flagged.
hashicorp · Jun 1, 2021 · 34f98c4 · 34f98c4
1 parent b62f3a9
commit 34f98c4
Show file tree

Hide file tree

Showing 3 changed files with 16 additions and 0 deletions.
diff --git a/decoder/attribute_candidates.go b/decoder/attribute_candidates.go
@@ -35,6 +35,10 @@ func detailForAttribute(attr *schema.AttributeSchema) string {
 		detail = "Optional"
 	}
 
+	if attr.IsSensitive {
+		detail += fmt.Sprintf(", sensitive")
+	}
+
 	friendlyName := attr.Expr.FriendlyName()
 	if friendlyName != "" {
 		if detail != "" {

diff --git a/schema/attribute_schema.go b/schema/attribute_schema.go
@@ -14,6 +14,7 @@ type AttributeSchema struct {
 	IsOptional   bool
 	IsDeprecated bool
 	IsComputed   bool
+	IsSensitive  bool
 
 	Expr ExprConstraints
 
@@ -85,6 +86,7 @@ func (as *AttributeSchema) Copy() *AttributeSchema {
 		IsOptional:   as.IsOptional,
 		IsDeprecated: as.IsDeprecated,
 		IsComputed:   as.IsComputed,
+		IsSensitive:  as.IsSensitive,
 		IsDepKey:     as.IsDepKey,
 		Description:  as.Description,
 		Expr:         as.Expr.Copy(),

diff --git a/schema/attribute_schema_test.go b/schema/attribute_schema_test.go
@@ -160,6 +160,16 @@ func TestAttributeSchema_Validate(t *testing.T) {
 			},
 			nil,
 		},
+		{
+			&AttributeSchema{
+				Expr: ExprConstraints{
+					TraversalExpr{OfType: cty.Number, OfScopeId: lang.ScopeId("blah")},
+				},
+				IsRequired:  true,
+				IsSensitive: true,
+			},
+			nil,
+		},
 	}
 
 	for i, tc := range testCases {