Create OrganizationTeamDataSource

Author: ddebeau

code_generator/provider_code_spec.json

@@ -365,6 +365,51 @@
         ]
       }
     },
+    {
+      "name": "organization_team",
+      "schema": {
+        "attributes": [
+          {
+            "name": "description",
+            "string": {
+              "computed_optional_required": "computed",
+              "description": "Markdown description"
+            }
+          },
+          {
+            "name": "members",
+            "list": {
+              "computed_optional_required": "computed",
+              "description": "List of team members",
+              "element_type": {
+                "string": {}
+              }
+            }
+          },
+          {
+            "name": "name",
+            "string": {
+              "computed_optional_required": "required",
+              "description": "Team name"
+            }
+          },
+          {
+            "name": "orgname",
+            "string": {
+              "computed_optional_required": "required",
+              "description": "Organization name"
+            }
+          },
+          {
+            "name": "role",
+            "string": {
+              "computed_optional_required": "computed",
+              "description": "Team role"
+            }
+          }
+        ]
+      }
+    },
     {
       "name": "organization_team_permission",
       "schema": {

internal/datasource_organization_team/organization_team_data_source_gen.go

@@ -0,0 +1,137 @@
+package provider
+
+import (
+	"context"
+	"encoding/json"
+	"fmt"
+	"io"
+
+	"github.com/hashicorp/terraform-plugin-framework/datasource"
+	"github.com/hashicorp/terraform-plugin-framework/types"
+
+	"github.com/enthought/terraform-provider-quay/quay_api"
+	"terraform-provider-quay/internal/datasource_organization_team"
+)
+
+var (
+	_ datasource.DataSource              = (*organizationTeamDataSource)(nil)
+	_ datasource.DataSourceWithConfigure = (*organizationTeamDataSource)(nil)
+)
+
+func NewOrganizationTeamDataSource() datasource.DataSource {
+	return &organizationTeamDataSource{}
+}
+
+type organizationTeamDataSource struct {
+	client *quay_api.APIClient
+}
+
+func (d *organizationTeamDataSource) Metadata(_ context.Context, req datasource.MetadataRequest, resp *datasource.MetadataResponse) {
+	resp.TypeName = req.ProviderTypeName + "_organization_team"
+}
+
+func (d *organizationTeamDataSource) Schema(ctx context.Context, _ datasource.SchemaRequest, resp *datasource.SchemaResponse) {
+	resp.Schema = datasource_organization_team.OrganizationTeamDataSourceSchema(ctx)
+}
+
+func (d *organizationTeamDataSource) Read(ctx context.Context, req datasource.ReadRequest, resp *datasource.ReadResponse) {
+	var data datasource_organization_team.OrganizationTeamModel
+	var resTeamData teamModelJSON
+	var resOrgData organizationModelJSON
+
+	// Read Terraform configuration data into the model
+	resp.Diagnostics.Append(req.Config.Get(ctx, &data)...)
+
+	if resp.Diagnostics.HasError() {
+		return
+	}
+
+	// Create variables
+	teamName := data.Name.ValueString()
+	orgName := data.Orgname.ValueString()
+
+	// Get members
+	httpRes, err := d.client.TeamAPI.GetOrganizationTeamMembers(context.Background(), orgName, teamName).Execute()
+	if err != nil {
+		errDetail := handleQuayAPIError(err)
+		resp.Diagnostics.AddError("Error reading Quay team", "Could not read Quay team, unexpected error: "+errDetail)
+		return
+	}
+
+	body, err := io.ReadAll(httpRes.Body)
+	if err != nil {
+		resp.Diagnostics.AddError(
+			"Error reading Quay team",
+			"Could not read Quay team, unexpected error: "+err.Error())
+		return
+	}
+	err = json.Unmarshal(body, &resTeamData)
+	if err != nil {
+		resp.Diagnostics.AddError(
+			"Error reading Quay team",
+			"Could not read Quay team, unexpected error: "+err.Error())
+		return
+	}
+
+	// Set members
+	var memList []string
+	for _, member := range resTeamData.Members {
+		memList = append(memList, member.Name)
+	}
+	members, diags := types.ListValueFrom(ctx, types.StringType, memList)
+
+	resp.Diagnostics.Append(diags...)
+	if resp.Diagnostics.HasError() {
+		return
+	}
+
+	data.Members = members
+
+	// Get org data
+	httpRes, err = d.client.OrganizationAPI.GetOrganization(context.Background(), orgName).Execute()
+	if err != nil {
+		errDetail := handleQuayAPIError(err)
+		resp.Diagnostics.AddError("Error reading Quay team", "Could not read Quay team, unexpected error: "+errDetail)
+		return
+	}
+
+	body, err = io.ReadAll(httpRes.Body)
+	if err != nil {
+		resp.Diagnostics.AddError(
+			"Error reading Quay team",
+			"Could not read Quay team, unexpected error: "+err.Error())
+		return
+	}
+	err = json.Unmarshal(body, &resOrgData)
+	if err != nil {
+		resp.Diagnostics.AddError(
+			"Error reading Quay team",
+			"Could not read Quay team, unexpected error: "+err.Error())
+		return
+	}
+
+	// Set Role
+	data.Role = types.StringValue(resOrgData.Teams[teamName].Role)
+
+	// Set Description
+	data.Description = types.StringValue(resOrgData.Teams[teamName].Description)
+
+	// Save data into Terraform state
+	resp.Diagnostics.Append(resp.State.Set(ctx, &data)...)
+}
+
+func (d *organizationTeamDataSource) Configure(_ context.Context, req datasource.ConfigureRequest, resp *datasource.ConfigureResponse) {
+	if req.ProviderData == nil {
+		return
+	}
+
+	client, ok := req.ProviderData.(*quay_api.APIClient)
+	if !ok {
+		resp.Diagnostics.AddError(
+			"Unexpected Data Source Configure Type",
+			fmt.Sprintf("Expected *quay_api.APIClient, got: %T. Please report this issue to the provider developers.", req.ProviderData),
+		)
+	}
+
+	d.client = client
+}

internal/provider/organization_team_data_source.go

@@ -0,0 +1,60 @@
+package provider
+
+import (
+	"testing"
+
+	"github.com/hashicorp/terraform-plugin-testing/helper/resource"
+)
+
+func TestAccOrganizationTeamDataSource(t *testing.T) {
+	resource.ParallelTest(t, resource.TestCase{
+		ProtoV6ProviderFactories: testAccProtoV6ProviderFactories,
+		Steps: []resource.TestStep{
+			// Create and Read
+			{
+				Config: providerConfig + `
+resource "quay_organization" "org_team_data" {
+  name = "org_team_data"
+  email = "[email protected]"
+}
+
+resource "quay_repository" "test" {
+  name = "test"
+  namespace = quay_organization.org_team_data.name
+}
+
+resource "quay_organization_robot" "test" {
+  name = "test"
+  orgname = quay_organization.org_team_data.name
+}
+
+resource "quay_organization_team" "test" {
+  name = "test"
+  orgname = quay_organization.org_team_data.name
+  role = "member"
+  description = "test"
+  members = [
+    quay_organization_robot.test.fullname
+  ]
+}
+
+data "quay_organization_team" "test" {
+  name = "test"
+  orgname = quay_organization.org_team_data.name
+  
+  depends_on = [
+    quay_organization_team.test
+  ]
+}
+`,
+				Check: resource.ComposeAggregateTestCheckFunc(
+					resource.TestCheckResourceAttr("data.quay_organization_team.test", "name", "test"),
+					resource.TestCheckResourceAttr("data.quay_organization_team.test", "orgname", "org_team_data"),
+					resource.TestCheckResourceAttr("data.quay_organization_team.test", "role", "member"),
+					resource.TestCheckResourceAttr("data.quay_organization_team.test", "description", "test"),
+					resource.TestCheckResourceAttr("data.quay_organization_team.test", "members.0", "org_team_data+test"),
+				),
+			},
+		},
+	})
+}

internal/provider/organization_team_data_source_test.go

@@ -153,6 +153,7 @@ func (p *quayProvider) DataSources(_ context.Context) []func() datasource.DataSo
 		NewOrganizationDataSource,
 		NewRepositoryDataSource,
 		NewOrganizationTeamPermissionDataSource,
+		NewOrganizationTeamDataSource,
 	}
 }