xiaozuoassistant 0.1.45 → 0.1.47

package/bin/cli.js

@@ -23,6 +23,7 @@ const packageRoot = path.resolve(__dirname, '..');
 
 const args = process.argv.slice(2);
 const command = args[0];
+const commandArgs = args.slice(1);
 
 const EXPORT_FILENAME = 'xiaozuoAssistant-backup.tar.gz';
 
@@ -58,6 +59,22 @@ function getPidFilePath() {
   return path.join(CWD, 'logs', 'server.pid');
 }
 
+function hasFlag(flag) {
+  return commandArgs.includes(flag);
+}
+
+function getFlagValue(name) {
+  const idx = commandArgs.findIndex(a => a === name);
+  if (idx < 0) return null;
+  const next = commandArgs[idx + 1];
+  if (!next || next.startsWith('-')) return null;
+  return next;
+}
+
+function getRegistry() {
+  return getFlagValue('--registry') || process.env.npm_config_registry || 'https://registry.npmjs.org';
+}
+
 function isProcessRunning(pid) {
   try {
     process.kill(pid, 0);
@@ -186,6 +203,131 @@ async function stopServer() {
   console.error('[CLI] ❌ 停止失败：端口仍在占用。');
 }
 
+async function runCommand(cmd, cmdArgs, options = {}) {
+  const child = spawn(cmd, cmdArgs, { stdio: 'inherit', ...options });
+  const code = await new Promise((resolve) => child.on('close', resolve));
+  return typeof code === 'number' ? code : 1;
+}
+
+async function runWithSudoIfNeeded(cmd, cmdArgs, options = {}) {
+  const code = await runCommand(cmd, cmdArgs, options);
+  if (code === 0) return 0;
+
+  const canSudo = process.platform !== 'win32' && process.stdin.isTTY;
+  if (!canSudo) return code;
+
+  console.log('[CLI] 权限不足，尝试使用 sudo 继续...');
+  return await runCommand('sudo', [cmd, ...cmdArgs], options);
+}
+
+async function updateApp() {
+  const registry = getRegistry();
+  const port = getPortFromConfig();
+  const pidFile = getPidFilePath();
+
+  const runningByPid = fs.existsSync(pidFile) && (() => {
+    try {
+      const pid = Number(fs.readFileSync(pidFile, 'utf-8').trim());
+      return Number.isFinite(pid) && pid > 0 && isProcessRunning(pid);
+    } catch {
+      return false;
+    }
+  })();
+  const runningByPort = await isPortOpen(port);
+  const wasRunning = runningByPid || runningByPort;
+
+  if (wasRunning) {
+    console.log('[CLI] 检测到服务正在运行，更新前将自动停止...');
+    await stopServer();
+  }
+
+  console.log(`[CLI] 正在更新 xiaozuoassistant（registry=${registry}）...`);
+  const code = await runWithSudoIfNeeded('npm', ['install', '-g', 'xiaozuoassistant@latest', `--registry=${registry}`]);
+  if (code !== 0) {
+    console.error('[CLI] ❌ 更新失败。');
+    if (wasRunning) {
+      console.log('[CLI] 更新失败，尝试恢复启动旧版本服务...');
+      await runCommand('node', [path.join(packageRoot, 'bin', 'cli.js'), 'start'], { cwd: CWD });
+    }
+    process.exit(code);
+  }
+
+  console.log('[CLI] ✅ 更新完成。');
+
+  if (wasRunning) {
+    console.log('[CLI] 正在自动重启服务...');
+    const restartCode = await runCommand('xiaozuoAssistant', ['start'], { cwd: CWD });
+    process.exit(restartCode);
+  }
+}
+
+async function removeApp() {
+  const registry = getRegistry();
+  const targets = ['config.json', 'memories', 'data', 'logs', 'sessions', 'workspace']
+    .map(p => path.join(CWD, p));
+
+  const doRemove = async () => {
+    await stopServer();
+
+    console.log('[CLI] 正在删除本地数据（当前目录）...');
+    for (const target of targets) {
+      try {
+        if (fs.existsSync(target)) {
+          fs.rmSync(target, { recursive: true, force: true });
+          console.log(`[CLI] 已删除：${target}`);
+        }
+      } catch (e) {
+        console.error(`[CLI] 删除失败：${target}`);
+        console.error(e);
+      }
+    }
+
+    console.log('[CLI] 正在卸载全局包 xiaozuoassistant...');
+    const code = await runWithSudoIfNeeded('npm', ['uninstall', '-g', 'xiaozuoassistant', `--registry=${registry}`]);
+    if (code !== 0) {
+      console.error('[CLI] ❌ 卸载失败。');
+      process.exit(code);
+    }
+
+    console.log('[CLI] ✅ 已卸载并清理数据。');
+  };
+
+  if (hasFlag('--yes') || hasFlag('-y')) {
+    await doRemove();
+    return;
+  }
+
+  console.log('[CLI] remove 将执行以下操作：');
+  console.log('  1) 停止后台服务');
+  console.log('  2) 删除当前目录下的数据：');
+  for (const target of targets) console.log(`     - ${target}`);
+  console.log('  3) 卸载全局 npm 包：xiaozuoassistant');
+  process.stdout.write('确认继续？(y/N) ');
+  process.stdin.once('data', async (data) => {
+    const answer = data.toString().trim().toLowerCase();
+    if (answer === 'y' || answer === 'yes') {
+      await doRemove();
+    } else {
+      console.log('已取消。');
+    }
+    process.exit(0);
+  });
+}
+
+function printVersion() {
+  try {
+    const pkgPath = path.join(packageRoot, 'package.json');
+    const pkg = JSON.parse(fs.readFileSync(pkgPath, 'utf-8'));
+    console.log(`${pkg.name}@${pkg.version}`);
+    console.log('Package Root:', packageRoot);
+    console.log('Node Version:', process.version);
+  } catch {
+    console.log('xiaozuoassistant@unknown');
+    console.log('Package Root:', packageRoot);
+    console.log('Node Version:', process.version);
+  }
+}
+
 async function exportData() {
     console.log('📦 Starting data export...');
     
@@ -420,6 +562,12 @@ if (command === 'start') {
     console.log('Package Root:', packageRoot);
     console.log('Node Version:', process.version);
     console.log('Doctor check complete.');
+} else if (command === 'version') {
+    printVersion();
+} else if (command === 'update') {
+    updateApp();
+} else if (command === 'remove') {
+    removeApp();
 } else if (command === 'export') {
     exportData();
 } else if (command === 'import') {
@@ -432,6 +580,9 @@ if (command === 'start') {
     console.log('  start   Start the xiaozuoAssistant server');
     console.log('  stop    Stop the xiaozuoAssistant server');
     console.log('  doctor  Check the health and configuration');
+    console.log('  version Print package version and environment');
+    console.log('  update  Update xiaozuoassistant and auto-restart if running');
+    console.log('  remove  Uninstall and delete data in current directory');
     console.log('  export  Backup local data (config, memories) to a file');
     console.log('  import  Restore data from a backup file');
 }

package/package.json

@@ -2,7 +2,7 @@
   "name": "xiaozuoassistant",
   "private": false,
   "description": "Your personal, locally-hosted AI assistant for office productivity.",
-  "version": "0.1.45",
+  "version": "0.1.47",
   "author": "mantle.lau",
   "license": "MIT",
   "repository": {