Merge pull request ThingsPanel#216 from November-12/main

GetCaptcha

controllers/tp_notification.go

@@ -2,13 +2,17 @@ package controllers
 
 import (
 	"ThingsPanel-Go/initialize/psql"
+	"ThingsPanel-Go/initialize/redis"
 	sendmessage "ThingsPanel-Go/initialize/send_message"
 	"ThingsPanel-Go/models"
 	"ThingsPanel-Go/services"
 	"ThingsPanel-Go/utils"
 	response "ThingsPanel-Go/utils"
 	valid "ThingsPanel-Go/validate"
 	"encoding/json"
+	"fmt"
+	"math/rand"
+	"strconv"
 	"strings"
 	"time"
 
@@ -523,3 +527,39 @@ func (c *TpNotification) HistoryList() {
 
 	response.SuccessWithDetailed(200, "success", ret, map[string]string{}, (*context2.Context)(c.Ctx))
 }
+
+func (c *TpNotification) GetCaptcha() {
+	var input struct {
+		PhoneNumber string `json:"phone_number" valid:"Required"`
+	}
+
+	err := json.Unmarshal(c.Ctx.Input.RequestBody, &input)
+
+	if err != nil {
+		response.SuccessWithMessage(400, "参数解析错误", (*context2.Context)(c.Ctx))
+		return
+	}
+
+	verificationCode := fmt.Sprintf("%04d", rand.Intn(10000))
+	err = redis.SetStr(input.PhoneNumber+"_code", verificationCode, 3*time.Minute)
+
+	if err != nil {
+		response.SuccessWithMessage(400, "redis err", (*context2.Context)(c.Ctx))
+		return
+	}
+
+	num, err := strconv.Atoi(input.PhoneNumber)
+	if err != nil {
+		response.SuccessWithMessage(400, "参数解析错误", (*context2.Context)(c.Ctx))
+		return
+	}
+
+	err = sendmessage.SendSMSVerificationCode(num, verificationCode, "")
+
+	if err != nil {
+		response.SuccessWithMessage(400, "message err", (*context2.Context)(c.Ctx))
+		return
+	}
+
+	response.SuccessWithDetailed(200, "success", nil, map[string]string{}, (*context2.Context)(c.Ctx))
+}

middleware/auth_middle.go

@@ -31,6 +31,7 @@ func AuthMiddle() {
 		"api/auth/refresh":                    0,
 		"api/auth/register":                   1,
 		"api/auth/tenant/register":            0,
+		"api/auth/captcha":                    0,
 		"/ws":                                 2,
 		"api/ota/download":                    0,
 	}
@@ -78,7 +79,7 @@ func isAuthExceptUrl(url string, m map[string]interface{}) bool {
 	return ok
 }
 
-//获取token
+// 获取token
 func GetToken(ctx *context.Context) (string, error) {
 	authorization := ctx.Input.Header("Authorization")
 	if len(authorization) == 0 {

routers/router.go

@@ -33,6 +33,10 @@ func init() {
 		web.NSRouter("/auth/logout", &controllers.AuthController{}, "*:Logout"),   //ty
 		web.NSRouter("/auth/refresh", &controllers.AuthController{}, "*:Refresh"), //ty
 		web.NSRouter("/auth/me", &controllers.AuthController{}, "*:Me"),           //ty
+
+		// 获取验证码
+		web.NSRouter("/auth/captcha", &controllers.TpNotification{}, "*:GetCaptcha"),
+
 		// 用户注册
 		web.NSRouter("/auth/register", &controllers.AuthController{}, "*:Register"), //ty
 		//租户注册TenantRegister