cty: add CanListFunc, CanSetFunc, CanMapFunc

cty/convert/conversion_collection.go

@@ -1,6 +1,8 @@
 package convert
 
 import (
+	"fmt"
+
 	"github.com/zclconf/go-cty/cty"
 )
 
@@ -51,6 +53,10 @@ func conversionCollectionToList(ety cty.Type, conv conversion) conversion {
 			return cty.ListValEmpty(ety), nil
 		}
 
+		if !cty.CanListVal(elems) {
+			return cty.NilVal, fmt.Errorf("element types must all match for conversion to list")
+		}
+
 		return cty.ListVal(elems), nil
 	}
 }
@@ -96,6 +102,10 @@ func conversionCollectionToSet(ety cty.Type, conv conversion) conversion {
 			return cty.SetValEmpty(ety), nil
 		}
 
+		if !cty.CanSetVal(elems) {
+			return cty.NilVal, fmt.Errorf("element types must all match for conversion to set")
+		}
+
 		return cty.SetVal(elems), nil
 	}
 }
@@ -152,8 +162,8 @@ func conversionCollectionToMap(ety cty.Type, conv conversion) conversion {
 			}
 		}
 
-		if err := conversionCheckMapElementTypes(elems, path); err != nil {
-			return cty.NilVal, err
+		if !cty.CanMapVal(elems) {
+			return cty.NilVal, fmt.Errorf("element types must all match for conversion to map")
 		}
 
 		return cty.MapVal(elems), nil
@@ -237,6 +247,10 @@ func conversionTupleToSet(tupleType cty.Type, setEty cty.Type, unsafe bool) conv
 			i++
 		}
 
+		if !cty.CanSetVal(elems) {
+			return cty.NilVal, fmt.Errorf("element types must all match for conversion to set")
+		}
+
 		return cty.SetVal(elems), nil
 	}
 }
@@ -324,6 +338,11 @@ func conversionTupleToList(tupleType cty.Type, listEty cty.Type, unsafe bool) co
 		if err != nil {
 			return cty.NilVal, err
 		}
+
+		if !cty.CanListVal(elems) {
+			return cty.NilVal, fmt.Errorf("element types must all match for conversion to list")
+		}
+
 		return cty.ListVal(elems), nil
 	}
 }
@@ -402,8 +421,8 @@ func conversionObjectToMap(objectType cty.Type, mapEty cty.Type, unsafe bool) co
 			}
 		}
 
-		if err := conversionCheckMapElementTypes(elems, path); err != nil {
-			return cty.NilVal, err
+		if !cty.CanMapVal(elems) {
+			return cty.NilVal, fmt.Errorf("attribute types must all match for conversion to map")
 		}
 
 		return cty.MapVal(elems), nil
@@ -487,7 +506,7 @@ func conversionUnifyCollectionElements(elems map[string]cty.Value, path cty.Path
 	}
 	unifiedType, _ := unify(elemTypes, unsafe)
 	if unifiedType == cty.NilType {
-		return nil, path.NewErrorf("collection elements cannot be unified")
+		return nil, path.NewErrorf("cannot find a common base type for all elements")
 	}
 
 	unifiedElems := make(map[string]cty.Value)
@@ -514,34 +533,14 @@ func conversionUnifyCollectionElements(elems map[string]cty.Value, path cty.Path
 	return unifiedElems, nil
 }
 
-func conversionCheckMapElementTypes(elems map[string]cty.Value, path cty.Path) error {
-	elementType := cty.NilType
-	elemPath := append(path.Copy(), nil)
-
-	for name, elem := range elems {
-		if elementType == cty.NilType {
-			elementType = elem.Type()
-			continue
-		}
-		if !elementType.Equals(elem.Type()) {
-			elemPath[len(elemPath)-1] = cty.IndexStep{
-				Key: cty.StringVal(name),
-			}
-			return elemPath.NewErrorf("%s is required", elementType.FriendlyName())
-		}
-	}
-
-	return nil
-}
-
 func conversionUnifyListElements(elems []cty.Value, path cty.Path, unsafe bool) ([]cty.Value, error) {
 	elemTypes := make([]cty.Type, len(elems))
 	for i, elem := range elems {
 		elemTypes[i] = elem.Type()
 	}
 	unifiedType, _ := unify(elemTypes, unsafe)
 	if unifiedType == cty.NilType {
-		return nil, path.NewErrorf("collection elements cannot be unified")
+		return nil, path.NewErrorf("cannot find a common base type for all elements")
 	}
 
 	ret := make([]cty.Value, len(elems))

cty/value_init.go

@@ -186,6 +186,20 @@ func ListValEmpty(element Type) Value {
 	}
 }
 
+// CanListVal returns false if the given Values can not be coalesced
+// into a single List due to inconsistent element types.
+func CanListVal(vals []Value) bool {
+	elementType := DynamicPseudoType
+	for _, val := range vals {
+		if elementType == DynamicPseudoType {
+			elementType = val.ty
+		} else if val.ty != DynamicPseudoType && !elementType.Equals(val.ty) {
+			return false
+		}
+	}
+	return true
+}
+
 // MapVal returns a Value of a map type whose element type is defined by
 // the types of the given values, which must be homogenous.
 //
@@ -227,6 +241,20 @@ func MapValEmpty(element Type) Value {
 	}
 }
 
+// CanMapVal returns false if the given Values can not be coalesced into a
+// single Map due to inconsistent element types.
+func CanMapVal(vals map[string]Value) bool {
+	elementType := DynamicPseudoType
+	for _, val := range vals {
+		if elementType == DynamicPseudoType {
+			elementType = val.ty
+		} else if val.ty != DynamicPseudoType && !elementType.Equals(val.ty) {
+			return false
+		}
+	}
+	return true
+}
+
 // SetVal returns a Value of set type whose element type is defined by
 // the types of the given values, which must be homogenous.
 //
@@ -267,6 +295,26 @@ func SetVal(vals []Value) Value {
 	}.WithMarks(markSets...)
 }
 
+// CanSetVal returns false if the given Values can not be coalesced
+// into a single Set due to inconsistent element types.
+func CanSetVal(vals []Value) bool {
+	elementType := DynamicPseudoType
+	var markSets []ValueMarks
+
+	for _, val := range vals {
+		if unmarkedVal, marks := val.UnmarkDeep(); len(marks) > 0 {
+			val = unmarkedVal
+			markSets = append(markSets, marks)
+		}
+		if elementType == DynamicPseudoType {
+			elementType = val.ty
+		} else if val.ty != DynamicPseudoType && !elementType.Equals(val.ty) {
+			return false
+		}
+	}
+	return true
+}
+
 // SetValFromValueSet returns a Value of set type based on an already-constructed
 // ValueSet.
 //