Modify imports that are included during code generation

[bendbennett]

Requirement for context import

With a spec that looks as follows:

{
  "datasources": [
    {
      "name": "example",
      "schema": {
        "attributes": [
          {
            "name": "bool_attribute",
            "bool": {
              "computed_optional_required": "computed"
            }
          }
        ]
      }
    }
  ],
  "provider": {
    "name": "provider"
  }
}

The generated code is:

// Code generated by terraform-plugin-framework-generator DO NOT EDIT.

package datasource_example

import (
	"context"
	"github.com/hashicorp/terraform-plugin-framework/types"

	"github.com/hashicorp/terraform-plugin-framework/datasource/schema"
)

func ExampleDataSourceSchema(ctx context.Context) schema.Schema {
	return schema.Schema{
		Attributes: map[string]schema.Attribute{
			"bool_attribute": schema.BoolAttribute{
				Computed: true,
			},
		},
	}
}

type ExampleModel struct {
	BoolAttribute types.Bool `tfsdk:"bool_attribute"`
}

The context import is required.

Requirement for github.com/hashicorp/terraform-plugin-framework/types/basetypes import

With a spec that looks as follows:

{
  "datasources": [
    {
      "name": "example",
      "schema": {
        "attributes": [
          {
            "name": "list_nested_attribute",
            "list_nested": {
              "computed_optional_required": "optional",
              "nested_object": {
                "attributes": [
                  {
                    "name": "bool_attribute",
                    "bool": {
                      "computed_optional_required": "optional"
                    }
                  }
                ]
              }
            }
          }
        ]
      }
    }
  ],
  "provider": {
    "name": "provider"
  }
}

The generated code includes:

/* ... */

var _ basetypes.ObjectTypable = ListNestedAttributeType{}

/* ... */

var _ basetypes.ObjectValuable = ListNestedAttributeValue{}

The github.com/hashicorp/terraform-plugin-framework/types/basetypes import is required.

[bflad]

🚀

[github-actions]

I'm going to lock this pull request because it has been closed for 30 days ⏳. This helps our maintainers find and focus on the active contributions.
If you have found a problem that seems related to this change, please open a new issue and complete the issue template so we can capture all the details necessary to investigate further.