78 lines
1.7 KiB
Go
78 lines
1.7 KiB
Go
|
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()
|
||
|
|
}
|
||
|
|
}
|