hummingbird/internal/tools/notify/qiyeweixin/client.go

/*******************************************************************************
 * Copyright 2017.
 *
 * Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except
 * in compliance with the License. You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software distributed under the License
 * is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express
 * or implied. See the License for the specific language governing permissions and limitations under
 * the License.
 *******************************************************************************/

package yiqiweixin

import (
	"encoding/json"
	"fmt"
	"github.com/kirinlabs/HttpRequest"
	"github.com/winc-link/hummingbird/internal/dtos"
	"github.com/winc-link/hummingbird/internal/models"
	"github.com/winc-link/hummingbird/internal/pkg/constants"
	"github.com/winc-link/hummingbird/internal/pkg/di"
	"github.com/winc-link/hummingbird/internal/pkg/logger"
	"time"
)

type WeixinClient struct {
	lc logger.LoggingClient
	p  *di.Container
}

func NewWeiXinClient(lc logger.LoggingClient, p *di.Container) *WeixinClient {
	return &WeixinClient{
		lc: lc,
		p:  p,
	}
}

type QiYeWeiXinTemplate struct {
	Msgtype  string   `json:"msgtype"`
	Markdown Markdown `json:"markdown"`
}

type Markdown struct {
	Content string `json:"content"`
}

func (d *WeixinClient) generateWeiXinTemplate(rule models.Rule, device models.Device, product models.Product, ruleName string) QiYeWeiXinTemplate {
	var qt QiYeWeiXinTemplate
	qt.Msgtype = "markdown"
	switch rule.Trigger {
	case constants.DeviceDataTrigger:
		var code string
		var propertyName string
		var data interface{}
		var unit string
		if rule.Option["code"] != "" {
			for _, property := range product.Properties {
				if property.Code == rule.Option["code"] {
					code = property.Code
					propertyName = property.Name
					break
				}
			}
			var persisterReq dtos.ThingModelPropertyDataRequest
			persisterReq.ThingModelDataBaseRequest.Last = true
			persisterReq.DeviceId = device.Id
			persisterReq.Code = rule.Option["code"]
		}
		if data == nil {
			qt.Markdown.Content = fmt.Sprintf(`设备已触发数据告警
         >设备名称:<font color=\"comment\">%s</font>
         >告警规则:<font color=\"comment\">%s</font>
		 >属性名称:<font color=\"comment\">%s</font>
 		 >触发时间:<font color=\"comment\">%s</font>
      `, device.Name, ruleName, fmt.Sprintf(propertyName+"[%s]", code),
				time.Now().Format("2006-01-02 15:04:05"))
		} else {
			qt.Markdown.Content = fmt.Sprintf(`设备已触发数据告警
         >设备名称:<font color=\"comment\">%s</font>
         >告警规则:<font color=\"comment\">%s</font>
		 >属性名称:<font color=\"comment\">%s</font>
         >当前属性值:<font color=\"comment\">%s</font>
 		 >触发时间:<font color=\"comment\">%s</font>
      `, device.Name, ruleName, fmt.Sprintf(propertyName+"[%s]", code),
				fmt.Sprintf("%s(%s)", data, unit),
				time.Now().Format("2006-01-02 15:04:05"))
		}
	case constants.DeviceEventTrigger:
		var code string
		var eventName string
		var eventType string
		var data interface{}
		if rule.Option["code"] != "" {
			for _, event := range product.Events {
				if event.Code == rule.Option["code"] {
					code = event.Code
					eventName = event.Name
					eventType = event.EventType
					break
				}
			}
			var persisterReq dtos.ThingModelEventDataRequest
			persisterReq.ThingModelDataBaseRequest.Last = true
			persisterReq.DeviceId = device.Id
			persisterReq.EventCode = rule.Option["code"]
			persisterReq.EventType = eventType
		}
		if data == nil {
			qt.Markdown.Content = fmt.Sprintf(`设备已触发事件告警
         >设备名称:<font color=\"comment\">%s</font>
         >告警规则:<font color=\"comment\">%s</font>
		 >事件名称:<font color=\"comment\">%s</font>
 	     >触发时间:<font color=\"comment\">%s</font>
         `, device.Name, ruleName, fmt.Sprintf(eventName+"[%s]", code), time.Now().Format("2006-01-02 15:04:05"))
		} else {
			qt.Markdown.Content = fmt.Sprintf(`设备已触发事件告警
         >设备名称:<font color=\"comment\">%s</font>
         >告警规则:<font color=\"comment\">%s</font>
		 >事件名称:<font color=\"comment\">%s</font>
         >事件内容:<font color=\"comment\">%s</font>
 	     >触发时间:<font color=\"comment\">%s</font>
         `, device.Name, ruleName, fmt.Sprintf(eventName+"[%s]", code), data, time.Now().Format("2006-01-02 15:04:05"))
		}
	case constants.DeviceStatusTrigger:
		qt.Markdown.Content = fmt.Sprintf(`设备已触发状态告警
         >设备名称:<font color=\"comment\">%s</font>
         >告警规则:<font color=\"comment\">%s</font>
		 >设备状态:<font color=\"comment\">%s</font>`, device.Name, ruleName, device.Status)
	}
	return qt
}

func (d *WeixinClient) Send(webhook string, rule models.Rule, device models.Device, product models.Product, ruleName string) {
	if webhook == "" {
		return
	}
	req := HttpRequest.NewRequest()
	req.JSON()
	context, _ := json.Marshal(d.generateWeiXinTemplate(rule, device, product, ruleName))
	d.lc.Info("weixin sent context", string(context))
	resp, err := req.Post(webhook, context)
	if err != nil {
		d.lc.Errorf("weixin send alert message error:", err.Error())
	}
	body, err := resp.Body()
	if err != nil {
		return
	}
	d.lc.Debug("weixin send message", string(body))
}