110 lines
3.0 KiB
Go
110 lines
3.0 KiB
Go
package app
|
||
|
||
import (
|
||
"Lee-WineList/client"
|
||
"Lee-WineList/common/constant"
|
||
"Lee-WineList/core"
|
||
"Lee-WineList/model/param"
|
||
"Lee-WineList/oauth2"
|
||
"Lee-WineList/repository"
|
||
"context"
|
||
"encoding/json"
|
||
"fmt"
|
||
"git.echol.cn/loser/logger/log"
|
||
"github.com/gin-gonic/gin"
|
||
"net/http"
|
||
"net/url"
|
||
)
|
||
|
||
type loginApi struct{}
|
||
|
||
// LoginApi 暴露接口
|
||
func LoginApi() *loginApi {
|
||
return &loginApi{}
|
||
}
|
||
|
||
// Login 登录
|
||
func (l loginApi) Login(ctx *gin.Context) {
|
||
var p param.Login
|
||
if err := ctx.ShouldBind(&p); err != nil {
|
||
core.R(ctx).FailWithMessage(err.Error())
|
||
return
|
||
}
|
||
|
||
// 获取用户基础账号信息
|
||
userId, err := oauth2.OAuthServer.UserAuthorizationHandler(ctx.Writer, ctx.Request)
|
||
if err != nil {
|
||
log.Errorf("获取用户基础账号信息失败: %v", err)
|
||
core.R(ctx).FailWithMessage(err.Error())
|
||
return
|
||
}
|
||
// 重新组装登录参数
|
||
ctx.Request.Form = url.Values{
|
||
"username": {userId},
|
||
"password": {p.Password},
|
||
"scope": {"ALL"},
|
||
"grant_type": {"password"},
|
||
}
|
||
|
||
// 参数解析成功，进行登录
|
||
if err = oauth2.OAuthServer.HandleTokenRequest(ctx.Writer, ctx.Request); err != nil {
|
||
log.Errorf("登录失败：%s", err.Error())
|
||
core.R(ctx).FailWithMessage("系统错误,登录失败")
|
||
return
|
||
}
|
||
// 登录成功才更新登录时间
|
||
if ctx.Writer.Status() == http.StatusOK {
|
||
// 登录成功，更新登录时间和IP
|
||
go repository.Login().UpdateLastLoginInfo(userId, ctx.ClientIP(), p.UserIdentity)
|
||
}
|
||
}
|
||
|
||
// Refresh 刷新登录Token
|
||
func (loginApi) Refresh(ctx *gin.Context) {
|
||
var p param.RefreshToken
|
||
if err := ctx.ShouldBind(&p); err != nil {
|
||
core.R(ctx).FailWithMessage("参数错误: " + err.Error())
|
||
return
|
||
}
|
||
|
||
// 刷新Token
|
||
if err := oauth2.OAuthServer.HandleTokenRequest(ctx.Writer, ctx.Request); err != nil {
|
||
log.Errorf("Token数据返回失败: %v", err.Error())
|
||
core.R(ctx).FailWithMessage("系统错误")
|
||
}
|
||
}
|
||
|
||
// Logout 退出登录
|
||
func (loginApi) Logout(ctx *gin.Context) {
|
||
log.Debug("退出登录啦")
|
||
r := core.R(ctx)
|
||
// Token字符串前缀
|
||
const bearerSchema string = "Bearer "
|
||
// 取出Token
|
||
tokenHeader := ctx.Request.Header.Get("Authorization")
|
||
tokenStr := tokenHeader[len(bearerSchema):]
|
||
// 取出原始RedisKey
|
||
baseDataId, err := client.Redis.Get(context.Background(), constant.OAuth2RedisKey+tokenStr).Result()
|
||
if err != nil {
|
||
r.FailWithMessage("Token信息获取失败")
|
||
return
|
||
}
|
||
baseDataStr, err := client.Redis.Get(context.Background(), constant.OAuth2RedisKey+baseDataId).Result()
|
||
if err != nil {
|
||
r.FailWithMessage("Token信息获取失败")
|
||
return
|
||
}
|
||
// 转换数据为Map
|
||
tokenData := make(map[string]any)
|
||
if err = json.Unmarshal([]byte(baseDataStr), &tokenData); err != nil {
|
||
r.FailWithMessage("系统错误: " + err.Error())
|
||
return
|
||
}
|
||
// 删除Redis缓存的数据
|
||
client.Redis.Del(context.Background(), constant.OAuth2RedisKey+baseDataId)
|
||
client.Redis.Del(context.Background(), fmt.Sprintf("%v%v", constant.OAuth2RedisKey, tokenData["Access"]))
|
||
client.Redis.Del(context.Background(), fmt.Sprintf("%v%v", constant.OAuth2RedisKey, tokenData["Refresh"]))
|
||
|
||
r.Ok()
|
||
}
|