napcat-sdk 0.1.1 → 0.2.0

package/dist/index.mjs

@@ -1,56 +1,35 @@
 import crypto from "node:crypto";
 import mitt from "mitt";
 
+//#region src/logger.ts
+const noop = () => {};
+const ABSTRACT_LOGGER = {
+	fatal: noop,
+	error: noop,
+	warn: noop,
+	info: noop,
+	debug: noop,
+	trace: noop
+};
+const CONSOLE_LOGGER = {
+	fatal: console.error.bind(console, "[FATAL]"),
+	error: console.error.bind(console, "[ERROR]"),
+	warn: console.warn.bind(console, "[WARN]"),
+	info: console.info.bind(console, "[INFO]"),
+	debug: console.debug.bind(console, "[DEBUG]"),
+	trace: console.trace.bind(console, "[TRACE]")
+};
+
+//#endregion
 //#region package.json
 var name$1 = "napcat-sdk";
-var version$1 = "0.1.1";
-var package_default = {
-	name: name$1,
-	type: "module",
-	version: version$1,
-	packageManager: "pnpm@10.26.0",
-	description: "A simple SDK for NapCat OneBot v11.",
-	keywords: [
-		"napcat",
-		"onebot",
-		"onebot v11",
-		"sdk"
-	],
-	homepage: "https://github.com/vikiboss/mioki#readme",
-	files: ["dist"],
-	bugs: { "url": "https://github.com/vikiboss/mioki/issues" },
-	repository: {
-		"type": "git",
-		"url": "git+https://github.com/vikiboss/mioki.git",
-		"directory": "packages/napcat-sdk"
-	},
-	scripts: {
-		"dev": "tsdown -w",
-		"build": "tsdown"
-	},
-	exports: {
-		".": {
-			"require": "./dist/index.cjs",
-			"import": "./dist/index.mjs",
-			"types": "./dist/index.d.ts"
-		},
-		"./package.json": "./package.json"
-	},
-	author: "Viki <hi@viki.moe> (https://github.com/vikiboss)",
-	license: "MIT",
-	devDependencies: {
-		"@types/node": "catalog:dev",
-		"tsdown": "catalog:dev",
-		"typescript": "catalog:dev"
-	},
-	dependencies: { "mitt": "^3.0.1" }
-};
+var version$1 = "0.2.0";
 
 //#endregion
 //#region src/segment.ts
-function createSegment(type$1, data) {
+function createSegment(type, data) {
 	return {
-		type: type$1,
+		type,
 		...data
 	};
 }
@@ -76,9 +55,9 @@ const segment = {
 	}),
 	mface: (options) => createSegment("mface", { ...options }),
 	bface: (id) => createSegment("bface", { id }),
-	contact: (type$1, id) => createSegment("contact", {
+	contact: (type, id) => createSegment("contact", {
 		id,
-		sub_type: type$1
+		sub_type: type
 	}),
 	poke: () => createSegment("poke", {}),
 	music: (platform, id) => createSegment("music", {
@@ -92,6 +71,7 @@ const segment = {
 		title,
 		...options
 	}),
+	node: (options) => createSegment("node", { ...options }),
 	forward: (id) => createSegment("forward", { id }),
 	json: (data) => createSegment("json", { data }),
 	file: (file, options) => createSegment("file", {
@@ -102,101 +82,6 @@ const segment = {
 	lightapp: () => createSegment("lightapp", {})
 };
 
-//#endregion
-//#region src/logger.ts
-const LOG_LEVEL_KEYS = Object.keys({
-	fatal: 0,
-	error: 1,
-	warn: 2,
-	info: 3,
-	debug: 4,
-	trace: 5
-});
-const noop = () => {};
-const ABSTRACT_LOGGER = Object.fromEntries(LOG_LEVEL_KEYS.map((level) => [level, noop]));
-const CONSOLE_LOGGER = Object.fromEntries(LOG_LEVEL_KEYS.map((level) => [level, (...args) => {
-	console[level === "fatal" ? "error" : level === "trace" ? "debug" : level](...args);
-}]));
-
-//#endregion
-//#region src/onebot.ts
-/**
-* NapCat 通知类型映射表（notify 类型）
-* @description 将 NapCat 特有的通知类型映射到标准的 notice_type 和 sub_type
-*/
-const NAPCAT_NOTICE_NOTIFY_MAP = {
-	input_status: {
-		notice_type: "friend",
-		sub_type: "input"
-	},
-	profile_like: {
-		notice_type: "friend",
-		sub_type: "like"
-	},
-	title: {
-		notice_type: "group",
-		sub_type: "title"
-	}
-};
-/**
-* NapCat 通知事件映射表（notice 类型）
-* @description 将 NapCat 的原始通知事件类型映射到标准的 notice_type 和 sub_type
-*/
-const NAPCAT_NOTICE_EVENT_MAP = {
-	friend_add: {
-		notice_type: "friend",
-		sub_type: "increase"
-	},
-	friend_recall: {
-		notice_type: "friend",
-		sub_type: "recall"
-	},
-	offline_file: {
-		notice_type: "friend",
-		sub_type: "offline_file"
-	},
-	client_status: {
-		notice_type: "client",
-		sub_type: "status"
-	},
-	group_admin: {
-		notice_type: "group",
-		sub_type: "admin"
-	},
-	group_ban: {
-		notice_type: "group",
-		sub_type: "ban"
-	},
-	group_card: {
-		notice_type: "group",
-		sub_type: "card"
-	},
-	group_upload: {
-		notice_type: "group",
-		sub_type: "upload"
-	},
-	group_decrease: {
-		notice_type: "group",
-		sub_type: "decrease"
-	},
-	group_increase: {
-		notice_type: "group",
-		sub_type: "increase"
-	},
-	group_msg_emoji_like: {
-		notice_type: "group",
-		sub_type: "reaction"
-	},
-	essence: {
-		notice_type: "group",
-		sub_type: "essence"
-	},
-	group_recall: {
-		notice_type: "group",
-		sub_type: "recall"
-	}
-};
-
 //#endregion
 //#region src/napcat.ts
 const name = name$1;
@@ -214,6 +99,14 @@ var NapCat = class {
 	#event = mitt();
 	/** Echo 事件发射器 */
 	#echoEvent = mitt();
+	/** 机器人 ID */
+	#uin = 0;
+	/** 机器人昵称  */
+	#nickname = "-";
+	/** 机器人状态 */
+	#online = false;
+	/** Cookies 缓存 */
+	#cookieCache = /* @__PURE__ */ new Map();
 	constructor(options) {
 		this.options = options;
 	}
@@ -243,6 +136,24 @@ var NapCat = class {
 	get segment() {
 		return segment;
 	}
+	/**
+	* 机器人 QQ 号
+	*/
+	get user_id() {
+		return this.uin;
+	}
+	/**
+	* 机器人 QQ 号
+	*/
+	get uin() {
+		return this.#uin;
+	}
+	/**
+	* 机器人昵称
+	*/
+	get nickname() {
+		return this.#nickname;
+	}
 	/** 生成唯一的 echo ID */
 	#echoId() {
 		return crypto.randomBytes(16).toString("hex");
@@ -279,9 +190,9 @@ var NapCat = class {
 				type: "at",
 				data: { qq: String(item.qq) }
 			};
-			const { type: type$1, ...data } = item;
+			const { type, ...data } = item;
 			return {
-				type: type$1,
+				type,
 				data
 			};
 		});
@@ -305,6 +216,7 @@ var NapCat = class {
 			...extraInfo,
 			group_id,
 			group_name,
+			napcat: this,
 			doSign: () => this.api("set_group_sign", { group_id }),
 			getInfo: () => this.api("get_group_info", { group_id }),
 			getMemberList: async () => this.api("get_group_member_list", { group_id }),
@@ -338,6 +250,7 @@ var NapCat = class {
 			...extraInfo,
 			user_id,
 			nickname,
+			napcat: this,
 			delete: (block, both) => this.api("delete_friend", {
 				user_id,
 				temp_block: block,
@@ -347,24 +260,37 @@ var NapCat = class {
 			getInfo: () => this.api("get_stranger_info", { user_id })
 		};
 	}
-	/** 构建群消息事件 */
+	/** 构建私聊消息事件 */
 	#buildPrivateMessageEvent(event) {
+		const quote_id = event.message.find((el) => el.type === "reply")?.data?.id || null;
 		return {
 			...event,
-			message: (event.message || []).map((el) => ({
+			quote_id,
+			getQuoteMessage: async () => {
+				if (!quote_id) return null;
+				const event$1 = await this.api("get_msg", { message_id: quote_id });
+				return this.#buildPrivateMessageEvent(event$1);
+			},
+			message: (event.message || []).filter((e) => e.type !== "reply").map((el) => ({
 				type: el.type,
 				...el.data
 			})),
 			friend: this.#buildFriend(event.user_id, event.sender?.nickname || ""),
-			recall: () => this.api("delete_msg", { message_id: event.message_id }),
 			reply: (sendable, reply = false) => this.sendPrivateMsg(event.user_id, this.#wrapReply(sendable, event.message_id, reply))
 		};
 	}
 	/** 构建群消息事件对象 */
 	#buildGroupMessageEvent(event) {
+		const quote_id = event.message.find((el) => el.type === "reply")?.data?.id || null;
 		return {
 			...event,
-			message: (event.message || []).map((el) => ({
+			quote_id,
+			getQuoteMessage: async () => {
+				if (!quote_id) return null;
+				const event$1 = await this.api("get_msg", { message_id: quote_id });
+				return this.#buildGroupMessageEvent(event$1);
+			},
+			message: (event.message || []).filter((e) => e.type !== "reply").map((el) => ({
 				type: el.type,
 				...el.data
 			})),
@@ -398,7 +324,20 @@ var NapCat = class {
 					this.#event.emit("meta_event", data);
 					if (data.meta_event_type) {
 						this.#event.emit(`meta_event.${data.meta_event_type}`, data);
-						if (data.sub_type) this.#event.emit(`meta_event.${data.meta_event_type}.${data.sub_type}`, data);
+						if (data.sub_type) {
+							if (data.sub_type === "connect") {
+								this.#uin = data.self_id;
+								this.#online = true;
+								this.#event.emit("napcat.connected", {
+									uin: this.#uin,
+									ts: data.time * 1e3
+								});
+								this.getLoginInfo().then((info) => {
+									this.#nickname = info.nickname;
+								});
+							}
+							this.#event.emit(`meta_event.${data.meta_event_type}.${data.sub_type}`, data);
+						}
 					}
 					break;
 				case "message":
@@ -494,24 +433,34 @@ var NapCat = class {
 	}
 	/** 获取一个群的信息，可以用于发送群消息等操作 */
 	async pickGroup(group_id) {
-		const groupInfo = await this.api("get_group_info", { group_id });
-		return this.#buildGroup(group_id, groupInfo.group_name, groupInfo);
+		try {
+			const groupInfo = await this.api("get_group_info", { group_id });
+			return this.#buildGroup(group_id, groupInfo.group_name, groupInfo);
+		} catch (err) {
+			this.logger.warn(`Error to pickGroup ${group_id}: ${err?.message || err}`);
+			return null;
+		}
 	}
 	/** 获取一个好友的信息，可以用于发送私聊消息等操作 */
 	async pickFriend(user_id) {
-		const friendInfo = await this.api("get_stranger_info", { user_id });
-		return this.#buildFriend(user_id, friendInfo.nickname, friendInfo);
+		try {
+			const friendInfo = await this.api("get_stranger_info", { user_id });
+			return this.#buildFriend(user_id, friendInfo.nickname, friendInfo);
+		} catch (err) {
+			this.logger.warn(`Error to pickFriend ${user_id}: ${err?.message || err}`);
+			return null;
+		}
 	}
 	/**
 	* 注册一次性事件监听器
 	*/
-	once(type$1, handler) {
+	once(type, handler) {
 		const onceHandler = (event) => {
 			handler(event);
-			this.#event.off(type$1, onceHandler);
+			this.#event.off(type, onceHandler);
 		};
-		this.logger.debug(`registering once: ${String(type$1)}`);
-		this.#event.on(type$1, onceHandler);
+		this.logger.debug(`registering once: ${String(type)}`);
+		this.#event.on(type, onceHandler);
 	}
 	/**
 	* 注册事件监听器，支持主类型或者点分子类型
@@ -520,17 +469,20 @@ var NapCat = class {
 	*
 	* 如果需要移除监听器，请调用 `off` 方法
 	*/
-	on(type$1, handler) {
-		this.logger.debug(`registering: ${String(type$1)}`);
-		this.#event.on(type$1, handler);
+	on(type, handler) {
+		this.logger.debug(`registering: ${String(type)}`);
+		this.#event.on(type, handler);
 	}
 	/**
 	* 移除事件监听器
 	*/
-	off(type$1, handler) {
-		this.logger.debug(`unregistering: ${String(type$1)}`);
-		this.#event.off(type$1, handler);
+	off(type, handler) {
+		this.logger.debug(`unregistering: ${String(type)}`);
+		this.#event.off(type, handler);
 	}
+	/**
+	* 调用 NapCat API
+	*/
 	api(action, params = {}) {
 		this.#ensureWsConnection(this.#ws);
 		this.logger.debug(`calling api action: ${action} with params: ${JSON.stringify(params)}`);
@@ -560,10 +512,81 @@ var NapCat = class {
 			message: this.#normalizeSendable(sendable)
 		});
 	}
+	/**
+	* 机器人是否在线
+	*/
+	isOnline() {
+		return this.#ws?.readyState === WebSocket.OPEN && this.#online;
+	}
+	/**
+	* 计算 GTK 值
+	*/
+	getGTk(pskey) {
+		let gkt = 5381;
+		for (let i = 0, len = pskey.length; i < len; ++i) gkt += (gkt << 5) + pskey.charCodeAt(i);
+		return gkt & 2147483647;
+	}
+	/**
+	* 获取 NapCat 原始 Cookie 相关信息
+	*/
+	getNapCatCookies(domain) {
+		return this.api("get_cookies", { domain });
+	}
+	/**
+	* 获取版本信息
+	*/
+	getVersionInfo() {
+		return this.api("get_version_info");
+	}
+	/**
+	* 获取登录信息
+	*/
+	getLoginInfo() {
+		return this.api("get_login_info");
+	}
+	/**
+	* 获取 Cookie 相关信息
+	*/
+	async getCookie(domain) {
+		const cache = this.#cookieCache.get(domain);
+		if (cache) return cache;
+		const { cookies: cookieString, bkn } = await this.getNapCatCookies(domain);
+		const skey = cookieString.match(/skey=([^;]*)/)?.[1] || "";
+		const pskey = cookieString.match(/pskey=([^;]*)/)?.[1] || "";
+		const gtk = this.getGTk(pskey);
+		const returns = {
+			pskey,
+			skey,
+			uin: this.uin,
+			gtk: String(gtk),
+			bkn,
+			cookie: `uin=${this.uin}; skey=${skey}; p_uin=${this.uin}; p_skey=${pskey};`,
+			legacyCookie: `uin=o${this.uin}; skey=${skey}; p_uin=o${this.uin}; p_skey=${pskey};`
+		};
+		this.#cookieCache.set(domain, returns);
+		setTimeout(() => {
+			this.#cookieCache.delete(domain);
+		}, 1e3 * 60 * 60);
+		return returns;
+	}
+	/**
+	* 通过域名获取 Pskey
+	*/
+	async getPskey(domain) {
+		const { pskey } = await this.getCookie(domain);
+		return pskey;
+	}
+	/**
+	* 获取 Bkn 值
+	*/
+	async getBkn() {
+		const { bkn } = await this.getCookie("vip.qq.com");
+		return bkn;
+	}
 	/** 启动 NapCat SDK 实例，建立 WebSocket 连接 */
-	async bootstrap() {
-		const { logger: _, ...config } = this.#config;
-		this.logger.info(`bootstrap with config: ${JSON.stringify(config)}`);
+	async run() {
+		const { logger: _, token: __, ...config } = this.#config;
+		this.logger.info(`run with config: ${JSON.stringify(config)}`);
 		return new Promise((resolve, reject) => {
 			const ws = new WebSocket(this.#buildWsUrl());
 			ws.onmessage = (event) => {
@@ -582,16 +605,18 @@ var NapCat = class {
 				this.#bindInternalEvents(data);
 			};
 			ws.onclose = () => {
+				this.#online = false;
 				this.logger.info("closed");
 				this.#event.emit("ws.close");
 			};
 			ws.onerror = (error) => {
+				this.#online = false;
 				this.logger.error(`error: ${error}`);
 				this.#event.emit("ws.error", error);
 				reject(error);
 			};
 			ws.onopen = () => {
-				this.logger.info("connected");
+				this.logger.info("NapCat connected");
 				this.#event.emit("ws.open");
 				resolve();
 			};
@@ -600,7 +625,7 @@ var NapCat = class {
 		});
 	}
 	/** 销毁 NapCat SDK 实例，关闭 WebSocket 连接 */
-	async destroy() {
+	close() {
 		if (this.#ws) {
 			this.logger.info("destroying NapCat SDK instance...");
 			this.#ws.close();
@@ -609,7 +634,83 @@ var NapCat = class {
 		} else this.logger.warn("NapCat SDK instance is not initialized.");
 	}
 };
+/**
+* NapCat 通知类型映射表（notify 类型）
+* @description 将 NapCat 特有的通知类型映射到标准的 notice_type 和 sub_type
+*/
+const NAPCAT_NOTICE_NOTIFY_MAP = {
+	input_status: {
+		notice_type: "friend",
+		sub_type: "input"
+	},
+	profile_like: {
+		notice_type: "friend",
+		sub_type: "like"
+	},
+	title: {
+		notice_type: "group",
+		sub_type: "title"
+	}
+};
+/**
+* NapCat 通知事件映射表（notice 类型）
+* @description 将 NapCat 的原始通知事件类型映射到标准的 notice_type 和 sub_type
+*/
+const NAPCAT_NOTICE_EVENT_MAP = {
+	friend_add: {
+		notice_type: "friend",
+		sub_type: "increase"
+	},
+	friend_recall: {
+		notice_type: "friend",
+		sub_type: "recall"
+	},
+	offline_file: {
+		notice_type: "friend",
+		sub_type: "offline_file"
+	},
+	client_status: {
+		notice_type: "client",
+		sub_type: "status"
+	},
+	group_admin: {
+		notice_type: "group",
+		sub_type: "admin"
+	},
+	group_ban: {
+		notice_type: "group",
+		sub_type: "ban"
+	},
+	group_card: {
+		notice_type: "group",
+		sub_type: "card"
+	},
+	group_upload: {
+		notice_type: "group",
+		sub_type: "upload"
+	},
+	group_decrease: {
+		notice_type: "group",
+		sub_type: "decrease"
+	},
+	group_increase: {
+		notice_type: "group",
+		sub_type: "increase"
+	},
+	group_msg_emoji_like: {
+		notice_type: "group",
+		sub_type: "reaction"
+	},
+	essence: {
+		notice_type: "group",
+		sub_type: "essence"
+	},
+	group_recall: {
+		notice_type: "group",
+		sub_type: "recall"
+	}
+};
 
 //#endregion
-export { ABSTRACT_LOGGER, CONSOLE_LOGGER, DEFAULT_NAPCAT_OPTIONS, NapCat, package_default as PKG, name, segment, version };
+export { ABSTRACT_LOGGER, CONSOLE_LOGGER, NapCat, name, noop, segment, version };
 //# sourceMappingURL=index.mjs.map