feat(server): first version
This commit is contained in:
Generated
+3348
File diff suppressed because it is too large
Load Diff
@@ -0,0 +1,25 @@
|
|||||||
|
{
|
||||||
|
"name": "bot-server",
|
||||||
|
"version": "1.0.0",
|
||||||
|
"description": "",
|
||||||
|
"main": "src/index.js",
|
||||||
|
"scripts": {
|
||||||
|
"start": "node .",
|
||||||
|
"test": "echo \"Error: no test specified\" && exit 1"
|
||||||
|
},
|
||||||
|
"author": "",
|
||||||
|
"license": "ISC",
|
||||||
|
"dependencies": {
|
||||||
|
"@journeyapps/sqlcipher": "^5.2.0",
|
||||||
|
"@koa/cors": "^3.1.0",
|
||||||
|
"@koa/router": "^10.1.1",
|
||||||
|
"commander": "^8.3.0",
|
||||||
|
"dayjs": "^1.10.7",
|
||||||
|
"inquirer": "^8.2.0",
|
||||||
|
"koa": "^2.13.4",
|
||||||
|
"qrcode-terminal": "^0.12.0",
|
||||||
|
"socket.io": "^4.3.1",
|
||||||
|
"sqlite": "^4.0.23",
|
||||||
|
"wechat4u": "^0.7.7"
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,95 @@
|
|||||||
|
const fs = require("fs");
|
||||||
|
const path = require("path");
|
||||||
|
const WeChat = require("wechat4u");
|
||||||
|
const { config, globalData } = require("./config.js");
|
||||||
|
const { ensureAvatarDir, composeAvatarFileName } = require("./util");
|
||||||
|
|
||||||
|
let sessionData;
|
||||||
|
try {
|
||||||
|
sessionData = JSON.parse(
|
||||||
|
fs.readFileSync(path.join(config.dataDir, "session.json"), {
|
||||||
|
encoding: "utf-8",
|
||||||
|
})
|
||||||
|
);
|
||||||
|
globalData.hasSessionData = true;
|
||||||
|
} catch {}
|
||||||
|
const bot = new WeChat(sessionData);
|
||||||
|
|
||||||
|
// 标志正在缓存头像
|
||||||
|
let cacheAvatarWorkingFlag = 0;
|
||||||
|
|
||||||
|
/** 判断是否群聊消息 */
|
||||||
|
function isGroupMessage(msg) {
|
||||||
|
return msg.getPeerUserName().startsWith("@@");
|
||||||
|
}
|
||||||
|
|
||||||
|
/** 获取联系人显示名称，备注优先 */
|
||||||
|
function getDisplayName(usernameOrContact) {
|
||||||
|
if (!usernameOrContact) {
|
||||||
|
return null;
|
||||||
|
}
|
||||||
|
let contact = usernameOrContact;
|
||||||
|
if (typeof usernameOrContact === "string") {
|
||||||
|
contact = bot.contacts[usernameOrContact];
|
||||||
|
}
|
||||||
|
return (
|
||||||
|
contact.RemarkName ||
|
||||||
|
contact.DisplayName ||
|
||||||
|
contact.NickName ||
|
||||||
|
contact.UserName
|
||||||
|
);
|
||||||
|
}
|
||||||
|
/** 获取用户昵称 */
|
||||||
|
function getNickName(usernameOrContact) {
|
||||||
|
let contact = usernameOrContact;
|
||||||
|
if (typeof usernameOrContact === "string") {
|
||||||
|
contact = bot.contacts[usernameOrContact];
|
||||||
|
}
|
||||||
|
return contact.NickName || contact.UserName;
|
||||||
|
}
|
||||||
|
|
||||||
|
/** 获取群聊消息的发言人 */
|
||||||
|
function getGroupMsgSpeaker(msg) {
|
||||||
|
if (!isGroupMessage(msg) || msg.isSendBySelf) {
|
||||||
|
return null;
|
||||||
|
}
|
||||||
|
const speakerID = msg["OriginalContent"].split(":", 1)[0];
|
||||||
|
return bot.contacts[speakerID];
|
||||||
|
}
|
||||||
|
|
||||||
|
/** 获取群聊消息发送的真实内容 */
|
||||||
|
function getGroupMsgContent(msg) {
|
||||||
|
const content = msg["Content"];
|
||||||
|
if (!isGroupMessage(msg) || msg.isSendBySelf) {
|
||||||
|
return content;
|
||||||
|
}
|
||||||
|
return content.replace(/^.*?:\n/, "");
|
||||||
|
}
|
||||||
|
|
||||||
|
/** 缓存用户头像 */
|
||||||
|
async function cacheAvatars() {
|
||||||
|
const flag = Date.now() + Math.random();
|
||||||
|
cacheAvatarWorkingFlag = flag;
|
||||||
|
const dir = ensureAvatarDir();
|
||||||
|
for (const uid of Object.keys(bot.contacts)) {
|
||||||
|
if (cacheAvatarWorkingFlag !== flag) {
|
||||||
|
// 联系人再次更新了，结束当前的工作，让新的调用来做缓存工作
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
const contact = bot.contacts[uid];
|
||||||
|
const filename = path.join(dir, composeAvatarFileName(contact));
|
||||||
|
if (!fs.existsSync(filename)) {
|
||||||
|
const { data } = await bot.getHeadImg(contact.HeadImgUrl);
|
||||||
|
fs.writeFileSync(filename, data);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
module.exports = {
|
||||||
|
bot,
|
||||||
|
getNickName,
|
||||||
|
getDisplayName,
|
||||||
|
getGroupMsgSpeaker,
|
||||||
|
getGroupMsgContent,
|
||||||
|
cacheAvatars,
|
||||||
|
};
|
||||||
@@ -0,0 +1,47 @@
|
|||||||
|
const fs = require("fs");
|
||||||
|
const path = require("path");
|
||||||
|
|
||||||
|
const config = {
|
||||||
|
dbFile: "./wxbot.db",
|
||||||
|
dataDir: "./wxbot-data",
|
||||||
|
mediaDir: "./wxbot-data/media",
|
||||||
|
cacheDir: "./wxbot-data/cache",
|
||||||
|
serve: true,
|
||||||
|
httpHost: "127.0.0.1",
|
||||||
|
httpPort: 9200,
|
||||||
|
baseUrl: "http:127.0.0.1",
|
||||||
|
debugMode: false,
|
||||||
|
appCode: {},
|
||||||
|
appAuthHash: {},
|
||||||
|
};
|
||||||
|
const globalData = {
|
||||||
|
// 是否有旧的登录数据可重用
|
||||||
|
hasSessionData: false,
|
||||||
|
};
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 加载配置文件
|
||||||
|
* @param file {string}
|
||||||
|
*/
|
||||||
|
function loadConfig(file) {
|
||||||
|
try {
|
||||||
|
const data = JSON.parse(fs.readFileSync(file, { encoding: "utf-8" }));
|
||||||
|
Object.assign(config, data);
|
||||||
|
if (!data.mediaDir) {
|
||||||
|
config.mediaDir = path.join(config.dataDir, "media");
|
||||||
|
}
|
||||||
|
if (!data.cacheDir) {
|
||||||
|
config.cacheDir = path.join(config.dataDir, "cache");
|
||||||
|
}
|
||||||
|
fs.mkdirSync(config.mediaDir, { recursive: true });
|
||||||
|
fs.mkdirSync(config.cacheDir, { recursive: true });
|
||||||
|
} catch (e) {
|
||||||
|
throw Error("加载配置文件失败：" + e.message);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
module.exports = {
|
||||||
|
globalData,
|
||||||
|
config,
|
||||||
|
loadConfig,
|
||||||
|
};
|
||||||
@@ -0,0 +1,88 @@
|
|||||||
|
const inquirer = require("inquirer");
|
||||||
|
const sqlite3 = require("@journeyapps/sqlcipher");
|
||||||
|
const sqlite = require("sqlite");
|
||||||
|
|
||||||
|
let pushSQL =
|
||||||
|
"INSERT INTO msg (\
|
||||||
|
msg_id,\
|
||||||
|
create_time,\
|
||||||
|
msg_type,\
|
||||||
|
composed_type,\
|
||||||
|
content,\
|
||||||
|
from_name,\
|
||||||
|
from_nickname,\
|
||||||
|
to_name,\
|
||||||
|
to_nickname,\
|
||||||
|
is_group_message,\
|
||||||
|
speaking_group_member_name,\
|
||||||
|
filename,\
|
||||||
|
display_filename\
|
||||||
|
) VALUES (\
|
||||||
|
$messageID,\
|
||||||
|
$createTimestamp,\
|
||||||
|
$messageType,\
|
||||||
|
$type,\
|
||||||
|
$content,\
|
||||||
|
$fromName,\
|
||||||
|
$fromNickName,\
|
||||||
|
$toName,\
|
||||||
|
$toNickName,\
|
||||||
|
$isGroupMessage,\
|
||||||
|
$speakingGroupMemberName,\
|
||||||
|
$filename,\
|
||||||
|
$displayFilename\
|
||||||
|
)";
|
||||||
|
|
||||||
|
let db, pushStatement;
|
||||||
|
async function initDB(dbFile) {
|
||||||
|
db = await sqlite.open({
|
||||||
|
filename: dbFile,
|
||||||
|
driver: sqlite3.Database,
|
||||||
|
});
|
||||||
|
while (true) {
|
||||||
|
try {
|
||||||
|
const { password } = await inquirer.prompt([
|
||||||
|
{ type: "password", name: "password", message: "密码:", prefix: "" },
|
||||||
|
]);
|
||||||
|
password && (await db.exec(`PRAGMA KEY=${password}`));
|
||||||
|
await db.exec(
|
||||||
|
`CREATE TABLE IF NOT EXISTS msg (
|
||||||
|
msg_id TEXT NOT NULL,
|
||||||
|
create_time INTEGER NOT NULL,
|
||||||
|
msg_type INTEGER NOT NULL,
|
||||||
|
composed_type TEXT NOT NULL,
|
||||||
|
content TEXT,
|
||||||
|
from_name TEXT,
|
||||||
|
from_nickname TEXT,
|
||||||
|
to_name TEXT,
|
||||||
|
to_nickname TEXT,
|
||||||
|
is_group_message INTEGER,
|
||||||
|
speaking_group_member_name TEXT,
|
||||||
|
filename TEXT,
|
||||||
|
display_filename TEXT)`
|
||||||
|
);
|
||||||
|
break;
|
||||||
|
} catch (e) {
|
||||||
|
if (e.name === "SQLITE_NOTADB") {
|
||||||
|
console.warn("密码错误");
|
||||||
|
} else {
|
||||||
|
throw e;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
pushStatement = await db.prepare(pushSQL);
|
||||||
|
}
|
||||||
|
|
||||||
|
/** 将消息存储到数据库 */
|
||||||
|
async function pushMessage(message) {
|
||||||
|
const m = {};
|
||||||
|
for (const key of Object.keys(message)) {
|
||||||
|
m["$" + key] = message[key];
|
||||||
|
}
|
||||||
|
await pushStatement.run(m);
|
||||||
|
}
|
||||||
|
|
||||||
|
module.exports = {
|
||||||
|
initDB,
|
||||||
|
pushMessage,
|
||||||
|
};
|
||||||
@@ -0,0 +1,63 @@
|
|||||||
|
const fs = require("fs");
|
||||||
|
const path = require("path");
|
||||||
|
const qrCode = require("qrcode-terminal");
|
||||||
|
const { program } = require("commander");
|
||||||
|
const { config, loadConfig, globalData } = require("./config.js");
|
||||||
|
|
||||||
|
// 在引入bot前加载配置，因为bot依赖config数据
|
||||||
|
program.requiredOption("-c, --config-file <string>", "配置文件");
|
||||||
|
const { configFile } = program.parse().opts();
|
||||||
|
try {
|
||||||
|
loadConfig(configFile);
|
||||||
|
} catch {
|
||||||
|
console.error("加载配置文件失败");
|
||||||
|
process.exit(1);
|
||||||
|
}
|
||||||
|
|
||||||
|
const { bot, cacheAvatars } = require("./bot.js");
|
||||||
|
const { processMessage } = require("./message.js");
|
||||||
|
const { initDB } = require("./db.js");
|
||||||
|
require("./socket.js");
|
||||||
|
|
||||||
|
async function start() {
|
||||||
|
await initDB(config.dbFile);
|
||||||
|
bot.on("uuid", (uuid) => {
|
||||||
|
console.log("登录二维码：", "https://login.weixin.qq.com/qrcode/" + uuid);
|
||||||
|
qrCode.generate("https://login.weixin.qq.com/l/" + uuid, {
|
||||||
|
small: true,
|
||||||
|
});
|
||||||
|
});
|
||||||
|
bot.on("login", () => {
|
||||||
|
console.log("登录成功");
|
||||||
|
fs.writeFileSync(
|
||||||
|
path.join(config.dataDir, "session.json"),
|
||||||
|
JSON.stringify(bot.botData)
|
||||||
|
);
|
||||||
|
});
|
||||||
|
bot.on("contacts-updated", () => {
|
||||||
|
cacheAvatars();
|
||||||
|
});
|
||||||
|
bot.on("message", (msg) => {
|
||||||
|
processMessage(msg).catch((e) => bot.emit("error", e));
|
||||||
|
});
|
||||||
|
bot.on("error", console.error);
|
||||||
|
try {
|
||||||
|
if (globalData.hasSessionData) {
|
||||||
|
await bot
|
||||||
|
.restart()
|
||||||
|
.then(() => {
|
||||||
|
return bot.sendText("[server restarted from session]", "filehelper");
|
||||||
|
})
|
||||||
|
.catch(async () => {
|
||||||
|
await bot.stop();
|
||||||
|
await bot.start();
|
||||||
|
});
|
||||||
|
} else {
|
||||||
|
await bot.start();
|
||||||
|
}
|
||||||
|
} catch (e) {
|
||||||
|
console.error("启动失败", e);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
start();
|
||||||
@@ -0,0 +1,209 @@
|
|||||||
|
const path = require("path");
|
||||||
|
const fs = require("fs");
|
||||||
|
const socket = require("./socket.js");
|
||||||
|
const db = require("./db.js");
|
||||||
|
const {
|
||||||
|
bot,
|
||||||
|
getDisplayName,
|
||||||
|
getGroupMsgSpeaker,
|
||||||
|
getNickName,
|
||||||
|
} = require("./bot.js");
|
||||||
|
const {
|
||||||
|
composeMessageID,
|
||||||
|
composeMediaFilePath,
|
||||||
|
composeFilename,
|
||||||
|
composeCacheFilePath,
|
||||||
|
fileMD5,
|
||||||
|
ensureMediaDir,
|
||||||
|
} = require("./util");
|
||||||
|
|
||||||
|
async function processMessage(msg) {
|
||||||
|
if ([50, 51, 52, 53, 9999].includes(msg.MsgType)) {
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
const peerContact = bot.contacts[msg.getPeerUserName()];
|
||||||
|
const fromContact = bot.contacts[msg["FromUserName"]];
|
||||||
|
const toContact = bot.contacts[msg["ToUserName"]];
|
||||||
|
if (!peerContact || !fromContact || !toContact) {
|
||||||
|
// TODO: notify self
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
if (
|
||||||
|
bot.Contact.isPublicContact(peerContact) ||
|
||||||
|
(bot.Contact.isSpContact(peerContact) &&
|
||||||
|
peerContact.UserName !== "filehelper")
|
||||||
|
) {
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
ensureMediaDir(msg);
|
||||||
|
|
||||||
|
const msgID = msg["MsgId"];
|
||||||
|
const msgType = msg["MsgType"];
|
||||||
|
|
||||||
|
const m = {
|
||||||
|
messageID: msgID,
|
||||||
|
composedID: composeMessageID(msg),
|
||||||
|
createTimestamp: msg["CreateTime"],
|
||||||
|
messageType: msgType,
|
||||||
|
fromID: msg["FromUserName"],
|
||||||
|
toID: msg["ToUserName"],
|
||||||
|
fromName: getDisplayName(fromContact),
|
||||||
|
fromNickName: getNickName(fromContact),
|
||||||
|
toName: getDisplayName(toContact),
|
||||||
|
toNickName: getNickName(toContact),
|
||||||
|
type: "unknown",
|
||||||
|
content: null,
|
||||||
|
group: bot.Contact.isRoomContact(peerContact) ? peerContact : null,
|
||||||
|
speakingGroupMember: null,
|
||||||
|
filename: null,
|
||||||
|
displayFilename: msg["FileName"] || "",
|
||||||
|
isGroupMessage: bot.Contact.isRoomContact(peerContact),
|
||||||
|
};
|
||||||
|
|
||||||
|
if (m.isGroupMessage) {
|
||||||
|
if (!msg.isSendBySelf) {
|
||||||
|
m.speakingGroupMember = getGroupMsgSpeaker(msg);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
switch (msgType) {
|
||||||
|
case bot.CONF.MSGTYPE_TEXT:
|
||||||
|
// 文本消息
|
||||||
|
m.type = "text";
|
||||||
|
m.content = msg["Content"];
|
||||||
|
break;
|
||||||
|
case bot.CONF.MSGTYPE_EMOTICON:
|
||||||
|
case bot.CONF.MSGTYPE_IMAGE: {
|
||||||
|
const suffix = msgType === bot.CONF.MSGTYPE_IMAGE ? ".jpg" : ".gif";
|
||||||
|
m.type = "img";
|
||||||
|
m.content =
|
||||||
|
`【${msgType === bot.CONF.MSGTYPE_IMAGE ? "图片" : "表情"}】` +
|
||||||
|
(msg["FileName"] || "");
|
||||||
|
if (msg["HasProductId"]) {
|
||||||
|
m.type = "unknown";
|
||||||
|
m.content = "【表情贴图】请在手机上查看";
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
const cachePath = composeCacheFilePath(msg, suffix);
|
||||||
|
// ENHANCE: 提前判断表情是否存在，节约带宽
|
||||||
|
const res = await bot.getMsgImg(msg.MsgId);
|
||||||
|
fs.writeFileSync(cachePath, res.data);
|
||||||
|
const md5 = await fileMD5(cachePath);
|
||||||
|
m.filename = md5 + suffix;
|
||||||
|
const filename = composeMediaFilePath(msg, m.filename);
|
||||||
|
if (fs.existsSync(filename)) {
|
||||||
|
fs.rmSync(cachePath);
|
||||||
|
} else {
|
||||||
|
fs.renameSync(cachePath, filename);
|
||||||
|
}
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
case bot.CONF.MSGTYPE_VOICE: {
|
||||||
|
m.type = "voice";
|
||||||
|
m.content = "【语音消息】";
|
||||||
|
m.filename = composeFilename(msg, ".mp3");
|
||||||
|
const { data } = await bot.getVoice(msg.MsgId);
|
||||||
|
fs.writeFileSync(composeMediaFilePath(msg, m.filename), data);
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
case bot.CONF.MSGTYPE_VIDEO:
|
||||||
|
case bot.CONF.MSGTYPE_MICROVIDEO: {
|
||||||
|
m.type = "video";
|
||||||
|
const suffix = ".mp4";
|
||||||
|
const cachePath = composeCacheFilePath(msg, suffix);
|
||||||
|
const res = await bot.getVideo(msg.MsgId);
|
||||||
|
fs.writeFileSync(cachePath, res.data);
|
||||||
|
const md5 = await fileMD5(cachePath);
|
||||||
|
m.filename = md5 + suffix;
|
||||||
|
const filename = composeMediaFilePath(msg, m.filename);
|
||||||
|
if (fs.existsSync(filename)) {
|
||||||
|
fs.rmSync(cachePath);
|
||||||
|
} else {
|
||||||
|
fs.renameSync(cachePath, filename);
|
||||||
|
}
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
case bot.CONF.MSGTYPE_APP:
|
||||||
|
m.type = "file";
|
||||||
|
if (msg.AppMsgType === 5) {
|
||||||
|
m.type = "text";
|
||||||
|
m.content = "【链接】" + msg["FileName"] + ": " + msg["Url"];
|
||||||
|
break;
|
||||||
|
} else if (msg.AppMsgType == 6) {
|
||||||
|
if (!(msg.FileSize < 10 * 1024 * 1024)) {
|
||||||
|
// 忽略10M以上的大文件
|
||||||
|
m.type = "unknown";
|
||||||
|
m.content = "【文件】" + m.displayFilename;
|
||||||
|
m.filename = null;
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
const suffix = path.extname(m.displayFilename);
|
||||||
|
const cachePath = composeCacheFilePath(msg, suffix);
|
||||||
|
const res = await bot.getDoc(
|
||||||
|
msg.FromUserName,
|
||||||
|
msg.MediaId,
|
||||||
|
msg.FileName
|
||||||
|
);
|
||||||
|
fs.writeFileSync(cachePath, res.data);
|
||||||
|
const md5 = await fileMD5(cachePath);
|
||||||
|
m.filename = md5 + suffix;
|
||||||
|
const filename = composeMediaFilePath(msg, m.filename);
|
||||||
|
if (fs.existsSync(filename)) {
|
||||||
|
fs.rmSync(cachePath);
|
||||||
|
} else {
|
||||||
|
fs.renameSync(cachePath, filename);
|
||||||
|
}
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
break;
|
||||||
|
default:
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
if (m.type === "unknown") {
|
||||||
|
m.content ||= "未知类型的消息，请在手机上查看";
|
||||||
|
}
|
||||||
|
|
||||||
|
const dbMsg = {
|
||||||
|
messageID: m.composedID,
|
||||||
|
createTimestamp: m.createTimestamp,
|
||||||
|
type: m.type,
|
||||||
|
messageType: m.messageType,
|
||||||
|
fromName: m.fromName,
|
||||||
|
fromNickName: m.fromNickName,
|
||||||
|
toName: m.toName,
|
||||||
|
toNickName: m.toNickName,
|
||||||
|
isGroupMessage: m.isGroupMessage,
|
||||||
|
speakingGroupMemberName:
|
||||||
|
(m.speakingGroupMember && getDisplayName(m.speakingGroupMember)) || null,
|
||||||
|
content: m.content,
|
||||||
|
filename: m.filename,
|
||||||
|
displayFilename: m.displayFilename,
|
||||||
|
};
|
||||||
|
const clientMsg = {
|
||||||
|
msgID: m.messageID,
|
||||||
|
composedID: m.composedID,
|
||||||
|
createTimestamp: m.createTimestamp,
|
||||||
|
type: m.type,
|
||||||
|
fromID: m.fromID,
|
||||||
|
fromName: m.fromName,
|
||||||
|
fromNickName: m.fromNickName,
|
||||||
|
toID: m.toID,
|
||||||
|
toName: m.toName,
|
||||||
|
toNickName: m.toNickName,
|
||||||
|
isGroupMessage: m.isGroupMessage,
|
||||||
|
speakingGroupMemberID:
|
||||||
|
m.speakingGroupMember && m.speakingGroupMember.UserName,
|
||||||
|
speakingGroupMemberName:
|
||||||
|
(m.speakingGroupMember && getDisplayName(m.speakingGroupMember)) || null,
|
||||||
|
content: m.content,
|
||||||
|
filename: m.filename,
|
||||||
|
displayFilename: m.displayFilename,
|
||||||
|
};
|
||||||
|
await db.pushMessage(dbMsg);
|
||||||
|
socket.pushMessage(clientMsg);
|
||||||
|
}
|
||||||
|
|
||||||
|
module.exports = {
|
||||||
|
processMessage,
|
||||||
|
};
|
||||||
@@ -0,0 +1,15 @@
|
|||||||
|
{
|
||||||
|
"dbFile": "/home/master/desktop/wxbot.db",
|
||||||
|
"dataDir": "/home/master/desktop",
|
||||||
|
"serve": true,
|
||||||
|
"httpHost": "127.0.0.1",
|
||||||
|
"httpPort": 9200,
|
||||||
|
"debugMode": false,
|
||||||
|
"baseUrl": "http:127.0.0.1",
|
||||||
|
"appCode": {
|
||||||
|
"mattuy-laptop": "iasdjfkwe3ds-qw3rjkwjfda8"
|
||||||
|
},
|
||||||
|
"appAuthHash": {
|
||||||
|
"mattuy-laptop": "b9c901e0ee98486361b2e5461442af3f3863961f"
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,250 @@
|
|||||||
|
const Koa = require("koa");
|
||||||
|
const fs = require("fs");
|
||||||
|
const path = require("path");
|
||||||
|
const { promisify } = require("util");
|
||||||
|
const Router = require("@koa/router");
|
||||||
|
const cors = require("@koa/cors");
|
||||||
|
const dayjs = require("dayjs");
|
||||||
|
const { Server } = require("socket.io");
|
||||||
|
const { createServer } = require("http");
|
||||||
|
const { bot, getDisplayName } = require("./bot.js");
|
||||||
|
const { config } = require("./config.js");
|
||||||
|
const {
|
||||||
|
textSHA1,
|
||||||
|
ensureAvatarDir,
|
||||||
|
ensureUploadDir,
|
||||||
|
composeAvatarFileName,
|
||||||
|
} = require("./util.js");
|
||||||
|
|
||||||
|
const DEFAULT_ROOM = "default_room";
|
||||||
|
|
||||||
|
const app = new Koa();
|
||||||
|
const router = new Router();
|
||||||
|
const httpServer = createServer(app.callback());
|
||||||
|
const io = new Server(httpServer, {
|
||||||
|
cors: true,
|
||||||
|
path: "/socket.io",
|
||||||
|
});
|
||||||
|
const activeSessions = new Set();
|
||||||
|
|
||||||
|
io.use((socket, next) => {
|
||||||
|
const auth = socket.handshake.auth;
|
||||||
|
if (!auth) {
|
||||||
|
next(Error("认证失败"));
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
const appName = auth.appName;
|
||||||
|
const appCode = config.appCode[appName];
|
||||||
|
if (!appCode || appCode !== auth.appCode) {
|
||||||
|
next(Error("认证失败"));
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
if (textSHA1(auth.passphrase) !== config.appAuthHash[appName]) {
|
||||||
|
next(Error("口令错误"));
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
if (bot?.state !== bot.CONF.STATE.login) {
|
||||||
|
next(Error("服务不可用"));
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
next().then(() => {
|
||||||
|
bot.sendText(
|
||||||
|
`【上线通知】
|
||||||
|
DATETIME: ${dayjs().format("YYYY-MM-DDTHH:MM:ss")}
|
||||||
|
APP_NAME: ${appName}
|
||||||
|
PLATFORM: ${auth.platform || ""}`,
|
||||||
|
"filehelper"
|
||||||
|
);
|
||||||
|
});
|
||||||
|
});
|
||||||
|
|
||||||
|
function deactivateSession() {
|
||||||
|
activeSessions.delete(socket.id);
|
||||||
|
}
|
||||||
|
|
||||||
|
async function pushMessage(message) {
|
||||||
|
if (message.filename) {
|
||||||
|
for (const sid of await io.allSockets()) {
|
||||||
|
const msg = {
|
||||||
|
...message,
|
||||||
|
fileUrl: `${config.baseUrl}/session/${sid}/media/${message.composedID}/${message.filename}`,
|
||||||
|
};
|
||||||
|
io.to(sid).emit("message", msg);
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
io.in(DEFAULT_ROOM).emit("message", message);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
function login(sessionID) {
|
||||||
|
const u = bot.user;
|
||||||
|
return {
|
||||||
|
id: u.UserName,
|
||||||
|
nickname: u.NickName,
|
||||||
|
displayName: u.NickName,
|
||||||
|
signature: u.Signature,
|
||||||
|
avatar: `${config.baseUrl}/session/${sessionID}/avatar/${u.UserName}`,
|
||||||
|
};
|
||||||
|
}
|
||||||
|
function getContacts(sessionID) {
|
||||||
|
function wrapContact(contact) {
|
||||||
|
return {
|
||||||
|
id: contact.UserName,
|
||||||
|
nickname: contact.NickName,
|
||||||
|
displayName: getDisplayName(contact),
|
||||||
|
signature: contact.Signature,
|
||||||
|
nameInitial: contact.PYInitial || contact.RemarkPYInitial,
|
||||||
|
nameQuanPin: contact.PYQuanPin || contact.RemarkPYQuanPin,
|
||||||
|
avatar: `${config.baseUrl}/session/${sessionID}/avatar/${contact.UserName}`,
|
||||||
|
isFriend: contact.ContactFlag === 1,
|
||||||
|
};
|
||||||
|
}
|
||||||
|
const data = { contacts: [], specialContacts: [], groups: [] };
|
||||||
|
for (const uid of Object.keys(bot.contacts)) {
|
||||||
|
const contact = bot.contacts[uid];
|
||||||
|
if (bot.Contact.isPublicContact(contact)) {
|
||||||
|
continue;
|
||||||
|
} else if (bot.Contact.isSpContact(contact)) {
|
||||||
|
if (["filehelper"].includes(contact.UserName)) {
|
||||||
|
data.specialContacts.push(wrapContact(contact));
|
||||||
|
}
|
||||||
|
continue;
|
||||||
|
} else if (bot.Contact.isRoomContact(contact)) {
|
||||||
|
data.groups.push(wrapContact(contact));
|
||||||
|
} else if (!contact.isSelf) {
|
||||||
|
data.contacts.push(wrapContact(contact));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return data;
|
||||||
|
}
|
||||||
|
async function sendMessage({ type, target, filename, fileUri, data }) {
|
||||||
|
const targetContact = bot.contacts[target];
|
||||||
|
if (!targetContact) {
|
||||||
|
return "联系人不存在";
|
||||||
|
}
|
||||||
|
const to = targetContact.UserName;
|
||||||
|
switch (type) {
|
||||||
|
case "text":
|
||||||
|
await bot.sendText(data, target);
|
||||||
|
break;
|
||||||
|
case "emotion":
|
||||||
|
await bot.sendEmoticon(data, to);
|
||||||
|
break;
|
||||||
|
case "img":
|
||||||
|
case "video":
|
||||||
|
case "file": {
|
||||||
|
const filepath = path.join(ensureUploadDir(), fileUri);
|
||||||
|
if (!fs.existsSync(filepath)) {
|
||||||
|
return "图片不存在";
|
||||||
|
}
|
||||||
|
let mediaId;
|
||||||
|
try {
|
||||||
|
const { mediaId: id } = await bot.uploadMedia(
|
||||||
|
fs.createReadStream(filepath),
|
||||||
|
filename,
|
||||||
|
targetContact.UserName
|
||||||
|
);
|
||||||
|
mediaId = id;
|
||||||
|
} catch {
|
||||||
|
return "上传文件到远程服务器失败";
|
||||||
|
}
|
||||||
|
try {
|
||||||
|
if (type === "img") {
|
||||||
|
await bot.sendPic(mediaId, to);
|
||||||
|
} else if (type === "video") {
|
||||||
|
await bot.sendVideo(mediaId, to);
|
||||||
|
} else {
|
||||||
|
await bot.sendDoc(
|
||||||
|
mediaId,
|
||||||
|
filename,
|
||||||
|
fs.statSync(filepath).size,
|
||||||
|
path.extname(filepath),
|
||||||
|
to
|
||||||
|
);
|
||||||
|
}
|
||||||
|
} catch (e) {
|
||||||
|
return e?.tips || e?.message || "消息发送失败";
|
||||||
|
}
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
async function revokeMessage({ msgID, target }) {
|
||||||
|
await bot.revokeMsg(msgID, target);
|
||||||
|
}
|
||||||
|
async function uploadFile({ data }) {
|
||||||
|
const uploadDir = ensureUploadDir();
|
||||||
|
const filename = dayjs().format("YYYYMMDD_") + String(Math.random()).slice(1);
|
||||||
|
const filepath = path.join(uploadDir, filename);
|
||||||
|
await promisify(fs.writeFile)(filepath, data);
|
||||||
|
// 10分钟后删除上传的临时文件
|
||||||
|
setTimeout(() => {
|
||||||
|
if (fs.existsSync(filepath)) {
|
||||||
|
fs.rmSync(filepath);
|
||||||
|
}
|
||||||
|
}, 600000);
|
||||||
|
return { localFileName: filename };
|
||||||
|
}
|
||||||
|
function handleNotImplemented(_, cb) {
|
||||||
|
cb?.call(null, "Not Implemented");
|
||||||
|
}
|
||||||
|
|
||||||
|
io.on("connection", (socket) => {
|
||||||
|
console.log("connected");
|
||||||
|
activeSessions.add(socket.id);
|
||||||
|
socket.join(DEFAULT_ROOM);
|
||||||
|
socket.on("disconnect", deactivateSession);
|
||||||
|
socket.on("login", login.bind(null, socket.id));
|
||||||
|
socket.on("get_contacts", getContacts.bind(null, socket.id));
|
||||||
|
socket.on("send_message", sendMessage);
|
||||||
|
socket.on("revoke_message", revokeMessage);
|
||||||
|
socket.on("upload_file", uploadFile);
|
||||||
|
socket.onAny(handleNotImplemented);
|
||||||
|
socket.use((_, next) => {
|
||||||
|
if (bot?.state !== bot.CONF.STATE.login) {
|
||||||
|
return next(Error("服务不可用"));
|
||||||
|
}
|
||||||
|
next();
|
||||||
|
});
|
||||||
|
});
|
||||||
|
|
||||||
|
/** 获取头像 */
|
||||||
|
router.get("/session/:sessionID/avatar/:userID", (ctx, next) => {
|
||||||
|
if (!activeSessions.has(ctx.params.sessionID)) {
|
||||||
|
ctx.throw(401);
|
||||||
|
}
|
||||||
|
const contact = bot.contacts[ctx.params.userID];
|
||||||
|
if (!contact) {
|
||||||
|
ctx.throw(404);
|
||||||
|
}
|
||||||
|
const filename = composeAvatarFileName(contact);
|
||||||
|
const filepath = filename && path.join(ensureAvatarDir(), filename);
|
||||||
|
if (filepath && fs.existsSync(filepath)) {
|
||||||
|
ctx.response.headers["Content-Type"] = "image/jpeg";
|
||||||
|
ctx.response.body = fs.createReadStream(filepath);
|
||||||
|
} else {
|
||||||
|
ctx.throw(404);
|
||||||
|
}
|
||||||
|
next();
|
||||||
|
});
|
||||||
|
/** 获取消息文件，包含图片 */
|
||||||
|
router.get("/session/:sessionID/media/:composedID/:filename", (ctx, next) => {
|
||||||
|
const filename = composeMediaFilePathFromID(
|
||||||
|
ctx.params.composedID,
|
||||||
|
ctx.params.filename
|
||||||
|
);
|
||||||
|
if (fs.existsSync(filename)) {
|
||||||
|
ctx.response.body = fs.createReadStream(filename);
|
||||||
|
} else {
|
||||||
|
ctx.throw(404);
|
||||||
|
}
|
||||||
|
next();
|
||||||
|
});
|
||||||
|
|
||||||
|
app.use(cors());
|
||||||
|
app.use(router.routes());
|
||||||
|
httpServer.listen(config.httpPort, config.httpHost);
|
||||||
|
|
||||||
|
module.exports = {
|
||||||
|
pushMessage,
|
||||||
|
};
|
||||||
@@ -0,0 +1,81 @@
|
|||||||
|
const crypto = require("crypto");
|
||||||
|
const fs = require("fs");
|
||||||
|
const path = require("path");
|
||||||
|
const dayjs = require("dayjs");
|
||||||
|
const { config } = require("./config");
|
||||||
|
|
||||||
|
function composeMessageID(msg) {
|
||||||
|
return dayjs.unix(msg["CreateTime"]).format("YYYYMMDD") + msg["MsgId"];
|
||||||
|
}
|
||||||
|
function composeFilename(msg, suffix = "") {
|
||||||
|
const dt = dayjs.unix(msg["CreateTime"]);
|
||||||
|
const datetime = dt.format("YYYYMMDD");
|
||||||
|
return path.join(datetime + msg["MsgId"] + suffix);
|
||||||
|
}
|
||||||
|
function composeMediaFilePath(msg, filename) {
|
||||||
|
const year = dayjs.unix(msg["CreateTime"]).format("YYYY");
|
||||||
|
return path.join(config.mediaDir, year, filename);
|
||||||
|
}
|
||||||
|
function composeMediaFilePathFromID(composedID, filename) {
|
||||||
|
return path.join(config.mediaDir, String(composedID).slice(0, 4), filename);
|
||||||
|
}
|
||||||
|
function composeCacheFilePath(msg, suffix = "") {
|
||||||
|
return path.join(config.cacheDir, composeFilename(msg, suffix));
|
||||||
|
}
|
||||||
|
function composeAvatarFileName(contact) {
|
||||||
|
const name =
|
||||||
|
contact.RemarkName ||
|
||||||
|
contact.DisplayName ||
|
||||||
|
contact.NickName ||
|
||||||
|
contact.UserName;
|
||||||
|
return textSHA1(name) + ".jpg";
|
||||||
|
}
|
||||||
|
function ensureMediaDir(msg) {
|
||||||
|
const dir = composeMediaFilePath(msg, "");
|
||||||
|
fs.mkdirSync(dir, { recursive: true });
|
||||||
|
return dir;
|
||||||
|
}
|
||||||
|
function ensureAvatarDir() {
|
||||||
|
const dir = path.join(config.cacheDir, "avatar");
|
||||||
|
fs.mkdirSync(dir, { recursive: true });
|
||||||
|
return dir;
|
||||||
|
}
|
||||||
|
function ensureUploadDir() {
|
||||||
|
const dir = path.join(config.cacheDir, "upload");
|
||||||
|
fs.mkdirSync(dir, { recursive: true });
|
||||||
|
return dir;
|
||||||
|
}
|
||||||
|
|
||||||
|
function fileMD5(filename) {
|
||||||
|
return new Promise((reslove) => {
|
||||||
|
let md5sum = crypto.createHash("md5");
|
||||||
|
let stream = fs.createReadStream(filename);
|
||||||
|
stream.on("data", function (chunk) {
|
||||||
|
md5sum.update(chunk);
|
||||||
|
});
|
||||||
|
stream.on("end", function () {
|
||||||
|
let md5 = md5sum.digest("hex");
|
||||||
|
reslove(md5);
|
||||||
|
});
|
||||||
|
});
|
||||||
|
}
|
||||||
|
function textSHA1(text) {
|
||||||
|
return crypto
|
||||||
|
.createHash("sha1")
|
||||||
|
.update(Buffer.from(text, "utf-8"))
|
||||||
|
.digest("hex");
|
||||||
|
}
|
||||||
|
|
||||||
|
module.exports = {
|
||||||
|
composeMessageID,
|
||||||
|
composeFilename,
|
||||||
|
composeMediaFilePath,
|
||||||
|
composeMediaFilePathFromID,
|
||||||
|
composeCacheFilePath,
|
||||||
|
composeAvatarFileName,
|
||||||
|
ensureMediaDir,
|
||||||
|
ensureAvatarDir,
|
||||||
|
ensureUploadDir,
|
||||||
|
fileMD5,
|
||||||
|
textSHA1,
|
||||||
|
};
|
||||||
Reference in New Issue
Block a user