package middleware

import (
	"fmt"
	"github.com/gin-gonic/gin"
	"kefu/tools"
	"time"
)

func JwtPageMiddleware(c *gin.Context) {
	//暂时不处理
	//token := c.Query("token")
	//userinfo := tools.ParseToken(token)
	//if userinfo == nil {
	//	c.Redirect(302,"/login")
	//	c.Abort()
	//}
}
func JwtApiMiddleware(c *gin.Context) {
	token := c.GetHeader("token")
	if token == "" {
		token = c.Query("token")
	}
	userinfo := tools.ParseToken(token)
	if userinfo == nil || userinfo["name"] == nil || userinfo["create_time"] == nil {
		c.JSON(200, gin.H{
			"code": 400,
			"msg":  "验证失败,请重新登录",
		})
		c.Abort()
		return
	}
	createTime := int64(userinfo["create_time"].(float64))
	var expire int64 = 24 * 60 * 60
	nowTime := time.Now().Unix()
	if (nowTime - createTime) >= expire {
		c.JSON(200, gin.H{
			"code": 401,
			"msg":  "token失效,请重新登录",
		})
		c.Abort()
	}
	c.Set("user", userinfo["name"])
	c.Set("kefu_id", userinfo["kefu_id"])
	c.Set("kefu_name", userinfo["name"])
	c.Set("role_id", userinfo["role_id"])
	c.Set("pid", userinfo["pid"])
	pid := userinfo["pid"].(float64)
	kefuId := userinfo["kefu_id"].(float64)
	entId := ""
	if userinfo["role_id"].(float64) == 3 {
		entId = fmt.Sprintf("%v", pid)
	} else {
		entId = fmt.Sprintf("%v", kefuId)
	}
	c.Set("ent_id", entId)
}
func JwtAdminMiddleware(c *gin.Context) {
	roleId, _ := c.Get("role_id")
	if roleId.(float64) != 1 {
		c.JSON(200, gin.H{
			"code": 401,
			"msg":  "权限不足，必须为管理员权限",
		})
		c.Abort()
	}
}
func JwtEntMiddleware(c *gin.Context) {
	roleId, _ := c.Get("role_id")
	if roleId.(float64) == 3 {
		c.JSON(200, gin.H{
			"code": 401,
			"msg":  "权限不足，当前账号为坐席权限",
		})
		c.Abort()
	}
}