v1.0.0

2023-11-08 21:53:07 +08:00
commit 211c3071dc
245 changed files with 39293 additions and 0 deletions
@@ -0,0 +1,183 @@
+package system
+
+import (
+	"fmt"
+	"math/rand"
+	"sun-panel/api/api_v1/common/apiData/commonApiStructs"
+	"sun-panel/api/api_v1/common/apiReturn"
+	"sun-panel/api/api_v1/common/base"
+	"sun-panel/global"
+	"sun-panel/lib/cmn"
+	"sun-panel/lib/cmn/systemSetting"
+	"sun-panel/lib/mail"
+	"sun-panel/models"
+	"time"
+
+	"github.com/gin-gonic/gin"
+	"gorm.io/gorm"
+)
+
+type registerInfo struct {
+	Email        string                               `json:"email"`
+	UserName     string                               `json:"userName"`
+	Password     string                               `json:"password"`
+	Vcode        string                               `json:"vcode"`
+	EmailVCode   string                               `json:"emailVCode"`
+	VCode        string                               `json:"vCode"`
+	Verification commonApiStructs.VerificationRequest `json:"verification"`
+	ReferralCode string                               `json:"referralCode"`
+}
+
+const EmailCodeCapacity = 1000
+
+type RegisterApi struct{}
+
+// 获取注册验证码
+func (l RegisterApi) SendRegisterVcode(c *gin.Context) {
+	req := registerInfo{}
+	err := c.ShouldBindJSON(&req)
+	req.Email = req.UserName
+	if err != nil {
+		apiReturn.ErrorParamFomat(c, err.Error())
+		return
+	}
+	errMsg, err := base.ValidateInputStruct(req)
+	if err != nil {
+		apiReturn.Error(c, errMsg)
+		return
+	}
+
+	// 验证码验证
+	{
+		errCode, verifcationId := base.VerificationCheck(req.Verification.CodeID, req.Verification.VCode)
+		if errCode != apiReturn.ERROR_CODE_SUCCESS {
+			apiReturn.ErrorVerification(c, errCode, verifcationId)
+			return
+		}
+	}
+
+	// 验证是否开启注册和后缀格式是否正确
+	{
+		systemSettingInfo := systemSetting.ApplicationSetting{}
+		if err := global.SystemSetting.GetValueByInterface("system_application", &systemSettingInfo); err != nil || !systemSettingInfo.Register.OpenRegister {
+			apiReturn.Error(c, global.Lang.Get("register.unopened_register"))
+			return
+		}
+
+		if systemSettingInfo.Register.EmailSuffix != "" && !cmn.VerifyFormat("^.*"+systemSettingInfo.Register.EmailSuffix+"$", req.Email) {
+			apiReturn.Error(c, global.Lang.GetWithFields("register.emailSuffix_error", map[string]string{"EmailSuffix": systemSettingInfo.Register.EmailSuffix}))
+			return
+		}
+	}
+
+	// 验证邮箱是否被注册
+	{
+		userCheck := &models.User{Mail: req.UserName}
+		if _, err := userCheck.GetUserInfoByUsername(req.UserName); err == nil && err != gorm.ErrRecordNotFound {
+			apiReturn.Error(c, global.Lang.Get("register.mail_exist"))
+			return
+		}
+	}
+
+	emailCode := generateEmailCode()
+	count, err := global.VerifyCodeCachePool.ItemCount()
+	if err != nil || count >= EmailCodeCapacity {
+		global.VerifyCodeCachePool.Flush()
+	}
+	global.VerifyCodeCachePool.Set(req.Email, emailCode, 0)
+	emailInfoConfig := systemSetting.Email{}
+	global.SystemSetting.GetValueByInterface("system_email", &emailInfoConfig)
+	emailInfo := mail.EmailInfo{
+		Username: emailInfoConfig.Mail,
+		Password: emailInfoConfig.Password,
+		Host:     emailInfoConfig.Host,
+		Port:     emailInfoConfig.Port,
+	}
+	err = mail.SendRegisterEmail(mail.NewEmailer(emailInfo), req.Email, emailCode)
+	if err != nil {
+		apiReturn.Error(c, global.Lang.Get("mail.send_mail_fail"))
+		global.Logger.Errorf("[register] fail to send email to%s", req.UserName)
+		return
+	}
+	apiReturn.Success(c)
+}
+
+// 注册提交（开始注册）
+func (l *RegisterApi) Commit(c *gin.Context) {
+	req := registerInfo{}
+	err := c.ShouldBindJSON(&req)
+	req.Email = req.UserName
+	if err != nil {
+		apiReturn.ErrorParamFomat(c, err.Error())
+		return
+	}
+	errMsg, err := base.ValidateInputStruct(req)
+	if err != nil {
+		apiReturn.Error(c, errMsg)
+		return
+	}
+
+	// 验证是否开启注册和后缀格式是否正确
+	{
+		systemSettingInfo := systemSetting.ApplicationSetting{}
+		if err := global.SystemSetting.GetValueByInterface("system_application", &systemSettingInfo); err != nil || !systemSettingInfo.Register.OpenRegister {
+			apiReturn.Error(c, global.Lang.Get("register.unopened_register"))
+			return
+		}
+
+		if systemSettingInfo.Register.EmailSuffix != "" && !cmn.VerifyFormat("^.*"+systemSettingInfo.Register.EmailSuffix+"$", req.Email) {
+			apiReturn.Error(c, global.Lang.GetWithFields("register.emailSuffix_error", map[string]string{"EmailSuffix": systemSettingInfo.Register.EmailSuffix}))
+			return
+		}
+	}
+
+	// 验证邮箱是否被注册
+	{
+		userCheck := &models.User{Mail: req.UserName}
+		if _, err := userCheck.GetUserInfoByUsername(req.UserName); err == nil && err != gorm.ErrRecordNotFound {
+			apiReturn.Error(c, global.Lang.Get("register.mail_exist"))
+			return
+		}
+	}
+
+	// 验证码验证
+	{
+		vCode, ok := global.VerifyCodeCachePool.Get(req.Email)
+		if !ok {
+			apiReturn.Error(c, global.Lang.Get("common.captcha_code_error"))
+			//验证码不存在
+			return
+		}
+		if vCode != req.EmailVCode {
+			apiReturn.Error(c, global.Lang.Get("common.captcha_code_error"))
+			return
+			//验证码有误
+		}
+	}
+
+	// 自动生成用户昵称
+	name := "用户" + cmn.BuildRandCode(4, cmn.RAND_CODE_MODE3)
+
+	//验证通过，注册
+	user := &models.User{
+		Mail:     req.UserName,
+		Name:     name,
+		Username: req.UserName,
+		Password: cmn.PasswordEncryption(req.Password),
+		Status:   1,
+		Role:     2,
+	}
+	_, err = user.CreateOne()
+	if err != nil {
+		apiReturn.ErrorDatabase(c, err.Error())
+		return
+	}
+	//删除旧的验证码
+	global.VerifyCodeCachePool.Delete(req.Email)
+
+	apiReturn.Success(c)
+}
+
+func generateEmailCode() string {
+	return fmt.Sprintf("%06v", rand.New(rand.NewSource(time.Now().UnixNano())).Int31n(1000000))
+}