package service

import (
	"fmt"
	"github.com/gin-gonic/gin"
	"github.com/google/uuid"
	"github.com/minio/minio-go/v7"
	"github.com/pkg/errors"
	"gitlab.wodcloud.com/smart-operation/so-operation-api/src/common/client"
	"gitlab.wodcloud.com/smart-operation/so-operation-api/src/common/conf"
	"mime/multipart"
	"net/http"
	"path"
)

var maxFileByte int64 = 20971520 //20M

type DocLibSvc struct{}

func GetFileContentType(out multipart.File) (contentType string, err error) {
	buffer := make([]byte, 512)
	_, err = out.Read(buffer)
	if err != nil {
		return
	}
	contentType = http.DetectContentType(buffer)
	return
}

func (d *DocLibSvc) AddFile(c *gin.Context) (fileUrl string, err error) {
	var (
		uploadFile  *multipart.FileHeader
		uploadFd    multipart.File
		minioClient *minio.Client
		ext         string
		contentType string
	)
	uploadFile, err = c.FormFile("uploadFile")
	if err != nil {
		err = errors.Wrap(err, "上传的文件不存在")
		return
	}
	if uploadFile.Size > maxFileByte {
		err = errors.New("超过上传最大限制")
		return
	}
	uploadFd, err = uploadFile.Open()
	if err != nil {
		err = errors.Wrap(err, "打开文件失败")
		return
	}
	defer uploadFd.Close()

	ext = path.Ext(uploadFile.Filename)
	contentType, err = GetFileContentType(uploadFd)
	if err != nil {
		err = errors.Wrap(err, "获取ContentType失败")
		return
	}
	uploadFd.Seek(0, 0)

	//上传文件到minio
	minioClient, err = client.GetMinioConnect()
	if err != nil {
		err = errors.Wrap(err, "getMinio")
		return
	}
	id := uuid.NewString()
	_, err = minioClient.PutObject(c,
		conf.Options.MinioBucket,
		fmt.Sprintf("%s%s", id, ext),
		uploadFd,
		uploadFile.Size,
		minio.PutObjectOptions{
			ContentType: contentType,
			UserMetadata: map[string]string{
				"name": uploadFile.Filename,
			},
		})
	if err != nil {
		err = errors.Wrap(err, "上传文件失败")
		return
	}
	fileUrl = fmt.Sprintf("https://%s/%s/%s%s", conf.Options.MinioServer, conf.Options.MinioBucket, id, ext)
	return
}