add cos operation (#1941)

* add cos operation

* add changelog

---------

Co-authored-by: mikatong <mikatong@tencent.com>

Author: tongyiming

.changelog/1941.txt

@@ -0,0 +1,15 @@
+```release-note:new-resource
+tencentcloud_cos_object_abort_multipart_upload_operation
+```
+
+```release-note:new-resource
+tencentcloud_cos_object_copy_operation
+```
+
+```release-note:new-resource
+tencentcloud_cos_object_restore_operation
+```
+
+```release-note:new-resource
+tencentcloud_cos_bucket_generate_inventory_immediately_operation
+```

tencentcloud/provider.go

@@ -0,0 +1,111 @@
+/*
+Provides a resource to generate a cos bucket inventory immediately
+
+Example Usage
+
+```hcl
+resource "tencentcloud_cos_bucket_generate_inventory_immediately_operation" "generate_inventory_immediately" {
+    inventory_id = "test"
+    bucket = "keep-test-xxxxxx"
+}
+```
+*/
+package tencentcloud
+
+import (
+	"context"
+	"fmt"
+	"log"
+	"strings"
+	"time"
+
+	"github.com/hashicorp/terraform-plugin-sdk/v2/helper/resource"
+	"github.com/hashicorp/terraform-plugin-sdk/v2/helper/schema"
+	"github.com/tencentyun/cos-go-sdk-v5"
+)
+
+func resourceTencentCloudCosBucketGenerateInventoryImmediatelyOperation() *schema.Resource {
+	return &schema.Resource{
+		Create: resourceTencentCloudCosBucketGenerateInventoryImmediatelyOperationCreate,
+		Read:   resourceTencentCloudCosBucketGenerateInventoryImmediatelyOperationRead,
+		Delete: resourceTencentCloudCosBucketGenerateInventoryImmediatelyOperationDelete,
+
+		Schema: map[string]*schema.Schema{
+			"inventory_id": {
+				Required:    true,
+				ForceNew:    true,
+				Type:        schema.TypeString,
+				Description: "The id of inventory.",
+			},
+			"bucket": {
+				Required:    true,
+				ForceNew:    true,
+				Type:        schema.TypeString,
+				Description: "Bucket.",
+			},
+		},
+	}
+}
+
+func resourceTencentCloudCosBucketGenerateInventoryImmediatelyOperationCreate(d *schema.ResourceData, meta interface{}) error {
+	defer logElapsed("resource.tencentcloud_cos_bucket_generate_inventory_immediately_operation.create")()
+	defer inconsistentCheck(d, meta)()
+
+	logId := getLogId(contextNil)
+	ctx := context.WithValue(context.TODO(), logIdKey, logId)
+	inventoryId := d.Get("inventory_id").(string)
+	bucket := d.Get("bucket").(string)
+	result, _, err := meta.(*TencentCloudClient).apiV3Conn.UseTencentCosClient(bucket).Bucket.GetInventory(ctx, inventoryId)
+	if err != nil {
+		log.Printf("[CRITAL]%s get cos bucketInventory failed, reason:%+v", logId, err)
+		return err
+	}
+	id := fmt.Sprintf("%s_instant_%s", result.ID, time.Now().Format("20060102150405"))
+	inventoryOpt := &cos.BucketPostInventoryOptions{
+		ID:                     id,
+		IncludedObjectVersions: result.IncludedObjectVersions,
+		Filter:                 result.Filter,
+		OptionalFields:         result.OptionalFields,
+		Destination:            result.Destination,
+	}
+	_, err = meta.(*TencentCloudClient).apiV3Conn.UseTencentCosClient(bucket).Bucket.PostInventory(ctx, id, inventoryOpt)
+	if err != nil {
+		return err
+	}
+	items := strings.Split(result.Destination.Bucket, "::")
+	targetBucket := items[len(items)-1]
+	targetBucketItems := strings.Split(targetBucket, "-")
+	objectId := fmt.Sprintf("%s/%s/%s/%s/%s/manifest.json", result.Destination.Prefix, targetBucketItems[len(targetBucketItems)-1], strings.Join(targetBucketItems[:len(targetBucketItems)-1], "-"), id, time.Now().Format("20060102"))
+	err = resource.Retry(10*readRetryTimeout, func() *resource.RetryError {
+		resp, e := meta.(*TencentCloudClient).apiV3Conn.UseTencentCosClient(targetBucket).Object.Head(ctx, objectId, nil)
+		if resp.StatusCode == 404 {
+			return resource.RetryableError(fmt.Errorf("Inventory still creating!"))
+		}
+		if e != nil {
+			return retryError(e)
+		}
+
+		return nil
+	})
+	if err != nil {
+		log.Printf("[CRITAL]%s PostInventory: object not exist, reason:%+v", logId, err)
+		return err
+	}
+	d.SetId(id)
+
+	return resourceTencentCloudCosBucketGenerateInventoryImmediatelyOperationRead(d, meta)
+}
+
+func resourceTencentCloudCosBucketGenerateInventoryImmediatelyOperationRead(d *schema.ResourceData, meta interface{}) error {
+	defer logElapsed("resource.tencentcloud_cos_bucket_generate_inventory_immediately_operation.read")()
+	defer inconsistentCheck(d, meta)()
+
+	return nil
+}
+
+func resourceTencentCloudCosBucketGenerateInventoryImmediatelyOperationDelete(d *schema.ResourceData, meta interface{}) error {
+	defer logElapsed("resource.tencentcloud_cos_bucket_generate_inventory_immediately_operation.delete")()
+	defer inconsistentCheck(d, meta)()
+
+	return nil
+}

tencentcloud/resource_tc_cos_bucket_generate_inventory_immediately_operation.go

@@ -0,0 +1,30 @@
+package tencentcloud
+
+import (
+	"testing"
+
+	"github.com/hashicorp/terraform-plugin-sdk/v2/helper/resource"
+)
+
+func TestAccTencentCloudCosBucketGenerateInventoryImmediatelyOperationResource(t *testing.T) {
+
+	resource.Test(t, resource.TestCase{
+		PreCheck:  func() { testAccPreCheck(t) },
+		Providers: testAccProviders,
+		Steps: []resource.TestStep{
+			{
+				Config: testAccCosBucketGenerateInventoryImmediatelyOperation,
+				Check: resource.ComposeAggregateTestCheckFunc(
+					resource.TestCheckResourceAttrSet("tencentcloud_cos_bucket_generate_inventory_immediately_operation.generate_inventory_immediately", "id"),
+				),
+			},
+		},
+	})
+}
+
+const testAccCosBucketGenerateInventoryImmediatelyOperation = `
+resource "tencentcloud_cos_bucket_generate_inventory_immediately_operation" "generate_inventory_immediately" {
+    inventory_id = "test"
+    bucket = "keep-test-1308919341"
+}
+`

tencentcloud/resource_tc_cos_bucket_generate_inventory_immediately_operation_test.go

@@ -0,0 +1,84 @@
+/*
+Provides a resource to abort multipart upload
+
+Example Usage
+
+```hcl
+resource "tencentcloud_cos_object_abort_multipart_upload_operation" "abort_multipart_upload" {
+    bucket = "keep-test-xxxxxx"
+    key = "object"
+    upload_id = "xxxxxx"
+}
+```
+*/
+package tencentcloud
+
+import (
+	"context"
+	"log"
+
+	"github.com/hashicorp/terraform-plugin-sdk/v2/helper/schema"
+)
+
+func resourceTencentCloudCosObjectAbortMultipartUploadOperation() *schema.Resource {
+	return &schema.Resource{
+		Create: resourceTencentCloudCosObjectAbortMultipartUploadOperationCreate,
+		Read:   resourceTencentCloudCosObjectAbortMultipartUploadOperationRead,
+		Delete: resourceTencentCloudCosObjectAbortMultipartUploadOperationDelete,
+
+		Schema: map[string]*schema.Schema{
+			"bucket": {
+				Required:    true,
+				ForceNew:    true,
+				Type:        schema.TypeString,
+				Description: "Bucket.",
+			},
+			"key": {
+				Required:    true,
+				ForceNew:    true,
+				Type:        schema.TypeString,
+				Description: "Object key.",
+			},
+			"upload_id": {
+				Required:    true,
+				ForceNew:    true,
+				Type:        schema.TypeString,
+				Description: "Multipart uploaded id.",
+			},
+		},
+	}
+}
+
+func resourceTencentCloudCosObjectAbortMultipartUploadOperationCreate(d *schema.ResourceData, meta interface{}) error {
+	defer logElapsed("resource.tencentcloud_cos_object_abort_multipart_upload_operation.create")()
+	defer inconsistentCheck(d, meta)()
+
+	logId := getLogId(contextNil)
+	ctx := context.WithValue(context.TODO(), logIdKey, logId)
+	bucket := d.Get("bucket").(string)
+	key := d.Get("key").(string)
+	uploadId := d.Get("upload_id").(string)
+	_, err := meta.(*TencentCloudClient).apiV3Conn.UseTencentCosClient(bucket).Object.AbortMultipartUpload(ctx, key, uploadId)
+	if err != nil {
+		log.Printf("[CRITAL]%s AbortMultipartUpload failed, reason:%+v", logId, err)
+		return err
+	}
+
+	d.SetId(uploadId)
+
+	return resourceTencentCloudCosObjectAbortMultipartUploadOperationRead(d, meta)
+}
+
+func resourceTencentCloudCosObjectAbortMultipartUploadOperationRead(d *schema.ResourceData, meta interface{}) error {
+	defer logElapsed("resource.tencentcloud_cos_object_abort_multipart_upload_operation.read")()
+	defer inconsistentCheck(d, meta)()
+
+	return nil
+}
+
+func resourceTencentCloudCosObjectAbortMultipartUploadOperationDelete(d *schema.ResourceData, meta interface{}) error {
+	defer logElapsed("resource.tencentcloud_cos_object_abort_multipart_upload_operation.delete")()
+	defer inconsistentCheck(d, meta)()
+
+	return nil
+}

tencentcloud/resource_tc_cos_object_abort_multipart_upload_operation.go

@@ -0,0 +1,84 @@
+/*
+Provides a resource to copy object
+
+Example Usage
+
+```hcl
+resource "tencentcloud_cos_object_copy_operation" "object_copy" {
+    bucket = "keep-copy-xxxxxxx"
+    key = "copy-acl.txt"
+    source_url = "keep-test-xxxxxx.cos.ap-guangzhou.myqcloud.com/acl.txt"
+}
+```
+*/
+package tencentcloud
+
+import (
+	"context"
+	"log"
+
+	"github.com/hashicorp/terraform-plugin-sdk/v2/helper/schema"
+)
+
+func resourceTencentCloudCosObjectCopyOperation() *schema.Resource {
+	return &schema.Resource{
+		Create: resourceTencentCloudCosObjectCopyOperationCreate,
+		Read:   resourceTencentCloudCosObjectCopyOperationRead,
+		Delete: resourceTencentCloudCosObjectCopyOperationDelete,
+
+		Schema: map[string]*schema.Schema{
+			"bucket": {
+				Required:    true,
+				ForceNew:    true,
+				Type:        schema.TypeString,
+				Description: "Bucket.",
+			},
+			"key": {
+				Required:    true,
+				ForceNew:    true,
+				Type:        schema.TypeString,
+				Description: "Object key.",
+			},
+			"source_url": {
+				Required:    true,
+				ForceNew:    true,
+				Type:        schema.TypeString,
+				Description: "Object key.",
+			},
+		},
+	}
+}
+
+func resourceTencentCloudCosObjectCopyOperationCreate(d *schema.ResourceData, meta interface{}) error {
+	defer logElapsed("resource.tencentcloud_cos_object_copy_operation.create")()
+	defer inconsistentCheck(d, meta)()
+
+	logId := getLogId(contextNil)
+	ctx := context.WithValue(context.TODO(), logIdKey, logId)
+	bucket := d.Get("bucket").(string)
+	key := d.Get("key").(string)
+	sourceURL := d.Get("source_url").(string)
+	_, _, err := meta.(*TencentCloudClient).apiV3Conn.UseTencentCosClient(bucket).Object.Copy(ctx, key, sourceURL, nil)
+	if err != nil {
+		log.Printf("[CRITAL]%s Restore failed, reason:%+v", logId, err)
+		return err
+	}
+
+	d.SetId(bucket + FILED_SP + key)
+
+	return resourceTencentCloudCosObjectCopyOperationRead(d, meta)
+}
+
+func resourceTencentCloudCosObjectCopyOperationRead(d *schema.ResourceData, meta interface{}) error {
+	defer logElapsed("resource.tencentcloud_cos_object_copy_operation.read")()
+	defer inconsistentCheck(d, meta)()
+
+	return nil
+}
+
+func resourceTencentCloudCosObjectCopyOperationDelete(d *schema.ResourceData, meta interface{}) error {
+	defer logElapsed("resource.tencentcloud_cos_object_copy_operation.delete")()
+	defer inconsistentCheck(d, meta)()
+
+	return nil
+}

tencentcloud/resource_tc_cos_object_copy_operation.go

@@ -0,0 +1,31 @@
+package tencentcloud
+
+import (
+	"testing"
+
+	"github.com/hashicorp/terraform-plugin-sdk/v2/helper/resource"
+)
+
+func TestAccTencentCloudCosObjectCopyOperationResource(t *testing.T) {
+
+	resource.Test(t, resource.TestCase{
+		PreCheck:  func() { testAccPreCheck(t) },
+		Providers: testAccProviders,
+		Steps: []resource.TestStep{
+			{
+				Config: testAccCosObjectCopyOperation,
+				Check: resource.ComposeAggregateTestCheckFunc(
+					resource.TestCheckResourceAttrSet("tencentcloud_cos_object_copy_operation.object_copy", "id"),
+				),
+			},
+		},
+	})
+}
+
+const testAccCosObjectCopyOperation = `
+resource "tencentcloud_cos_object_copy_operation" "object_copy" {
+    bucket = "keep-copy-1308919341"
+    key = "copy-acl.txt"
+    source_url = "keep-test-1308919341.cos.ap-guangzhou.myqcloud.com/acl.txt"
+}
+`