Cluster data source

docs/data-sources/cluster.md

@@ -0,0 +1,82 @@
+---
+# generated by https://github.com/hashicorp/terraform-plugin-docs
+page_title: "astronomer_cluster Data Source - astronomer"
+subcategory: ""
+description: |-
+  Cluster data source
+---
+
+# astronomer_cluster (Data Source)
+
+Cluster data source
+
+## Example Usage
+
+```terraform
+data "astronomer_cluster" "example" {
+  id = "clozc036j01to01jrlgvueo8t"
+}
+```
+
+<!-- schema generated by tfplugindocs -->
+## Schema
+
+### Required
+
+- `id` (String) Cluster identifier
+
+### Read-Only
+
+- `cloud_provider` (String) Cluster cloud provider
+- `created_at` (String) Cluster creation timestamp
+- `db_instance_type` (String) Cluster database instance type
+- `is_limited` (Boolean) Whether the cluster is limited
+- `metadata` (Attributes) Cluster metadata (see [below for nested schema](#nestedatt--metadata))
+- `name` (String) Cluster name
+- `node_pools` (Attributes List) Cluster node pools (see [below for nested schema](#nestedatt--node_pools))
+- `pod_subnet_range` (String) Cluster pod subnet range
+- `provider_account` (String) Cluster provider account
+- `region` (String) Cluster region
+- `service_peering_range` (String) Cluster service peering range
+- `service_subnet_range` (String) Cluster service subnet range
+- `status` (String) Cluster status
+- `tags` (Attributes List) Cluster tags (see [below for nested schema](#nestedatt--tags))
+- `tenant_id` (String) Cluster tenant ID
+- `type` (String) Cluster type
+- `updated_at` (String) Cluster last updated timestamp
+- `vpc_subnet_range` (String) Cluster VPC subnet range
+- `workspace_ids` (List of String) Cluster workspace IDs
+
+<a id="nestedatt--metadata"></a>
+### Nested Schema for `metadata`
+
+Read-Only:
+
+- `external_ips` (List of String) Cluster external IPs
+- `oidc_issuer_url` (String) Cluster OIDC issuer URL
+
+
+<a id="nestedatt--node_pools"></a>
+### Nested Schema for `node_pools`
+
+Read-Only:
+
+- `cloud_provider` (String) Node pool cloud provider
+- `cluster_id` (String) Node pool cluster identifier
+- `created_at` (String) Node pool creation timestamp
+- `id` (String) Node pool identifier
+- `is_default` (Boolean) Whether the node pool is the default node pool of the cluster
+- `max_node_count` (Number) Node pool maximum node count
+- `name` (String) Node pool name
+- `node_instance_type` (String) Node pool node instance type
+- `supported_astro_machines` (List of String) Node pool supported Astro machines
+- `updated_at` (String) Node pool last updated timestamp
+
+
+<a id="nestedatt--tags"></a>
+### Nested Schema for `tags`
+
+Read-Only:
+
+- `key` (String) Cluster tag key
+- `value` (String) Cluster tag value

examples/data-sources/astronomer_cluster/data-source.tf

@@ -0,0 +1,3 @@
+data "astronomer_cluster" "example" {
+  id = "clozc036j01to01jrlgvueo8t"
+}

internal/provider/datasources/data_source_cluster.go

@@ -0,0 +1,117 @@
+package datasources
+
+import (
+	"context"
+	"fmt"
+
+	"github.com/astronomer/astronomer-terraform-provider/internal/clients"
+	"github.com/astronomer/astronomer-terraform-provider/internal/clients/platform"
+	"github.com/astronomer/astronomer-terraform-provider/internal/provider/models"
+	"github.com/astronomer/astronomer-terraform-provider/internal/provider/schemas"
+	"github.com/astronomer/astronomer-terraform-provider/internal/utils"
+	"github.com/hashicorp/terraform-plugin-framework/datasource"
+	"github.com/hashicorp/terraform-plugin-framework/datasource/schema"
+	"github.com/hashicorp/terraform-plugin-log/tflog"
+)
+
+// Ensure provider defined types fully satisfy framework interfaces.
+var _ datasource.DataSource = &clusterDataSource{}
+var _ datasource.DataSourceWithConfigure = &clusterDataSource{}
+
+func NewClusterDataSource() datasource.DataSource {
+	return &clusterDataSource{}
+}
+
+// clusterDataSource defines the data source implementation.
+type clusterDataSource struct {
+	PlatformClient platform.ClientWithResponsesInterface
+	OrganizationId string
+}
+
+func (d *clusterDataSource) Metadata(
+	ctx context.Context,
+	req datasource.MetadataRequest,
+	resp *datasource.MetadataResponse,
+) {
+	resp.TypeName = req.ProviderTypeName + "_cluster"
+}
+
+func (d *clusterDataSource) Schema(
+	ctx context.Context,
+	req datasource.SchemaRequest,
+	resp *datasource.SchemaResponse,
+) {
+	resp.Schema = schema.Schema{
+		// This description is used by the documentation generator and the language server.
+		MarkdownDescription: "Cluster data source",
+		Attributes:          schemas.ClusterDataSourceSchemaAttributes(),
+	}
+}
+
+func (d *clusterDataSource) Configure(
+	ctx context.Context,
+	req datasource.ConfigureRequest,
+	resp *datasource.ConfigureResponse,
+) {
+	// Prevent panic if the provider has not been configured.
+	if req.ProviderData == nil {
+		return
+	}
+
+	apiClients, ok := req.ProviderData.(models.ApiClientsModel)
+	if !ok {
+		utils.DataSourceApiClientConfigureError(ctx, req, resp)
+		return
+	}
+
+	d.PlatformClient = apiClients.PlatformClient
+	d.OrganizationId = apiClients.OrganizationId
+}
+
+func (d *clusterDataSource) Read(
+	ctx context.Context,
+	req datasource.ReadRequest,
+	resp *datasource.ReadResponse,
+) {
+	var data models.Cluster
+
+	// Read Terraform configuration data into the model
+	resp.Diagnostics.Append(req.Config.Get(ctx, &data)...)
+	if resp.Diagnostics.HasError() {
+		return
+	}
+
+	cluster, err := d.PlatformClient.GetClusterWithResponse(
+		ctx,
+		d.OrganizationId,
+		data.Id.ValueString(),
+	)
+	if err != nil {
+		tflog.Error(ctx, "failed to get cluster", map[string]interface{}{"error": err})
+		resp.Diagnostics.AddError(
+			"Client Error",
+			fmt.Sprintf("Unable to read cluster, got error: %s", err),
+		)
+		return
+	}
+	_, diagnostic := clients.NormalizeAPIError(ctx, cluster.HTTPResponse, cluster.Body)
+	if diagnostic != nil {
+		resp.Diagnostics.Append(diagnostic)
+		return
+	}
+	if cluster.JSON200 == nil {
+		tflog.Error(ctx, "failed to get cluster", map[string]interface{}{"error": "nil response"})
+		resp.Diagnostics.AddError("Client Error", "Unable to read cluster, got nil response")
+		return
+	}
+
+	// Populate the model with the response data
+	diags := data.ReadFromResponse(ctx, cluster.JSON200)
+	if diags.HasError() {
+		resp.Diagnostics.Append(diags...)
+		return
+	}
+
+	// Save data into Terraform state
+	resp.Diagnostics.Append(resp.State.Set(ctx, &data)...)
+}

internal/provider/datasources/data_source_cluster_test.go

@@ -0,0 +1,49 @@
+package datasources_test
+
+import (
+	"fmt"
+	"os"
+	"testing"
+
+	astronomerprovider "github.com/astronomer/astronomer-terraform-provider/internal/provider"
+	"github.com/hashicorp/terraform-plugin-testing/helper/resource"
+)
+
+func TestAcc_DataSourceCluster(t *testing.T) {
+	hybridClusterId := os.Getenv("HYBRID_CLUSTER_ID")
+	resourceName := "test_data_cluster_hybrid"
+	resourceVar := fmt.Sprintf("data.astronomer_cluster.%v", resourceName)
+	resource.Test(t, resource.TestCase{
+		PreCheck: func() {
+			astronomerprovider.TestAccPreCheck(t)
+		},
+		ProtoV6ProviderFactories: astronomerprovider.TestAccProtoV6ProviderFactories,
+		Steps: []resource.TestStep{
+			{
+				Config: astronomerprovider.ProviderConfig(t, false) + cluster(resourceName, hybridClusterId),
+				Check: resource.ComposeTestCheckFunc(
+					resource.TestCheckResourceAttrSet(resourceVar, "id"),
+					resource.TestCheckResourceAttrSet(resourceVar, "name"),
+					resource.TestCheckResourceAttrSet(resourceVar, "cloud_provider"),
+					resource.TestCheckResourceAttrSet(resourceVar, "db_instance_type"),
+					resource.TestCheckResourceAttrSet(resourceVar, "region"),
+					resource.TestCheckResourceAttrSet(resourceVar, "vpc_subnet_range"),
+					resource.TestCheckResourceAttrSet(resourceVar, "created_at"),
+					resource.TestCheckResourceAttrSet(resourceVar, "updated_at"),
+					resource.TestCheckResourceAttr(resourceVar, "type", "HYBRID"),
+					resource.TestCheckResourceAttrSet(resourceVar, "provider_account"),
+					resource.TestCheckResourceAttrSet(resourceVar, "node_pools.0.id"),
+					resource.TestCheckResourceAttrSet(resourceVar, "node_pools.0.name"),
+					resource.TestCheckResourceAttrSet(resourceVar, "metadata.external_ips.0"),
+				),
+			},
+		},
+	})
+}
+
+func cluster(resourceName, clusterId string) string {
+	return fmt.Sprintf(`
+data astronomer_cluster "%v" {
+	  id = "%v"
+}`, resourceName, clusterId)
+}