Remove attribute and block schema templates

go.mod

@@ -5,8 +5,9 @@ go 1.21
 require (
 	github.com/google/go-cmp v0.6.0
 	github.com/hashicorp/cli v1.1.6
-	github.com/hashicorp/terraform-plugin-codegen-spec v0.1.1-0.20231024091233-c659ac8a54fc
+	github.com/hashicorp/terraform-plugin-codegen-spec v0.1.1-0.20240104173918-8a979efa31b3
 	github.com/hashicorp/terraform-plugin-framework v1.4.2
+	github.com/hashicorp/terraform-plugin-go v0.19.0
 	github.com/mattn/go-colorable v0.1.13
 )
 
@@ -21,7 +22,6 @@ require (
 	github.com/hashicorp/errwrap v1.1.0 // indirect
 	github.com/hashicorp/go-hclog v1.5.0 // indirect
 	github.com/hashicorp/go-multierror v1.1.1 // indirect
-	github.com/hashicorp/terraform-plugin-go v0.19.0 // indirect
 	github.com/hashicorp/terraform-plugin-log v0.9.0 // indirect
 	github.com/huandu/xstrings v1.4.0 // indirect
 	github.com/imdario/mergo v0.3.16 // indirect

go.sum

@@ -32,8 +32,8 @@ github.com/hashicorp/go-hclog v1.5.0/go.mod h1:W4Qnvbt70Wk/zYJryRzDRU/4r0kIg0PVH
 github.com/hashicorp/go-multierror v1.0.0/go.mod h1:dHtQlpGsu+cZNNAkkCN/P3hoUDHhCYQXV3UM06sGGrk=
 github.com/hashicorp/go-multierror v1.1.1 h1:H5DkEtf6CXdFp0N0Em5UCwQpXMWke8IA0+lD48awMYo=
 github.com/hashicorp/go-multierror v1.1.1/go.mod h1:iw975J/qwKPdAO1clOe2L8331t/9/fmwbPZ6JB6eMoM=
-github.com/hashicorp/terraform-plugin-codegen-spec v0.1.1-0.20231024091233-c659ac8a54fc h1:VmMk5vOSJgpWOuBsI4ZBZkcsrkLq0fKoyKvKFnbBuxk=
-github.com/hashicorp/terraform-plugin-codegen-spec v0.1.1-0.20231024091233-c659ac8a54fc/go.mod h1:PQn6bDD8UWoAVJoHXqFk2i/RmLbeQBjbiP38i+E+YIw=
+github.com/hashicorp/terraform-plugin-codegen-spec v0.1.1-0.20240104173918-8a979efa31b3 h1:2yq1HJPw4LpV/vIm++wcCYZfbNGQftwVWIv/FAAxHZ4=
+github.com/hashicorp/terraform-plugin-codegen-spec v0.1.1-0.20240104173918-8a979efa31b3/go.mod h1:PQn6bDD8UWoAVJoHXqFk2i/RmLbeQBjbiP38i+E+YIw=
 github.com/hashicorp/terraform-plugin-framework v1.4.2 h1:P7a7VP1GZbjc4rv921Xy5OckzhoiO3ig6SGxwelD2sI=
 github.com/hashicorp/terraform-plugin-framework v1.4.2/go.mod h1:GWl3InPFZi2wVQmdVnINPKys09s9mLmTZr95/ngLnbY=
 github.com/hashicorp/terraform-plugin-go v0.19.0 h1:BuZx/6Cp+lkmiG0cOBk6Zps0Cb2tmqQpDM3iAtnhDQU=

internal/convert/computed_optional_required.go

@@ -0,0 +1,62 @@
+// Copyright (c) HashiCorp, Inc.
+// SPDX-License-Identifier: MPL-2.0
+
+package convert
+
+import (
+	"bytes"
+
+	specschema "github.com/hashicorp/terraform-plugin-codegen-spec/schema"
+)
+
+type ComputedOptionalRequired struct {
+	computedOptionalRequired specschema.ComputedOptionalRequired
+}
+
+func NewComputedOptionalRequired(c specschema.ComputedOptionalRequired) ComputedOptionalRequired {
+	return ComputedOptionalRequired{
+		computedOptionalRequired: c,
+	}
+}
+
+func (c ComputedOptionalRequired) Equal(other ComputedOptionalRequired) bool {
+	return c.computedOptionalRequired.Equal(other.computedOptionalRequired)
+}
+
+func (c ComputedOptionalRequired) IsComputed() bool {
+	if c.computedOptionalRequired == specschema.Computed || c.computedOptionalRequired == specschema.ComputedOptional {
+		return true
+	}
+
+	return false
+}
+
+func (c ComputedOptionalRequired) IsOptional() bool {
+	if c.computedOptionalRequired == specschema.Optional || c.computedOptionalRequired == specschema.ComputedOptional {
+		return true
+	}
+
+	return false
+}
+
+func (c ComputedOptionalRequired) IsRequired() bool {
+	return c.computedOptionalRequired == specschema.Required
+}
+
+func (c ComputedOptionalRequired) Schema() []byte {
+	var b bytes.Buffer
+
+	if c.IsRequired() {
+		b.WriteString("Required: true,\n")
+	}
+
+	if c.IsOptional() {
+		b.WriteString("Optional: true,\n")
+	}
+
+	if c.IsComputed() {
+		b.WriteString("Computed: true,\n")
+	}
+
+	return b.Bytes()
+}

internal/convert/custom_type_collection.go

@@ -0,0 +1,88 @@
+// Copyright (c) HashiCorp, Inc.
+// SPDX-License-Identifier: MPL-2.0
+
+package convert
+
+import (
+	"fmt"
+
+	specschema "github.com/hashicorp/terraform-plugin-codegen-spec/schema"
+
+	"github.com/hashicorp/terraform-plugin-codegen-framework/internal/format"
+)
+
+const (
+	CustomCollectionTypeList CustomCollectionTypes = "List"
+	CustomCollectionTypeMap  CustomCollectionTypes = "Map"
+	CustomCollectionTypeSet  CustomCollectionTypes = "Set"
+)
+
+type CustomCollectionTypes string
+
+func (c CustomCollectionTypes) Equal(other CustomCollectionTypes) bool {
+	return c == other
+}
+
+type CustomTypeCollection struct {
+	associatedExternalType *specschema.AssociatedExternalType
+	customCollectionType   CustomCollectionTypes
+	customType             *specschema.CustomType
+	elementType            string
+	name                   string
+}
+
+// NewCustomTypeCollection constructs an CustomTypeCollection which is used to determine whether a CustomType
+// should be assigned to a collection attribute in the schema.
+//
+// If a CustomType has been declared in the spec, then the CustomType.Type will be used as
+// the CustomType in the schema.
+//
+// If the spec CustomType is nil, and the spec AssociatedExternalType is not nil, the generator
+// will create custom Type and Value types using the attribute name, and the generated custom
+// Type type will be used as the CustomType in the schema.
+func NewCustomTypeCollection(c *specschema.CustomType, a *specschema.AssociatedExternalType, cct CustomCollectionTypes, elemType, name string) CustomTypeCollection {
+	return CustomTypeCollection{
+		associatedExternalType: a,
+		customCollectionType:   cct,
+		customType:             c,
+		elementType:            elemType,
+		name:                   name,
+	}
+}
+
+func (c CustomTypeCollection) Equal(other CustomTypeCollection) bool {
+	if !c.associatedExternalType.Equal(other.associatedExternalType) {
+		return false
+	}
+
+	if !c.customCollectionType.Equal(other.customCollectionType) {
+		return false
+	}
+
+	if !c.customType.Equal(other.customType) {
+		return false
+	}
+
+	if c.elementType != other.elementType {
+		return false
+	}
+
+	return c.name == other.name
+}
+
+func (c CustomTypeCollection) Schema() []byte {
+	var customType string
+
+	switch {
+	case c.customType != nil:
+		customType = c.customType.Type
+	case c.associatedExternalType != nil:
+		customType = fmt.Sprintf("%sType{\ntypes.%sType{\nElemType: %s,\n},\n}", format.ToPascalCase(c.name), c.customCollectionType, c.elementType)
+	}
+
+	if customType != "" {
+		return []byte(fmt.Sprintf("CustomType: %s,\n", customType))
+	}
+
+	return nil
+}

internal/convert/custom_type_nested_collection.go

@@ -0,0 +1,40 @@
+// Copyright (c) HashiCorp, Inc.
+// SPDX-License-Identifier: MPL-2.0
+
+package convert
+
+import (
+	"fmt"
+
+	specschema "github.com/hashicorp/terraform-plugin-codegen-spec/schema"
+)
+
+type CustomTypeNestedCollection struct {
+	customType *specschema.CustomType
+}
+
+// NewCustomTypeNestedCollection constructs an CustomTypeNestedCollection which is used to determine whether a CustomType
+// should be assigned to a nested attribute in the schema.
+//
+// If a CustomType has been declared in the spec, then the CustomType.Type will be used as
+// the CustomType in the schema.
+//
+// If the spec CustomType is nil, the generator will create custom Type and Value types using the attribute
+// name, and the generated custom Type type will be used as the CustomType in the schema.
+func NewCustomTypeNestedCollection(c *specschema.CustomType) CustomTypeNestedCollection {
+	return CustomTypeNestedCollection{
+		customType: c,
+	}
+}
+
+func (a CustomTypeNestedCollection) Equal(other CustomTypeNestedCollection) bool {
+	return a.customType.Equal(other.customType)
+}
+
+func (a CustomTypeNestedCollection) Schema() []byte {
+	if a.customType != nil && a.customType.Type != "" {
+		return []byte(fmt.Sprintf("CustomType: %s,\n", a.customType.Type))
+	}
+
+	return nil
+}

internal/convert/custom_type_nested_object.go

@@ -0,0 +1,57 @@
+// Copyright (c) HashiCorp, Inc.
+// SPDX-License-Identifier: MPL-2.0
+
+package convert
+
+import (
+	"fmt"
+
+	specschema "github.com/hashicorp/terraform-plugin-codegen-spec/schema"
+
+	"github.com/hashicorp/terraform-plugin-codegen-framework/internal/format"
+)
+
+type CustomTypeNestedObject struct {
+	customType *specschema.CustomType
+	name       string
+}
+
+// NewCustomTypeNestedObject constructs an CustomTypeNestedObject which is used to determine whether a CustomType
+// should be assigned to a nested attribute object in the schema.
+//
+// If a CustomType has been declared in the spec, then the CustomType.Type will be used as
+// the CustomType in the schema.
+//
+// If the spec CustomType is nil, the generator will create custom Type and Value types using the attribute
+// name, and the generated custom Type type will be used as the CustomType in the schema.
+func NewCustomTypeNestedObject(c *specschema.CustomType, name string) CustomTypeNestedObject {
+	return CustomTypeNestedObject{
+		customType: c,
+		name:       name,
+	}
+}
+
+func (c CustomTypeNestedObject) Equal(other CustomTypeNestedObject) bool {
+	if !c.customType.Equal(other.customType) {
+		return false
+	}
+
+	return c.name == other.name
+}
+
+func (c CustomTypeNestedObject) Schema() []byte {
+	var customTypeType string
+
+	switch {
+	case c.customType != nil:
+		customTypeType = c.customType.Type
+	default:
+		customTypeType = fmt.Sprintf("%sType{\nObjectType: types.ObjectType{\nAttrTypes: %sValue{}.AttributeTypes(ctx),\n},\n}", format.ToPascalCase(c.name), format.ToPascalCase(c.name))
+	}
+
+	if customTypeType != "" {
+		return []byte(fmt.Sprintf("CustomType: %s,\n", customTypeType))
+	}
+
+	return nil
+}