eta_chart_lib/controllers/base_auth.go

package controllers

import (
	"encoding/json"
	"eta/eta_chart_lib/models"
	"eta/eta_chart_lib/utils"
	"fmt"
	"github.com/beego/beego/v2/server/web"
	"github.com/sirupsen/logrus"
	"net/http"
	"net/url"
	"strings"
)

type BaseAuthController struct {
	web.Controller
}

func (c *BaseAuthController) Prepare() {
	fmt.Println("enter prepare")
	method := c.Ctx.Input.Method()
	uri := c.Ctx.Input.URI()
	ip := c.Ctx.Input.IP()
	fmt.Println("Url:", uri+";ip:"+ip)
	if method != "HEAD" {
		if method == "POST" || method == "GET" {
			//authorization := this.Ctx.Input.Header("authorization")
			//if authorization == "" {
			//	authorization = this.Ctx.Input.Header("Authorization")
			//}
			//if authorization == "" {
			//	newAuthorization := this.GetString("authorization")
			//	account := this.GetString("account")
			//	authorization = "authorization=" + newAuthorization + ";account=" + account
			//	if newAuthorization == "" {
			//		newAuthorization = this.GetString("Authorization")
			//		if account == "" {
			//			account = this.GetString("Account")
			//		}
			//		authorization = "authorization=" + newAuthorization + ";account=" + account
			//	}
			//}
			//fmt.Println("authorization##################")
			//fmt.Println(authorization)
			//fmt.Println("authorization############")
			//if authorization == "" {
			//	strArr := strings.Split(uri, "?")
			//	for k, v := range strArr {
			//		fmt.Println(k, v)
			//	}
			//	if len(strArr) > 1 {
			//		authorization := strArr[1]
			//		authorization = strings.Replace(authorization, "Authorization", "authorization", -1)
			//		fmt.Println(authorization)
			//	}
			//}
			//fmt.Println("authorization------------")
			//fmt.Println(authorization)
			//fmt.Println("authorization-------------")
			//if authorization == "" {
			//	this.JSON(models.BaseResponse{Ret: 408, Msg: "请重新授权！", ErrMsg: "请重新授权:Token is empty or account is empty"}, false, false)
			//	this.StopRun()
			//	return
			//}
			//authorizationArr := strings.Split(authorization, ";")
			//tokenStr := authorizationArr[0]
			//tokenArr := strings.Split(tokenStr, "=")
			//token := tokenArr[1]
			//
			//accountStr := authorizationArr[1]
			//accountArr := strings.Split(accountStr, "=")
			//account := accountArr[1]
			//
			//fmt.Println("token:", token)
			//fmt.Println("account:", account)
			////校验token是否合法
			//// JWT校验Token和Account
			//if !utils.CheckToken(account, token) {
			//	fmt.Println("CheckToken Err")
			//	this.JSON(models.BaseResponse{Ret: 408, Msg: "鉴权失败，请重新登录！", ErrMsg: "登录失效，请重新登陆!,CheckToken Fail"}, false, false)
			//	this.StopRun()
			//	return
			//}
			//
			//fmt.Println("GetUserByToken")
			//session, err := system.GetSysSessionByToken(token)
			//if err != nil {
			//	if err.Error() == utils.ErrNoRow() {
			//		this.JSON(models.BaseResponse{Ret: 408, Msg: "信息已变更,请重新登陆！", ErrMsg: "Token 信息已变更:Token: " + token}, false, false)
			//		this.StopRun()
			//		return
			//	}
			//	this.JSON(models.BaseResponse{Ret: 408, Msg: "网络异常，请稍后重试!", ErrMsg: "获取用户信息异常，Eerr:" + err.Error()}, false, false)
			//	this.StopRun()
			//	return
			//}
			//if session == nil {
			//	this.JSON(models.BaseResponse{Ret: 408, Msg: "网络异常，请稍后重试!", ErrMsg: "sesson is empty "}, false, false)
			//	this.StopRun()
			//	return
			//}
			//if time.Now().After(session.ExpiredTime) {
			//	this.JSON(models.BaseResponse{Ret: 408, Msg: "请重新登录!", ErrMsg: "获取用户信息异常，Eerr:" + err.Error()}, false, false)
			//	this.StopRun()
			//	return
			//}
			//admin, err := system.GetSysUserById(session.SysUserId)
			//if err != nil {
			//	if err.Error() == utils.ErrNoRow() {
			//		this.JSON(models.BaseResponse{Ret: 408, Msg: "信息已变更,请重新登陆！", ErrMsg: "获取admin 信息失败 " + strconv.Itoa(session.SysUserId)}, false, false)
			//		this.StopRun()
			//		return
			//	}
			//	this.JSON(models.BaseResponse{Ret: 408, Msg: "网络异常，请稍后重试!", ErrMsg: "获取admin信息异常，Eerr:" + err.Error()}, false, false)
			//	this.StopRun()
			//	return
			//}
			//if admin == nil {
			//	this.JSON(models.BaseResponse{Ret: 408, Msg: "网络异常，请稍后重试!", ErrMsg: "admin is empty "}, false, false)
			//	this.StopRun()
			//	return
			//}
			//admin.RoleTypeCode = GetSysUserRoleTypeCode(admin.RoleTypeCode)
			//this.SysUser = admin
		} else {
			c.JSON(models.BaseResponse{Ret: 408, Msg: "请求异常，请联系客服!", ErrMsg: "POST之外的请求，暂不支持"}, false, false)
			c.StopRun()
			return
		}
	}
}

func (c *BaseAuthController) ServeJSON(encoding ...bool) {
	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)
		return
	}
	baseRes := c.Data["json"].(*models.BaseResponse)
	if baseRes != nil && baseRes.Ret != 200 && baseRes.Ret != 408 && baseRes.IsSendEmail {
		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))
		}
		go utils.SendEmail(utils.APPNAME+"【"+utils.RunMode+"】"+"失败提醒", "URI:"+c.Ctx.Input.URI()+"<br/> "+"Params"+requestBody+" <br/>"+"ErrMsg:"+baseRes.ErrMsg+";<br/>Msg:"+baseRes.Msg+";<br/> Body:"+string(body)+"<br/>", utils.EmailSendToUsers)
	}
	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(data interface{}, requestBody, ip string) {
	authorization := ""
	method := c.Ctx.Input.Method()
	uri := c.Ctx.Input.URI()
	fmt.Println("Url:", uri)
	if method != "HEAD" {
		if method == "POST" || method == "GET" {
			authorization = c.Ctx.Input.Header("authorization")
			if authorization == "" {
				authorization = c.Ctx.Input.Header("Authorization")
			}
			if authorization == "" {
				newAuthorization := c.GetString("authorization")
				if newAuthorization != `` {
					authorization = "authorization=" + newAuthorization
				} else {
					newAuthorization = c.GetString("Authorization")
					authorization = "authorization=" + newAuthorization
				}
			} else {
				if strings.Contains(authorization, ";") {
					authorization = strings.Replace(authorization, ";", "$", 1)
				}
			}
			if authorization == "" {
				strArr := strings.Split(uri, "?")
				for k, v := range strArr {
					fmt.Println(k, v)
				}
				if len(strArr) > 1 {
					authorization = strArr[1]
					authorization = strings.Replace(authorization, "Authorization", "authorization", -1)
					fmt.Println(authorization)
				}
			}
		}
	}

	var reqData interface{}
	err := json.Unmarshal([]byte(requestBody), &reqData)
	if err != nil {
		utils.ApiLog.WithFields(logrus.Fields{
			"uri":           c.Ctx.Input.URI(),
			"authorization": authorization,
			"requestBody":   requestBody,
			"responseBody":  data,
			"ip":            ip,
		}).Info("请求详情")
	} else {
		utils.ApiLog.WithFields(logrus.Fields{
			"uri":           c.Ctx.Input.URI(),
			"authorization": authorization,
			"requestBody":   reqData,
			"responseBody":  data,
			"ip":            ip,
		}).Info("请求详情")
	}
	return
}