mirror of
https://github.com/LLOneBot/LLOneBot.git
synced 2024-11-22 01:56:33 +00:00
445 lines
18 KiB
TypeScript
445 lines
18 KiB
TypeScript
import {AtType, ChatType, Group, RawMessage, SendArkElement, SendMessageElement} from "../../ntqqapi/types";
|
||
import {friends, getFriend, getGroup, getGroupMember, getUidByUin, selfInfo,} from "../../common/data";
|
||
import {
|
||
OB11MessageCustomMusic,
|
||
OB11MessageData,
|
||
OB11MessageDataType,
|
||
OB11MessageMixType,
|
||
OB11MessageNode,
|
||
OB11PostSendMsg
|
||
} from '../types';
|
||
import {NTQQApi, Peer} from "../../ntqqapi/ntcall";
|
||
import {SendMsgElementConstructor} from "../../ntqqapi/constructor";
|
||
import {uri2local} from "../utils";
|
||
import BaseAction from "./BaseAction";
|
||
import {ActionName, BaseCheckResult} from "./types";
|
||
import * as fs from "node:fs";
|
||
import {log, sleep} from "../../common/utils";
|
||
import {decodeCQCode} from "../cqcode";
|
||
import {dbUtil} from "../../common/db";
|
||
import {ALLOW_SEND_TEMP_MSG} from "../../common/config";
|
||
import {FileCache} from "../../common/types";
|
||
|
||
function checkSendMessage(sendMsgList: OB11MessageData[]) {
|
||
function checkUri(uri: string): boolean {
|
||
const pattern = /^(file:\/\/|http:\/\/|https:\/\/|base64:\/\/)/;
|
||
return pattern.test(uri);
|
||
}
|
||
|
||
for (let msg of sendMsgList) {
|
||
if (msg["type"] && msg["data"]) {
|
||
let type = msg["type"];
|
||
let data = msg["data"];
|
||
if (type === "text" && !data["text"]) {
|
||
return 400;
|
||
} else if (["image", "voice", "record"].includes(type)) {
|
||
if (!data["file"]) {
|
||
return 400;
|
||
} else {
|
||
if (checkUri(data["file"])) {
|
||
return 200;
|
||
} else {
|
||
return 400;
|
||
}
|
||
}
|
||
|
||
} else if (type === "at" && !data["qq"]) {
|
||
return 400;
|
||
} else if (type === "reply" && !data["id"]) {
|
||
return 400;
|
||
}
|
||
} else {
|
||
return 400
|
||
}
|
||
}
|
||
return 200;
|
||
}
|
||
|
||
export interface ReturnDataType {
|
||
message_id: number
|
||
}
|
||
|
||
export class SendMsg extends BaseAction<OB11PostSendMsg, ReturnDataType> {
|
||
actionName = ActionName.SendMsg
|
||
|
||
protected async check(payload: OB11PostSendMsg): Promise<BaseCheckResult> {
|
||
const messages = this.convertMessage2List(payload.message);
|
||
const fmNum = this.getSpecialMsgNum(payload, OB11MessageDataType.node)
|
||
if (fmNum && fmNum != messages.length) {
|
||
return {
|
||
valid: false,
|
||
message: "转发消息不能和普通消息混在一起发送,转发需要保证message只有type为node的元素"
|
||
}
|
||
}
|
||
if (payload.group_id && !(await getGroup(payload.group_id))) {
|
||
return {
|
||
valid: false,
|
||
message: `群${payload.group_id}不存在`
|
||
}
|
||
}
|
||
if (payload.user_id && payload.message_type !== "group") {
|
||
if (!(await getFriend(payload.user_id))) {
|
||
if (!ALLOW_SEND_TEMP_MSG) {
|
||
return {
|
||
valid: false,
|
||
message: `不能发送临时消息`
|
||
}
|
||
}
|
||
}
|
||
}
|
||
return {
|
||
valid: true,
|
||
}
|
||
}
|
||
|
||
protected async _handle(payload: OB11PostSendMsg) {
|
||
const peer: Peer = {
|
||
chatType: ChatType.friend,
|
||
peerUid: ""
|
||
}
|
||
let isTempMsg = false;
|
||
let group: Group | undefined = undefined;
|
||
const genGroupPeer = async () => {
|
||
group = await getGroup(payload.group_id.toString())
|
||
peer.chatType = ChatType.group
|
||
// peer.name = group.name
|
||
peer.peerUid = group.groupCode
|
||
}
|
||
|
||
const genFriendPeer = () => {
|
||
const friend = friends.find(f => f.uin == payload.user_id.toString())
|
||
if (friend) {
|
||
// peer.name = friend.nickName
|
||
peer.peerUid = friend.uid
|
||
} else {
|
||
peer.chatType = ChatType.temp
|
||
const tempUserUid = getUidByUin(payload.user_id.toString())
|
||
if (!tempUserUid) {
|
||
throw (`找不到私聊对象${payload.user_id}`)
|
||
}
|
||
// peer.name = tempUser.nickName
|
||
isTempMsg = true;
|
||
peer.peerUid = tempUserUid;
|
||
}
|
||
}
|
||
if (payload?.group_id && payload.message_type === "group") {
|
||
await genGroupPeer()
|
||
|
||
} else if (payload?.user_id) {
|
||
genFriendPeer()
|
||
} else if (payload.group_id) {
|
||
await genGroupPeer()
|
||
} else {
|
||
throw ("发送消息参数错误, 请指定group_id或user_id")
|
||
}
|
||
const messages = this.convertMessage2List(payload.message);
|
||
if (this.getSpecialMsgNum(payload, OB11MessageDataType.node)) {
|
||
try {
|
||
const returnMsg = await this.handleForwardNode(peer, messages as OB11MessageNode[], group)
|
||
return {message_id: returnMsg.msgShortId}
|
||
} catch (e) {
|
||
throw ("发送转发消息失败 " + e.toString())
|
||
}
|
||
} else {
|
||
if (this.getSpecialMsgNum(payload, OB11MessageDataType.music)) {
|
||
const music: OB11MessageCustomMusic = messages[0] as OB11MessageCustomMusic
|
||
if (music) {
|
||
const {url, audio, title, content, image} = music.data;
|
||
const selfPeer: Peer = {peerUid: selfInfo.uid, chatType: ChatType.friend}
|
||
// 搞不定！
|
||
// const musicMsg = await this.send(selfPeer, [this.genMusicElement(url, audio, title, content, image)], [], false)
|
||
// 转发
|
||
// const res = await NTQQApi.forwardMsg(selfPeer, peer, [musicMsg.msgId])
|
||
// log("转发音乐消息成功", res);
|
||
// return {message_id: musicMsg.msgShortId}
|
||
}
|
||
}
|
||
}
|
||
// log("send msg:", peer, sendElements)
|
||
const {sendElements, deleteAfterSentFiles} = await this.createSendElements(messages, group)
|
||
try {
|
||
const returnMsg = await this.send(peer, sendElements, deleteAfterSentFiles)
|
||
deleteAfterSentFiles.map(f => fs.unlink(f, () => {}));
|
||
return {message_id: returnMsg.msgShortId}
|
||
} catch (e) {
|
||
log("发送消息失败", e.stack.toString())
|
||
throw (e.toString())
|
||
}
|
||
}
|
||
|
||
protected convertMessage2List(message: OB11MessageMixType) {
|
||
if (typeof message === "string") {
|
||
// message = [{
|
||
// type: OB11MessageDataType.text,
|
||
// data: {
|
||
// text: message
|
||
// }
|
||
// }] as OB11MessageData[]
|
||
message = decodeCQCode(message.toString())
|
||
} else if (!Array.isArray(message)) {
|
||
message = [message]
|
||
}
|
||
return message;
|
||
}
|
||
|
||
private getSpecialMsgNum(payload: OB11PostSendMsg, msgType: OB11MessageDataType): number {
|
||
if (Array.isArray(payload.message)) {
|
||
return payload.message.filter(msg => msg.type == msgType).length
|
||
}
|
||
return 0
|
||
}
|
||
|
||
// 返回一个合并转发的消息id
|
||
private async handleForwardNode(destPeer: Peer, messageNodes: OB11MessageNode[], group: Group | undefined) {
|
||
const selfPeer: Peer = {
|
||
chatType: ChatType.friend,
|
||
peerUid: selfInfo.uid
|
||
}
|
||
let selfNodeMsgList: RawMessage[] = []; // 自己给自己发的消息
|
||
let originalNodeMsgList: RawMessage[] = [];
|
||
let sendForwardElements: SendMessageElement[] = []
|
||
for (const messageNode of messageNodes) {
|
||
// 一个node表示一个人的消息
|
||
let nodeId = messageNode.data.id;
|
||
// 有nodeId表示一个子转发消息卡片
|
||
if (nodeId) {
|
||
let nodeMsg = await dbUtil.getMsgByShortId(parseInt(nodeId));
|
||
if (nodeMsg) {
|
||
originalNodeMsgList.push(nodeMsg);
|
||
}
|
||
} else {
|
||
// 自定义的消息
|
||
// 提取消息段，发给自己生成消息id
|
||
try {
|
||
const {
|
||
sendElements,
|
||
deleteAfterSentFiles
|
||
} = await this.createSendElements(this.convertMessage2List(messageNode.data.content), group);
|
||
log("开始生成转发节点", sendElements);
|
||
sendForwardElements.push(...sendElements);
|
||
const nodeMsg = await this.send(selfPeer, sendElements, deleteAfterSentFiles, true);
|
||
selfNodeMsgList.push(nodeMsg);
|
||
log("转发节点生成成功", nodeMsg.msgId);
|
||
await sleep(500);
|
||
} catch (e) {
|
||
log("生效转发消息节点失败", e)
|
||
}
|
||
}
|
||
}
|
||
|
||
let nodeIds: string[] = []
|
||
// 检查是否需要克隆直接引用消息id的节点
|
||
let needSendSelf = false;
|
||
if (sendForwardElements.length) {
|
||
needSendSelf = true
|
||
} else {
|
||
needSendSelf = !originalNodeMsgList.every((msg, index) => msg.peerUid === originalNodeMsgList[0].peerUid && msg.recallTime.length < 2)
|
||
}
|
||
if (needSendSelf) {
|
||
nodeIds = selfNodeMsgList.map(msg => msg.msgId);
|
||
let sendElements: SendMessageElement[] = [];
|
||
for (const originalNodeMsg of originalNodeMsgList) {
|
||
if (originalNodeMsg.peerUid === selfInfo.uid && originalNodeMsg.recallTime.length < 2) {
|
||
nodeIds.push(originalNodeMsg.msgId)
|
||
} else { // 需要进行克隆
|
||
Object.keys(originalNodeMsg.elements).forEach((eleKey) => {
|
||
if (eleKey !== "elementId") {
|
||
sendForwardElements.push(originalNodeMsg.elements[eleKey])
|
||
sendElements.push(originalNodeMsg.elements[eleKey])
|
||
}
|
||
})
|
||
try {
|
||
const nodeMsg = await NTQQApi.sendMsg(selfPeer, sendElements, true);
|
||
nodeIds.push(nodeMsg.msgId)
|
||
log("克隆转发消息到节点")
|
||
} catch (e) {
|
||
log("克隆转发消息失败", e)
|
||
}
|
||
}
|
||
}
|
||
} else {
|
||
nodeIds = originalNodeMsgList.map(msg => msg.msgId)
|
||
}
|
||
|
||
let srcPeer = selfPeer;
|
||
if (!needSendSelf) {
|
||
srcPeer = {
|
||
chatType: originalNodeMsgList[0].chatType === ChatType.group ? ChatType.group : ChatType.friend,
|
||
peerUid: originalNodeMsgList[0].peerUid
|
||
}
|
||
}
|
||
// elements之间用换行符分隔
|
||
// let _sendForwardElements: SendMessageElement[] = []
|
||
// for(let i = 0; i < sendForwardElements.length; i++){
|
||
// _sendForwardElements.push(sendForwardElements[i])
|
||
// _sendForwardElements.push(SendMsgElementConstructor.text("\n\n"))
|
||
// }
|
||
// const nodeMsg = await NTQQApi.sendMsg(selfPeer, _sendForwardElements, true);
|
||
// nodeIds.push(nodeMsg.msgId)
|
||
// await sleep(500);
|
||
// 开发转发
|
||
try {
|
||
return await NTQQApi.multiForwardMsg(srcPeer, destPeer, nodeIds)
|
||
} catch (e) {
|
||
log("forward failed", e)
|
||
return null;
|
||
}
|
||
}
|
||
|
||
private async createSendElements(messageData: OB11MessageData[], group: Group | undefined, ignoreTypes: OB11MessageDataType[] = []) {
|
||
let sendElements: SendMessageElement[] = []
|
||
let deleteAfterSentFiles: string[] = []
|
||
for (let sendMsg of messageData) {
|
||
if (ignoreTypes.includes(sendMsg.type)) {
|
||
continue
|
||
}
|
||
switch (sendMsg.type) {
|
||
case OB11MessageDataType.text: {
|
||
const text = sendMsg.data?.text;
|
||
if (text) {
|
||
sendElements.push(SendMsgElementConstructor.text(sendMsg.data!.text))
|
||
}
|
||
}
|
||
break;
|
||
case OB11MessageDataType.at: {
|
||
if (!group){
|
||
continue
|
||
}
|
||
let atQQ = sendMsg.data?.qq;
|
||
if (atQQ) {
|
||
atQQ = atQQ.toString()
|
||
if (atQQ === "all") {
|
||
sendElements.push(SendMsgElementConstructor.at(atQQ, atQQ, AtType.atAll, "全体成员"))
|
||
} else {
|
||
// const atMember = group?.members.find(m => m.uin == atQQ)
|
||
const atMember = await getGroupMember(group?.groupCode, atQQ);
|
||
if (atMember) {
|
||
sendElements.push(SendMsgElementConstructor.at(atQQ, atMember.uid, AtType.atUser, atMember.cardName || atMember.nick))
|
||
}
|
||
}
|
||
}
|
||
}
|
||
break;
|
||
case OB11MessageDataType.reply: {
|
||
let replyMsgId = sendMsg.data.id;
|
||
if (replyMsgId) {
|
||
const replyMsg = await dbUtil.getMsgByShortId(parseInt(replyMsgId))
|
||
if (replyMsg) {
|
||
sendElements.push(SendMsgElementConstructor.reply(replyMsg.msgSeq, replyMsg.msgId, replyMsg.senderUin, replyMsg.senderUin))
|
||
}
|
||
}
|
||
}
|
||
break;
|
||
case OB11MessageDataType.face: {
|
||
const faceId = sendMsg.data?.id
|
||
if (faceId) {
|
||
sendElements.push(SendMsgElementConstructor.face(parseInt(faceId)))
|
||
}
|
||
}
|
||
break;
|
||
|
||
case OB11MessageDataType.image:
|
||
case OB11MessageDataType.file:
|
||
case OB11MessageDataType.video:
|
||
case OB11MessageDataType.voice: {
|
||
let file = sendMsg.data?.file
|
||
const payloadFileName = sendMsg.data?.name
|
||
if (file) {
|
||
const cache = await dbUtil.getFileCache(file)
|
||
if (cache){
|
||
if (fs.existsSync(cache.filePath)){
|
||
file = "file://" + cache.filePath
|
||
}
|
||
else if (cache.downloadFunc){
|
||
await cache.downloadFunc()
|
||
file = cache.filePath;
|
||
log("找到文件缓存", file);
|
||
}
|
||
}
|
||
const {path, isLocal, fileName, errMsg} = (await uri2local(file))
|
||
if (errMsg){
|
||
throw errMsg
|
||
}
|
||
if (path) {
|
||
if (!isLocal) { // 只删除http和base64转过来的文件
|
||
deleteAfterSentFiles.push(path)
|
||
}
|
||
const constructorMap = {
|
||
[OB11MessageDataType.image]: SendMsgElementConstructor.pic,
|
||
[OB11MessageDataType.voice]: SendMsgElementConstructor.ptt,
|
||
[OB11MessageDataType.video]: SendMsgElementConstructor.video,
|
||
[OB11MessageDataType.file]: SendMsgElementConstructor.file,
|
||
}
|
||
if (sendMsg.type === OB11MessageDataType.file) {
|
||
log("发送文件", path, payloadFileName || fileName)
|
||
sendElements.push(await SendMsgElementConstructor.file(path, payloadFileName || fileName));
|
||
} else if (sendMsg.type === OB11MessageDataType.video) {
|
||
log("发送视频", path, payloadFileName || fileName)
|
||
sendElements.push(await SendMsgElementConstructor.video(path, payloadFileName || fileName));
|
||
} else {
|
||
sendElements.push(await constructorMap[sendMsg.type](path));
|
||
}
|
||
}
|
||
}
|
||
}
|
||
break;
|
||
}
|
||
|
||
}
|
||
|
||
return {
|
||
sendElements,
|
||
deleteAfterSentFiles
|
||
}
|
||
}
|
||
|
||
private async send(peer: Peer, sendElements: SendMessageElement[], deleteAfterSentFiles: string[], waitComplete = true) {
|
||
if (!sendElements.length) {
|
||
throw ("消息体无法解析")
|
||
}
|
||
const returnMsg = await NTQQApi.sendMsg(peer, sendElements, waitComplete, 20000);
|
||
log("消息发送结果", returnMsg)
|
||
returnMsg.msgShortId = await dbUtil.addMsg(returnMsg)
|
||
deleteAfterSentFiles.map(f => fs.unlink(f, () => {
|
||
}))
|
||
return returnMsg
|
||
}
|
||
|
||
private genMusicElement(url: string, audio: string, title: string, content: string, image: string): SendArkElement {
|
||
const musicJson = {
|
||
app: 'com.tencent.structmsg',
|
||
config: {
|
||
ctime: 1709689928,
|
||
forward: 1,
|
||
token: '5c1e4905f926dd3a64a4bd3841460351',
|
||
type: 'normal'
|
||
},
|
||
extra: {app_type: 1, appid: 100497308, uin: selfInfo.uin},
|
||
meta: {
|
||
news: {
|
||
action: '',
|
||
android_pkg_name: '',
|
||
app_type: 1,
|
||
appid: 100497308,
|
||
ctime: 1709689928,
|
||
desc: content || title,
|
||
jumpUrl: url,
|
||
musicUrl: audio,
|
||
preview: image,
|
||
source_icon: 'https://p.qpic.cn/qqconnect/0/app_100497308_1626060999/100?max-age=2592000&t=0',
|
||
source_url: '',
|
||
tag: 'QQ音乐',
|
||
title: title,
|
||
uin: selfInfo.uin,
|
||
}
|
||
},
|
||
prompt: content || title,
|
||
ver: '0.0.0.1',
|
||
view: 'news'
|
||
}
|
||
|
||
return SendMsgElementConstructor.ark(musicJson)
|
||
}
|
||
}
|
||
|
||
export default SendMsg |