From cdf7469efd3edfba3cd2c096eff758bf94041e19 Mon Sep 17 00:00:00 2001
From: James Bardin <j.bardin@gmail.com>
Date: Fri, 25 Jun 2021 13:35:17 -0400
Subject: [PATCH] marks.Has and marks.Contains

---
 internal/lang/marks/marks.go | 21 +++++++++++++++++++++
 1 file changed, 21 insertions(+)

diff --git a/internal/lang/marks/marks.go b/internal/lang/marks/marks.go
index bdda72ee5..93a731e71 100644
--- a/internal/lang/marks/marks.go
+++ b/internal/lang/marks/marks.go
@@ -2,6 +2,8 @@ package marks
 
 import (
 	"strings"
+
+	"github.com/zclconf/go-cty/cty"
 )
 
 // valueMarks allow creating strictly typed values for use as cty.Value marks.
@@ -13,6 +15,25 @@ func (m valueMark) GoString() string {
 	return "marks." + strings.Title(string(m))
 }
 
+// Has returns true if and only if the cty.Value has the given mark.
+func Has(val cty.Value, mark valueMark) bool {
+	return val.HasMark(mark)
+}
+
+// Contains returns true if the cty.Value or any any value within it contains
+// the given mark.
+func Contains(val cty.Value, mark valueMark) bool {
+	ret := false
+	cty.Walk(val, func(_ cty.Path, v cty.Value) (bool, error) {
+		if v.HasMark(mark) {
+			ret = true
+			return false, nil
+		}
+		return true, nil
+	})
+	return ret
+}
+
 // Sensitive indicates that this value is marked as sensitive in the context of
 // Terraform.
 var Sensitive = valueMark("sensitive")