eta_api/services/user_login.go

package services

import (
	"encoding/json"
	"errors"
	"eta/eta_api/models"
	"eta/eta_api/models/company"
	"eta/eta_api/models/system"
	"eta/eta_api/utils"
	"fmt"
	"io"
	"net/http"
	"strconv"
	"strings"
	"time"
)

// SendAdminMobileVerifyCode 发送用户手机验证码
func SendAdminMobileVerifyCode(source int, mobile, areaCode string) (ok bool, err error) {
	verifyCode := utils.GetRandDigit(6)
	record := new(system.AdminVerifyCodeRecord)
	record.VerifyType = system.AdminVerifyCodeRecordTypeMobile
	record.Mobile = mobile
	record.Source = source
	record.Code = verifyCode
	record.ExpiredTime = time.Now().Add(utils.VerifyCodeExpireMinute * time.Minute)
	record.CreateTime = time.Now().Local()
	record.ModifyTime = time.Now().Local()
	if e := record.Create(); e != nil {
		err = fmt.Errorf("新增验证码记录失败, Err: %s", e.Error())
		return
	}
	// 获取配置好的短信模版
	smsCond := ` AND conf_key in (?,?) `
	smsPars := make([]interface{}, 0)
	smsPars = append(smsPars, "LoginSmsTpId", "LoginSmsGjTpId")
	conf := new(models.BusinessConf)
	confList, e := conf.GetItemsByCondition(smsCond, smsPars, []string{"conf_key", "conf_val"}, "")
	if e != nil {
		if e.Error() == utils.ErrNoRow() {
			err = fmt.Errorf("请先配置短信模版")
			return
		}
		err = fmt.Errorf("获取短信模版失败, Err: %s", e.Error())
		return
	}

	tplId := ""
	gjTplId := ""
	for _, v := range confList {
		if v.ConfKey == "LoginSmsTpId" {
			tplId = v.ConfVal
		} else if v.ConfKey == "LoginSmsGjTpId" {
			gjTplId = v.ConfVal
		}
	}

	if tplId == "" {
		err = fmt.Errorf("请先配置短信模版")
		return
	}
	if areaCode == "86" {
		ok = SendSmsCode(mobile, verifyCode, tplId)
	} else {
		if gjTplId == "" {
			err = fmt.Errorf("请先配置国际短信模版")
			return
		}
		ok = SendSmsCodeGj(mobile, verifyCode, areaCode, gjTplId)
	}
	record.SendStatus = system.AdminVerifyCodeRecordStatusSuccess
	if !ok {
		record.SendStatus = system.AdminVerifyCodeRecordStatusFail
	}
	cols := []string{"SendStatus"}
	if e := record.Update(cols); e != nil {
		err = fmt.Errorf("更新验证码记录失败, Err: %s", e.Error())
	}
	return
}

// SendAdminEmailVerifyCode 发送用户邮箱验证码
func SendAdminEmailVerifyCode(source int, email string) (ok bool, err error) {
	verifyCode := utils.GetRandDigit(6)
	record := new(system.AdminVerifyCodeRecord)
	record.VerifyType = system.AdminVerifyCodeRecordTypeEmail
	record.Email = email
	record.Source = source
	record.Code = verifyCode
	record.ExpiredTime = time.Now().Add(utils.VerifyCodeExpireMinute * time.Minute)
	record.CreateTime = time.Now().Local()
	record.ModifyTime = time.Now().Local()
	if e := record.Create(); e != nil {
		err = fmt.Errorf("新增验证码记录失败, Err: %s", e.Error())
		return
	}

	// 获取邮件配置
	authKey := "english_report_email_conf"
	emailConf, e := company.GetConfigDetailByCode(authKey)
	if e != nil {
		err = fmt.Errorf("获取群发邮件权限失败, Err: %s", e.Error())
		return
	}
	if emailConf.ConfigValue == "" {
		err = fmt.Errorf("邮件配置为空, 不可推送")
		return
	}
	conf := new(models.EnglishReportEmailConf)
	if e = json.Unmarshal([]byte(emailConf.ConfigValue), &conf); e != nil {
		err = fmt.Errorf("邮件配置有误, 不可推送")
		return
	}

	// 获取邮箱模板
	// 获取配置好的短信模版
	cond := ` AND (conf_key = ? OR conf_key = ?)`
	pars := make([]interface{}, 0)
	pars = append(pars, "LoginEmailTemplateSubject", "LoginEmailTemplateContent")
	busiConf := new(models.BusinessConf)
	emailConfList, e := busiConf.GetItemsByCondition(cond, pars, []string{"conf_key, conf_val"}, "")
	if e != nil {
		if e.Error() == utils.ErrNoRow() {
			err = fmt.Errorf("请先配置邮件模版")
			return
		}
		err = fmt.Errorf("获取邮件模版失败, Err: %s", e.Error())
		return
	}
	var emaiContent, emailSubject string
	for _, v := range emailConfList {
		if v.ConfKey == "LoginEmailTemplateContent" {
			emaiContent = v.ConfVal
		} else if v.ConfKey == "LoginEmailTemplateSubject" {
			emailSubject = v.ConfVal
		}
	}
	if emailSubject == "" {
		err = fmt.Errorf("请先配置邮件模版主题")
		return
	}
	if emaiContent == "" {
		err = fmt.Errorf("请先配置邮件模版内容")
		return
	}

	req := new(EnglishReportSendEmailRequest)
	req.Subject = emailSubject
	req.Email = email
	// todo 发信人昵称
	req.FromAlias = conf.FromAlias
	// 填充模板
	t := time.Now().Format("2006年01月02日")
	ct := emaiContent
	ct = strings.Replace(ct, "{{VERIFY_CODE}}", verifyCode, 1)
	ct = strings.Replace(ct, "{{EXPIRED_MINUTE}}", strconv.Itoa(utils.VerifyCodeExpireMinute), 1)
	ct = strings.Replace(ct, "{{DATE_TIME}}", t, 1)
	req.HtmlBody = ct

	aliEmail := new(AliyunEmail)
	o, result, e := aliEmail.SendEmail(req)
	if e != nil {
		err = fmt.Errorf("邮箱推送失败, Err: %s", e.Error())
		return
	}
	ok = o

	record.SendStatus = system.AdminVerifyCodeRecordStatusSuccess
	if !ok {
		record.SendStatus = system.AdminVerifyCodeRecordStatusFail
	}
	record.SendResult = result
	cols := []string{"SendStatus", "SendResult"}
	if e = record.Update(cols); e != nil {
		err = fmt.Errorf("更新验证码记录失败, Err: %s", e.Error())
	}
	return
}

func ThirdLogin(req map[string]interface{}) (data GetCrmTokenData, err error, errMsg string) {
	if utils.BusinessCode == utils.BusinessCodeRelease || utils.BusinessCode == utils.BusinessCodeSandbox {
		authCode, ok := req["AuthCode"]
		if !ok {
			errMsg = "参数有误"
			err = errors.New("参数缺失, AuthCode")
			return
		}
		authCodeStr := fmt.Sprint(authCode)
		if authCodeStr == "" {
			errMsg = "参数有误"
			err = errors.New("参数缺失, AuthCode")
			return
		}

		data, err = CodeLoginFromMiddleServer(authCodeStr)

		return
	}

	data, err = ThirdCodeLoginFromMiddleServer(req)

	// 普通的第三方

	return
}

// ThirdCodeLoginFromMiddleServer 中间服务-编码登录
func ThirdCodeLoginFromMiddleServer(param map[string]interface{}) (tokenResp GetCrmTokenData, err error) {
	data, e := json.Marshal(param)
	if e != nil {
		err = fmt.Errorf("data json marshal err: %s", e.Error())
		return
	}

	body := io.NopCloser(strings.NewReader(string(data)))
	client := &http.Client{}
	req, e := http.NewRequest("POST", utils.EtaBridgeLoginUrl, body)
	if e != nil {
		err = fmt.Errorf("http create request err: %s", e.Error())
		return
	}

	contentType := "application/json;charset=utf-8"
	req.Header.Set("Content-Type", contentType)
	req.Header.Set("Authorization", utils.CrmEtaAuthorization)
	resp, e := client.Do(req)
	if e != nil {
		err = fmt.Errorf("http client do err: %s", e.Error())
		return
	}
	defer func() {
		_ = resp.Body.Close()
	}()
	b, e := io.ReadAll(resp.Body)
	if e != nil {
		err = fmt.Errorf("resp body read err: %s", e.Error())
		return
	}
	if len(b) == 0 {
		err = fmt.Errorf("resp body is empty")
		return
	}
	// 生产环境解密, 注意有个坑前后的双引号
	if utils.RunMode == "release" {
		str := string(b)
		str = strings.Trim(str, `"`)
		b = utils.DesBase64Decrypt([]byte(str), utils.DesKey)
	}

	result := new(GetCrmTokenDataResp)
	if e = json.Unmarshal(b, &result); e != nil {
		err = fmt.Errorf("result unmarshal err: %s\nresult: %s", e.Error(), string(b))
		return
	}
	if result.Code != 200 {
		err = fmt.Errorf("result: %s", string(b))
		return
	}
	tokenResp = result.Data
	return
}