486 lines
12 KiB
Go
486 lines
12 KiB
Go
|
package wechat_kf_sdk
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"bytes"
|
|||
|
|
"encoding/json"
|
|||
|
|
"encoding/xml"
|
|||
|
|
"errors"
|
|||
|
|
"fmt"
|
|||
|
|
"github.com/patrickmn/go-cache"
|
|||
|
|
"github.com/silenceper/wechat/v2/util"
|
|||
|
|
"github.com/tidwall/gjson"
|
|||
|
|
"io"
|
|||
|
|
"io/ioutil"
|
|||
|
|
"log"
|
|||
|
|
"net/http"
|
|||
|
|
"os"
|
|||
|
|
"strings"
|
|||
|
|
"sync"
|
|||
|
|
"time"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
// 定义获取access_token的响应数据结构体
|
|||
|
|
type AccessTokenResponse struct {
|
|||
|
|
ErrCode int `json:"errcode"` // 错误码
|
|||
|
|
ErrMsg string `json:"errmsg"` // 错误信息
|
|||
|
|
AccessToken string `json:"access_token"` // access_token
|
|||
|
|
ExpiresIn int `json:"expires_in"` // 过期时间
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 定义微信客服API的封装结构体
|
|||
|
|
type KefuWework struct {
|
|||
|
|
corpid string // 企业ID
|
|||
|
|
corpsecret string // 企业密钥
|
|||
|
|
Token string // 令牌
|
|||
|
|
EncodingAESKey string // AES加密密钥
|
|||
|
|
mutex sync.Mutex // 互斥锁，用于保护access_token的获取
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 微信回调消息
|
|||
|
|
type WeixinUserAskMsg struct {
|
|||
|
|
ToUserName string `xml:"ToUserName"`
|
|||
|
|
CreateTime int64 `xml:"CreateTime"`
|
|||
|
|
MsgType string `xml:"MsgType"`
|
|||
|
|
Event string `xml:"Event"`
|
|||
|
|
Token string `xml:"Token"`
|
|||
|
|
OpenKfId string `xml:"OpenKfId"`
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 同步消息结果
|
|||
|
|
type SyncMsgRet struct {
|
|||
|
|
Errcode int `json:"errcode"`
|
|||
|
|
Errmsg string `json:"errmsg"`
|
|||
|
|
NextCursor string `json:"next_cursor"`
|
|||
|
|
MsgList []SyncMsg `json:"msg_list"`
|
|||
|
|
}
|
|||
|
|
type SyncMsg struct {
|
|||
|
|
Msgid string `json:"msgid"`
|
|||
|
|
SendTime int64 `json:"send_time"`
|
|||
|
|
Origin int `json:"origin"`
|
|||
|
|
Msgtype string `json:"msgtype"`
|
|||
|
|
Event struct {
|
|||
|
|
EventType string `json:"event_type"`
|
|||
|
|
Scene string `json:"scene"`
|
|||
|
|
OpenKfid string `json:"open_kfid"`
|
|||
|
|
ExternalUserid string `json:"external_userid"`
|
|||
|
|
WelcomeCode string `json:"welcome_code"`
|
|||
|
|
} `json:"event"`
|
|||
|
|
Text struct {
|
|||
|
|
Content string `json:"content"`
|
|||
|
|
} `json:"text"`
|
|||
|
|
Image struct {
|
|||
|
|
MediaId string `json:"media_id"`
|
|||
|
|
} `json:"image"`
|
|||
|
|
Voice struct {
|
|||
|
|
MediaId string `json:"media_id"`
|
|||
|
|
} `json:"voice"`
|
|||
|
|
OpenKfid string `json:"open_kfid"`
|
|||
|
|
ExternalUserid string `json:"external_userid"`
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 发送的文本消息
|
|||
|
|
type SendMsgText struct {
|
|||
|
|
Touser string `json:"touser,omitempty"`
|
|||
|
|
OpenKfid string `json:"open_kfid,omitempty"`
|
|||
|
|
Msgid string `json:"msgid,omitempty"`
|
|||
|
|
Msgtype string `json:"msgtype,omitempty"`
|
|||
|
|
Text struct {
|
|||
|
|
Content string `json:"content,omitempty"`
|
|||
|
|
} `json:"text,omitempty"`
|
|||
|
|
Image struct {
|
|||
|
|
MediaId string `json:"media_id,omitempty"`
|
|||
|
|
} `json:"image,omitempty"`
|
|||
|
|
Voice struct {
|
|||
|
|
MediaId string `json:"media_id,omitempty"`
|
|||
|
|
} `json:"voice,omitempty"`
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
var weworkCache = cache.New(5*time.Minute, 10*time.Minute) // 缓存，用于存储access_token
|
|||
|
|
|
|||
|
|
// 创建微信客服API的封装结构体实例
|
|||
|
|
func NewKefuWework(corpid string, corpsecret, Token, EncodingAESKey string) *KefuWework {
|
|||
|
|
return &KefuWework{
|
|||
|
|
corpid: corpid,
|
|||
|
|
corpsecret: corpsecret,
|
|||
|
|
Token: Token,
|
|||
|
|
EncodingAESKey: EncodingAESKey,
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 获取access_token的函数
|
|||
|
|
func (s *KefuWework) GetAccessToken() (string, error) {
|
|||
|
|
// 加锁，避免并发调用获取access_token接口
|
|||
|
|
s.mutex.Lock()
|
|||
|
|
defer s.mutex.Unlock()
|
|||
|
|
|
|||
|
|
// 判断access_token是否过期，如果未过期则直接返回
|
|||
|
|
cacheKey := "wework_access_" + s.corpid
|
|||
|
|
if accessToken, ok := weworkCache.Get(cacheKey); ok {
|
|||
|
|
return accessToken.(string), nil
|
|||
|
|
}
|
|||
|
|
// 发送GET请求，构建请求URL
|
|||
|
|
reqURL := fmt.Sprintf("%s?corpid=%s&corpsecret=%s", "https://qyapi.weixin.qq.com/cgi-bin/gettoken", s.corpid, s.corpsecret)
|
|||
|
|
resp, err := http.Get(reqURL)
|
|||
|
|
if err != nil {
|
|||
|
|
return "", err
|
|||
|
|
}
|
|||
|
|
defer resp.Body.Close()
|
|||
|
|
// 解析响应数据到AccessTokenResponse结构体
|
|||
|
|
var tokenResp AccessTokenResponse
|
|||
|
|
if err := json.NewDecoder(resp.Body).Decode(&tokenResp); err != nil {
|
|||
|
|
return "", err
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 判断获取access_token是否成功
|
|||
|
|
if tokenResp.ErrCode != 0 {
|
|||
|
|
return "", fmt.Errorf("GetAccessToken failed: %s", tokenResp.ErrMsg)
|
|||
|
|
}
|
|||
|
|
weworkCache.Set(cacheKey, tokenResp.AccessToken, time.Duration(tokenResp.ExpiresIn-3600)*time.Second)
|
|||
|
|
log.Printf("GetAccessToken kefuWework:%s\n", tokenResp.AccessToken)
|
|||
|
|
// 返回access_token
|
|||
|
|
return tokenResp.AccessToken, nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// GetAccountList 获取客服帐号列表，包括所有的客服帐号的客服ID、名称和头像。
|
|||
|
|
func (s *KefuWework) GetAccountList(offset, limit uint32) (string, error) {
|
|||
|
|
var respData string
|
|||
|
|
// 获取access_token
|
|||
|
|
accessToken, err := s.GetAccessToken()
|
|||
|
|
if err != nil {
|
|||
|
|
return respData, err
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 构建请求数据
|
|||
|
|
reqData := map[string]uint32{
|
|||
|
|
"offset": offset, // 偏移量
|
|||
|
|
"limit": limit, // 限制数量
|
|||
|
|
}
|
|||
|
|
reqBody, err := json.Marshal(reqData)
|
|||
|
|
if err != nil {
|
|||
|
|
return respData, err
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 发送POST请求
|
|||
|
|
reqURL := fmt.Sprintf("https://qyapi.weixin.qq.com/cgi-bin/kf/account/list?access_token=%s", accessToken)
|
|||
|
|
resp, err := http.Post(reqURL, "application/json", bytes.NewReader(reqBody))
|
|||
|
|
if err != nil {
|
|||
|
|
return respData, err
|
|||
|
|
}
|
|||
|
|
defer resp.Body.Close()
|
|||
|
|
|
|||
|
|
ioBytes, _ := io.ReadAll(resp.Body)
|
|||
|
|
|
|||
|
|
return string(ioBytes), nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// GetAccountList 获取客服帐号链接
|
|||
|
|
func (s *KefuWework) GetAccountLink(openKfId string) (string, error) {
|
|||
|
|
var respData string
|
|||
|
|
// 获取access_token
|
|||
|
|
accessToken, err := s.GetAccessToken()
|
|||
|
|
if err != nil {
|
|||
|
|
return respData, err
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 构建请求数据
|
|||
|
|
reqData := map[string]string{
|
|||
|
|
"open_kfid": openKfId,
|
|||
|
|
"scene": "123456",
|
|||
|
|
}
|
|||
|
|
reqBody, err := json.Marshal(reqData)
|
|||
|
|
if err != nil {
|
|||
|
|
return respData, err
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 发送POST请求
|
|||
|
|
reqURL := fmt.Sprintf("https://qyapi.weixin.qq.com/cgi-bin/kf/add_contact_way?access_token=%s", accessToken)
|
|||
|
|
resp, err := http.Post(reqURL, "application/json", bytes.NewReader(reqBody))
|
|||
|
|
if err != nil {
|
|||
|
|
return respData, err
|
|||
|
|
}
|
|||
|
|
defer resp.Body.Close()
|
|||
|
|
|
|||
|
|
ioBytes, _ := io.ReadAll(resp.Body)
|
|||
|
|
|
|||
|
|
return string(ioBytes), nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 发送同步客服消息请求
|
|||
|
|
func (s *KefuWework) SyncMsg(reqData map[string]interface{}) (SyncMsgRet, error) {
|
|||
|
|
var msgRet SyncMsgRet
|
|||
|
|
// 获取access_token
|
|||
|
|
accessToken, err := s.GetAccessToken()
|
|||
|
|
if err != nil {
|
|||
|
|
return msgRet, err
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
reqBody, err := json.Marshal(reqData)
|
|||
|
|
if err != nil {
|
|||
|
|
return msgRet, err
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 发送POST请求
|
|||
|
|
reqURL := fmt.Sprintf("https://qyapi.weixin.qq.com/cgi-bin/kf/sync_msg?access_token=%s", accessToken)
|
|||
|
|
resp, err := http.Post(reqURL, "application/json", bytes.NewReader(reqBody))
|
|||
|
|
if err != nil {
|
|||
|
|
return msgRet, err
|
|||
|
|
}
|
|||
|
|
defer resp.Body.Close()
|
|||
|
|
|
|||
|
|
ioBytes, _ := ioutil.ReadAll(resp.Body)
|
|||
|
|
json.Unmarshal(ioBytes, &msgRet)
|
|||
|
|
return msgRet, nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 发送消息
|
|||
|
|
func (s *KefuWework) SendMsg(reply interface{}) error {
|
|||
|
|
// 获取access_token
|
|||
|
|
accessToken, err := s.GetAccessToken()
|
|||
|
|
if err != nil {
|
|||
|
|
return err
|
|||
|
|
}
|
|||
|
|
url := "https://qyapi.weixin.qq.com/cgi-bin/kf/send_msg?access_token=" + accessToken
|
|||
|
|
reqBody, err := json.Marshal(reply)
|
|||
|
|
if err != nil {
|
|||
|
|
return err
|
|||
|
|
}
|
|||
|
|
resp, err := http.Post(url, "application/json", bytes.NewReader(reqBody))
|
|||
|
|
if err != nil {
|
|||
|
|
return err
|
|||
|
|
}
|
|||
|
|
defer resp.Body.Close()
|
|||
|
|
body, err := ioutil.ReadAll(resp.Body)
|
|||
|
|
if err != nil {
|
|||
|
|
log.Println("SendMsg：", body, err)
|
|||
|
|
return err
|
|||
|
|
}
|
|||
|
|
log.Println("SendMsg：", string(body))
|
|||
|
|
return nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 发送文本消息
|
|||
|
|
func (s *KefuWework) SendTextMsg(kfId, toUser, content string) error {
|
|||
|
|
reply := SendMsgText{
|
|||
|
|
Touser: toUser,
|
|||
|
|
OpenKfid: kfId,
|
|||
|
|
Msgid: "",
|
|||
|
|
Msgtype: "text",
|
|||
|
|
Text: struct {
|
|||
|
|
Content string `json:"content,omitempty"`
|
|||
|
|
}{
|
|||
|
|
Content: content,
|
|||
|
|
},
|
|||
|
|
}
|
|||
|
|
err := s.SendMsg(reply)
|
|||
|
|
return err
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 发送图片消息消息
|
|||
|
|
func (s *KefuWework) SendImagesMsg(kfId, toUser, path string) error {
|
|||
|
|
// 获取access_token
|
|||
|
|
accessToken, err := s.GetAccessToken()
|
|||
|
|
if err != nil {
|
|||
|
|
return err
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
uri := fmt.Sprintf("https://qyapi.weixin.qq.com/cgi-bin/media/upload?access_token=%s&type=%s", accessToken, "image")
|
|||
|
|
response, err := util.PostFile("media", path, uri)
|
|||
|
|
mediaId := gjson.Get(string(response), "media_id").String()
|
|||
|
|
reply := SendMsgText{
|
|||
|
|
Touser: toUser,
|
|||
|
|
OpenKfid: kfId,
|
|||
|
|
Msgid: "",
|
|||
|
|
Msgtype: "image",
|
|||
|
|
Image: struct {
|
|||
|
|
MediaId string `json:"media_id,omitempty"`
|
|||
|
|
}{
|
|||
|
|
MediaId: mediaId,
|
|||
|
|
},
|
|||
|
|
}
|
|||
|
|
err = s.SendMsg(reply)
|
|||
|
|
return err
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 发送语音消息
|
|||
|
|
func (s *KefuWework) SendVoiceMsg(kfId, toUser, path string) error {
|
|||
|
|
// 获取access_token
|
|||
|
|
accessToken, err := s.GetAccessToken()
|
|||
|
|
if err != nil {
|
|||
|
|
return err
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
uri := fmt.Sprintf("https://qyapi.weixin.qq.com/cgi-bin/media/upload?access_token=%s&type=%s", accessToken, "voice")
|
|||
|
|
response, err := util.PostFile("media", path, uri)
|
|||
|
|
mediaId := gjson.Get(string(response), "media_id").String()
|
|||
|
|
reply := SendMsgText{
|
|||
|
|
Touser: toUser,
|
|||
|
|
OpenKfid: kfId,
|
|||
|
|
Msgid: "",
|
|||
|
|
Msgtype: "voice",
|
|||
|
|
Voice: struct {
|
|||
|
|
MediaId string `json:"media_id,omitempty"`
|
|||
|
|
}{
|
|||
|
|
MediaId: mediaId,
|
|||
|
|
},
|
|||
|
|
}
|
|||
|
|
err = s.SendMsg(reply)
|
|||
|
|
return err
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 发送客服欢迎语
|
|||
|
|
func (s *KefuWework) SendWelcomeMsg(content, welcomeCode string) error {
|
|||
|
|
// 获取access_token
|
|||
|
|
accessToken, err := s.GetAccessToken()
|
|||
|
|
if err != nil {
|
|||
|
|
return err
|
|||
|
|
}
|
|||
|
|
// 构建请求数据
|
|||
|
|
reqData := map[string]interface{}{
|
|||
|
|
"code": welcomeCode,
|
|||
|
|
"msgtype": "text",
|
|||
|
|
"text": map[string]string{
|
|||
|
|
"content": content,
|
|||
|
|
},
|
|||
|
|
}
|
|||
|
|
reqBody, err := json.Marshal(reqData)
|
|||
|
|
if err != nil {
|
|||
|
|
log.Println("SendWelcomeMsg：", err)
|
|||
|
|
return err
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 发送POST请求
|
|||
|
|
reqURL := fmt.Sprintf("https://qyapi.weixin.qq.com/cgi-bin/kf/send_msg_on_event?access_token=%s", accessToken)
|
|||
|
|
resp, err := http.Post(reqURL, "application/json", bytes.NewReader(reqBody))
|
|||
|
|
if err != nil {
|
|||
|
|
log.Println("SendWelcomeMsg：", err)
|
|||
|
|
return err
|
|||
|
|
}
|
|||
|
|
defer resp.Body.Close()
|
|||
|
|
|
|||
|
|
resBody, err := io.ReadAll(resp.Body)
|
|||
|
|
log.Println("SendWelcomeMsg：", string(resBody), err)
|
|||
|
|
if err != nil {
|
|||
|
|
log.Println("SendWelcomeMsg：", err)
|
|||
|
|
return err
|
|||
|
|
}
|
|||
|
|
return nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 发送WEBHOOK
|
|||
|
|
func (s *KefuWework) SendWebHook(url, content string) error {
|
|||
|
|
// 获取access_token
|
|||
|
|
_, err := s.GetAccessToken()
|
|||
|
|
if err != nil {
|
|||
|
|
return err
|
|||
|
|
}
|
|||
|
|
return nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 获取客户列表
|
|||
|
|
func (s *KefuWework) BatchGet(reqData map[string]interface{}) (string, error) {
|
|||
|
|
var respData string
|
|||
|
|
// 获取access_token
|
|||
|
|
accessToken, err := s.GetAccessToken()
|
|||
|
|
if err != nil {
|
|||
|
|
return respData, err
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
reqBody, err := json.Marshal(reqData)
|
|||
|
|
if err != nil {
|
|||
|
|
return respData, err
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 发送POST请求
|
|||
|
|
reqURL := fmt.Sprintf("https://qyapi.weixin.qq.com/cgi-bin/kf/customer/batchget?access_token=%s", accessToken)
|
|||
|
|
resp, err := http.Post(reqURL, "application/json", bytes.NewReader(reqBody))
|
|||
|
|
if err != nil {
|
|||
|
|
return respData, err
|
|||
|
|
}
|
|||
|
|
defer resp.Body.Close()
|
|||
|
|
|
|||
|
|
ioBytes, _ := io.ReadAll(resp.Body)
|
|||
|
|
|
|||
|
|
return string(ioBytes), nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 验证签名
|
|||
|
|
func (s *KefuWework) CheckSign(signature, timestamp, nonce, echostr string) (string, error) {
|
|||
|
|
wxcpt := NewWXBizMsgCrypt(s.Token, s.EncodingAESKey, s.corpid, XmlType)
|
|||
|
|
echoStr, cryptErr := wxcpt.VerifyURL(signature, timestamp, nonce, echostr)
|
|||
|
|
if cryptErr != nil {
|
|||
|
|
return "", errors.New(cryptErr.ErrMsg)
|
|||
|
|
}
|
|||
|
|
return string(echoStr), nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 解析数据
|
|||
|
|
func (s *KefuWework) DecryptMsg(signature, timestamp, nonce, data string) (WeixinUserAskMsg, error) {
|
|||
|
|
var weixinUserAskMsg WeixinUserAskMsg
|
|||
|
|
wxcpt := NewWXBizMsgCrypt(s.Token, s.EncodingAESKey, s.corpid, XmlType)
|
|||
|
|
msg, cryptErr := wxcpt.DecryptMsg(signature, timestamp, nonce, []byte(data))
|
|||
|
|
if cryptErr != nil {
|
|||
|
|
return weixinUserAskMsg, errors.New(cryptErr.ErrMsg)
|
|||
|
|
}
|
|||
|
|
err := xml.Unmarshal(msg, &weixinUserAskMsg)
|
|||
|
|
if err != nil {
|
|||
|
|
return weixinUserAskMsg, err
|
|||
|
|
}
|
|||
|
|
return weixinUserAskMsg, nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 下载临时文件
|
|||
|
|
func (s *KefuWework) DownloadTempFileByMediaID(mediaId, path string) (string, error) {
|
|||
|
|
respData := ""
|
|||
|
|
// 获取access_token
|
|||
|
|
accessToken, err := s.GetAccessToken()
|
|||
|
|
if err != nil {
|
|||
|
|
return respData, err
|
|||
|
|
}
|
|||
|
|
// 发起HTTP GET请求
|
|||
|
|
url := fmt.Sprintf("https://qyapi.weixin.qq.com/cgi-bin/media/get?access_token=%s&media_id=%s", accessToken, mediaId)
|
|||
|
|
response, err := http.Get(url)
|
|||
|
|
if err != nil {
|
|||
|
|
return respData, err
|
|||
|
|
}
|
|||
|
|
defer response.Body.Close()
|
|||
|
|
|
|||
|
|
// 创建本地文件用于保存下载的文件
|
|||
|
|
file, err := os.Create(path)
|
|||
|
|
if err != nil {
|
|||
|
|
return respData, err
|
|||
|
|
}
|
|||
|
|
defer file.Close()
|
|||
|
|
|
|||
|
|
// 将HTTP响应的内容复制到本地文件
|
|||
|
|
_, err = io.Copy(file, response.Body)
|
|||
|
|
if err != nil {
|
|||
|
|
return respData, err
|
|||
|
|
}
|
|||
|
|
return path, nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// ConvertXMLToJSON 将 XML 转换为 JSON 字符串
|
|||
|
|
func ConvertXMLToJSON(xmlStr string) (string, error) {
|
|||
|
|
var result interface{}
|
|||
|
|
err := xml.Unmarshal([]byte(xmlStr), &result)
|
|||
|
|
if err != nil {
|
|||
|
|
return "", err
|
|||
|
|
}
|
|||
|
|
jsonStr, err := json.Marshal(result)
|
|||
|
|
if err != nil {
|
|||
|
|
return "", err
|
|||
|
|
}
|
|||
|
|
return string(jsonStr), nil
|
|||
|
|
}
|
|||
|
|
func parseFileName(response *http.Response) string {
|
|||
|
|
headers := response.Header["Content-Disposition"]
|
|||
|
|
if len(headers) > 0 {
|
|||
|
|
// 提取文件名
|
|||
|
|
fileParts := strings.Split(headers[0], "=")
|
|||
|
|
if len(fileParts) > 1 {
|
|||
|
|
return fileParts[1]
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
return ""
|
|||
|
|
}
|