65 lines
1.8 KiB
Go
65 lines
1.8 KiB
Go
package message
|
||
|
||
import (
|
||
"encoding/xml"
|
||
"gitee.ltd/lxh/wechat-robot/internal/minio"
|
||
"gitee.ltd/lxh/wechat-robot/internal/model"
|
||
"gitee.ltd/lxh/wechat-robot/internal/types"
|
||
"gitee.ltd/lxh/wechat-robot/internal/wechat"
|
||
"gitee.ltd/wechat-client/xybot"
|
||
"github.com/gofiber/fiber/v2/log"
|
||
)
|
||
|
||
// handlerMedia
|
||
// @description: 处理媒体消息
|
||
// @param msg *model.Message 消息记录
|
||
// @param client *xybot.Client 机器人客户端
|
||
func handlerMedia(msg *model.Message, client *xybot.Client) {
|
||
var err error
|
||
|
||
// 解析消息xml文件
|
||
// 解析为结构体
|
||
var media wechat.MediaMessage
|
||
if err = xml.Unmarshal([]byte(msg.Content), &media); err != nil {
|
||
log.Errorf("%v解析消息失败: %v", msg.Type, err.Error())
|
||
return
|
||
}
|
||
|
||
var md5Str, bs64Str string
|
||
switch msg.Type {
|
||
case types.MsgTypeImage:
|
||
fileUrl := ""
|
||
md5Str = media.Img.Md5
|
||
if media.Img.CdnBigImgUrl != "" {
|
||
fileUrl = media.Img.CdnBigImgUrl
|
||
} else if media.Img.CdnMidImgUrl != "" {
|
||
fileUrl = media.Img.CdnMidImgUrl
|
||
} else if media.Img.CdnThumbUrl != "" {
|
||
fileUrl = media.Img.CdnThumbUrl
|
||
} else {
|
||
return
|
||
}
|
||
if bs64Str, err = client.Tool.CdnDownloadImg(media.Img.AesKey, fileUrl); err != nil {
|
||
log.Errorf("图片文件下载失败: %s", err.Error())
|
||
return
|
||
}
|
||
case types.MsgTypeVideo:
|
||
if bs64Str, err = client.Tool.DownloadVideo(msg.ClientMsgId); err != nil {
|
||
log.Errorf("视频文件下载失败: %s", err.Error())
|
||
return
|
||
}
|
||
}
|
||
|
||
// 下载完成，保存到OSS
|
||
msg.FileUrl, err = minio.SaveBase64(bs64Str, md5Str)
|
||
if err != nil {
|
||
log.Errorf("文件保存到Minio失败: %s", err.Error())
|
||
return
|
||
}
|
||
// 更新数据库
|
||
if err = model.GetDB().Model(&msg).Where("id = ?", msg.ID).Update("file_url", msg.FileUrl).Error; err != nil {
|
||
log.Errorf("更新数据库失败: %s", err.Error())
|
||
return
|
||
}
|
||
}
|