eta_api/controllers/resource.go

package controllers

import (
	"bufio"
	"encoding/base64"
	"eta_gn/eta_api/models"
	"eta_gn/eta_api/services"
	"eta_gn/eta_api/utils"
	"fmt"
	"github.com/h2non/filetype"
	"github.com/kgiannakakis/mp3duration/src/mp3duration"
	"io"
	"io/ioutil"
	"net/http"
	"os"
	"path"
	"regexp"
	"strconv"
	"strings"
	"time"

	rdHttp "github.com/rdlucklib/rdluck_tools/http"
)

type ResourceController struct {
	BaseCommonController
}

type ResourceAuthController struct {
	BaseAuthController
}

// @router /image/upload [post]
func (this *ResourceController) Upload() {
	br := new(models.BaseResponse).Init()
	defer func() {
		this.Data["json"] = br
		this.ServeJSON()
	}()
	f, h, err := this.GetFile("file")
	if err != nil {
		br.Msg = "获取资源信息失败"
		br.ErrMsg = "获取资源信息失败,Err:" + err.Error()
		return
	}

	fileData, e := ioutil.ReadAll(f)
	if e != nil {
		br.Msg = "上传失败"
		br.ErrMsg = "读取文件失败, Err: " + e.Error()
		return
	}
	pass := filetype.IsImage(fileData)
	if !pass {
		br.Msg = "文件格式有误"
		br.ErrMsg = "文件格式有误"
		return
	}

	uploadFileName := h.Filename //上传的文件名
	ext := path.Ext(h.Filename)
	dateDir := time.Now().Format("20060102")
	uploadDir := utils.STATIC_DIR + "hongze/" + dateDir
	err = os.MkdirAll(uploadDir, utils.DIR_MOD)
	if err != nil {
		br.Msg = "存储目录创建失败"
		br.ErrMsg = "存储目录创建失败,Err:" + err.Error()
		return
	}
	randStr := utils.GetRandStringNoSpecialChar(28)
	fileName := randStr + ext
	fpath := uploadDir + "/" + fileName
	defer f.Close() //关闭上传文件
	err = this.SaveToFile("file", fpath)
	if err != nil {
		br.Msg = "文件上传失败"
		br.ErrMsg = "文件上传失败，Err:" + err.Error()
		return
	}
	resourceUrl := ``
	ossClient := services.NewOssClient()
	if ossClient == nil {
		br.Msg = "上传失败"
		br.ErrMsg = "初始化OSS服务失败"
		return
	}
	resourceUrl, err = ossClient.UploadFile(fileName, fpath, "")
	if err != nil {
		br.Msg = "文件上传失败"
		br.ErrMsg = "文件上传失败，Err:" + err.Error()
		return
	}

	defer func() {
		os.Remove(fpath)
	}()

	item := new(models.Resource)
	item.ResourceUrl = resourceUrl
	item.ResourceType = 1
	item.CreateTime = time.Now()
	newId, err := models.AddResource(item)
	if err != nil {
		br.Msg = "资源上传失败"
		br.ErrMsg = "资源上传失败,Err:" + err.Error()
		return
	}
	resp := models.ResourceResp{
		Id:           newId,
		ResourceUrl:  resourceUrl,
		ResourceName: uploadFileName,
	}

	br.Msg = "上传成功"
	br.Ret = 200
	br.Success = true
	br.Data = resp
	return
}

// @router /video/upload [post]
func (this *ResourceController) VideoUpload() {
	utils.FileLog.Info("start VideoUpload ")
	utils.FileLog.Info("%s:", time.Now().Format(utils.FormatDateTime))

	br := new(models.BaseResponse).Init()
	defer func() {
		this.Data["json"] = br
		this.ServeJSON()
	}()
	f, h, err := this.GetFile("file")
	if err != nil {
		br.Msg = "获取资源信息失败"
		br.ErrMsg = "获取资源信息失败,Err:" + err.Error()
		return
	}

	fileData, e := ioutil.ReadAll(f)
	if e != nil {
		br.Msg = "上传失败"
		br.ErrMsg = "读取文件失败, Err: " + e.Error()
		return
	}
	pass := filetype.IsVideo(fileData)
	if !pass {
		br.Msg = "文件格式有误"
		br.ErrMsg = "文件格式有误"
		return
	}

	ext := path.Ext(h.Filename)
	dateDir := time.Now().Format("20060102")
	uploadDir := utils.STATIC_DIR + "hongze/" + dateDir
	fmt.Println(uploadDir)
	err = os.MkdirAll(uploadDir, utils.DIR_MOD)
	if err != nil {
		br.Msg = "存储目录创建失败"
		br.ErrMsg = "存储目录创建失败,Err:" + err.Error()
		return
	}
	randStr := utils.GetRandStringNoSpecialChar(28)
	fileName := randStr + ext
	fpath := uploadDir + "/" + fileName
	defer f.Close() //关闭上传文件

	utils.FileLog.Info("start save")
	utils.FileLog.Info("%s:", time.Now().Format(utils.FormatDateTime))
	utils.FileLog.Info("fpath %s:", fpath)
	fileHandle, err := os.OpenFile(fpath, os.O_RDWR|os.O_CREATE|os.O_APPEND, os.ModeAppend)
	if err != nil {
		br.Msg = "创建文件失败"
		br.ErrMsg = "创建文件失败,Err:" + err.Error()
		return
	}
	defer fileHandle.Close()
	for {
		buffer := make([]byte, 4096)
		_, err := f.Read(buffer)
		if err != nil && err != io.EOF {
			br.Msg = "上传失败"
			br.ErrMsg = "读取文件内容失败,Err:" + err.Error()
			return
		}
		if err == io.EOF {
			break
		}

		buf := bufio.NewWriter(fileHandle)
		buf.WriteString(string(buffer))
		err = buf.Flush()
		if err != nil {
			br.Msg = "上传失败"
			br.ErrMsg = "写入文件内容失败,Err:" + err.Error()
			return
		}
	}

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

	resourceUrl := ``
	ossClient := services.NewOssClient()
	if ossClient == nil {
		br.Msg = "上传失败"
		br.ErrMsg = "初始化OSS服务失败"
		return
	}
	resourceUrl, err = ossClient.UploadFile(fileName, fpath, "")
	if err != nil {
		br.Msg = "文件上传失败"
		br.ErrMsg = "文件上传失败，Err:" + err.Error()
		return
	}

	utils.FileLog.Info("%s:", time.Now().Format(utils.FormatDateTime))
	utils.FileLog.Info("end update oss ")

	item := new(models.Resource)
	item.ResourceUrl = resourceUrl
	item.ResourceType = 3
	item.CreateTime = time.Now()
	newId, err := models.AddResource(item)
	if err != nil {
		br.Msg = "资源上传失败"
		br.ErrMsg = "资源上传失败,Err:" + err.Error()
		return
	}
	resp := new(models.ResourceResp)
	resp.Id = newId
	resp.ResourceUrl = resourceUrl

	utils.FileLog.Info("start GetMP4Duration ")
	utils.FileLog.Info("%s:", time.Now().Format(utils.FormatDateTime))

	duration, err := services.GetMP4Duration(f)
	if err != nil {
		panic(err)
	}
	utils.FileLog.Info("end GetMP4Duration ")
	utils.FileLog.Info("%s:", time.Now().Format(utils.FormatDateTime))
	resp.PlaySeconds = duration
	br.Msg = "上传成功"
	br.Ret = 200
	br.Success = true
	br.Data = resp
	utils.FileLog.Info("end VideoUpload ")
	utils.FileLog.Info("%s:", time.Now().Format(utils.FormatDateTime))
	return
}

// @router /voice/upload [post]
func (this *ResourceController) VoiceUpload() {
	utils.FileLog.Info("start VoiceUpload ")
	utils.FileLog.Info(fmt.Sprintf("%s:", time.Now().Format(utils.FormatDateTime)))

	br := new(models.BaseResponse).Init()
	defer func() {
		this.Data["json"] = br
		this.ServeJSON()
	}()
	f, h, err := this.GetFile("file")
	if err != nil {
		br.Msg = "获取资源信息失败"
		br.ErrMsg = "获取资源信息失败,Err:" + err.Error()
		return
	}
	uploadFileName := h.Filename //上传的文件名

	fileData, e := ioutil.ReadAll(f)
	if e != nil {
		br.Msg = "上传失败"
		br.ErrMsg = "读取文件失败, Err: " + e.Error()
		return
	}
	pass := filetype.IsAudio(fileData)
	if !pass {
		br.Msg = "文件格式有误"
		br.ErrMsg = "文件格式有误"
		return
	}

	ext := path.Ext(h.Filename)
	dateDir := time.Now().Format("20060102")
	uploadDir := utils.STATIC_DIR + "hongze/" + dateDir
	fmt.Println(uploadDir)
	err = os.MkdirAll(uploadDir, utils.DIR_MOD)
	if err != nil {
		br.Msg = "存储目录创建失败"
		br.ErrMsg = "存储目录创建失败,Err:" + err.Error()
		return
	}
	randStr := utils.GetRandStringNoSpecialChar(28)
	fileName := randStr + ext
	fpath := uploadDir + "/" + fileName
	defer f.Close() //关闭上传文件

	utils.FileLog.Info("start save")
	utils.FileLog.Info(fmt.Sprintf("%s:", time.Now().Format(utils.FormatDateTime)))
	utils.FileLog.Info(fmt.Sprintf("fpath %s:", fpath))
	fileHandle, err := os.OpenFile(fpath, os.O_RDWR|os.O_CREATE|os.O_APPEND, 0755)
	if err != nil {
		br.Msg = "创建文件失败"
		br.ErrMsg = "创建文件失败,Err:" + err.Error()
		return
	}
	defer fileHandle.Close()

	localFileBuffer := make([]byte, 0)
	for {
		buffer := make([]byte, 4096)
		maxLen, err := f.Read(buffer)
		if err != nil && err != io.EOF {
			br.Msg = "上传失败"
			br.ErrMsg = "读取文件内容失败,Err:" + err.Error()
			return
		}
		if err == io.EOF {
			break
		}

		buf := bufio.NewWriter(fileHandle)
		buf.WriteString(string(buffer))
		localFileBuffer = append(localFileBuffer, buffer[0:maxLen]...)
		err = buf.Flush()
		if err != nil {
			br.Msg = "上传失败"
			br.ErrMsg = "写入文件内容失败,Err:" + err.Error()
			return
		}
	}
	utils.FileLog.Info(fmt.Sprintf("%s:", time.Now().Format(utils.FormatDateTime)))
	utils.FileLog.Info("end save")

	defer func() {
		if utils.FileIsExist(fpath) {
			os.Remove(fpath)
		}
	}()

	var playSeconds float64

	if ext == ".m4a" { //m4a格式的走这个计算逻辑
		utils.FileLog.Info("start calculate m4a duration ")
		tmpPlaySeconds, err := services.GetMP4Duration(f)
		if err != nil {
			br.Msg = "音频资源时长计算失败"
			br.ErrMsg = "音频资源时长计算失败,Err:" + err.Error()
			return
		}
		playSeconds = float64(tmpPlaySeconds)
		utils.FileLog.Info("end calculate m4a duration ")
	} else {
		utils.FileLog.Info("start calculate  mp3duration ")
		playSeconds, err = mp3duration.Calculate(fpath)
		if playSeconds <= 0 {
			playSeconds, err = utils.GetVideoPlaySeconds(fpath)
			if err != nil {
				br.Msg = "音频资源时长计算失败"
				br.ErrMsg = "音频资源时长计算失败,Err:" + err.Error()
				return
			}
		}
		utils.FileLog.Info("end calculate  mp3duration ")
	}

	utils.FileLog.Info("start update oss ")
	utils.FileLog.Info(fmt.Sprintf("%s:", time.Now().Format(utils.FormatDateTime)))

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

	resourceUrl := ``
	ossClient := services.NewOssClient()
	if ossClient == nil {
		br.Msg = "上传失败"
		br.ErrMsg = "初始化OSS服务失败"
		return
	}
	resourceUrl, err = ossClient.UploadFile(fileName, fpath, "")
	if err != nil {
		br.Msg = "文件上传失败"
		br.ErrMsg = "文件上传失败，Err:" + err.Error()
		return
	}
	utils.FileLog.Info(fmt.Sprintf("%s:", time.Now().Format(utils.FormatDateTime)))
	utils.FileLog.Info("end update oss ")

	item := new(models.Resource)
	item.ResourceUrl = resourceUrl
	item.ResourceType = 3
	item.CreateTime = time.Now()
	newId, err := models.AddResource(item)
	if err != nil {
		br.Msg = "音频资源上传失败"
		br.ErrMsg = "音频资源上传失败,Err:" + err.Error()
		return
	}
	resp := models.ResourceResp{
		Id:           newId,
		ResourceUrl:  resourceUrl,
		ResourceName: uploadFileName,
		PlaySeconds:  uint32(playSeconds),
	}

	br.Msg = "上传成功"
	br.Ret = 200
	br.Success = true
	br.Data = resp
	utils.FileLog.Info("end VoiceUpload ")
	utils.FileLog.Info(fmt.Sprintf("%s:", time.Now().Format(utils.FormatDateTime)))
	return
}

// @router /upload_image_base64 [post]
func (this *ResourceController) UploadImageBase64() {
	br := new(models.BaseResponse).Init()
	defer func() {
		this.Data["json"] = br
		this.ServeJSON()
	}()

	var err error
	ext := ".png"
	uploadDir := "./static"
	randStr := utils.GetRandStringNoSpecialChar(28)
	fileName := randStr + ext
	fpath := uploadDir + "/" + fileName

	image := this.Ctx.Request.FormValue("Image")

	if image == "" {
		f, h, e := this.GetFile("Image")
		if e != nil {
			br.Msg = "参数错误"
			br.ErrMsg = "获取资源信息失败,Err:" + e.Error()
			return
		}

		fileData, e := ioutil.ReadAll(f)
		if e != nil {
			br.Msg = "上传失败"
			br.ErrMsg = "读取文件失败, Err: " + e.Error()
			return
		}
		pass := filetype.IsImage(fileData)
		if !pass {
			br.Msg = "文件格式有误"
			br.ErrMsg = "文件格式有误"
			return
		}

		ext = path.Ext(h.Filename)
		dateDir := time.Now().Format("20060102")
		uploadDir = utils.STATIC_DIR + "hongze/" + dateDir
		err = os.MkdirAll(uploadDir, utils.DIR_MOD)
		if err != nil {
			br.Msg = "存储目录创建失败"
			br.ErrMsg = "存储目录创建失败,Err:" + err.Error()
			return
		}
		randStr = utils.GetRandStringNoSpecialChar(28)
		fileName = randStr + ext
		fpath = uploadDir + "/" + fileName
		defer f.Close() //关闭上传文件
		err = this.SaveToFile("Image", fpath)
		if err != nil {
			br.Msg = "文件上传失败"
			br.ErrMsg = "文件上传失败，Err:" + err.Error()
			return
		}
	} else {
		b, _ := regexp.MatchString(`^data:\s*image\/(\w+);base64,`, image)
		if !b {
			br.Msg = "图片格式不正确"
			br.ErrMsg = "图片格式不正确"
			return
		}
		re, _ := regexp.Compile(`^data:\s*image\/(\w+);base64,`)
		base64Str := re.ReplaceAllString(image, "")
		base64Str = strings.Replace(base64Str, " ", "", -1)

		err = utils.SaveBase64ToFile(base64Str, fpath)
		if err != nil {
			br.Msg = "图片保存失败"
			br.ErrMsg = "图片保存失败,Err:" + err.Error()
			return
		}
	}

	fmt.Println("end")

	defer os.Remove(fpath)

	hzUploadDir := utils.RESOURCE_DIR + "images/"
	savePath := hzUploadDir + time.Now().Format("200601/20060102/")
	savePath += fileName

	resourceUrl := ``
	ossClient := services.NewOssClient()
	if ossClient == nil {
		br.Msg = "上传失败"
		br.ErrMsg = "初始化OSS服务失败"
		return
	}
	resourceUrl, err = ossClient.UploadFile(fileName, fpath, savePath)
	if err != nil {
		br.Msg = "文件上传失败"
		br.ErrMsg = "文件上传失败，Err:" + err.Error()
		return
	}
	utils.FileLog.Info("%s:", time.Now().Format(utils.FormatDateTime))
	utils.FileLog.Info("end update oss ")

	item := new(models.Resource)
	item.ResourceUrl = resourceUrl
	item.ResourceType = 1
	item.CreateTime = time.Now()
	newId, err := models.AddResource(item)
	if err != nil {
		br.Msg = "资源上传失败"
		br.ErrMsg = "资源上传失败,Err:" + err.Error()
		return
	}
	resp := new(models.ResourceResp)
	resp.Id = newId
	resp.ResourceUrl = resourceUrl
	br.Msg = "上传成功"
	br.Ret = 200
	br.Success = true
	br.Data = resp
	utils.FileLog.Info("end VideoUpload ")
	utils.FileLog.Info("%s:", time.Now().Format(utils.FormatDateTime))
}

func IsFileExist(filename string) bool {
	_, err := os.Stat(filename)
	if os.IsNotExist(err) {
		return false
	}
	return true
}

// @router /image/uploadV2 [post]
func (this *ResourceController) UploadV2() {
	br := new(models.BaseResponse).Init()
	defer func() {
		this.Data["json"] = br
		this.ServeJSON()
	}()
	businessType := this.Ctx.Request.Form.Get("business_type")
	fmt.Println("businessType:", businessType)
	fmt.Println(this.Ctx.Request.Form)
	fmt.Println("===========")
	br.Data = businessType

	f, h, err := this.GetFile("file")
	if err != nil {
		br.Msg = "获取资源信息失败"
		br.ErrMsg = "获取资源信息失败,Err:" + err.Error()
		return
	}

	fileData, e := ioutil.ReadAll(f)
	if e != nil {
		br.Msg = "上传失败"
		br.ErrMsg = "读取文件失败, Err: " + e.Error()
		return
	}
	pass := filetype.IsImage(fileData)
	if !pass {
		br.Msg = "文件格式有误"
		br.ErrMsg = "文件格式有误"
		return
	}

	ext := path.Ext(h.Filename)
	dateDir := time.Now().Format("20060102")
	uploadDir := utils.STATIC_DIR + "hongze/" + dateDir
	err = os.MkdirAll(uploadDir, utils.DIR_MOD)
	if err != nil {
		br.Msg = "存储目录创建失败"
		br.ErrMsg = "存储目录创建失败,Err:" + err.Error()
		return
	}
	randStr := utils.GetRandStringNoSpecialChar(28)
	fileName := randStr + ext
	fpath := uploadDir + "/" + fileName
	defer f.Close() //关闭上传文件
	err = this.SaveToFile("file", fpath)
	if err != nil {
		br.Msg = "文件上传失败"
		br.ErrMsg = "文件上传失败，Err:" + err.Error()
		return
	}
	resourceUrl := ``
	ossClient := services.NewOssClient()
	if ossClient == nil {
		br.Msg = "上传失败"
		br.ErrMsg = "初始化OSS服务失败"
		return
	}
	resourceUrl, err = ossClient.UploadFile(fileName, fpath, "")
	if err != nil {
		br.Msg = "文件上传失败"
		br.ErrMsg = "文件上传失败，Err:" + err.Error()
		return
	}

	defer func() {
		os.Remove(fpath)
	}()

	item := new(models.Resource)
	item.ResourceUrl = resourceUrl
	item.ResourceType = 1
	item.CreateTime = time.Now()
	newId, err := models.AddResource(item)
	if err != nil {
		br.Msg = "资源上传失败"
		br.ErrMsg = "资源上传失败,Err:" + err.Error()
		return
	}
	resp := new(models.ResourceResp)
	resp.Id = newId
	resp.ResourceUrl = resourceUrl

	br.Msg = "上传成功"
	br.Ret = 200
	br.Success = true
	return
}

// @router /oss/get_sts_token [get]
func (this *ResourceController) OssSTSToken() {
	br := new(models.BaseResponse).Init()
	defer func() {
		this.Data["json"] = br
		this.ServeJSON()
	}()

	ossClient := services.NewOssClient()
	if ossClient == nil {
		br.Msg = "上传失败"
		br.ErrMsg = "初始化OSS服务失败"
		return
	}
	resp, e := ossClient.GetUploadToken()
	if e != nil {
		br.Msg = "获取失败"
		br.ErrMsg = "获取OSS上传Token失败, Err: " + e.Error()
		return
	}

	br.Data = resp
	br.Msg = "获取成功"
	br.Ret = 200
	br.Success = true

}

// @router /file/download [get]
func (this *ResourceAuthController) FileDownload() {
	br := new(models.BaseResponse).Init()
	defer func() {
		if br.ErrMsg == "" {
			br.IsSendEmail = false
		}
		this.Data["json"] = br
		this.ServeJSON()
	}()
	sysUser := this.SysUser
	if sysUser == nil {
		br.Msg = "请登录"
		br.ErrMsg = "请登录，SysUser Is Empty"
		br.Ret = 408
		return
	}
	fileEncode := this.GetString("FileUrl")
	fileEncode = strings.TrimSpace(fileEncode)
	if fileEncode == "" {
		br.Msg = "参数有误"
		return
	}
	fileByte, e := base64.StdEncoding.DecodeString(fileEncode)
	if e != nil {
		br.Msg = "下载失败"
		br.ErrMsg = "文件地址解析失败, Err: " + e.Error()
		return
	}
	fileUrl := string(fileByte)
	fileArr := strings.Split(fileUrl, "/")
	if len(fileArr) == 0 {
		br.Msg = "文件地址有误"
		return
	}
	fileName := fileArr[len(fileArr)-1]

	down, e := http.Get(fileUrl)
	if e != nil {
		br.Msg = "下载失败"
		br.ErrMsg = "文件下载失败, http get: " + e.Error()
		return
	}
	defer down.Body.Close()
	if down.StatusCode != http.StatusOK {
		br.Msg = "下载失败"
		br.ErrMsg = fmt.Sprintf("文件下载失败, http status: %d", down.StatusCode)
		return
	}

	localFilePath := fmt.Sprintf("%s%s", utils.GetRandStringNoSpecialChar(6), fileName)
	localFile, e := os.Create(localFilePath)
	if e != nil {
		br.Msg = "下载失败"
		br.ErrMsg = "生成本地文件失败, Err: " + e.Error()
		return
	}
	defer func() {
		if e = localFile.Close(); e != nil {
			fmt.Println("local file close err: ", e.Error())
		}
		if e = os.Remove(localFilePath); e != nil {
			fmt.Println("local file remove err: ", e.Error())
		}
	}()

	_, e = io.Copy(localFile, down.Body)
	if e != nil {
		br.Msg = "下载失败"
		br.ErrMsg = "复制文件资源失败, Err: " + e.Error()
		return
	}

	br.Ret = 200
	br.Msg = "下载成功"
	br.Success = true
	this.Ctx.Output.Download(localFilePath, fileName)
}

// @router /file/reload [get]
func (this *ResourceController) FileReload() {

	fileUrl := `https://hzstatic.hzinsights.com/static/images/202202/20220208/Qr3dulOoVTJgyl8skH1jL1kv36RX.png`

	body, err := rdHttp.Get(fileUrl)
	if err != nil {
		fmt.Println("err: ", err)
		this.Ctx.WriteString("err: " + err.Error() + "")
		return
	}
	this.Ctx.ResponseWriter.Header().Set("Content-Type", "image/png")
	this.Ctx.ResponseWriter.Header().Set("Content-Length", strconv.Itoa(len(body)))

	_, err = this.Ctx.ResponseWriter.Write(body)
	if err != nil {
		http.Error(this.Ctx.ResponseWriter, "Failed to write response", http.StatusInternalServerError)
		return
	}
}