dsh-version-check 0.0.0-stage → 0.3.0

package/lib/deployment.d.ts

@@ -0,0 +1,36 @@
+/**
+ * 定位 DSH 安装目录、npm 全局前缀与 systemd 服务单元。
+ */
+/** DSH 部署形态。 */
+export interface DeploymentInfo {
+  packageRoot: string;
+  version: string;
+  npmPrefix: string | null;
+  unitName: string | null;
+  restartMode: 'systemd' | 'manual';
+}
+/**
+ * 从 `/proc/self/cgroup` 的内容中取所在服务的单元名。
+ * 按 `/` 切分 cgroup 路径，从末尾向前找第一个 `.service` 段；
+ * `.scope` 段继续向上查找（服务内嵌套的 scope 仍指向所属服务）。
+ * `user@N.service` 是 systemd 用户管理器，重启它等于注销整个用户会话，
+ * 因此该单元被排除并返回 null。
+ * @param text cgroup 文件内容。
+ * @returns 没有可用的 service 单元时返回 null。
+ */
+export declare function unitNameFromCgroup(text: unknown): string | null;
+/**
+ * 探测 DSH 的部署形态，同步执行。
+ */
+export declare function detectDeployment(): DeploymentInfo;
+/**
+ * 读取磁盘上当前的 DSH 版本。每次调用都重新读取 package.json。
+ * @param packageRoot 包根目录。
+ */
+export declare function readInstalledVersion(packageRoot: string): string;
+/**
+ * 在 PATH 中查找 npm 可执行文件。
+ * @param env 环境变量，默认使用 process.env。
+ * @returns 找不到返回 null。
+ */
+export declare function findNpmExecutable(env?: NodeJS.ProcessEnv): string | null;

package/lib/deployment.js

@@ -0,0 +1,154 @@
+import { existsSync, readFileSync, realpathSync, statSync } from "node:fs";
+import { spawnSync } from "node:child_process";
+import { delimiter, dirname, join } from "node:path";
+// src/deployment.ts
+/**
+* 定位 DSH 安装目录、npm 全局前缀与 systemd 服务单元。
+*/
+const PACKAGE_NAME = "@deepseek-ai/dsh";
+const ROOT_SEARCH_DEPTH = 8;
+const CGROUP_PATH = "/proc/self/cgroup";
+const SYSTEMCTL_TIMEOUT_MS = 5e3;
+const USER_MANAGER_PREFIX = "user@";
+/**
+* 从 `/proc/self/cgroup` 的内容中取所在服务的单元名。
+* 按 `/` 切分 cgroup 路径，从末尾向前找第一个 `.service` 段；
+* `.scope` 段继续向上查找（服务内嵌套的 scope 仍指向所属服务）。
+* `user@N.service` 是 systemd 用户管理器，重启它等于注销整个用户会话，
+* 因此该单元被排除并返回 null。
+* @param text cgroup 文件内容。
+* @returns 没有可用的 service 单元时返回 null。
+*/
+function unitNameFromCgroup(text) {
+	if (typeof text !== "string") return null;
+	for (const line of text.split("\n")) {
+		const trimmed = line.trim();
+		if (trimmed.length === 0) continue;
+		const segments = trimmed.slice(trimmed.lastIndexOf(":") + 1).split("/");
+		for (let index = segments.length - 1; index >= 0; index -= 1) {
+			const segment = segments[index];
+			if (!segment.endsWith(".service")) continue;
+			if (segment.startsWith(USER_MANAGER_PREFIX)) return null;
+			return segment;
+		}
+	}
+	return null;
+}
+/**
+* 读取 package.json；文件缺失或 JSON 非法就地抛出。
+* @param packageRoot 包目录。
+*/
+function readManifest(packageRoot) {
+	const text = readFileSync(join(packageRoot, "package.json"), "utf8");
+	const manifest = JSON.parse(text);
+	if (manifest === null || typeof manifest !== "object" || Array.isArray(manifest)) throw new Error(`package.json 不是对象: ${packageRoot}`);
+	return manifest;
+}
+/**
+* 从入口文件所在目录向上查找 `@deepseek-ai/dsh` 的安装目录。
+* @param startDir 查找起点目录。
+* @returns 包根目录。
+*/
+function findPackageRoot(startDir) {
+	let current = startDir;
+	for (let depth = 0; depth <= ROOT_SEARCH_DEPTH; depth += 1) {
+		const manifestPath = join(current, "package.json");
+		if (existsSync(manifestPath)) {
+			const manifest = JSON.parse(readFileSync(manifestPath, "utf8"));
+			if (manifest !== null && typeof manifest === "object" && manifest.name === PACKAGE_NAME) return current;
+		}
+		const parent = dirname(current);
+		if (parent === current) break;
+		current = parent;
+	}
+	throw new Error(`在 ${startDir} 的上级目录中找不到 ${PACKAGE_NAME} 的安装目录`);
+}
+/**
+* 从安装目录推断 npm 全局前缀。
+* POSIX 安装位置形如 `<prefix>/lib/node_modules/@deepseek-ai/dsh`，
+* Windows 安装位置形如 `<prefix>/node_modules/@deepseek-ai/dsh`。
+* @param packageRoot 包根目录。
+* @returns 推断失败返回 null。
+*/
+function deriveNpmPrefix(packageRoot) {
+	const match = (process.platform === "win32" ? /^(.+)[\\/]node_modules[\\/]@deepseek-ai[\\/]dsh$/ : /^(.+)\/lib\/node_modules\/@deepseek-ai\/dsh$/).exec(packageRoot);
+	if (match === null) return null;
+	return match[1];
+}
+/**
+* 读取当前进程所属的 systemd 服务单元名。
+*/
+function readUnitName() {
+	if (!existsSync(CGROUP_PATH)) return null;
+	return unitNameFromCgroup(readFileSync(CGROUP_PATH, "utf8"));
+}
+/**
+* 确认当前进程是该服务的 MainPID。
+* 只有服务的主进程才允许触发自身重启，否则会重启到错误的单元。
+* @param unitName systemd 服务单元名。
+*/
+function isServiceMainPid(unitName) {
+	const result = spawnSync("systemctl", [
+		"--user",
+		"show",
+		unitName,
+		"-p",
+		"MainPID",
+		"--value"
+	], {
+		encoding: "utf8",
+		timeout: SYSTEMCTL_TIMEOUT_MS
+	});
+	if (result.error !== void 0 || result.status !== 0) return false;
+	return String(result.stdout).trim() === String(process.pid);
+}
+/**
+* 探测 DSH 的部署形态，同步执行。
+*/
+function detectDeployment() {
+	const entryPath = process.argv[1];
+	if (typeof entryPath !== "string" || entryPath.length === 0) throw new Error("无法确定 DSH 入口文件: process.argv[1] 为空");
+	const realEntry = realpathSync(entryPath);
+	const packageRoot = findPackageRoot(dirname(realEntry));
+	const version = readManifest(packageRoot).version;
+	if (typeof version !== "string" || version.length === 0) throw new Error(`package.json 缺少 version 字段: ${packageRoot}`);
+	const candidateUnit = process.platform === "linux" ? readUnitName() : null;
+	const unitName = candidateUnit !== null && isServiceMainPid(candidateUnit) ? candidateUnit : null;
+	return {
+		packageRoot,
+		version,
+		npmPrefix: deriveNpmPrefix(packageRoot),
+		unitName,
+		restartMode: unitName === null ? "manual" : "systemd"
+	};
+}
+/**
+* 读取磁盘上当前的 DSH 版本。每次调用都重新读取 package.json。
+* @param packageRoot 包根目录。
+*/
+function readInstalledVersion(packageRoot) {
+	const version = readManifest(packageRoot).version;
+	if (typeof version !== "string" || version.length === 0) throw new Error(`package.json 缺少 version 字段: ${packageRoot}`);
+	return version;
+}
+/**
+* 在 PATH 中查找 npm 可执行文件。
+* @param env 环境变量，默认使用 process.env。
+* @returns 找不到返回 null。
+*/
+function findNpmExecutable(env = process.env) {
+	const rawPath = env === null || env === void 0 ? "" : env.PATH ?? "";
+	const candidates = process.platform === "win32" ? ["npm.cmd"] : ["npm"];
+	for (const directory of rawPath.split(delimiter)) {
+		if (directory.length === 0) continue;
+		for (const name of candidates) {
+			const candidate = join(directory, name);
+			const stat = statSync(candidate, { throwIfNoEntry: false });
+			if (stat === void 0 || !stat.isFile()) continue;
+			if (process.platform !== "win32" && (stat.mode & 73) === 0) continue;
+			return candidate;
+		}
+	}
+	return null;
+}
+export { detectDeployment, findNpmExecutable, readInstalledVersion, unitNameFromCgroup };

package/lib/gate.d.ts

@@ -0,0 +1,22 @@
+/**
+ * 请求来源校验：只接受本机 loopback 页面发起的请求。
+ */
+/** 来源校验只读取请求的 socket 与 headers，测试可以构造同形状的对象。 */
+export interface GateRequestLike {
+  socket?: {
+    remoteAddress?: string | null;
+    encrypted?: boolean;
+  } | null;
+  headers: {
+    host?: string | string[] | undefined;
+    origin?: string | string[] | undefined;
+  };
+}
+/** 拒绝原因；通过时返回 null。 */
+export type GateRefusal = 'bad-remote' | 'bad-host' | 'cross-origin' | null;
+/**
+ * 校验请求来源。
+ * @param req 请求对象。
+ * @param allowedOrigins 反向代理或隧道场景下额外允许的源。
+ */
+export declare function gateRequest(req: GateRequestLike, allowedOrigins?: string[]): GateRefusal;

package/lib/gate.js

@@ -0,0 +1,92 @@
+// src/gate.ts
+/**
+* 请求来源校验：只接受本机 loopback 页面发起的请求。
+*/
+const LOOPBACK_IPV4 = /^127(?:\.\d{1,3}){3}$/;
+/**
+* 取可比较的主机名：转小写并去掉 IPv6 方括号。
+*/
+function bareHostname(value) {
+	return value.toLowerCase().replace(/^\[/, "").replace(/\]$/, "");
+}
+/**
+* 判断主机名是否指向 loopback 接口。
+*/
+function isLoopbackHostname(hostname) {
+	const text = bareHostname(hostname);
+	if (text === "localhost" || text === "::1") return true;
+	return LOOPBACK_IPV4.test(text);
+}
+/**
+* 解析 `Host` 头为 `{hostname, port}`。
+* @param value `Host` 头原始值。
+*/
+function splitHost(value) {
+	if (typeof value !== "string") return null;
+	const text = value.trim();
+	const bracketed = text.match(/^\[([^\]]+)\](?::(\d{1,5}))?$/);
+	if (bracketed !== null) return {
+		hostname: bracketed[1],
+		port: bracketed[2] ?? ""
+	};
+	const plain = text.match(/^([A-Za-z0-9._-]+)(?::(\d{1,5}))?$/);
+	if (plain === null) return null;
+	return {
+		hostname: plain[1],
+		port: plain[2] ?? ""
+	};
+}
+/**
+* 解析 Origin 为可比较的形式。
+* @param value Origin 头原始值。
+* @returns 非 http/https 源返回 null。
+*/
+function parseOrigin(value) {
+	let parsed;
+	try {
+		parsed = new URL(value);
+	} catch {
+		return null;
+	}
+	if (parsed.protocol !== "http:" && parsed.protocol !== "https:") return null;
+	const hostname = bareHostname(parsed.hostname);
+	const port = parsed.port;
+	return {
+		key: `${parsed.protocol}//${hostname}${port.length > 0 ? `:${port}` : ""}`,
+		hostname,
+		scheme: parsed.protocol,
+		port
+	};
+}
+/**
+* 校验请求来源。
+* @param req 请求对象。
+* @param allowedOrigins 反向代理或隧道场景下额外允许的源。
+*/
+function gateRequest(req, allowedOrigins = []) {
+	const allowedKeys = /* @__PURE__ */ new Set();
+	const allowedHostnames = /* @__PURE__ */ new Set();
+	for (const entry of Array.isArray(allowedOrigins) ? allowedOrigins : []) {
+		const origin = parseOrigin(entry);
+		if (origin === null) continue;
+		allowedKeys.add(origin.key);
+		allowedHostnames.add(origin.hostname);
+	}
+	if (!isLoopbackHostname(String(req.socket?.remoteAddress ?? "").replace(/^::ffff:/i, ""))) return "bad-remote";
+	const host = splitHost(req.headers.host);
+	if (host === null) return "bad-host";
+	const hostname = bareHostname(host.hostname);
+	if (!isLoopbackHostname(hostname) && !allowedHostnames.has(hostname)) return "bad-host";
+	const origin = req.headers.origin;
+	if (typeof origin !== "string" || origin.length === 0) return null;
+	if (origin === "dsh-app://app") return null;
+	const parsed = parseOrigin(origin);
+	if (parsed === null) return "cross-origin";
+	if (allowedKeys.has(parsed.key)) return null;
+	if (parsed.hostname !== hostname) return "cross-origin";
+	const originPort = parsed.port.length > 0 ? parsed.port : parsed.scheme === "https:" ? "443" : "80";
+	const connectionPort = req.socket?.encrypted === true ? "443" : "80";
+	if (originPort !== (host.port.length > 0 ? host.port : connectionPort)) return "cross-origin";
+	return null;
+}
+export { gateRequest };

package/lib/index.d.ts

@@ -0,0 +1,7 @@
+import { Context } from "@deepseek-ai/cordis";
+/**
+ * Host 入口。
+ * @param ctx cordis 上下文。
+ * @param config patch 层配置。
+ */
+export declare function apply(ctx: Context, config: unknown): void;