feat(aad): add a data source to get unblock quota statistics (#6876)

Author: shi2duxinxia

docs/data-sources/aad_unblock_quota_statistics.md

@@ -0,0 +1,45 @@
+---
+subcategory: "Advanced Anti-DDoS"
+layout: "huaweicloud"
+page_title: "HuaweiCloud: huaweicloud_aad_unblock_quota_statistics"
+description: |-
+  Use this data source to get information about the AAD unblock quota statistics within HuaweiCloud.
+---
+
+# huaweicloud_aad_unblock_quota_statistics
+
+Use this data source to get information about the AAD unblock quota statistics within HuaweiCloud.
+
+## Example Usage
+
+```hcl
+variable "domain_id" {}
+
+data "huaweicloud_aad_unblock_quota_statistics" "test" {
+  domain_id = var.domain_id
+}
+```
+
+## Argument Reference
+
+The following arguments are supported:
+
+* `domain_id` - (Required, String) Specified the account ID of IAM user.
+
+## Attribute Reference
+
+The following attributes are exported:
+
+* `id` - The data source ID.
+
+* `type` - The user type. The valid values are as follows:
+  + **common_user**: Indicates common user.
+  + **native_protection_user**: Indicates native basic protection user.
+
+* `total_unblocking_quota` - The total unblocking quota.
+
+* `remaining_unblocking_quota` - The remaining unblocking quota.
+
+* `unblocking_quota_today` - The unblocking quota of today.
+
+* `remaining_unblocking_quota_today` - The remaining unblocking quota of today.

huaweicloud/provider.go

@@ -452,9 +452,10 @@ func Provider() *schema.Provider {
 			"huaweicloud_access_analyzers":              accessanalyzer.DataSourceAccessAnalyzers(),
 			"huaweicloud_access_analyzer_archive_rules": accessanalyzer.DataSourceAccessAnalyzerArchiveRules(),
 
-			"huaweicloud_aad_instances":               aad.DataSourceAADInstances(),
-			"huaweicloud_aad_black_white_lists":       aad.DataSourceAadBlackWhiteLists(),
-			"huaweicloud_aad_web_protection_policies": aad.DataSourceAadWebProtectionPolicies(),
+			"huaweicloud_aad_instances":                aad.DataSourceAADInstances(),
+			"huaweicloud_aad_unblock_quota_statistics": aad.DataSourceUnblockQuotaStatistics(),
+			"huaweicloud_aad_black_white_lists":        aad.DataSourceAadBlackWhiteLists(),
+			"huaweicloud_aad_web_protection_policies":  aad.DataSourceAadWebProtectionPolicies(),
 
 			"huaweicloud_antiddos_config_ranges":                antiddos.DataSourceConfigRanges(),
 			"huaweicloud_antiddos_weekly_protection_statistics": antiddos.DataSourceWeeklyProtectionStatistics(),

huaweicloud/services/aad/data_source_huaweicloud_aad_unblock_quota_statistics.go

@@ -0,0 +1,105 @@
+package aad
+
+import (
+	"context"
+	"strings"
+
+	"github.com/hashicorp/go-multierror"
+	"github.com/hashicorp/go-uuid"
+	"github.com/hashicorp/terraform-plugin-sdk/v2/diag"
+	"github.com/hashicorp/terraform-plugin-sdk/v2/helper/schema"
+
+	"github.com/chnsz/golangsdk"
+
+	"github.com/huaweicloud/terraform-provider-huaweicloud/huaweicloud/config"
+	"github.com/huaweicloud/terraform-provider-huaweicloud/huaweicloud/utils"
+)
+
+// @API AAD GET /v1/unblockservice/{domain_id}/unblock-quota-statistics
+func DataSourceUnblockQuotaStatistics() *schema.Resource {
+	return &schema.Resource{
+		ReadContext: dataSourceUnblockQuotaStatisticsRead,
+
+		Schema: map[string]*schema.Schema{
+			"domain_id": {
+				Type:        schema.TypeString,
+				Required:    true,
+				Description: "Specified the account ID of IAM user.",
+			},
+			"type": {
+				Type:        schema.TypeString,
+				Computed:    true,
+				Description: "The user type.",
+			},
+			"total_unblocking_quota": {
+				Type:        schema.TypeInt,
+				Computed:    true,
+				Description: "The total unblocking quota.",
+			},
+			"remaining_unblocking_quota": {
+				Type:        schema.TypeInt,
+				Computed:    true,
+				Description: "The remaining unblocking quota.",
+			},
+			"unblocking_quota_today": {
+				Type:        schema.TypeInt,
+				Computed:    true,
+				Description: "The unblocking quota of today.",
+			},
+			"remaining_unblocking_quota_today": {
+				Type:        schema.TypeInt,
+				Computed:    true,
+				Description: "The remaining unblocking quota of today.",
+			},
+		},
+	}
+}
+
+func dataSourceUnblockQuotaStatisticsRead(_ context.Context, d *schema.ResourceData, meta interface{}) diag.Diagnostics {
+	var (
+		cfg    = meta.(*config.Config)
+		region = cfg.GetRegion(d)
+		mErr   *multierror.Error
+	)
+
+	client, err := cfg.NewServiceClient("aad", region)
+	if err != nil {
+		return diag.Errorf("error creating AAD client: %s", err)
+	}
+
+	httpUrl := "v1/unblockservice/{domain_id}/unblock-quota-statistics"
+	httpUrl = strings.ReplaceAll(httpUrl, "{domain_id}", d.Get("domain_id").(string))
+	listPath := client.Endpoint + httpUrl
+	opt := golangsdk.RequestOpts{
+		KeepResponseBody: true,
+		MoreHeaders: map[string]string{
+			"Content-Type": "application/json",
+		},
+	}
+
+	requestResp, err := client.Request("GET", listPath, &opt)
+	if err != nil {
+		return diag.Errorf("error retrieving AAD unblock quota statistics: %s", err)
+	}
+
+	respBody, err := utils.FlattenResponse(requestResp)
+	if err != nil {
+		return diag.FromErr(err)
+	}
+
+	id, err := uuid.GenerateUUID()
+	if err != nil {
+		return diag.Errorf("unable to generate ID: %s", err)
+	}
+	d.SetId(id)
+
+	mErr = multierror.Append(
+		mErr,
+		d.Set("type", utils.PathSearch("type", respBody, nil)),
+		d.Set("total_unblocking_quota", utils.PathSearch("total_unblocking_quota", respBody, nil)),
+		d.Set("remaining_unblocking_quota", utils.PathSearch("remaining_unblocking_quota", respBody, nil)),
+		d.Set("unblocking_quota_today", utils.PathSearch("unblocking_quota_today", respBody, nil)),
+		d.Set("remaining_unblocking_quota_today", utils.PathSearch("remaining_unblocking_quota_today", respBody, nil)),
+	)
+	return diag.FromErr(mErr.ErrorOrNil())
+}

huaweicloud/services/acceptance/aad/data_source_huaweicloud_aad_unblock_quota_statistics_test.go

@@ -0,0 +1,47 @@
+package antiddos
+
+import (
+	"fmt"
+	"testing"
+
+	"github.com/hashicorp/terraform-plugin-sdk/v2/helper/resource"
+
+	"github.com/huaweicloud/terraform-provider-huaweicloud/huaweicloud/services/acceptance"
+)
+
+func TestAccUnblockQuotaStatisticsDataSource_basic(t *testing.T) {
+	var (
+		dataSourceName = "data.huaweicloud_aad_unblock_quota_statistics.test"
+		dc             = acceptance.InitDataSourceCheck(dataSourceName)
+	)
+
+	resource.ParallelTest(t, resource.TestCase{
+		PreCheck: func() {
+			acceptance.TestAccPreCheck(t)
+			// Please prepare domain_id before running this test cases.
+			acceptance.TestAccPrecheckDomainId(t)
+		},
+		ProviderFactories: acceptance.TestAccProviderFactories,
+		Steps: []resource.TestStep{
+			{
+				Config: testUnblockQuotaStatistics_basic(),
+				Check: resource.ComposeTestCheckFunc(
+					dc.CheckResourceExists(),
+					resource.TestCheckResourceAttrSet(dataSourceName, "type"),
+					resource.TestCheckResourceAttrSet(dataSourceName, "total_unblocking_quota"),
+					resource.TestCheckResourceAttrSet(dataSourceName, "remaining_unblocking_quota"),
+					resource.TestCheckResourceAttrSet(dataSourceName, "unblocking_quota_today"),
+					resource.TestCheckResourceAttrSet(dataSourceName, "remaining_unblocking_quota_today"),
+				),
+			},
+		},
+	})
+}
+
+func testUnblockQuotaStatistics_basic() string {
+	return fmt.Sprintf(`
+data "huaweicloud_aad_unblock_quota_statistics" "test" {
+  domain_id = "%[1]s"
+}
+`, acceptance.HW_DOMAIN_ID)
+}