ellm-proxy 0.0.3 → 0.0.7

package/utils/paths.js

@@ -1,44 +1,45 @@
-// 共享常量与路径工具：包根/应用目录常量、配置查找链、首次配置自愈、选项绝对化。
-// 供 CLI（bin/）与管理服务（manager/）两侧复用；工具一律返回数据，人读文案由调用方输出。
-import { readFileSync, mkdirSync, existsSync, copyFileSync } from "node:fs";
-import { homedir } from "node:os";
-import { dirname, join, resolve } from "node:path";
-import { fileURLToPath } from "node:url";
-
-/** 本文件在 utils/ 下，包根为上一级 */
-export const pkgRoot = resolve(dirname(fileURLToPath(import.meta.url)), "..");
-export const defaultAppDir = join(homedir(), ".ellm-proxy");
-export const pkg = JSON.parse(readFileSync(join(pkgRoot, "package.json"), "utf-8"));
-export const APP_NAME = "ellm-proxy"; // 主服务（dist/server.js）的 pm2 进程名
-export const MANAGER_APP_NAME = "ellm-proxy-manager"; // 管理服务（manager/index.js）的 pm2 进程名
-export const MIN_NODE = ">=22.13.0"; // node:sqlite 最低版（与包 engines 一致）
-
-/** 配置文件路径：--config > ELLM_CONFIG 环境变量 > ~/.ellm-proxy/ellm.config.json（与 server 端查找链一致） */
-export function configPath(override) {
-  if (override) return resolve(override);
-  return process.env["ELLM_CONFIG"] ? resolve(process.env["ELLM_CONFIG"]) : join(defaultAppDir, "ellm.config.json");
-}
-
-/** 首次配置自愈：不存在则从包内示例复制创建，并提示两种修改方式 */
-export function ensureConfig(override) {
-  const config = configPath(override);
-  if (existsSync(config)) return config;
-  const example = join(pkgRoot, "ellm.config.example.json");
-  if (!existsSync(example)) {
-    throw new Error(`缺少示例配置 ${example}，无法自动创建配置文件`);
-  }
-  mkdirSync(dirname(config), { recursive: true });
-  copyFileSync(example, config);
-  console.log(`已创建配置文件: ${config}`);
-  console.log("可直接编辑该文件（填入供应商 apiKey），或经管理台在线修改（实时热更新）。");
-  return config;
-}
-
-/** 路径类选项统一转绝对路径（相对按调用方当前目录展开；CLI cwd 与 node 进程工作目录基准不同，透传相对串会错位） */
-export function absolutize(opts) {
-  const out = { ...opts };
-  for (const key of ["config", "dataDir", "workingDir", "serverJs"]) {
-    if (out[key]) out[key] = resolve(out[key]);
-  }
-  return out;
-}
+// 共享常量与路径工具：包根/应用目录常量、配置查找链、首次配置自愈、选项绝对化。
+// 供 CLI（bin/）与管理服务（manager/）两侧复用；工具一律返回数据，人读文案由调用方输出。
+import { readFileSync, mkdirSync, existsSync, copyFileSync } from "node:fs";
+import { homedir } from "node:os";
+import { dirname, join, resolve } from "node:path";
+import { fileURLToPath } from "node:url";
+
+/** 本文件在 utils/ 下，包根为上一级 */
+export const pkgRoot = resolve(dirname(fileURLToPath(import.meta.url)), "..");
+export const defaultAppDir = join(homedir(), ".ellm-proxy");
+export const pkg = JSON.parse(readFileSync(join(pkgRoot, "package.json"), "utf-8"));
+export const APP_NAME = "ellm-proxy"; // 主服务（dist/server.js）的 pm2 进程名
+export const MANAGER_APP_NAME = "ellm-proxy-manager"; // 管理服务（manager/index.js）的 pm2 进程名
+export const MIN_NODE = ">=22.13.0"; // node:sqlite 最低版（与包 engines 一致）
+export const FALLBACK_NODE = "22"; // 兜底解释器锁定的主版本（满足 MIN_NODE，与 nve 22 口径一致）
+
+/** 配置文件路径：--config > ELLM_CONFIG 环境变量 > ~/.ellm-proxy/ellm.config.json（与 server 端查找链一致） */
+export function configPath(override) {
+  if (override) return resolve(override);
+  return process.env["ELLM_CONFIG"] ? resolve(process.env["ELLM_CONFIG"]) : join(defaultAppDir, "ellm.config.json");
+}
+
+/** 首次配置自愈：不存在则从包内示例复制创建，并提示两种修改方式 */
+export function ensureConfig(override) {
+  const config = configPath(override);
+  if (existsSync(config)) return config;
+  const example = join(pkgRoot, "ellm.config.example.json");
+  if (!existsSync(example)) {
+    throw new Error(`缺少示例配置 ${example}，无法自动创建配置文件`);
+  }
+  mkdirSync(dirname(config), { recursive: true });
+  copyFileSync(example, config);
+  console.log(`已创建配置文件: ${config}`);
+  console.log("可直接编辑该文件（填入供应商 apiKey），或经管理台在线修改（实时热更新）。");
+  return config;
+}
+
+/** 路径类选项统一转绝对路径（相对按调用方当前目录展开；CLI cwd 与 node 进程工作目录基准不同，透传相对串会错位） */
+export function absolutize(opts) {
+  const out = { ...opts };
+  for (const key of ["config", "dataDir", "workingDir", "serverJs"]) {
+    if (out[key]) out[key] = resolve(out[key]);
+  }
+  return out;
+}

package/utils/pm2.js

@@ -1,130 +1,130 @@
-// pm2 JS API 封装：连接包裹/回调转 promise/应用查询/探活轮询/安全 dump/包内 pm2 CLI 定位/env 白名单。
-import { rmSync } from "node:fs";
-import { homedir } from "node:os";
-import { dirname, join } from "node:path";
-import { createRequire } from "node:module";
-import pm2 from "pm2";
-import { APP_NAME, MANAGER_APP_NAME } from "./paths.js";
-import { dot, endOk, endFail } from "./ui.js";
-
-/** pm2.connect 包裹：fn 内可调 pm2.xxx（回调式），结束后自动 disconnect */
-export function withPm2(fn) {
-  return new Promise((resolvePromise, reject) => {
-    pm2.connect((connectErr) => {
-      if (connectErr) return reject(connectErr);
-      Promise.resolve(fn(pm2)).then(
-        (value) => pm2.disconnect(() => resolvePromise(value)),
-        (err) => {
-          try {
-            pm2.disconnect();
-          } catch {
-            /* 已断开 */
-          }
-          reject(err);
-        },
-      );
-    });
-  });
-}
-
-/** pm2 回调式方法转 promise */
-export function pm2cb(method, ...args) {
-  return new Promise((resolvePromise, reject) => {
-    pm2[method](...args, (err, ...out) => (err ? reject(err) : resolvePromise(...out)));
-  });
-}
-
-/** 查询应用当前进程记录（list 中按 name 匹配；不存在返回 undefined）。
- *  name 缺省 = 主服务 ellm-proxy；管理服务传 MANAGER_APP_NAME */
-export async function findApp(name = APP_NAME) {
-  const list = (await pm2cb("list")) || [];
-  return list.find((p) => p.name === name);
-}
-
-/** 启动探活：轮询 status 至 online（每 pollMs 一次，全程打进度点）——比固定等待更快成功
- *  （online 即返回），也能容忍慢启动（默认 15s 上限，固定 1.5s 在慢机器上会误报）；errored/stopped
- *  提前失败（脚本秒崩不必等满超时）。failHint 是失败文案的线索部分（文案是契约的一部分）。
- *  quiet = 静默轮询不打点（多服务并行等待时输出由调用方合并，避免多组进度点交错）。 */
-export async function waitOnline(name, failHint, { timeoutMs = 15000, pollMs = 300, quiet = false } = {}) {
-  const deadline = Date.now() + timeoutMs;
-  let last = null;
-  for (;;) {
-    last = await findApp(name);
-    const status = last?.pm2_env?.status;
-    if (status === "online") {
-      if (!quiet) endOk();
-      return last;
-    }
-    if (status === "errored" || status === "stopped" || Date.now() >= deadline) {
-      if (!quiet) endFail();
-      throw new Error(`${name} 启动后状态异常（${status ?? "已退出"}）。${failHint}`);
-    }
-    if (!quiet) dot();
-    await new Promise((r) => setTimeout(r, pollMs));
-  }
-}
-
-/** 安全 dump（start/stop 成功后调用）：pm2 不允许 dump 空列表会报错——此时说明全部进程已删，
- *  清掉旧 dump.pm2 与滚动备份 dump.pm2.bak（resurrect 无 dump.pm2 时会回落 .bak），防止开机
- *  resurrect 把已停止的服务拉起来 */
-export async function safeDump() {
-  try {
-    await pm2cb("dump");
-  } catch {
-    try {
-      rmSync(join(homedir(), ".pm2", "dump.pm2"), { force: true });
-      rmSync(join(homedir(), ".pm2", "dump.pm2.bak"), { force: true });
-    } catch {
-      /* 清理失败不影响 stop 语义 */
-    }
-  }
-}
-
-/** 当前包内 pm2 CLI 入口绝对路径（autostart cmd 用；resolve 兼容 npm 全局平铺与 pnpm 布局） */
-export function pm2BinJs() {
-  const pm2PkgRoot = dirname(createRequire(import.meta.url).resolve("pm2/package.json"));
-  return join(pm2PkgRoot, "bin", "pm2");
-}
-
-/** 清洗 pm2 子进程透传的 env：pm2 托管的进程自身 env 带 pm_id/pm2_env/unique_id 等
- *  内部变量与 bash 注入的 `_`，原样传回 pm2.start 会污染新进程注册——实测主服务
- *  （name=ellm-proxy）被注册成管理服务的名字，导致探活误判「已退出」。
- *  黑名单清洗实测仍有漏网（多键组合触发），故改用白名单：
- *  只透传主服务必需的业务键（ELLM_* 前缀，含 ELLM_PORT / ELLM_MANAGER_PORT）与 Windows node 运行所需的系统键。
- *  PM2_HOME 保留（pm2 API 连接 daemon 需要，主服务自身不用）。 */
-const PM2_ENV_WHITELIST = new Set([
-  "PATH",
-  "SYSTEMROOT",
-  "WINDIR",
-  "SYSTEMDRIVE",
-  "COMSPEC",
-  "PATHEXT",
-  "OS",
-  "TEMP",
-  "TMP",
-  "USERPROFILE",
-  "HOMEDRIVE",
-  "HOMEPATH",
-  "APPDATA",
-  "LOCALAPPDATA",
-  "PROGRAMDATA",
-  "PROGRAMFILES",
-  "ALLUSERSPROFILE",
-  "PUBLIC",
-  "USERNAME",
-  "USERDOMAIN",
-  "COMPUTERNAME",
-  "NUMBER_OF_PROCESSORS",
-  "HTTP_PROXY",
-  "HTTPS_PROXY",
-  "NO_PROXY",
-]);
-
-export function sanitizePm2Env(env) {
-  const out = {};
-  for (const [key, value] of Object.entries(env)) {
-    if (key === "PM2_HOME" || key.startsWith("ELLM_")) out[key] = value;
-    else if (PM2_ENV_WHITELIST.has(key.toUpperCase())) out[key] = value;
-  }
-  return out;
-}
+// pm2 JS API 封装：连接包裹/回调转 promise/应用查询/探活轮询/安全 dump/包内 pm2 CLI 定位/env 白名单。
+import { rmSync } from "node:fs";
+import { homedir } from "node:os";
+import { dirname, join } from "node:path";
+import { createRequire } from "node:module";
+import pm2 from "pm2";
+import { APP_NAME, MANAGER_APP_NAME } from "./paths.js";
+import { dot, endOk, endFail } from "./ui.js";
+
+/** pm2.connect 包裹：fn 内可调 pm2.xxx（回调式），结束后自动 disconnect */
+export function withPm2(fn) {
+  return new Promise((resolvePromise, reject) => {
+    pm2.connect((connectErr) => {
+      if (connectErr) return reject(connectErr);
+      Promise.resolve(fn(pm2)).then(
+        (value) => pm2.disconnect(() => resolvePromise(value)),
+        (err) => {
+          try {
+            pm2.disconnect();
+          } catch {
+            /* 已断开 */
+          }
+          reject(err);
+        },
+      );
+    });
+  });
+}
+
+/** pm2 回调式方法转 promise */
+export function pm2cb(method, ...args) {
+  return new Promise((resolvePromise, reject) => {
+    pm2[method](...args, (err, ...out) => (err ? reject(err) : resolvePromise(...out)));
+  });
+}
+
+/** 查询应用当前进程记录（list 中按 name 匹配；不存在返回 undefined）。
+ *  name 缺省 = 主服务 ellm-proxy；管理服务传 MANAGER_APP_NAME */
+export async function findApp(name = APP_NAME) {
+  const list = (await pm2cb("list")) || [];
+  return list.find((p) => p.name === name);
+}
+
+/** 启动探活：轮询 status 至 online（每 pollMs 一次，全程打进度点）——比固定等待更快成功
+ *  （online 即返回），也能容忍慢启动（默认 15s 上限，固定 1.5s 在慢机器上会误报）；errored/stopped
+ *  提前失败（脚本秒崩不必等满超时）。failHint 是失败文案的线索部分（文案是契约的一部分）。
+ *  quiet = 静默轮询不打点（多服务并行等待时输出由调用方合并，避免多组进度点交错）。 */
+export async function waitOnline(name, failHint, { timeoutMs = 15000, pollMs = 300, quiet = false } = {}) {
+  const deadline = Date.now() + timeoutMs;
+  let last = null;
+  for (;;) {
+    last = await findApp(name);
+    const status = last?.pm2_env?.status;
+    if (status === "online") {
+      if (!quiet) endOk();
+      return last;
+    }
+    if (status === "errored" || status === "stopped" || Date.now() >= deadline) {
+      if (!quiet) endFail();
+      throw new Error(`${name} 启动后状态异常（${status ?? "已退出"}）。${failHint}`);
+    }
+    if (!quiet) dot();
+    await new Promise((r) => setTimeout(r, pollMs));
+  }
+}
+
+/** 安全 dump（start/stop 成功后调用）：pm2 不允许 dump 空列表会报错——此时说明全部进程已删，
+ *  清掉旧 dump.pm2 与滚动备份 dump.pm2.bak（resurrect 无 dump.pm2 时会回落 .bak），防止开机
+ *  resurrect 把已停止的服务拉起来 */
+export async function safeDump() {
+  try {
+    await pm2cb("dump");
+  } catch {
+    try {
+      rmSync(join(homedir(), ".pm2", "dump.pm2"), { force: true });
+      rmSync(join(homedir(), ".pm2", "dump.pm2.bak"), { force: true });
+    } catch {
+      /* 清理失败不影响 stop 语义 */
+    }
+  }
+}
+
+/** 当前包内 pm2 CLI 入口绝对路径（autostart cmd 用；resolve 兼容 npm 全局平铺与 pnpm 布局） */
+export function pm2BinJs() {
+  const pm2PkgRoot = dirname(createRequire(import.meta.url).resolve("pm2/package.json"));
+  return join(pm2PkgRoot, "bin", "pm2");
+}
+
+/** 清洗 pm2 子进程透传的 env：pm2 托管的进程自身 env 带 pm_id/pm2_env/unique_id 等
+ *  内部变量与 bash 注入的 `_`，原样传回 pm2.start 会污染新进程注册——实测主服务
+ *  （name=ellm-proxy）被注册成管理服务的名字，导致探活误判「已退出」。
+ *  黑名单清洗实测仍有漏网（多键组合触发），故改用白名单：
+ *  只透传主服务必需的业务键（ELLM_* 前缀，含 ELLM_PORT / ELLM_MANAGER_PORT）与 Windows node 运行所需的系统键。
+ *  PM2_HOME 保留（pm2 API 连接 daemon 需要，主服务自身不用）。 */
+const PM2_ENV_WHITELIST = new Set([
+  "PATH",
+  "SYSTEMROOT",
+  "WINDIR",
+  "SYSTEMDRIVE",
+  "COMSPEC",
+  "PATHEXT",
+  "OS",
+  "TEMP",
+  "TMP",
+  "USERPROFILE",
+  "HOMEDRIVE",
+  "HOMEPATH",
+  "APPDATA",
+  "LOCALAPPDATA",
+  "PROGRAMDATA",
+  "PROGRAMFILES",
+  "ALLUSERSPROFILE",
+  "PUBLIC",
+  "USERNAME",
+  "USERDOMAIN",
+  "COMPUTERNAME",
+  "NUMBER_OF_PROCESSORS",
+  "HTTP_PROXY",
+  "HTTPS_PROXY",
+  "NO_PROXY",
+]);
+
+export function sanitizePm2Env(env) {
+  const out = {};
+  for (const [key, value] of Object.entries(env)) {
+    if (key === "PM2_HOME" || key.startsWith("ELLM_")) out[key] = value;
+    else if (PM2_ENV_WHITELIST.has(key.toUpperCase())) out[key] = value;
+  }
+  return out;
+}