499 lines
14 KiB
Go
499 lines
14 KiB
Go
package kefuWework
|
||
|
||
import (
|
||
"bytes"
|
||
"encoding/json"
|
||
"encoding/xml"
|
||
"fmt"
|
||
"github.com/tidwall/gjson"
|
||
"io/ioutil"
|
||
"kefu/lib"
|
||
"kefu/models"
|
||
"kefu/service"
|
||
"kefu/tools"
|
||
"kefu/ws"
|
||
"log"
|
||
"net/http"
|
||
"net/url"
|
||
"strings"
|
||
"time"
|
||
|
||
"github.com/gin-gonic/gin"
|
||
"github.com/patrickmn/go-cache"
|
||
)
|
||
|
||
// 验证企业微信回调的token
|
||
//var token = "taooshihan"
|
||
//
|
||
//// 验证企业微信回调的key
|
||
//var encodingAesKey = "ORSH1J7CaQtQi2b4dGY5gAJDUJFDOd5cO7Yd7knIxUF"
|
||
//
|
||
//// 企业微信企业id
|
||
//var corpid = "wwa4266261c4ea2c08"
|
||
//
|
||
//// 企业微信secret
|
||
//var corpsecret = "SPU2s8t5cmI3GfkAuggiTyPk5LsuPY22hgep4cGBzlQ"
|
||
|
||
// 企业微信的重试缓存，如果服务器延迟低，可以去掉该变量以及 isRetry 逻辑
|
||
var retryCache = cache.New(60*time.Minute, 10*time.Minute)
|
||
|
||
// 企业微信 token 缓存，请求频次过高可能有一些额外的问题
|
||
var tokenCache = cache.New(5*time.Minute, 5*time.Minute)
|
||
|
||
// 上下文对话能力，默认是 3, 可以根据需要修改对话长度
|
||
var weworkConversationSize = 3
|
||
|
||
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 AccessToken struct {
|
||
Errcode int `json:"errcode"`
|
||
Errmsg string `json:"errmsg"`
|
||
AccessToken string `json:"access_token"`
|
||
ExpiresIn int `json:"expires_in"`
|
||
}
|
||
|
||
type MsgRet struct {
|
||
Errcode int `json:"errcode"`
|
||
Errmsg string `json:"errmsg"`
|
||
NextCursor string `json:"next_cursor"`
|
||
MsgList []Msg `json:"msg_list"`
|
||
}
|
||
type Msg 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"`
|
||
OpenKfid string `json:"open_kfid"`
|
||
ExternalUserid string `json:"external_userid"`
|
||
}
|
||
|
||
type ReplyMsg 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"`
|
||
}
|
||
|
||
// 定义微信客服API的封装结构体
|
||
type KefuWework struct {
|
||
corpid string
|
||
corpsecret string
|
||
Token string
|
||
EncodingAESKey string
|
||
}
|
||
|
||
// 创建微信客服API的封装结构体实例
|
||
func NewKefuWework(corpid, corpsecret, Token, EncodingAESKey string) *KefuWework {
|
||
return &KefuWework{
|
||
corpid: corpid,
|
||
corpsecret: corpsecret,
|
||
Token: Token,
|
||
EncodingAESKey: EncodingAESKey,
|
||
}
|
||
}
|
||
|
||
// 微信客服消息
|
||
func TalkWeixin(c *gin.Context) {
|
||
entId := c.Param("entId")
|
||
verifyMsgSign := c.Query("msg_signature")
|
||
verifyTimestamp := c.Query("timestamp")
|
||
verifyNonce := c.Query("nonce")
|
||
|
||
//获取配置
|
||
configs := models.GetEntConfigsMap(entId, "kefuWeworkCorpid", "kefuWeworkSecret", "kefuWeworkToken", "kefuWeworkEncodingAESKey")
|
||
kefuWework := NewKefuWework(configs["kefuWeworkCorpid"], configs["kefuWeworkSecret"], configs["kefuWeworkToken"], configs["kefuWeworkEncodingAESKey"])
|
||
//读取xml数据
|
||
bodyBytes, _ := ioutil.ReadAll(c.Request.Body)
|
||
crypt := NewWXBizMsgCrypt(kefuWework.Token, kefuWework.EncodingAESKey, kefuWework.corpid, 1)
|
||
data, _ := crypt.DecryptMsg(verifyMsgSign, verifyTimestamp, verifyNonce, bodyBytes)
|
||
var weixinUserAskMsg WeixinUserAskMsg
|
||
err := xml.Unmarshal(data, &weixinUserAskMsg)
|
||
if err != nil {
|
||
log.Println("微信客服xml解析失败，error: " + err.Error())
|
||
return
|
||
}
|
||
accessToken, err := kefuWework.accessToken()
|
||
if err != nil {
|
||
log.Println("微信客服获取access_token失败，error: " + err.Error())
|
||
return
|
||
}
|
||
//获取最新的消息
|
||
msgToken := weixinUserAskMsg.Token
|
||
msgRet, err := getMsgs(accessToken, msgToken)
|
||
if err != nil {
|
||
log.Println("微信客服获取最新消息失败，error: " + err.Error())
|
||
return
|
||
}
|
||
if isRetry(verifyMsgSign) {
|
||
log.Println("微信客服重试机制 ")
|
||
return
|
||
}
|
||
//访客入库
|
||
handleVisitor(msgRet, entId, c, kefuWework)
|
||
//返回消息
|
||
go handleMsgRet(msgRet, entId, kefuWework)
|
||
c.JSON(200, "ok")
|
||
}
|
||
|
||
func (this *KefuWework) TalkToUser(external_userid, open_kfid, content string) {
|
||
reply := ReplyMsg{
|
||
Touser: external_userid,
|
||
OpenKfid: open_kfid,
|
||
Msgtype: "text",
|
||
Text: struct {
|
||
Content string `json:"content,omitempty"`
|
||
}{Content: content},
|
||
}
|
||
atoken, err := this.accessToken()
|
||
if err != nil {
|
||
return
|
||
}
|
||
callTalk(reply, atoken)
|
||
}
|
||
|
||
// 获取最新的消息处理是否自动回复
|
||
func handleMsgRet(msgRet MsgRet, entId string, wework *KefuWework) {
|
||
size := len(msgRet.MsgList)
|
||
if size < 1 {
|
||
return
|
||
}
|
||
current := msgRet.MsgList[size-1]
|
||
userId := current.ExternalUserid
|
||
kfId := current.OpenKfid
|
||
//消息类型
|
||
msgtype := current.Msgtype
|
||
//事件消息
|
||
if msgtype == "event" {
|
||
//用户进入会话事件
|
||
//current.Event.WelcomeCode
|
||
}
|
||
//文本消息
|
||
content := current.Text.Content
|
||
if content == "" {
|
||
go wework.TalkToUser(userId, kfId, "非常抱歉，当前仅支持文本消息，请输入文本。")
|
||
return
|
||
}
|
||
|
||
//查询机器人信息
|
||
configs := models.GetEntConfigsMap(entId, "QiyeWechatKefuPreRobot", "RobotStatus", "chatGPTSecret", "QdrantAIStatus")
|
||
if configs["RobotStatus"] == "3" {
|
||
return
|
||
}
|
||
kefuInfo := models.FindUser(kfId)
|
||
if kefuInfo.ID == 0 {
|
||
kefuInfo = models.FindUserByUid(entId)
|
||
}
|
||
visitorId := fmt.Sprintf("wxkf|%s|%s", entId, userId)
|
||
vistorInfo := models.FindVisitorByVistorId(visitorId)
|
||
ret := SearchLearn(entId, content)
|
||
//判断chatGPT
|
||
if ret == "" && configs["QdrantAIStatus"] == "true" && models.CheckVisitorRobotReply(vistorInfo.State) {
|
||
if configs["QiyeWechatKefuPreRobot"] != "" {
|
||
go wework.TalkToUser(userId, kfId, configs["QiyeWechatKefuPreRobot"])
|
||
}
|
||
ret = Gpt3Knowledge(entId, visitorId, kefuInfo, content)
|
||
}
|
||
//if ret == "" && configs["chatGPTSecret"] != "" {
|
||
// ret = service.Gpt3dot5Message(entId, "robot_message", content)
|
||
// //ret, _ = AskOnConversation(configs["chatGPTSecret"], content, userId, weworkConversationSize)
|
||
// //TalkToUser(userId, kfId, content, "该问题暂时无法回答，已收录到学习库")
|
||
//}
|
||
if ret == "" {
|
||
return
|
||
}
|
||
|
||
models.CreateMessage(kefuInfo.Name, visitorId, ret, "kefu", entId, "read")
|
||
go ws.KefuMessage(visitorId, ret, kefuInfo)
|
||
wework.TalkToUser(userId, kfId, ret)
|
||
}
|
||
func handleVisitor(msgRet MsgRet, entId string, c *gin.Context, wework *KefuWework) {
|
||
size := len(msgRet.MsgList)
|
||
if size < 1 {
|
||
return
|
||
}
|
||
current := msgRet.MsgList[size-1]
|
||
userId := current.ExternalUserid
|
||
kfId := current.OpenKfid
|
||
content := current.Text.Content
|
||
if content == "" {
|
||
return
|
||
}
|
||
|
||
//查找访客插入访客表
|
||
visitorId := fmt.Sprintf("wxkf|%s|%s", entId, userId)
|
||
vistorInfo := models.FindVisitorByVistorId(visitorId)
|
||
if vistorInfo.ID == 0 {
|
||
accessToken, _ := wework.accessToken()
|
||
ret, _ := getVisitor(accessToken, userId)
|
||
visitorName := gjson.Get(ret, "customer_list.0.nickname").String()
|
||
avator := gjson.Get(ret, "customer_list.0.avatar").String()
|
||
if visitorName == "" {
|
||
visitorName = "微信客服用户"
|
||
}
|
||
if avator == "" {
|
||
avator = "/static/images/we-chat-wx.png"
|
||
}
|
||
vistorInfo = models.Visitor{
|
||
Model: models.Model{
|
||
CreatedAt: time.Now(),
|
||
UpdatedAt: time.Now(),
|
||
},
|
||
Name: visitorName,
|
||
Avator: avator,
|
||
ToId: kfId,
|
||
VisitorId: visitorId,
|
||
Status: 1,
|
||
Refer: "来自微信客服",
|
||
EntId: entId,
|
||
VisitNum: 0,
|
||
City: "微信客服",
|
||
ClientIp: c.ClientIP(),
|
||
}
|
||
vistorInfo.AddVisitor()
|
||
} else {
|
||
go models.UpdateVisitorStatus(visitorId, 3)
|
||
}
|
||
|
||
kefuInfo := models.FindUser(kfId)
|
||
if kefuInfo.ID == 0 {
|
||
kefuInfo = models.FindUserByUid(entId)
|
||
}
|
||
go ws.VisitorOnline(kefuInfo.Name, vistorInfo)
|
||
go ws.VisitorSendToKefu(kefuInfo.Name, vistorInfo, content)
|
||
go service.SendWorkWechatMesage(vistorInfo.EntId, vistorInfo, kefuInfo, content, c)
|
||
}
|
||
func isRetry(signature string) bool {
|
||
var base = "retry:signature:%s"
|
||
key := fmt.Sprintf(base, signature)
|
||
_, found := retryCache.Get(key)
|
||
if found {
|
||
return true
|
||
}
|
||
retryCache.Set(key, "1", 1*time.Minute)
|
||
return false
|
||
}
|
||
|
||
func getMsgs(accessToken, msgToken string) (MsgRet, error) {
|
||
var msgRet MsgRet
|
||
url := "https://qyapi.weixin.qq.com/cgi-bin/kf/sync_msg?access_token=" + accessToken
|
||
method := "POST"
|
||
payload := strings.NewReader(fmt.Sprintf(`{"token" : "%s"}`, msgToken))
|
||
client := &http.Client{}
|
||
req, err := http.NewRequest(method, url, payload)
|
||
if err != nil {
|
||
fmt.Println(err)
|
||
return msgRet, err
|
||
}
|
||
req.Header.Add("Content-Type", "application/json")
|
||
|
||
res, err := client.Do(req)
|
||
if err != nil {
|
||
return msgRet, err
|
||
}
|
||
defer res.Body.Close()
|
||
|
||
body, err := ioutil.ReadAll(res.Body)
|
||
if err != nil {
|
||
log.Println("微信客服读取消息失败，error:", err)
|
||
return msgRet, err
|
||
}
|
||
json.Unmarshal(body, &msgRet)
|
||
return msgRet, nil
|
||
}
|
||
func getVisitor(accessToken, visitorId string) (string, error) {
|
||
var respData string
|
||
url := "https://qyapi.weixin.qq.com/cgi-bin/kf/customer/batchget?access_token=" + accessToken
|
||
method := "POST"
|
||
reqData := map[string]interface{}{
|
||
"external_userid_list": []string{
|
||
visitorId,
|
||
},
|
||
}
|
||
reqBody, err := json.Marshal(reqData)
|
||
if err != nil {
|
||
return respData, err
|
||
}
|
||
|
||
client := &http.Client{}
|
||
req, err := http.NewRequest(method, url, bytes.NewReader(reqBody))
|
||
if err != nil {
|
||
fmt.Println(err)
|
||
return respData, err
|
||
}
|
||
req.Header.Add("Content-Type", "application/json")
|
||
|
||
res, err := client.Do(req)
|
||
if err != nil {
|
||
return respData, err
|
||
}
|
||
defer res.Body.Close()
|
||
|
||
body, err := ioutil.ReadAll(res.Body)
|
||
if err != nil {
|
||
fmt.Println(err)
|
||
return respData, err
|
||
}
|
||
return string(body), nil
|
||
}
|
||
func (this *KefuWework) accessToken() (string, error) {
|
||
var tokenCacheKey = "tokenCache"
|
||
data, found := tokenCache.Get(tokenCacheKey)
|
||
if found {
|
||
return fmt.Sprintf("%v", data), nil
|
||
}
|
||
urlBase := "https://qyapi.weixin.qq.com/cgi-bin/gettoken?corpid=%s&corpsecret=%s"
|
||
url := fmt.Sprintf(urlBase, this.corpid, this.corpsecret)
|
||
method := "GET"
|
||
client := &http.Client{}
|
||
req, err := http.NewRequest(method, url, nil)
|
||
if err != nil {
|
||
fmt.Println(err)
|
||
return "", err
|
||
}
|
||
res, err := client.Do(req)
|
||
if err != nil {
|
||
fmt.Println(err)
|
||
return "", err
|
||
}
|
||
defer res.Body.Close()
|
||
|
||
body, err := ioutil.ReadAll(res.Body)
|
||
if err != nil {
|
||
fmt.Println(err)
|
||
return "", err
|
||
}
|
||
s := string(body)
|
||
var accessToken AccessToken
|
||
json.Unmarshal([]byte(s), &accessToken)
|
||
token := accessToken.AccessToken
|
||
tokenCache.Set(tokenCacheKey, token, 5*time.Minute)
|
||
return token, nil
|
||
}
|
||
func Gpt3Knowledge(entId, visitorId string, kefuInfo models.User, content string) string {
|
||
config := models.GetEntConfigsMap(entId, "chatGPTUrl", "chatGPTSecret", "QdrantAIStatus", "QdrantAICollect", "chatGPTSystem", "chatGPTPrompt",
|
||
"RobotName", "RobotAvator",
|
||
"chatGPTHistory")
|
||
api := models.FindConfig("BaseGPTKnowledge")
|
||
|
||
messageContent := ""
|
||
if config["QdrantAIStatus"] == "true" && config["QdrantAICollect"] != "" && api != "" {
|
||
kefuInfo.Nickname = tools.Ifelse(config["RobotName"] != "", config["RobotName"], kefuInfo.Nickname).(string)
|
||
path := fmt.Sprintf("%s/%s/searchStream", api, config["QdrantAICollect"])
|
||
data := url.Values{}
|
||
data.Set("keywords", content)
|
||
system := fmt.Sprintf("我希望你扮演%s客服人员", kefuInfo.Nickname)
|
||
if config["chatGPTSystem"] != "" {
|
||
system = config["chatGPTSystem"]
|
||
}
|
||
data.Set("system", system)
|
||
data.Set("prompt", config["chatGPTPrompt"])
|
||
data.Set("gptUrl", config["chatGPTUrl"])
|
||
data.Set("gptSecret", config["chatGPTSecret"])
|
||
//是否传递历史记录
|
||
if config["chatGPTHistory"] == "true" {
|
||
messages := models.FindMessageByQueryPage(1, 6, "visitor_id = ?", visitorId)
|
||
gptMessages := make([]lib.Gpt3Dot5Message, 0)
|
||
for i := len(messages) - 1; i >= 1; i-- {
|
||
reqContent := messages[i].Content
|
||
if messages[i].MesType == "visitor" {
|
||
gptMessages = append(gptMessages, lib.Gpt3Dot5Message{
|
||
Role: "user",
|
||
Content: reqContent,
|
||
})
|
||
} else {
|
||
gptMessages = append(gptMessages, lib.Gpt3Dot5Message{
|
||
Role: "assistant",
|
||
Content: reqContent,
|
||
})
|
||
}
|
||
}
|
||
historyJson, _ := json.Marshal(gptMessages)
|
||
data.Set("history", string(historyJson))
|
||
}
|
||
messageContent, _ = tools.PostForm(path, data)
|
||
}
|
||
return messageContent
|
||
}
|
||
|
||
func callTalk(reply ReplyMsg, accessToken string) error {
|
||
url := "https://qyapi.weixin.qq.com/cgi-bin/kf/send_msg?access_token=" + accessToken
|
||
method := "POST"
|
||
data, err := json.Marshal(reply)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
reqBody := string(data)
|
||
fmt.Println(reqBody)
|
||
payload := strings.NewReader(reqBody)
|
||
client := &http.Client{}
|
||
req, err := http.NewRequest(method, url, payload)
|
||
|
||
if err != nil {
|
||
return err
|
||
}
|
||
req.Header.Add("Content-Type", "application/json")
|
||
|
||
res, err := client.Do(req)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
defer res.Body.Close()
|
||
body, err := ioutil.ReadAll(res.Body)
|
||
if err != nil {
|
||
fmt.Println(err)
|
||
return err
|
||
}
|
||
s := string(body)
|
||
fmt.Println(s)
|
||
return nil
|
||
}
|
||
|
||
// 查询自己知识库内容
|
||
func SearchLearn(entId, content string) string {
|
||
articles := models.FindArticleList(1, 10, "score desc", "ent_id= ? and title like ?", entId, "%"+content+"%")
|
||
|
||
result := ""
|
||
articleLen := len(articles)
|
||
//未匹配，进入学习库
|
||
if articleLen == 0 {
|
||
if len([]rune(content)) > 1 {
|
||
go service.AddUpdateLearn(entId, content)
|
||
}
|
||
} else if articleLen == 1 {
|
||
result, _ = tools.ReplaceStringByRegex(articles[0].Content, `<[^a>]+>|target="[^"]+"`, "")
|
||
article := models.Article{
|
||
Score: articles[0].Score + 1,
|
||
}
|
||
go article.SaveArticle("ent_id = ? and id = ?", entId, articles[0].Id)
|
||
} else if articleLen > 1 {
|
||
result = "您是不是想问："
|
||
for _, article := range articles {
|
||
title := strings.Split(strings.ReplaceAll(article.Title, "，", ","), ",")[0]
|
||
result += fmt.Sprintf("\r\n<a href='weixin://bizmsgmenu?msgmenucontent=%s&msgmenuid=%d'>%s</a>", title, article.Id, title)
|
||
}
|
||
}
|
||
return result
|
||
}
|
||
|
||
//
|