256 lines
7.3 KiB
Go
256 lines
7.3 KiB
Go
|
package routers
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"encoding/json"
|
|||
|
|
"errors"
|
|||
|
|
"github.com/gin-gonic/gin"
|
|||
|
|
"github.com/tidwall/gjson"
|
|||
|
|
"log"
|
|||
|
|
"net/url"
|
|||
|
|
"strings"
|
|||
|
|
"time"
|
|||
|
|
"wechathook/models"
|
|||
|
|
"wechathook/service"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
var currentWxid string
|
|||
|
|
var currentWxNickname string
|
|||
|
|
var contactList map[string]string
|
|||
|
|
|
|||
|
|
func init() {
|
|||
|
|
//当前微信信息
|
|||
|
|
wechatinfo := Wechat39825Info()
|
|||
|
|
currentWxid = gjson.Get(wechatinfo, "data.wxid").String()
|
|||
|
|
currentWxNickname = gjson.Get(wechatinfo, "data.name").String()
|
|||
|
|
contacts := gjson.Get(Wechat39825ContactList(), "data").Array()
|
|||
|
|
contactList = make(map[string]string, 0)
|
|||
|
|
for _, contact := range contacts {
|
|||
|
|
nickname := contact.Get("nickname").String()
|
|||
|
|
wxid := contact.Get("wxid").String()
|
|||
|
|
contactList[wxid] = nickname
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
func Wechat39825(c *gin.Context) {
|
|||
|
|
// 声明一个map变量来存储JSON数据
|
|||
|
|
var jsonData map[string]interface{}
|
|||
|
|
// 解析JSON数据
|
|||
|
|
if err := c.BindJSON(&jsonData); err != nil {
|
|||
|
|
// 如果解析失败，返回错误信息
|
|||
|
|
c.JSON(400, gin.H{"error": err.Error()})
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
// 将JSON数据转换为字符串
|
|||
|
|
jsonBytes, _ := json.Marshal(jsonData)
|
|||
|
|
jsonString := string(jsonBytes)
|
|||
|
|
|
|||
|
|
messageType := gjson.Get(jsonString, "type").Int()
|
|||
|
|
log.Println("收到数据：", jsonString)
|
|||
|
|
|
|||
|
|
//好友请求
|
|||
|
|
if messageType == 37 {
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
if messageType == 1 {
|
|||
|
|
|
|||
|
|
content := gjson.Get(jsonString, "content").String()
|
|||
|
|
//if messageType == 3 {
|
|||
|
|
// content, _ = BaiduOCR(gjson.Get(jsonString, "base64Img").String())
|
|||
|
|
//}
|
|||
|
|
|
|||
|
|
displayFullContent := gjson.Get(jsonString, "displayFullContent").String()
|
|||
|
|
fromUser := gjson.Get(jsonString, "fromUser").String()
|
|||
|
|
|
|||
|
|
if currentWxid == fromUser {
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
//微信账号白名单
|
|||
|
|
wechatWhiteList := models.FindSetting("wechatWhiteList").Value
|
|||
|
|
reply := ""
|
|||
|
|
//群聊里@
|
|||
|
|
if strings.Contains(fromUser, "@chatroom") {
|
|||
|
|
if !strings.Contains(displayFullContent, "在群聊中@了") && !strings.Contains(wechatWhiteList, fromUser) {
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
wxidContent := strings.Split(content, ":\n")
|
|||
|
|
wxid := wxidContent[0]
|
|||
|
|
content := strings.Trim(wxidContent[1], "\n")
|
|||
|
|
content = strings.Replace(content, "@"+currentWxNickname, "", -1)
|
|||
|
|
log.Println("解析数据：", fromUser, displayFullContent, wxid, content)
|
|||
|
|
|
|||
|
|
if models.FindSettingDefault("groupKeywordStatus", "yes") != "no" {
|
|||
|
|
reply = service.GetReplyFromLocal(content)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
if reply == "" && models.FindSettingDefault("groupAIStatus", "yes") != "no" {
|
|||
|
|
reply, _ = service.GetReplyByAPI(fromUser, contactList[fromUser], "", content)
|
|||
|
|
}
|
|||
|
|
if reply != "" {
|
|||
|
|
SendWechat39825AtMsg(wxid, fromUser, reply)
|
|||
|
|
images := service.ExtractMarkdownImagePaths(reply)
|
|||
|
|
if len(images) != 0 {
|
|||
|
|
filePath, _ := service.DownloadImageToDailyDir(images[0].Path, "./resource", images[0].Desc)
|
|||
|
|
if filePath != "" {
|
|||
|
|
SendWechat39825AtMsg(wxid, fromUser, service.GetRootPath()+"\\"+filePath)
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
//私聊
|
|||
|
|
//if wechatWhiteList != "" && !strings.Contains(wechatWhiteList, fromUser) {
|
|||
|
|
// return
|
|||
|
|
//}
|
|||
|
|
if models.FindSettingDefault("personKeywordStatus", "yes") != "no" {
|
|||
|
|
reply = service.GetReplyFromLocal(content)
|
|||
|
|
}
|
|||
|
|
if reply == "" && models.FindSettingDefault("personAIStatus", "yes") != "no" {
|
|||
|
|
reply, _ = service.GetReplyByAPI(fromUser, contactList[fromUser], "", content)
|
|||
|
|
}
|
|||
|
|
if reply != "" {
|
|||
|
|
SendWechat39825Msg(fromUser, reply)
|
|||
|
|
images := service.ExtractMarkdownImagePaths(reply)
|
|||
|
|
if len(images) != 0 {
|
|||
|
|
filePath, _ := service.DownloadImageToDailyDir(images[0].Path, "./resource", images[0].Desc)
|
|||
|
|
if filePath != "" {
|
|||
|
|
SendWechat39825Msg(fromUser, service.GetRootPath()+"\\"+filePath)
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
c.JSON(200, gin.H{
|
|||
|
|
"message": "pong",
|
|||
|
|
})
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 发送微信消息
|
|||
|
|
func SendWechat39825AtMsg(wxids, chatRoomId, msg string) (string, error) {
|
|||
|
|
if msg == "" {
|
|||
|
|
return "", errors.New("回复内容为空！")
|
|||
|
|
}
|
|||
|
|
if service.IsWindowsPath(msg) {
|
|||
|
|
SendWechat39825File(chatRoomId, msg)
|
|||
|
|
return "", nil
|
|||
|
|
}
|
|||
|
|
sleepTime, _ := service.GetRandomValue(models.FindSettingDefault("delayTime", "0"))
|
|||
|
|
time.Sleep(time.Duration(sleepTime) * time.Second)
|
|||
|
|
url := "http://127.0.0.1:19088/api/sendAtText"
|
|||
|
|
data := map[string]interface{}{
|
|||
|
|
"wxids": wxids,
|
|||
|
|
"chatRoomId": chatRoomId,
|
|||
|
|
"msg": msg,
|
|||
|
|
}
|
|||
|
|
respBytes, err := service.PostJSON(url, data)
|
|||
|
|
respString := string(respBytes)
|
|||
|
|
log.Println("发送信息：", respString, msg)
|
|||
|
|
if err != nil {
|
|||
|
|
return "", err
|
|||
|
|
}
|
|||
|
|
return respString, nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 发送微信消息
|
|||
|
|
func SendWechat39825Msg(wxid, msg string) (string, error) {
|
|||
|
|
if msg == "" {
|
|||
|
|
return "", errors.New("回复内容为空！")
|
|||
|
|
}
|
|||
|
|
if service.IsWindowsPath(msg) {
|
|||
|
|
SendWechat39825File(wxid, msg)
|
|||
|
|
return "", nil
|
|||
|
|
}
|
|||
|
|
sleepTime, _ := service.GetRandomValue(models.FindSettingDefault("delayTime", "0"))
|
|||
|
|
time.Sleep(time.Duration(sleepTime) * time.Second)
|
|||
|
|
url := "http://127.0.0.1:19088/api/sendTextMsg"
|
|||
|
|
data := map[string]interface{}{
|
|||
|
|
"wxid": wxid,
|
|||
|
|
"msg": msg,
|
|||
|
|
}
|
|||
|
|
respBytes, err := service.PostJSON(url, data)
|
|||
|
|
respString := string(respBytes)
|
|||
|
|
log.Println("发送信息：", respString, msg)
|
|||
|
|
if err != nil {
|
|||
|
|
return "", err
|
|||
|
|
}
|
|||
|
|
return respString, nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 发送微信文件
|
|||
|
|
func SendWechat39825File(wxid, filePath string) (string, error) {
|
|||
|
|
if filePath == "" {
|
|||
|
|
return "", errors.New("回复内容为空！")
|
|||
|
|
}
|
|||
|
|
sleepTime, _ := service.GetRandomValue(models.FindSettingDefault("delayTime", "0"))
|
|||
|
|
time.Sleep(time.Duration(sleepTime) * time.Second)
|
|||
|
|
|
|||
|
|
url := "http://127.0.0.1:19088/api/sendFileMsg"
|
|||
|
|
data := map[string]interface{}{
|
|||
|
|
"wxid": wxid,
|
|||
|
|
"filePath": filePath,
|
|||
|
|
}
|
|||
|
|
if service.IsImagePath(filePath) {
|
|||
|
|
url = "http://127.0.0.1:19088/api/sendImagesMsg"
|
|||
|
|
data = map[string]interface{}{
|
|||
|
|
"wxid": wxid,
|
|||
|
|
"imagePath": filePath,
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
respBytes, err := service.PostJSON(url, data)
|
|||
|
|
respString := string(respBytes)
|
|||
|
|
log.Println("发送信息：", respString, filePath)
|
|||
|
|
if err != nil {
|
|||
|
|
return "", err
|
|||
|
|
}
|
|||
|
|
return respString, nil
|
|||
|
|
}
|
|||
|
|
func Wechat39825Info() string {
|
|||
|
|
url := "http://127.0.0.1:19088/api/userInfo"
|
|||
|
|
data := map[string]interface{}{}
|
|||
|
|
respBytes, _ := service.PostJSON(url, data)
|
|||
|
|
respString := string(respBytes)
|
|||
|
|
log.Println("当前微信信息：", respString)
|
|||
|
|
return respString
|
|||
|
|
}
|
|||
|
|
func Wechat39825ContactList() string {
|
|||
|
|
url := "http://127.0.0.1:19088/api/getContactList"
|
|||
|
|
data := map[string]interface{}{}
|
|||
|
|
respBytes, _ := service.PostJSON(url, data)
|
|||
|
|
respString := string(respBytes)
|
|||
|
|
log.Println("当前微信联系人：", respString)
|
|||
|
|
return respString
|
|||
|
|
}
|
|||
|
|
func Wechat39825Hook(c *gin.Context) {
|
|||
|
|
url := "http://127.0.0.1:19088/api/hookSyncMsg"
|
|||
|
|
data := map[string]interface{}{
|
|||
|
|
"port": "19099",
|
|||
|
|
"ip": "127.0.0.1",
|
|||
|
|
"url": "http://localhost:8089/wechat39825",
|
|||
|
|
"timeout": "3000",
|
|||
|
|
"enableHttp": true,
|
|||
|
|
}
|
|||
|
|
respBytes, _ := service.PostJSON(url, data)
|
|||
|
|
respString := string(respBytes)
|
|||
|
|
log.Println("发送Hook信息：", respString)
|
|||
|
|
c.Writer.Write(respBytes)
|
|||
|
|
}
|
|||
|
|
func Wechat39825Contacts(c *gin.Context) {
|
|||
|
|
resp := Wechat39825ContactList()
|
|||
|
|
c.Writer.Write([]byte(resp))
|
|||
|
|
}
|
|||
|
|
func Wechat39825Remark(c *gin.Context) {
|
|||
|
|
url := "http://127.0.0.1:19088/api/getDBInfo"
|
|||
|
|
data := map[string]interface{}{}
|
|||
|
|
respBytes, _ := service.PostJSON(url, data)
|
|||
|
|
respString := string(respBytes)
|
|||
|
|
log.Println("当前微信数据库信息：", respString)
|
|||
|
|
}
|
|||
|
|
func BaiduOCR(imgbase64 string) (string, error) {
|
|||
|
|
b := service.BaiduOCR{
|
|||
|
|
API_KEY: "NrtkiynNnqkQOwm6isip58gS",
|
|||
|
|
SECRET_KEY: "CO9594vqWsXVdhQEfXSNlDeHu5Jk7lWE",
|
|||
|
|
}
|
|||
|
|
r, err := b.OCR(url.QueryEscape(imgbase64))
|
|||
|
|
log.Println(r, err)
|
|||
|
|
return r, err
|
|||
|
|
}
|