Message-Push-Nest/service/send_message_service/send_message.go

package send_message_service

import (
	"fmt"
	"message-nest/models"
	"message-nest/pkg/logging"
	"message-nest/pkg/message"
	"message-nest/service/send_ins_service"
	"message-nest/service/send_task_service"
	"message-nest/service/send_way_service"
	"strings"
)

type SendMessageService struct {
	TaskID   string
	Text     string
	HTML     string
	MarkDown string
}

func (sm *SendMessageService) Send() string {
	var logOutput []string
	status := 1

	sendTaskService := send_task_service.SendTaskService{
		ID: sm.TaskID,
	}
	task, err := sendTaskService.GetTaskWithIns()
	if err != nil {
		return fmt.Sprintf("任务不存在！任务id: %s", sm.TaskID)
	}

	for idx, ins := range task.InsData {
		way, err := models.GetWayByID(ins.WayID)
		if err != nil {
			logOutput = append(logOutput, fmt.Sprintf("渠道信息不存在！渠道id：%s", ins.WayID))
		}
		wayService := send_way_service.SendWay{
			ID:   fmt.Sprintf("%s", way.ID),
			Name: way.Name,
			Auth: way.Auth,
			Type: way.Type,
		}

		logOutput = append(logOutput, fmt.Sprintf(">> 实例 %d", idx+1))
		logOutput = append(logOutput, fmt.Sprintf("开始发送，实例: %s", ins.WayID))
		logOutput = append(logOutput, fmt.Sprintf("实例类型: %s + %s", ins.WayType, ins.ContentType))
		logOutput = append(logOutput, fmt.Sprintf("实例配置: %s", ins.Config))

		errStr, msgObj := wayService.ValidateDiffWay()
		if errStr != "" {
			sm.MarkStatus(errStr, &status)
			logOutput = append(logOutput, fmt.Sprintf("实例渠道认证校验失败: %s", errStr))
			continue
		}

		// 邮箱类型的实例
		emailAuth, ok := msgObj.(send_way_service.WayDetailEmail)
		if ok {
			errMsg := sm.SendTaskEmail(emailAuth, ins.SendTasksIns)
			sm.MarkStatus(errMsg, &status)
			logOutput = append(logOutput, sm.TransError(errMsg))
			continue
		}

		logOutput = append(logOutput, fmt.Sprintf("未知渠道的发信实例: %s", ins.ID))

	}

	logOutput = sm.FormatSendContent(logOutput)
	sm.RecordSendLog(logOutput, status)

	return ""
}

// FormatSendContent 格式化输出的发送内容
func (sm *SendMessageService) FormatSendContent(logOutput []string) []string {
	logOutput = append(logOutput, fmt.Sprintf(">> 发送的内容:"))
	if sm.Text != "" {
		logOutput = append(logOutput, fmt.Sprintf("Text: %s", sm.Text))
	}
	if sm.HTML != "" {
		logOutput = append(logOutput, fmt.Sprintf("HTML: %s", sm.HTML))
	}
	if sm.MarkDown != "" {
		logOutput = append(logOutput, fmt.Sprintf("MarkDown: %s", sm.MarkDown))
	}
	return logOutput
}

// MarkStatus 标记任务状态
func (sm *SendMessageService) MarkStatus(errStr string, status *int) {
	if errStr != "" {
		*status = 0
	}
}

// RecordSendLog 记录发送日志
func (sm *SendMessageService) RecordSendLog(logOutput []string, status int) {
	if len(logOutput) <= 0 {
		return
	}
	log := models.SendTasksLogs{
		Log:    strings.Join(logOutput, "\n"),
		TaskID: sm.TaskID,
		Status: status,
	}
	err := log.Add()
	if err != nil {
		logging.Logger.Error(fmt.Sprintf("添加日志失败！原因是：%s", err))
	}
}

// TransError 转化错误
func (sm *SendMessageService) TransError(err string) string {
	if err == "" {
		return "发送成功！\n"
	} else {
		return fmt.Sprintf("发送失败：%s！", err)
	}
}

// GetSendMsg
func (sm *SendMessageService) GetSendMsg(ins models.SendTasksIns) string {
	data := map[string]string{}
	data["text"] = sm.Text
	data["html"] = sm.HTML
	data["markdown"] = sm.MarkDown
	content, ok := data[ins.ContentType]
	if !ok || len(content) == 0 {
		content, ok := data["text"]
		if !ok {
			logging.Logger.Error("text节点数据为空！")
			return ""
		} else {
			return content
		}
	} else {
		return content
	}
}

// SendTaskEmail 执行发送邮件
func (sm *SendMessageService) SendTaskEmail(auth send_way_service.WayDetailEmail, ins models.SendTasksIns) string {
	insService := send_ins_service.SendTaskInsService{}
	errStr, c := insService.ValidateDiffIns(ins)
	if errStr != "" {
		return errStr
	}
	config, ok := c.(models.InsEmailConfig)
	if !ok {
		return "邮箱config校验失败"
	}
	var emailer message.EmailMessage
	emailer.Init(auth.Server, auth.Port, auth.Account, auth.Passwd)
	errMsg := emailer.SendTextMessage(config.ToAccount, config.Title, sm.GetSendMsg(ins))
	return errMsg
}