mirror of
https://github.com/openimsdk/open-im-server.git
synced 2025-04-06 04:15:46 +08:00
168 lines
6.9 KiB
Go
168 lines
6.9 KiB
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"
|
||
"encoding/json"
|
||
"fmt"
|
||
openapi "github.com/alibabacloud-go/darabonba-openapi/client"
|
||
dysmsapi20170525 "github.com/alibabacloud-go/dysmsapi-20170525/v2/client"
|
||
"github.com/alibabacloud-go/tea/tea"
|
||
"github.com/gin-gonic/gin"
|
||
"github.com/tencentcloud/tencentcloud-sdk-go/tencentcloud/common"
|
||
"github.com/tencentcloud/tencentcloud-sdk-go/tencentcloud/common/errors"
|
||
"github.com/tencentcloud/tencentcloud-sdk-go/tencentcloud/common/profile"
|
||
sms "github.com/tencentcloud/tencentcloud-sdk-go/tencentcloud/sms/v20210111"
|
||
"gopkg.in/gomail.v2"
|
||
"math/rand"
|
||
"net/http"
|
||
"time"
|
||
)
|
||
|
||
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(¶ms); 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
|
||
}
|
||
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
|
||
}
|
||
ok, err := db.DB.JudgeAccountEXISTS(accountKey)
|
||
if ok || err != nil {
|
||
log.NewError(params.OperationID, "The phone number has been registered", params)
|
||
c.JSON(http.StatusOK, gin.H{"errCode": constant.RepeatSendCode, "errMsg": "The phone number has been registered"})
|
||
return
|
||
}
|
||
accountKey += "_" + constant.VerificationCodeForRegisterSuffix
|
||
|
||
case constant.VerificationCodeForReset:
|
||
accountKey += "_" + constant.VerificationCodeForResetSuffix
|
||
}
|
||
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("", 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 := client.SendSms(sendSmsRequest)
|
||
if err != nil {
|
||
log.NewError(params.OperationID, "sendSms error", account, "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
|
||
}
|
||
if *response.Body.Code != "OK" {
|
||
log.NewError(params.OperationID, "alibabacloud sendSms error", account, "err", response.Body.Code, response.Body.Message)
|
||
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
|
||
}
|
||
}
|
||
|
||
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
|
||
}
|