|
|
@@ -0,0 +1,161 @@
|
|
|
+package channel
|
|
|
+
|
|
|
+import (
|
|
|
+ "bytes"
|
|
|
+ "encoding/json"
|
|
|
+ "errors"
|
|
|
+ "fmt"
|
|
|
+ "message-pusher/common"
|
|
|
+ "message-pusher/model"
|
|
|
+ "net/http"
|
|
|
+ "strings"
|
|
|
+)
|
|
|
+
|
|
|
+type larkAppTokenRequest struct {
|
|
|
+ AppID string `json:"app_id"`
|
|
|
+ AppSecret string `json:"app_secret"`
|
|
|
+}
|
|
|
+
|
|
|
+type larkAppTokenResponse struct {
|
|
|
+ Code int `json:"code"`
|
|
|
+ Msg string `json:"msg"`
|
|
|
+ TenantAccessToken string `json:"tenant_access_token"`
|
|
|
+ Expire int `json:"expire"`
|
|
|
+}
|
|
|
+
|
|
|
+type LarkAppTokenStoreItem struct {
|
|
|
+ AppID string
|
|
|
+ AppSecret string
|
|
|
+ AccessToken string
|
|
|
+}
|
|
|
+
|
|
|
+func (i *LarkAppTokenStoreItem) Key() string {
|
|
|
+ return i.AppID + i.AppSecret
|
|
|
+}
|
|
|
+
|
|
|
+func (i *LarkAppTokenStoreItem) IsShared() bool {
|
|
|
+ var count int64 = 0
|
|
|
+ model.DB.Model(&model.Channel{}).Where("secret = ? and app_id = ? and type = ?",
|
|
|
+ i.AppSecret, i.AppID, model.TypeLarkApp).Count(&count)
|
|
|
+ return count > 1
|
|
|
+}
|
|
|
+
|
|
|
+func (i *LarkAppTokenStoreItem) IsFilled() bool {
|
|
|
+ return i.AppID != "" && i.AppSecret != ""
|
|
|
+}
|
|
|
+
|
|
|
+func (i *LarkAppTokenStoreItem) Token() string {
|
|
|
+ return i.AccessToken
|
|
|
+}
|
|
|
+
|
|
|
+func (i *LarkAppTokenStoreItem) Refresh() {
|
|
|
+ // https://open.feishu.cn/document/ukTMukTMukTM/ukDNz4SO0MjL5QzM/auth-v3/auth/tenant_access_token_internal
|
|
|
+ tokenRequest := larkAppTokenRequest{
|
|
|
+ AppID: i.AppID,
|
|
|
+ AppSecret: i.AppSecret,
|
|
|
+ }
|
|
|
+ tokenRequestData, err := json.Marshal(tokenRequest)
|
|
|
+ responseData, err := http.Post("https://open.feishu.cn/open-apis/auth/v3/tenant_access_token/internal",
|
|
|
+ "application/json; charset=utf-8", bytes.NewBuffer(tokenRequestData))
|
|
|
+ if err != nil {
|
|
|
+ common.SysError("failed to refresh access token: " + err.Error())
|
|
|
+ return
|
|
|
+ }
|
|
|
+ defer responseData.Body.Close()
|
|
|
+ var res larkAppTokenResponse
|
|
|
+ err = json.NewDecoder(responseData.Body).Decode(&res)
|
|
|
+ if err != nil {
|
|
|
+ common.SysError("failed to decode larkAppTokenResponse: " + err.Error())
|
|
|
+ return
|
|
|
+ }
|
|
|
+ if res.Code != 0 {
|
|
|
+ common.SysError(res.Msg)
|
|
|
+ return
|
|
|
+ }
|
|
|
+ i.AccessToken = res.TenantAccessToken
|
|
|
+ common.SysLog("access token refreshed")
|
|
|
+}
|
|
|
+
|
|
|
+type larkAppMessageRequest struct {
|
|
|
+ ReceiveId string `json:"receive_id"`
|
|
|
+ MsgType string `json:"msg_type"`
|
|
|
+ Content string `json:"content"`
|
|
|
+}
|
|
|
+
|
|
|
+type larkAppMessageResponse struct {
|
|
|
+ Code int `json:"code"`
|
|
|
+ Msg string `json:"msg"`
|
|
|
+}
|
|
|
+
|
|
|
+func parseLarkAppTarget(target string) (string, string, error) {
|
|
|
+ parts := strings.Split(target, ":")
|
|
|
+ if len(parts) != 2 {
|
|
|
+ return "", "", errors.New("无效的飞书应用号消息接收者参数")
|
|
|
+ }
|
|
|
+ return parts[0], parts[1], nil
|
|
|
+}
|
|
|
+
|
|
|
+func SendLarkAppMessage(message *model.Message, user *model.User, channel_ *model.Channel) error {
|
|
|
+ // https://open.feishu.cn/document/uAjLw4CM/ukTMukTMukTM/reference/im-v1/message/create
|
|
|
+ rawTarget := message.To
|
|
|
+ if rawTarget == "" {
|
|
|
+ rawTarget = channel_.AccountId
|
|
|
+ }
|
|
|
+ targetType, target, err := parseLarkAppTarget(rawTarget)
|
|
|
+ if err != nil {
|
|
|
+ return err
|
|
|
+ }
|
|
|
+ request := larkAppMessageRequest{
|
|
|
+ ReceiveId: target,
|
|
|
+ }
|
|
|
+ atPrefix := getLarkAtPrefix(message)
|
|
|
+ if message.Description != "" {
|
|
|
+ request.MsgType = "text"
|
|
|
+ content := larkTextContent{Text: atPrefix + message.Description}
|
|
|
+ contentData, err := json.Marshal(content)
|
|
|
+ if err != nil {
|
|
|
+ return err
|
|
|
+ }
|
|
|
+ request.Content = string(contentData)
|
|
|
+ } else {
|
|
|
+ request.MsgType = "interactive"
|
|
|
+ content := larkCardContent{}
|
|
|
+ content.Config.WideScreenMode = true
|
|
|
+ content.Config.EnableForward = true
|
|
|
+ content.Elements = append(content.Elements, larkMessageRequestCardElement{
|
|
|
+ Tag: "div",
|
|
|
+ Text: larkMessageRequestCardElementText{
|
|
|
+ Content: atPrefix + message.Content,
|
|
|
+ Tag: "lark_md",
|
|
|
+ },
|
|
|
+ })
|
|
|
+ contentData, err := json.Marshal(content)
|
|
|
+ if err != nil {
|
|
|
+ return err
|
|
|
+ }
|
|
|
+ request.Content = string(contentData)
|
|
|
+ }
|
|
|
+ requestData, err := json.Marshal(request)
|
|
|
+ if err != nil {
|
|
|
+ return err
|
|
|
+ }
|
|
|
+ key := fmt.Sprintf("%s%s", channel_.AppId, channel_.Secret)
|
|
|
+ accessToken := TokenStoreGetToken(key)
|
|
|
+ url := fmt.Sprintf("https://open.feishu.cn/open-apis/im/v1/messages?receive_id_type=%s", targetType)
|
|
|
+ req, _ := http.NewRequest("POST", url, bytes.NewReader(requestData))
|
|
|
+ req.Header.Set("Authorization", "Bearer "+accessToken)
|
|
|
+ req.Header.Set("Content-Type", "application/json; charset=utf-8")
|
|
|
+ resp, err := http.DefaultClient.Do(req)
|
|
|
+ if err != nil {
|
|
|
+ return err
|
|
|
+ }
|
|
|
+ var res larkAppMessageResponse
|
|
|
+ err = json.NewDecoder(resp.Body).Decode(&res)
|
|
|
+ if err != nil {
|
|
|
+ return err
|
|
|
+ }
|
|
|
+ if res.Code != 0 {
|
|
|
+ return errors.New(res.Msg)
|
|
|
+ }
|
|
|
+ return nil
|
|
|
+}
|
JustSong