cc-viewer 1.6.292 → 1.6.294

package/server/lib/im-process-manager.js

@@ -0,0 +1,161 @@
+// IM 进程管理（父进程侧）—— 由主交互式 ccv 调用，负责把每个启用的 IM 作为 detached 常驻
+// 子进程拉起 / 停止 / 查询状态 / 启动时 reconcile。worker 自身（cli.js runImMode）持有 im.lock；
+// 本模块只读锁 + 探活 + spawn/kill。
+//
+// 设计：detached + unref（主 ccv 关闭后 worker 仍在线）；停止时杀**进程组**清理 worker 的
+// Claude PTY + proxy 等孙进程；env 用「前缀剥离 + 白名单回填」杜绝 CCV_*/CCVIEWER_* 泄漏。
+// v1 不做常驻 supervisor（reconcile-on-startup + 配置保存即驱动 已覆盖主路径，且避免多主进程互斗）。
+import { spawn as nodeSpawn, execSync } from 'node:child_process';
+import { openSync, closeSync, mkdirSync } from 'node:fs';
+import { join, dirname, resolve } from 'node:path';
+import { fileURLToPath } from 'node:url';
+import { listPlatforms, loadConfig } from './im-config.js';
+import {
+  imDir, readImLock, clearImLock, isPidAlive, getImLiveness, defaultProbe,
+} from './im-lock.js';
+
+const CLI_JS = resolve(dirname(fileURLToPath(import.meta.url)), '..', '..', 'cli.js');
+
+const sleep = (ms) => new Promise((r) => setTimeout(r, ms));
+
+/** 解析真实 node 二进制（Electron 下 process.execPath 是 Electron，需要 `which node`）。 */
+export function resolveNodeBinary() {
+  if (!process.versions.electron) return process.execPath;
+  try {
+    const out = execSync(process.platform === 'win32' ? 'where node' : 'which node', { encoding: 'utf-8' });
+    const p = process.platform === 'win32' ? out.split('\n')[0].trim() : out.trim();
+    if (p) return p;
+  } catch { /* fall through */ }
+  return process.platform === 'win32' ? 'node' : '/usr/local/bin/node';
+}
+
+/**
+ * 构建 worker 子进程环境（前缀剥离 + 白名单回填）。
+ * - 先继承全部 process.env（保住 PATH / HOME / ANTHROPIC_ * / 代理 / locale 等）。
+ * - 删除所有 CCV_ 与 CCVIEWER_ 前缀的内部变量（面向未来：新增的 CCV_ 变量也不会泄漏）。
+ *   尤其防住：CCV_BYPASS_PERMISSIONS、CCV_ELECTRON_MULTITAB（会让 worker 不起 Claude PTY）、
+ *   CCV_PASSWORD、CCV_USER_NAME/AVATAR、CCV_PROXY_MODE、CCV_PROJECT_DIR 等误导项。
+ * - 仅 CCV_LOG_DIR 在父进程显式设置时回填（共享同一份 prefs 与日志根；生产默认不设，worker 走默认）。
+ * - 再写入 worker 专属 env。
+ */
+export function buildChildEnv(id, base = process.env) {
+  const env = { ...base };
+  const inheritedLogDir = base.CCV_LOG_DIR;
+  for (const k of Object.keys(env)) {
+    if (k.startsWith('CCV_') || k.startsWith('CCVIEWER_')) delete env[k];
+  }
+  if (inheritedLogDir) env.CCV_LOG_DIR = inheritedLogDir;
+  env.CCV_IM_PLATFORM = id;
+  env.CCV_START_PORT = '7050';
+  env.CCV_MAX_PORT = '7099';
+  env.CCV_HOST = '127.0.0.1';
+  env.CCV_IM_DENY = '1';
+  return env;
+}
+
+/**
+ * 以 detached 子进程拉起一个 IM worker：`node cli.js --im <id> --no-open`，cwd=IM_<id>/。
+ * @param {string} id
+ * @param {{ spawnImpl?: Function }} [opts] 测试可注入 spawnImpl
+ * @returns {{ pid:number|undefined, dir:string, outLog:string }}
+ */
+export function spawnImProcess(id, opts = {}) {
+  const spawnImpl = opts.spawnImpl || nodeSpawn;
+  const dir = imDir(id);
+  mkdirSync(dir, { recursive: true });
+  const outLog = join(dir, 'process.out.log');
+
+  let stdio = 'ignore';
+  let fd = null;
+  try {
+    fd = openSync(outLog, 'a');
+    stdio = ['ignore', fd, fd];
+  } catch { /* 无法开日志文件时退化为 ignore */ }
+
+  const child = spawnImpl(resolveNodeBinary(), [CLI_JS, '--im', id, '--no-open'], {
+    cwd: dir,
+    env: buildChildEnv(id),
+    stdio,
+    detached: true,        // 自成进程组 leader → 停止时可整组杀，清理孙进程
+    windowsHide: true,
+  });
+  try { child.unref?.(); } catch { /* noop */ }
+  if (fd !== null) { try { closeSync(fd); } catch { /* 子进程已 dup */ } }
+  return { pid: child?.pid, dir, outLog };
+}
+
+/** 默认进程组杀（worker 是 detached group leader）；失败回退单进程。 */
+function defaultKill(pid, signal) {
+  try { process.kill(-pid, signal); return; } catch { /* not a group leader / windows */ }
+  try { process.kill(pid, signal); } catch { /* already gone */ }
+}
+
+/**
+ * 停止一个 IM worker：SIGTERM（worker cleanup 断适配器 + 释放锁）→ 轮询锁释放/进程死 →
+ * 超时 SIGKILL 进程组 → 清锁。
+ * @param {string} id
+ * @param {{ killImpl?:Function, isAlive?:Function, timeoutMs?:number, pollIntervalMs?:number }} [opts]
+ */
+export async function stopImProcess(id, opts = {}) {
+  const kill = opts.killImpl || defaultKill;
+  const isAlive = opts.isAlive || isPidAlive;
+  const timeoutMs = opts.timeoutMs ?? 8000;
+  const pollMs = opts.pollIntervalMs ?? 200;
+
+  const lock = readImLock(id);
+  if (!lock || !lock.pid || !isAlive(lock.pid)) {
+    clearImLock(id);
+    return { stopped: true, alreadyDead: true };
+  }
+
+  kill(lock.pid, 'SIGTERM');
+
+  const deadline = Date.now() + timeoutMs;
+  while (Date.now() < deadline) {
+    await sleep(pollMs);
+    const l = readImLock(id);
+    if (!l || l.pid !== lock.pid) return { stopped: true };   // worker 干净退出已释放锁
+    if (!isAlive(lock.pid)) { clearImLock(id); return { stopped: true }; }
+  }
+  kill(lock.pid, 'SIGKILL');
+  await sleep(pollMs);
+  clearImLock(id);
+  return { stopped: true, forced: true };
+}
+
+/**
+ * 查询某 IM 的进程状态（供 header chip / 路由）。
+ * @returns {Promise<{state:string, running:boolean, connected:boolean, pid:number|null, port:number|null, startedAt:string|null}>}
+ */
+export async function getImProcessStatus(id, opts = {}) {
+  const live = await getImLiveness(id, opts);
+  return {
+    state: live.state,
+    running: live.state === 'ready' || live.state === 'booting' || live.state === 'hung',
+    connected: live.state === 'ready' && !!live.connected,
+    pid: live.lock?.pid ?? null,
+    port: live.lock?.port ?? null,
+    startedAt: live.lock?.startedAt ?? null,
+  };
+}
+
+/**
+ * 启动时对账：为每个 enabled 且当前无活进程（state==='dead'）的 IM 拉起 worker。幂等（worker 侧 wx 锁兜底）。
+ * @returns {Promise<string[]>} 本次拉起的平台 id 列表
+ */
+export async function reconcileImProcesses(opts = {}) {
+  const spawned = [];
+  for (const id of listPlatforms()) {
+    let cfg;
+    try { cfg = loadConfig(id); } catch { continue; }
+    if (!cfg.enabled) continue;
+    const live = await getImLiveness(id, opts);
+    if (live.state === 'dead') {
+      try { spawnImProcess(id, opts); spawned.push(id); }
+      catch (e) { console.error(`[CC Viewer] reconcile spawn failed for IM ${id}:`, e?.message || e); }
+    }
+  }
+  return spawned;
+}
+
+export { defaultProbe };

package/server/lib/im-senders.js

@@ -0,0 +1,73 @@
+// IM 发送者身份持久化 —— 每个 IM worker 把解析到的「发送者 senderId → {name, avatar}」写到
+// ~/.claude/cc-viewer/IM_<id>/im-senders.json，供主进程的 /api/im/:platform/senders 读取，
+// 让「对话记录」弹窗能按 senderId 显示真实姓名 + 头像。
+//
+// 设计（与 im-lock.js 同风格）：
+// - 原子写：temp + renameSyncWithRetry，读方永不见半写 JSON。
+// - 读容忍：坏 / 缺 / 非对象一律降级为 {}，绝不抛。
+// - 容量上限：按 ts 保留最近 MAX_SENDERS 个，避免无限增长（群聊发送者可能很多）。
+// - 这些是本地数据（不入 preferences、不外发），与现有本地日志同级。
+import { openSync, closeSync, readFileSync, writeFileSync, unlinkSync, mkdirSync } from 'node:fs';
+import { join } from 'node:path';
+import { randomBytes } from 'node:crypto';
+import { imDir } from './im-lock.js';
+import { renameSyncWithRetry } from './file-api.js';
+
+export const MAX_SENDERS = 500;
+
+function sendersPath(id) { return join(imDir(id), 'im-senders.json'); }
+
+/** 读发送者映射；不存在 / 坏 JSON / 非对象 → {}。 */
+export function readSenders(id) {
+  try {
+    const o = JSON.parse(readFileSync(sendersPath(id), 'utf-8'));
+    return (o && typeof o === 'object' && !Array.isArray(o)) ? o : {};
+  } catch { return {}; }
+}
+
+/** 原子写整张映射（temp + rename）。 */
+function writeSenders(id, map) {
+  const dir = imDir(id);
+  mkdirSync(dir, { recursive: true });
+  const tmp = join(dir, `im-senders.json.tmp-${process.pid}-${randomBytes(4).toString('hex')}`);
+  try {
+    writeFileSync(tmp, JSON.stringify(map), { mode: 0o600 });
+    renameSyncWithRetry(tmp, sendersPath(id));
+  } catch (err) {
+    try { unlinkSync(tmp); } catch { /* best-effort */ }
+    throw err;
+  }
+}
+
+/**
+ * upsert 一个发送者。merge 进现有映射，盖上 ts；超过 MAX_SENDERS 时丢弃最旧的。
+ * name/avatar 任一为空都接受（部分平台只有名字没头像）。整段失败静默（持久化失败非致命）。
+ * @returns {boolean} 是否实际写入
+ */
+export function upsertSender(id, senderId, profile = {}) {
+  if (typeof senderId !== 'string' || !senderId) return false;
+  const name = profile.name != null ? String(profile.name) : null;
+  const avatar = profile.avatar != null ? String(profile.avatar) : null;
+  try {
+    const map = readSenders(id);
+    const prev = map[senderId];
+    // 无新信息（name/avatar 都没变且都已存在）→ 跳过写盘，省 IO。
+    if (prev && prev.name === name && prev.avatar === avatar) return false;
+    map[senderId] = { name, avatar, ts: tsNow() };
+
+    const keys = Object.keys(map);
+    if (keys.length > MAX_SENDERS) {
+      keys
+        .sort((a, b) => (map[a].ts || 0) - (map[b].ts || 0))
+        .slice(0, keys.length - MAX_SENDERS)
+        .forEach((k) => { delete map[k]; });
+    }
+    writeSenders(id, map);
+    return true;
+  } catch {
+    return false; // 持久化失败不影响消息流
+  }
+}
+
+// 单独抽出便于测试注入（避免直接用 Date.now 影响可测性，但保持简单）。
+function tsNow() { return Date.now(); }

package/server/lib/interceptor-core.js

@@ -247,7 +247,9 @@ export function createStreamAssembler() {
 export function findRecentLog(dir, projectName) {
   try {
     const files = readdirSync(dir)
-      .filter(f => f.startsWith(projectName + '_') && f.endsWith('.jsonl'))
+      // 排除 *_temp.jsonl：临时文件是未完成的写入态（resume 流程中途产物），
+      // 不应被当作"最近完整日志"（否则 _temp 因 sort 排在正式文件之后会被误选）。
+      .filter(f => f.startsWith(projectName + '_') && f.endsWith('.jsonl') && !f.endsWith('_temp.jsonl'))
       .sort()
       .reverse();
     if (files.length === 0) return null;

package/server/lib/perm-bridge.js

@@ -14,6 +14,7 @@
 import { readFileSync } from 'node:fs';
 import http from 'node:http';
 import https from 'node:https';
+import { evaluateImDeny } from './im-deny.js';
 
 const port = process.env.CCVIEWER_PORT;
 const rawProtocol = process.env.CCVIEWER_PROTOCOL;
@@ -62,6 +63,22 @@ if (!toolName || !toolInput) {
 const isPublishCmd = toolName === 'Bash' && toolInput.command &&
   /npm\s+publish/i.test(toolInput.command);
 
+// IM worker (skip-permissions) 硬拦截 —— 必须在下面的 bypass auto-allow 之前求值，
+// 否则 CCV_BYPASS_PERMISSIONS=1 会把一切先放行（见 plan §安全 2）。仅对 IM worker 生效。
+if (process.env.CCV_IM_DENY === '1') {
+  const verdict = evaluateImDeny(toolName, toolInput);
+  if (verdict.deny) {
+    process.stdout.write(JSON.stringify({
+      hookSpecificOutput: {
+        hookEventName: 'PreToolUse',
+        permissionDecision: 'deny',
+        permissionDecisionReason: `cc-viewer IM guard: ${verdict.reason}. 该操作在 IM 机器人(--dangerously-skip-permissions)下被禁止；请改用更安全的方式，或在回复中说明并请用户手动执行。`,
+      },
+    }) + '\n');
+    process.exit(0);
+  }
+}
+
 // Bypass mode: auto-allow all tools except publish commands
 // 使用显式 allow 而非 exit(1) fallback，避免 Claude Code 记录 hook error 日志
 if (process.env.CCV_BYPASS_PERMISSIONS === '1' && !isPublishCmd) {

package/server/pty-manager.js

@@ -2,7 +2,7 @@ import { resolveNativePath, LOG_DIR } from '../findcc.js';
 import { fileURLToPath } from 'node:url';
 import { join, dirname } from 'node:path';
 import { chmodSync, statSync } from 'node:fs';
-import { platform, arch } from 'node:os';
+import { platform, arch, homedir } from 'node:os';
 import { createRequire } from 'node:module';
 import { prepareEmbeddedShellSpawn, stripClaudeNoFlickerUnlessOptedIn } from './lib/terminal-env.js';
 
@@ -208,11 +208,32 @@ export async function spawnClaude(proxyPort, cwd, extraArgs = [], claudePath = n
 
   // 通过 --settings 注入 ANTHROPIC_BASE_URL，确保覆盖 settings.json 中的配置。
   // 仅覆盖 env.ANTHROPIC_BASE_URL，不影响其他 settings 字段。
-  const settingsJson = JSON.stringify({
+  const settingsObj = {
     env: {
       ANTHROPIC_BASE_URL: env.ANTHROPIC_BASE_URL
     }
-  });
+  };
+  // IM worker（skip-permissions）注入 permissions.deny 作为第二道防御（见 plan §安全 3）。
+  // 仅追加 deny 规则（deny 优先级最高、只会收紧不会放宽），不会破坏用户既有 permissions。
+  // 注：bypass 模式下 deny 是否仍被消费取决于 Claude Code 行为；真正可靠的强制层是
+  // perm-bridge.js 的 PreToolUse deny（CCV_IM_DENY）。此处为纵深防御的 best-effort 一层。
+  if (process.env.CCV_IM_DENY === '1') {
+    const home = homedir();
+    settingsObj.permissions = {
+      deny: [
+        'Bash(sudo:*)', 'Bash(rm -rf:*)', 'Bash(rm -fr:*)',
+        'Bash(git push:*)', 'Bash(npm publish:*)', 'Bash(ssh:*)', 'Bash(scp:*)',
+        `Read(${home}/.ssh/**)`, `Edit(${home}/.ssh/**)`, `Write(${home}/.ssh/**)`,
+        `Read(${home}/.aws/**)`, `Edit(${home}/.aws/**)`, `Write(${home}/.aws/**)`,
+        // 精确到文件：保护 deny 机制本身(settings/hooks)与 IM 密钥库(preferences.json)，
+        // 但不封禁整个 ~/.claude —— worker 的工作目录就在 ~/.claude/cc-viewer/IM_<id>/ 下，需保持可写。
+        `Edit(${home}/.claude/settings.json)`, `Write(${home}/.claude/settings.json)`,
+        `Edit(${home}/.claude/settings.local.json)`, `Write(${home}/.claude/settings.local.json)`,
+        `Edit(${home}/.claude/cc-viewer/preferences.json)`, `Write(${home}/.claude/cc-viewer/preferences.json)`,
+      ],
+    };
+  }
+  const settingsJson = JSON.stringify(settingsObj);
 
   // 注入 --thinking-display summarized；以下任一情况跳过注入：
   // - 路径在拒绝集里（上次因此 crash 过）

package/server/routes/dingtalk.js

@@ -18,13 +18,19 @@ function readBody(req, deps, cb) {
   req.on('end', () => cb(body));
 }
 
-function dingtalkStatus(req, res, parsedUrl, isLocal, deps) {
-  const conn = deps.dingtalk.getBridgeStatus();
+async function dingtalkStatus(req, res, parsedUrl, isLocal, deps) {
+  // 与 /api/im/:platform/status 一致：worker 报自身在进程适配器状态（manager 据此探活）；
+  // 主进程经 manager 报 detached worker 的进程/连接状态。
+  let conn;
+  let processInfo = null;
+  if (deps.dingtalk.isWorker) {
+    conn = deps.dingtalk.getBridgeStatus();
+  } else {
+    processInfo = await deps.dingtalk.getProcessStatus();
+    conn = { running: processInfo.running, connected: processInfo.connected };
+  }
   res.writeHead(200, JSON_HEADERS);
   if (!isLocal) {
-    // Loopback gate: a token-authorized LAN client must not see the appKey, the staffId
-    // allowlist, the bound conversation id, or raw error strings. Expose only the minimum the
-    // header status chip needs. (config/test are already loopback-only.)
     res.end(JSON.stringify({
       enabled: loadDingTalkState().enabled,
       hasSecret: loadDingTalkState().hasSecret,
@@ -32,9 +38,14 @@ function dingtalkStatus(req, res, parsedUrl, isLocal, deps) {
     }));
     return;
   }
-  // 本机(127.0.0.1)= admin：附带明文 appSecret 供本人查阅/复制（镜像 /api/auth/state 的密码、
-  // /api/proxy-profiles 的 apiKey 策略）。上方 !isLocal 分支只下发 hasSecret，绝不下发 secret。
-  res.end(JSON.stringify({ ...loadDingTalkState(), appSecret: loadDingTalkConfig().appSecret, connection: conn }));
+  // 本机(127.0.0.1)= admin / manager 探活：附带明文 appSecret 与 pid（供身份匹配）。
+  res.end(JSON.stringify({
+    ...loadDingTalkState(),
+    appSecret: loadDingTalkConfig().appSecret,
+    connection: conn,
+    process: processInfo,
+    pid: deps.dingtalk.isWorker ? process.pid : (processInfo?.pid ?? null),
+  }));
 }
 
 function dingtalkConfigPost(req, res, parsedUrl, isLocal, deps) {
@@ -44,7 +55,7 @@ function dingtalkConfigPost(req, res, parsedUrl, isLocal, deps) {
     res.end(JSON.stringify({ error: 'Loopback only' }));
     return;
   }
-  readBody(req, deps, (body) => {
+  readBody(req, deps, async (body) => {
     let incoming;
     try { incoming = JSON.parse(body); }
     catch {
@@ -52,8 +63,17 @@ function dingtalkConfigPost(req, res, parsedUrl, isLocal, deps) {
       res.end(JSON.stringify({ error: 'Invalid JSON' }));
       return;
     }
+    // 发送者白名单为非必填：启用时若 allowStaffIds 为空不再硬拦截（前端弹安全警告）。worker 以
+    // --dangerously-skip-permissions 运行，空白名单运行期退化为 bind-first-conversation。打一条
+    // 服务端审计（curl/headless 启用走不到前端 toast）；PreToolUse permissions.deny 硬拦截仍生效。
+    const staff = Array.isArray(incoming.allowStaffIds)
+      ? incoming.allowStaffIds.filter((s) => typeof s === 'string' && s.trim())
+      : [];
+    if (incoming.enabled && staff.length === 0) {
+      console.warn('[CC Viewer] IM dingtalk enabled with EMPTY allowlist — bind-first-conversation; the first conversation to message can drive this --dangerously-skip-permissions session');
+    }
     // saveDingTalkConfig preserves the stored secret when appSecret is empty/omitted.
-    saveDingTalkConfig({
+    const saved = saveDingTalkConfig({
       enabled: incoming.enabled,
       appKey: incoming.appKey,
       appSecret: incoming.appSecret,
@@ -61,10 +81,15 @@ function dingtalkConfigPost(req, res, parsedUrl, isLocal, deps) {
       maxChunkChars: incoming.maxChunkChars,
       blockOnSkipPermissions: incoming.blockOnSkipPermissions,
     });
-    // Apply immediately: stop the old Stream connection and (re)start with the new config.
-    Promise.resolve(deps.dingtalk.reloadBridge()).catch(() => {});
+    // 驱动进程管理器（替代旧的在进程 reloadBridge）：启用→重启 worker，停用→停 worker。
+    try {
+      if (saved?.enabled ?? incoming.enabled) await deps.dingtalk.restartProcess();
+      else await deps.dingtalk.stopProcess();
+    } catch (e) {
+      console.error('[CC Viewer] IM config apply failed for dingtalk:', e?.message || e);
+    }
     res.writeHead(200, JSON_HEADERS);
-    res.end(JSON.stringify({ ...loadDingTalkState(), connection: deps.dingtalk.getBridgeStatus() }));
+    res.end(JSON.stringify({ ...loadDingTalkState(), connection: { running: !!(saved?.enabled ?? incoming.enabled), connected: false } }));
   });
 }