110 lines
3.4 KiB
Go
110 lines
3.4 KiB
Go
package db
|
||
|
||
import (
|
||
"bytes"
|
||
"crypto/hmac"
|
||
"crypto/sha256"
|
||
"encoding/hex"
|
||
"encoding/json"
|
||
"fmt"
|
||
"io/ioutil"
|
||
"net/http"
|
||
"regexp"
|
||
"strconv"
|
||
"time"
|
||
"xiawan/wx/protobuf/wechat"
|
||
|
||
"github.com/lunny/log"
|
||
)
|
||
|
||
// MessageCallbackPayload 消息回调内容体
|
||
type MessageCallbackPayload struct {
|
||
UUID string `json:"uuid"` // 微信UUID标识
|
||
MsgId string `json:"msg_id"` // 消息ID
|
||
FromUser string `json:"from_user"` // 发送方
|
||
ToUser string `json:"to_user"` // 接收方
|
||
MsgType int `json:"msg_type"` // 消息类型
|
||
Content string `json:"content"` // 消息内容
|
||
CreateTime int64 `json:"create_time"` // 消息创建时间
|
||
IsGroup bool `json:"is_group"` // 是否群消息
|
||
Attachments map[string]interface{} `json:"attachments,omitempty"` // 附件信息
|
||
RawData map[string]interface{} `json:"raw_data,omitempty"` // 原始数据
|
||
}
|
||
|
||
// decodeUnicodeEscapes 解码Unicode转义字符（复用自 wxsocketmsg.go）
|
||
func decodeUnicodeEscapes(s string) string {
|
||
// 处理JSON中的Unicode转义字符，如\u003c
|
||
re := regexp.MustCompile(`\\u([0-9a-fA-F]{4})`)
|
||
result := re.ReplaceAllStringFunc(s, func(match string) string {
|
||
// 提取十六进制数字部分
|
||
hexStr := match[2:] // 去掉\u前缀
|
||
// 将十六进制转换为整数
|
||
if codePoint, err := strconv.ParseInt(hexStr, 16, 32); err == nil {
|
||
// 转换为Unicode字符
|
||
return string(rune(codePoint))
|
||
}
|
||
return match // 如果转换失败，返回原字符串
|
||
})
|
||
|
||
return result
|
||
}
|
||
|
||
// GenerateSignature 生成签名
|
||
func GenerateSignature(payload []byte, key string) string {
|
||
h := hmac.New(sha256.New, []byte(key))
|
||
h.Write(payload)
|
||
return hex.EncodeToString(h.Sum(nil))
|
||
}
|
||
|
||
// SendMessageCallback 发送消息回调
|
||
func SendMessageCallback(msg *wechat.AddMsg, uuid string) {
|
||
// 获取回调配置
|
||
config, err := GetMessageCallbackConfig(uuid)
|
||
if err != nil || config == nil || !config.Enabled {
|
||
return // 没有配置或未启用回调，直接返回
|
||
}
|
||
|
||
// 使用消息包装器包含UUID信息
|
||
msgWrapper := NewCallbackMessageWrapper(uuid, msg, "message")
|
||
jsonBytes, err := json.Marshal(msgWrapper)
|
||
if err != nil {
|
||
log.Errorf("Failed to marshal CallbackMessageWrapper: %v", err)
|
||
return
|
||
}
|
||
|
||
// 对JSON内容进行Unicode解码处理（与 WebSocket 保持一致）
|
||
decodedJSON := decodeUnicodeEscapes(string(jsonBytes))
|
||
|
||
// 创建HTTP请求
|
||
req, err := http.NewRequest("POST", config.CallbackURL, bytes.NewBuffer([]byte(decodedJSON)))
|
||
if err != nil {
|
||
log.Errorf("Failed to create callback request: %v", err)
|
||
return
|
||
}
|
||
|
||
// 设置请求头
|
||
req.Header.Set("Content-Type", "application/json")
|
||
req.Header.Set("X-Timestamp", fmt.Sprintf("%d", time.Now().Unix()))
|
||
|
||
// 发送请求
|
||
client := &http.Client{
|
||
Timeout: 5 * time.Second,
|
||
}
|
||
resp, err := client.Do(req)
|
||
if err != nil {
|
||
log.Errorf("Failed to send callback request: %v", err)
|
||
return
|
||
}
|
||
defer resp.Body.Close()
|
||
|
||
// 读取响应内容
|
||
_, err = ioutil.ReadAll(resp.Body)
|
||
if err != nil {
|
||
log.Errorf("Failed to read callback response: %v", err)
|
||
return
|
||
}
|
||
|
||
// 记录回调结果
|
||
log.Infof("Message callback sent to %s for message %d, status: %d", config.CallbackURL, msg.GetMsgId(), resp.StatusCode)
|
||
}
|