open-im-server/internal/demo/register/send_code.go

package register

import (
	"Open_IM/pkg/common/config"
	"Open_IM/pkg/common/constant"
	"Open_IM/pkg/common/db"
	"Open_IM/pkg/common/db/mysql_model/im_mysql_model"
	"Open_IM/pkg/common/log"
	"Open_IM/pkg/common/utils"
	"fmt"
	"github.com/gin-gonic/gin"
	"gopkg.in/gomail.v2"
	"math/rand"
	"net/http"

	"time"
)

var sms SMS

func init() {
	var err error
	if config.Config.Demo.AliSMSVerify.Enable {
		sms, err = NewAliSMS()
		if err != nil {
			panic(err)
		}
	} else {
		sms, err = NewTencentSMS()
		if err != nil {
			panic(err)
		}
	}
}

type paramsVerificationCode struct {
	Email       string `json:"email"`
	PhoneNumber string `json:"phoneNumber"`
	OperationID string `json:"operationID" binding:"required"`
	UsedFor     int    `json:"usedFor"`
	AreaCode    string `json:"areaCode"`
}

func SendVerificationCode(c *gin.Context) {
	params := paramsVerificationCode{}

	if err := c.BindJSON(&params); err != nil {
		log.NewError("", "BindJSON failed", "err:", err.Error(), "phoneNumber", params.PhoneNumber, "email", params.Email)
		c.JSON(http.StatusBadRequest, gin.H{"errCode": constant.FormattingError, "errMsg": err.Error()})
		return
	}
	operationID := params.OperationID
	if operationID == "" {
		operationID = utils.OperationIDGenerator()
	}
	log.Info(operationID, "SendVerificationCode args: ", "area code: ", params.AreaCode, "Phone Number: ", params.PhoneNumber)
	var account string
	if params.Email != "" {
		account = params.Email
	} else {
		account = params.PhoneNumber
	}
	var accountKey = params.AreaCode + account
	if params.UsedFor == 0 {
		params.UsedFor = constant.VerificationCodeForRegister
	}
	switch params.UsedFor {
	case constant.VerificationCodeForRegister:
		_, err := im_mysql_model.GetRegister(account, params.AreaCode, "")
		if err == nil {
			log.NewError(params.OperationID, "The phone number has been registered", params)
			c.JSON(http.StatusOK, gin.H{"errCode": constant.HasRegistered, "errMsg": "The phone number has been registered"})
			return
		}
		accountKey = accountKey + "_" + constant.VerificationCodeForRegisterSuffix
		ok, err := db.DB.JudgeAccountEXISTS(accountKey)
		if ok || err != nil {
			log.NewError(params.OperationID, "Repeat send code", params, accountKey)
			c.JSON(http.StatusOK, gin.H{"errCode": constant.RepeatSendCode, "errMsg": "Repeat send code"})
			return
		}

	case constant.VerificationCodeForReset:
		accountKey = accountKey + "_" + constant.VerificationCodeForResetSuffix
		ok, err := db.DB.JudgeAccountEXISTS(accountKey)
		if ok || err != nil {
			log.NewError(params.OperationID, "Repeat send code", params, accountKey)
			c.JSON(http.StatusOK, gin.H{"errCode": constant.RepeatSendCode, "errMsg": "Repeat send code"})
			return
		}
	}
	rand.Seed(time.Now().UnixNano())
	code := 100000 + rand.Intn(900000)
	log.NewInfo(params.OperationID, params.UsedFor, "begin store redis", accountKey, code)
	err := db.DB.SetAccountCode(accountKey, code, config.Config.Demo.CodeTTL)
	if err != nil {
		log.NewError(params.OperationID, "set redis error", accountKey, "err", err.Error())
		c.JSON(http.StatusOK, gin.H{"errCode": constant.SmsSendCodeErr, "errMsg": "Enter the superCode directly in the verification code box, SuperCode can be configured in config.xml"})
		return
	}
	log.NewDebug(params.OperationID, config.Config.Demo)
	if params.Email != "" {
		m := gomail.NewMessage()
		m.SetHeader(`From`, config.Config.Demo.Mail.SenderMail)
		m.SetHeader(`To`, []string{account}...)
		m.SetHeader(`Subject`, config.Config.Demo.Mail.Title)
		m.SetBody(`text/html`, fmt.Sprintf("%d", code))
		if err := gomail.NewDialer(config.Config.Demo.Mail.SmtpAddr, config.Config.Demo.Mail.SmtpPort, config.Config.Demo.Mail.SenderMail, config.Config.Demo.Mail.SenderAuthorizationCode).DialAndSend(m); err != nil {
			log.Error(params.OperationID, "send mail error", account, err.Error())
			c.JSON(http.StatusOK, gin.H{"errCode": constant.MailSendCodeErr, "errMsg": ""})
			return
		}
	} else {
		//client, err := CreateClient(tea.String(config.Config.Demo.AliSMSVerify.AccessKeyID), tea.String(config.Config.Demo.AliSMSVerify.AccessKeySecret))
		//if err != nil {
		//	log.NewError(params.OperationID, "create sendSms client err", "err", err.Error())
		//	c.JSON(http.StatusOK, gin.H{"errCode": constant.SmsSendCodeErr, "errMsg": "Enter the superCode directly in the verification code box, SuperCode can be configured in config.xml"})
		//	return
		//}

		//sendSmsRequest := &dysmsapi20170525.SendSmsRequest{
		//	PhoneNumbers:  tea.String(accountKey),
		//	SignName:      tea.String(config.Config.Demo.AliSMSVerify.SignName),
		//	TemplateCode:  tea.String(config.Config.Demo.AliSMSVerify.VerificationCodeTemplateCode),
		//	TemplateParam: tea.String(fmt.Sprintf("{\"code\":\"%d\"}", code)),
		//}
		response, err := sms.SendSms(code, params.AreaCode+params.PhoneNumber)
		//response, err := client.SendSms(sendSmsRequest)
		if err != nil {
			log.NewError(params.OperationID, "sendSms error", account, "err", err.Error(), response)
			c.JSON(http.StatusOK, gin.H{"errCode": constant.SmsSendCodeErr, "errMsg": "Enter the superCode directly in the verification code box, SuperCode can be configured in config.xml"})
			return
		}
	}
	log.Debug(params.OperationID, "send sms success", code, accountKey)
	data := make(map[string]interface{})
	data["account"] = account
	c.JSON(http.StatusOK, gin.H{"errCode": constant.NoError, "errMsg": "Verification code has been set!", "data": data})
}

//func CreateClient(accessKeyId *string, accessKeySecret *string) (result *dysmsapi20170525.Client, err error) {
//	c := &openapi.Config{
//		// 您的AccessKey ID
//		AccessKeyId: accessKeyId,
//		// 您的AccessKey Secret
//		AccessKeySecret: accessKeySecret,
//	}
//
//	// 访问的域名
//	c.Endpoint = tea.String("dysmsapi.aliyuncs.com")
//	result = &dysmsapi20170525.Client{}
//	result, err = dysmsapi20170525.NewClient(c)
//	return result, err
//}
//func CreateTencentSMSClient() (string, error) {
//	credential := common.NewCredential(
//		config.Config.Demo.TencentSMS.SecretID,
//		config.Config.Demo.TencentSMS.SecretKey,
//	)
//	cpf := profile.NewClientProfile()
//	client, err := sms.NewClient(credential, config.Config.Demo.TencentSMS.Region, cpf)
//	if err != nil {
//		return "", err
//	}
//	request := sms.NewSendSmsRequest()
//	request.SmsSdkAppId = common.StringPtr(config.Config.Demo.TencentSMS.AppID)
//	request.SignName = common.StringPtr(config.Config.Demo.TencentSMS.SignName)
//	request.TemplateId = common.StringPtr(config.Config.Demo.TencentSMS.VerificationCodeTemplateCode)
//	request.TemplateParamSet = common.StringPtrs([]string{"666666"})
//	request.PhoneNumberSet = common.StringPtrs([]string{"+971588232183"})
//	// 通过client对象调用想要访问的接口，需要传入请求对象
//	response, err := client.SendSms(request)
//	// 非SDK异常，直接失败。实际代码中可以加入其他的处理。
//	if err != nil {
//		log.Error("test", "send code to tencent err", err.Error())
//	}
//	// 处理异常
//	if _, ok := err.(*errors.TencentCloudSDKError); ok {
//		log.Error("test", "An API error has returned:", err.Error())
//		return "", err
//	}
//
//	b, _ := json.Marshal(response.Response)
//	return string(b), nil
//}