初始化

Author: zy84338719

biz/bot/dingtalk/action_card.go

@@ -0,0 +1,50 @@
+package dingtalk
+
+type ActionCardMessage struct {
+	MsgType    MsgType     `json:"msgtype"`
+	ActionCard *ActionCard `json:"actionCard"`
+}
+
+type ActionCard struct {
+	Title          string `json:"title"`
+	Text           string `json:"text"`
+	HideAvatar     string `json:"hideAvatar"`
+	BtnOrientation string `json:"btnOrientation"`
+	SingleTitle    string `json:"singleTitle"`
+	SingleURL      string `json:"singleURL"`
+	Btns           []*Btn `json:"btns"`
+}
+
+/*** ActionCard ***/
+func NewActionCard() *ActionCardMessage {
+	return &ActionCardMessage{ACTIONCARD, &ActionCard{}}
+}
+
+func (a *ActionCardMessage) SetContent(title, text string) *ActionCardMessage {
+	a.ActionCard.Title = title
+	a.ActionCard.Text = text
+	return a
+}
+
+func (a *ActionCardMessage) AddBtn(singleTitle, singleURL string) *ActionCardMessage {
+	a.ActionCard.SingleTitle = singleTitle
+	a.ActionCard.SingleURL = singleURL
+	return a
+}
+
+func (a *ActionCardMessage) AddBtns(btns [][]string) *ActionCardMessage {
+	for _, item := range btns {
+		a.ActionCard.Btns = append(a.ActionCard.Btns, &Btn{item[0], item[1]})
+	}
+	return a
+}
+
+func (a *ActionCardMessage) HideAvatar() *ActionCardMessage {
+	a.ActionCard.HideAvatar = "1"
+	return a
+}
+
+func (a *ActionCardMessage) BtnOrientation() *ActionCardMessage {
+	a.ActionCard.BtnOrientation = "0"
+	return a
+}

biz/bot/dingtalk/bot.go

@@ -0,0 +1,119 @@
+package dingtalk
+
+import (
+	"bytes"
+	"crypto/hmac"
+	"crypto/sha256"
+	"encoding/base64"
+	"encoding/json"
+	"errors"
+	"fmt"
+	"github.com/yi-nology/common/utils/xjson"
+	"io"
+	"io/ioutil"
+	"net/http"
+	"net/url"
+	"strconv"
+	"time"
+)
+
+type Robot struct {
+	Key        string
+	RequestUrl string
+	Client     *http.Client
+}
+
+type Resp struct {
+	ErrCode int    `json:"errcode"`
+	ErrMsg  string `json:"errmsg"`
+}
+
+const (
+	WEBHOOK_URL = "https://oapi.dingtalk.com/robot/send?access_token=%s"
+)
+
+// New 获取一个 Robot instance
+func New(botKey string) *Robot {
+	return &Robot{
+		Key:        botKey,
+		RequestUrl: fmt.Sprintf(WEBHOOK_URL, botKey),
+		Client:     &http.Client{Timeout: 5 * time.Second},
+	}
+}
+
+// AddSign 更新SDK 支持Sign模式
+func (r *Robot) AddSign(secret string) *Robot {
+	if len(secret) == 0 {
+		return r
+	}
+	microTimestamp := time.Now().UnixNano() / 1e6
+	h := hmac.New(sha256.New, []byte(secret))
+	io.WriteString(h, fmt.Sprintf("%d\n%s", microTimestamp, secret))
+	if r.RequestUrl != "" {
+		r.RequestUrl = r.RequestUrl + "&timestamp=" + strconv.Itoa(int(microTimestamp)) + "&sign=" + url.QueryEscape(base64.StdEncoding.EncodeToString(h.Sum(nil)))
+	}
+	return r
+}
+
+// Send 发送notification
+func (r *Robot) Send(message interface{}) (bool, error) {
+	b, err := json.Marshal(message)
+	if err != nil {
+		return false, err
+	}
+	// log.Println(string(b))
+	return r.SendRaw(b)
+}
+
+func (r *Robot) SendRaw(msgBytes []byte) (bool, error) {
+	resp, err := r.Client.Post(r.RequestUrl, "application/json", bytes.NewReader(msgBytes))
+	if err != nil {
+		return false, err
+	}
+
+	defer resp.Body.Close()
+	body, err := ioutil.ReadAll(resp.Body)
+	if err != nil {
+		return false, err
+	}
+
+	ret := &Resp{}
+
+	if err := json.Unmarshal(body, &ret); err != nil {
+		return false, err
+	}
+
+	if ret.ErrCode == 0 {
+		return true, nil
+	} else {
+		return false, errors.New(ret.ErrMsg)
+	}
+}
+
+func (r *Robot) CheckMessage(msg string) bool {
+
+	if len(msg) == 0 {
+		return false
+	}
+	text := TextMessage{}
+	markdown := MarkDownMessage{}
+	link := LinkMessage{}
+	feedcard := FeedCardMessage{}
+	actionCard := ActionCardMessage{}
+	if xjson.UnmarshalFromString(msg, &text) == nil {
+		return true
+	}
+	if xjson.UnmarshalFromString(msg, &markdown) == nil {
+		return true
+	}
+	if xjson.UnmarshalFromString(msg, &link) == nil {
+		return true
+	}
+	if xjson.UnmarshalFromString(msg, &feedcard) == nil {
+		return true
+	}
+	if xjson.UnmarshalFromString(msg, &actionCard) == nil {
+		return true
+	}
+	return false
+}

biz/bot/dingtalk/dingtalk.go

@@ -0,0 +1,19 @@
+package dingtalk
+
+type MsgType string
+
+const TEXT MsgType = "text"
+const LINK MsgType = "link"
+const MARKDOWN MsgType = "markdown"
+const ACTIONCARD MsgType = "actionCard"
+const FEEDCARD MsgType = "feedCard"
+
+type At struct {
+	AtMobiles []string `json:"atMobiles"`
+	IsAtAll   bool     `json:"isAtAll"`
+}
+
+type Btn struct {
+	Title     string `json:"title"`
+	ActionURL string `json:"actionURL"`
+}

biz/bot/dingtalk/feed_card.go

@@ -0,0 +1,33 @@
+package dingtalk
+
+type FeedCardMessage struct {
+	MsgType  MsgType   `json:"msgtype"`
+	FeedCard *FeedCard `json:"feedCard"`
+}
+
+type FeedCard struct {
+	Links []*FeedCardLink `json:"links"`
+}
+
+type FeedCardLink struct {
+	Title      string `json:"title"`
+	MessageURL string `json:"messageURL"`
+	PicURL     string `json:"picURL"`
+}
+
+/*** FeedCard ***/
+func NewFeedCard() *FeedCardMessage {
+	return &FeedCardMessage{FEEDCARD, &FeedCard{}}
+}
+
+func (f *FeedCardMessage) AddCard(title, messageURL, picURL string) *FeedCardMessage {
+	f.FeedCard.Links = append(f.FeedCard.Links, &FeedCardLink{title, messageURL, picURL})
+	return f
+}
+
+func (f *FeedCardMessage) AddCards(cards [][]string) *FeedCardMessage {
+	for _, item := range cards {
+		f.FeedCard.Links = append(f.FeedCard.Links, &FeedCardLink{item[0], item[1], item[2]})
+	}
+	return f
+}

biz/bot/dingtalk/link.go

@@ -0,0 +1,26 @@
+package dingtalk
+
+type LinkMessage struct {
+	MsgType MsgType `json:"msgtype"`
+	Link    *Link   `json:"link"`
+}
+
+type Link struct {
+	Title      string `json:"title"`
+	Text       string `json:"text"`
+	MessageUrl string `json:"messageUrl"`
+	PicUrl     string `json:"picUrl"`
+}
+
+/*** Link ***/
+func NewLink() *LinkMessage {
+	return &LinkMessage{LINK, &Link{}}
+}
+
+func (l *LinkMessage) SetContent(title, text, messageUrl, picUrl string) *LinkMessage {
+	l.Link.Title = title
+	l.Link.Text = text
+	l.Link.MessageUrl = messageUrl
+	l.Link.PicUrl = picUrl
+	return l
+}

biz/bot/dingtalk/markdown.go

@@ -0,0 +1,33 @@
+package dingtalk
+
+type MarkDownMessage struct {
+	MsgType  MsgType   `json:"msgtype"`
+	At       *At       `json:"at"`
+	MarkDown *MarkDown `json:"markdown"`
+}
+
+type MarkDown struct {
+	Title string `json:"title"`
+	Text  string `json:"text"`
+}
+
+/*** markdown ***/
+func NewMarkDown() *MarkDownMessage {
+	return &MarkDownMessage{MARKDOWN, &At{}, &MarkDown{}}
+}
+
+func (m *MarkDownMessage) SetContent(title, text string) *MarkDownMessage {
+	m.MarkDown.Title = title
+	m.MarkDown.Text = text
+	return m
+}
+
+func (m *MarkDownMessage) AtMobiles(mobiles []string) *MarkDownMessage {
+	m.At.AtMobiles = mobiles
+	return m
+}
+
+func (m *MarkDownMessage) AtAll() *MarkDownMessage {
+	m.At.IsAtAll = true
+	return m
+}

biz/bot/dingtalk/text.go

@@ -0,0 +1,31 @@
+package dingtalk
+
+type TextMessage struct {
+	MsgType MsgType `json:"msgtype"`
+	At      *At     `json:"at"`
+	Text    *Text   `json:"text"`
+}
+
+type Text struct {
+	Content string `json:"content"`
+}
+
+/*** Text ***/
+func NewText() *TextMessage {
+	return &TextMessage{TEXT, &At{}, &Text{}}
+}
+
+func (t *TextMessage) SetContent(content string) *TextMessage {
+	t.Text.Content = content
+	return t
+}
+
+func (t *TextMessage) AtMobiles(mobiles []string) *TextMessage {
+	t.At.AtMobiles = mobiles
+	return t
+}
+
+func (t *TextMessage) AtAll() *TextMessage {
+	t.At.IsAtAll = true
+	return t
+}

biz/bot/interface.go

@@ -0,0 +1,30 @@
+package bot
+
+import (
+	"github.com/yi-nology/common/biz/bot/dingtalk"
+	"github.com/yi-nology/common/biz/bot/wxwork"
+)
+
+type BotOne interface {
+	Send(interface{}) (bool, error)        // 发送
+	SendRaw(msgBytes []byte) (bool, error) // 发送原始消息
+	CheckMessage(msg string) bool          // 检查消息是否合法
+}
+
+type BotType string
+
+const (
+	WXWork   BotType = "wx"
+	Dingtalk BotType = "dd"
+)
+
+func SwitchOne(botType BotType, botKey string, secret string) BotOne {
+	switch botType {
+	case WXWork:
+		return wxwork.New(botKey)
+	case Dingtalk:
+		return dingtalk.New(botKey).AddSign(secret)
+	default:
+		return wxwork.New(botKey)
+	}
+}