forked from Shiloh/githaven
362 lines
8.7 KiB
Go
362 lines
8.7 KiB
Go
|
// Copyright 2017 Lunny Xiao. All rights reserved.
|
|||
|
|
// Use of this source code is governed by a MIT-style
|
|||
|
|
// license that can be found in the LICENSE file.
|
|||
|
|
|
|||
|
|
package dingtalk
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"bytes"
|
|||
|
|
"encoding/json"
|
|||
|
|
"errors"
|
|||
|
|
"fmt"
|
|||
|
|
"io/ioutil"
|
|||
|
|
"net/http"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
/*
|
|||
|
|
{
|
|||
|
|
"msgtype": "text",
|
|||
|
|
"text": {
|
|||
|
|
"content": "我就是我, 是不一样的烟火"
|
|||
|
|
},
|
|||
|
|
"at": {
|
|||
|
|
"atMobiles": [
|
|||
|
|
"156xxxx8827",
|
|||
|
|
"189xxxx8325"
|
|||
|
|
],
|
|||
|
|
"isAtAll": false
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
{
|
|||
|
|
"msgtype": "link",
|
|||
|
|
"link": {
|
|||
|
|
"text": "这个即将发布的新版本，创始人陈航（花名“无招”）称它为“红树林”。
|
|||
|
|
而在此之前，每当面临重大升级，产品经理们都会取一个应景的代号，这一次，为什么是“红树林”？",
|
|||
|
|
"title": "时代的火车向前开",
|
|||
|
|
"picUrl": "",
|
|||
|
|
"messageUrl": "https://mp.weixin.qq.com/s?__biz=MzA4NjMwMTA2Ng==&mid=2650316842&idx=1&sn=60da3ea2b29f1dcc43a7c8e4a7c97a16&scene=2&srcid=09189AnRJEdIiWVaKltFzNTw&from=timeline&isappinstalled=0&key=&ascene=2&uin=&devicetype=android-23&version=26031933&nettype=WIFI"
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
{
|
|||
|
|
"msgtype": "markdown",
|
|||
|
|
"markdown": {
|
|||
|
|
"title":"杭州天气",
|
|||
|
|
"text": "#### 杭州天气 @156xxxx8827\n" +
|
|||
|
|
"> 9度，西北风1级，空气良89，相对温度73%\n\n" +
|
|||
|
|
"> \n" +
|
|||
|
|
"> ###### 10点20分发布 [天气](http://www.thinkpage.cn/) \n"
|
|||
|
|
},
|
|||
|
|
"at": {
|
|||
|
|
"atMobiles": [
|
|||
|
|
"156xxxx8827",
|
|||
|
|
"189xxxx8325"
|
|||
|
|
],
|
|||
|
|
"isAtAll": false
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
{
|
|||
|
|
"actionCard": {
|
|||
|
|
"title": "乔布斯 20 年前想打造一间苹果咖啡厅，而它正是 Apple Store 的前身",
|
|||
|
|
"text": "
|
|||
|
|
### 乔布斯 20 年前想打造的苹果咖啡厅
|
|||
|
|
Apple Store 的设计正从原来满满的科技感走向生活化，而其生活化的走向其实可以追溯到 20 年前苹果一个建立咖啡馆的计划",
|
|||
|
|
"hideAvatar": "0",
|
|||
|
|
"btnOrientation": "0",
|
|||
|
|
"singleTitle" : "阅读全文",
|
|||
|
|
"singleURL" : "https://www.dingtalk.com/",
|
|||
|
|
"btns": [
|
|||
|
|
{
|
|||
|
|
"title": "内容不错",
|
|||
|
|
"actionURL": "https://www.dingtalk.com/"
|
|||
|
|
},
|
|||
|
|
{
|
|||
|
|
"title": "不感兴趣",
|
|||
|
|
"actionURL": "https://www.dingtalk.com/"
|
|||
|
|
}
|
|||
|
|
]
|
|||
|
|
},
|
|||
|
|
"msgtype": "actionCard"
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
{
|
|||
|
|
"feedCard": {
|
|||
|
|
"links": [
|
|||
|
|
{
|
|||
|
|
"title": "时代的火车向前开",
|
|||
|
|
"messageURL": "https://mp.weixin.qq.com/s?__biz=MzA4NjMwMTA2Ng==&mid=2650316842&idx=1&sn=60da3ea2b29f1dcc43a7c8e4a7c97a16&scene=2&srcid=09189AnRJEdIiWVaKltFzNTw&from=timeline&isappinstalled=0&key=&ascene=2&uin=&devicetype=android-23&version=26031933&nettype=WIFI",
|
|||
|
|
"picURL": "https://www.dingtalk.com/"
|
|||
|
|
},
|
|||
|
|
{
|
|||
|
|
"title": "时代的火车向前开2",
|
|||
|
|
"messageURL": "https://mp.weixin.qq.com/s?__biz=MzA4NjMwMTA2Ng==&mid=2650316842&idx=1&sn=60da3ea2b29f1dcc43a7c8e4a7c97a16&scene=2&srcid=09189AnRJEdIiWVaKltFzNTw&from=timeline&isappinstalled=0&key=&ascene=2&uin=&devicetype=android-23&version=26031933&nettype=WIFI",
|
|||
|
|
"picURL": "https://www.dingtalk.com/"
|
|||
|
|
}
|
|||
|
|
]
|
|||
|
|
},
|
|||
|
|
"msgtype": "feedCard"
|
|||
|
|
}
|
|||
|
|
*/
|
|||
|
|
|
|||
|
|
type LinkMsg struct {
|
|||
|
|
Title string `json:"title"`
|
|||
|
|
MessageURL string `json:"messageURL"`
|
|||
|
|
PicURL string `json:"picURL"`
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
type ActionCard struct {
|
|||
|
|
Text string `json:"text"`
|
|||
|
|
Title string `json:"title"`
|
|||
|
|
HideAvatar string `json:"hideAvatar"`
|
|||
|
|
BtnOrientation string `json:"btnOrientation"`
|
|||
|
|
SingleTitle string `json:"singleTitle"`
|
|||
|
|
SingleURL string `json:"singleURL"`
|
|||
|
|
Buttons []struct {
|
|||
|
|
Title string `json:"title"`
|
|||
|
|
ActionURL string `json:"actionURL"`
|
|||
|
|
} `json:"btns"`
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// Payload struct
|
|||
|
|
type Payload struct {
|
|||
|
|
MsgType string `json:"msgtype"`
|
|||
|
|
Text struct {
|
|||
|
|
Content string `json:"content"`
|
|||
|
|
} `json:"text"`
|
|||
|
|
Link struct {
|
|||
|
|
Text string `json:"text"`
|
|||
|
|
Title string `json:"title"`
|
|||
|
|
PicURL string `json:"picUrl"`
|
|||
|
|
MessageURL string `json:"messageUrl"`
|
|||
|
|
} `json:"link"`
|
|||
|
|
Markdown struct {
|
|||
|
|
Text string `json:"text"`
|
|||
|
|
Title string `json:"title"`
|
|||
|
|
} `json:"markdown"`
|
|||
|
|
ActionCard ActionCard `json:"actionCard"`
|
|||
|
|
FeedCard struct {
|
|||
|
|
Links []LinkMsg `json:"links"`
|
|||
|
|
} `json:"feedCard"`
|
|||
|
|
At struct {
|
|||
|
|
AtMobiles []string `json:"atMobiles"`
|
|||
|
|
IsAtAll bool `json:"isAtAll"`
|
|||
|
|
} `json:"at"`
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
type Webhook struct {
|
|||
|
|
accessToken string
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func NewWebhook(accessToken string) *Webhook {
|
|||
|
|
return &Webhook{accessToken}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
type Response struct {
|
|||
|
|
ErrorCode int `json:"errcode"`
|
|||
|
|
ErrorMessage string `json:"errmsg"`
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// SendPayload 发送消息
|
|||
|
|
func (w *Webhook) SendPayload(payload *Payload) error {
|
|||
|
|
bs, err := json.Marshal(payload)
|
|||
|
|
if err != nil {
|
|||
|
|
return err
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
resp, err := http.Post("https://oapi.dingtalk.com/robot/send?access_token="+w.accessToken, "application/json", bytes.NewReader(bs))
|
|||
|
|
if err != nil {
|
|||
|
|
return err
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
bs, err = ioutil.ReadAll(resp.Body)
|
|||
|
|
if err != nil {
|
|||
|
|
return err
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
if resp.StatusCode != 200 {
|
|||
|
|
return fmt.Errorf("%d: %s", resp.StatusCode, string(bs))
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
var result Response
|
|||
|
|
err = json.Unmarshal(bs, &result)
|
|||
|
|
if err != nil {
|
|||
|
|
return err
|
|||
|
|
}
|
|||
|
|
if result.ErrorCode != 0 {
|
|||
|
|
return fmt.Errorf("%d: %s", result.ErrorCode, result.ErrorMessage)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// SendTextMsg 发送文本消息
|
|||
|
|
func (w *Webhook) SendTextMsg(content string, isAtAll bool, mobiles ...string) error {
|
|||
|
|
return w.SendPayload(&Payload{
|
|||
|
|
MsgType: "text",
|
|||
|
|
Text: struct {
|
|||
|
|
Content string `json:"content"`
|
|||
|
|
}{
|
|||
|
|
Content: content,
|
|||
|
|
},
|
|||
|
|
At: struct {
|
|||
|
|
AtMobiles []string `json:"atMobiles"`
|
|||
|
|
IsAtAll bool `json:"isAtAll"`
|
|||
|
|
}{
|
|||
|
|
AtMobiles: mobiles,
|
|||
|
|
IsAtAll: isAtAll,
|
|||
|
|
},
|
|||
|
|
})
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// SendLinkMsg 发送链接消息
|
|||
|
|
func (w *Webhook) SendLinkMsg(title, content, picURL, msgURL string) error {
|
|||
|
|
return w.SendPayload(&Payload{
|
|||
|
|
MsgType: "link",
|
|||
|
|
Link: struct {
|
|||
|
|
Text string `json:"text"`
|
|||
|
|
Title string `json:"title"`
|
|||
|
|
PicURL string `json:"picUrl"`
|
|||
|
|
MessageURL string `json:"messageUrl"`
|
|||
|
|
}{
|
|||
|
|
Text: content,
|
|||
|
|
Title: title,
|
|||
|
|
PicURL: picURL,
|
|||
|
|
MessageURL: msgURL,
|
|||
|
|
},
|
|||
|
|
})
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// SendMarkdownMsg 发送markdown消息，仅支持以下格式
|
|||
|
|
/*
|
|||
|
|
标题
|
|||
|
|
# 一级标题
|
|||
|
|
## 二级标题
|
|||
|
|
### 三级标题
|
|||
|
|
#### 四级标题
|
|||
|
|
##### 五级标题
|
|||
|
|
###### 六级标题
|
|||
|
|
|
|||
|
|
引用
|
|||
|
|
> A man who stands for nothing will fall for anything.
|
|||
|
|
|
|||
|
|
文字加粗、斜体
|
|||
|
|
**bold**
|
|||
|
|
*italic*
|
|||
|
|
|
|||
|
|
链接
|
|||
|
|
[this is a link](http://name.com)
|
|||
|
|
|
|||
|
|
图片
|
|||
|
|
|
|||
|
|
|
|||
|
|
无序列表
|
|||
|
|
- item1
|
|||
|
|
- item2
|
|||
|
|
|
|||
|
|
有序列表
|
|||
|
|
1. item1
|
|||
|
|
2. item2
|
|||
|
|
*/
|
|||
|
|
func (w *Webhook) SendMarkdownMsg(title, content string, isAtAll bool, mobiles ...string) error {
|
|||
|
|
return w.SendPayload(&Payload{
|
|||
|
|
MsgType: "markdown",
|
|||
|
|
Markdown: struct {
|
|||
|
|
Text string `json:"text"`
|
|||
|
|
Title string `json:"title"`
|
|||
|
|
}{
|
|||
|
|
Text: content,
|
|||
|
|
Title: title,
|
|||
|
|
},
|
|||
|
|
At: struct {
|
|||
|
|
AtMobiles []string `json:"atMobiles"`
|
|||
|
|
IsAtAll bool `json:"isAtAll"`
|
|||
|
|
}{
|
|||
|
|
AtMobiles: mobiles,
|
|||
|
|
IsAtAll: isAtAll,
|
|||
|
|
},
|
|||
|
|
})
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// SendSingleActionCardMsg 发送整体跳转ActionCard类型消息
|
|||
|
|
func (w *Webhook) SendSingleActionCardMsg(title, content, linkTitle, linkURL string, hideAvatar, btnOrientation bool) error {
|
|||
|
|
var strHideAvatar = "0"
|
|||
|
|
if hideAvatar {
|
|||
|
|
strHideAvatar = "1"
|
|||
|
|
}
|
|||
|
|
var strBtnOrientation = "0"
|
|||
|
|
if btnOrientation {
|
|||
|
|
strBtnOrientation = "1"
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return w.SendPayload(&Payload{
|
|||
|
|
MsgType: "actionCard",
|
|||
|
|
ActionCard: ActionCard{
|
|||
|
|
Text: content,
|
|||
|
|
Title: title,
|
|||
|
|
HideAvatar: strHideAvatar,
|
|||
|
|
BtnOrientation: strBtnOrientation,
|
|||
|
|
SingleTitle: linkTitle,
|
|||
|
|
SingleURL: linkURL,
|
|||
|
|
},
|
|||
|
|
})
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// SendActionCardMsg 独立跳转ActionCard类型
|
|||
|
|
func (w *Webhook) SendActionCardMsg(title, content string, linkTitles, linkURLs []string, hideAvatar, btnOrientation bool) error {
|
|||
|
|
if len(linkTitles) == 0 || len(linkURLs) == 0 {
|
|||
|
|
return errors.New("链接参数不能为空")
|
|||
|
|
}
|
|||
|
|
if len(linkTitles) != len(linkURLs) {
|
|||
|
|
return errors.New("链接数量不匹配")
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
var strHideAvatar = "0"
|
|||
|
|
if hideAvatar {
|
|||
|
|
strHideAvatar = "1"
|
|||
|
|
}
|
|||
|
|
var strBtnOrientation = "0"
|
|||
|
|
if btnOrientation {
|
|||
|
|
strBtnOrientation = "1"
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
var btns []struct {
|
|||
|
|
Title string `json:"title"`
|
|||
|
|
ActionURL string `json:"actionURL"`
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
for i := 0; i < len(linkTitles); i++ {
|
|||
|
|
btns = append(btns, struct {
|
|||
|
|
Title string `json:"title"`
|
|||
|
|
ActionURL string `json:"actionURL"`
|
|||
|
|
}{
|
|||
|
|
Title: linkTitles[i],
|
|||
|
|
ActionURL: linkURLs[i],
|
|||
|
|
})
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return w.SendPayload(&Payload{
|
|||
|
|
MsgType: "actionCard",
|
|||
|
|
ActionCard: ActionCard{
|
|||
|
|
Text: content,
|
|||
|
|
Title: title,
|
|||
|
|
HideAvatar: strHideAvatar,
|
|||
|
|
BtnOrientation: strBtnOrientation,
|
|||
|
|
Buttons: btns,
|
|||
|
|
},
|
|||
|
|
})
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// SendLinkCardMsg 发送链接消息
|
|||
|
|
func (w *Webhook) SendLinkCardMsg(msgs []LinkMsg) error {
|
|||
|
|
return w.SendPayload(&Payload{
|
|||
|
|
MsgType: "feedCard",
|
|||
|
|
FeedCard: struct {
|
|||
|
|
Links []LinkMsg `json:"links"`
|
|||
|
|
}{
|
|||
|
|
Links: msgs,
|
|||
|
|
},
|
|||
|
|
})
|
|||
|
|
}
|