node-karin 1.15.5 → 1.16.1

package/dist/uptime-C121X_rq.mjs

@@ -0,0 +1,210 @@
+import fs from "node:fs";
+import os from "node:os";
+import { exec } from "child_process";
+import { createServer } from "node:net";
+
+//#region src/utils/system/error.ts
+/**
+* @description 拆解错误对象 用于`JSON`序列化
+* @param error 错误对象
+* @returns 拆解后的错误对象
+*/
+const stringifyError = (error) => {
+	const { name, message, stack } = error || {};
+	return {
+		name,
+		message,
+		stack
+	};
+};
+/**
+* 将错误对象转为字符串
+* @param error 错误对象
+* @returns 错误字符串
+*/
+const errorToString = (error) => {
+	if (!error) return "";
+	return [
+		`name: ${error.name}`,
+		`message: ${error.message}`,
+		`stack: ${error.stack?.toString()}`
+	].join("\n");
+};
+
+//#endregion
+//#region src/utils/system/exec.ts
+/**
+* 执行 shell 命令
+* @param cmd 命令
+* @param options 选项
+* @param options.log 是否打印日志 默认不打印
+* @param options.booleanResult 是否只返回布尔值 表示命令是否成功执行 默认返回完整的结果
+* @example
+* ```ts
+* const { status, error, stdout, stderr } = await exec('ls -al')
+* // -> { status: true, error: null, stdout: '...', stderr: '...' }
+*
+* const status = await exec('ls -al', { booleanResult: true })
+* // -> true
+*
+* const { status, error, stdout, stderr } = await exec('ls -al', { log: true })
+* // -> 打印执行命令和结果
+* ```
+*/
+const exec$1 = (cmd, options) => {
+	const logger = global?.logger || console;
+	return new Promise((resolve) => {
+		if (options?.log) logger.info([
+			"[exec] 执行命令:",
+			`pwd: ${options?.cwd || process.cwd()}`,
+			`cmd: ${cmd}`,
+			`options: ${JSON.stringify(options)}`
+		].join("\n"));
+		exec(cmd, options, (error, stdout, stderr) => {
+			if (options?.log) {
+				const info = stringifyError(error);
+				if (info.message) info.message = `\x1b[91m${info.message}\x1b[0m`;
+				logger.info([
+					"[exec] 执行结果:",
+					`stderr: ${stderr.toString()}`,
+					`stdout: ${stdout.toString()}`,
+					`error: ${JSON.stringify(info, null, 2)}`
+				].join("\n"));
+			}
+			if (options?.booleanResult) return resolve(!error);
+			stdout = stdout.toString();
+			stderr = stderr.toString();
+			if (options?.trim) {
+				stdout = stdout.trim();
+				stderr = stderr.trim();
+			}
+			resolve({
+				status: !error,
+				error,
+				stdout,
+				stderr
+			});
+		});
+	});
+};
+
+//#endregion
+//#region src/utils/system/system.ts
+/** 是否为windows */
+const isWin = os.platform() === "win32";
+/** 是否为linux */
+const isLinux = os.platform() === "linux";
+/** 是否为mac */
+const isMac = os.platform() === "darwin";
+/** 是否为docker */
+const isDocker = fs.existsSync("/.dockerenv");
+/** 是否为root用户 仅linux */
+const isRoot = os.userInfo().uid === 0;
+
+//#endregion
+//#region src/utils/system/pid.ts
+/**
+* 传入端口号，返回对应的pid
+* @param port - 端口号
+*/
+const getPid = async (port) => {
+	const { stdout } = await exec$1(isWin ? `netstat -ano | findstr :${port}` : `lsof -i:${port} | grep LISTEN | awk '{print $2}'`);
+	if (!stdout) return null;
+	if (isWin) {
+		const pid = stdout.toString().split(/\s+/).filter(Boolean).pop();
+		return isNaN(Number(pid)) ? null : Number(pid);
+	}
+	return isNaN(Number(stdout)) ? null : Number(stdout);
+};
+
+//#endregion
+//#region src/utils/common/sleep.ts
+/**
+* 休眠函数
+* @param ms 毫秒
+* @example
+* ```ts
+* await sleep(1000)
+* ```
+*/
+const sleep = (ms) => new Promise((resolve) => setTimeout(resolve, ms));
+
+//#endregion
+//#region src/utils/system/port.ts
+/**
+* 检查端口是否可用
+* @param port - 端口号
+* @returns 端口可用返回true，否则返回false
+*/
+const checkPort = (port) => {
+	return new Promise((resolve) => {
+		const server = createServer();
+		server.once("error", () => {
+			server.close();
+			resolve(false);
+		});
+		server.once("listening", () => {
+			server.close();
+			resolve(true);
+		});
+		server.listen(port, "127.0.0.1");
+	});
+};
+/**
+* 等待端口可用
+* @param port - 端口号
+* @param maxAttempts - 最大尝试次数
+* @param interval - 检查间隔(ms)
+* @returns 端口是否可用
+*/
+const waitPort = async (port, maxAttempts = 30, interval = 1e3) => {
+	for (let attempt = 0; attempt < maxAttempts; attempt++) {
+		if (await checkPort(port)) return true;
+		logger.debug(`端口 ${port} 被占用，等待释放... (${attempt + 1}/${maxAttempts})`);
+		await sleep(interval);
+	}
+	return false;
+};
+/**
+* 结束指定PID或端口的程序
+* @param identifier - PID号或端口号
+* @param isPort - 是否为端口号
+* @returns 是否成功结束程序
+*/
+const killApp = async (identifier, isPort = false) => {
+	const pid = isPort ? await getPid(Number(identifier)) : Number(identifier);
+	if (!pid) return false;
+	const { stderr } = await exec$1(isWin ? `taskkill /F /PID ${pid}` : `kill -9 ${pid}`);
+	if (stderr) return false;
+	return true;
+};
+
+//#endregion
+//#region src/utils/common/uptime.ts
+/**
+* 获取运行时间
+* @example
+* ```ts
+* uptime()
+* // -> '1天2小时3分钟4秒'
+* // -> '2小时3分钟4秒'
+* // -> '3分钟4秒'
+* // -> '4秒'
+* ```
+*/
+const uptime = () => {
+	const time = process.uptime();
+	const day = Math.floor(time / 86400);
+	const hour = Math.floor(time % 86400 / 3600);
+	const min = Math.floor(time % 3600 / 60);
+	const sec = Math.floor(time % 60);
+	return [
+		day ? `${day}天` : "",
+		hour ? `${hour}小时` : "",
+		min ? `${min}分钟` : "",
+		!day && sec ? `${sec}秒` : ""
+	].filter(Boolean).join("");
+};
+
+//#endregion
+export { sleep as a, isLinux as c, isWin as d, exec$1 as f, waitPort as i, isMac as l, stringifyError as m, checkPort as n, getPid as o, errorToString as p, killApp as r, isDocker as s, uptime as t, isRoot as u };