data_source_tc_cos_bucket_multipart_uploads.go

package cos

import (
	"context"
	"encoding/json"
	"log"

	tccommon "github.com/tencentcloudstack/terraform-provider-tencentcloud/tencentcloud/common"

	"github.com/hashicorp/terraform-plugin-sdk/v2/helper/schema"
	"github.com/tencentyun/cos-go-sdk-v5"

	"github.com/tencentcloudstack/terraform-provider-tencentcloud/tencentcloud/internal/helper"
)

func DataSourceTencentCloudCosBucketMultipartUploads() *schema.Resource {
	return &schema.Resource{
		Read: dataSourceTencentCloudCosBucketMultipartUploadsRead,

		Schema: map[string]*schema.Schema{
			"bucket": {
				Type:        schema.TypeString,
				Required:    true,
				Description: "Bucket.",
			},
			"delimiter": {
				Type:        schema.TypeString,
				Optional:    true,
				Description: "The delimiter is a symbol, and the Object name contains the Object between the specified prefix and the first occurrence of delimiter characters as a set of elements: common prefix. If there is no prefix, start from the beginning of the path.",
			},
			"encoding_type": {
				Type:        schema.TypeString,
				Optional:    true,
				Description: "Specifies the encoding format of the return value. Legal value: url.",
			},
			"prefix": {
				Type:        schema.TypeString,
				Optional:    true,
				Description: "The returned Object key must be prefixed with Prefix. Note that when using the prefix query, the returned key still contains Prefix.",
			},
			"uploads": {
				Type:        schema.TypeList,
				Computed:    true,
				Description: "Information for each Upload.",
				Elem: &schema.Resource{
					Schema: map[string]*schema.Schema{
						"key": {
							Type:        schema.TypeString,
							Computed:    true,
							Description: "Name of the Object.",
						},
						"upload_id": {
							Type:        schema.TypeString,
							Computed:    true,
							Description: "Mark the ID of this multipart upload.",
						},
						"storage_class": {
							Type:        schema.TypeString,
							Computed:    true,
							Description: "Used to represent the storage level of a chunk. Enumerated value: STANDARD,STANDARD_IA,ARCHIVE.",
						},
						"initiated": {
							Type:        schema.TypeString,
							Computed:    true,
							Description: "The starting time of multipart upload.",
						},
						"owner": {
							Type:        schema.TypeList,
							Computed:    true,
							Description: "Information used to represent the owner of these chunks.",
							Elem: &schema.Resource{
								Schema: map[string]*schema.Schema{
									"id": {
										Type:        schema.TypeString,
										Computed:    true,
										Description: "The user's unique CAM identity ID.",
									},
									"display_name": {
										Type:        schema.TypeString,
										Computed:    true,
										Description: "Abbreviation for user identity ID (UIN).",
									},
								},
							},
						},
						"initiator": {
							Type:        schema.TypeList,
							Computed:    true,
							Description: "Used to represent the information of the initiator of this upload.",
							Elem: &schema.Resource{
								Schema: map[string]*schema.Schema{
									"id": {
										Type:        schema.TypeString,
										Computed:    true,
										Description: "The user's unique CAM identity ID.",
									},
									"display_name": {
										Type:        schema.TypeString,
										Computed:    true,
										Description: "Abbreviation for user identity ID (UIN).",
									},
								},
							},
						},
					},
				},
			},
			"result_output_file": {
				Type:        schema.TypeString,
				Optional:    true,
				Description: "Used to save results.",
			},
		},
	}
}

func dataSourceTencentCloudCosBucketMultipartUploadsRead(d *schema.ResourceData, meta interface{}) error {
	defer tccommon.LogElapsed("data_source.tencentcloud_cos_bucket_multipart_uploads.read")()

	logId := tccommon.GetLogId(tccommon.ContextNil)
	ctx := context.WithValue(context.TODO(), tccommon.LogIdKey, logId)
	bucket := d.Get("bucket").(string)
	multipartUploads := make([]map[string]interface{}, 0)
	opt := &cos.ListMultipartUploadsOptions{}
	if v, ok := d.GetOk("delimiter"); ok {
		opt.Delimiter = v.(string)
	}
	if v, ok := d.GetOk("encoding_type"); ok {
		opt.EncodingType = v.(string)
	}
	if v, ok := d.GetOk("prefix"); ok {
		opt.Prefix = v.(string)
	}
	ids := make([]string, 0)
	for {
		result, response, err := meta.(tccommon.ProviderMeta).GetAPIV3Conn().UseTencentCosClient(bucket).Bucket.ListMultipartUploads(ctx, opt)
		responseBody, _ := json.Marshal(response.Body)
		if err != nil {
			return err
		}
		log.Printf("[DEBUG]%s api[ListMultipartUploads] success, response body [%s]\n", logId, responseBody)
		for _, item := range result.Uploads {
			itemMap := make(map[string]interface{})
			itemMap["key"] = item.Key
			itemMap["upload_id"] = item.UploadID
			itemMap["initiated"] = item.Initiated
			itemMap["storage_class"] = item.StorageClass
			if item.Owner != nil {
				owner := map[string]interface{}{
					"display_name": item.Owner.DisplayName,
					"id":           item.Owner.ID,
				}
				itemMap["owner"] = []map[string]interface{}{owner}
			}
			if item.Initiator != nil {
				initiator := map[string]interface{}{
					"display_name": item.Initiator.DisplayName,
					"id":           item.Initiator.ID,
				}
				itemMap["initiator"] = []map[string]interface{}{initiator}
			}
			ids = append(ids, item.UploadID)
			multipartUploads = append(multipartUploads, itemMap)
		}
		if result.IsTruncated {
			opt.KeyMarker = result.KeyMarker
			opt.UploadIDMarker = result.UploadIDMarker
		} else {
			break
		}
	}

	d.SetId(helper.DataResourceIdsHash(ids))
	_ = d.Set("uploads", multipartUploads)
	output, ok := d.GetOk("result_output_file")
	if ok && output.(string) != "" {
		if err := tccommon.WriteToFile(output.(string), multipartUploads); err != nil {
			return err
		}
	}

	return nil
}