package service

import (
	"fmt"
	"mime/multipart"
	"os"
	"path"
	"runtime"
	"slices"
	"strings"
	"time"
	err2 "yintai-api/app/common/err"
	"yintai-api/app/common/service"

	"github.com/gin-gonic/gin"
	log "github.com/sirupsen/logrus"
	"github.com/smbrave/goutil"
)

type Files struct{}

// 直接上传到七牛云上。返回保存的地址，访问的url每次都用新的

func (f Files) UploadToken() string {
	return service.NewQiniu().UploadToken()
}

func (f Files) Upload(file *multipart.FileHeader, ctx *gin.Context) any {
	start := time.Now()
	defer log.Debug(" [all] time cost:", time.Since(start))
	if file == nil {
		err2.ParamErr("file is nil")
		return nil
	}

	// support check
	if err := f.Support(path.Ext(file.Filename), file.Size); err != nil {
		err2.ParamErr(err)
	}

	// 保存本地临时文件
	dst := f.Dst(file.Filename)
	if runtime.GOOS == "windows" {
		dst = dst[1:]
	}

	err := ctx.SaveUploadedFile(file, dst)
	if err != nil {
		err2.InnerErr(err)
	}

	objectKey := service.NewQiniu().Save(dst, f.ObjectKey(file.Filename))

	defer func() {
		_ = os.Remove(dst)
	}()

	return service.NewQiniu().ForeverUrl(objectKey)
}

func (f Files) ObjectKey(ext string) string {
	return fmt.Sprintf("yintai/%s", f.Name(ext))
}

func (f Files) Dst(ext string) string {
	return "/tmp/" + f.Name(ext)
}

func (f Files) Name(ext string) string {
	now := time.Now().Format("20060102150405")
	return fmt.Sprintf("%s_%s%s", now, goutil.RandomStr(4), ext)

}

func (f Files) Support(ext string, size int64) error {
	video := []string{"mp4", "m3u8"}
	audio := []string{"mp3"}
	img := []string{"png", "jpg", "jpeg", "jpg", "svg", "gif", "webp"}
	doc := []string{"ppt", "xls", "xlsx", "pptx", "pdf"}

	type T struct {
		support []string
		size    int64
	}

	x := make([]T, 0)
	x = append(x, T{support: video, size: 30 << 20})
	x = append(x, T{support: audio, size: 30 << 20})
	x = append(x, T{support: img, size: 10 << 20})
	x = append(x, T{support: doc, size: 120 << 20})

	for _, t := range x {
		ext = strings.TrimLeft(ext, ".")
		if slices.Contains(t.support, ext) {
			if size > t.size {
				return fmt.Errorf("file too large")
			}
		}
	}

	return nil
}