🎨 update project structure

2024-09-29 15:46:35 +08:00
parent 2769467ce2
commit 87f1ff6961
85 changed files with 693 additions and 431 deletions
--- a/controller/sms_controller/sms_controller.go
+++ b/controller/sms_controller/sms_controller.go
@@ -0,0 +1,162 @@
+package sms_controller
+
+import (
+	ginI18n "github.com/gin-contrib/i18n"
+	"github.com/gin-gonic/gin"
+	gosms "github.com/pkg6/go-sms"
+	"github.com/pkg6/go-sms/gateways"
+	"github.com/pkg6/go-sms/gateways/aliyun"
+	"github.com/pkg6/go-sms/gateways/smsbao"
+	"schisandra-cloud-album/common/constant"
+	"schisandra-cloud-album/common/redis"
+	"schisandra-cloud-album/common/result"
+	"schisandra-cloud-album/global"
+	"schisandra-cloud-album/utils"
+	"time"
+)
+
+// SendMessageByAli 发送短信验证码
+// @Summary 发送短信验证码
+// @Description 发送短信验证码
+// @Tags 短信验证码
+// @Produce json
+// @Param phone query string true "手机号"
+// @Router /controller/sms/ali/send [get]
+func (SmsController) SendMessageByAli(c *gin.Context) {
+	smsRequest := SmsRequest{}
+	err := c.ShouldBindJSON(&smsRequest)
+	if err != nil {
+		result.FailWithMessage(ginI18n.MustGetMessage(c, "CaptchaSendFailed"), c)
+		return
+	}
+	checkRotateData := utils.CheckRotateData(smsRequest.Angle, smsRequest.Key)
+	if !checkRotateData {
+		result.FailWithMessage(ginI18n.MustGetMessage(c, "CaptchaVerifyError"), c)
+		return
+	}
+	isPhone := utils.IsPhone(smsRequest.Phone)
+	if !isPhone {
+		result.FailWithMessage(ginI18n.MustGetMessage(c, "PhoneErrorFormat"), c)
+		return
+	}
+	val := redis.Get(constant.UserLoginSmsRedisKey + smsRequest.Phone).Val()
+	if val != "" {
+		result.FailWithMessage(ginI18n.MustGetMessage(c, "CaptchaTooOften"), c)
+		return
+	}
+	sms := gosms.NewParser(gateways.Gateways{
+		ALiYun: aliyun.ALiYun{
+			Host:            global.CONFIG.SMS.Ali.Host,
+			AccessKeyId:     global.CONFIG.SMS.Ali.AccessKeyID,
+			AccessKeySecret: global.CONFIG.SMS.Ali.AccessKeySecret,
+		},
+	})
+	code := utils.GenValidateCode(6)
+	wrong := redis.Set(constant.UserLoginSmsRedisKey+smsRequest.Phone, code, time.Minute).Err()
+	if wrong != nil {
+		global.LOG.Error(wrong)
+		return
+	}
+	_, err = sms.Send(smsRequest.Phone, gosms.MapStringAny{
+		"content":  "您的验证码是：****。请不要把验证码泄露给其他人。",
+		"template": global.CONFIG.SMS.Ali.TemplateID,
+		//"signName": global.CONFIG.SMS.Ali.Signature,
+		"data": gosms.MapStrings{
+			"code": code,
+		},
+	}, nil)
+	if err != nil {
+		global.LOG.Error(err)
+		result.FailWithMessage(ginI18n.MustGetMessage(c, "CaptchaSendFailed"), c)
+		return
+	}
+	result.OkWithMessage(ginI18n.MustGetMessage(c, "CaptchaSendSuccess"), c)
+
+}
+
+// SendMessageBySmsBao 短信宝发送短信验证码
+// @Summary 短信宝发送短信验证码
+// @Description 短信宝发送短信验证码
+// @Tags 短信验证码
+// @Produce json
+// @Param phone query string true "手机号"
+// @Router /controller/sms/smsbao/send [post]
+func (SmsController) SendMessageBySmsBao(c *gin.Context) {
+	smsRequest := SmsRequest{}
+	err := c.ShouldBindJSON(&smsRequest)
+	if err != nil {
+		result.FailWithMessage(ginI18n.MustGetMessage(c, "CaptchaSendFailed"), c)
+		return
+	}
+	checkRotateData := utils.CheckRotateData(smsRequest.Angle, smsRequest.Key)
+	if !checkRotateData {
+		result.FailWithMessage(ginI18n.MustGetMessage(c, "CaptchaVerifyError"), c)
+		return
+	}
+	isPhone := utils.IsPhone(smsRequest.Phone)
+	if !isPhone {
+		result.FailWithMessage(ginI18n.MustGetMessage(c, "PhoneErrorFormat"), c)
+		return
+	}
+	val := redis.Get(constant.UserLoginSmsRedisKey + smsRequest.Phone).Val()
+	if val != "" {
+		result.FailWithMessage(ginI18n.MustGetMessage(c, "CaptchaTooOften"), c)
+		return
+	}
+	sms := gosms.NewParser(gateways.Gateways{
+		SmsBao: smsbao.SmsBao{
+			User:     global.CONFIG.SMS.SmsBao.User,
+			Password: global.CONFIG.SMS.SmsBao.Password,
+		},
+	})
+	code := utils.GenValidateCode(6)
+	wrong := redis.Set(constant.UserLoginSmsRedisKey+smsRequest.Phone, code, time.Minute).Err()
+	if wrong != nil {
+		global.LOG.Error(wrong)
+		return
+	}
+	_, err = sms.Send(smsRequest.Phone, gosms.MapStringAny{
+		"content": "您的验证码是：" + code + "。请不要把验证码泄露给其他人。",
+	}, nil)
+	if err != nil {
+		global.LOG.Error(err)
+		result.FailWithMessage(ginI18n.MustGetMessage(c, "CaptchaSendFailed"), c)
+		return
+	}
+	result.OkWithMessage(ginI18n.MustGetMessage(c, "CaptchaSendSuccess"), c)
+}
+
+// SendMessageTest 发送测试短信验证码
+// @Summary 发送测试短信验证码
+// @Description 发送测试短信验证码
+// @Tags 短信验证码
+// @Produce json
+// @Param phone query string true "手机号"
+// @Router /controller/sms/test/send [post]
+func (SmsController) SendMessageTest(c *gin.Context) {
+	smsRequest := SmsRequest{}
+	err := c.ShouldBindJSON(&smsRequest)
+	if err != nil {
+		result.FailWithMessage(ginI18n.MustGetMessage(c, "CaptchaSendFailed"), c)
+		return
+	}
+	checkRotateData := utils.CheckRotateData(smsRequest.Angle, smsRequest.Key)
+	if !checkRotateData {
+		result.FailWithMessage(ginI18n.MustGetMessage(c, "CaptchaVerifyError"), c)
+		return
+	}
+	isPhone := utils.IsPhone(smsRequest.Phone)
+	if !isPhone {
+		result.FailWithMessage(ginI18n.MustGetMessage(c, "PhoneError"), c)
+		return
+	}
+	code := utils.GenValidateCode(6)
+	err = redis.Set(constant.UserLoginSmsRedisKey+smsRequest.Phone, code, time.Minute).Err()
+	if err != nil {
+		global.LOG.Error(err)
+		result.FailWithMessage(ginI18n.MustGetMessage(c, "CaptchaSendFailed"), c)
+		return
+	}
+	result.OkWithMessage(ginI18n.MustGetMessage(c, "CaptchaSendSuccess"), c)
+
+}