mirror of
https://github.com/openimsdk/open-im-server.git
synced 2025-04-06 04:15:46 +08:00
186 lines
7.0 KiB
Go
186 lines
7.0 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"
|
||
"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(¶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
|
||
}
|
||
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
|
||
//}
|