package controllers
import (
"encoding/json"
"eta/eta_mini_bridge/models"
"eta/eta_mini_bridge/utils"
"fmt"
"net/http"
"net/url"
"github.com/beego/beego/v2/server/web"
)
type BaseAuthController struct {
web.Controller
Token string
}
func (this *BaseAuthController) Prepare() {
method := this.Ctx.Input.Method()
uri := this.Ctx.Input.URI()
fmt.Println("Url:", uri)
header := this.Ctx.Request.Header
headerBody, err := json.Marshal(header)
fmt.Println(err)
fmt.Println("header:" + string(headerBody))
if method != "HEAD" {
//校验签名
nonce := this.Ctx.Input.Header("Nonce")
timestamp := this.Ctx.Input.Header("Timestamp")
appid := this.Ctx.Input.Header("Appid")
signature := this.Ctx.Input.Header("Signature")
if nonce == "" {
errMsg := "随机字符串不能为空"
this.JSON(models.BaseResponse{Ret: 400, Msg: "", ErrMsg: errMsg}, false, false)
this.StopRun()
return
}
if timestamp == "" {
errMsg := "时间戳不能为空"
this.JSON(models.BaseResponse{Ret: 400, Msg: "", ErrMsg: errMsg}, false, false)
this.StopRun()
return
}
checkSign := utils.GetSign(nonce, timestamp, appid, utils.ETA_MINI_API_SECRET)
if signature != checkSign {
errMsg := "签名错误"
this.JSON(models.BaseResponse{Ret: 401, Msg: "", ErrMsg: errMsg}, false, false)
this.StopRun()
return
}
if method != "GET" && method != "POST" {
errMsg := "无效的请求方式"
this.JSON(models.BaseResponse{Ret: 501, Msg: "", ErrMsg: errMsg}, false, false)
this.StopRun()
return
}
} else {
this.JSON(models.BaseResponse{Ret: 500, Msg: "系统异常,请联系客服!", ErrMsg: "method:" + method}, false, false)
this.StopRun()
return
}
}
func (c *BaseAuthController) ServeJSON(encoding ...bool) {
//所有请求都做这么个处理吧,目前这边都是做编辑、刷新逻辑处理(新增的话,并没有指标id,不会有影响)
var (
hasIndent = false
hasEncoding = false
)
if web.BConfig.RunMode == web.PROD {
hasIndent = false
}
if len(encoding) > 0 && encoding[0] == true {
hasEncoding = true
}
if c.Data["json"] == nil {
//go utils.SendEmail("异常提醒:", "接口:"+"URI:"+c.Ctx.Input.URI()+";无返回值", utils.EmailSendToUsers)
//body := "接口:" + "URI:" + c.Ctx.Input.URI() + ";无返回值"
//go alarm_msg.SendAlarmMsg(body, 1)
return
}
baseRes := c.Data["json"].(*models.BaseResponse)
if baseRes != nil && baseRes.Ret != 408 {
//body, _ := json.Marshal(baseRes)
//var requestBody string
//method := c.Ctx.Input.Method()
//if method == "GET" {
// requestBody = c.Ctx.Request.RequestURI
//} else {
// requestBody, _ = url.QueryUnescape(string(c.Ctx.Input.RequestBody))
//}
//if baseRes.Ret != 200 && baseRes.IsSendEmail {
// go utils.SendEmail(utils.APP_NAME_CN+"【"+utils.RunMode+"】"+"失败提醒", "URI:"+c.Ctx.Input.URI()+"
"+"Params"+requestBody+"
"+"ErrMsg:"+baseRes.ErrMsg+";
Msg:"+baseRes.Msg+";
Body:"+string(body)+"
", utils.EmailSendToUsers)
//}
// 记录错误日志, 并清掉错误信息避免暴露给外部
if baseRes.ErrMsg != "" {
utils.FileLog.Info(baseRes.ErrMsg)
baseRes.ErrMsg = ""
}
}
c.JSON(c.Data["json"], hasIndent, hasEncoding)
}
func (c *BaseAuthController) JSON(data interface{}, hasIndent bool, coding bool) error {
c.Ctx.Output.Header("Content-Type", "application/json; charset=utf-8")
var content []byte
var err error
if hasIndent {
content, err = json.MarshalIndent(data, "", " ")
} else {
content, err = json.Marshal(data)
}
if err != nil {
http.Error(c.Ctx.Output.Context.ResponseWriter, err.Error(), http.StatusInternalServerError)
return err
}
ip := c.Ctx.Input.IP()
requestBody, err := url.QueryUnescape(string(c.Ctx.Input.RequestBody))
if err != nil {
requestBody = string(c.Ctx.Input.RequestBody)
}
if requestBody == "" {
requestBody = c.Ctx.Input.URI()
}
c.logUri(data, requestBody, ip)
if coding {
content = []byte(utils.StringsToJSON(string(content)))
}
return c.Ctx.Output.Body(content)
}
func (c *BaseAuthController) logUri(respContent interface{}, requestBody, ip string) {
utils.ApiLog.Info("uri:%s, requestBody:%s, responseBody:%s, ip:%s", c.Ctx.Input.URI(), requestBody, respContent, ip)
return
}