package alyoss

import (
	"fmt"
	"gitee.com/h79/goutils/common/filemgr/structs"
	"gitee.com/h79/goutils/common/filemgr/tools"
	"gitee.com/h79/goutils/common/logger"
	"gitee.com/h79/goutils/common/result"
	"github.com/aliyun/aliyun-oss-go-sdk/oss"
	"io"
	"os"
)

// FileUpload 文件上传
func (of *OssFile) FileUpload(fileResult *structs.FileResult, data *structs.FileName, bucket *oss.Bucket) error {

	fd, err := os.Open(data.Name)
	defer fd.Close()

	if err != nil {
		return result.Error(result.ErrUpload, fmt.Sprintf("[File]open file size : %s", err))
	}

	f, err := fd.Stat()
	if err != nil {
		return result.Error(result.ErrUpload, fmt.Sprintf("[File]file stat %s", err))
	}

	if f.Size() <= of.config.FileConfig.MinChunkSize {
		//直接文件上传
		if err := bucket.PutObjectFromFile(data.Key, data.Name); err != nil {
			return result.Error(result.ErrUpload, fmt.Sprintf("[File]Bucket put file %s", err))
		}
		return nil
	}

	//分片处理
	imur, err := bucket.InitiateMultipartUpload(data.Key)
	if err != nil {
		return result.Error(result.ErrUpload, fmt.Sprintf("[File]Bucket InitiateMultipartUpload %s", err))
	}

	chunks := tools.Chunk(f.Size(), tools.ChunkNum(f.Size(), of.config.FileConfig.MinChunkSize))

	// 对每个分片调用UploadPart方法上传。
	var parts []oss.UploadPart
	for _, chunk := range chunks {
		_, err := fd.Seek(chunk.Offset, io.SeekStart)
		if err != nil {
			return result.Error(result.ErrUpload, fmt.Sprintf("[File]Bucket part file seek %s", err))
		}
		part, err := bucket.UploadPart(imur, fd, chunk.Size, chunk.Number)
		if err != nil {
			return result.Error(result.ErrUpload, fmt.Sprintf("[File]Bucket part file %s", err))
		}
		parts = append(parts, part)
	}
	// 完成分片上传
	cmur, err := bucket.CompleteMultipartUpload(imur, parts)
	if err != nil {
		return result.Error(result.ErrUpload, fmt.Sprintf("[File]Bucket complete part %s", err))
	}
	logger.Info("Upload: Complete part %v", cmur)
	return nil
}