从原代码库剥离为单独的仓库

Author: Athurg Feng

department.go

@@ -0,0 +1,116 @@
+package dingtalk
+
+import (
+	"encoding/json"
+	"fmt"
+	"net/http"
+	"net/url"
+)
+
+// API地址
+const (
+	DEPARTMENT_LIST_URL = DINGTALK_API_URL + "/department/list"
+)
+
+// 部门
+type Department struct {
+	Id              int
+	Name            string
+	ParentId        int
+	CreateDeptGroup bool
+	AutoAddUser     bool
+	Ext             string
+}
+
+// 获取部门列表API的返回数据
+type DepartmentListResponse struct {
+	CommonResponse
+	Department []Department
+}
+
+// 获取指定部门的子部门列表
+func GetDepartmentList(id int, fetchChild bool) ([]Department, error) {
+	param := url.Values{
+		"access_token": {accessToken},
+		"fetch_child":  {fmt.Sprintf("%t", fetchChild)},
+	}
+
+	if id != 0 {
+		param.Set("id", fmt.Sprintf("%d", id))
+
+	}
+
+	resp, err := http.Get(DEPARTMENT_LIST_URL + "?" + param.Encode())
+	if err != nil {
+		return nil, err
+	}
+	defer resp.Body.Close()
+
+	var info DepartmentListResponse
+	err = json.NewDecoder(resp.Body).Decode(&info)
+	if err != nil {
+		return nil, err
+	}
+
+	if err := info.CommonResponse.Error(); err != nil {
+		return nil, err
+	}
+
+	return info.Department, nil
+}
+
+// 获取所有部门列表
+func GetAllDepartments() ([]Department, error) {
+	return GetDepartmentList(0, true)
+}
+
+// 获取重复（部门名称）的部门信息
+func DumpDuplicatedDepartments() (map[string]int, error) {
+	deptCount := map[string]int{}
+
+	depts, err := GetDepartmentList(0, true)
+	if err != nil {
+		return nil, err
+	}
+
+	for _, dept := range depts {
+		deptCount[dept.Name] += 1
+	}
+
+	for dept, count := range deptCount {
+		if count == 1 {
+			delete(deptCount, dept)
+		}
+	}
+
+	return deptCount, nil
+}
+
+// 获取部门名称-ID映射表
+func GetDepartmentIdMap() (map[string]int, error) {
+	// 获取所有部门信息
+	depts, err := GetDepartmentList(0, true)
+	if err != nil {
+		return nil, err
+	}
+
+	//检查是否有重名部门
+	deptCount := map[string]int{}
+	nameIdMap := map[string]int{}
+	for _, dept := range depts {
+		deptCount[dept.Name] += 1
+		nameIdMap[dept.Name] = dept.Id
+	}
+
+	for dept, count := range deptCount {
+		if count == 1 {
+			delete(deptCount, dept)
+		}
+	}
+
+	if len(deptCount) > 0 {
+		return nil, fmt.Errorf("部分部门重名，重名部门包括%s", deptCount)
+	}
+
+	return nameIdMap, nil
+}

dingtalk.go

@@ -0,0 +1,24 @@
+package dingtalk
+
+import (
+	"fmt"
+)
+
+// 钉钉API地址
+const (
+	DINGTALK_API_URL = "https://oapi.dingtalk.com"
+)
+
+// 钉钉API公共响应
+type CommonResponse struct {
+	ErrCode int
+	ErrMsg  string
+}
+
+func (resp *CommonResponse) Error() error {
+	if resp.ErrCode != 0 || (resp.ErrMsg != "ok" && resp.ErrMsg != "") {
+		return fmt.Errorf("[%d]%s", resp.ErrCode, resp.ErrMsg)
+	}
+
+	return nil
+}

gettoken.go

@@ -0,0 +1,45 @@
+package dingtalk
+
+import (
+	"encoding/json"
+	"net/http"
+	"net/url"
+)
+
+// API地址
+const (
+	GETTOKEN_URL = DINGTALK_API_URL + "/gettoken"
+)
+
+var accessToken string
+
+// 刷新AccessToken
+func RefreshAccessToken(appkey, appsecret string) error {
+	param := url.Values{
+		"appkey":    {appkey},
+		"appsecret": {appsecret},
+	}
+
+	resp, err := http.Get(GETTOKEN_URL + "?" + param.Encode())
+	if err != nil {
+		return err
+	}
+	defer resp.Body.Close()
+
+	var info struct {
+		CommonResponse
+		AccessToken string `json:"access_token"`
+	}
+
+	err = json.NewDecoder(resp.Body).Decode(&info)
+	if err != nil {
+		return err
+	}
+
+	if err := info.CommonResponse.Error(); err != nil {
+		return err
+	}
+
+	accessToken = info.AccessToken
+	return nil
+}

message.go

@@ -0,0 +1,56 @@
+package dingtalk
+
+// 钉钉消息
+type Message struct {
+	MsgType string `json:"msgtype"`
+
+	Markdown *MessagePartialMarkdown `json:"markdown,omitempty"`
+	Text     *MessagePartialText     `json:"text,omitempty"`
+
+	At *MessagePartialAt `json:"at,omitempty"`
+}
+
+// 钉钉消息AT部分
+type MessagePartialAt struct {
+	AtMobiles     []string `json:"atMobiles"`
+	AtDingtalkIds []string `json:"atDingtalkIds"`
+	IsAtAll       bool     `json:"isAtAll"`
+}
+
+// 钉钉消息文本部分
+type MessagePartialText struct {
+	Content string `json:"content"`
+}
+
+// 钉钉消息Markdown部分
+type MessagePartialMarkdown struct {
+	Title string `json:"title"`
+	Text  string `json:"text"`
+}
+
+// 创建一条Markdown消息
+func NewMarkdownMessage(title, text string) Message {
+	partial := &MessagePartialMarkdown{
+		Text:  text,
+		Title: title,
+	}
+	msg := Message{
+		MsgType:  "markdown",
+		Markdown: partial,
+	}
+
+	return msg
+}
+
+// 创建一条文本消息
+func NewTextMessage(content string) Message {
+	partial := &MessagePartialText{
+		Content: content,
+	}
+	msg := Message{
+		MsgType: "text",
+		Text:    partial,
+	}
+
+	return msg
+}

robot.go

@@ -0,0 +1,43 @@
+package dingtalk
+
+import (
+	"crypto/hmac"
+	"crypto/sha256"
+	"encoding/base64"
+	"net/http"
+)
+
+// 钉钉企业内部机器人请求结构
+type RobotRequest struct {
+	MsgType string
+	Text    struct {
+		Content string
+	}
+
+	MsgId    string
+	CreateAt int
+
+	ConversationType  string
+	ConversationId    string
+	ConversationTitle string
+	SenderId          string
+	SenderNick        string
+	SenderCorpId      string
+	SenderStaffId     string
+	ChatbotUserId     string
+	AtUsers           []struct {
+		DingtalkId string
+		StaffId    string
+	}
+}
+
+func IsValidateRobotRequest(appSecret string, r *http.Request) bool {
+	s := r.Header.Get("timestamp") + "\n" + appSecret
+
+	mac := hmac.New(sha256.New, []byte(appSecret))
+	mac.Write([]byte(s))
+	expectMac := mac.Sum(nil)
+
+	sign := base64.StdEncoding.EncodeToString(expectMac)
+	return sign == r.Header.Get("sign")
+}