marks.Has and marks.Contains

This commit is contained in:
James Bardin 2021-06-25 13:35:17 -04:00
parent d9dfd451ea
commit cdf7469efd
1 changed files with 21 additions and 0 deletions

View File

@ -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")