auxpi/utils/upload.go

package utils

import (
	"auxpi/auxpiAll"
	"auxpi/bootstrap"
	"auxpi/server"
	"github.com/astaxie/beego"
	"mime/multipart"
	"strings"
)

type UpLoadTools struct {
	beego.Controller
	server.Smms
	server.SouGou
	server.Sina
}

var picType = []string{"png", "jpg", "jpeg", "gif", "bmp"}
var siteConfig = bootstrap.Config()
//返回不同图床的 URL
func (this *UpLoadTools) HandleUrl(apiSelect string, f multipart.File, h *multipart.FileHeader) string {
	imgMime := h.Header.Get("Content-Type")
	imgInfo := h.Header.Get("Content-Disposition")
	//读取文件
	size := h.Size
	fileContent := make([]byte, size)
	f.Read(fileContent)
	url := ""
	switch apiSelect {
	case "SouGou":
		url = this.UpLoadToSouGou(fileContent)
	case "Sina":
		if siteConfig.SiteUploadWay.OpenSinaPicStore == false {
			url = ""
		}
		url = this.UpLoadToSina(fileContent, imgMime)
	case "Smms":
		url = this.UpLoatToSmms(fileContent, imgInfo)
	default:
		switch siteConfig.ApiDefault {
		case "SouGou":
			url = this.UpLoadToSouGou(fileContent)
		case "Smms":
			url = this.UpLoatToSmms(fileContent, imgInfo)
		}

	}
	return url
}

//验证文件后缀&文件MIME
func (this *UpLoadTools) Validate(contentType string, fileName string) bool {
	//首先检测文件的后缀
	isSuffix := false
	for _, pType := range picType {
		if strings.HasSuffix(fileName, pType) {
			isSuffix = true
			break
		}
	}
	//然后检测 MIME 类型
	//beego.Alert(contentType)
	if strings.HasPrefix(contentType, "image") && isSuffix {
		for _, pType := range picType {
			if strings.HasSuffix(contentType, pType) {
				return true
			}
		}

	}
	return false
}

//错误resp
func (this *UpLoadTools) ErrorResp(code int, msg string) {
	result := &auxpi.ErrorJson{}
	result.Code = code
	result.Msg = msg
	this.Data["json"] = result
	this.ServeJSON()
}

func (this *UpLoadTools) SuccResp(code int, msg string, url string, name string) {
	result := &auxpi.ResultJson{}
	result.Code = code
	result.Msg = msg
	result.Data.Url = url
	result.Data.Name = name
	//beego.Alert(result)
	this.Data["json"] = result
	this.ServeJSON()
}