ferry/apis/public/file.go

package public

import (
	"encoding/base64"
	"errors"
	"ferry/tools/app"
	"fmt"
	"io/ioutil"
	"os"
	"strings"

	"github.com/spf13/viper"

	"github.com/gin-gonic/gin"
	"github.com/google/uuid"
)

// @Summary 上传图片
// @Description 获取JSON
// @Tags 公共接口
// @Accept multipart/form-data
// @Param type query string true "type" (1：单图，2：多图, 3：base64图片)
// @Param file formData file true "file"
// @Success 200 {string} string	"{"code": 200, "message": "添加成功"}"
// @Success 200 {string} string	"{"code": -1, "message": "添加失败"}"
// @Router /api/v1/public/uploadFile [post]

func UploadFile(c *gin.Context) {
	var (
		urlPrefix    string
		tag          string
		fileType     string
		saveFilePath string
		err          error
		protocol     string = "http"
		requestHost  string
	)
	tag, _ = c.GetPostForm("type")
	fileType = c.DefaultQuery("file_type", "images")

	if fileType != "images" && fileType != "files" {
		app.Error(c, -1, fmt.Errorf("上传接口目前，仅支持图片上传和文件上传"), "")
		return
	}

	if strings.HasPrefix(c.Request.Header.Get("Origin"), "https") {
		protocol = "https"
	}

	requestHostList := strings.Split(c.Request.Host, ":")
	if len(requestHostList) > 1 && requestHostList[1] == "80" {
		requestHost = requestHostList[0]
	} else {
		requestHost = c.Request.Host
	}

	if viper.GetBool("settings.domain.getHost") {
		urlPrefix = fmt.Sprintf("%s://%s/", protocol, requestHost)
	} else {
		urlPrefix = fmt.Sprintf("%s://%s", protocol, viper.GetString("settings.domain.url"))
		if !strings.HasSuffix(viper.GetString("settings.domain.url"), "/") {
			urlPrefix = urlPrefix + "/"
		}
	}

	if tag == "" {
		tag = "1"
	}

	saveFilePath = "static/uploadfile/" + fileType + "/"
	_, err = os.Stat(saveFilePath)
	if os.IsNotExist(err) {
		err = os.MkdirAll(saveFilePath, 0755)
		if err != nil {
			app.Error(c, -1, err, fmt.Sprintf("创建图片目录失败，%v", err.Error()))
			return
		}
	}

	guid := strings.ReplaceAll(uuid.New().String(), "-", "")

	switch tag {
	case "1": // 单图
		files, err := c.FormFile("file")
		if err != nil {
			app.Error(c, 200, errors.New(""), "图片不能为空")
			return
		}
		// 上传文件至指定目录
		singleFile := saveFilePath + guid + "-" + files.Filename
		_ = c.SaveUploadedFile(files, singleFile)
		app.OK(c, urlPrefix+singleFile, "上传成功")
		return
	case "2": // 多图
		files := c.Request.MultipartForm.File["file"]
		multipartFile := make([]string, len(files))
		for _, f := range files {
			guid = strings.ReplaceAll(uuid.New().String(), "-", "")
			multipartFileName := saveFilePath + guid + "-" + f.Filename
			_ = c.SaveUploadedFile(f, multipartFileName)
			multipartFile = append(multipartFile, urlPrefix+multipartFileName)
		}
		app.OK(c, multipartFile, "上传成功")
		return
	case "3": // base64
		files, _ := c.GetPostForm("file")
		ddd, _ := base64.StdEncoding.DecodeString(files)
		_ = ioutil.WriteFile(saveFilePath+guid+".jpg", ddd, 0666)
		app.OK(c, urlPrefix+saveFilePath+guid+".jpg", "上传成功")
	default:
		app.Error(c, 200, errors.New(""), "标识不正确")
		return
	}
}