hongze_yb/controller/public.go

package controller

import (
	"fmt"
	"github.com/gin-gonic/gin"
	"hongze/hongze_yb/controller/response"
	"hongze/hongze_yb/global"
	"hongze/hongze_yb/logic"
	"hongze/hongze_yb/models/request"
	"hongze/hongze_yb/models/tables/yb_suncode_pars"
	"hongze/hongze_yb/services"
	"hongze/hongze_yb/services/alarm_msg"
	"hongze/hongze_yb/utils"
	"os"
	"path"
	"time"
)

// GetApplyVarietyList
// @Tags 公共模块
// @Summary  获取所有可以申请的品种权限列表
// @Description 获取所有可以申请的品种权限列表
// @Security ApiKeyAuth
// @securityDefinitions.basic BasicAuth
// @Param Authorization	header string true "微信登录后获取到的token"
// @Accept  json
// @Product json
// @Success 200 {object} []logic.ApplyVariety "获取成功"
// @failure 400 {string} string "获取失败"
// @Router /public/get_apply_variety_list [get]
func GetApplyVarietyList(c *gin.Context) {
	list := logic.GetApplyVarietyList()
	response.OkData("获取成功", list, c)
}

// Upload
// @Tags 公共模块
// @Summary  文件上传
// @Description 文件上传
// @Security ApiKeyAuth
// @securityDefinitions.basic BasicAuth
// @Param Authorization	header string true "微信登录后获取到的token"
// @Param file formData file false "操作描述"
// @Accept  multipart/form-data
// @Product json
// @Success 200 {object} string "上传成功"
// @failure 400 {string} string "上传失败，存储目录创建失败"
// @Router /public/upload [post]
func Upload(c *gin.Context) {
	// 单文件
	file, err := c.FormFile("file")
	fmt.Println("file", file)
	if err != nil {
		response.FailData("获取资源失败", "获取资源失败，Err:"+err.Error(), c)
		return
	}

	ext := path.Ext(file.Filename)
	dateDir := time.Now().Format("20060102")
	uploadDir := global.CONFIG.Serve.StaticDir + "hongze/" + dateDir
	err = os.MkdirAll(uploadDir, 0766)
	if err != nil {
		response.FailData("存储目录创建失败", "存储目录创建失败，Err:"+err.Error(), c)
		return
	}
	randStr := utils.GetRandStringNoSpecialChar(28)
	fileName := randStr + ext
	fpath := uploadDir + "/" + fileName

	// 上传文件至指定目录
	err = c.SaveUploadedFile(file, fpath)
	if err != nil {
		response.FailData("文件上传失败", "文件上传失败，Err:"+err.Error(), c)
		return
	}
	defer func() {
		os.Remove(fpath)
	}()

	//上传到阿里云
	resourceUrl, err := services.UploadAliyun(fileName, fpath)
	if err != nil {
		response.FailData("文件上传失败", "文件上传失败，Err:"+err.Error(), c)
		return
	}
	response.OkData("文件上传成功", resourceUrl, c)

}

// GetSharePoster
// @Tags 公共模块
// @Summary  获取分享海报
// @Description 获取分享海报
// @Security ApiKeyAuth
// @securityDefinitions.basic BasicAuth
// @Param	request	body services.SharePosterReq true "type json string"
// @Accept  application/json
// @Success 200 {object} string "获取成功"
// @failure 400 {string} string "获取失败"
// @Router /public/get_share_poster [post]
func GetSharePoster(c *gin.Context) {
	var req services.SharePosterReq
	if c.ShouldBind(&req) != nil {
		response.Fail("参数异常", c)
		return
	}
	if req.Source == "" {
		response.Fail("来源有误", c)
		return
	}
	imgUrl, err := services.CreatePosterFromSourceV2(req.CodePage, req.CodeScene, req.Source, req.Version, req.Pars)
	if err != nil {
		response.FailData("获取分享海报失败", "获取分享海报失败, Err: "+err.Error(), c)
		return
	}
	response.OkData("获取成功", imgUrl, c)
}

// GetSuncodeScene 获取小程序太阳码scene值
// @Tags 公共模块
// @Summary  获取小程序太阳码scene值
// @Description 获取小程序太阳码scene值
// @Security ApiKeyAuth
// @Param Authorization	header string true "Bearer 31a165baebe6dec616b1f8f3207b4273"
// @Param scene_key query string true "scene_key值"
// @Success 200 {string} string "获取成功"
// @failure 400 {string} string "获取失败"
// @Router /public/get_suncode_scene [get]
func GetSuncodeScene(c *gin.Context) {
	reqKey := c.DefaultQuery("scene_key", "")
	if reqKey == "" {
		response.Fail("参数有误", c)
	}
	pars, err := yb_suncode_pars.GetSceneByKey(reqKey)
	if err != nil && err != utils.ErrNoRow {
		response.FailMsg("获取失败", "GetSuncodeScene获取失败, Err: "+err.Error(), c)
		return
	}
	scene := ""
	if pars != nil {
		scene = pars.Scene
	}
	response.OkData("获取成功", scene, c)
}

// GetVarietyTagTree 标签树
// @Tags 公共模块
// @Summary  标签树
// @Description 标签树
// @Security ApiKeyAuth
// @Param Authorization	header string true "Bearer 31a165baebe6dec616b1f8f3207b4273"
// @Param scene_key query string true "scene_key值"
// @Success 200 {string} string "获取成功"
// @failure 400 {string} string "获取失败"
// @Router /public/get_variety_tag_tree [get]
func GetVarietyTagTree(c *gin.Context) {
	list, err := services.GetTagTree()
	if err != nil {
		response.FailMsg("获取标签树失败", "获取标签树失败, Err: "+err.Error(), c)
		return
	}
	response.OkData("获取成功", list, c)
}

// WechatWarning 小程序前端预警提示
// @Tags 公共模块
// @Description 小程序前端预警提示
// @Param content	query  string  true  "预警信息"
// @Success 200 {string} string "操作成功"
// @failure 400 {string} string "操作失败"
// @Router /public/wechat_warning [post]
func WechatWarning(c *gin.Context) {
	var req request.WechatWarningReq
	if err := c.Bind(&req); err != nil {
		response.Fail("参数有误", c)
		return
	}
	if req.Content != "" {
		tips := fmt.Sprintf("研报小程序前端报错预警-ErrMsg: %s", req.Content)
		global.LOG.Warning(tips)
		go alarm_msg.SendAlarmMsg(tips, 2)
	}
	response.Ok("操作成功", c)
}