forked from lxh/go-wxhelper
121 lines
3.3 KiB
Go
121 lines
3.3 KiB
Go
package login
|
||
|
||
import (
|
||
"context"
|
||
"encoding/json"
|
||
"gitee.ltd/lxh/logger/log"
|
||
"github.com/gin-gonic/gin"
|
||
"net/http"
|
||
"net/url"
|
||
"wechat-robot/internal/redis"
|
||
"wechat-robot/model/param/login"
|
||
"wechat-robot/pkg/auth"
|
||
"wechat-robot/pkg/captcha"
|
||
"wechat-robot/pkg/response"
|
||
userService "wechat-robot/service/adminuser"
|
||
)
|
||
|
||
// Login
|
||
// @description: 登录
|
||
// @param ctx
|
||
// @return err
|
||
func Login(ctx *gin.Context) {
|
||
log.Debugf("收到登录请求")
|
||
var p login.GetTokenWithPassword
|
||
if err := ctx.ShouldBind(&p); err != nil {
|
||
response.New(ctx).SetMsg("参数错误").SetError(err).Fail()
|
||
return
|
||
}
|
||
|
||
// 验证验证码是否正确
|
||
if !new(captcha.RedisStore).Verify(p.VerifyId, p.VerifyCode, true) {
|
||
response.New(ctx).SetMsg("验证码错误").Fail()
|
||
return
|
||
}
|
||
|
||
// 重写参数
|
||
ctx.Request.Form = url.Values{
|
||
"username": {p.Username},
|
||
"password": {p.Password},
|
||
"scope": {"ALL"},
|
||
"grant_type": {"password"},
|
||
}
|
||
// 参数解析成功，进行登录
|
||
if err := auth.OAuthServer.HandleTokenRequest(ctx.Writer, ctx.Request); err != nil {
|
||
log.Errorf("登录失败：%s", err.Error())
|
||
response.New(ctx).SetMsg("系统错误").SetError(err).Fail()
|
||
return
|
||
}
|
||
|
||
if ctx.Writer.Status() == http.StatusOK {
|
||
go userService.UpdateLastLoginInfo(p.Username, ctx.ClientIP())
|
||
}
|
||
}
|
||
|
||
// Refresh
|
||
// @description: 刷新Token
|
||
// @param ctx
|
||
func Refresh(ctx *gin.Context) {
|
||
var p login.RefreshToken
|
||
if err := ctx.ShouldBind(&p); err != nil {
|
||
response.New(ctx).SetMsg("参数错误").SetError(err).Fail()
|
||
return
|
||
}
|
||
|
||
// 取出用户Id
|
||
userId := auth.GetUserIdWithRefreshToken(p.RefreshToken)
|
||
|
||
// 重写参数
|
||
ctx.Request.Form = url.Values{
|
||
"refresh_token": {p.RefreshToken},
|
||
"grant_type": {"refresh_token"},
|
||
}
|
||
|
||
// 刷新Token
|
||
if err := auth.OAuthServer.HandleTokenRequest(ctx.Writer, ctx.Request); err != nil {
|
||
log.Errorf("Token数据返回失败: %v", err.Error())
|
||
response.New(ctx).SetMsg("系统错误").Fail()
|
||
}
|
||
|
||
// 登录成功才更新登录时间
|
||
if ctx.Writer.Status() == http.StatusOK {
|
||
// 登录成功，更新登录时间和IP
|
||
go userService.UpdateLastLoginInfo(userId, ctx.ClientIP())
|
||
}
|
||
}
|
||
|
||
// Logout
|
||
// @description: 退出登录
|
||
// @param ctx
|
||
func Logout(ctx *gin.Context) {
|
||
log.Debug("退出登录啦")
|
||
// Token字符串前缀
|
||
const bearerSchema string = "Bearer "
|
||
// 取出Token
|
||
tokenHeader := ctx.GetHeader("Authorization")
|
||
tokenStr := tokenHeader[len(bearerSchema):]
|
||
// 取出原始RedisKey
|
||
baseDataId, err := redis.Client.Get(context.Background(), "oauth:token:"+tokenStr).Result()
|
||
if err != nil {
|
||
response.New(ctx).SetMsg("Token信息获取失败").Fail()
|
||
return
|
||
}
|
||
baseDataStr, err := redis.Client.Get(context.Background(), "oauth:token:"+baseDataId).Result()
|
||
if err != nil {
|
||
response.New(ctx).SetMsg("Token信息获取失败").Fail()
|
||
return
|
||
}
|
||
// 转换数据为Map
|
||
tokenData := make(map[string]interface{})
|
||
if err = json.Unmarshal([]byte(baseDataStr), &tokenData); err != nil {
|
||
response.New(ctx).SetMsg("系统错误").SetError(err).Fail()
|
||
return
|
||
}
|
||
// 删除Redis缓存的数据
|
||
redis.Client.Del(context.Background(), "oauth:token:"+baseDataId)
|
||
redis.Client.Del(context.Background(), "oauth:token:"+tokenData["Access"].(string))
|
||
redis.Client.Del(context.Background(), "oauth:token:"+tokenData["Refresh"].(string))
|
||
|
||
response.New(ctx).Success()
|
||
}
|