ferry/apis/public/file.go

package public

import (
	"encoding/base64"
	"errors"
	"ferry/pkg/utils"
	"ferry/tools/app"
	"fmt"
	"io/ioutil"
	"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 (
		tag       string
		urlPrefix string
	)
	tag, _ = c.GetPostForm("type")
	if strings.HasSuffix(viper.GetString("settings.domain"), "/") {
		urlPrefix = viper.GetString("settings.domain")
	} else {
		urlPrefix = fmt.Sprintf("%v/", viper.GetString("settings.domain"))
	}
	if tag == "" {
		app.Error(c, 200, errors.New(""), "缺少标识")
		return
	} else {
		switch tag {
		case "1": // 单图
			files, err := c.FormFile("file")
			if err != nil {
				app.Error(c, 200, errors.New(""), "图片不能为空")
				return
			}
			// 上传文件至指定目录
			guid := uuid.New().String()

			singleFile := "static/uploadfile/" + guid + utils.GetExt(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 := uuid.New().String()
				multipartFileName := "static/uploadfile/" + guid + utils.GetExt(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)
			guid := uuid.New().String()
			_ = ioutil.WriteFile("static/uploadfile/"+guid+".jpg", ddd, 0666)
			app.OK(c, urlPrefix+"static/uploadfile/"+guid+".jpg", "上传成功")
		}
	}
}