package middleware

import (
	"github.com/gin-gonic/gin"
	"github.com/sirupsen/logrus"
	"go-admin/models"
	"go-admin/tools"
	"strings"
	"time"
)

//实例化
var logger = logrus.New()

// 日志记录到文件
func LoggerToFile() gin.HandlerFunc {

	return func(c *gin.Context) {
		// 开始时间
		startTime := time.Now()

		// 处理请求
		c.Next()

		// 结束时间
		endTime := time.Now()

		// 执行时间
		latencyTime := endTime.Sub(startTime)

		// 请求方式
		reqMethod := c.Request.Method

		// 请求路由
		reqUri := c.Request.RequestURI

		// 状态码
		statusCode := c.Writer.Status()

		// 请求IP
		clientIP := c.ClientIP()

		// 日志格式
		logger.Infof("%s [%s] %3d %13v %15s %s %s \r\n",
			startTime.Format("2006-01-02 15:04:05.9999"),
			strings.ToUpper(logger.Level.String()),
			statusCode,
			latencyTime,
			clientIP,
			reqMethod,
			reqUri,
		)

		if c.Request.Method != "GET" && c.Request.Method != "OPTIONS" {
			menu := models.Menu{}
			menu.Path = reqUri
			menu.Action = reqMethod
			menuList, _ := menu.Get()
			sysOperLog := models.SysOperLog{}
			sysOperLog.OperIp = clientIP
			sysOperLog.OperLocation = tools.GetLocation(clientIP)
			sysOperLog.Status = tools.IntToString(statusCode)
			sysOperLog.OperName = tools.GetUserName(c)
			sysOperLog.RequestMethod = c.Request.Method
			sysOperLog.OperUrl = reqUri
			if reqUri == "/login" {
				sysOperLog.BusinessType = "10"
				sysOperLog.Title = "用户登录"
				sysOperLog.OperName = "-"
			} else if strings.Contains(reqUri, "/api/v1/logout") {
				sysOperLog.BusinessType = "11"
			} else if strings.Contains(reqUri, "/api/v1/getCaptcha") {
				sysOperLog.BusinessType = "12"
				sysOperLog.Title = "验证码"
			} else {
				if reqMethod == "POST" {
					sysOperLog.BusinessType = "1"
				} else if reqMethod == "PUT" {
					sysOperLog.BusinessType = "2"
				} else if reqMethod == "DELETE" {
					sysOperLog.BusinessType = "3"
				}
			}
			sysOperLog.Method = reqMethod
			if len(menuList) > 0 {
				sysOperLog.Title = menuList[0].Title
			}
			b, _ := c.Get("body")
			sysOperLog.OperParam, _ = tools.StructToJsonStr(b)
			sysOperLog.CreateBy = tools.GetUserName(c)
			sysOperLog.OperTime = tools.GetCurrntTime()
			sysOperLog.LatencyTime = (latencyTime).String()
			sysOperLog.UserAgent = c.Request.UserAgent()
			if c.Err() == nil {
				sysOperLog.Status = "0"
			} else {
				sysOperLog.Status = "1"
			}
			_, _ = sysOperLog.Create()
		}
	}
}