feat: 添加是否需要校验验证码的功能。

feat: 添加是否需要校验验证码的功能。

README.md

@@ -107,7 +107,7 @@ QQ群 3：767524537
 
 [兰玉磊的技术博客](https://www.fdevops.com/)
 
-###个人微信，添加好友请描述地区、公司及名字，例如：北京-国美-xxx。
+### 个人微信，添加好友请描述地区、公司及名字，例如：北京-国美-xxx。
 
 微信号：fdevops
 

handler/auth.go

@@ -7,6 +7,7 @@ import (
 	jwt "ferry/pkg/jwtauth"
 	ldap1 "ferry/pkg/ldap"
 	"ferry/pkg/logger"
+	"ferry/pkg/settings"
 	"ferry/tools"
 	"fmt"
 	"net/http"
@@ -66,6 +67,7 @@ func Authenticator(c *gin.Context) (interface{}, error) {
 		authUserCount int
 		addUserInfo   system.SysUser
 		ldapUserInfo  *ldap.Entry
+		isVerifyCode  interface{}
 	)
 
 	ua := user_agent.New(c.Request.UserAgent())
@@ -91,12 +93,20 @@ func Authenticator(c *gin.Context) (interface{}, error) {
 	}
 	loginLog.Username = loginVal.Username
 
-	// 校验验证码
-	if !store.Verify(loginVal.UUID, loginVal.Code, true) {
-		loginLog.Status = "1"
-		loginLog.Msg = "验证码错误"
-		_, _ = loginLog.Create()
-		return nil, jwt.ErrInvalidVerificationode
+	// 查询设置 is_verify_code
+	isVerifyCode, err = settings.GetContentByKey(1, "is_verify_code")
+	if err != nil {
+		return nil, errors.New("获取是否需要验证码校验失败")
+	}
+
+	if isVerifyCode.(bool) {
+		// 校验验证码
+		if !store.Verify(loginVal.UUID, loginVal.Code, true) {
+			loginLog.Status = "1"
+			loginLog.Msg = "验证码错误"
+			_, _ = loginLog.Create()
+			return nil, jwt.ErrInvalidVerificationode
+		}
 	}
 
 	// ldap 验证

models/system/login.go

@@ -12,7 +12,7 @@ import (
 type Login struct {
 	Username  string `form:"UserName" json:"username" binding:"required"`
 	Password  string `form:"Password" json:"password" binding:"required"`
-	Code      string `form:"Code" json:"code" binding:"required"`
+	Code      string `form:"Code" json:"code"`
 	UUID      string `form:"UUID" json:"uuid" binding:"required"`
 	LoginType int    `form:"LoginType" json:"loginType"`
 }

pkg/settings/get_content.go

@@ -0,0 +1,40 @@
+package settings
+
+import (
+	"encoding/json"
+	"ferry/global/orm"
+	"ferry/models/system"
+)
+
+func GetContent(classify int) (content map[string]interface{}, err error) {
+	var (
+		settings system.Settings
+	)
+
+	err = orm.Eloquent.Where("classify = ?", classify).Find(&settings).Error
+	if err != nil {
+		return
+	}
+
+	err = json.Unmarshal(settings.Content, &content)
+	if err != nil {
+		return
+	}
+
+	return
+}
+
+func GetContentByKey(classify int, key string) (value interface{}, err error) {
+	var (
+		content map[string]interface{}
+	)
+
+	content, err = GetContent(classify)
+	if err != nil {
+		return
+	}
+
+	value = content[key]
+
+	return
+}