eta_mini_bridge/controllers/wechat.go

package controllers

import (
	"encoding/json"
	"eta/eta_mini_bridge/models"
	"eta/eta_mini_bridge/services"
	"eta/eta_mini_bridge/services/alarm_msg"
	"eta/eta_mini_bridge/utils"
	"fmt"
	"html"
	"strconv"
	"time"
)

type WeChatController struct {
	BaseAuthController
}

// @Title 发送微信模板接口
// @Description 发送微信模板接口
// @Param	ReportId query int true "报告id"
// @Success 200 {object} models.WechatSign
// @router /send_template_msg [get]
func (this *WeChatController) SendTemplateMsg() {
	br := new(models.BaseResponse).Init()
	defer func() {
		if br.Ret != 200 {
			b, _ := json.Marshal(br)
			alarm_msg.SendAlarmMsg(string(b), 1)
		}
		this.Data["json"] = br
		this.ServeJSON()
	}()
	reportId, _ := this.GetInt("ReportId")
	if reportId <= 0 {
		br.Msg = "参数错误"
		br.ErrMsg = "参数错误，报告id小于等于0"
		return
	}
	report, err := models.GetReportById(reportId)
	if err != nil {
		br.Msg = "该报告已删除"
		br.ErrMsg = "获取报告详情失败,Err:" + err.Error()
		return
	}
	report.ContentSub = html.UnescapeString(report.ContentSub)
	report.Content = html.UnescapeString(report.Content)
	if report == nil {
		br.Msg = "报告不存在"
		return
	}
	reportChartPermissionIdStrs, err := models.GetChartPermissionIdsListByClassifyId(report.ClassifyIdSecond)
	if err != nil {
		br.Msg = "获取研报权限失败"
		br.ErrMsg = "获取研报权限失败，Err:" + err.Error()
		return
	}
	chartPermissionList, err := models.GetChartPermissionIdsByIds(reportChartPermissionIdStrs)
	if err != nil {
		br.Msg = "获取研报权限失败"
		br.ErrMsg = "获取研报权限失败，Err:" + err.Error()
		return
	}
	permissionMapping, err := models.GetUserChartPermissionMapping()
	if err != nil {
		br.Msg = "获取用户权限失败"
		br.ErrMsg = "获取用户权限失败，Err:" + err.Error()
		return
	}
	// 获取不同二级品种分类下的的小程序用户列表映射
	permissionMap := make(map[int][]int)
	for _, v := range permissionMapping {
		if _, ok := permissionMap[v.ChartPermissionId]; !ok {
			permissionMap[v.ChartPermissionId] = make([]int, 0)
		} else {
			permissionMap[v.ChartPermissionId] = append(permissionMap[v.ChartPermissionId], v.UserId)
		}
	}
	var IsPublic bool
	for _, v := range chartPermissionList {
		if v.IsPublic == 1 {
			IsPublic = true
			break
		}
	}
	userList, err := models.GetUserBySubscribe()
	if err != nil {
		br.Msg = "获取用户列表失败"
		br.ErrMsg = "获取用户列表失败，Err:" + err.Error()
		return
	}
	userUnionMap := make(map[int]string)
	for _, v := range userList {
		userUnionMap[v.UserId] = v.UnionId
	}
	var openIds []*services.OpenIdList
	if IsPublic {
		var unionIds []string
		for _, v := range userList {
			unionIds = append(unionIds, v.UnionId)
		}
		// 给所有人，发送模板消息
		records, err := models.GetUserRecordByUnionids(unionIds)
		if err != nil {
			br.Msg = "获取用户记录失败"
			br.ErrMsg = "获取用户记录失败，Err:" + err.Error()
			return
		}
		for _, v := range records {
			openIds = append(openIds, &services.OpenIdList{
				OpenId: v.OpenId,
				UserId: v.UserId,
			})
		}
	} else {
		// 给指定用户，发送模板消息
		// 报告所属的二级品种分类
		var permissionIds []int
		for _, v := range reportChartPermissionIdStrs {
			vv, _ := strconv.Atoi(v)
			permissionIds = append(permissionIds, vv)
		}
		var sendUserIds []int
		// 获取报告所属的二级品种分类对应的用户列表
		for _, v := range permissionIds {
			if _, ok := permissionMap[v]; ok {
				sendUserIds = append(sendUserIds, permissionMap[v]...)
			}
		}
		sendUserIds = utils.Unique(sendUserIds)
		var unionIds []string
		for _, v := range sendUserIds {
			if id, ok := userUnionMap[v]; ok {
				unionIds = append(unionIds, id)
			}
		}
		records, err := models.GetUserRecordByUnionids(unionIds)
		if err != nil {
			br.Msg = "获取用户记录失败"
			br.ErrMsg = "获取用户记录失败，Err:" + err.Error()
			return
		}
		for _, v := range records {
			openIds = append(openIds, &services.OpenIdList{
				OpenId: v.OpenId,
				UserId: v.UserId,
			})
		}
	}
	sendMap := make(map[string]interface{})
	sendMap["template_id"] = utils.TEMPLATE_ID_BY_PRODUCT
	// sendMap["first"] = fmt.Sprintf("Hi,最新一期%s已上线，欢迎查看", report.ClassifyNameFirst)
	// sendMap["keyword1"] = map[string]interface{}{"value": fmt.Sprintf("弘则%s", report.ClassifyNameSecond), "color": "#173177"}
	// sendMap["keyword2"] = map[string]interface{}{"value": report.ContentSub, "color": "#173177"}
	sendMap["first"] = "测试first"
	sendMap["keyword1"] = "测试k1"
	sendMap["keyword2"] = "测试k2"
	sendMap["remark"] = time.Now().Format("2006-01-02")
	// sendMap["keyword3"] = map[string]interface{}{"value": report.PublishTime, "color": "#173177"}
	// sendMap["keyword4"] = map[string]interface{}{"value": report.Abstract, "color": "#173177"}
	var WxAppId = `wxb059c872d79b9967`
	// var WxAppSecret = `1737c73e9f69a21de1a345b8f0800258`
	var wxAppPath string
	if utils.RunMode == "debug" {
		wxAppPath = fmt.Sprintf("pages-question/answerDetail?id=%d", 3800)
	} else {
		wxAppPath = fmt.Sprintf("pages-report/reportDetail/index?id=%d", report.Id)
	}
	sendMap["miniprogram"] = map[string]string{
		"appid":    WxAppId,
		"pagepath": wxAppPath,
	}
	alarm_msg.SendAlarmMsg(fmt.Sprintf("发送模板消息，报告id:%d", report.Id), 1)
	err = services.SendMultiTemplateMsg(sendMap, openIds, 1)
	if err != nil {
		br.Msg = "发送失败"
		br.ErrMsg = "发送失败，Err:" + err.Error()
		return
	}

	br.Ret = 200
	br.Msg = "发送成功"
	br.Success = true
}