hongze_yb_en_api/services/aliyun_email.go

package services

import (
	"context"
	"encoding/json"
	openapi "github.com/alibabacloud-go/darabonba-openapi/v2/client"
	dm "github.com/alibabacloud-go/dm-20151123/v2/client"
	util "github.com/alibabacloud-go/tea-utils/v2/service"
	"github.com/alibabacloud-go/tea/tea"
	"hongze/hongze_yb_en_api/global"
	"hongze/hongze_yb_en_api/utils"
	"time"
)

var (
	AliyunEmailAccountName       = "ficcemail@hzmail.hzinsights.com"
	AliyunEmailAccessKeyId       = "LTAIFMZYQhS2BTvW"
	AliyunEmailAccessKeySecret   = "12kk1ptCHoGWedhBnKRVW5hRJzq9Fq"
	AliyunEmailReplyAddress      = "ficcemail@hzinsights.com"
	AliyunEmailReplyAddressAlias = "弘则研究"
)

// AliyunEmail 阿里云邮件
type AliyunEmail struct {
	Client *dm.Client
}

// NewClient
func (em *AliyunEmail) NewClient() (err error) {
	config := &openapi.Config{
		AccessKeyId:     tea.String(AliyunEmailAccessKeyId),
		AccessKeySecret: tea.String(AliyunEmailAccessKeySecret),
	}
	// 访问的域名
	config.Endpoint = tea.String("dm.aliyuncs.com")
	_result, _err := dm.NewClient(config)
	em.Client = _result
	err = _err
	return
}

// AliyunEmailResult 邮件推送响应体
type AliyunEmailResult struct {
	Code        string `description:"错误码"`
	StatusCode  int    `description:"状态码"`
	Message     string `description:"响应信息"`
	Data        string `description:"请求体"`
	Description string `description:"描述"`
	EnvId       string `description:"请求成功-事件ID"`
	RequestId   string `description:"请求成功-请求ID"`
}

// AliyunEmailResultData 邮件推送响应内容
type AliyunEmailResultData struct {
	Code       string `description:"错误码"`
	HostId     string `description:""`
	Message    string `description:"响应信息"`
	Recommend  string `description:""`
	RequestId  string `description:"请求ID"`
	StatusCode int    `description:"状态码"`
}
//
//// SendEmail 邮件推送
//func (em *AliyunEmail) SendEmail(item *english_report.EnglishReportSendEmailRequest) (ok bool, result string, err error) {
//	if em.Client == nil {
//		if e := em.NewClient(); e != nil {
//			err = e
//			return
//		}
//	}
//	singleSendMailRequest := &dm.SingleSendMailRequest{}
//	singleSendMailRequest.SetAccountName(AliyunEmailAccountName)
//	singleSendMailRequest.SetAddressType(1)
//	singleSendMailRequest.SetReplyToAddress(false)
//	singleSendMailRequest.SetSubject(item.Subject)
//	singleSendMailRequest.SetToAddress(item.Email)
//	singleSendMailRequest.SetHtmlBody(item.HtmlBody)
//	singleSendMailRequest.SetFromAlias(item.FromAlias)
//
//	runtime := &util.RuntimeOptions{}
//	tryErr := func() error {
//		res, e := em.Client.SingleSendMailWithOptions(singleSendMailRequest, runtime)
//		if e != nil {
//			return e
//		}
//		// 请求成功
//		if tea.Int32Value(res.StatusCode) == 200 {
//			ok = true
//		}
//		resByte, e := json.Marshal(res.Body)
//		if e != nil {
//			return e
//		}
//		result = string(resByte)
//		return nil
//	}()
//
//	if tryErr != nil {
//		var e = &tea.SDKError{}
//		if t, ok := tryErr.(*tea.SDKError); ok {
//			e = t
//		} else {
//			e.Message = tea.String(tryErr.Error())
//		}
//		err = e
//		errByte, _ := json.Marshal(err)
//		result = string(errByte)
//	}
//	return
//}
//
//// BatchSendEmail 批量推送邮件
//func (em *AliyunEmail) BatchSendEmail(list []*english_report.EnglishReportSendEmailRequest) (results []*english_report.EnglishReportSendEmailResult, err error) {
//	results = make([]*english_report.EnglishReportSendEmailResult, 0)
//	if len(list) == 0 {
//		return
//	}
//	if e := em.NewClient(); e != nil {
//		err = e
//		return
//	}
//
//	// sendSingleMail接口有QPS100的限制, 保险起见每秒只请求50次接口
//	max := 50
//	c := 0
//	for i := range list {
//		if c >= max {
//			time.Sleep(time.Second)
//			c = 0
//		}
//		c += 1
//
//		dataByte, _ := json.Marshal(list[i])
//		ok, result, _ := em.SendEmail(list[i])
//		results = append(results, &english_report.EnglishReportSendEmailResult{
//			ReportId:   list[i].ReportId,
//			EmailId:    list[i].EmailId,
//			Email:      list[i].Email,
//			Ok:         ok,
//			SendData:   string(dataByte),
//			ResultData: result,
//			Source:     1,
//		})
//	}
//	return
//}

// sendEmailVerificationCode 邮件验证码推送
func (em *AliyunEmail) sendEmailVerificationCode(email, content string) (ok bool, result string, err error) {
	if em.Client == nil {
		if e := em.NewClient(); e != nil {
			err = e
			return
		}
	}

	singleSendMailRequest := &dm.SingleSendMailRequest{}
	singleSendMailRequest.SetAccountName(AliyunEmailAccountName)
	singleSendMailRequest.SetAddressType(1)
	singleSendMailRequest.SetReplyToAddress(false)
	singleSendMailRequest.SetSubject("Your Email Verification Code")
	singleSendMailRequest.SetToAddress(email)
	singleSendMailRequest.SetHtmlBody(content)
	singleSendMailRequest.SetFromAlias("Horizon FICC")

	runtime := &util.RuntimeOptions{}
	tryErr := func() error {
		res, e := em.Client.SingleSendMailWithOptions(singleSendMailRequest, runtime)
		if e != nil {
			return e
		}
		// 请求成功
		if tea.Int32Value(res.StatusCode) == 200 {
			ok = true
		}
		resByte, e := json.Marshal(res.Body)
		if e != nil {
			return e
		}
		result = string(resByte)
		return nil
	}()

	if tryErr != nil {
		var e = &tea.SDKError{}
		if t, ok := tryErr.(*tea.SDKError); ok {
			e = t
		} else {
			e.Message = tea.String(tryErr.Error())
		}
		err = e
		errByte, _ := json.Marshal(err)
		result = string(errByte)
	}
	return
}

// SendEmailVerificationCode 邮件验证码
func (em *AliyunEmail) SendEmailCode(name, email string) (results []*EnglishReportSendEmailCodeResult, err error) {
	results = make([]*EnglishReportSendEmailCodeResult, 0)

	if e := em.NewClient(); e != nil {
		err = e
		return
	}

	msgCode := utils.GetRandDigit(4)
	content := "Hi " + name + "：</br>Please enter this verification code on the registration page so we can make sure it's you：</br><p>" + msgCode + "</p>This code will expire in 15 minutes.</br>" +
		"If you didn't initiate this request, or need help with your registration, please let us know at: stephanie@hzinsights.com </br>Horizon Research Team"


	dataByte, _ := json.Marshal(content)
	ok, result, _ := em.sendEmailVerificationCode(email, content)
	results = append(results, &EnglishReportSendEmailCodeResult{
		Email:      email,
		Ok:         ok,
		SendData:   string(dataByte),
		ResultData: result,
		Source:     1,
	})

	_ = global.Redis.SetEX(context.TODO(), email, msgCode, 15*time.Minute)

	return
}


// EnglishReportSendEmailCodeResult 英文研报验证码-推送邮件响应体
type EnglishReportSendEmailCodeResult struct {
	Email      string `description:"邮箱地址"`
	Ok         bool   `description:"是否推送成功"`
	SendData   string `description:"请求数据-JSON"`
	ResultData string `description:"推送结果-JSON"`
	Source     int    `description:"服务来源：1-阿里云；2-腾讯云"`
}