onebots
Version:
基于icqq的多例oneBot实现
275 lines (274 loc) • 10.6 kB
JavaScript
;
var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
if (k2 === undefined) k2 = k;
var desc = Object.getOwnPropertyDescriptor(m, k);
if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
desc = { enumerable: true, get: function() { return m[k]; } };
}
Object.defineProperty(o, k2, desc);
}) : (function(o, m, k, k2) {
if (k2 === undefined) k2 = k;
o[k2] = m[k];
}));
var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
Object.defineProperty(o, "default", { enumerable: true, value: v });
}) : function(o, v) {
o["default"] = v;
});
var __importStar = (this && this.__importStar) || function (mod) {
if (mod && mod.__esModule) return mod;
var result = {};
if (mod != null) for (var k in mod) if (k !== "default" && Object.prototype.hasOwnProperty.call(mod, k)) __createBinding(result, mod, k);
__setModuleDefault(result, mod);
return result;
};
var __classPrivateFieldGet = (this && this.__classPrivateFieldGet) || function (receiver, state, kind, f) {
if (kind === "a" && !f) throw new TypeError("Private accessor was defined without a getter");
if (typeof state === "function" ? receiver !== state || !f : !state.has(receiver)) throw new TypeError("Cannot read private member from an object whose class did not declare it");
return kind === "m" ? f : kind === "a" ? f.call(receiver) : f ? f.value : state.get(receiver);
};
var _WechatAdapter_disposes;
Object.defineProperty(exports, "__esModule", { value: true });
const adapter_1 = require("../../adapter");
const onebot_1 = require("../../onebot");
const web_wechat_1 = require("web-wechat");
const path = __importStar(require("path"));
class WechatAdapter extends adapter_1.Adapter {
constructor(app, config) {
super(app, "wechat", config);
_WechatAdapter_disposes.set(this, new Map());
this.icon = `https://res.wx.qq.com/a/wx_fed/assets/res/NTI4MWU5.ico`;
}
async startOneBot(oneBot) {
await this.setOnline(oneBot.uin);
const pkg = require(path.resolve(path.dirname(require.resolve("web-wechat")), "../package.json"));
oneBot.dependency = `web-wechat v${pkg.version}`;
const disposeArr = [];
const clean = () => {
while (disposeArr.length > 0) {
disposeArr.pop()();
}
oneBot.internal.stop();
};
const messageHandler = event => {
this.emit("message.receive", oneBot.uin, event);
};
oneBot.internal.on("message", messageHandler);
disposeArr.push(() => {
oneBot.internal.off("message", messageHandler);
});
return clean;
}
async setOnline(uin) {
const oneBot = this.getOneBot(uin);
await oneBot?.internal.start();
const { data } = await oneBot?.internal.getAvatar(oneBot.internal.info.nickname);
oneBot.avatar = `data:image/png;base64,${data.toString("base64")}`;
oneBot.nickname = oneBot.internal.info.nickname;
oneBot.status = onebot_1.OneBotStatus.Online;
}
async setOffline(uin) {
const oneBot = this.getOneBot(uin);
await oneBot?.internal.stop();
oneBot.status = onebot_1.OneBotStatus.OffLine;
}
createOneBot(uin, protocol, versions) {
const oneBot = super.createOneBot(uin, protocol, versions);
oneBot.internal = new web_wechat_1.Client({
log_level: this.app.config.log_level,
...protocol,
});
oneBot.status = onebot_1.OneBotStatus.Pending;
return oneBot;
}
call(uin, version, method, args) {
const oneBot = this.oneBots.get(uin);
if (!oneBot) {
throw new Error(`未找到账号${uin}`);
}
if (typeof this[method] === "function")
return this[method](uin, version, args);
if (typeof oneBot.internal[method] !== "function")
throw onebot_1.OneBot.UnsupportedMethodError;
try {
return oneBot.internal[method](...(args || []));
}
catch (e) {
throw new Error(`call internal method error:${e.message}`);
}
}
async sendPrivateMessage(uin, version, args) {
const [user_id, message] = args;
const bot = this.getOneBot(uin);
let result = await bot.internal.sendPrivateMsg(user_id, message);
if (Array.isArray(result))
result = JSON.stringify(result);
return {
message_id: version === "V11"
? bot.V11.transformToInt("message_id", `${user_id}:${result}`)
: `${user_id}:${result}`,
};
}
async sendGroupMessage(uin, version, args) {
const [group_id, message] = args;
const bot = this.getOneBot(uin);
let result = await bot.internal.sendGroupMsg(group_id, message);
if (Array.isArray(result))
result = JSON.stringify(result);
return {
message_id: version === "V11"
? bot.V11.transformToInt("message_id", `${group_id}:${result}`)
: `${group_id}:${result}`,
};
}
async deleteMessage(uin, version, [str]) {
const [username, ...message_idArr] = str.split(":");
const bot = this.getOneBot(uin);
try {
const message_ids = JSON.parse(message_idArr.join(":"));
if (Array.isArray(message_ids)) {
let success = false;
for (const message_id in message_ids) {
success = await bot.internal.recallMsg(username, message_id);
if (!success)
return success;
}
return success;
}
else if (message_ids && typeof message_ids === "string") {
return await bot.internal.recallMsg(username, message_ids);
}
}
catch {
return bot.internal.recallMsg(username, message_idArr.join(":"));
}
}
fromSegment(onebot, version, segment) {
return []
.concat(segment)
.map(segment => {
if (version === "V12" && ["image", "video", "audio"].includes(segment.type))
return onebot.V12.transformMedia(segment);
return segment;
})
.map(item => {
if (typeof item === "string")
return item;
const { type, data } = item;
return {
type,
...data,
};
});
}
toSegment(version, message) {
return [].concat(message).map(item => {
if (!item || typeof item !== "object")
return {
type: "text",
data: {
text: item,
},
};
const { type, ...data } = item;
return {
type,
data,
};
});
}
async getFriendList(uin, version) {
const bot = this.getOneBot(uin);
const result = bot.internal.getFriendList();
return result.map(friend => {
return {
...friend,
user_name: friend.nickname,
user_id: version === "V11"
? bot.V11.transformToInt("friend", friend.user_id)
: friend.user_id,
};
});
}
async getGroupList(uin, version) {
const bot = this.getOneBot(uin);
const result = bot.internal.getGroupList();
return result.map(group => {
return {
...group,
group_id: version === "V11"
? bot.V11.transformToInt("group", group.group_id)
: group.group_id,
};
});
}
formatEventPayload(uin, version, event, data) {
const bot = this.getOneBot(uin);
const result = {
id: data.id || Math.random().toString(36).slice(2),
[version === "V12" ? "type" : "post_type"]: event,
version: version,
self: {
platform: "wechat",
user_id: data.self_id,
},
detail_type: data.message_type || data.notice_type || data.request_type,
platform: "wechat",
group: data.group?.info,
member: data.member?.info,
friend: data.friend?.info,
time: data.timestamp,
...data,
sender: {
...(data?.sender || {}),
},
user_id: data.user_id || data.sender?.user_id,
};
delete result.bot;
delete result.c;
delete result.parser;
if (event === "message") {
result.alt_message = result.raw_message || "";
}
if (version === "V11") {
bot.V11.transformStrToIntForObj(result, ["user_id", "group_id", "message_id"]);
bot.V11.transformStrToIntForObj(result.self, ["user_id"]);
bot.V11.transformStrToIntForObj(result.sender, ["user_id"]);
bot.V11.transformStrToIntForObj(result.group, ["user_id", "group_id"]);
bot.V11.transformStrToIntForObj(result.member, ["user_id", "member_id"]);
bot.V11.transformStrToIntForObj(result.friend, ["user_id"]);
}
return result;
}
async start(uin) {
const startOneBots = [...this.oneBots.values()].filter(oneBot => {
return uin ? oneBot.uin === uin : true;
});
for (const oneBot of startOneBots) {
__classPrivateFieldGet(this, _WechatAdapter_disposes, "f").set(oneBot.uin, await this.startOneBot(oneBot));
}
const { protocol } = this.config;
await super.start();
}
async stop(uin) {
const stopOneBots = [...this.oneBots.values()].filter(oneBot => {
return uin ? oneBot.uin === uin : true;
});
for (const oneBot of stopOneBots) {
const dispose = __classPrivateFieldGet(this, _WechatAdapter_disposes, "f").get(oneBot.uin);
if (dispose) {
dispose();
}
}
await super.stop();
}
getSelfInfo(uin, version) {
const oneBot = this.oneBots.get(uin);
return {
nickname: oneBot?.internal?.nickname,
status: onebot_1.OneBotStatus.Online,
};
}
}
_WechatAdapter_disposes = new WeakMap();
exports.default = WechatAdapter;