123 lines
3.1 KiB
TypeScript
123 lines
3.1 KiB
TypeScript
import db from "../../db";
|
||
import { sendMsg } from "../../utils/sendMsg";
|
||
|
||
interface BaseMsg {
|
||
msg_type: MsgType;
|
||
content: string;
|
||
}
|
||
|
||
interface GroupMsg extends BaseMsg {
|
||
group_id: string;
|
||
}
|
||
|
||
interface NormalMsg extends BaseMsg {
|
||
receive_id: string;
|
||
receive_id_type: ReceiveIDType;
|
||
}
|
||
|
||
type MessageReqJson = GroupMsg & NormalMsg;
|
||
|
||
const validateMessageReq = (body: MessageReqJson) => {
|
||
if (!body.group_id && !body.receive_id) {
|
||
return new Response("group_id or receive_id is required");
|
||
}
|
||
if (body.receive_id && !body.receive_id_type) {
|
||
return new Response("receive_id_type is required");
|
||
}
|
||
if (!body.msg_type) {
|
||
return new Response("msg_type is required");
|
||
}
|
||
if (!body.content) {
|
||
return new Response("content is required");
|
||
}
|
||
return false;
|
||
};
|
||
|
||
export const manageMessageReq = async (req: Request) => {
|
||
const body = (await req.json()) as MessageReqJson;
|
||
// 校验参数
|
||
const validateRes = validateMessageReq(body);
|
||
if (validateRes) return validateRes;
|
||
|
||
// 遍历所有id发送消息，保存所有对应的messageId
|
||
const sendRes = {
|
||
chat_id: {} as Record<string, any>,
|
||
open_id: {} as Record<string, any>,
|
||
union_id: {} as Record<string, any>,
|
||
user_id: {} as Record<string, any>,
|
||
email: {} as Record<string, any>,
|
||
};
|
||
|
||
// 发送消息列表
|
||
const sendList = [] as Promise<any>[];
|
||
|
||
// 处理消息内容
|
||
const finalContent =
|
||
typeof body.content !== "string"
|
||
? JSON.stringify(body.content)
|
||
: body.content;
|
||
|
||
if (body.group_id) {
|
||
// 获取所有接收者
|
||
const group = (await db.messageGroup.getOne(
|
||
body.group_id!
|
||
)) as PBMessageGroup;
|
||
if (!group) {
|
||
return new Response("group not found");
|
||
}
|
||
|
||
const { chat_id, open_id, union_id, user_id, email } = group;
|
||
|
||
// 构造发送消息函数
|
||
const makeSendFunc = (receive_id_type: ReceiveIDType) => {
|
||
return (receive_id: string) => {
|
||
sendList.push(
|
||
sendMsg(
|
||
receive_id_type,
|
||
receive_id,
|
||
body.msg_type,
|
||
finalContent
|
||
).then((res) => {
|
||
sendRes[receive_id_type][receive_id] = res;
|
||
})
|
||
);
|
||
};
|
||
};
|
||
|
||
// 创建消息列表
|
||
if (chat_id) chat_id.map(makeSendFunc("chat_id"));
|
||
if (open_id) open_id.map(makeSendFunc("open_id"));
|
||
if (union_id) union_id.map(makeSendFunc("union_id"));
|
||
if (user_id) user_id.map(makeSendFunc("user_id"));
|
||
if (email) email.map(makeSendFunc("email"));
|
||
}
|
||
|
||
if (body.receive_id && body.receive_id_type) {
|
||
sendList.push(
|
||
sendMsg(
|
||
body.receive_id_type,
|
||
body.receive_id,
|
||
body.msg_type,
|
||
finalContent
|
||
).then((res) => {
|
||
sendRes[body.receive_id_type][body.receive_id] = res;
|
||
})
|
||
);
|
||
}
|
||
|
||
try {
|
||
await Promise.all(sendList);
|
||
return Response.json({
|
||
code: 200,
|
||
msg: "ok",
|
||
data: sendRes,
|
||
});
|
||
} catch {
|
||
return Response.json({
|
||
code: 400,
|
||
msg: "send msg failed",
|
||
data: sendRes,
|
||
});
|
||
}
|
||
};
|