package alyoss

import (
	"context"
	fileconfig "gitee.com/h79/goutils/common/file/config"
	filestream "gitee.com/h79/goutils/common/file/stream"
	"github.com/aliyun/aliyun-oss-go-sdk/oss"
)

const minChunkSize = 5 * 1024 * 1024

type OssFile struct {
	config *fileconfig.Config
	bf     *fileconfig.BucketFile
}

func NewOssFile(config *fileconfig.Config, file *fileconfig.BucketFile) *OssFile {
	if config.MinChunkSize <= minChunkSize {
		config.MinChunkSize = minChunkSize
	}
	return &OssFile{config: config, bf: file}
}

func (of *OssFile) UploadFile(ctx context.Context, fileResult *fileconfig.Result) error {
	// 阿里云主账号AccessKey拥有所有API的访问权限，风险很高。强烈建议您创建并使用RAM账号进行API访问或日常运维，请登录 https://ram.console.aliyun.com 创建RAM账号。
	client, err := oss.New(of.config.EndPoint, of.config.AcsKeyID, of.config.AcsSecret)
	if err != nil {
		return err
	}

	bucket, err := client.Bucket(of.bf.Bucket)
	if err != nil {
		return err
	}

	return of.FileUpload(ctx, bucket, fileResult)
}

// IsObjectExist 文件是否存在
func (of *OssFile) IsObjectExist() (bool, error) {

	// 阿里云主账号AccessKey拥有所有API的访问权限，风险很高。强烈建议您创建并使用RAM账号进行API访问或日常运维，请登录 https://ram.console.aliyun.com 创建RAM账号。
	client, err := oss.New(of.config.EndPoint, of.config.AcsKeyID, of.config.AcsSecret)
	if err != nil {
		return false, err
	}

	bucket, err := client.Bucket(of.bf.Bucket)
	if err != nil {
		return false, err
	}
	return bucket.IsObjectExist(of.bf.Key)
}

func (of *OssFile) UploadStream(ctx context.Context, stream *filestream.Stream, fileResult *fileconfig.Result) error {
	// 传入oss key, 文件路径
	// 阿里云主账号AccessKey拥有所有API的访问权限，风险很高。强烈建议您创建并使用RAM账号进行API访问或日常运维，请登录 https://ram.console.aliyun.com 创建RAM账号。
	client, err := oss.New(of.config.EndPoint, of.config.AcsKeyID, of.config.AcsSecret)
	if err != nil {
		return err
	}

	bucket, err := client.Bucket(of.bf.Bucket)
	if err != nil {
		return err
	}

	return of.StreamUpload(ctx, bucket, stream, fileResult)
}