HFish/core/alert/alert.go

package alert

import (
	"HFish/utils/try"
	"HFish/core/dbUtil"
	"strconv"
	"strings"
	"HFish/utils/send"
	"bytes"
	"net/http"
	"HFish/utils/log"
	"encoding/json"
	"HFish/view/data"
	"github.com/gin-gonic/gin"
	"HFish/error"
	"sync"
)

func AlertMail(model string, typex string, agent string, ipx string, country string, region string, city string, infox string, wg *sync.WaitGroup) {
	// 判断邮件通知
	try.Try(func() {
		// 只有新加入才会发送邮件通知
		if (model == "new") {
			result, err := dbUtil.DB().Table("hfish_setting").Fields("status", "info").Where("type", "=", "alertMail").First()

			if err != nil {
				log.Pr("HFish", "127.0.0.1", "获取邮件告警配置失败", err)
			}

			status := strconv.FormatInt(result["status"].(int64), 10)

			// 判断是否启用通知
			if status == "1" {
				info := result["info"]
				config := strings.Split(info.(string), "&&")

				if (country == "本地地址") {
					region = ""
					city = ""
				} else if (country == "局域网") {
					region = ""
					city = ""
				}

				text := `
				<div><b>Hi，上钩了！</b></div>
				<div><b><br /></b></div>
				<div><b>集群名称：</b>` + agent + `</div>
				<div><b>攻击IP：</b>` + ipx + `</div>
				<div><b>地理信息：</b>` + country + ` ` + region + ` ` + city + `</div>
				<div><b>上钩内容：</b>` + infox + `</div>
				<div><br /></div>
				<div><span style="color: rgb(128, 128, 128); font-size: 10px;">(HFish 自动发送)</span></div>
				`

				send.SendMail(config[4:], "[HFish]提醒你，"+typex+"有鱼上钩!", text, config)
			}
		}

		wg.Done()
	}).Catch(func() {
		wg.Done()
	})
}

func AlertWebHook(id string, model string, typex string, projectName string, agent string, ipx string, country string, region string, city string, infox string, time string, wg *sync.WaitGroup) {
	// 判断 WebHook 通知
	try.Try(func() {
		result, err := dbUtil.DB().Table("hfish_setting").Fields("status", "info").Where("type", "=", "webHook").First()

		if err != nil {
			log.Pr("HFish", "127.0.0.1", "获取WebHook配置失败", err)
		}

		status := strconv.FormatInt(result["status"].(int64), 10)

		// 判断是否启用通知
		if status == "1" {
			info := result["info"]

			song := make(map[string]interface{})
			song["id"] = id
			song["model"] = model
			song["project"] = projectName
			song["type"] = typex
			song["agent"] = agent
			song["ip"] = ipx
			song["country"] = country
			song["region"] = region
			song["city"] = city
			song["info"] = infox
			song["time"] = time

			bytesData, _ := json.Marshal(song)

			reader := bytes.NewReader(bytesData)

			request, _ := http.NewRequest("POST", info.(string), reader)
			request.Header.Set("Content-Type", "application/json;charset=UTF-8")

			client := http.Client{}
			resp, err := client.Do(request)

			if err != nil {
				log.Pr("HFish", "127.0.0.1", "WebHook 调用失败", err)
			} else {
				log.Pr("HFish", "127.0.0.1", "WebHook 调用成功")
			}

			defer resp.Body.Close()
			//defer request.Body.Close()
		}

		wg.Done()
	}).Catch(func() {
		wg.Done()
	})
}

// 大数据展示
func AlertDataWs(model string, typex string, projectName string, agent string, ipx string, country string, region string, city string, time string) {
	if (model == "new") {
		// 拼接字典
		d := data.MakeDataJson("center_data", map[string]interface{}{
			"type":        typex,
			"projectName": projectName,
			"agent":       agent,
			"ipx":         ipx,
			"country":     country,
			"region":      region,
			"city":        city,
			"time":        time,
		})

		// 发送到客户端
		data.Send(gin.H{
			"code": error.ErrSuccessCode,
			"msg":  error.ErrSuccessMsg,
			"data": d,
		})
	}
}