mirror of
https://github.com/tiger1103/gfast.git
synced 2026-10-01 13:21:58 +00:00
105 lines
2.9 KiB
Go
105 lines
2.9 KiB
Go
package controller
|
|
|
|
import (
|
|
"context"
|
|
"github.com/gogf/gf/v2/crypto/gmd5"
|
|
"github.com/gogf/gf/v2/errors/gerror"
|
|
"github.com/gogf/gf/v2/frame/g"
|
|
"github.com/gogf/gf/v2/os/genv"
|
|
"github.com/gogf/gf/v2/util/gconv"
|
|
"github.com/tiger1103/gfast/v3/api/v1/system"
|
|
commonService "github.com/tiger1103/gfast/v3/internal/app/common/service"
|
|
"github.com/tiger1103/gfast/v3/internal/app/system/model"
|
|
"github.com/tiger1103/gfast/v3/internal/app/system/service"
|
|
"github.com/tiger1103/gfast/v3/library/libUtils"
|
|
)
|
|
|
|
var (
|
|
User = UserController{}
|
|
)
|
|
|
|
type UserController struct {
|
|
baseController
|
|
}
|
|
|
|
func (c *UserController) Login(ctx context.Context, req *system.UserLoginReq) (res *system.UserLoginRes, err error) {
|
|
var (
|
|
user *model.LoginUserRes
|
|
token string
|
|
permissions []string
|
|
menuList []*model.UserMenus
|
|
)
|
|
//判断验证码是否正确
|
|
debug := genv.GetWithCmd("gf.debug")
|
|
if debug.Int() != 1 {
|
|
if !commonService.Captcha().VerifyString(req.VerifyKey, req.VerifyCode) {
|
|
err = gerror.New("验证码输入错误")
|
|
return
|
|
}
|
|
}
|
|
ip := libUtils.GetClientIp(ctx)
|
|
userAgent := libUtils.GetUserAgent(ctx)
|
|
user, err = service.User().GetAdminUserByUsernamePassword(ctx, req)
|
|
if err != nil {
|
|
// 保存登录失败的日志信息
|
|
service.SysLoginLog().Invoke(ctx, &model.LoginLogParams{
|
|
Status: 0,
|
|
Username: req.Username,
|
|
Ip: ip,
|
|
UserAgent: userAgent,
|
|
Msg: err.Error(),
|
|
Module: "系统后台",
|
|
})
|
|
return
|
|
}
|
|
err = service.User().UpdateLoginInfo(ctx, user.Id, ip)
|
|
if err != nil {
|
|
return
|
|
}
|
|
// 报存登录成功的日志信息
|
|
service.SysLoginLog().Invoke(ctx, &model.LoginLogParams{
|
|
Status: 1,
|
|
Username: req.Username,
|
|
Ip: ip,
|
|
UserAgent: userAgent,
|
|
Msg: "登录成功",
|
|
Module: "系统后台",
|
|
})
|
|
key := gconv.String(user.Id) + "-" + gmd5.MustEncryptString(user.UserName) + gmd5.MustEncryptString(user.UserPassword)
|
|
if g.Cfg().MustGet(ctx, "gfToken.multiLogin").Bool() {
|
|
key = gconv.String(user.Id) + "-" + gmd5.MustEncryptString(user.UserName) + gmd5.MustEncryptString(user.UserPassword+ip+userAgent)
|
|
}
|
|
user.UserPassword = ""
|
|
token, err = service.GfToken(ctx).GenerateToken(ctx, key, user)
|
|
if err != nil {
|
|
return
|
|
}
|
|
//获取用户菜单数据
|
|
menuList, permissions, err = service.User().GetAdminRules(ctx, user.Id)
|
|
if err != nil {
|
|
return
|
|
}
|
|
res = &system.UserLoginRes{
|
|
UserInfo: user,
|
|
Token: token,
|
|
MenuList: menuList,
|
|
Permissions: permissions,
|
|
}
|
|
return
|
|
}
|
|
|
|
// GetUserMenus 获取用户菜单及按钮权限
|
|
func (c *UserController) GetUserMenus(ctx context.Context, req *system.UserMenusReq) (res *system.UserMenusRes, err error) {
|
|
var (
|
|
permissions []string
|
|
menuList []*model.UserMenus
|
|
)
|
|
userId := service.Context().GetUserId(ctx)
|
|
menuList, permissions, err = service.User().GetAdminRules(ctx, userId)
|
|
res = &system.UserMenusRes{
|
|
MenuList: menuList,
|
|
Permissions: permissions,
|
|
}
|
|
return
|
|
}
|