mirror of
https://github.com/silenceper/wechat.git
synced 2026-03-01 00:35:26 +08:00
Compare commits
1 Commits
v2.0.7-rc.
...
c5c764dcc9
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
c5c764dcc9 |
21
.github/ISSUE_TEMPLATE/bug.md
vendored
21
.github/ISSUE_TEMPLATE/bug.md
vendored
@@ -1,21 +0,0 @@
|
|||||||
---
|
|
||||||
name: 报告Bug
|
|
||||||
about: 反馈BUG信息
|
|
||||||
title: "[BUG]"
|
|
||||||
labels: bug
|
|
||||||
assignees: ''
|
|
||||||
|
|
||||||
---
|
|
||||||
|
|
||||||
**描述**
|
|
||||||
|
|
||||||
**如何复现**
|
|
||||||
步骤:
|
|
||||||
1、
|
|
||||||
2、
|
|
||||||
|
|
||||||
**关联日志信息**
|
|
||||||
|
|
||||||
|
|
||||||
**使用的版本**
|
|
||||||
- SDK版本: [比如 v0.0.0]
|
|
||||||
15
.github/ISSUE_TEMPLATE/feature.md
vendored
15
.github/ISSUE_TEMPLATE/feature.md
vendored
@@ -1,15 +0,0 @@
|
|||||||
---
|
|
||||||
name: API需求
|
|
||||||
about: 待实现的API接口,SDK的强大离不开社区的帮助,欢迎为项目贡献PR
|
|
||||||
title: "[Feature]"
|
|
||||||
labels: enhancement
|
|
||||||
assignees: ''
|
|
||||||
|
|
||||||
---
|
|
||||||
|
|
||||||
<!--
|
|
||||||
!!!SDK的强大离不开社区的帮助,欢迎为本项目贡献PR!!!
|
|
||||||
-->
|
|
||||||
**你想要实现的模块或API**
|
|
||||||
|
|
||||||
|
|
||||||
15
.github/ISSUE_TEMPLATE/question.md
vendored
15
.github/ISSUE_TEMPLATE/question.md
vendored
@@ -1,15 +0,0 @@
|
|||||||
---
|
|
||||||
name: 使用咨询
|
|
||||||
about: 关于SDK使用相关的咨询,在使用前请先阅读官方微信文档
|
|
||||||
title: "[咨询]"
|
|
||||||
labels: question
|
|
||||||
assignees: ''
|
|
||||||
|
|
||||||
---
|
|
||||||
|
|
||||||
<!--
|
|
||||||
重要:
|
|
||||||
1、在使用本SDK前请先阅读对应的官方微信API文档:https://developers.weixin.qq.com/doc/offiaccount/Getting_Started/Overview.html
|
|
||||||
2、本SDK部分接口文档: https://silenceper.com/wechat/
|
|
||||||
-->
|
|
||||||
**请描述您的问题**
|
|
||||||
3
.gitignore
vendored
3
.gitignore
vendored
@@ -26,5 +26,4 @@ _testmain.go
|
|||||||
.vscode/
|
.vscode/
|
||||||
vendor
|
vendor
|
||||||
.idea/
|
.idea/
|
||||||
example/*
|
example/*
|
||||||
/test
|
|
||||||
@@ -12,15 +12,11 @@ import (
|
|||||||
|
|
||||||
const (
|
const (
|
||||||
//AccessTokenURL 获取access_token的接口
|
//AccessTokenURL 获取access_token的接口
|
||||||
accessTokenURL = "https://api.weixin.qq.com/cgi-bin/token?grant_type=client_credential&appid=%s&secret=%s"
|
accessTokenURL = "https://api.weixin.qq.com/cgi-bin/token"
|
||||||
//AccessTokenURL 企业微信获取access_token的接口
|
|
||||||
workAccessTokenURL = "https://qyapi.weixin.qq.com/cgi-bin/gettoken?corpid=%s&corpsecret=%s"
|
|
||||||
//CacheKeyOfficialAccountPrefix 微信公众号cache key前缀
|
//CacheKeyOfficialAccountPrefix 微信公众号cache key前缀
|
||||||
CacheKeyOfficialAccountPrefix = "gowechat_officialaccount_"
|
CacheKeyOfficialAccountPrefix = "gowechat_officialaccount_"
|
||||||
//CacheKeyMiniProgramPrefix 小程序cache key前缀
|
//CacheKeyMiniProgramPrefix 小程序cache key前缀
|
||||||
CacheKeyMiniProgramPrefix = "gowechat_miniprogram_"
|
CacheKeyMiniProgramPrefix = "gowechat_miniprogram_"
|
||||||
//CacheKeyWorkPrefix 企业微信cache key前缀
|
|
||||||
CacheKeyWorkPrefix = "gowechat_work_"
|
|
||||||
)
|
)
|
||||||
|
|
||||||
//DefaultAccessToken 默认AccessToken 获取
|
//DefaultAccessToken 默认AccessToken 获取
|
||||||
@@ -69,58 +65,7 @@ func (ak *DefaultAccessToken) GetAccessToken() (accessToken string, err error) {
|
|||||||
|
|
||||||
//cache失效,从微信服务器获取
|
//cache失效,从微信服务器获取
|
||||||
var resAccessToken ResAccessToken
|
var resAccessToken ResAccessToken
|
||||||
resAccessToken, err = GetTokenFromServer(fmt.Sprintf(accessTokenURL, ak.appID, ak.appSecret))
|
resAccessToken, err = GetTokenFromServer(ak.appID, ak.appSecret)
|
||||||
if err != nil {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
expires := resAccessToken.ExpiresIn - 1500
|
|
||||||
err = ak.cache.Set(accessTokenCacheKey, resAccessToken.AccessToken, time.Duration(expires)*time.Second)
|
|
||||||
if err != nil {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
accessToken = resAccessToken.AccessToken
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
//WorkAccessToken 企业微信AccessToken 获取
|
|
||||||
type WorkAccessToken struct {
|
|
||||||
CorpID string
|
|
||||||
CorpSecret string
|
|
||||||
cacheKeyPrefix string
|
|
||||||
cache cache.Cache
|
|
||||||
accessTokenLock *sync.Mutex
|
|
||||||
}
|
|
||||||
|
|
||||||
//NewWorkAccessToken new WorkAccessToken
|
|
||||||
func NewWorkAccessToken(corpID, corpSecret, cacheKeyPrefix string, cache cache.Cache) AccessTokenHandle {
|
|
||||||
if cache == nil {
|
|
||||||
panic("cache the not exist")
|
|
||||||
}
|
|
||||||
return &WorkAccessToken{
|
|
||||||
CorpID: corpID,
|
|
||||||
CorpSecret: corpSecret,
|
|
||||||
cache: cache,
|
|
||||||
cacheKeyPrefix: cacheKeyPrefix,
|
|
||||||
accessTokenLock: new(sync.Mutex),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
//GetAccessToken 企业微信获取access_token,先从cache中获取,没有则从服务端获取
|
|
||||||
func (ak *WorkAccessToken) GetAccessToken() (accessToken string, err error) {
|
|
||||||
//加上lock,是为了防止在并发获取token时,cache刚好失效,导致从微信服务器上获取到不同token
|
|
||||||
ak.accessTokenLock.Lock()
|
|
||||||
defer ak.accessTokenLock.Unlock()
|
|
||||||
accessTokenCacheKey := fmt.Sprintf("%s_access_token_%s", ak.cacheKeyPrefix, ak.CorpID)
|
|
||||||
val := ak.cache.Get(accessTokenCacheKey)
|
|
||||||
if val != nil {
|
|
||||||
accessToken = val.(string)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
//cache失效,从微信服务器获取
|
|
||||||
var resAccessToken ResAccessToken
|
|
||||||
resAccessToken, err = GetTokenFromServer(fmt.Sprintf(workAccessTokenURL, ak.CorpID, ak.CorpSecret))
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
@@ -135,7 +80,8 @@ func (ak *WorkAccessToken) GetAccessToken() (accessToken string, err error) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
//GetTokenFromServer 强制从微信服务器获取token
|
//GetTokenFromServer 强制从微信服务器获取token
|
||||||
func GetTokenFromServer(url string) (resAccessToken ResAccessToken, err error) {
|
func GetTokenFromServer(appID, appSecret string) (resAccessToken ResAccessToken, err error) {
|
||||||
|
url := fmt.Sprintf("%s?grant_type=client_credential&appid=%s&secret=%s", accessTokenURL, appID, appSecret)
|
||||||
var body []byte
|
var body []byte
|
||||||
body, err = util.HTTPGet(url)
|
body, err = util.HTTPGet(url)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
@@ -145,7 +91,7 @@ func GetTokenFromServer(url string) (resAccessToken ResAccessToken, err error) {
|
|||||||
if err != nil {
|
if err != nil {
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
if resAccessToken.ErrCode != 0 {
|
if resAccessToken.ErrMsg != "" {
|
||||||
err = fmt.Errorf("get access_token error : errcode=%v , errormsg=%v", resAccessToken.ErrCode, resAccessToken.ErrMsg)
|
err = fmt.Errorf("get access_token error : errcode=%v , errormsg=%v", resAccessToken.ErrCode, resAccessToken.ErrMsg)
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|||||||
1
go.sum
1
go.sum
@@ -24,6 +24,7 @@ github.com/sirupsen/logrus v1.8.1 h1:dJKuHgqk1NNQlqoA6BTlM1Wf9DOH3NBjQyu0h9+AZZE
|
|||||||
github.com/sirupsen/logrus v1.8.1/go.mod h1:yWOB1SBYBC5VeMP7gHvWumXLIWorT60ONWic61uBYv0=
|
github.com/sirupsen/logrus v1.8.1/go.mod h1:yWOB1SBYBC5VeMP7gHvWumXLIWorT60ONWic61uBYv0=
|
||||||
github.com/spf13/cast v1.3.1 h1:nFm6S0SMdyzrzcmThSipiEubIDy8WEXKNZ0UOgiRpng=
|
github.com/spf13/cast v1.3.1 h1:nFm6S0SMdyzrzcmThSipiEubIDy8WEXKNZ0UOgiRpng=
|
||||||
github.com/spf13/cast v1.3.1/go.mod h1:Qx5cxh0v+4UWYiBimWS+eyWzqEqokIECu5etghLkUJE=
|
github.com/spf13/cast v1.3.1/go.mod h1:Qx5cxh0v+4UWYiBimWS+eyWzqEqokIECu5etghLkUJE=
|
||||||
|
github.com/stretchr/objx v0.1.0 h1:4G4v2dO3VZwixGIRoQ5Lfboy6nUhCyYzaqnIAPPhYs4=
|
||||||
github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME=
|
github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME=
|
||||||
github.com/stretchr/testify v1.2.2/go.mod h1:a8OnRcib4nhh0OaRAV+Yts87kKdq0PP7pXfy6kDkUVs=
|
github.com/stretchr/testify v1.2.2/go.mod h1:a8OnRcib4nhh0OaRAV+Yts87kKdq0PP7pXfy6kDkUVs=
|
||||||
github.com/stretchr/testify v1.5.1/go.mod h1:5W2xD1RspED5o8YsWQXVCued0rvSQ+mT+I5cxcmMvtA=
|
github.com/stretchr/testify v1.5.1/go.mod h1:5W2xD1RspED5o8YsWQXVCued0rvSQ+mT+I5cxcmMvtA=
|
||||||
|
|||||||
@@ -7,7 +7,9 @@ import (
|
|||||||
|
|
||||||
// Config config for 小程序
|
// Config config for 小程序
|
||||||
type Config struct {
|
type Config struct {
|
||||||
AppID string `json:"app_id"` // appid
|
AppID string `json:"app_id"` // app_id
|
||||||
AppSecret string `json:"app_secret"` // appsecret
|
AppSecret string `json:"app_secret"` // app_secret
|
||||||
Cache cache.Cache
|
Token string `json:"token"` // token
|
||||||
|
EncodingAESKey string `json:"encoding_aes_key"` // encoding_aes_key
|
||||||
|
Cache cache.Cache
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -1,61 +0,0 @@
|
|||||||
package content
|
|
||||||
|
|
||||||
import (
|
|
||||||
"fmt"
|
|
||||||
|
|
||||||
"github.com/silenceper/wechat/v2/miniprogram/context"
|
|
||||||
"github.com/silenceper/wechat/v2/util"
|
|
||||||
)
|
|
||||||
|
|
||||||
const (
|
|
||||||
checkTextURL = "https://api.weixin.qq.com/wxa/msg_sec_check?access_token=%s"
|
|
||||||
checkImageURL = "https://api.weixin.qq.com/wxa/img_sec_check?access_token=%s"
|
|
||||||
)
|
|
||||||
|
|
||||||
//Content 内容安全
|
|
||||||
type Content struct {
|
|
||||||
*context.Context
|
|
||||||
}
|
|
||||||
|
|
||||||
//NewContent 内容安全接口
|
|
||||||
func NewContent(ctx *context.Context) *Content {
|
|
||||||
return &Content{ctx}
|
|
||||||
}
|
|
||||||
|
|
||||||
//CheckText 检测文字
|
|
||||||
//@text 需要检测的文字
|
|
||||||
func (content *Content) CheckText(text string) error {
|
|
||||||
accessToken, err := content.GetAccessToken()
|
|
||||||
if err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
response, err := util.PostJSON(
|
|
||||||
fmt.Sprintf(checkTextURL, accessToken),
|
|
||||||
map[string]string{
|
|
||||||
"content": text,
|
|
||||||
},
|
|
||||||
)
|
|
||||||
if err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
return util.DecodeWithCommonError(response, "ContentCheckText")
|
|
||||||
}
|
|
||||||
|
|
||||||
//CheckImage 检测图片
|
|
||||||
//所传参数为要检测的图片文件的绝对路径,图片格式支持PNG、JPEG、JPG、GIF, 像素不超过 750 x 1334,同时文件大小以不超过 300K 为宜,否则可能报错
|
|
||||||
//@media 图片文件的绝对路径
|
|
||||||
func (content *Content) CheckImage(media string) error {
|
|
||||||
accessToken, err := content.GetAccessToken()
|
|
||||||
if err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
response, err := util.PostFile(
|
|
||||||
"media",
|
|
||||||
media,
|
|
||||||
fmt.Sprintf(checkImageURL, accessToken),
|
|
||||||
)
|
|
||||||
if err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
return util.DecodeWithCommonError(response, "ContentCheckImage")
|
|
||||||
}
|
|
||||||
@@ -20,13 +20,29 @@ const (
|
|||||||
MsgTypeLink = "link"
|
MsgTypeLink = "link"
|
||||||
//MsgTypeMiniProgramPage 小程序卡片
|
//MsgTypeMiniProgramPage 小程序卡片
|
||||||
MsgTypeMiniProgramPage = "miniprogrampage"
|
MsgTypeMiniProgramPage = "miniprogrampage"
|
||||||
|
//MsgTypeEvent 表示事件推送消息
|
||||||
|
MsgTypeEvent = "event"
|
||||||
|
)
|
||||||
|
|
||||||
|
const (
|
||||||
|
//EventSubscribePopup 用户操作订阅通知弹窗事件推送,用户在图文等场景内订阅通知的操作
|
||||||
|
EventSubscribePopup EventType = "subscribe_msg_popup_event"
|
||||||
|
//EventSubscribeChange 用户管理订阅通知,用户在服务通知管理页面做通知管理时的操作
|
||||||
|
EventSubscribeChange = "subscribe_msg_change_event"
|
||||||
|
//EventSubscribeSent 发送订阅通知,调用 bizsend 接口发送通知
|
||||||
|
EventSubscribeSent = "subscribe_msg_sent_event"
|
||||||
|
)
|
||||||
|
|
||||||
|
const (
|
||||||
|
InfoTypeAcceptSubscribeMessage InfoType = "accept"
|
||||||
|
InfoTypeRejectSubscribeMessage = "reject"
|
||||||
)
|
)
|
||||||
|
|
||||||
// CommonToken 消息中通用的结构
|
// CommonToken 消息中通用的结构
|
||||||
type CommonToken struct {
|
type CommonToken struct {
|
||||||
XMLName xml.Name `xml:"xml"`
|
XMLName xml.Name `xml:"xml"`
|
||||||
ToUserName string `xml:"ToUserName"`
|
ToUserName CDATA `xml:"ToUserName"`
|
||||||
FromUserName string `xml:"FromUserName"`
|
FromUserName CDATA `xml:"FromUserName"`
|
||||||
CreateTime int64 `xml:"CreateTime"`
|
CreateTime int64 `xml:"CreateTime"`
|
||||||
MsgType MsgType `xml:"MsgType"`
|
MsgType MsgType `xml:"MsgType"`
|
||||||
}
|
}
|
||||||
@@ -54,4 +70,35 @@ type MiniProgramMixMessage struct {
|
|||||||
// 进入会话事件
|
// 进入会话事件
|
||||||
Event string `xml:"Event"`
|
Event string `xml:"Event"`
|
||||||
SessionFrom string `xml:"SessionFrom"`
|
SessionFrom string `xml:"SessionFrom"`
|
||||||
|
|
||||||
|
// 用户操作订阅通知弹窗消息回调
|
||||||
|
SubscribeMsgPopupEvent struct {
|
||||||
|
List []SubscribeMessageList `xml:"List"`
|
||||||
|
} `xml:"SubscribeMsgPopupEvent"`
|
||||||
|
|
||||||
|
// 用户管理订阅通知回调
|
||||||
|
SubscribeMsgChangeEvent struct {
|
||||||
|
List []SubscribeMessageList `xml:"List"`
|
||||||
|
} `xml:"SubscribeMsgChangeEvent"`
|
||||||
|
|
||||||
|
// 用户发送订阅通知回调
|
||||||
|
SubscribeMsgSentEvent struct {
|
||||||
|
List []SubscribeMessageList `xml:"List"`
|
||||||
|
} `xml:"SubscribeMsgSentEvent"`
|
||||||
}
|
}
|
||||||
|
|
||||||
|
type SubscribeMessageList struct {
|
||||||
|
TemplateId string `xml:"TemplateId"`
|
||||||
|
SubscribeStatusString string `xml:"SubscribeStatusString"`
|
||||||
|
PopupScene string `xml:"PopupScene"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// EncryptedXMLMsg 安全模式下的消息体
|
||||||
|
type EncryptedXMLMsg struct {
|
||||||
|
XMLName struct{} `xml:"xml" json:"-"`
|
||||||
|
ToUserName string `xml:"ToUserName" json:"toUserName"`
|
||||||
|
EncryptedMsg string `xml:"Encrypt" json:"Encrypt"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// CDATA 使用这种类型,在序列化 xml 时文本会被解析器忽略
|
||||||
|
type CDATA string
|
||||||
|
|||||||
15
miniprogram/message/reply.go
Normal file
15
miniprogram/message/reply.go
Normal file
@@ -0,0 +1,15 @@
|
|||||||
|
package message
|
||||||
|
|
||||||
|
import "errors"
|
||||||
|
|
||||||
|
//ErrInvalidReply 无效的回复
|
||||||
|
var ErrInvalidReply = errors.New("无效的回复信息")
|
||||||
|
|
||||||
|
//ErrUnsupportedReply 不支持的回复类型
|
||||||
|
var ErrUnsupportedReply = errors.New("不支持的回复消息")
|
||||||
|
|
||||||
|
//Reply 消息回复
|
||||||
|
type Reply struct {
|
||||||
|
MsgType MsgType
|
||||||
|
MsgData interface{}
|
||||||
|
}
|
||||||
@@ -5,14 +5,15 @@ import (
|
|||||||
"github.com/silenceper/wechat/v2/miniprogram/analysis"
|
"github.com/silenceper/wechat/v2/miniprogram/analysis"
|
||||||
"github.com/silenceper/wechat/v2/miniprogram/auth"
|
"github.com/silenceper/wechat/v2/miniprogram/auth"
|
||||||
"github.com/silenceper/wechat/v2/miniprogram/config"
|
"github.com/silenceper/wechat/v2/miniprogram/config"
|
||||||
"github.com/silenceper/wechat/v2/miniprogram/content"
|
|
||||||
"github.com/silenceper/wechat/v2/miniprogram/context"
|
"github.com/silenceper/wechat/v2/miniprogram/context"
|
||||||
"github.com/silenceper/wechat/v2/miniprogram/encryptor"
|
"github.com/silenceper/wechat/v2/miniprogram/encryptor"
|
||||||
"github.com/silenceper/wechat/v2/miniprogram/message"
|
"github.com/silenceper/wechat/v2/miniprogram/message"
|
||||||
"github.com/silenceper/wechat/v2/miniprogram/qrcode"
|
"github.com/silenceper/wechat/v2/miniprogram/qrcode"
|
||||||
|
"github.com/silenceper/wechat/v2/miniprogram/server"
|
||||||
"github.com/silenceper/wechat/v2/miniprogram/subscribe"
|
"github.com/silenceper/wechat/v2/miniprogram/subscribe"
|
||||||
"github.com/silenceper/wechat/v2/miniprogram/tcb"
|
"github.com/silenceper/wechat/v2/miniprogram/tcb"
|
||||||
"github.com/silenceper/wechat/v2/miniprogram/werun"
|
"github.com/silenceper/wechat/v2/miniprogram/werun"
|
||||||
|
"net/http"
|
||||||
)
|
)
|
||||||
|
|
||||||
//MiniProgram 微信小程序相关API
|
//MiniProgram 微信小程序相关API
|
||||||
@@ -80,7 +81,10 @@ func (miniProgram *MiniProgram) GetWeRun() *werun.WeRun {
|
|||||||
return werun.NewWeRun(miniProgram.ctx)
|
return werun.NewWeRun(miniProgram.ctx)
|
||||||
}
|
}
|
||||||
|
|
||||||
// GetContentSecurity 内容安全接口
|
// GetServer 小程序微信回调处理,接收事件,回复消息管理
|
||||||
func (miniProgram *MiniProgram) GetContentSecurity() *content.Content {
|
func (miniProgram *MiniProgram) GetServer(req *http.Request, write http.ResponseWriter) *server.Server {
|
||||||
return content.NewContent(miniProgram.ctx)
|
srv := server.NewServer(miniProgram.ctx)
|
||||||
|
srv.Request = req
|
||||||
|
srv.Write = write
|
||||||
|
return srv
|
||||||
}
|
}
|
||||||
|
|||||||
199
miniprogram/server/server.go
Normal file
199
miniprogram/server/server.go
Normal file
@@ -0,0 +1,199 @@
|
|||||||
|
package server
|
||||||
|
|
||||||
|
import (
|
||||||
|
"encoding/xml"
|
||||||
|
"errors"
|
||||||
|
"fmt"
|
||||||
|
"github.com/silenceper/wechat/v2/miniprogram/context"
|
||||||
|
"github.com/silenceper/wechat/v2/miniprogram/message"
|
||||||
|
"github.com/silenceper/wechat/v2/util"
|
||||||
|
"io/ioutil"
|
||||||
|
"net/http"
|
||||||
|
"reflect"
|
||||||
|
"runtime/debug"
|
||||||
|
"strconv"
|
||||||
|
)
|
||||||
|
|
||||||
|
// Server struct
|
||||||
|
type Server struct {
|
||||||
|
*context.Context
|
||||||
|
Write http.ResponseWriter
|
||||||
|
Request *http.Request
|
||||||
|
skipValidate bool
|
||||||
|
openID string
|
||||||
|
|
||||||
|
messageHandler func(mixMessage *message.MiniProgramMixMessage) *message.Reply
|
||||||
|
|
||||||
|
RequestRawXMLMsg []byte
|
||||||
|
RequestMsg *message.MiniProgramMixMessage
|
||||||
|
|
||||||
|
ResponseRawXMLMsg []byte
|
||||||
|
ResponseMsg interface{}
|
||||||
|
|
||||||
|
isSafeMode bool
|
||||||
|
random []byte
|
||||||
|
nonce string
|
||||||
|
timestamp int64
|
||||||
|
}
|
||||||
|
|
||||||
|
func NewServer(context *context.Context) *Server {
|
||||||
|
srv := new(Server)
|
||||||
|
srv.Context = context
|
||||||
|
return srv
|
||||||
|
}
|
||||||
|
|
||||||
|
func (srv *Server) Server() error {
|
||||||
|
if !srv.Validate() {
|
||||||
|
return fmt.Errorf("请求签名校验失败")
|
||||||
|
}
|
||||||
|
echoStr := srv.Query("echostr")
|
||||||
|
if echoStr != "" {
|
||||||
|
srv.SetResponseWrite(echoStr)
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
response, err := srv.handleRequest()
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
return srv.buildResponse(response)
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
// SkipValidate 设置跳过签名校验
|
||||||
|
func (srv *Server) SkipValidate(skip bool) {
|
||||||
|
srv.skipValidate = skip
|
||||||
|
}
|
||||||
|
|
||||||
|
// Validate 校验请求是否合法
|
||||||
|
func (srv *Server) Validate() bool {
|
||||||
|
if srv.skipValidate {
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
timestamp := srv.Query("timestamp")
|
||||||
|
nonce := srv.Query("nonce")
|
||||||
|
signature := srv.Query("signature")
|
||||||
|
return signature == util.Signature(srv.Token, timestamp, nonce)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (srv *Server) handleRequest() (reply *message.Reply, err error) {
|
||||||
|
//set isSafeMode
|
||||||
|
srv.isSafeMode = false
|
||||||
|
encryptType := srv.Query("encrypt_type")
|
||||||
|
if encryptType == "aes" {
|
||||||
|
srv.isSafeMode = true
|
||||||
|
}
|
||||||
|
//set openID
|
||||||
|
srv.openID = srv.Query("openid")
|
||||||
|
|
||||||
|
var msg interface{}
|
||||||
|
msg, err = srv.getMessage()
|
||||||
|
if err != nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
mixMessage, success := msg.(*message.MiniProgramMixMessage)
|
||||||
|
if !success {
|
||||||
|
err = errors.New("消息类型转换失败")
|
||||||
|
}
|
||||||
|
srv.RequestMsg = mixMessage
|
||||||
|
reply = srv.messageHandler(mixMessage)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
//GetOpenID return openID
|
||||||
|
func (srv *Server) GetOpenID() string {
|
||||||
|
return srv.openID
|
||||||
|
}
|
||||||
|
|
||||||
|
func (srv *Server) getMessage() (interface{}, error) {
|
||||||
|
var rawXMLMsgBytes []byte
|
||||||
|
var err error
|
||||||
|
if srv.isSafeMode {
|
||||||
|
var encryptedXMLMsg message.EncryptedXMLMsg
|
||||||
|
if err := xml.NewDecoder(srv.Request.Body).Decode(&encryptedXMLMsg); err != nil {
|
||||||
|
return nil, fmt.Errorf("从body中解析xml失败,err=%v", err)
|
||||||
|
}
|
||||||
|
//验证消息签名
|
||||||
|
timestamp := srv.Query("timestamp")
|
||||||
|
srv.timestamp, err = strconv.ParseInt(timestamp, 10, 32)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
nonce := srv.Query("nonce")
|
||||||
|
srv.nonce = nonce
|
||||||
|
msgSignature := srv.Query("msg_signature")
|
||||||
|
msgSignatureGen := util.Signature(srv.Token, timestamp, nonce, encryptedXMLMsg.EncryptedMsg)
|
||||||
|
if msgSignature != msgSignatureGen {
|
||||||
|
return nil, fmt.Errorf("消息不合法,验证签名失败")
|
||||||
|
}
|
||||||
|
|
||||||
|
// 解密
|
||||||
|
srv.random, rawXMLMsgBytes, err = util.DecryptMsg(srv.AppID, encryptedXMLMsg.EncryptedMsg, srv.EncodingAESKey)
|
||||||
|
if err != nil {
|
||||||
|
return nil, fmt.Errorf("消息解密失败, err=%v", err)
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
rawXMLMsgBytes, err = ioutil.ReadAll(srv.Request.Body)
|
||||||
|
if err != nil {
|
||||||
|
return nil, fmt.Errorf("从body中解析xml失败, err=%v", err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
srv.RequestRawXMLMsg = rawXMLMsgBytes
|
||||||
|
return srv.parseRequestMessage(rawXMLMsgBytes)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (srv *Server) parseRequestMessage(rawXMLMsgBytes []byte) (msg *message.MiniProgramMixMessage, err error) {
|
||||||
|
msg = &message.MiniProgramMixMessage{}
|
||||||
|
err = xml.Unmarshal(rawXMLMsgBytes, msg)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
func (srv *Server) SetMessageHandler(handler func(*message.MiniProgramMixMessage) *message.Reply) {
|
||||||
|
srv.messageHandler = handler
|
||||||
|
}
|
||||||
|
|
||||||
|
func (srv *Server) buildResponse(reply *message.Reply) (err error) {
|
||||||
|
defer func() {
|
||||||
|
if e := recover(); e != nil {
|
||||||
|
err = fmt.Errorf("panic error: %v\n%s", e, debug.Stack())
|
||||||
|
}
|
||||||
|
}()
|
||||||
|
if reply == nil {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
msgType := reply.MsgType
|
||||||
|
switch msgType {
|
||||||
|
case message.MsgTypeEvent:
|
||||||
|
case message.MsgTypeImage:
|
||||||
|
case message.MsgTypeLink:
|
||||||
|
case message.MsgTypeText:
|
||||||
|
case message.MsgTypeMiniProgramPage:
|
||||||
|
default:
|
||||||
|
err = message.ErrUnsupportedReply
|
||||||
|
return
|
||||||
|
}
|
||||||
|
msgData := reply.MsgData
|
||||||
|
value := reflect.ValueOf(msgData)
|
||||||
|
//msgData must be a ptr
|
||||||
|
kind := value.Kind().String()
|
||||||
|
if kind != "ptr" {
|
||||||
|
return message.ErrUnsupportedReply
|
||||||
|
}
|
||||||
|
params := make([]reflect.Value, 1)
|
||||||
|
params[0] = reflect.ValueOf(srv.RequestMsg.FromUserName)
|
||||||
|
value.MethodByName("SetToUserName").Call(params)
|
||||||
|
|
||||||
|
params[0] = reflect.ValueOf(srv.RequestMsg.ToUserName)
|
||||||
|
value.MethodByName("SetFromUserName").Call(params)
|
||||||
|
|
||||||
|
params[0] = reflect.ValueOf(srv.RequestMsg.MsgType)
|
||||||
|
value.MethodByName("SetMsgType").Call(params)
|
||||||
|
|
||||||
|
params[0] = reflect.ValueOf(util.GetCurrTS())
|
||||||
|
value.MethodByName("SetCreateTime").Call(params)
|
||||||
|
|
||||||
|
srv.ResponseMsg = msgData
|
||||||
|
srv.ResponseRawXMLMsg, err = xml.Marshal(msgData)
|
||||||
|
return
|
||||||
|
}
|
||||||
29
miniprogram/server/util.go
Normal file
29
miniprogram/server/util.go
Normal file
@@ -0,0 +1,29 @@
|
|||||||
|
package server
|
||||||
|
|
||||||
|
import "net/http"
|
||||||
|
|
||||||
|
var textContentType = []string{"text/plain; charset=utf-8"}
|
||||||
|
|
||||||
|
//Set http response Content-Type
|
||||||
|
func setContentType(w http.ResponseWriter, value []string) {
|
||||||
|
header := w.Header()
|
||||||
|
if val := header["Content-Type"]; len(val) == 0 {
|
||||||
|
header["Content-Type"] = value
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Query 查询 URL query string
|
||||||
|
func (srv *Server) Query(key string) string {
|
||||||
|
req := srv.Request
|
||||||
|
return req.URL.Query().Get(key)
|
||||||
|
}
|
||||||
|
|
||||||
|
// SetResponseWrite 设置回调返回值
|
||||||
|
func (srv *Server) SetResponseWrite(str string) {
|
||||||
|
setContentType(srv.Write, textContentType)
|
||||||
|
srv.Write.WriteHeader(http.StatusOK)
|
||||||
|
_, err := srv.Write.Write([]byte(str))
|
||||||
|
if err != nil {
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -46,9 +46,9 @@ type PaidResult struct {
|
|||||||
CouponID0 *string `xml:"coupon_id_0"`
|
CouponID0 *string `xml:"coupon_id_0"`
|
||||||
CouponID1 *string `xml:"coupon_id_1"`
|
CouponID1 *string `xml:"coupon_id_1"`
|
||||||
CouponID2 *string `xml:"coupon_id_2"`
|
CouponID2 *string `xml:"coupon_id_2"`
|
||||||
CouponFee0 *string `xml:"coupon_fee_0"`
|
CouponFeed0 *string `xml:"coupon_fee_0"`
|
||||||
CouponFee1 *string `xml:"coupon_fee_1"`
|
CouponFeed1 *string `xml:"coupon_fee_1"`
|
||||||
CouponFee2 *string `xml:"coupon_fee_2"`
|
CouponFeed2 *string `xml:"coupon_fee_2"`
|
||||||
|
|
||||||
TransactionID *string `xml:"transaction_id"`
|
TransactionID *string `xml:"transaction_id"`
|
||||||
OutTradeNo *string `xml:"out_trade_no"`
|
OutTradeNo *string `xml:"out_trade_no"`
|
||||||
|
|||||||
43
util/rsa.go
43
util/rsa.go
@@ -1,43 +0,0 @@
|
|||||||
package util
|
|
||||||
|
|
||||||
import (
|
|
||||||
"crypto/rand"
|
|
||||||
"crypto/rsa"
|
|
||||||
"crypto/x509"
|
|
||||||
"encoding/base64"
|
|
||||||
"encoding/pem"
|
|
||||||
"errors"
|
|
||||||
"fmt"
|
|
||||||
)
|
|
||||||
|
|
||||||
// RSADecrypt 数据解密
|
|
||||||
func RSADecrypt(privateKey string, ciphertext []byte) ([]byte, error) {
|
|
||||||
block, _ := pem.Decode([]byte(privateKey))
|
|
||||||
if block == nil {
|
|
||||||
return nil, errors.New("PrivateKey format error")
|
|
||||||
}
|
|
||||||
priv, err := x509.ParsePKCS1PrivateKey(block.Bytes)
|
|
||||||
if err != nil {
|
|
||||||
oldErr := err
|
|
||||||
key, err := x509.ParsePKCS8PrivateKey(block.Bytes)
|
|
||||||
if err != nil {
|
|
||||||
return nil, fmt.Errorf("ParsePKCS1PrivateKey error: %s, ParsePKCS8PrivateKey error: %s", oldErr.Error(), err.Error())
|
|
||||||
}
|
|
||||||
switch t := key.(type) {
|
|
||||||
case *rsa.PrivateKey:
|
|
||||||
priv = key.(*rsa.PrivateKey)
|
|
||||||
default:
|
|
||||||
return nil, fmt.Errorf("ParsePKCS1PrivateKey error: %s, ParsePKCS8PrivateKey error: Not supported privatekey format, should be *rsa.PrivateKey, got %T", oldErr.Error(), t)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return rsa.DecryptPKCS1v15(rand.Reader, priv, ciphertext)
|
|
||||||
}
|
|
||||||
|
|
||||||
// RSADecryptBase64 Base64解码后再次进行RSA解密
|
|
||||||
func RSADecryptBase64(privateKey string, cryptoText string) ([]byte, error) {
|
|
||||||
encryptedData, err := base64.StdEncoding.DecodeString(cryptoText)
|
|
||||||
if err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
return RSADecrypt(privateKey, encryptedData)
|
|
||||||
}
|
|
||||||
@@ -12,8 +12,6 @@ import (
|
|||||||
openConfig "github.com/silenceper/wechat/v2/openplatform/config"
|
openConfig "github.com/silenceper/wechat/v2/openplatform/config"
|
||||||
"github.com/silenceper/wechat/v2/pay"
|
"github.com/silenceper/wechat/v2/pay"
|
||||||
payConfig "github.com/silenceper/wechat/v2/pay/config"
|
payConfig "github.com/silenceper/wechat/v2/pay/config"
|
||||||
"github.com/silenceper/wechat/v2/work"
|
|
||||||
workConfig "github.com/silenceper/wechat/v2/work/config"
|
|
||||||
log "github.com/sirupsen/logrus"
|
log "github.com/sirupsen/logrus"
|
||||||
)
|
)
|
||||||
|
|
||||||
@@ -69,8 +67,3 @@ func (wc *Wechat) GetPay(cfg *payConfig.Config) *pay.Pay {
|
|||||||
func (wc *Wechat) GetOpenPlatform(cfg *openConfig.Config) *openplatform.OpenPlatform {
|
func (wc *Wechat) GetOpenPlatform(cfg *openConfig.Config) *openplatform.OpenPlatform {
|
||||||
return openplatform.NewOpenPlatform(cfg)
|
return openplatform.NewOpenPlatform(cfg)
|
||||||
}
|
}
|
||||||
|
|
||||||
// GetWork 获取企业微信的实例
|
|
||||||
func (wc *Wechat) GetWork(cfg *workConfig.Config) *work.Work {
|
|
||||||
return work.NewWork(cfg)
|
|
||||||
}
|
|
||||||
|
|||||||
@@ -1,15 +0,0 @@
|
|||||||
// Package config 企业微信config配置
|
|
||||||
package config
|
|
||||||
|
|
||||||
import (
|
|
||||||
"github.com/silenceper/wechat/v2/cache"
|
|
||||||
)
|
|
||||||
|
|
||||||
// Config config for 企业微信
|
|
||||||
type Config struct {
|
|
||||||
CorpID string `json:"corp_id"` // corp_id
|
|
||||||
CorpSecret string `json:"corp_secret"` // corp_secret,如果需要获取会话存档实例,当前参数请填写聊天内容存档的Secret,可以在企业微信管理端--管理工具--聊天内容存档查看
|
|
||||||
AgentID string `json:"agent_id"` // agent_id
|
|
||||||
Cache cache.Cache
|
|
||||||
RasPrivateKey string // 消息加密私钥,可以在企业微信管理端--管理工具--消息加密公钥查看对用公钥,私钥一般由自己保存
|
|
||||||
}
|
|
||||||
@@ -1,12 +0,0 @@
|
|||||||
package context
|
|
||||||
|
|
||||||
import (
|
|
||||||
"github.com/silenceper/wechat/v2/credential"
|
|
||||||
"github.com/silenceper/wechat/v2/work/config"
|
|
||||||
)
|
|
||||||
|
|
||||||
// Context struct
|
|
||||||
type Context struct {
|
|
||||||
*config.Config
|
|
||||||
credential.AccessTokenHandle
|
|
||||||
}
|
|
||||||
@@ -1,95 +0,0 @@
|
|||||||
企业微信会话存档SDK(基于企业微信C版官方SDK封装),暂时只支持在`linux`环境下使用当前SDK。
|
|
||||||
|
|
||||||
### 官方文档地址
|
|
||||||
https://open.work.weixin.qq.com/api/doc/90000/90135/91774
|
|
||||||
|
|
||||||
### 使用方式
|
|
||||||
|
|
||||||
1、安装 go module
|
|
||||||
> go get -u github.com/silenceper/wechat/v2
|
|
||||||
|
|
||||||
2、从 `github.com/silenceper/wechat/v2/work/msgaudit/lib` 文件夹下复制 `libWeWorkFinanceSdk_C.so` 动态库文件到系统动态链接库默认文件夹下,或者复制到任意文件夹并在当前文件夹下执行 `export LD_LIBRARY_PATH=$(pwd)`命令设置动态链接库检索地址后即可正常使用
|
|
||||||
|
|
||||||
### Example
|
|
||||||
|
|
||||||
```go
|
|
||||||
package main
|
|
||||||
|
|
||||||
import (
|
|
||||||
"bytes"
|
|
||||||
"fmt"
|
|
||||||
"github.com/silenceper/wechat/v2"
|
|
||||||
"github.com/silenceper/wechat/v2/work/msgaudit"
|
|
||||||
"github.com/silenceper/wechat/v2/work/config"
|
|
||||||
"io/ioutil"
|
|
||||||
"os"
|
|
||||||
"path"
|
|
||||||
)
|
|
||||||
|
|
||||||
func main() {
|
|
||||||
//初始化客户端
|
|
||||||
wechatClient := wechat.NewWechat()
|
|
||||||
|
|
||||||
workClient := wechatClient.NewWork(&config.Config{
|
|
||||||
CorpID: "XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX",
|
|
||||||
CorpSecret: "XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX",
|
|
||||||
RasPrivateKey: "XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX",
|
|
||||||
})
|
|
||||||
|
|
||||||
client, err := workClient.GetMsgAudit()
|
|
||||||
if err != nil {
|
|
||||||
fmt.Printf("SDK 初始化失败:%v \n", err)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
//同步消息
|
|
||||||
chatDataList, err := client.GetChatData(0, 100, "", "", 3)
|
|
||||||
if err != nil {
|
|
||||||
fmt.Printf("消息同步失败:%v \n", err)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
for _, chatData := range chatDataList {
|
|
||||||
//消息解密
|
|
||||||
chatInfo, err := client.DecryptData(chatData.EncryptRandomKey, chatData.EncryptChatMsg)
|
|
||||||
if err != nil {
|
|
||||||
fmt.Printf("消息解密失败:%v \n", err)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
if chatInfo.Type == "image" {
|
|
||||||
image, _ := chatInfo.GetImageMessage()
|
|
||||||
sdkfileid := image.Image.SdkFileId
|
|
||||||
|
|
||||||
isFinish := false
|
|
||||||
buffer := bytes.Buffer{}
|
|
||||||
for !isFinish {
|
|
||||||
//获取媒体数据
|
|
||||||
mediaData, err := client.GetMediaData("", sdkfileid, "", "", 5)
|
|
||||||
if err != nil {
|
|
||||||
fmt.Printf("媒体数据拉取失败:%v \n", err)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
buffer.Write(mediaData.Data)
|
|
||||||
if mediaData.IsFinish {
|
|
||||||
isFinish = mediaData.IsFinish
|
|
||||||
}
|
|
||||||
}
|
|
||||||
filePath, _ := os.Getwd()
|
|
||||||
filePath = path.Join(filePath, "test.png")
|
|
||||||
err := ioutil.WriteFile(filePath, buffer.Bytes(), 0666)
|
|
||||||
if err != nil {
|
|
||||||
fmt.Printf("文件存储失败:%v \n", err)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
break
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
//释放SDK实例
|
|
||||||
client.Free()
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
```
|
|
||||||
@@ -1,207 +0,0 @@
|
|||||||
package msgaudit
|
|
||||||
|
|
||||||
import "encoding/json"
|
|
||||||
|
|
||||||
// ChatDataResponse 会话存档消息响应数据
|
|
||||||
type ChatDataResponse struct {
|
|
||||||
Error
|
|
||||||
ChatDataList []ChatData `json:"chatdata,omitempty"`
|
|
||||||
}
|
|
||||||
|
|
||||||
// IsError 判断是否正确响应
|
|
||||||
func (c ChatDataResponse) IsError() bool {
|
|
||||||
return c.ErrCode != 0
|
|
||||||
}
|
|
||||||
|
|
||||||
// ChatData 会话存档原始数据
|
|
||||||
type ChatData struct {
|
|
||||||
Seq uint64 `json:"seq,omitempty"` // 消息的seq值,标识消息的序号。再次拉取需要带上上次回包中最大的seq。Uint64类型,范围0-pow(2,64)-1
|
|
||||||
MsgID string `json:"msgid,omitempty"` // 消息id,消息的唯一标识,企业可以使用此字段进行消息去重。
|
|
||||||
PublickeyVer uint32 `json:"publickey_ver,omitempty"` // 加密此条消息使用的公钥版本号。
|
|
||||||
EncryptRandomKey string `json:"encrypt_random_key,omitempty"` // 使用publickey_ver指定版本的公钥进行非对称加密后base64加密的内容,需要业务方先base64 decode处理后,再使用指定版本的私钥进行解密,得出内容。
|
|
||||||
EncryptChatMsg string `json:"encrypt_chat_msg,omitempty"` // 消息密文。需要业务方使用将encrypt_random_key解密得到的内容,与encrypt_chat_msg,传入sdk接口DecryptData,得到消息明文。
|
|
||||||
}
|
|
||||||
|
|
||||||
// ChatMessage 会话存档消息
|
|
||||||
type ChatMessage struct {
|
|
||||||
ID string // 消息id,消息的唯一标识,企业可以使用此字段进行消息去重。
|
|
||||||
From string // 消息发送方id。同一企业内容为userid,非相同企业为external_userid。消息如果是机器人发出,也为external_userid。
|
|
||||||
ToList []string // 消息接收方列表,可能是多个,同一个企业内容为userid,非相同企业为external_userid。
|
|
||||||
Action string // 消息动作,目前有send(发送消息)/recall(撤回消息)/switch(切换企业日志)三种类型。
|
|
||||||
Type string // 消息类型
|
|
||||||
originData []byte // 原始消息对象
|
|
||||||
}
|
|
||||||
|
|
||||||
// GetOriginMessage 获取消息原始数据
|
|
||||||
func (c ChatMessage) GetOriginMessage() (msg map[string]interface{}, err error) {
|
|
||||||
err = json.Unmarshal(c.originData, &msg)
|
|
||||||
return msg, err
|
|
||||||
}
|
|
||||||
|
|
||||||
// GetTextMessage 获取文本消息
|
|
||||||
func (c ChatMessage) GetTextMessage() (msg TextMessage, err error) {
|
|
||||||
err = json.Unmarshal(c.originData, &msg)
|
|
||||||
return msg, err
|
|
||||||
}
|
|
||||||
|
|
||||||
// GetImageMessage 获取图片消息
|
|
||||||
func (c ChatMessage) GetImageMessage() (msg ImageMessage, err error) {
|
|
||||||
err = json.Unmarshal(c.originData, &msg)
|
|
||||||
return msg, err
|
|
||||||
}
|
|
||||||
|
|
||||||
// GetRevokeMessage 获取撤回消息
|
|
||||||
func (c ChatMessage) GetRevokeMessage() (msg RevokeMessage, err error) {
|
|
||||||
err = json.Unmarshal(c.originData, &msg)
|
|
||||||
return msg, err
|
|
||||||
}
|
|
||||||
|
|
||||||
// GetAgreeMessage 获取同意会话聊天内容
|
|
||||||
func (c ChatMessage) GetAgreeMessage() (msg AgreeMessage, err error) {
|
|
||||||
err = json.Unmarshal(c.originData, &msg)
|
|
||||||
return msg, err
|
|
||||||
}
|
|
||||||
|
|
||||||
// GetVoiceMessage 获取语音消息
|
|
||||||
func (c ChatMessage) GetVoiceMessage() (msg VoiceMessage, err error) {
|
|
||||||
err = json.Unmarshal(c.originData, &msg)
|
|
||||||
return msg, err
|
|
||||||
}
|
|
||||||
|
|
||||||
// GetVideoMessage 获取视频消息
|
|
||||||
func (c ChatMessage) GetVideoMessage() (msg VideoMessage, err error) {
|
|
||||||
err = json.Unmarshal(c.originData, &msg)
|
|
||||||
return msg, err
|
|
||||||
}
|
|
||||||
|
|
||||||
// GetCardMessage 获取名片消息
|
|
||||||
func (c ChatMessage) GetCardMessage() (msg CardMessage, err error) {
|
|
||||||
err = json.Unmarshal(c.originData, &msg)
|
|
||||||
return msg, err
|
|
||||||
}
|
|
||||||
|
|
||||||
// GetLocationMessage 获取位置消息
|
|
||||||
func (c ChatMessage) GetLocationMessage() (msg LocationMessage, err error) {
|
|
||||||
err = json.Unmarshal(c.originData, &msg)
|
|
||||||
return msg, err
|
|
||||||
}
|
|
||||||
|
|
||||||
// GetEmotionMessage 获取表情消息
|
|
||||||
func (c ChatMessage) GetEmotionMessage() (msg EmotionMessage, err error) {
|
|
||||||
err = json.Unmarshal(c.originData, &msg)
|
|
||||||
return msg, err
|
|
||||||
}
|
|
||||||
|
|
||||||
// GetFileMessage 获取文件消息
|
|
||||||
func (c ChatMessage) GetFileMessage() (msg FileMessage, err error) {
|
|
||||||
err = json.Unmarshal(c.originData, &msg)
|
|
||||||
return msg, err
|
|
||||||
}
|
|
||||||
|
|
||||||
// GetLinkMessage 获取链接消息
|
|
||||||
func (c ChatMessage) GetLinkMessage() (msg LinkMessage, err error) {
|
|
||||||
err = json.Unmarshal(c.originData, &msg)
|
|
||||||
return msg, err
|
|
||||||
}
|
|
||||||
|
|
||||||
// GetWeappMessage 获取小程序消息
|
|
||||||
func (c ChatMessage) GetWeappMessage() (msg WeappMessage, err error) {
|
|
||||||
err = json.Unmarshal(c.originData, &msg)
|
|
||||||
return msg, err
|
|
||||||
}
|
|
||||||
|
|
||||||
// GetChatRecordMessage 获取会话记录消息
|
|
||||||
func (c ChatMessage) GetChatRecordMessage() (msg ChatRecordMessage, err error) {
|
|
||||||
err = json.Unmarshal(c.originData, &msg)
|
|
||||||
return msg, err
|
|
||||||
}
|
|
||||||
|
|
||||||
// GetTodoMessage 获取待办消息
|
|
||||||
func (c ChatMessage) GetTodoMessage() (msg TodoMessage, err error) {
|
|
||||||
err = json.Unmarshal(c.originData, &msg)
|
|
||||||
return msg, err
|
|
||||||
}
|
|
||||||
|
|
||||||
// GetVoteMessage 获取投票消息
|
|
||||||
func (c ChatMessage) GetVoteMessage() (msg VoteMessage, err error) {
|
|
||||||
err = json.Unmarshal(c.originData, &msg)
|
|
||||||
return msg, err
|
|
||||||
}
|
|
||||||
|
|
||||||
// GetCollectMessage 获取填表消息
|
|
||||||
func (c ChatMessage) GetCollectMessage() (msg CollectMessage, err error) {
|
|
||||||
err = json.Unmarshal(c.originData, &msg)
|
|
||||||
return msg, err
|
|
||||||
}
|
|
||||||
|
|
||||||
// GetRedpacketMessage 获取红包消息
|
|
||||||
func (c ChatMessage) GetRedpacketMessage() (msg RedpacketMessage, err error) {
|
|
||||||
err = json.Unmarshal(c.originData, &msg)
|
|
||||||
return msg, err
|
|
||||||
}
|
|
||||||
|
|
||||||
// GetMeetingMessage 获取会议邀请消息
|
|
||||||
func (c ChatMessage) GetMeetingMessage() (msg MeetingMessage, err error) {
|
|
||||||
err = json.Unmarshal(c.originData, &msg)
|
|
||||||
return msg, err
|
|
||||||
}
|
|
||||||
|
|
||||||
// GetDocMessage 获取在线文档消息
|
|
||||||
func (c ChatMessage) GetDocMessage() (msg DocMessage, err error) {
|
|
||||||
err = json.Unmarshal(c.originData, &msg)
|
|
||||||
return msg, err
|
|
||||||
}
|
|
||||||
|
|
||||||
// GetMarkdownMessage 获取MarkDown格式消息
|
|
||||||
func (c ChatMessage) GetMarkdownMessage() (msg MarkdownMessage, err error) {
|
|
||||||
err = json.Unmarshal(c.originData, &msg)
|
|
||||||
return msg, err
|
|
||||||
}
|
|
||||||
|
|
||||||
// GetNewsMessage 获取图文消息
|
|
||||||
func (c ChatMessage) GetNewsMessage() (msg NewsMessage, err error) {
|
|
||||||
err = json.Unmarshal(c.originData, &msg)
|
|
||||||
return msg, err
|
|
||||||
}
|
|
||||||
|
|
||||||
// GetCalendarMessage 获取日程消息
|
|
||||||
func (c ChatMessage) GetCalendarMessage() (msg CalendarMessage, err error) {
|
|
||||||
err = json.Unmarshal(c.originData, &msg)
|
|
||||||
return msg, err
|
|
||||||
}
|
|
||||||
|
|
||||||
// GetMixedMessage 获取混合消息
|
|
||||||
func (c ChatMessage) GetMixedMessage() (msg MixedMessage, err error) {
|
|
||||||
err = json.Unmarshal(c.originData, &msg)
|
|
||||||
return msg, err
|
|
||||||
}
|
|
||||||
|
|
||||||
// GetMeetingVoiceCallMessage 获取音频存档消息
|
|
||||||
func (c ChatMessage) GetMeetingVoiceCallMessage() (msg MeetingVoiceCallMessage, err error) {
|
|
||||||
err = json.Unmarshal(c.originData, &msg)
|
|
||||||
return msg, err
|
|
||||||
}
|
|
||||||
|
|
||||||
// GetVoipDocShareMessage 获取音频共享消息
|
|
||||||
func (c ChatMessage) GetVoipDocShareMessage() (msg VoipDocShareMessage, err error) {
|
|
||||||
err = json.Unmarshal(c.originData, &msg)
|
|
||||||
return msg, err
|
|
||||||
}
|
|
||||||
|
|
||||||
// GetExternalRedPacketMessage 获取互通红包消息
|
|
||||||
func (c ChatMessage) GetExternalRedPacketMessage() (msg ExternalRedPacketMessage, err error) {
|
|
||||||
err = json.Unmarshal(c.originData, &msg)
|
|
||||||
return msg, err
|
|
||||||
}
|
|
||||||
|
|
||||||
// GetSphFeedMessage 获取视频号消息
|
|
||||||
func (c ChatMessage) GetSphFeedMessage() (msg SphFeedMessage, err error) {
|
|
||||||
err = json.Unmarshal(c.originData, &msg)
|
|
||||||
return msg, err
|
|
||||||
}
|
|
||||||
|
|
||||||
// GetSwitchMessage 获取切换企业日志
|
|
||||||
func (c ChatMessage) GetSwitchMessage() (msg SwitchMessage, err error) {
|
|
||||||
err = json.Unmarshal(c.originData, &msg)
|
|
||||||
return msg, err
|
|
||||||
}
|
|
||||||
@@ -1,260 +0,0 @@
|
|||||||
package msgaudit
|
|
||||||
|
|
||||||
// #cgo LDFLAGS: -L${SRCDIR}/lib -lWeWorkFinanceSdk_C
|
|
||||||
// #cgo CFLAGS: -I ./lib/
|
|
||||||
// #include <stdlib.h>
|
|
||||||
// #include "WeWorkFinanceSdk_C.h"
|
|
||||||
import "C"
|
|
||||||
import (
|
|
||||||
"encoding/json"
|
|
||||||
"github.com/silenceper/wechat/v2/util"
|
|
||||||
"github.com/silenceper/wechat/v2/work/config"
|
|
||||||
"unsafe"
|
|
||||||
)
|
|
||||||
|
|
||||||
// Client 会话存档
|
|
||||||
type Client struct {
|
|
||||||
ptr *C.WeWorkFinanceSdk_t
|
|
||||||
privateKey string
|
|
||||||
}
|
|
||||||
|
|
||||||
// NewClient 初始会话会话存档实例
|
|
||||||
/**
|
|
||||||
* 初始化函数
|
|
||||||
* Return值=0表示该API调用成功
|
|
||||||
*
|
|
||||||
* @param [in] sdk NewSdk返回的sdk指针
|
|
||||||
* @param [in] corpid 调用企业的企业id,例如:wwd08c8exxxx5ab44d,可以在企业微信管理端--我的企业--企业信息查看
|
|
||||||
* @param [in] secret 聊天内容存档的Secret,可以在企业微信管理端--管理工具--聊天内容存档查看
|
|
||||||
* @param [in] privateKey 消息加密私钥,可以在企业微信管理端--管理工具--消息加密公钥查看对用公钥,私钥一般由自己保存
|
|
||||||
*
|
|
||||||
*
|
|
||||||
* @return 返回是否初始化成功
|
|
||||||
* 0 - 成功
|
|
||||||
* !=0 - 失败
|
|
||||||
*/
|
|
||||||
func NewClient(cfg *config.Config) (*Client, error) {
|
|
||||||
ptr := C.NewSdk()
|
|
||||||
corpIDC := C.CString(cfg.CorpID)
|
|
||||||
corpSecretC := C.CString(cfg.CorpSecret)
|
|
||||||
defer func() {
|
|
||||||
C.free(unsafe.Pointer(corpIDC))
|
|
||||||
C.free(unsafe.Pointer(corpSecretC))
|
|
||||||
}()
|
|
||||||
retC := C.Init(ptr, corpIDC, corpSecretC)
|
|
||||||
ret := int(retC)
|
|
||||||
if ret != 0 {
|
|
||||||
return nil, NewSDKErr(ret)
|
|
||||||
}
|
|
||||||
return &Client{
|
|
||||||
ptr: ptr,
|
|
||||||
privateKey: cfg.RasPrivateKey,
|
|
||||||
}, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
// Free 释放SDK实例是可调用该方法释放内存
|
|
||||||
func (s *Client) Free() {
|
|
||||||
if s.ptr == nil {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
C.DestroySdk(s.ptr)
|
|
||||||
s.ptr = nil
|
|
||||||
}
|
|
||||||
|
|
||||||
// GetChatData 拉取聊天记录函数
|
|
||||||
/**
|
|
||||||
* 拉取聊天记录函数
|
|
||||||
*
|
|
||||||
*
|
|
||||||
* @param [in] seq 从指定的seq开始拉取消息,注意的是返回的消息从seq+1开始返回,seq为之前接口返回的最大seq值。首次使用请使用seq:0
|
|
||||||
* @param [in] limit 一次拉取的消息条数,最大值1000条,超过1000条会返回错误
|
|
||||||
* @param [in] proxy 使用代理的请求,需要传入代理的链接。如:socks5://10.0.0.1:8081 或者 http://10.0.0.1:8081
|
|
||||||
* @param [in] passwd 代理账号密码,需要传入代理的账号密码。如 user_name:passwd_123
|
|
||||||
* @param [in] timeout 超时时间,单位秒
|
|
||||||
* @return chatDatas 返回本次拉取消息的数据,slice结构体.内容包括errcode/errmsg,以及每条消息内容。示例如下:
|
|
||||||
|
|
||||||
{"errcode":0,"errmsg":"ok","chatdata":[{"seq":196,"msgid":"CAQQ2fbb4QUY0On2rYSAgAMgip/yzgs=","publickey_ver":3,"encrypt_random_key":"ftJ+uz3n/z1DsxlkwxNgE+mL38H42/KCvN8T60gbbtPD+Rta1hKTuQPzUzO6Hzne97MgKs7FfdDxDck/v8cDT6gUVjA2tZ/M7euSD0L66opJ/IUeBtpAtvgVSD5qhlaQjvfKJc/zPMGNK2xCLFYqwmQBZXbNT7uA69Fflm512nZKW/piK2RKdYJhRyvQnA1ISxK097sp9WlEgDg250fM5tgwMjujdzr7ehK6gtVBUFldNSJS7ndtIf6aSBfaLktZgwHZ57ONewWq8GJe7WwQf1hwcDbCh7YMG8nsweEwhDfUz+u8rz9an+0lgrYMZFRHnmzjgmLwrR7B/32Qxqd79A==","encrypt_chat_msg":"898WSfGMnIeytTsea7Rc0WsOocs0bIAerF6de0v2cFwqo9uOxrW9wYe5rCjCHHH5bDrNvLxBE/xOoFfcwOTYX0HQxTJaH0ES9OHDZ61p8gcbfGdJKnq2UU4tAEgGb8H+Q9n8syRXIjaI3KuVCqGIi4QGHFmxWenPFfjF/vRuPd0EpzUNwmqfUxLBWLpGhv+dLnqiEOBW41Zdc0OO0St6E+JeIeHlRZAR+E13Isv9eS09xNbF0qQXWIyNUi+ucLr5VuZnPGXBrSfvwX8f0QebTwpy1tT2zvQiMM2MBugKH6NuMzzuvEsXeD+6+3VRqL"}]}
|
|
||||||
*/
|
|
||||||
func (s *Client) GetChatData(seq uint64, limit uint64, proxy string, passwd string, timeout int) ([]ChatData, error) {
|
|
||||||
proxyC := C.CString(proxy)
|
|
||||||
passwdC := C.CString(passwd)
|
|
||||||
chatSlice := C.NewSlice()
|
|
||||||
defer func() {
|
|
||||||
C.free(unsafe.Pointer(proxyC))
|
|
||||||
C.free(unsafe.Pointer(passwdC))
|
|
||||||
C.FreeSlice(chatSlice)
|
|
||||||
}()
|
|
||||||
|
|
||||||
if s.ptr == nil {
|
|
||||||
return nil, NewSDKErr(10002)
|
|
||||||
}
|
|
||||||
|
|
||||||
retC := C.GetChatData(s.ptr, C.ulonglong(seq), C.uint(limit), proxyC, passwdC, C.int(timeout), chatSlice)
|
|
||||||
ret := int(retC)
|
|
||||||
if ret != 0 {
|
|
||||||
return nil, NewSDKErr(ret)
|
|
||||||
}
|
|
||||||
buf := s.GetContentFromSlice(chatSlice)
|
|
||||||
|
|
||||||
var data ChatDataResponse
|
|
||||||
err := json.Unmarshal(buf, &data)
|
|
||||||
if err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
if data.ErrCode != 0 {
|
|
||||||
return nil, data.Error
|
|
||||||
}
|
|
||||||
return data.ChatDataList, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
// GetRawChatData 拉取聊天记录函数
|
|
||||||
/**
|
|
||||||
* 拉取聊天记录函数
|
|
||||||
*
|
|
||||||
*
|
|
||||||
* @param [in] seq 从指定的seq开始拉取消息,注意的是返回的消息从seq+1开始返回,seq为之前接口返回的最大seq值。首次使用请使用seq:0
|
|
||||||
* @param [in] limit 一次拉取的消息条数,最大值1000条,超过1000条会返回错误
|
|
||||||
* @param [in] proxy 使用代理的请求,需要传入代理的链接。如:socks5://10.0.0.1:8081 或者 http://10.0.0.1:8081
|
|
||||||
* @param [in] passwd 代理账号密码,需要传入代理的账号密码。如 user_name:passwd_123
|
|
||||||
* @param [in] timeout 超时时间,单位秒
|
|
||||||
* @return chatDatas 返回本次拉取消息的数据,slice结构体.内容包括errcode/errmsg,以及每条消息内容。示例如下:
|
|
||||||
|
|
||||||
{"errcode":0,"errmsg":"ok","chatdata":[{"seq":196,"msgid":"CAQQ2fbb4QUY0On2rYSAgAMgip/yzgs=","publickey_ver":3,"encrypt_random_key":"ftJ+uz3n/z1DsxlkwxNgE+mL38H42/KCvN8T60gbbtPD+Rta1hKTuQPzUzO6Hzne97MgKs7FfdDxDck/v8cDT6gUVjA2tZ/M7euSD0L66opJ/IUeBtpAtvgVSD5qhlaQjvfKJc/zPMGNK2xCLFYqwmQBZXbNT7uA69Fflm512nZKW/piK2RKdYJhRyvQnA1ISxK097sp9WlEgDg250fM5tgwMjujdzr7ehK6gtVBUFldNSJS7ndtIf6aSBfaLktZgwHZ57ONewWq8GJe7WwQf1hwcDbCh7YMG8nsweEwhDfUz+u8rz9an+0lgrYMZFRHnmzjgmLwrR7B/32Qxqd79A==","encrypt_chat_msg":"898WSfGMnIeytTsea7Rc0WsOocs0bIAerF6de0v2cFwqo9uOxrW9wYe5rCjCHHH5bDrNvLxBE/xOoFfcwOTYX0HQxTJaH0ES9OHDZ61p8gcbfGdJKnq2UU4tAEgGb8H+Q9n8syRXIjaI3KuVCqGIi4QGHFmxWenPFfjF/vRuPd0EpzUNwmqfUxLBWLpGhv+dLnqiEOBW41Zdc0OO0St6E+JeIeHlRZAR+E13Isv9eS09xNbF0qQXWIyNUi+ucLr5VuZnPGXBrSfvwX8f0QebTwpy1tT2zvQiMM2MBugKH6NuMzzuvEsXeD+6+3VRqL"}]}
|
|
||||||
*/
|
|
||||||
func (s *Client) GetRawChatData(seq uint64, limit uint64, proxy string, passwd string, timeout int) (ChatDataResponse, error) {
|
|
||||||
proxyC := C.CString(proxy)
|
|
||||||
passwdC := C.CString(passwd)
|
|
||||||
chatSlice := C.NewSlice()
|
|
||||||
defer func() {
|
|
||||||
C.free(unsafe.Pointer(proxyC))
|
|
||||||
C.free(unsafe.Pointer(passwdC))
|
|
||||||
C.FreeSlice(chatSlice)
|
|
||||||
}()
|
|
||||||
|
|
||||||
if s.ptr == nil {
|
|
||||||
return ChatDataResponse{}, NewSDKErr(10002)
|
|
||||||
}
|
|
||||||
|
|
||||||
retC := C.GetChatData(s.ptr, C.ulonglong(seq), C.uint(limit), proxyC, passwdC, C.int(timeout), chatSlice)
|
|
||||||
ret := int(retC)
|
|
||||||
if ret != 0 {
|
|
||||||
return ChatDataResponse{}, NewSDKErr(ret)
|
|
||||||
}
|
|
||||||
buf := s.GetContentFromSlice(chatSlice)
|
|
||||||
|
|
||||||
var data ChatDataResponse
|
|
||||||
err := json.Unmarshal(buf, &data)
|
|
||||||
if err != nil {
|
|
||||||
return ChatDataResponse{}, err
|
|
||||||
}
|
|
||||||
return data, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
// DecryptData 解析密文.企业微信自有解密内容
|
|
||||||
/**
|
|
||||||
* @brief 解析密文.企业微信自有解密内容
|
|
||||||
* @param [in] encrypt_key, getchatdata返回的encrypt_random_key,使用企业自持对应版本秘钥RSA解密后的内容
|
|
||||||
* @param [in] encrypt_msg, getchatdata返回的encrypt_chat_msg
|
|
||||||
* @param [out] msg, 解密的消息明文
|
|
||||||
* @return 返回是否调用成功
|
|
||||||
* 0 - 成功
|
|
||||||
* !=0 - 失败
|
|
||||||
*/
|
|
||||||
func (s *Client) DecryptData(encryptRandomKey string, encryptMsg string) (msg ChatMessage, err error) {
|
|
||||||
encryptKey, err := util.RSADecryptBase64(s.privateKey, encryptRandomKey)
|
|
||||||
if err != nil {
|
|
||||||
return msg, err
|
|
||||||
}
|
|
||||||
encryptKeyC := C.CString(string(encryptKey))
|
|
||||||
encryptMsgC := C.CString(encryptMsg)
|
|
||||||
msgSlice := C.NewSlice()
|
|
||||||
defer func() {
|
|
||||||
C.free(unsafe.Pointer(encryptKeyC))
|
|
||||||
C.free(unsafe.Pointer(encryptMsgC))
|
|
||||||
C.FreeSlice(msgSlice)
|
|
||||||
}()
|
|
||||||
|
|
||||||
retC := C.DecryptData(encryptKeyC, encryptMsgC, msgSlice)
|
|
||||||
ret := int(retC)
|
|
||||||
if ret != 0 {
|
|
||||||
return msg, NewSDKErr(ret)
|
|
||||||
}
|
|
||||||
buf := s.GetContentFromSlice(msgSlice)
|
|
||||||
|
|
||||||
// handle illegal escape character in text
|
|
||||||
for i := 0; i < len(buf); {
|
|
||||||
if buf[i] < 0x20 {
|
|
||||||
buf = append(buf[:i], buf[i+1:]...)
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
i++
|
|
||||||
}
|
|
||||||
|
|
||||||
var baseMessage BaseMessage
|
|
||||||
err = json.Unmarshal(buf, &baseMessage)
|
|
||||||
if err != nil {
|
|
||||||
return msg, err
|
|
||||||
}
|
|
||||||
|
|
||||||
msg.ID = baseMessage.MsgID
|
|
||||||
msg.From = baseMessage.From
|
|
||||||
msg.ToList = baseMessage.ToList
|
|
||||||
msg.Action = baseMessage.Action
|
|
||||||
msg.Type = baseMessage.MsgType
|
|
||||||
msg.originData = buf
|
|
||||||
return msg, err
|
|
||||||
}
|
|
||||||
|
|
||||||
// GetMediaData 拉取媒体消息函数
|
|
||||||
/**
|
|
||||||
* 拉取媒体消息函数
|
|
||||||
* Return值=0表示该API调用成功
|
|
||||||
*
|
|
||||||
*
|
|
||||||
* @param [in] sdk NewSdk返回的sdk指针
|
|
||||||
* @param [in] sdkFileid 从GetChatData返回的聊天消息中,媒体消息包括的sdkfileid
|
|
||||||
* @param [in] proxy 使用代理的请求,需要传入代理的链接。如:socks5://10.0.0.1:8081 或者 http://10.0.0.1:8081
|
|
||||||
* @param [in] passwd 代理账号密码,需要传入代理的账号密码。如 user_name:passwd_123
|
|
||||||
* @param [in] indexbuf 媒体消息分片拉取,需要填入每次拉取的索引信息。首次不需要填写,默认拉取512k,后续每次调用只需要将上次调用返回的outindexbuf填入即可。
|
|
||||||
* @param [in] timeout 超时时间,单位秒
|
|
||||||
* @param [out] media_data 返回本次拉取的媒体数据.MediaData结构体.内容包括data(数据内容)/outindexbuf(下次索引)/is_finish(拉取完成标记)
|
|
||||||
|
|
||||||
*
|
|
||||||
* @return 返回是否调用成功
|
|
||||||
* 0 - 成功
|
|
||||||
* !=0 - 失败
|
|
||||||
*/
|
|
||||||
func (s *Client) GetMediaData(indexBuf string, sdkFileID string, proxy string, passwd string, timeout int) (*MediaData, error) {
|
|
||||||
indexBufC := C.CString(indexBuf)
|
|
||||||
sdkFileIDC := C.CString(sdkFileID)
|
|
||||||
proxyC := C.CString(proxy)
|
|
||||||
passwdC := C.CString(passwd)
|
|
||||||
mediaDataC := C.NewMediaData()
|
|
||||||
defer func() {
|
|
||||||
C.free(unsafe.Pointer(indexBufC))
|
|
||||||
C.free(unsafe.Pointer(sdkFileIDC))
|
|
||||||
C.free(unsafe.Pointer(proxyC))
|
|
||||||
C.free(unsafe.Pointer(passwdC))
|
|
||||||
C.FreeMediaData(mediaDataC)
|
|
||||||
}()
|
|
||||||
|
|
||||||
if s.ptr == nil {
|
|
||||||
return nil, NewSDKErr(10002)
|
|
||||||
}
|
|
||||||
|
|
||||||
retC := C.GetMediaData(s.ptr, indexBufC, sdkFileIDC, proxyC, passwdC, C.int(timeout), mediaDataC)
|
|
||||||
ret := int(retC)
|
|
||||||
if ret != 0 {
|
|
||||||
return nil, NewSDKErr(ret)
|
|
||||||
}
|
|
||||||
return &MediaData{
|
|
||||||
OutIndexBuf: C.GoString(C.GetOutIndexBuf(mediaDataC)),
|
|
||||||
Data: C.GoBytes(unsafe.Pointer(C.GetData(mediaDataC)), C.GetDataLen(mediaDataC)),
|
|
||||||
IsFinish: int(C.IsMediaDataFinish(mediaDataC)) == 1,
|
|
||||||
}, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
// GetContentFromSlice 从切片内获取内容
|
|
||||||
func (s *Client) GetContentFromSlice(slice *C.struct_Slice_t) []byte {
|
|
||||||
return C.GoBytes(unsafe.Pointer(C.GetContentFromSlice(slice)), C.GetSliceLen(slice))
|
|
||||||
}
|
|
||||||
@@ -1,8 +0,0 @@
|
|||||||
package msgaudit
|
|
||||||
|
|
||||||
// Config 会话存档初始化参数
|
|
||||||
type Config struct {
|
|
||||||
CorpID string // 调用企业的企业id,例如:wwd08c8exxxx5ab44d,可以在企业微信管理端--我的企业--企业信息查看
|
|
||||||
CorpSecret string // 聊天内容存档的Secret,可以在企业微信管理端--管理工具--聊天内容存档查看
|
|
||||||
RasPrivateKey string // 消息加密私钥,可以在企业微信管理端--管理工具--消息加密公钥查看对用公钥,私钥一般由自己保存
|
|
||||||
}
|
|
||||||
@@ -1,76 +0,0 @@
|
|||||||
package msgaudit
|
|
||||||
|
|
||||||
import (
|
|
||||||
"fmt"
|
|
||||||
)
|
|
||||||
|
|
||||||
//返回码 错误说明
|
|
||||||
//10000 参数错误,请求参数错误
|
|
||||||
//10001 网络错误,网络请求错误
|
|
||||||
//10002 数据解析失败
|
|
||||||
//10003 系统失败
|
|
||||||
//10004 密钥错误导致加密失败
|
|
||||||
//10005 fileid错误
|
|
||||||
//10006 解密失败
|
|
||||||
//10007 找不到消息加密版本的私钥,需要重新传入私钥对
|
|
||||||
//10008 解析encrypt_key出错
|
|
||||||
//10009 ip非法
|
|
||||||
//10010 数据过期
|
|
||||||
//10011 证书错误
|
|
||||||
const (
|
|
||||||
SDKErrMsg = "sdk failed"
|
|
||||||
SDKParamsErrMsg = "参数错误,请求参数错误"
|
|
||||||
SDKNetworkErrMsg = "网络错误,网络请求错误"
|
|
||||||
SDKParseErrMsg = "数据解析失败"
|
|
||||||
SDKSystemErrMsg = "系统失败"
|
|
||||||
SDKSecretErrMsg = "密钥错误导致加密失败"
|
|
||||||
SDKFileIDErrMsg = "fileid错误"
|
|
||||||
SDKDecryptErrMsg = "解密失败"
|
|
||||||
SDKSecretMissErrMsg = "找不到消息加密版本的私钥,需要重新传入私钥对"
|
|
||||||
SDKEncryptKeyErrMsg = "解析encrypt_key出错"
|
|
||||||
SDKIPNotWhiteListErrMsg = "ip非法"
|
|
||||||
SDKDataExpiredErrMsg = "数据过期"
|
|
||||||
SDKTokenExpiredErrMsg = "证书过期"
|
|
||||||
)
|
|
||||||
|
|
||||||
// Error 错误
|
|
||||||
type Error struct {
|
|
||||||
ErrCode int `json:"errcode,omitempty"`
|
|
||||||
ErrMsg string `json:"errmsg,omitempty"`
|
|
||||||
}
|
|
||||||
|
|
||||||
func (e Error) Error() string {
|
|
||||||
return fmt.Sprintf("%d:%s", e.ErrCode, e.ErrMsg)
|
|
||||||
}
|
|
||||||
|
|
||||||
// NewSDKErr 初始化新的SDK错误
|
|
||||||
func NewSDKErr(code int) Error {
|
|
||||||
msg := ""
|
|
||||||
switch code {
|
|
||||||
case 10000:
|
|
||||||
msg = SDKParamsErrMsg
|
|
||||||
case 10001:
|
|
||||||
msg = SDKNetworkErrMsg
|
|
||||||
case 10002:
|
|
||||||
msg = SDKParseErrMsg
|
|
||||||
case 10003:
|
|
||||||
msg = SDKSystemErrMsg
|
|
||||||
case 10004:
|
|
||||||
msg = SDKSecretErrMsg
|
|
||||||
case 10005:
|
|
||||||
msg = SDKFileIDErrMsg
|
|
||||||
case 10006:
|
|
||||||
msg = SDKDecryptErrMsg
|
|
||||||
case 10007:
|
|
||||||
msg = SDKSecretMissErrMsg
|
|
||||||
case 10008:
|
|
||||||
msg = SDKEncryptKeyErrMsg
|
|
||||||
case 10009:
|
|
||||||
msg = SDKIPNotWhiteListErrMsg
|
|
||||||
case 10010:
|
|
||||||
msg = SDKDataExpiredErrMsg
|
|
||||||
case 10011:
|
|
||||||
msg = SDKTokenExpiredErrMsg
|
|
||||||
}
|
|
||||||
return Error{ErrCode: code, ErrMsg: msg}
|
|
||||||
}
|
|
||||||
@@ -1,152 +0,0 @@
|
|||||||
// All Rights Reserved.
|
|
||||||
// *File : WeWorkFinanceSdk_C.h
|
|
||||||
// @Brief:拉取企业聊天记录与媒体消息sdk头文件
|
|
||||||
|
|
||||||
#pragma once
|
|
||||||
//返回码 错误说明
|
|
||||||
//10000 参数错误,请求参数错误
|
|
||||||
//10001 网络错误,网络请求错误
|
|
||||||
//10002 数据解析失败
|
|
||||||
//10003 系统失败
|
|
||||||
//10004 密钥错误导致加密失败
|
|
||||||
//10005 fileid错误
|
|
||||||
//10006 解密失败
|
|
||||||
//10007 找不到消息加密版本的私钥,需要重新传入私钥对
|
|
||||||
//10008 解析encrypt_key出错
|
|
||||||
//10009 ip非法
|
|
||||||
//10010 数据过期
|
|
||||||
//10011 证书错误
|
|
||||||
|
|
||||||
typedef struct WeWorkFinanceSdk_t WeWorkFinanceSdk_t;
|
|
||||||
|
|
||||||
// 数据
|
|
||||||
typedef struct Slice_t {
|
|
||||||
char* buf;
|
|
||||||
int len;
|
|
||||||
} Slice_t;
|
|
||||||
|
|
||||||
typedef struct MediaData {
|
|
||||||
char* outindexbuf;
|
|
||||||
int out_len;
|
|
||||||
char* data;
|
|
||||||
int data_len;
|
|
||||||
int is_finish;
|
|
||||||
} MediaData_t;
|
|
||||||
|
|
||||||
|
|
||||||
#ifdef __cplusplus
|
|
||||||
extern "C" {
|
|
||||||
#endif
|
|
||||||
|
|
||||||
WeWorkFinanceSdk_t* NewSdk();
|
|
||||||
|
|
||||||
|
|
||||||
/**
|
|
||||||
* 初始化函数
|
|
||||||
* Return值=0表示该API调用成功
|
|
||||||
*
|
|
||||||
* @param [in] sdk NewSdk返回的sdk指针
|
|
||||||
* @param [in] corpid 调用企业的企业id,例如:wwd08c8exxxx5ab44d,可以在企业微信管理端--我的企业--企业信息查看
|
|
||||||
* @param [in] secret 聊天内容存档的Secret,可以在企业微信管理端--管理工具--聊天内容存档查看
|
|
||||||
*
|
|
||||||
*
|
|
||||||
* @return 返回是否初始化成功
|
|
||||||
* 0 - 成功
|
|
||||||
* !=0 - 失败
|
|
||||||
*/
|
|
||||||
int Init(WeWorkFinanceSdk_t* sdk, const char* corpid, const char* secret);
|
|
||||||
|
|
||||||
/**
|
|
||||||
* 拉取聊天记录函数
|
|
||||||
* Return值=0表示该API调用成功
|
|
||||||
*
|
|
||||||
*
|
|
||||||
* @param [in] sdk NewSdk返回的sdk指针
|
|
||||||
* @param [in] seq 从指定的seq开始拉取消息,注意的是返回的消息从seq+1开始返回,seq为之前接口返回的最大seq值。首次使用请使用seq:0
|
|
||||||
* @param [in] limit 一次拉取的消息条数,最大值1000条,超过1000条会返回错误
|
|
||||||
* @param [in] proxy 使用代理的请求,需要传入代理的链接。如:socks5://10.0.0.1:8081 或者 http://10.0.0.1:8081
|
|
||||||
* @param [in] passwd 代理账号密码,需要传入代理的账号密码。如 user_name:passwd_123
|
|
||||||
* @param [in] timeout 超时时间,单位秒
|
|
||||||
* @param [out] chatDatas 返回本次拉取消息的数据,slice结构体.内容包括errcode/errmsg,以及每条消息内容。示例如下:
|
|
||||||
|
|
||||||
{"errcode":0,"errmsg":"ok","chatdata":[{"seq":196,"msgid":"CAQQ2fbb4QUY0On2rYSAgAMgip/yzgs=","publickey_ver":3,"encrypt_random_key":"ftJ+uz3n/z1DsxlkwxNgE+mL38H42/KCvN8T60gbbtPD+Rta1hKTuQPzUzO6Hzne97MgKs7FfdDxDck/v8cDT6gUVjA2tZ/M7euSD0L66opJ/IUeBtpAtvgVSD5qhlaQjvfKJc/zPMGNK2xCLFYqwmQBZXbNT7uA69Fflm512nZKW/piK2RKdYJhRyvQnA1ISxK097sp9WlEgDg250fM5tgwMjujdzr7ehK6gtVBUFldNSJS7ndtIf6aSBfaLktZgwHZ57ONewWq8GJe7WwQf1hwcDbCh7YMG8nsweEwhDfUz+u8rz9an+0lgrYMZFRHnmzjgmLwrR7B/32Qxqd79A==","encrypt_chat_msg":"898WSfGMnIeytTsea7Rc0WsOocs0bIAerF6de0v2cFwqo9uOxrW9wYe5rCjCHHH5bDrNvLxBE/xOoFfcwOTYX0HQxTJaH0ES9OHDZ61p8gcbfGdJKnq2UU4tAEgGb8H+Q9n8syRXIjaI3KuVCqGIi4QGHFmxWenPFfjF/vRuPd0EpzUNwmqfUxLBWLpGhv+dLnqiEOBW41Zdc0OO0St6E+JeIeHlRZAR+E13Isv9eS09xNbF0qQXWIyNUi+ucLr5VuZnPGXBrSfvwX8f0QebTwpy1tT2zvQiMM2MBugKH6NuMzzuvEsXeD+6+3VRqL"}]}
|
|
||||||
|
|
||||||
*
|
|
||||||
* @return 返回是否调用成功
|
|
||||||
* 0 - 成功
|
|
||||||
* !=0 - 失败
|
|
||||||
*/
|
|
||||||
int GetChatData(WeWorkFinanceSdk_t* sdk, unsigned long long seq, unsigned int limit, const char *proxy,const char* passwd, int timeout,Slice_t* chatDatas);
|
|
||||||
|
|
||||||
/**
|
|
||||||
* @brief 解析密文.企业微信自有解密内容
|
|
||||||
* @param [in] encrypt_key, getchatdata返回的encrypt_random_key,使用企业自持对应版本秘钥RSA解密后的内容
|
|
||||||
* @param [in] encrypt_msg, getchatdata返回的encrypt_chat_msg
|
|
||||||
* @param [out] msg, 解密的消息明文
|
|
||||||
* @return 返回是否调用成功
|
|
||||||
* 0 - 成功
|
|
||||||
* !=0 - 失败
|
|
||||||
*/
|
|
||||||
int DecryptData(const char* encrypt_key, const char* encrypt_msg, Slice_t* msg);
|
|
||||||
|
|
||||||
/**
|
|
||||||
* 拉取媒体消息函数
|
|
||||||
* Return值=0表示该API调用成功
|
|
||||||
*
|
|
||||||
*
|
|
||||||
* @param [in] sdk NewSdk返回的sdk指针
|
|
||||||
* @param [in] sdkFileid 从GetChatData返回的聊天消息中,媒体消息包括的sdkfileid
|
|
||||||
* @param [in] proxy 使用代理的请求,需要传入代理的链接。如:socks5://10.0.0.1:8081 或者 http://10.0.0.1:8081
|
|
||||||
* @param [in] passwd 代理账号密码,需要传入代理的账号密码。如 user_name:passwd_123
|
|
||||||
* @param [in] indexbuf 媒体消息分片拉取,需要填入每次拉取的索引信息。首次不需要填写,默认拉取512k,后续每次调用只需要将上次调用返回的outindexbuf填入即可。
|
|
||||||
* @param [in] timeout 超时时间,单位秒
|
|
||||||
* @param [out] media_data 返回本次拉取的媒体数据.MediaData结构体.内容包括data(数据内容)/outindexbuf(下次索引)/is_finish(拉取完成标记)
|
|
||||||
|
|
||||||
*
|
|
||||||
* @return 返回是否调用成功
|
|
||||||
* 0 - 成功
|
|
||||||
* !=0 - 失败
|
|
||||||
*/
|
|
||||||
int GetMediaData(WeWorkFinanceSdk_t* sdk, const char* indexbuf,
|
|
||||||
const char* sdkFileid,const char *proxy,const char* passwd, int timeout, MediaData_t* media_data);
|
|
||||||
|
|
||||||
/**
|
|
||||||
* @brief 释放sdk,和NewSdk成对使用
|
|
||||||
* @return
|
|
||||||
*/
|
|
||||||
void DestroySdk(WeWorkFinanceSdk_t* sdk);
|
|
||||||
|
|
||||||
|
|
||||||
//--------------下面接口为了其他语言例如python等调用c接口,酌情使用--------------
|
|
||||||
Slice_t* NewSlice();
|
|
||||||
|
|
||||||
/**
|
|
||||||
* @brief 释放slice,和NewSlice成对使用
|
|
||||||
* @return
|
|
||||||
*/
|
|
||||||
void FreeSlice(Slice_t* slice);
|
|
||||||
|
|
||||||
/**
|
|
||||||
* @brief 为其他语言提供读取接口
|
|
||||||
* @return 返回buf指针
|
|
||||||
* !=NULL - 成功
|
|
||||||
* NULL - 失败
|
|
||||||
*/
|
|
||||||
char* GetContentFromSlice(Slice_t* slice);
|
|
||||||
int GetSliceLen(Slice_t* slice);
|
|
||||||
|
|
||||||
// 媒体记录相关工具
|
|
||||||
|
|
||||||
MediaData_t* NewMediaData();
|
|
||||||
|
|
||||||
void FreeMediaData(MediaData_t* media_data);
|
|
||||||
|
|
||||||
char* GetOutIndexBuf(MediaData_t* media_data);
|
|
||||||
char* GetData(MediaData_t* media_data);
|
|
||||||
int GetIndexLen(MediaData_t* media_data);
|
|
||||||
int GetDataLen(MediaData_t* media_data);
|
|
||||||
int IsMediaDataFinish(MediaData_t* media_data);
|
|
||||||
|
|
||||||
#ifdef __cplusplus
|
|
||||||
}
|
|
||||||
#endif
|
|
||||||
Binary file not shown.
@@ -1 +0,0 @@
|
|||||||
781ec3cbad904b1527023cc9df0f279b
|
|
||||||
@@ -1,148 +0,0 @@
|
|||||||
#include "WeWorkFinanceSdk_C.h"
|
|
||||||
#include <dlfcn.h>
|
|
||||||
#include <stdint.h>
|
|
||||||
#include <stdio.h>
|
|
||||||
#include <stdlib.h>
|
|
||||||
#include <string>
|
|
||||||
using std::string;
|
|
||||||
|
|
||||||
typedef WeWorkFinanceSdk_t* newsdk_t();
|
|
||||||
typedef int Init_t(WeWorkFinanceSdk_t*, const char*, const char*);
|
|
||||||
typedef void DestroySdk_t(WeWorkFinanceSdk_t*);
|
|
||||||
|
|
||||||
typedef int GetChatData_t(WeWorkFinanceSdk_t*, unsigned long long, unsigned int, const char*, const char*, int, Slice_t*);
|
|
||||||
typedef Slice_t* NewSlice_t();
|
|
||||||
typedef void FreeSlice_t(Slice_t*);
|
|
||||||
|
|
||||||
typedef int GetMediaData_t(WeWorkFinanceSdk_t*, const char*, const char*, const char*, const char*, int, MediaData_t*);
|
|
||||||
typedef int DecryptData_t(const char*, const char*, Slice_t*);
|
|
||||||
typedef MediaData_t* NewMediaData_t();
|
|
||||||
typedef void FreeMediaData_t(MediaData_t*);
|
|
||||||
|
|
||||||
int main(int argc, char* argv[])
|
|
||||||
{
|
|
||||||
int ret = 0;
|
|
||||||
//seq 表示该企业存档消息序号,该序号单调递增,拉取序号建议设置为上次拉取返回结果中最大序号。首次拉取时seq传0,sdk会返回有效期内最早的消息。
|
|
||||||
//limit 表示本次拉取的最大消息条数,取值范围为1~1000
|
|
||||||
//proxy与passwd为代理参数,如果运行sdk的环境不能直接访问外网,需要配置代理参数。sdk访问的域名是"https://qyapi.weixin.qq.com"。
|
|
||||||
//建议先通过curl访问"https://qyapi.weixin.qq.com",验证代理配置正确后,再传入sdk。
|
|
||||||
//timeout 为拉取会话存档的超时时间,单位为秒,建议超时时间设置为5s。
|
|
||||||
//sdkfileid 媒体文件id,从解密后的会话存档中得到
|
|
||||||
//savefile 媒体文件保存路径
|
|
||||||
//encrypt_key 拉取会话存档返回的encrypt_random_key,使用配置在企业微信管理台的rsa公钥对应的私钥解密后得到encrypt_key。
|
|
||||||
//encrypt_chat_msg 拉取会话存档返回的encrypt_chat_msg
|
|
||||||
if (argc < 2) {
|
|
||||||
printf("./sdktools 1(chatmsg) 2(mediadata) 3(decryptdata)\n");
|
|
||||||
printf("./sdktools 1 seq limit proxy passwd timeout\n");
|
|
||||||
printf("./sdktools 2 fileid proxy passwd timeout savefile\n");
|
|
||||||
printf("./sdktools 3 encrypt_key encrypt_chat_msg\n");
|
|
||||||
return -1;
|
|
||||||
}
|
|
||||||
|
|
||||||
void* so_handle = dlopen("./libWeWorkFinanceSdk_C.so", RTLD_LAZY);
|
|
||||||
if (!so_handle) {
|
|
||||||
printf("load sdk so fail:%s\n", dlerror());
|
|
||||||
return -1;
|
|
||||||
}
|
|
||||||
newsdk_t* newsdk_fn = (newsdk_t*)dlsym(so_handle, "NewSdk");
|
|
||||||
WeWorkFinanceSdk_t* sdk = newsdk_fn();
|
|
||||||
|
|
||||||
//使用sdk前需要初始化,初始化成功后的sdk可以一直使用。
|
|
||||||
//如需并发调用sdk,建议每个线程持有一个sdk实例。
|
|
||||||
//初始化时请填入自己企业的corpid与secrectkey。
|
|
||||||
Init_t* init_fn = (Init_t*)dlsym(so_handle, "Init");
|
|
||||||
DestroySdk_t* destroysdk_fn = (DestroySdk_t*)dlsym(so_handle, "DestroySdk");
|
|
||||||
ret = init_fn(sdk, "wwd08c8e7c775ab44d", "zJ6k0naVVQ--gt9PUSSEvs03zW_nlDVmjLCTOTAfrew");
|
|
||||||
if (ret != 0) {
|
|
||||||
//sdk需要主动释放
|
|
||||||
destroysdk_fn(sdk);
|
|
||||||
printf("init sdk err ret:%d\n", ret);
|
|
||||||
return -1;
|
|
||||||
}
|
|
||||||
|
|
||||||
int type = strtoul(argv[1], NULL, 10);
|
|
||||||
if (type == 1) {
|
|
||||||
//拉取会话存档
|
|
||||||
uint64_t iSeq = strtoul(argv[2], NULL, 10);
|
|
||||||
uint64_t iLimit = strtoul(argv[3], NULL, 10);
|
|
||||||
uint64_t timeout = strtoul(argv[6], NULL, 10);
|
|
||||||
|
|
||||||
NewSlice_t* newslice_fn = (NewSlice_t*)dlsym(so_handle, "NewSlice");
|
|
||||||
FreeSlice_t* freeslice_fn = (FreeSlice_t*)dlsym(so_handle, "FreeSlice");
|
|
||||||
|
|
||||||
//每次使用GetChatData拉取存档前需要调用NewSlice获取一个chatDatas,在使用完chatDatas中数据后,还需要调用FreeSlice释放。
|
|
||||||
Slice_t* chatDatas = newslice_fn();
|
|
||||||
GetChatData_t* getchatdata_fn = (GetChatData_t*)dlsym(so_handle, "GetChatData");
|
|
||||||
ret = getchatdata_fn(sdk, iSeq, iLimit, argv[4], argv[5], timeout, chatDatas);
|
|
||||||
if (ret != 0) {
|
|
||||||
freeslice_fn(chatDatas);
|
|
||||||
printf("GetChatData err ret:%d\n", ret);
|
|
||||||
return -1;
|
|
||||||
}
|
|
||||||
printf("GetChatData len:%d data:%s\n", chatDatas->len, chatDatas->buf);
|
|
||||||
freeslice_fn(chatDatas);
|
|
||||||
}
|
|
||||||
else if (type == 2) {
|
|
||||||
//拉取媒体文件
|
|
||||||
std::string index;
|
|
||||||
uint64_t timeout = strtoul(argv[5], NULL, 10);
|
|
||||||
int isfinish = 0;
|
|
||||||
|
|
||||||
GetMediaData_t* getmediadata_fn = (GetMediaData_t*)dlsym(so_handle, "GetMediaData");
|
|
||||||
NewMediaData_t* newmediadata_fn = (NewMediaData_t*)dlsym(so_handle, "NewMediaData");
|
|
||||||
FreeMediaData_t* freemediadata_fn = (FreeMediaData_t*)dlsym(so_handle, "FreeMediaData");
|
|
||||||
|
|
||||||
//媒体文件每次拉取的最大size为512k,因此超过512k的文件需要分片拉取。若该文件未拉取完整,mediaData中的is_finish会返回0,同时mediaData中的outindexbuf会返回下次拉取需要传入GetMediaData的indexbuf。
|
|
||||||
//indexbuf一般格式如右侧所示,”Range:bytes=524288-1048575“,表示这次拉取的是从524288到1048575的分片。单个文件首次拉取填写的indexbuf为空字符串,拉取后续分片时直接填入上次返回的indexbuf即可。
|
|
||||||
while (isfinish == 0) {
|
|
||||||
//每次使用GetMediaData拉取存档前需要调用NewMediaData获取一个mediaData,在使用完mediaData中数据后,还需要调用FreeMediaData释放。
|
|
||||||
printf("index:%s\n", index.c_str());
|
|
||||||
MediaData_t* mediaData = newmediadata_fn();
|
|
||||||
ret = getmediadata_fn(sdk, index.c_str(), argv[2], argv[3], argv[4], timeout, mediaData);
|
|
||||||
if (ret != 0) {
|
|
||||||
//单个分片拉取失败建议重试拉取该分片,避免从头开始拉取。
|
|
||||||
freemediadata_fn(mediaData);
|
|
||||||
printf("GetMediaData err ret:%d\n", ret);
|
|
||||||
return -1;
|
|
||||||
}
|
|
||||||
printf("content size:%d isfin:%d outindex:%s\n", mediaData->data_len, mediaData->is_finish, mediaData->outindexbuf);
|
|
||||||
|
|
||||||
//大于512k的文件会分片拉取,此处需要使用追加写,避免后面的分片覆盖之前的数据。
|
|
||||||
char file[200];
|
|
||||||
snprintf(file, sizeof(file), "%s", argv[6]);
|
|
||||||
FILE* fp = fopen(file, "ab+");
|
|
||||||
printf("filename:%s \n", file);
|
|
||||||
if (NULL == fp) {
|
|
||||||
freemediadata_fn(mediaData);
|
|
||||||
printf("open file err\n");
|
|
||||||
return -1;
|
|
||||||
}
|
|
||||||
|
|
||||||
fwrite(mediaData->data, mediaData->data_len, 1, fp);
|
|
||||||
fclose(fp);
|
|
||||||
|
|
||||||
//获取下次拉取需要使用的indexbuf
|
|
||||||
index.assign(string(mediaData->outindexbuf));
|
|
||||||
isfinish = mediaData->is_finish;
|
|
||||||
freemediadata_fn(mediaData);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
else if (type == 3) {
|
|
||||||
//解密会话存档内容
|
|
||||||
//sdk不会要求用户传入rsa私钥,保证用户会话存档数据只有自己能够解密。
|
|
||||||
//此处需要用户先用rsa私钥解密encrypt_random_key后,作为encrypt_key参数传入sdk来解密encrypt_chat_msg获取会话存档明文。
|
|
||||||
//每次使用DecryptData解密会话存档前需要调用NewSlice获取一个Msgs,在使用完Msgs中数据后,还需要调用FreeSlice释放。
|
|
||||||
NewSlice_t* newslice_fn = (NewSlice_t*)dlsym(so_handle, "NewSlice");
|
|
||||||
FreeSlice_t* freeslice_fn = (FreeSlice_t*)dlsym(so_handle, "FreeSlice");
|
|
||||||
|
|
||||||
Slice_t* Msgs = newslice_fn();
|
|
||||||
// decryptdata api
|
|
||||||
DecryptData_t* decryptdata_fn = (DecryptData_t*)dlsym(so_handle, "DecryptData");
|
|
||||||
ret = decryptdata_fn(argv[2], argv[3], Msgs);
|
|
||||||
printf("chatdata :%s ret :%d\n", Msgs->buf, ret);
|
|
||||||
|
|
||||||
freeslice_fn(Msgs);
|
|
||||||
}
|
|
||||||
|
|
||||||
return ret;
|
|
||||||
}
|
|
||||||
@@ -1 +0,0 @@
|
|||||||
200215
|
|
||||||
@@ -1,8 +0,0 @@
|
|||||||
package msgaudit
|
|
||||||
|
|
||||||
// MediaData 媒体文件数据
|
|
||||||
type MediaData struct {
|
|
||||||
OutIndexBuf string `json:"outindexbuf,omitempty"`
|
|
||||||
IsFinish bool `json:"is_finish,omitempty"`
|
|
||||||
Data []byte `json:"data,omitempty"`
|
|
||||||
}
|
|
||||||
@@ -1,352 +0,0 @@
|
|||||||
package msgaudit
|
|
||||||
|
|
||||||
// BaseMessage 基础消息
|
|
||||||
type BaseMessage struct {
|
|
||||||
MsgID string `json:"msgid,omitempty"` // 消息id,消息的唯一标识,企业可以使用此字段进行消息去重。
|
|
||||||
Action string `json:"action,omitempty"` // 消息动作,目前有send(发送消息)/recall(撤回消息)/switch(切换企业日志)三种类型。
|
|
||||||
From string `json:"from,omitempty"` // 消息发送方id。同一企业内容为userid,非相同企业为external_userid。消息如果是机器人发出,也为external_userid。
|
|
||||||
ToList []string `json:"tolist,omitempty"` // 消息接收方列表,可能是多个,同一个企业内容为userid,非相同企业为external_userid。
|
|
||||||
RoomID string `json:"roomid,omitempty"` // 群聊消息的群id。如果是单聊则为空。
|
|
||||||
MsgTime int64 `json:"msgtime,omitempty"` // 消息发送时间戳,utc时间,ms单位。
|
|
||||||
MsgType string `json:"msgtype,omitempty"` // 文本消息为:text。
|
|
||||||
}
|
|
||||||
|
|
||||||
// TextMessage 文本消息
|
|
||||||
type TextMessage struct {
|
|
||||||
BaseMessage
|
|
||||||
Text struct {
|
|
||||||
Content string `json:"content,omitempty"` // 消息内容。
|
|
||||||
} `json:"text,omitempty"`
|
|
||||||
}
|
|
||||||
|
|
||||||
// ImageMessage 图片消息
|
|
||||||
type ImageMessage struct {
|
|
||||||
BaseMessage
|
|
||||||
Image struct {
|
|
||||||
SdkFileID string `json:"sdkfileid,omitempty"` // 媒体资源的id信息。
|
|
||||||
Md5Sum string `json:"md5sum,omitempty"` // 图片资源的md5值,供进行校验。
|
|
||||||
FileSize uint32 `json:"filesize,omitempty"` // 图片资源的文件大小。
|
|
||||||
} `json:"image,omitempty"`
|
|
||||||
}
|
|
||||||
|
|
||||||
// RevokeMessage 撤回消息
|
|
||||||
type RevokeMessage struct {
|
|
||||||
BaseMessage
|
|
||||||
Revoke struct {
|
|
||||||
PreMsgID string `json:"pre_msgid,omitempty"` // 标识撤回的原消息的msgid
|
|
||||||
} `json:"revoke,omitempty"`
|
|
||||||
}
|
|
||||||
|
|
||||||
// AgreeMessage 同意会话聊天内容
|
|
||||||
type AgreeMessage struct {
|
|
||||||
BaseMessage
|
|
||||||
Agree struct {
|
|
||||||
UserID string `json:"userid,omitempty"` // 同意/不同意协议者的userid,外部企业默认为external_userid。
|
|
||||||
AgreeTime int64 `json:"agree_time,omitempty"` // 同意/不同意协议的时间,utc时间,ms单位。
|
|
||||||
} `json:"agree,omitempty"`
|
|
||||||
}
|
|
||||||
|
|
||||||
// VoiceMessage 语音消息
|
|
||||||
type VoiceMessage struct {
|
|
||||||
BaseMessage
|
|
||||||
Voice struct {
|
|
||||||
SdkFileID string `json:"sdkfileid,omitempty"` // 媒体资源的id信息。
|
|
||||||
VoiceSize uint32 `json:"voice_size,omitempty"` // 语音消息大小。
|
|
||||||
PlayLength uint32 `json:"play_length,omitempty"` // 播放长度。
|
|
||||||
Md5Sum string `json:"md5sum,omitempty"` // 图片资源的md5值,供进行校验。
|
|
||||||
} `json:"voice,omitempty"`
|
|
||||||
}
|
|
||||||
|
|
||||||
// VideoMessage 视频消息
|
|
||||||
type VideoMessage struct {
|
|
||||||
BaseMessage
|
|
||||||
Video struct {
|
|
||||||
SdkFileID string `json:"sdkfileid,omitempty"` // 媒体资源的id信息。
|
|
||||||
FileSize uint32 `json:"filesize,omitempty"` // 图片资源的文件大小。
|
|
||||||
PlayLength uint32 `json:"play_length,omitempty"` // 播放长度。
|
|
||||||
Md5Sum string `json:"md5sum,omitempty"` // 图片资源的md5值,供进行校验。
|
|
||||||
} `json:"video,omitempty"`
|
|
||||||
}
|
|
||||||
|
|
||||||
// CardMessage 名片消息
|
|
||||||
type CardMessage struct {
|
|
||||||
BaseMessage
|
|
||||||
Card struct {
|
|
||||||
CorpName string `json:"corpname,omitempty"` // 名片所有者所在的公司名称。
|
|
||||||
UserID string `json:"userid,omitempty"` // 名片所有者的id,同一公司是userid,不同公司是external_userid
|
|
||||||
} `json:"card,omitempty"`
|
|
||||||
}
|
|
||||||
|
|
||||||
// LocationMessage 位置消息
|
|
||||||
type LocationMessage struct {
|
|
||||||
BaseMessage
|
|
||||||
Location struct {
|
|
||||||
Lng float64 `json:"longitude,omitempty"` // 经度,单位double
|
|
||||||
Lat float64 `json:"latitude,omitempty"` // 纬度,单位double
|
|
||||||
Address string `json:"address,omitempty"` // 地址信息
|
|
||||||
Title string `json:"title,omitempty"` // 位置信息的title。
|
|
||||||
Zoom uint32 `json:"zoom,omitempty"` // 缩放比例。
|
|
||||||
} `json:"location,omitempty"`
|
|
||||||
}
|
|
||||||
|
|
||||||
// EmotionMessage 表情消息
|
|
||||||
type EmotionMessage struct {
|
|
||||||
BaseMessage
|
|
||||||
Emotion struct {
|
|
||||||
Type uint32 `json:"type,omitempty"` // 表情类型,png或者gif.1表示gif 2表示png。
|
|
||||||
Width uint32 `json:"width,omitempty"` // 表情图片宽度。
|
|
||||||
Height uint32 `json:"height,omitempty"` // 表情图片高度。
|
|
||||||
ImageSize uint32 `json:"imagesize,omitempty"` // 资源的文件大小。
|
|
||||||
SdkFileID string `json:"sdkfileid,omitempty"` // 媒体资源的id信息。
|
|
||||||
Md5Sum string `json:"md5sum,omitempty"` // 图片资源的md5值,供进行校验。
|
|
||||||
} `json:"emotion,omitempty"`
|
|
||||||
}
|
|
||||||
|
|
||||||
// FileMessage 文件消息
|
|
||||||
type FileMessage struct {
|
|
||||||
BaseMessage
|
|
||||||
File struct {
|
|
||||||
FileName string `json:"filename,omitempty"` // 文件名称。
|
|
||||||
FileExt string `json:"fileext,omitempty"` // 文件类型后缀。
|
|
||||||
SdkFileID string `json:"sdkfileid,omitempty"` // 媒体资源的id信息。
|
|
||||||
FileSize uint32 `json:"filesize,omitempty"` // 文件大小。
|
|
||||||
Md5Sum string `json:"md5sum,omitempty"` // 资源的md5值,供进行校验。
|
|
||||||
} `json:"file,omitempty"`
|
|
||||||
}
|
|
||||||
|
|
||||||
// LinkMessage 链接消息
|
|
||||||
type LinkMessage struct {
|
|
||||||
BaseMessage
|
|
||||||
Link struct {
|
|
||||||
Title string `json:"title,omitempty"` // 消息标题。
|
|
||||||
Desc string `json:"description,omitempty"` // 消息描述。
|
|
||||||
LinkURL string `json:"link_url,omitempty"` // 链接url地址
|
|
||||||
ImageURL string `json:"image_url,omitempty"` // 链接图片url。
|
|
||||||
} `json:"link,omitempty"`
|
|
||||||
}
|
|
||||||
|
|
||||||
// WeappMessage 小程序消息
|
|
||||||
type WeappMessage struct {
|
|
||||||
BaseMessage
|
|
||||||
WeApp struct {
|
|
||||||
Title string `json:"title,omitempty"` // 消息标题。
|
|
||||||
Desc string `json:"description,omitempty"` // 消息描述。
|
|
||||||
Username string `json:"username,omitempty"` // 用户名称。
|
|
||||||
DisplayName string `json:"displayname,omitempty"` // 小程序名称
|
|
||||||
} `json:"weapp,omitempty"`
|
|
||||||
}
|
|
||||||
|
|
||||||
// ChatRecordMessage 会话记录消息
|
|
||||||
type ChatRecordMessage struct {
|
|
||||||
BaseMessage
|
|
||||||
ChatRecord struct {
|
|
||||||
Title string `json:"title,omitempty"` // 聊天记录标题
|
|
||||||
Item []ChatRecord `json:"item,omitempty"` // 消息记录内的消息内容,批量数据
|
|
||||||
} `json:"chatrecord,omitempty"`
|
|
||||||
}
|
|
||||||
|
|
||||||
// TodoMessage 待办消息
|
|
||||||
type TodoMessage struct {
|
|
||||||
BaseMessage
|
|
||||||
Todo struct {
|
|
||||||
Title string `json:"title,omitempty"` // 代办的来源文本
|
|
||||||
Content string `json:"content,omitempty"` // 代办的具体内容
|
|
||||||
} `json:"todo,omitempty"`
|
|
||||||
}
|
|
||||||
|
|
||||||
// VoteMessage 投票消息
|
|
||||||
type VoteMessage struct {
|
|
||||||
BaseMessage
|
|
||||||
VoteTitle string `json:"votetitle,omitempty"` // 投票主题。
|
|
||||||
VoteItem []string `json:"voteitem,omitempty"` // 投票选项,可能多个内容。
|
|
||||||
VoteType uint32 `json:"votetype,omitempty"` // 投票类型.101发起投票、102参与投票。
|
|
||||||
VoteID string `json:"voteid,omitempty"` // 投票id,方便将参与投票消息与发起投票消息进行前后对照。
|
|
||||||
}
|
|
||||||
|
|
||||||
// CollectMessage 填表消息
|
|
||||||
type CollectMessage struct {
|
|
||||||
BaseMessage
|
|
||||||
Collect struct {
|
|
||||||
RoomName string `json:"room_name,omitempty"` // 填表消息所在的群名称。
|
|
||||||
Creator string `json:"creator,omitempty"` // 创建者在群中的名字
|
|
||||||
CreateTime string `json:"create_time,omitempty"` // 创建的时间
|
|
||||||
Details []CollectDetails `json:"details,omitempty"` // 表内容
|
|
||||||
} `json:"collect,omitempty"`
|
|
||||||
}
|
|
||||||
|
|
||||||
// RedpacketMessage 红包消息
|
|
||||||
type RedpacketMessage struct {
|
|
||||||
BaseMessage
|
|
||||||
RedPacket struct {
|
|
||||||
Type uint32 `json:"type,omitempty"` // 红包消息类型。1 普通红包、2 拼手气群红包、3 激励群红包。
|
|
||||||
Wish string `json:"wish,omitempty"` // 红包祝福语
|
|
||||||
TotalCnt uint32 `json:"totalcnt,omitempty"` // 红包总个数
|
|
||||||
TotalAmount uint32 `json:"totalamount,omitempty"` // 红包总金额。单位为分。
|
|
||||||
} `json:"redpacket,omitempty"`
|
|
||||||
}
|
|
||||||
|
|
||||||
// MeetingMessage 会议邀请消息
|
|
||||||
type MeetingMessage struct {
|
|
||||||
BaseMessage
|
|
||||||
Meeting struct {
|
|
||||||
Topic string `json:"topic,omitempty"` // 会议主题
|
|
||||||
StartTime int64 `json:"starttime,omitempty"` // 会议开始时间。Utc时间
|
|
||||||
EndTime int64 `json:"endtime,omitempty"` // 会议结束时间。Utc时间
|
|
||||||
Address string `json:"address,omitempty"` // 会议地址
|
|
||||||
Remarks string `json:"remarks,omitempty"` // 会议备注
|
|
||||||
MeetingType uint32 `json:"meetingtype,omitempty"` // 会议消息类型。101发起会议邀请消息、102处理会议邀请消息
|
|
||||||
MeetingID uint64 `json:"meetingid,omitempty"` // 会议id。方便将发起、处理消息进行对照
|
|
||||||
Status uint32 `json:"status,omitempty"` // 会议邀请处理状态。1 参加会议、2 拒绝会议、3 待定、4 未被邀请、5 会议已取消、6 会议已过期、7 不在房间内。
|
|
||||||
} `json:"meeting,omitempty"`
|
|
||||||
}
|
|
||||||
|
|
||||||
// DocMessage 在线文档消息
|
|
||||||
type DocMessage struct {
|
|
||||||
BaseMessage
|
|
||||||
Doc struct {
|
|
||||||
Title string `json:"title,omitempty"` // 在线文档名称
|
|
||||||
LinkURL string `json:"link_url,omitempty"` // 在线文档链接
|
|
||||||
DocCreator string `json:"doc_creator,omitempty"` // 在线文档创建者。本企业成员创建为userid;外部企业成员创建为external_userid
|
|
||||||
} `json:"doc,omitempty"`
|
|
||||||
}
|
|
||||||
|
|
||||||
// MarkdownMessage MarkDown消息
|
|
||||||
type MarkdownMessage struct {
|
|
||||||
BaseMessage
|
|
||||||
Info struct {
|
|
||||||
Content string `json:"content,omitempty"` // markdown消息内容,目前为机器人发出的消息
|
|
||||||
} `json:"info,omitempty"`
|
|
||||||
}
|
|
||||||
|
|
||||||
// NewsMessage 图文消息
|
|
||||||
type NewsMessage struct {
|
|
||||||
BaseMessage
|
|
||||||
Info struct {
|
|
||||||
Item []News `json:"item,omitempty"` // 图文消息数组
|
|
||||||
} `json:"info,omitempty"` // 图文消息的内容
|
|
||||||
}
|
|
||||||
|
|
||||||
// CalendarMessage 日程消息
|
|
||||||
type CalendarMessage struct {
|
|
||||||
BaseMessage
|
|
||||||
Calendar struct {
|
|
||||||
Title string `json:"title,omitempty"` // 日程主题
|
|
||||||
CreatorName string `json:"creatorname,omitempty"` // 日程组织者
|
|
||||||
AttendeeName []string `json:"attendeename,omitempty"` // 日程参与人。数组,内容为String类型
|
|
||||||
StartTime int64 `json:"starttime,omitempty"` // 日程开始时间。Utc时间,单位秒
|
|
||||||
EndTime int64 `json:"endtime,omitempty"` // 日程结束时间。Utc时间,单位秒
|
|
||||||
Place string `json:"place,omitempty"` // 日程地点
|
|
||||||
Remarks string `json:"remarks,omitempty"` // 日程备注
|
|
||||||
} `json:"calendar,omitempty"`
|
|
||||||
}
|
|
||||||
|
|
||||||
// MixedMessage 混合消息
|
|
||||||
type MixedMessage struct {
|
|
||||||
BaseMessage
|
|
||||||
Mixed struct {
|
|
||||||
Item []MixedMsg `json:"item,omitempty"`
|
|
||||||
} `json:"mixed,omitempty"` // 消息内容。可包含图片、文字、表情等多种消息。Object类型
|
|
||||||
}
|
|
||||||
|
|
||||||
// MeetingVoiceCallMessage 音频存档消息
|
|
||||||
type MeetingVoiceCallMessage struct {
|
|
||||||
BaseMessage
|
|
||||||
VoiceID string `json:"voiceid,omitempty"` // 音频id
|
|
||||||
MeetingVoiceCall *MeetingVoiceCall `json:"meeting_voice_call,omitempty"` // 音频消息内容。包括结束时间、fileid,可能包括多个demofiledata、sharescreendata消息,demofiledata表示文档共享信息,sharescreendata表示屏幕共享信息。Object类型
|
|
||||||
}
|
|
||||||
|
|
||||||
// VoipDocShareMessage 音频共享消息
|
|
||||||
type VoipDocShareMessage struct {
|
|
||||||
BaseMessage
|
|
||||||
VoipID string `json:"voipid,omitempty"` // 音频id
|
|
||||||
VoipDocShare *VoipDocShare `json:"voip_doc_share,omitempty"` // 共享文档消息内容。包括filename、md5sum、filesize、sdkfileid字段。Object类型
|
|
||||||
}
|
|
||||||
|
|
||||||
// ExternalRedPacketMessage 互通小红包消息
|
|
||||||
type ExternalRedPacketMessage struct {
|
|
||||||
BaseMessage
|
|
||||||
RedPacket struct {
|
|
||||||
Type int32 `json:"type,omitempty"` // 红包消息类型。1 普通红包、2 拼手气群红包。Uint32类型
|
|
||||||
Wish int32 `json:"wish,omitempty"` // 红包祝福语。String类型
|
|
||||||
TotalCnt int32 `json:"totalcnt,omitempty"` // 红包总个数。Uint32类型
|
|
||||||
TotalAmount int32 `json:"totalamount,omitempty"` // 红包消息类型。1 普通红包、2 拼手气群红包。Uint32类型
|
|
||||||
} `json:"redpacket,omitempty"`
|
|
||||||
}
|
|
||||||
|
|
||||||
// SphFeedMessage 视频号消息
|
|
||||||
type SphFeedMessage struct {
|
|
||||||
BaseMessage
|
|
||||||
SphFeed struct {
|
|
||||||
FeedType string `json:"feed_type,omitempty"` // 视频号消息类型
|
|
||||||
SphName string `json:"sph_name,omitempty"` // 视频号账号名称
|
|
||||||
FeedDesc uint64 `json:"feed_desc,omitempty"` // 视频号账号名称
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// SwitchMessage 企业切换日志
|
|
||||||
type SwitchMessage struct {
|
|
||||||
MsgID string `json:"msgid,omitempty"` // 消息id,消息的唯一标识,企业可以使用此字段进行消息去重
|
|
||||||
Action string `json:"action,omitempty"` // 消息动作,切换企业为switch
|
|
||||||
Time int64 `json:"time,omitempty"` // 消息发送时间戳,utc时间,ms单位。
|
|
||||||
User string `json:"user,omitempty"` // 具体为切换企业的成员的userid。
|
|
||||||
}
|
|
||||||
|
|
||||||
// ChatRecord 会话记录消息
|
|
||||||
type ChatRecord struct {
|
|
||||||
Type string `json:"type,omitempty"` // 每条聊天记录的具体消息类型:ChatRecordText/ ChatRecordFile/ ChatRecordImage/ ChatRecordVideo/ ChatRecordLink/ ChatRecordLocation/ ChatRecordMixed ….
|
|
||||||
Content string `json:"content,omitempty"` // 消息内容。Json串,内容为对应类型的json
|
|
||||||
MsgTime int64 `json:"msgtime,omitempty"` // 消息时间,utc时间,ms单位。
|
|
||||||
FromChatroom bool `json:"from_chatroom,omitempty"` // 是否来自群会话。
|
|
||||||
}
|
|
||||||
|
|
||||||
// CollectDetails 填表消息
|
|
||||||
type CollectDetails struct {
|
|
||||||
ID uint64 `json:"id,omitempty"` // 表项id
|
|
||||||
Ques string `json:"ques,omitempty"` // 表项名称
|
|
||||||
Type string `json:"type,omitempty"` // 表项类型,有Text(文本),Number(数字),Date(日期),Time(时间)
|
|
||||||
}
|
|
||||||
|
|
||||||
// News 图文消息
|
|
||||||
type News struct {
|
|
||||||
Title string `json:"title,omitempty"` // 图文消息标题
|
|
||||||
Desc string `json:"description,omitempty"` // 图文消息描述
|
|
||||||
URL string `json:"url,omitempty"` // 图文消息点击跳转地址
|
|
||||||
PicURL string `json:"picurl,omitempty"` // 图文消息配图的url
|
|
||||||
}
|
|
||||||
|
|
||||||
// MixedMsg 混合消息
|
|
||||||
type MixedMsg struct {
|
|
||||||
Type string `json:"type,omitempty"`
|
|
||||||
Content string `json:"content,omitempty"`
|
|
||||||
}
|
|
||||||
|
|
||||||
// MeetingVoiceCall 音频存档消息
|
|
||||||
type MeetingVoiceCall struct {
|
|
||||||
EndTime int64 `json:"endtime,omitempty"` // 音频结束时间
|
|
||||||
SdkFileID string `json:"sdkfileid,omitempty"` // 音频媒体下载的id
|
|
||||||
DemoFileData []DemoFileData `json:"demofiledata,omitempty"` // 文档分享对象,Object类型
|
|
||||||
ShareScreenData []ShareScreenData `json:"sharescreendata,omitempty"` // 屏幕共享对象,Object类型
|
|
||||||
}
|
|
||||||
|
|
||||||
// DemoFileData 文档共享消息
|
|
||||||
type DemoFileData struct {
|
|
||||||
FileName string `json:"filename,omitempty"` // 文档共享名称
|
|
||||||
DemoOperator string `json:"demooperator,omitempty"` // 文档共享操作用户的id
|
|
||||||
StartTime int64 `json:"starttime,omitempty"` // 文档共享开始时间
|
|
||||||
EndTime int64 `json:"endtime,omitempty"` // 文档共享结束时间
|
|
||||||
}
|
|
||||||
|
|
||||||
// ShareScreenData 屏幕共享信息
|
|
||||||
type ShareScreenData struct {
|
|
||||||
Share string `json:"share,omitempty"` // 屏幕共享用户的id
|
|
||||||
StartTime int64 `json:"starttime,omitempty"` // 屏幕共享开始时间
|
|
||||||
EndTime int64 `json:"endtime,omitempty"` // 屏幕共享结束时间
|
|
||||||
}
|
|
||||||
|
|
||||||
// VoipDocShare 音频共享文档消息
|
|
||||||
type VoipDocShare struct {
|
|
||||||
FileName string `json:"filename,omitempty"` // 文档共享文件名称
|
|
||||||
Md5Sum string `json:"md5sum,omitempty"` // 共享文件的md5值
|
|
||||||
FileSize uint64 `json:"filesize,omitempty"` // 共享文件的大小
|
|
||||||
SdkFileID string `json:"sdkfileid,omitempty"` // 共享文件的sdkfile,通过此字段进行媒体数据下载
|
|
||||||
}
|
|
||||||
@@ -1,87 +0,0 @@
|
|||||||
package oauth
|
|
||||||
|
|
||||||
import (
|
|
||||||
"encoding/json"
|
|
||||||
"fmt"
|
|
||||||
"net/url"
|
|
||||||
|
|
||||||
"github.com/silenceper/wechat/v2/util"
|
|
||||||
"github.com/silenceper/wechat/v2/work/context"
|
|
||||||
)
|
|
||||||
|
|
||||||
//Oauth auth
|
|
||||||
type Oauth struct {
|
|
||||||
*context.Context
|
|
||||||
}
|
|
||||||
|
|
||||||
var (
|
|
||||||
//oauthTargetURL 企业微信内跳转地址
|
|
||||||
oauthTargetURL = "https://open.weixin.qq.com/connect/oauth2/authorize?appid=%s&redirect_uri=%s&response_type=code&scope=snsapi_base&state=STATE#wechat_redirect"
|
|
||||||
//oauthUserInfoURL 获取用户信息地址
|
|
||||||
oauthUserInfoURL = "https://qyapi.weixin.qq.com/cgi-bin/user/getuserinfo?access_token=%s&code=%s"
|
|
||||||
//oauthQrContentTargetURL 构造独立窗口登录二维码
|
|
||||||
oauthQrContentTargetURL = "https://open.work.weixin.qq.com/wwopen/sso/qrConnect?appid=%s&agentid=%s&redirect_uri=%s&state=%s"
|
|
||||||
)
|
|
||||||
|
|
||||||
//NewOauth new init oauth
|
|
||||||
func NewOauth(ctx *context.Context) *Oauth {
|
|
||||||
return &Oauth{
|
|
||||||
ctx,
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
//GetTargetURL 获取授权地址
|
|
||||||
func (ctr *Oauth) GetTargetURL(callbackURL string) string {
|
|
||||||
//url encode
|
|
||||||
urlStr := url.QueryEscape(callbackURL)
|
|
||||||
return fmt.Sprintf(
|
|
||||||
oauthTargetURL,
|
|
||||||
ctr.CorpID,
|
|
||||||
urlStr,
|
|
||||||
)
|
|
||||||
}
|
|
||||||
|
|
||||||
//GetQrContentTargetURL 构造独立窗口登录二维码
|
|
||||||
func (ctr *Oauth) GetQrContentTargetURL(callbackURL string) string {
|
|
||||||
//url encode
|
|
||||||
urlStr := url.QueryEscape(callbackURL)
|
|
||||||
return fmt.Sprintf(
|
|
||||||
oauthQrContentTargetURL,
|
|
||||||
ctr.CorpID,
|
|
||||||
ctr.AgentID,
|
|
||||||
urlStr,
|
|
||||||
util.RandomStr(16),
|
|
||||||
)
|
|
||||||
}
|
|
||||||
|
|
||||||
//ResUserInfo 返回得用户信息
|
|
||||||
type ResUserInfo struct {
|
|
||||||
util.CommonError
|
|
||||||
//当用户为企业成员时返回
|
|
||||||
UserID string `json:"UserId"`
|
|
||||||
DeviceID string `json:"DeviceId"`
|
|
||||||
//非企业成员授权时返回
|
|
||||||
OpenID string `json:"OpenId"`
|
|
||||||
}
|
|
||||||
|
|
||||||
//UserFromCode 根据code获取用户信息
|
|
||||||
func (ctr *Oauth) UserFromCode(code string) (result ResUserInfo, err error) {
|
|
||||||
var accessToken string
|
|
||||||
accessToken, err = ctr.GetAccessToken()
|
|
||||||
if err != nil {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
var response []byte
|
|
||||||
response, err = util.HTTPGet(
|
|
||||||
fmt.Sprintf(oauthUserInfoURL, accessToken, code),
|
|
||||||
)
|
|
||||||
if err != nil {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
err = json.Unmarshal(response, &result)
|
|
||||||
if result.ErrCode != 0 {
|
|
||||||
err = fmt.Errorf("GetUserAccessToken error : errcode=%v , errmsg=%v", result.ErrCode, result.ErrMsg)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
return
|
|
||||||
}
|
|
||||||
39
work/work.go
39
work/work.go
@@ -1,39 +0,0 @@
|
|||||||
package work
|
|
||||||
|
|
||||||
import (
|
|
||||||
"github.com/silenceper/wechat/v2/credential"
|
|
||||||
"github.com/silenceper/wechat/v2/work/config"
|
|
||||||
"github.com/silenceper/wechat/v2/work/context"
|
|
||||||
"github.com/silenceper/wechat/v2/work/msgaudit"
|
|
||||||
"github.com/silenceper/wechat/v2/work/oauth"
|
|
||||||
)
|
|
||||||
|
|
||||||
// Work 企业微信
|
|
||||||
type Work struct {
|
|
||||||
ctx *context.Context
|
|
||||||
}
|
|
||||||
|
|
||||||
//NewWork init work
|
|
||||||
func NewWork(cfg *config.Config) *Work {
|
|
||||||
defaultAkHandle := credential.NewWorkAccessToken(cfg.CorpID, cfg.CorpSecret, credential.CacheKeyWorkPrefix, cfg.Cache)
|
|
||||||
ctx := &context.Context{
|
|
||||||
Config: cfg,
|
|
||||||
AccessTokenHandle: defaultAkHandle,
|
|
||||||
}
|
|
||||||
return &Work{ctx: ctx}
|
|
||||||
}
|
|
||||||
|
|
||||||
//GetContext get Context
|
|
||||||
func (wk *Work) GetContext() *context.Context {
|
|
||||||
return wk.ctx
|
|
||||||
}
|
|
||||||
|
|
||||||
//GetOauth get oauth
|
|
||||||
func (wk *Work) GetOauth() *oauth.Oauth {
|
|
||||||
return oauth.NewOauth(wk.ctx)
|
|
||||||
}
|
|
||||||
|
|
||||||
// GetMsgAudit get msgAudit
|
|
||||||
func (wk *Work) GetMsgAudit() (*msgaudit.Client, error) {
|
|
||||||
return msgaudit.NewClient(wk.ctx.Config)
|
|
||||||
}
|
|
||||||
Reference in New Issue
Block a user