@vitex/nook 0.0.0-stage → 0.1.0

package/src/backend/service.js

@@ -0,0 +1,337 @@
+/**
+ * 后台服务管理。
+ *
+ * macOS 使用 launchd，Linux 使用 systemd 用户服务（Ubuntu 24.04+）。
+ * 所有路径都根据当前运行环境动态生成，不写死任何本机路径。
+ */
+import { execFileSync } from 'node:child_process';
+import fs from 'node:fs';
+import os from 'node:os';
+import path from 'node:path';
+import { PROJECT_DIR, getLogsDir, getNodeBin, ensureHome } from './config.js';
+import { log, logErr } from './logging.js';
+
+const LABEL_DEFAULT = 'com.nook.server';
+const IS_MACOS = process.platform === 'darwin';
+const IS_LINUX = process.platform === 'linux';
+
+function label() {
+  return process.env.NOOK_SERVICE_LABEL || LABEL_DEFAULT;
+}
+
+function manager() {
+  if (IS_MACOS) return 'launchd';
+  if (IS_LINUX) return 'systemd';
+  return 'unsupported';
+}
+
+function serviceFile() {
+  if (IS_MACOS) {
+    return path.join(os.homedir(), 'Library', 'LaunchAgents', `${label()}.plist`);
+  }
+  if (IS_LINUX) {
+    const configHome = process.env.XDG_CONFIG_HOME
+      ? path.resolve(process.env.XDG_CONFIG_HOME)
+      : path.join(os.homedir(), '.config');
+    return path.join(configHome, 'systemd', 'user', `${label()}.service`);
+  }
+  return '';
+}
+
+function scriptPath() {
+  return path.resolve(process.env.NOOK_LAUNCHER || path.join(PROJECT_DIR, 'bin', 'nook.js'));
+}
+
+function run(cmd, args, { allowFailure = false, ...options } = {}) {
+  try {
+    return execFileSync(cmd, args, {
+      encoding: 'utf8',
+      stdio: ['ignore', 'pipe', 'pipe'],
+      ...options,
+    });
+  } catch (e) {
+    if (allowFailure) return String(e?.stdout || '');
+    const detail = String(e?.stderr || e?.message || e).trim();
+    throw new Error(`${cmd} ${args.join(' ')} 失败：${detail}`);
+  }
+}
+
+function systemctl(args, options = {}) {
+  const env = { ...process.env };
+  if (!env.XDG_RUNTIME_DIR && typeof process.getuid === 'function') {
+    const runtimeDir = `/run/user/${process.getuid()}`;
+    if (fs.existsSync(runtimeDir)) env.XDG_RUNTIME_DIR = runtimeDir;
+  }
+  return run('systemctl', ['--user', ...args], { env, ...options });
+}
+
+function hasSystemd() {
+  if (!IS_LINUX || !fs.existsSync('/run/systemd/system')) return false;
+  try {
+    execFileSync('systemctl', ['--version'], { stdio: 'ignore' });
+    return true;
+  } catch {
+    return false;
+  }
+}
+
+function requireSupported() {
+  if (IS_MACOS) return;
+  if (IS_LINUX) {
+    if (!hasSystemd()) throw new Error('Linux 后台服务需要 systemd（Ubuntu 24.04+）');
+    return;
+  }
+  throw new Error(`当前平台不支持后台服务：${process.platform}`);
+}
+
+function xml(s) {
+  return String(s).replace(/&/g, '&amp;').replace(/</g, '&lt;').replace(/>/g, '&gt;');
+}
+
+function systemdEscape(s) {
+  return String(s)
+    .replace(/\\/g, '\\\\')
+    .replace(/"/g, '\\"')
+    .replace(/\n/g, '\\n')
+    .replace(/\r/g, '\\r')
+    .replace(/\t/g, '\\t')
+    .replace(/%/g, '%%');
+}
+
+function systemdQuote(s) {
+  return `"${systemdEscape(s)}"`;
+}
+
+export function renderPlist({ label: serviceLabel, nodeBin, script, workingDir, port, host, stdout, stderr, env }) {
+  const envLines = Object.entries(env || {}).map(([k, v]) => `        <key>${xml(k)}</key>\n        <string>${xml(v)}</string>`);
+  return [
+    '<?xml version="1.0" encoding="UTF-8"?>',
+    '<!DOCTYPE plist PUBLIC "-//Apple//DTD PLIST 1.0//EN" "http://www.apple.com/DTDs/PropertyList-1.0.dtd">',
+    '<plist version="1.0">',
+    '<dict>',
+    `    <key>Label</key>\n    <string>${xml(serviceLabel)}</string>`,
+    '    <key>ProgramArguments</key>',
+    '    <array>',
+    `        <string>${xml(nodeBin)}</string>`,
+    `        <string>${xml(script)}</string>`,
+    '        <string>start</string>',
+    '        <string>--port</string>',
+    `        <string>${xml(String(port))}</string>`,
+    '        <string>--host</string>',
+    `        <string>${xml(String(host))}</string>`,
+    '    </array>',
+    '    <key>WorkingDirectory</key>',
+    `    <string>${xml(workingDir)}</string>`,
+    '    <key>RunAtLoad</key>\n    <true/>',
+    '    <key>KeepAlive</key>\n    <true/>',
+    '    <key>ThrottleInterval</key>\n    <integer>5</integer>',
+    '    <key>EnvironmentVariables</key>',
+    '    <dict>',
+    ...envLines,
+    '    </dict>',
+    `    <key>StandardOutPath</key>\n    <string>${xml(stdout)}</string>`,
+    `    <key>StandardErrorPath</key>\n    <string>${xml(stderr)}</string>`,
+    '</dict>',
+    '</plist>',
+    '',
+  ].join('\n');
+}
+
+export function renderSystemdUnit({ label: serviceLabel, nodeBin, script, workingDir, port, host, env }) {
+  const envLines = Object.entries(env || {})
+    .map(([k, v]) => `Environment=${systemdQuote(`${k}=${v}`)}`);
+  return [
+    '[Unit]',
+    'Description=Nook - Pi web interface',
+    '',
+    '[Service]',
+    'Type=simple',
+    `WorkingDirectory=${systemdQuote(workingDir)}`,
+    `ExecStart=${systemdQuote(nodeBin)} ${systemdQuote(script)} start --host ${systemdQuote(host)} --port ${systemdQuote(String(port))}`,
+    'Restart=always',
+    'RestartSec=5',
+    ...envLines,
+    '',
+    '[Install]',
+    'WantedBy=default.target',
+    '',
+  ].join('\n');
+}
+
+function isLoaded() {
+  if (IS_MACOS) {
+    try {
+      const out = run('launchctl', ['list', label()], { allowFailure: true });
+      return out.includes(label());
+    } catch {
+      return false;
+    }
+  }
+  if (IS_LINUX) {
+    if (!hasSystemd()) return false;
+    return systemctl(['is-active', label()], { allowFailure: true }).trim() === 'active';
+  }
+  return false;
+}
+
+function plistArgs(text) {
+  const host = text.match(/<string>--host<\/string>\s*<string>([^<]*)<\/string>/)?.[1] || '';
+  const port = Number(text.match(/<string>--port<\/string>\s*<string>([^<]*)<\/string>/)?.[1] || 0);
+  return { host, port: Number.isFinite(port) ? port : 0 };
+}
+
+function unitArgs(text) {
+  const exec = text.match(/^ExecStart=(.*)$/m)?.[1] || '';
+  const hostMatch = exec.match(/--host\s+(?:"((?:\\.|[^"\\])*)"|(\S+))/);
+  const portMatch = exec.match(/--port\s+(?:"(\d+)"|(\d+))/);
+  const port = Number(portMatch?.[1] || portMatch?.[2] || 0);
+  return {
+    host: hostMatch?.[1] ?? hostMatch?.[2] ?? '',
+    port: Number.isFinite(port) ? port : 0,
+  };
+}
+
+function serviceArgs(file) {
+  try {
+    const text = fs.readFileSync(file, 'utf8');
+    return IS_MACOS ? plistArgs(text) : unitArgs(text);
+  } catch {
+    return { host: '', port: 0 };
+  }
+}
+
+export function serviceStatus() {
+  const file = serviceFile();
+  const installed = Boolean(file) && fs.existsSync(file);
+  const args = installed ? serviceArgs(file) : { host: '', port: 0 };
+  return {
+    label: label(),
+    manager: manager(),
+    file,
+    installed,
+    running: isLoaded(),
+    host: args.host,
+    port: args.port,
+  };
+}
+
+function enableLinger() {
+  try {
+    execFileSync('loginctl', ['enable-linger', os.userInfo().username], { stdio: 'ignore' });
+  } catch {
+    logErr(`提示：未能启用 linger；未登录时服务可能不会运行。可手动执行：sudo loginctl enable-linger ${os.userInfo().username}`);
+  }
+}
+
+export function installService({ port, host }) {
+  requireSupported();
+  ensureHome();
+  const file = serviceFile();
+  const logs = getLogsDir();
+  fs.mkdirSync(path.dirname(file), { recursive: true });
+  fs.mkdirSync(logs, { recursive: true });
+
+  const env = {
+    NOOK_SERVICE: '1',
+    ...(process.env.PATH ? { PATH: process.env.PATH } : {}),
+    ...(process.env.NOOK_HOME ? { NOOK_HOME: process.env.NOOK_HOME } : {}),
+  };
+
+  const content = IS_MACOS
+    ? renderPlist({
+      label: label(),
+      nodeBin: getNodeBin(),
+      script: scriptPath(),
+      workingDir: PROJECT_DIR,
+      port,
+      host,
+      stdout: path.join(logs, 'stdout.log'),
+      stderr: path.join(logs, 'stderr.log'),
+      env,
+    })
+    : renderSystemdUnit({
+      label: label(),
+      nodeBin: getNodeBin(),
+      script: scriptPath(),
+      workingDir: PROJECT_DIR,
+      port,
+      host,
+      env,
+    });
+
+  fs.writeFileSync(file, content);
+  log(`已写入 ${file}`);
+
+  if (IS_MACOS) {
+    run('launchctl', ['unload', file], { allowFailure: true });
+    run('launchctl', ['load', file]);
+  } else {
+    enableLinger();
+    systemctl(['daemon-reload']);
+    systemctl(['enable', label()]);
+    systemctl(['restart', label()]);
+  }
+
+  log(`服务已安装并启动（${isLoaded() ? 'running' : 'stopped'}）`);
+  return serviceStatus();
+}
+
+export function uninstallService() {
+  const file = serviceFile();
+  if (file && fs.existsSync(file)) {
+    if (IS_MACOS) {
+      run('launchctl', ['unload', file], { allowFailure: true });
+    } else if (IS_LINUX) {
+      systemctl(['disable', '--now', label()], { allowFailure: true });
+    }
+    fs.rmSync(file, { force: true });
+    if (IS_LINUX) systemctl(['daemon-reload'], { allowFailure: true });
+    log('服务已卸载');
+  } else {
+    log('没有已安装的服务');
+  }
+  return serviceStatus();
+}
+
+export function startService() {
+  const file = serviceFile();
+  if (!file || !fs.existsSync(file)) throw new Error('尚未安装服务，请先运行 nook service install');
+  if (isLoaded()) { log('服务已在运行'); return serviceStatus(); }
+  if (IS_MACOS) {
+    run('launchctl', ['load', file]);
+  } else {
+    systemctl(['start', label()]);
+  }
+  log('服务已启动');
+  return serviceStatus();
+}
+
+export function stopService() {
+  const file = serviceFile();
+  if (!file || !fs.existsSync(file)) { log('尚未安装服务'); return serviceStatus(); }
+  if (!isLoaded()) { log('服务未在运行'); return serviceStatus(); }
+  try {
+    if (IS_MACOS) {
+      run('launchctl', ['unload', file]);
+    } else {
+      systemctl(['stop', label()]);
+    }
+    log('服务已停止');
+  } catch (e) {
+    logErr(e.message);
+  }
+  return serviceStatus();
+}
+
+export function restartService() {
+  const file = serviceFile();
+  if (!file || !fs.existsSync(file)) throw new Error('尚未安装服务，请先运行 nook service install');
+  if (IS_MACOS) {
+    run('launchctl', ['unload', file], { allowFailure: true });
+    run('launchctl', ['load', file]);
+  } else {
+    systemctl(['restart', label()]);
+  }
+  log('服务已重启');
+  return serviceStatus();
+}

package/src/backend/sessions.js

@@ -0,0 +1,187 @@
+/**
+ * Pi 会话读取。
+ *
+ * Nook 不发明会话格式，只通过 Pi SDK 的 SessionManager 读写：
+ *   ~/.pi/agent/sessions/<encoded-cwd>/
+ *
+ * 新建会话也写入 Pi 默认目录，所以 Pi CLI 的 `pi -r` 也能看到。
+ */
+import fs from 'node:fs';
+import { execFileSync } from 'node:child_process';
+import { SessionManager } from '@earendil-works/pi-coding-agent';
+import { logErr } from './logging.js';
+import { stripAnsi } from './text.js';
+
+const FEISHU_PREFIX = /^当前用户通过飞书通道以机器人的方式和你沟通，用户的 open_id 是\s*\S+\s*/;
+const TOOL_RESULT_LIMIT = 8000;
+
+export function newSessionId() {
+  return cryptoRandomUUID();
+}
+
+function cryptoRandomUUID() {
+  return globalThis.crypto?.randomUUID?.() || `${Date.now()}-${Math.random().toString(16).slice(2)}`;
+}
+
+export function cleanUserText(text) {
+  return String(text ?? '').replace(FEISHU_PREFIX, '').trim();
+}
+
+function textFromContent(content) {
+  if (typeof content === 'string') return content;
+  if (!Array.isArray(content)) return '';
+  return content
+    .map((block) => {
+      if (!block || typeof block !== 'object') return '';
+      if (block.type === 'text') return block.text || '';
+      if (block.type === 'image') return '[图片]';
+      return '';
+    })
+    .filter(Boolean)
+    .join('\n');
+}
+
+export function truncateText(text, limit = TOOL_RESULT_LIMIT) {
+  const s = stripAnsi(text);
+  if (s.length <= limit) return { text: s, truncated: false, length: s.length };
+  return { text: s.slice(0, limit), truncated: true, length: s.length };
+}
+
+function titleFrom(firstMessage, name) {
+  const named = String(name || '').replace(/\s+/g, ' ').trim();
+  if (named) return named;
+  const clean = cleanUserText(firstMessage).replace(/\s+/g, ' ').trim();
+  return clean || '新会话';
+}
+
+function serializeInfo(info) {
+  return {
+    id: info.id,
+    name: info.name || '',
+    title: titleFrom(info.firstMessage, info.name),
+    firstMessage: cleanUserText(info.firstMessage),
+    created: info.created instanceof Date ? info.created.toISOString() : info.created,
+    modified: info.modified instanceof Date ? info.modified.toISOString() : info.modified,
+    messageCount: info.messageCount || 0,
+    path: info.path,
+  };
+}
+
+/** 列出项目下由 Pi 管理的会话 */
+export async function listSessions(project) {
+  try {
+    const infos = await SessionManager.list(project.path);
+    return infos
+      .map(serializeInfo)
+      .sort((a, b) => String(b.modified).localeCompare(String(a.modified)));
+  } catch (e) {
+    logErr('读取 Pi 会话目录失败：', e?.message || e);
+    return [];
+  }
+}
+
+export function isValidSessionId(id) {
+  return /^[a-zA-Z0-9_-]{1,120}$/.test(String(id || ''));
+}
+
+/** 找到会话文件；找不到返回 null（表示这是一个尚未落盘的新会话） */
+export function findSessionFile(project, sessionId) {
+  if (!isValidSessionId(sessionId)) return null;
+  try {
+    const file = SessionManager.findById(project.path, sessionId);
+    if (file) return { path: file };
+  } catch { }
+  return null;
+}
+
+export function openSessionManager(project, sessionId, file = findSessionFile(project, sessionId)) {
+  if (file) return SessionManager.open(file.path, undefined, project.path);
+  return SessionManager.create(project.path, undefined, { id: sessionId });
+}
+
+/** 把 Pi 的 session tree 转成 UI 消息流 */
+export function readTranscript(sm) {
+  const entries = sm.buildContextEntries();
+  const messages = [];
+  const toolIndex = new Map();
+
+  for (const entry of entries) {
+    if (!entry || entry.type !== 'message') continue;
+    const msg = entry.message;
+    if (!msg || typeof msg !== 'object') continue;
+
+    if (msg.role === 'user') {
+      const text = cleanUserText(textFromContent(msg.content));
+      if (!text) continue;
+      messages.push({ role: 'user', text, timestamp: msg.timestamp || null });
+      continue;
+    }
+
+    if (msg.role === 'assistant') {
+      const blocks = [];
+      for (const block of Array.isArray(msg.content) ? msg.content : []) {
+        if (!block || typeof block !== 'object') continue;
+        if (block.type === 'thinking') {
+          if (block.thinking && !block.redacted) blocks.push({ type: 'thinking', text: block.thinking });
+        } else if (block.type === 'text') {
+          if (block.text) blocks.push({ type: 'text', text: block.text });
+        } else if (block.type === 'toolCall') {
+          const tool = {
+            type: 'tool',
+            id: block.id || '',
+            name: block.name || 'tool',
+            args: block.arguments ?? {},
+            result: null,
+            isError: false,
+            resultTruncated: false,
+          };
+          blocks.push(tool);
+          if (tool.id) toolIndex.set(tool.id, tool);
+        }
+      }
+      if (msg.stopReason === 'error' && msg.errorMessage) {
+        blocks.push({ type: 'error', text: msg.errorMessage });
+      }
+      if (blocks.length === 0) continue;
+      messages.push({ role: 'assistant', blocks, timestamp: msg.timestamp || null });
+      continue;
+    }
+
+    if (msg.role === 'toolResult') {
+      const tool = toolIndex.get(msg.toolCallId);
+      if (!tool) continue;
+      const raw = stripAnsi(textFromContent(msg.content));
+      const cut = truncateText(raw);
+      tool.result = cut.text;
+      tool.resultTruncated = cut.truncated;
+      tool.resultLength = cut.length;
+      tool.isError = Boolean(msg.isError);
+      continue;
+    }
+
+    if (msg.role === 'custom' && msg.display) {
+      const text = cleanUserText(textFromContent(msg.content));
+      if (text) messages.push({ role: 'assistant', blocks: [{ type: 'text', text }], timestamp: msg.timestamp || null });
+    }
+  }
+
+  const name = sm.getSessionName?.() || '';
+  const firstUser = messages.find((m) => m.role === 'user')?.text || '';
+  return {
+    id: sm.getSessionId(),
+    name,
+    title: titleFrom(firstUser, name),
+    cwd: sm.getCwd(),
+    messages,
+  };
+}
+
+export function deleteSessionFile(file) {
+  if (!file?.path) return false;
+  try {
+    execFileSync('trash', [file.path], { stdio: 'ignore', timeout: 5000 });
+    return true;
+  } catch {
+    try { fs.rmSync(file.path, { force: true }); return true; } catch (e) { logErr('删除会话失败：', e?.message || e); return false; }
+  }
+}

package/src/backend/text.js

@@ -0,0 +1,4 @@
+/** 去掉终端 ANSI 控制序列，避免工具结果在 H5 里出现乱码。 */
+export function stripAnsi(text) {
+  return String(text ?? '').replace(/\u001b\[[0-?]*[ -/]*[@-~]/g, '');
+}