mirror of
https://github.com/1Panel-dev/1Panel.git
synced 2024-12-15 17:29:00 +08:00
169 lines
4.9 KiB
Go
169 lines
4.9 KiB
Go
package v1
|
|
|
|
import (
|
|
"encoding/base64"
|
|
|
|
"github.com/1Panel-dev/1Panel/backend/app/api/v1/helper"
|
|
"github.com/1Panel-dev/1Panel/backend/app/dto"
|
|
"github.com/1Panel-dev/1Panel/backend/app/model"
|
|
"github.com/1Panel-dev/1Panel/backend/constant"
|
|
"github.com/1Panel-dev/1Panel/backend/global"
|
|
"github.com/1Panel-dev/1Panel/backend/utils/captcha"
|
|
"github.com/1Panel-dev/1Panel/backend/utils/qqwry"
|
|
"github.com/gin-gonic/gin"
|
|
)
|
|
|
|
type BaseApi struct{}
|
|
|
|
// @Tags Auth
|
|
// @Summary User login
|
|
// @Description 用户登录
|
|
// @Accept json
|
|
// @Param EntranceCode header string true "安全入口 base64 加密串"
|
|
// @Param request body dto.Login true "request"
|
|
// @Success 200 {object} dto.UserLoginInfo
|
|
// @Router /auth/login [post]
|
|
func (b *BaseApi) Login(c *gin.Context) {
|
|
var req dto.Login
|
|
if err := c.ShouldBindJSON(&req); err != nil {
|
|
helper.ErrorWithDetail(c, constant.CodeErrBadRequest, constant.ErrTypeInvalidParams, err)
|
|
return
|
|
}
|
|
if req.AuthMethod != "jwt" && !req.IgnoreCaptcha {
|
|
if err := captcha.VerifyCode(req.CaptchaID, req.Captcha); err != nil {
|
|
helper.ErrorWithDetail(c, constant.CodeErrInternalServer, constant.ErrTypeInternalServer, err)
|
|
return
|
|
}
|
|
}
|
|
entranceItem := c.Request.Header.Get("EntranceCode")
|
|
var entrance []byte
|
|
if len(entranceItem) != 0 {
|
|
entrance, _ = base64.StdEncoding.DecodeString(entranceItem)
|
|
}
|
|
|
|
user, err := authService.Login(c, req, string(entrance))
|
|
go saveLoginLogs(c, err)
|
|
if err != nil {
|
|
helper.ErrorWithDetail(c, constant.CodeErrInternalServer, constant.ErrTypeInternalServer, err)
|
|
return
|
|
}
|
|
helper.SuccessWithData(c, user)
|
|
}
|
|
|
|
// @Tags Auth
|
|
// @Summary User login with mfa
|
|
// @Description 用户 mfa 登录
|
|
// @Accept json
|
|
// @Param request body dto.MFALogin true "request"
|
|
// @Success 200 {object} dto.UserLoginInfo
|
|
// @Router /auth/mfalogin [post]
|
|
// @Header 200 {string} EntranceCode "安全入口"
|
|
func (b *BaseApi) MFALogin(c *gin.Context) {
|
|
var req dto.MFALogin
|
|
if err := c.ShouldBindJSON(&req); err != nil {
|
|
helper.ErrorWithDetail(c, constant.CodeErrBadRequest, constant.ErrTypeInvalidParams, err)
|
|
return
|
|
}
|
|
if err := global.VALID.Struct(req); err != nil {
|
|
helper.ErrorWithDetail(c, constant.CodeErrBadRequest, constant.ErrTypeInvalidParams, err)
|
|
return
|
|
}
|
|
entranceItem := c.Request.Header.Get("EntranceCode")
|
|
var entrance []byte
|
|
if len(entranceItem) != 0 {
|
|
entrance, _ = base64.StdEncoding.DecodeString(entranceItem)
|
|
}
|
|
|
|
user, err := authService.MFALogin(c, req, string(entrance))
|
|
if err != nil {
|
|
helper.ErrorWithDetail(c, constant.CodeErrInternalServer, constant.ErrTypeInternalServer, err)
|
|
return
|
|
}
|
|
helper.SuccessWithData(c, user)
|
|
}
|
|
|
|
// @Tags Auth
|
|
// @Summary User logout
|
|
// @Description 用户登出
|
|
// @Success 200
|
|
// @Security ApiKeyAuth
|
|
// @Router /auth/logout [post]
|
|
func (b *BaseApi) LogOut(c *gin.Context) {
|
|
if err := authService.LogOut(c); err != nil {
|
|
helper.ErrorWithDetail(c, constant.CodeErrInternalServer, constant.ErrTypeInternalServer, err)
|
|
return
|
|
}
|
|
helper.SuccessWithData(c, nil)
|
|
}
|
|
|
|
// @Tags Auth
|
|
// @Summary Load captcha
|
|
// @Description 加载验证码
|
|
// @Success 200 {object} dto.CaptchaResponse
|
|
// @Router /auth/captcha [get]
|
|
func (b *BaseApi) Captcha(c *gin.Context) {
|
|
captcha, err := captcha.CreateCaptcha()
|
|
if err != nil {
|
|
helper.ErrorWithDetail(c, constant.CodeErrInternalServer, constant.ErrTypeInternalServer, err)
|
|
return
|
|
}
|
|
helper.SuccessWithData(c, captcha)
|
|
}
|
|
|
|
// @Tags Auth
|
|
// @Summary Load safety status
|
|
// @Description 获取系统安全登录状态
|
|
// @Success 200
|
|
// @Router /auth/issafety [get]
|
|
func (b *BaseApi) CheckIsSafety(c *gin.Context) {
|
|
code := c.DefaultQuery("code", "")
|
|
status, err := authService.CheckIsSafety(code)
|
|
if err != nil {
|
|
helper.ErrorWithDetail(c, constant.CodeErrInternalServer, constant.ErrTypeInternalServer, err)
|
|
return
|
|
}
|
|
helper.SuccessWithData(c, status)
|
|
}
|
|
|
|
// @Tags Auth
|
|
// @Summary Check System isDemo
|
|
// @Description 判断是否为demo环境
|
|
// @Success 200
|
|
// @Router /auth/demo [get]
|
|
func (b *BaseApi) CheckIsDemo(c *gin.Context) {
|
|
helper.SuccessWithData(c, global.CONF.System.IsDemo)
|
|
}
|
|
|
|
// @Tags Auth
|
|
// @Summary Load System Language
|
|
// @Description 获取系统语言设置
|
|
// @Success 200
|
|
// @Router /auth/language [get]
|
|
func (b *BaseApi) GetLanguage(c *gin.Context) {
|
|
settingInfo, err := settingService.GetSettingInfo()
|
|
if err != nil {
|
|
helper.ErrorWithDetail(c, constant.CodeErrInternalServer, constant.ErrTypeInternalServer, err)
|
|
return
|
|
}
|
|
helper.SuccessWithData(c, settingInfo.Language)
|
|
}
|
|
|
|
func saveLoginLogs(c *gin.Context, err error) {
|
|
var logs model.LoginLog
|
|
if err != nil {
|
|
logs.Status = constant.StatusFailed
|
|
logs.Message = err.Error()
|
|
} else {
|
|
logs.Status = constant.StatusSuccess
|
|
}
|
|
logs.IP = c.ClientIP()
|
|
qqWry, err := qqwry.NewQQwry()
|
|
if err != nil {
|
|
global.LOG.Errorf("load qqwry datas failed: %s", err)
|
|
}
|
|
res := qqWry.Find(logs.IP)
|
|
logs.Agent = c.GetHeader("User-Agent")
|
|
logs.Address = res.Area
|
|
_ = logService.CreateLoginLog(logs)
|
|
}
|