67 lines
1.7 KiB
Go
67 lines
1.7 KiB
Go
package api
|
|
|
|
import (
|
|
"fmt"
|
|
"net/http"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
|
|
"message-nest/pkg/app"
|
|
"message-nest/pkg/e"
|
|
"message-nest/pkg/util"
|
|
"message-nest/service/auth_service"
|
|
"message-nest/service/settings_service"
|
|
"message-nest/models"
|
|
)
|
|
|
|
type auth struct {
|
|
Username string `valid:"Required; MaxSize(50)"`
|
|
Password string `valid:"Required; MaxSize(50)"`
|
|
}
|
|
|
|
type ReqAuth struct {
|
|
Username string `json:"username" validate:"required,max=50" label:"用户名"`
|
|
Password string `json:"passwd" validate:"required,max=50" label:"密码"`
|
|
}
|
|
|
|
func GetAuth(c *gin.Context) {
|
|
var (
|
|
appG = app.Gin{C: c}
|
|
req ReqAuth
|
|
)
|
|
|
|
errCode, errMsg := app.BindJsonAndPlayValid(c, &req)
|
|
if errCode != e.SUCCESS {
|
|
appG.CResponse(errCode, errMsg, nil)
|
|
return
|
|
}
|
|
|
|
authService := auth_service.Auth{Username: req.Username, Password: req.Password}
|
|
isExist, err := authService.Check()
|
|
if err != nil {
|
|
appG.CResponse(http.StatusInternalServerError, fmt.Sprintf("校验失败：%s", err), nil)
|
|
return
|
|
}
|
|
if !isExist {
|
|
appG.CResponse(http.StatusUnauthorized, "账号或密码不正确！", nil)
|
|
return
|
|
}
|
|
|
|
// 获取配置的 cookie 过期天数
|
|
expDays := settings_service.GetCookieExpDays()
|
|
token, err := util.GenerateToken(req.Username, req.Password, expDays)
|
|
if err != nil {
|
|
appG.CResponse(http.StatusInternalServerError, fmt.Sprintf("生成token失败：%s", err), nil)
|
|
return
|
|
}
|
|
|
|
// 查询用户ID并记录登录日志
|
|
if u, _ := models.GetUserByUsername(req.Username); u != nil {
|
|
_ = models.AddLoginLog(u.ID, req.Username, c.ClientIP(), c.GetHeader("User-Agent"))
|
|
}
|
|
|
|
appG.CResponse(http.StatusOK, "登录成功!", map[string]string{
|
|
"token": token,
|
|
})
|
|
}
|