:toda:

http/api/login.go

@@ -0,0 +1,87 @@
+package api
+
+import (
+	"fmt"
+	"gitee.ltd/lxh/logger/log"
+	"github.com/gin-gonic/gin"
+	"github.com/mojocn/base64Captcha"
+	"wireguard-ui/component"
+	"wireguard-ui/http/param"
+	"wireguard-ui/http/response"
+	"wireguard-ui/service"
+	"wireguard-ui/utils"
+)
+
+type LoginApi struct{}
+
+func Login() LoginApi {
+	return LoginApi{}
+}
+
+// Captcha
+// @description: 获取验证码
+// @receiver login
+// @param c
+func (LoginApi) Captcha(c *gin.Context) {
+	math := base64Captcha.DriverMath{Height: 120, Width: 480, Fonts: []string{"ApothecaryFont.ttf", "3Dumb.ttf"}}
+	mathDriver := math.ConvertFonts()
+
+	capt := base64Captcha.NewCaptcha(mathDriver, component.Captcha{})
+
+	id, base64Str, _, err := capt.Generate()
+	if err != nil {
+		response.R(c).FailedWithError(fmt.Errorf("生成验证码失败: %s", err.Error()))
+		return
+	}
+
+	response.R(c).OkWithData(map[string]any{
+		"id":      id,
+		"captcha": base64Str,
+	})
+}
+
+// Login
+// @description: 登陆
+// @receiver login
+// @param c
+func (LoginApi) Login(c *gin.Context) {
+	var p param.Login
+	if err := c.ShouldBind(&p); err != nil {
+		response.R(c).Validator(err)
+		return
+	}
+
+	// 验证验证码是否正确
+	ok := component.Captcha{}.Verify(p.CaptchaId, p.CaptchaCode, true)
+	if !ok {
+		response.R(c).FailedWithError("验证码错误")
+		return
+	}
+
+	// 验证码正确，查询用户信息
+	user, err := service.User().GetUserByAccount(p.Account)
+	if err != nil {
+		response.R(c).FailedWithError("获取用户信息失败")
+		return
+	}
+
+	// 对比密码
+	if !utils.Password().ComparePassword(user.Password, p.Password) {
+		response.R(c).FailedWithError("密码错误")
+		return
+	}
+
+	// 生成token
+	token, expireAt, err := component.JWT().GenerateToken(user.Id)
+	if err != nil {
+		log.Errorf("用户[%s]生成token失败: %v", user.Account, err.Error())
+		response.R(c).FailedWithError("登陆失败！")
+		return
+	}
+
+	response.R(c).OkWithData(map[string]any{
+		"token":    token,
+		"type":     "Bearer",
+		"expireAt": expireAt,
+	})
+}