demo merge into project

2021-12-01 12:16:02 +08:00
parent bd62da8e1a
commit 15660663f6
20 changed files with 788 additions and 7 deletions
@@ -0,0 +1,112 @@
+package register
+
+import (
+	"Open_IM/pkg/common/config"
+	"Open_IM/pkg/common/constant"
+	"Open_IM/pkg/common/db/mysql_model/im_mysql_model"
+	"Open_IM/pkg/common/log"
+	"Open_IM/pkg/utils"
+	"bytes"
+	"encoding/json"
+	"fmt"
+	"github.com/gin-gonic/gin"
+	"io/ioutil"
+	"net/http"
+)
+
+type ParamsLogin struct {
+	Email       string `json:"email"`
+	PhoneNumber string `json:"phoneNumber"`
+	Password    string `json:"password"`
+	Platform    int32  `json:"platform"`
+}
+
+func Login(c *gin.Context) {
+	log.NewDebug("Login api is statrting...")
+	params := ParamsLogin{}
+	if err := c.BindJSON(&params); err != nil {
+		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
+	}
+
+	log.InfoByKv("api Login get params", account)
+
+	queryParams := im_mysql_model.Register{
+		Account:  account,
+		Password: params.Password,
+	}
+
+	canLogin := im_mysql_model.Login(&queryParams)
+	if canLogin == 1 {
+		log.ErrorByKv("Incorrect phone number password", account, "err", "Mobile phone number is not registered")
+		c.JSON(http.StatusOK, gin.H{"errCode": constant.LogicalError, "errMsg": "Mobile phone number is not registered"})
+		return
+	}
+	if canLogin == 2 {
+		log.ErrorByKv("Incorrect phone number password", account, "err", "Incorrect password")
+		c.JSON(http.StatusOK, gin.H{"errCode": constant.LogicalError, "errMsg": "Incorrect password"})
+		return
+	}
+
+	resp, err := OpenIMToken(account, params.Platform)
+	if err != nil {
+		log.ErrorByKv("get token by phone number err", account, "err", err.Error())
+		c.JSON(http.StatusOK, gin.H{"errCode": constant.HttpError, "errMsg": err.Error()})
+		return
+	}
+	response, err := ioutil.ReadAll(resp.Body)
+	defer resp.Body.Close()
+	if err != nil {
+		log.ErrorByKv("Failed to read file", account, "err", err.Error())
+		c.JSON(http.StatusOK, gin.H{"errCode": constant.IoErrot, "errMsg": err.Error()})
+		return
+	}
+	imRep := IMRegisterResp{}
+	err = json.Unmarshal(response, &imRep)
+	if err != nil {
+		log.ErrorByKv("json parsing failed", account, "err", err.Error())
+		c.JSON(http.StatusOK, gin.H{"errCode": constant.FormattingError, "errMsg": err.Error()})
+		return
+	}
+
+	if imRep.ErrCode != 0 {
+		log.ErrorByKv("openIM Login request failed", account, "err")
+		c.JSON(http.StatusOK, gin.H{"errCode": constant.HttpError, "errMsg": imRep.ErrMsg})
+		return
+	}
+
+	c.JSON(http.StatusOK, gin.H{"errCode": constant.NoError, "errMsg": "", "data": imRep.Data})
+	return
+
+}
+
+func OpenIMToken(Account string, platform int32) (*http.Response, error) {
+	url := fmt.Sprintf("http://%s:10000/auth/user_token", utils.ServerIP)
+
+	client := &http.Client{}
+	params := make(map[string]interface{})
+
+	params["secret"] = config.Config.Secret
+	params["platform"] = platform
+	params["uid"] = Account
+	con, err := json.Marshal(params)
+	if err != nil {
+		log.ErrorByKv("json parsing failed", Account, "err", err.Error())
+		return nil, err
+	}
+	req, err := http.NewRequest("POST", url, bytes.NewBuffer([]byte(con)))
+	if err != nil {
+		log.ErrorByKv("request error", "/auth/user_token", "err", err.Error())
+		return nil, err
+	}
+
+	resp, err := client.Do(req)
+	return resp, err
+}
@@ -0,0 +1,151 @@
+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"
+	"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/garyburd/redigo/redis"
+	"github.com/gin-gonic/gin"
+	"gopkg.in/gomail.v2"
+	"math/rand"
+	"net/http"
+	"time"
+)
+
+type paramsVerificationCode struct {
+	Email       string `json:"email"`
+	PhoneNumber string `json:"phoneNumber"`
+}
+
+func SendVerificationCode(c *gin.Context) {
+	log.InfoByKv("sendCode api is statrting...", "")
+	params := paramsVerificationCode{}
+
+	if err := c.BindJSON(&params); err != nil {
+		log.ErrorByKv("request params json parsing failed", params.PhoneNumber, params.Email, "err", err.Error())
+		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
+	}
+
+	queryParams := im_mysql_model.GetRegisterParams{
+		Account: account,
+	}
+	_, err, rowsAffected := im_mysql_model.GetRegister(&queryParams)
+
+	if err == nil && rowsAffected != 0 {
+		log.ErrorByKv("The phone number has been registered", queryParams.Account, "err")
+		c.JSON(http.StatusOK, gin.H{"errCode": constant.LogicalError, "errMsg": "The phone number has been registered"})
+		return
+	}
+
+	log.InfoByKv("begin sendSms", account)
+	rand.Seed(time.Now().UnixNano())
+	code := 100000 + rand.Intn(900000)
+	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.ErrorByKv("send mail error", account, "err", err.Error())
+			c.JSON(http.StatusOK, gin.H{"errCode": constant.IntentionalError, "errMsg": err.Error()})
+			return
+		}
+	} else {
+		client, err := CreateClient(tea.String(config.Config.Demo.AliSMSVerify.AccessKeyID), tea.String(config.Config.Demo.AliSMSVerify.AccessKeySecret))
+		if err != nil {
+			log.ErrorByKv("create sendSms client err", "", "err", err.Error())
+			c.JSON(http.StatusOK, gin.H{"errCode": constant.IntentionalError, "errMsg": "Enter the superCode directly in the verification code box, SuperCode can be configured in config.xml"})
+			return
+		}
+
+		sendSmsRequest := &dysmsapi20170525.SendSmsRequest{
+			PhoneNumbers:  tea.String(account),
+			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.ErrorByKv("sendSms error", account, "err", err.Error())
+			c.JSON(http.StatusOK, gin.H{"errCode": constant.IntentionalError, "errMsg": "Enter the superCode directly in the verification code box, SuperCode can be configured in config.xml"})
+			return
+		}
+		if *response.Body.Code != "OK" {
+			log.ErrorByKv("alibabacloud sendSms error", account, "err", response.Body.Code, response.Body.Message)
+			c.JSON(http.StatusOK, gin.H{"errCode": constant.IntentionalError, "errMsg": "Enter the superCode directly in the verification code box, SuperCode can be configured in config.xml"})
+			return
+		}
+	}
+
+	log.InfoByKv("begin store redis", account)
+	v, err := redis.Int(db.DB.Exec("TTL", account))
+	if err != nil {
+		log.ErrorByKv("get account from redis error", account, "err", err.Error())
+		c.JSON(http.StatusOK, gin.H{"errCode": constant.IntentionalError, "errMsg": "Enter the superCode directly in the verification code box, SuperCode can be configured in config.xml"})
+		return
+	}
+	switch {
+	case v == -2:
+		_, err = db.DB.Exec("SET", account, code, "EX", 600)
+		if err != nil {
+			log.ErrorByKv("set redis error", account, "err", err.Error())
+			c.JSON(http.StatusOK, gin.H{"errCode": constant.IntentionalError, "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 sent successfully!", "data": data})
+		log.InfoByKv("send new verification code", account)
+		return
+	case v > 540:
+		data := make(map[string]interface{})
+		data["account"] = account
+		c.JSON(http.StatusOK, gin.H{"errCode": constant.LogicalError, "errMsg": "Frequent operation!", "data": data})
+		log.InfoByKv("frequent operation", account)
+		return
+	case v < 540:
+		_, err = db.DB.Exec("SET", account, code, "EX", 600)
+		if err != nil {
+			c.JSON(http.StatusOK, gin.H{"errCode": constant.IntentionalError, "errMsg": "Enterthe 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 reset!", "data": data})
+		log.InfoByKv("Reset verification code", account)
+		return
+	}
+
+}
+
+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
+}
@@ -0,0 +1,155 @@
+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/utils"
+	"bytes"
+	"encoding/json"
+	"fmt"
+	"github.com/garyburd/redigo/redis"
+	"github.com/gin-gonic/gin"
+	"io/ioutil"
+	"net/http"
+)
+
+type ParamsSetPassword struct {
+	Email            string `json:"email"`
+	PhoneNumber      string `json:"phoneNumber"`
+	Password         string `json:"password"`
+	VerificationCode string `json:"verificationCode"`
+}
+
+type Data struct {
+	ExpiredTime int64  `json:"expiredTime"`
+	Token       string `json:"token"`
+	Uid         string `json:"uid"`
+}
+
+type IMRegisterResp struct {
+	Data    Data   `json:"data"`
+	ErrCode int32  `json:"errCode"`
+	ErrMsg  string `json:"errMsg"`
+}
+
+func SetPassword(c *gin.Context) {
+	log.InfoByKv("setPassword api is statrting...", "")
+	params := ParamsSetPassword{}
+	if err := c.BindJSON(&params); err != nil {
+		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
+	}
+
+	log.InfoByKv("begin store redis", account)
+	v, err := redis.String(db.DB.Exec("GET", account))
+
+	if params.VerificationCode == config.Config.Demo.SuperCode {
+		goto openIMRegisterTab
+	}
+
+	fmt.Println("Get Redis:", v, err)
+	if err != nil {
+		log.ErrorByKv("password Verification code expired", account, "err", err.Error())
+		data := make(map[string]interface{})
+		data["phoneNumber"] = account
+		c.JSON(http.StatusOK, gin.H{"errCode": constant.LogicalError, "errMsg": "Verification expired!", "data": data})
+		return
+	}
+	if v != params.VerificationCode {
+		log.InfoByKv("password Verification code error", account, params.VerificationCode)
+		data := make(map[string]interface{})
+		data["PhoneNumber"] = account
+		c.JSON(http.StatusOK, gin.H{"errCode": constant.LogicalError, "errMsg": "Verification code error!", "data": data})
+		return
+	}
+
+openIMRegisterTab:
+	log.InfoByKv("openIM register begin", account)
+	resp, err := OpenIMRegister(account)
+
+	log.InfoByKv("openIM register resp", account, resp, err)
+	if err != nil {
+		log.ErrorByKv("request openIM register error", account, "err", err.Error())
+		c.JSON(http.StatusOK, gin.H{"errCode": constant.HttpError, "errMsg": err.Error()})
+		return
+	}
+	response, err := ioutil.ReadAll(resp.Body)
+	defer resp.Body.Close()
+	if err != nil {
+		c.JSON(http.StatusOK, gin.H{"errCode": constant.IoErrot, "errMsg": err.Error()})
+		return
+	}
+	imrep := IMRegisterResp{}
+	err = json.Unmarshal(response, &imrep)
+	if err != nil {
+		c.JSON(http.StatusOK, gin.H{"errCode": constant.FormattingError, "errMsg": err.Error()})
+		return
+	}
+	if imrep.ErrCode != 0 {
+		c.JSON(http.StatusOK, gin.H{"errCode": constant.HttpError, "errMsg": imrep.ErrMsg})
+		return
+	}
+
+	queryParams := im_mysql_model.SetPasswordParams{
+		Account:  account,
+		Password: params.Password,
+	}
+
+	log.InfoByKv("begin store mysql", account, params.Password)
+	_, err = im_mysql_model.SetPassword(&queryParams)
+	if err != nil {
+		log.ErrorByKv("set phone number password error", account, "err", err.Error())
+		c.JSON(http.StatusOK, gin.H{"errCode": constant.DatabaseError, "errMsg": err.Error()})
+		return
+	}
+
+	log.InfoByKv("end setPassword", account)
+	c.JSON(http.StatusOK, gin.H{"errCode": constant.NoError, "errMsg": "", "data": imrep.Data})
+	return
+}
+
+func OpenIMRegister(account string) (*http.Response, error) {
+	url := fmt.Sprintf("http://%s:10000/auth/user_register", utils.ServerIP)
+	fmt.Println("1:", config.Config.Secret)
+
+	client := &http.Client{}
+
+	params := make(map[string]interface{})
+
+	params["secret"] = config.Config.Secret
+	params["platform"] = 7
+	params["uid"] = account
+	params["name"] = account
+	params["icon"] = ""
+	params["gender"] = 0
+
+	params["mobile"] = ""
+
+	params["email"] = ""
+	params["birth"] = ""
+	params["ex"] = ""
+	con, err := json.Marshal(params)
+	if err != nil {
+		return nil, err
+	}
+
+	log.InfoByKv("openIM register params", account, params)
+	req, err := http.NewRequest("POST", url, bytes.NewBuffer([]byte(con)))
+	if err != nil {
+		return nil, err
+	}
+
+	resp, err := client.Do(req)
+
+	return resp, err
+}
@@ -0,0 +1,69 @@
+package register
+
+import (
+	"Open_IM/pkg/common/config"
+	"Open_IM/pkg/common/constant"
+	"Open_IM/pkg/common/db"
+	"Open_IM/pkg/common/log"
+
+	"github.com/garyburd/redigo/redis"
+	"github.com/gin-gonic/gin"
+	"net/http"
+)
+
+type paramsCertification struct {
+	Email            string `json:"email"`
+	PhoneNumber      string `json:"phoneNumber"`
+	VerificationCode string `json:"verificationCode"`
+}
+
+func Verify(c *gin.Context) {
+	log.InfoByKv("Verify api is statrting...", "")
+	params := paramsCertification{}
+
+	if err := c.BindJSON(&params); err != nil {
+		log.ErrorByKv("request params json parsing failed", "", "err", err.Error())
+		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
+	}
+
+	if params.VerificationCode == config.Config.Demo.SuperCode {
+		log.InfoByKv("Super Code Verified successfully", account)
+		data := make(map[string]interface{})
+		data["account"] = account
+		data["verificationCode"] = params.VerificationCode
+		c.JSON(http.StatusOK, gin.H{"errCode": constant.NoError, "errMsg": "Verified successfully!", "data": data})
+		return
+	}
+
+	log.InfoByKv("begin get form redis", account)
+	v, err := redis.String(db.DB.Exec("GET", account))
+	log.InfoByKv("redis phone number and verificating Code", account, v)
+	if err != nil {
+		log.ErrorByKv("Verification code expired", account, "err", err.Error())
+		data := make(map[string]interface{})
+		data["account"] = account
+		c.JSON(http.StatusOK, gin.H{"errCode": constant.LogicalError, "errMsg": "Verification code expired!", "data": data})
+		return
+	}
+	if params.VerificationCode == v {
+		log.InfoByKv("Verified successfully", account)
+		data := make(map[string]interface{})
+		data["account"] = account
+		data["verificationCode"] = params.VerificationCode
+		c.JSON(http.StatusOK, gin.H{"errCode": constant.NoError, "errMsg": "Verified successfully!", "data": data})
+		return
+	} else {
+		log.InfoByKv("Verification code error", account, params.VerificationCode)
+		data := make(map[string]interface{})
+		data["account"] = account
+		c.JSON(http.StatusOK, gin.H{"errCode": constant.LogicalError, "errMsg": "Verification code error!", "data": data})
+	}
+}