package storage

import (
	"context"
	"github.com/tencentyun/cos-go-sdk-v5"
	"net/http"
	"net/url"
	"uploader/config"
)

type CosStorage struct {
	Config *config.Cos
}

func NewCosStorage(config *config.Cos) *CosStorage {
	return &CosStorage{Config: config}
}

func (s *CosStorage) getClient() (*cos.Client, error) {
	u, _ := url.Parse(s.Config.BucketURL)
	b := &cos.BaseURL{BucketURL: u}
	client := cos.NewClient(b, &http.Client{
		Transport: &cos.AuthorizationTransport{
			// 通过环境变量获取密钥
			// 环境变量 SECRETID 表示用户的 SecretId，登录访问管理控制台查看密钥，https://console.cloud.tencent.com/cam/capi
			SecretID: s.Config.SecretID, // 用户的 SecretId，建议使用子账号密钥，授权遵循最小权限指引，降低使用风险。子账号密钥获取可参见 https://cloud.tencent.com/document/product/598/37140
			// 环境变量 SECRETKEY 表示用户的 SecretKey，登录访问管理控制台查看密钥，https://console.cloud.tencent.com/cam/capi
			SecretKey: s.Config.SecretKey, // 用户的 SecretKey，建议使用子账号密钥，授权遵循最小权限指引，降低使用风险。子账号密钥获取可参见 https://cloud.tencent.com/document/product/598/37140
		},
	})
	return client, nil
}

func (s *CosStorage) Upload(filePath, destPath string) (string, error) {
	client, err := s.getClient()
	if err != nil {
		return "", err
	}

	_, _, err = client.Object.Upload(
		context.Background(), destPath, filePath, &cos.MultiUploadOptions{ThreadPoolSize: 5, PartSize: 30},
	)
	if err != nil {
		return "", err
	}

	return s.Config.Domain + "/" + destPath, err
}

func (s *CosStorage) Delete(objectName string) error {
	client, err := s.getClient()
	if err != nil {
		return err
	}
	_, err = client.Object.Delete(context.Background(), objectName)
	return err
}