xiaozuoassistant 0.2.20 → 0.2.22

package/bin/cli.js

@@ -1,896 +1,896 @@
-#!/usr/bin/env node
-
-import { spawn } from 'child_process';
-import path from 'path';
-import { fileURLToPath } from 'url';
-import fs from 'fs';
-import os from 'os';
-import { createGzip } from 'zlib';
-import { pipeline } from 'stream';
-import { promisify } from 'util';
-import * as tar from 'tar';
-import http from 'http';
-import https from 'https';
-import net from 'net';
-
-const pipe = promisify(pipeline);
-
-const __filename = fileURLToPath(import.meta.url);
-const __dirname = path.dirname(__filename);
-
-// Determine the root directory of the package
-// When running from source: bin/cli.js -> ../
-// When installed globally: lib/node_modules/xiaozuoassistant/bin/cli.js -> ../
-const packageRoot = path.resolve(__dirname, '..');
-
-const binName = path.basename(process.argv[1] || 'xiaozuoAssistant');
-
-const args = process.argv.slice(2);
-const command = args[0];
-const commandArgs = args.slice(1);
-
-const EXPORT_FILENAME_PREFIX = 'xiaozuoAssistant-backup';
-const PRODUCTION_PORT = 3021; // 生产环境（CLI）默认端口
-const DEV_PORT = 3001; // 开发环境默认端口
-
-// Helper to get user's current working directory where they ran the command
-// This is kept only for display/backward compatibility; runtime data is stored in APP_HOME.
-const CWD = process.cwd();
-
-function getAppHome() {
-  const fromFlag = getFlagValue('--home');
-  const fromEnv = process.env.XIAOZUOASSISTANT_HOME;
-  const base = (fromFlag || fromEnv || path.join(os.homedir(), '.xiaozuoassistant')).trim();
-  return path.resolve(base);
-}
-
-const APP_HOME = getAppHome();
-
-// Unified runtime data directory
-const DATA_PATHS = [
-  'config.json',
-  'memories',
-  'data',
-  'logs',
-  'sessions',
-  'workspace',
-  'plugins'
-];
-
-function ensureAppHome() {
-  try {
-    fs.mkdirSync(APP_HOME, { recursive: true });
-  } catch (e) {
-    console.error(`[CLI] 无法创建数据目录：${APP_HOME}`);
-    console.error(e);
-    process.exit(1);
-  }
-}
-
-function ensureDefaultConfig() {
-  const configPath = path.join(APP_HOME, 'config.json');
-  
-  // 如果配置文件已存在，检查是否需要迁移端口（例如从 3001 迁移到 3021）
-  if (fs.existsSync(configPath)) {
-    try {
-        const configContent = fs.readFileSync(configPath, 'utf-8');
-        const config = JSON.parse(configContent);
-        // 如果端口是旧的默认值 3001，自动更新为 3021
-        if (config.server && config.server.port === DEV_PORT) {
-            console.log(`[CLI] 检测到旧端口配置 (${DEV_PORT})，正在迁移到生产环境端口 (${PRODUCTION_PORT})...`);
-            config.server.port = PRODUCTION_PORT;
-            fs.writeFileSync(configPath, JSON.stringify(config, null, 2));
-            console.log(`[CLI] ✅ 端口已更新为 ${PRODUCTION_PORT}`);
-        }
-    } catch (e) {
-        console.warn('[CLI] 检查现有配置失败:', e);
-    }
-    return;
-  }
-
-  // 如果配置文件不存在，创建新的默认配置
-  const templatePath = path.join(packageRoot, 'config.json');
-  try {
-    if (fs.existsSync(templatePath)) {
-      // 复制模板，但强制修改端口为 3021
-      const templateContent = fs.readFileSync(templatePath, 'utf-8');
-      const templateConfig = JSON.parse(templateContent);
-      
-      if (!templateConfig.server) templateConfig.server = {};
-      templateConfig.server.port = PRODUCTION_PORT;
-      
-      fs.writeFileSync(configPath, JSON.stringify(templateConfig, null, 2));
-      return;
-    }
-  } catch (e) {
-    // ignore
-  }
-
-  const fallback = {
-    server: { port: PRODUCTION_PORT, host: 'localhost' },
-    llm: { apiKey: '', baseURL: '', model: '', temperature: 0.7 },
-    logging: { level: 'info' },
-    channels: {},
-    systemPrompt: ''
-  };
-  try {
-    fs.writeFileSync(configPath, JSON.stringify(fallback, null, 2));
-  } catch (e) {
-    console.error(`[CLI] 无法写入默认配置：${configPath}`);
-    console.error(e);
-    process.exit(1);
-  }
-}
-
-function ensureDefaultDirs() {
-  for (const dir of ['logs', 'data', 'memories', 'sessions', 'workspace']) {
-    try {
-      fs.mkdirSync(path.join(APP_HOME, dir), { recursive: true });
-    } catch {
-      // ignore
-    }
-  }
-}
-
-function getPortFromConfig() {
-  let port = PRODUCTION_PORT;
-  const configPath = path.join(APP_HOME, 'config.json');
-  if (fs.existsSync(configPath)) {
-    try {
-      const config = JSON.parse(fs.readFileSync(configPath, 'utf-8'));
-      if (config.server && config.server.port) {
-        port = config.server.port;
-      }
-    } catch (e) {
-      // ignore
-    }
-  }
-  return port;
-}
-
-function getPidFilePath() {
-  return path.join(APP_HOME, '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() {
-  if (hasFlag('-cn')) {
-    return 'https://registry.npmmirror.com';
-  } else if (hasFlag('-en')) {
-    return 'https://registry.npmjs.org';
-  }
-  return getFlagValue('--registry') || process.env.npm_config_registry || 'https://registry.npmjs.org';
-}
-
-function isProcessRunning(pid) {
-  try {
-    process.kill(pid, 0);
-    return true;
-  } catch (e) {
-    return false;
-  }
-}
-
-async function isPortOpen(port) {
-  return await new Promise((resolve) => {
-    const socket = net.connect({ host: '127.0.0.1', port }, () => {
-      socket.destroy();
-      resolve(true);
-    });
-    socket.on('error', () => {
-      resolve(false);
-    });
-    socket.setTimeout(500, () => {
-      socket.destroy();
-      resolve(false);
-    });
-  });
-}
-
-async function killProcessTree(pid) {
-  if (process.platform === 'win32') {
-    try {
-      await runCommand('taskkill', ['/pid', pid.toString(), '/T', '/F'], { stdio: 'ignore' });
-      return true;
-    } catch {
-      return false;
-    }
-  }
-
-  // Unix-like implementation
-  const tryKill = (targetPid, signal) => {
-    try {
-      process.kill(targetPid, signal);
-      return true;
-    } catch (e) {
-      return false;
-    }
-  };
-
-  tryKill(-pid, 'SIGTERM');
-  tryKill(pid, 'SIGTERM');
-  
-  // Wait a bit
-  await new Promise(r => setTimeout(r, 500));
-  
-  if (isProcessRunning(pid)) {
-      tryKill(-pid, 'SIGKILL');
-      tryKill(pid, 'SIGKILL');
-  }
-  
-  return !isProcessRunning(pid);
-}
-
-async function stopServer() {
-  const pidFile = getPidFilePath();
-  const port = getPortFromConfig();
-
-  const killByPort = async () => {
-    if (process.platform === 'win32') {
-       // Windows implementation using netstat
-       try {
-         const netstat = spawn('netstat', ['-ano'], { stdio: ['ignore', 'pipe', 'ignore'] });
-         let output = '';
-         netstat.stdout.on('data', d => output += d.toString());
-         await new Promise(r => netstat.on('close', r));
-         
-         const lines = output.split('\n');
-         const pids = new Set();
-         for (const line of lines) {
-             if (line.includes(`:${port}`)) {
-                 const parts = line.trim().split(/\s+/);
-                 const pid = parts[parts.length - 1];
-                 if (pid && !isNaN(Number(pid)) && Number(pid) > 0) {
-                     pids.add(Number(pid));
-                 }
-             }
-         }
-         
-         if (pids.size === 0) return false;
-         
-         for (const pid of pids) {
-             await killProcessTree(pid);
-         }
-         return true;
-       } catch (e) {
-           console.error('[CLI] Windows port kill failed:', e);
-           return false;
-       }
-    } else {
-        // Unix implementation using lsof
-        const lsof = spawn('lsof', ['-ti', `tcp:${port}`], { stdio: ['ignore', 'pipe', 'ignore'] });
-        let output = '';
-        lsof.stdout.on('data', (d) => { output += d.toString(); });
-        await new Promise((resolve) => lsof.on('close', resolve));
-
-        const pids = output
-          .split(/\s+/)
-          .map(s => s.trim())
-          .filter(Boolean)
-          .map(Number)
-          .filter(n => Number.isFinite(n) && n > 0);
-
-        if (pids.length === 0) return false;
-
-        for (const pid of pids) {
-          await killProcessTree(pid);
-        }
-        return true;
-    }
-  };
-
-  if (fs.existsSync(pidFile)) {
-    const pidStr = fs.readFileSync(pidFile, 'utf-8').trim();
-    const pid = Number(pidStr);
-    
-    // ... existing pid checks ...
-    if (!Number.isFinite(pid) || pid <= 0) {
-      fs.unlinkSync(pidFile);
-      console.log('[CLI] 未找到可用的 PID（已清理 pid 文件）。');
-      return;
-    }
-
-    if (!isProcessRunning(pid)) {
-      fs.unlinkSync(pidFile);
-      console.log('[CLI] 服务未在运行（已清理 pid 文件）。');
-      return;
-    }
-
-    console.log(`[CLI] 正在停止服务（PID: ${pid}）...`);
-    await killProcessTree(pid);
-    await killByPort();
-    try { fs.unlinkSync(pidFile); } catch (e) {}
-
-    if (!(await isPortOpen(port))) {
-      console.log('[CLI] ✅ 服务已停止。');
-      return;
-    }
-    
-    // Fallback message
-    console.error('[CLI] ❌ 停止失败：端口仍在占用。');
-    if (process.platform !== 'win32') {
-        console.error(`[CLI] 你可以手动执行：lsof -ti tcp:${port} | xargs kill -9`);
-    } else {
-        console.error(`[CLI] 你可以手动执行：netstat -ano | findstr :${port} 并 kill 对应 PID`);
-    }
-    return;
-  }
-
-  console.log('[CLI] 未找到 pid 文件，尝试按端口停止...');
-  const ok = await killByPort();
-  if (!ok) {
-    console.log(`[CLI] 未发现占用端口 ${port} 的进程。`);
-    return;
-  }
-  if (!(await isPortOpen(port))) {
-    console.log('[CLI] ✅ 服务已停止。');
-    return;
-  }
-  console.error('[CLI] ❌ 停止失败：端口仍在占用。');
-}
-
-async function runCommand(cmd, cmdArgs, options = {}) {
-  // On Windows, use npm.cmd instead of npm
-  const command = process.platform === 'win32' && cmd === 'npm' ? 'npm.cmd' : cmd;
-  // For Windows, use shell: true to ensure commands are executed correctly
-  const child = spawn(command, cmdArgs, { stdio: 'inherit', shell: process.platform === 'win32', ...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 getRemoteVersion(registry) {
-  return new Promise((resolve, reject) => {
-    const options = {
-      hostname: registry.replace('https://', ''),
-      path: '/xiaozuoassistant/latest',
-      method: 'GET',
-      headers: {
-        'Content-Type': 'application/json'
-      }
-    };
-
-    const req = https.request(options, (res) => {
-      let data = '';
-      res.on('data', (chunk) => {
-        data += chunk;
-      });
-      res.on('end', () => {
-        try {
-          const info = JSON.parse(data);
-          resolve(info.version);
-        } catch (e) {
-          reject(new Error('Failed to parse remote version info'));
-        }
-      });
-    });
-
-    req.on('error', (e) => {
-      reject(e);
-    });
-
-    req.end();
-  });
-}
-
-function compareVersions(version1, version2) {
-  const v1 = version1.split('.').map(Number);
-  const v2 = version2.split('.').map(Number);
-  for (let i = 0; i < Math.max(v1.length, v2.length); i++) {
-    const num1 = v1[i] || 0;
-    const num2 = v2[i] || 0;
-    if (num1 > num2) return 1;
-    if (num1 < num2) return -1;
-  }
-  return 0;
-}
-
-async function updateApp() {
-  const registry = getRegistry();
-  const port = getPortFromConfig();
-  const pidFile = getPidFilePath();
-
-  // Get local version
-  const packageJsonPath = path.join(__dirname, '..', 'package.json');
-  const packageJsonContent = fs.readFileSync(packageJsonPath, 'utf-8');
-  const packageJson = JSON.parse(packageJsonContent);
-  const localVersion = packageJson.version;
-  console.log(`[CLI] 当前版本: ${localVersion}`);
-
-  // Get remote version
-  try {
-    console.log(`[CLI] 检查远程版本（registry=${registry}）...`);
-    const remoteVersion = await getRemoteVersion(registry);
-    console.log(`[CLI] 远程版本: ${remoteVersion}`);
-
-    // Compare versions
-    const comparison = compareVersions(localVersion, remoteVersion);
-    if (comparison >= 0) {
-      console.log('[CLI] ✅ 已经是最新版本，无需更新。');
-      process.exit(0);
-    }
-
-    console.log(`[CLI] 发现新版本 ${remoteVersion}，开始更新...`);
-  } catch (e) {
-    console.warn('[CLI] ⚠️ 版本检查失败，继续执行更新:', e);
-  }
-
-  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();
-  }
-
-  // Backup user config before update
-  const configPath = path.join(APP_HOME, 'config.json');
-  const configBackupPath = path.join(APP_HOME, 'config.json.bak');
-  let configBackedUp = false;
-  if (fs.existsSync(configPath)) {
-      try {
-          fs.copyFileSync(configPath, configBackupPath);
-          configBackedUp = true;
-          console.log('[CLI] 用户配置已备份。');
-      } catch (e) {
-          console.warn('[CLI] ⚠️ 用户配置备份失败:', e);
-      }
-  }
-
-  console.log(`[CLI] 正在更新 xiaozuoassistant（registry=${registry}）...`);
-  const code = await runWithSudoIfNeeded('npm', ['install', '-g', 'xiaozuoassistant@latest', `--registry=${registry}`]);
-  
-  // Restore user config after update
-  if (configBackedUp && fs.existsSync(configBackupPath)) {
-      try {
-          fs.copyFileSync(configBackupPath, configPath);
-          fs.unlinkSync(configBackupPath);
-          console.log('[CLI] 用户配置已恢复。');
-      } catch (e) {
-          console.warn('[CLI] ⚠️ 用户配置恢复失败:', e);
-      }
-  }
-
-  if (code !== 0) {
-    console.error('[CLI] ❌ 更新失败。');
-    if (wasRunning) {
-      console.log('[CLI] 更新失败，尝试恢复启动旧版本服务...');
-      await runCommand('node', [path.join(packageRoot, 'bin', 'cli.js'), 'start'], { cwd: APP_HOME });
-    }
-    process.exit(code);
-  }
-
-  console.log('[CLI] ✅ 更新完成。');
-
-  if (wasRunning) {
-    console.log('[CLI] 正在自动重启服务...');
-    const restartCode = await runCommand('xiaozuoAssistant', ['start'], { cwd: APP_HOME });
-    process.exit(restartCode);
-  }
-}
-
-async function removeApp() {
-  const registry = getRegistry();
-  const targets = ['config.json', 'memories', 'data', 'logs', 'sessions', 'workspace']
-    .map(p => path.join(APP_HOME, 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() {
-    ensureAppHome();
-    ensureDefaultDirs();
-    console.log('📦 Starting data export...');
-    
-    // Generate filename with timestamp
-    const now = new Date();
-    const timestamp = now.toISOString().replace(/[-:T.]/g, '').slice(0, 14); // YYYYMMDDHHmmss
-    const exportFilename = `${EXPORT_FILENAME_PREFIX}-${timestamp}.tar.gz`;
-    
-    const filesToArchive = [];
-    
-    for (const p of DATA_PATHS) {
-        if (fs.existsSync(path.join(APP_HOME, p))) {
-            filesToArchive.push(p);
-            console.log(`   - Found: ${p}`);
-        }
-    }
-
-    if (filesToArchive.length === 0) {
-        console.warn('⚠️ No data files found to export in current directory.');
-        return;
-    }
-
-    const exportPath = path.join(APP_HOME, exportFilename);
-
-    try {
-        await tar.c(
-            {
-                gzip: true,
-                file: exportPath,
-                cwd: APP_HOME
-            },
-            filesToArchive
-        );
-        console.log(`✅ Export successful!`);
-        console.log(`   Backup file: ${exportPath}`);
-        console.log(`   Copy this file to your new machine to import.`);
-    } catch (err) {
-        console.error('❌ Export failed:', err);
-    }
-}
-
-async function importData() {
-    ensureAppHome();
-    
-    // Find backup file
-    let backupFile = getFlagValue('--file');
-    
-    if (!backupFile) {
-        // Try to find the latest backup file in APP_HOME
-        try {
-            const files = fs.readdirSync(APP_HOME)
-                .filter(f => f.startsWith(EXPORT_FILENAME_PREFIX) && f.endsWith('.tar.gz'))
-                .sort()
-                .reverse();
-                
-            if (files.length > 0) {
-                backupFile = path.join(APP_HOME, files[0]);
-                console.log(`[CLI] No file specified, using latest found: ${files[0]}`);
-            } else {
-                // Fallback to old filename for compatibility
-                const oldFile = path.join(APP_HOME, 'xiaozuoAssistant-backup.tar.gz');
-                if (fs.existsSync(oldFile)) {
-                    backupFile = oldFile;
-                    console.log(`[CLI] Using legacy backup file: xiaozuoAssistant-backup.tar.gz`);
-                }
-            }
-        } catch (e) {
-            // ignore
-        }
-    }
-
-    if (!backupFile) {
-        console.error(`❌ No backup file found or specified.`);
-        console.log(`   Use --file <path> to specify a backup file.`);
-        return;
-    }
-    
-    // Resolve absolute path
-    const backupPath = path.resolve(process.cwd(), backupFile);
-    
-    if (!fs.existsSync(backupPath)) {
-        console.error(`❌ Backup file not found: ${backupPath}`);
-        return;
-    }
-
-    console.log(`📦 Starting data import from: ${backupPath}`);
-    console.log('⚠️  Warning: This will overwrite existing data files (config.json, memories, etc.)');
-    
-    // Simple prompt implementation for Node.js
-    process.stdout.write('   Are you sure? (y/N) ');
-    process.stdin.once('data', async (data) => {
-        const answer = data.toString().trim().toLowerCase();
-        if (answer === 'y' || answer === 'yes') {
-            try {
-                // Stop server if running before import
-                const pidFile = getPidFilePath();
-                if (fs.existsSync(pidFile)) {
-                     console.log('[CLI] Stopping server before import...');
-                     await stopServer();
-                }
-
-                await tar.x({
-                    file: backupPath,
-                    cwd: APP_HOME
-                });
-                console.log('✅ Import successful! Data restored.');
-                
-                // Auto-configure workspace path in config.json
-                const configPath = path.join(APP_HOME, 'config.json');
-                if (fs.existsSync(configPath)) {
-                    try {
-                        const configContent = fs.readFileSync(configPath, 'utf-8');
-                        const config = JSON.parse(configContent);
-                        
-                        // Update workspace to current directory
-                        const oldWorkspace = config.workspace;
-                        config.workspace = APP_HOME;
-                        
-                        // Also update System Prompt if it contains the old workspace path
-                        if (config.systemPrompt && typeof config.systemPrompt === 'string') {
-                            if (oldWorkspace && config.systemPrompt.includes(oldWorkspace)) {
-                                config.systemPrompt = config.systemPrompt.replace(oldWorkspace, APP_HOME);
-                            } else if (config.systemPrompt.includes('Current Workspace:')) {
-                                // Fallback regex replacement if exact string match fails
-                                config.systemPrompt = config.systemPrompt.replace(/Current Workspace: .*/, `Current Workspace: ${APP_HOME}`);
-                            }
-                        }
-
-                        // Ensure port is set to production port if it was dev port
-                        if (config.server && config.server.port === DEV_PORT) {
-                            config.server.port = PRODUCTION_PORT;
-                            console.log(`[CLI] Updated port from ${DEV_PORT} to ${PRODUCTION_PORT}`);
-                        }
-
-                        fs.writeFileSync(configPath, JSON.stringify(config, null, 2));
-                        console.log(`✅ Auto-configured workspace path to: ${APP_HOME}`);
-                    } catch (e) {
-                        console.warn('⚠️ Failed to auto-update config.json path:', e);
-                    }
-                }
-
-                console.log('   You can now start the application.');
-            } catch (err) {
-                console.error('❌ Import failed:', err);
-            }
-        } else {
-            console.log('   Import cancelled.');
-        }
-        process.exit(0);
-    });
-}
-
-if (command === 'start') {
-  ensureAppHome();
-  ensureDefaultDirs();
-  ensureDefaultConfig();
-  console.log('Starting xiaozuoAssistant...');
-  
-  const serverPath = path.join(packageRoot, 'dist', 'server', 'index.js');
-  
-  // Ensure logs directory exists in APP_HOME
-  const logDir = path.join(APP_HOME, 'logs');
-  if (!fs.existsSync(logDir)) {
-    try {
-      fs.mkdirSync(logDir, { recursive: true });
-    } catch (e) {
-      console.error(`[CLI] Failed to create logs directory at ${logDir}:`, e);
-      process.exit(1);
-    }
-  }
-
-  const outLog = path.join(logDir, 'stdout.log');
-  const errLog = path.join(logDir, 'stderr.log');
-
-  console.log(`[CLI] Launching server from: ${serverPath}`);
-  console.log(`[CLI] Logs redirected to:`);
-  console.log(`   - Stdout: ${outLog}`);
-  console.log(`   - Stderr: ${errLog}`);
-  console.log(`   - App Logs (Rotated): ${path.join(logDir, 'app.log')}`);
-
-  const port = getPortFromConfig();
-  const pidFile = getPidFilePath();
-
-  if (await isPortOpen(port)) {
-    console.error(`[CLI] 端口 ${port} 已被占用，可能服务已在运行。`);
-    console.error('[CLI] 可先执行：xiaozuoAssistant stop');
-    process.exit(1);
-  }
-
-  if (fs.existsSync(pidFile)) {
-    try {
-      const pid = Number(fs.readFileSync(pidFile, 'utf-8').trim());
-      if (Number.isFinite(pid) && pid > 0 && isProcessRunning(pid)) {
-        console.log(`[CLI] 服务已在运行（PID: ${pid}）。`);
-        console.log(`[CLI] 访问地址: http://localhost:${port}`);
-        process.exit(0);
-      }
-      fs.unlinkSync(pidFile);
-    } catch (e) {
-      // ignore
-    }
-  }
-
-  console.log(`[CLI] Waiting for server to start on port ${port}...`);
-
-  try {
-    const out = fs.openSync(outLog, 'a');
-    const err = fs.openSync(errLog, 'a');
-
-    const child = spawn(process.execPath, [serverPath, ...args.slice(1)], {
-      detached: true, // Allow child to run independently
-      stdio: ['ignore', out, err], // Disconnect stdin, redirect stdout/stderr
-      cwd: APP_HOME, // Run in unified app home
-      env: {
-        ...process.env,
-        NODE_ENV: 'production'
-      }
-    });
-
-    try {
-      fs.writeFileSync(pidFile, String(child.pid));
-    } catch (e) {
-      // ignore
-    }
-
-    let childExited = false;
-    child.on('exit', (code) => {
-      childExited = true;
-      console.error(`\n[CLI] Server process exited unexpectedly with code ${code}.`);
-      console.error(`[CLI] Check logs at ${errLog} or ${outLog}`);
-      process.exit(1);
-    });
-
-    // Health check function
-    const checkHealth = async (port, retries = 30) => {
-        const check = () => new Promise((resolve, reject) => {
-            const req = http.get(`http://localhost:${port}/api/health`, (res) => {
-                if (res.statusCode === 200) {
-                    resolve(true);
-                } else {
-                    reject(new Error(`Status ${res.statusCode}`));
-                }
-                res.resume();
-            });
-            
-            req.on('error', (err) => reject(err));
-            req.setTimeout(500, () => {
-                req.destroy();
-                reject(new Error('Timeout'));
-            });
-        });
-
-        for (let i = 0; i < retries; i++) {
-            if (childExited) throw new Error('Child process exited');
-            try {
-                await check();
-                return true;
-            } catch (e) {
-                await new Promise(r => setTimeout(r, 1000));
-                process.stdout.write('.');
-            }
-        }
-        throw new Error('Health check timed out');
-    };
-
-    // Perform health check
-    await checkHealth(port);
-    console.log(`\n[CLI] ✅ Health check passed!`);
-
-    // Unreference the child so the parent process can exit
-    child.unref();
-
-    console.log(`[CLI] Server running in background.`);
-    console.log(`[CLI] Access the Web UI at: http://localhost:${port}`);
-    process.exit(0);
-  } catch (error) {
-    try { if (fs.existsSync(pidFile)) fs.unlinkSync(pidFile); } catch (e) {}
-    console.error(`\n[CLI] Failed to start server: ${error.message}`);
-    process.exit(1);
-  }
-} else if (command === 'doctor') {
-    ensureAppHome();
-    ensureDefaultDirs();
-    console.log('Running doctor check...');
-    try {
-        const pkgPath = path.join(packageRoot, 'package.json');
-        const pkg = JSON.parse(fs.readFileSync(pkgPath, 'utf-8'));
-        console.log('Package Name:', pkg.name);
-        console.log('Package Version:', pkg.version);
-    } catch (e) {
-        console.log('Package Version: Unknown');
-    }
-    console.log('Package Root:', packageRoot);
-    console.log('Node Version:', process.version);
-    console.log('Doctor check complete.');
-} else if (command === 'version') {
-    ensureAppHome();
-    printVersion();
-} else if (command === 'update') {
-    updateApp();
-} else if (command === 'remove') {
-    removeApp();
-} else if (command === 'export') {
-    exportData();
-} else if (command === 'import') {
-    importData();
-} else if (command === 'stop') {
-    ensureAppHome();
-    stopServer();
-} else {
-    console.log(`Usage: ${binName} <command>`);
-    console.log('Commands:');
-    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('    Options:');
-    console.log('      -cn  Use Chinese npm registry (https://registry.npmmirror.com)');
-    console.log('      -en  Use English npm registry (https://registry.npmjs.org)');
-    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');
-    console.log('');
-    console.log(`Data Dir: ${APP_HOME}`);
-    console.log('  Use --home <path> or env XIAOZUOASSISTANT_HOME to override.');
-}
+#!/usr/bin/env node
+
+import { spawn } from 'child_process';
+import path from 'path';
+import { fileURLToPath } from 'url';
+import fs from 'fs';
+import os from 'os';
+import { createGzip } from 'zlib';
+import { pipeline } from 'stream';
+import { promisify } from 'util';
+import * as tar from 'tar';
+import http from 'http';
+import https from 'https';
+import net from 'net';
+
+const pipe = promisify(pipeline);
+
+const __filename = fileURLToPath(import.meta.url);
+const __dirname = path.dirname(__filename);
+
+// Determine the root directory of the package
+// When running from source: bin/cli.js -> ../
+// When installed globally: lib/node_modules/xiaozuoassistant/bin/cli.js -> ../
+const packageRoot = path.resolve(__dirname, '..');
+
+const binName = path.basename(process.argv[1] || 'xiaozuoAssistant');
+
+const args = process.argv.slice(2);
+const command = args[0];
+const commandArgs = args.slice(1);
+
+const EXPORT_FILENAME_PREFIX = 'xiaozuoAssistant-backup';
+const PRODUCTION_PORT = 3021; // 生产环境（CLI）默认端口
+const DEV_PORT = 3001; // 开发环境默认端口
+
+// Helper to get user's current working directory where they ran the command
+// This is kept only for display/backward compatibility; runtime data is stored in APP_HOME.
+const CWD = process.cwd();
+
+function getAppHome() {
+  const fromFlag = getFlagValue('--home');
+  const fromEnv = process.env.XIAOZUOASSISTANT_HOME;
+  const base = (fromFlag || fromEnv || path.join(os.homedir(), '.xiaozuoassistant')).trim();
+  return path.resolve(base);
+}
+
+const APP_HOME = getAppHome();
+
+// Unified runtime data directory
+const DATA_PATHS = [
+  'config.json',
+  'memories',
+  'data',
+  'logs',
+  'sessions',
+  'workspace',
+  'plugins'
+];
+
+function ensureAppHome() {
+  try {
+    fs.mkdirSync(APP_HOME, { recursive: true });
+  } catch (e) {
+    console.error(`[CLI] 无法创建数据目录：${APP_HOME}`);
+    console.error(e);
+    process.exit(1);
+  }
+}
+
+function ensureDefaultConfig() {
+  const configPath = path.join(APP_HOME, 'config.json');
+  
+  // 如果配置文件已存在，检查是否需要迁移端口（例如从 3001 迁移到 3021）
+  if (fs.existsSync(configPath)) {
+    try {
+        const configContent = fs.readFileSync(configPath, 'utf-8');
+        const config = JSON.parse(configContent);
+        // 如果端口是旧的默认值 3001，自动更新为 3021
+        if (config.server && config.server.port === DEV_PORT) {
+            console.log(`[CLI] 检测到旧端口配置 (${DEV_PORT})，正在迁移到生产环境端口 (${PRODUCTION_PORT})...`);
+            config.server.port = PRODUCTION_PORT;
+            fs.writeFileSync(configPath, JSON.stringify(config, null, 2));
+            console.log(`[CLI] ✅ 端口已更新为 ${PRODUCTION_PORT}`);
+        }
+    } catch (e) {
+        console.warn('[CLI] 检查现有配置失败:', e);
+    }
+    return;
+  }
+
+  // 如果配置文件不存在，创建新的默认配置
+  const templatePath = path.join(packageRoot, 'config.json');
+  try {
+    if (fs.existsSync(templatePath)) {
+      // 复制模板，但强制修改端口为 3021
+      const templateContent = fs.readFileSync(templatePath, 'utf-8');
+      const templateConfig = JSON.parse(templateContent);
+      
+      if (!templateConfig.server) templateConfig.server = {};
+      templateConfig.server.port = PRODUCTION_PORT;
+      
+      fs.writeFileSync(configPath, JSON.stringify(templateConfig, null, 2));
+      return;
+    }
+  } catch (e) {
+    // ignore
+  }
+
+  const fallback = {
+    server: { port: PRODUCTION_PORT, host: 'localhost' },
+    llm: { apiKey: '', baseURL: '', model: '', temperature: 0.7 },
+    logging: { level: 'info' },
+    channels: {},
+    systemPrompt: ''
+  };
+  try {
+    fs.writeFileSync(configPath, JSON.stringify(fallback, null, 2));
+  } catch (e) {
+    console.error(`[CLI] 无法写入默认配置：${configPath}`);
+    console.error(e);
+    process.exit(1);
+  }
+}
+
+function ensureDefaultDirs() {
+  for (const dir of ['logs', 'data', 'memories', 'sessions', 'workspace']) {
+    try {
+      fs.mkdirSync(path.join(APP_HOME, dir), { recursive: true });
+    } catch {
+      // ignore
+    }
+  }
+}
+
+function getPortFromConfig() {
+  let port = PRODUCTION_PORT;
+  const configPath = path.join(APP_HOME, 'config.json');
+  if (fs.existsSync(configPath)) {
+    try {
+      const config = JSON.parse(fs.readFileSync(configPath, 'utf-8'));
+      if (config.server && config.server.port) {
+        port = config.server.port;
+      }
+    } catch (e) {
+      // ignore
+    }
+  }
+  return port;
+}
+
+function getPidFilePath() {
+  return path.join(APP_HOME, '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() {
+  if (hasFlag('-cn')) {
+    return 'https://registry.npmmirror.com';
+  } else if (hasFlag('-en')) {
+    return 'https://registry.npmjs.org';
+  }
+  return getFlagValue('--registry') || process.env.npm_config_registry || 'https://registry.npmjs.org';
+}
+
+function isProcessRunning(pid) {
+  try {
+    process.kill(pid, 0);
+    return true;
+  } catch (e) {
+    return false;
+  }
+}
+
+async function isPortOpen(port) {
+  return await new Promise((resolve) => {
+    const socket = net.connect({ host: '127.0.0.1', port }, () => {
+      socket.destroy();
+      resolve(true);
+    });
+    socket.on('error', () => {
+      resolve(false);
+    });
+    socket.setTimeout(500, () => {
+      socket.destroy();
+      resolve(false);
+    });
+  });
+}
+
+async function killProcessTree(pid) {
+  if (process.platform === 'win32') {
+    try {
+      await runCommand('taskkill', ['/pid', pid.toString(), '/T', '/F'], { stdio: 'ignore' });
+      return true;
+    } catch {
+      return false;
+    }
+  }
+
+  // Unix-like implementation
+  const tryKill = (targetPid, signal) => {
+    try {
+      process.kill(targetPid, signal);
+      return true;
+    } catch (e) {
+      return false;
+    }
+  };
+
+  tryKill(-pid, 'SIGTERM');
+  tryKill(pid, 'SIGTERM');
+  
+  // Wait a bit
+  await new Promise(r => setTimeout(r, 500));
+  
+  if (isProcessRunning(pid)) {
+      tryKill(-pid, 'SIGKILL');
+      tryKill(pid, 'SIGKILL');
+  }
+  
+  return !isProcessRunning(pid);
+}
+
+async function stopServer() {
+  const pidFile = getPidFilePath();
+  const port = getPortFromConfig();
+
+  const killByPort = async () => {
+    if (process.platform === 'win32') {
+       // Windows implementation using netstat
+       try {
+         const netstat = spawn('netstat', ['-ano'], { stdio: ['ignore', 'pipe', 'ignore'] });
+         let output = '';
+         netstat.stdout.on('data', d => output += d.toString());
+         await new Promise(r => netstat.on('close', r));
+         
+         const lines = output.split('\n');
+         const pids = new Set();
+         for (const line of lines) {
+             if (line.includes(`:${port}`)) {
+                 const parts = line.trim().split(/\s+/);
+                 const pid = parts[parts.length - 1];
+                 if (pid && !isNaN(Number(pid)) && Number(pid) > 0) {
+                     pids.add(Number(pid));
+                 }
+             }
+         }
+         
+         if (pids.size === 0) return false;
+         
+         for (const pid of pids) {
+             await killProcessTree(pid);
+         }
+         return true;
+       } catch (e) {
+           console.error('[CLI] Windows port kill failed:', e);
+           return false;
+       }
+    } else {
+        // Unix implementation using lsof
+        const lsof = spawn('lsof', ['-ti', `tcp:${port}`], { stdio: ['ignore', 'pipe', 'ignore'] });
+        let output = '';
+        lsof.stdout.on('data', (d) => { output += d.toString(); });
+        await new Promise((resolve) => lsof.on('close', resolve));
+
+        const pids = output
+          .split(/\s+/)
+          .map(s => s.trim())
+          .filter(Boolean)
+          .map(Number)
+          .filter(n => Number.isFinite(n) && n > 0);
+
+        if (pids.length === 0) return false;
+
+        for (const pid of pids) {
+          await killProcessTree(pid);
+        }
+        return true;
+    }
+  };
+
+  if (fs.existsSync(pidFile)) {
+    const pidStr = fs.readFileSync(pidFile, 'utf-8').trim();
+    const pid = Number(pidStr);
+    
+    // ... existing pid checks ...
+    if (!Number.isFinite(pid) || pid <= 0) {
+      fs.unlinkSync(pidFile);
+      console.log('[CLI] 未找到可用的 PID（已清理 pid 文件）。');
+      return;
+    }
+
+    if (!isProcessRunning(pid)) {
+      fs.unlinkSync(pidFile);
+      console.log('[CLI] 服务未在运行（已清理 pid 文件）。');
+      return;
+    }
+
+    console.log(`[CLI] 正在停止服务（PID: ${pid}）...`);
+    await killProcessTree(pid);
+    await killByPort();
+    try { fs.unlinkSync(pidFile); } catch (e) {}
+
+    if (!(await isPortOpen(port))) {
+      console.log('[CLI] ✅ 服务已停止。');
+      return;
+    }
+    
+    // Fallback message
+    console.error('[CLI] ❌ 停止失败：端口仍在占用。');
+    if (process.platform !== 'win32') {
+        console.error(`[CLI] 你可以手动执行：lsof -ti tcp:${port} | xargs kill -9`);
+    } else {
+        console.error(`[CLI] 你可以手动执行：netstat -ano | findstr :${port} 并 kill 对应 PID`);
+    }
+    return;
+  }
+
+  console.log('[CLI] 未找到 pid 文件，尝试按端口停止...');
+  const ok = await killByPort();
+  if (!ok) {
+    console.log(`[CLI] 未发现占用端口 ${port} 的进程。`);
+    return;
+  }
+  if (!(await isPortOpen(port))) {
+    console.log('[CLI] ✅ 服务已停止。');
+    return;
+  }
+  console.error('[CLI] ❌ 停止失败：端口仍在占用。');
+}
+
+async function runCommand(cmd, cmdArgs, options = {}) {
+  // On Windows, use npm.cmd instead of npm
+  const command = process.platform === 'win32' && cmd === 'npm' ? 'npm.cmd' : cmd;
+  // For Windows, use shell: true to ensure commands are executed correctly
+  const child = spawn(command, cmdArgs, { stdio: 'inherit', shell: process.platform === 'win32', ...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 getRemoteVersion(registry) {
+  return new Promise((resolve, reject) => {
+    const options = {
+      hostname: registry.replace('https://', ''),
+      path: '/xiaozuoassistant/latest',
+      method: 'GET',
+      headers: {
+        'Content-Type': 'application/json'
+      }
+    };
+
+    const req = https.request(options, (res) => {
+      let data = '';
+      res.on('data', (chunk) => {
+        data += chunk;
+      });
+      res.on('end', () => {
+        try {
+          const info = JSON.parse(data);
+          resolve(info.version);
+        } catch (e) {
+          reject(new Error('Failed to parse remote version info'));
+        }
+      });
+    });
+
+    req.on('error', (e) => {
+      reject(e);
+    });
+
+    req.end();
+  });
+}
+
+function compareVersions(version1, version2) {
+  const v1 = version1.split('.').map(Number);
+  const v2 = version2.split('.').map(Number);
+  for (let i = 0; i < Math.max(v1.length, v2.length); i++) {
+    const num1 = v1[i] || 0;
+    const num2 = v2[i] || 0;
+    if (num1 > num2) return 1;
+    if (num1 < num2) return -1;
+  }
+  return 0;
+}
+
+async function updateApp() {
+  const registry = getRegistry();
+  const port = getPortFromConfig();
+  const pidFile = getPidFilePath();
+
+  // Get local version
+  const packageJsonPath = path.join(__dirname, '..', 'package.json');
+  const packageJsonContent = fs.readFileSync(packageJsonPath, 'utf-8');
+  const packageJson = JSON.parse(packageJsonContent);
+  const localVersion = packageJson.version;
+  console.log(`[CLI] 当前版本: ${localVersion}`);
+
+  // Get remote version
+  try {
+    console.log(`[CLI] 检查远程版本（registry=${registry}）...`);
+    const remoteVersion = await getRemoteVersion(registry);
+    console.log(`[CLI] 远程版本: ${remoteVersion}`);
+
+    // Compare versions
+    const comparison = compareVersions(localVersion, remoteVersion);
+    if (comparison >= 0) {
+      console.log('[CLI] ✅ 已经是最新版本，无需更新。');
+      process.exit(0);
+    }
+
+    console.log(`[CLI] 发现新版本 ${remoteVersion}，开始更新...`);
+  } catch (e) {
+    console.warn('[CLI] ⚠️ 版本检查失败，继续执行更新:', e);
+  }
+
+  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();
+  }
+
+  // Backup user config before update
+  const configPath = path.join(APP_HOME, 'config.json');
+  const configBackupPath = path.join(APP_HOME, 'config.json.bak');
+  let configBackedUp = false;
+  if (fs.existsSync(configPath)) {
+      try {
+          fs.copyFileSync(configPath, configBackupPath);
+          configBackedUp = true;
+          console.log('[CLI] 用户配置已备份。');
+      } catch (e) {
+          console.warn('[CLI] ⚠️ 用户配置备份失败:', e);
+      }
+  }
+
+  console.log(`[CLI] 正在更新 xiaozuoassistant（registry=${registry}）...`);
+  const code = await runWithSudoIfNeeded('npm', ['install', '-g', 'xiaozuoassistant@latest', `--registry=${registry}`]);
+  
+  // Restore user config after update
+  if (configBackedUp && fs.existsSync(configBackupPath)) {
+      try {
+          fs.copyFileSync(configBackupPath, configPath);
+          fs.unlinkSync(configBackupPath);
+          console.log('[CLI] 用户配置已恢复。');
+      } catch (e) {
+          console.warn('[CLI] ⚠️ 用户配置恢复失败:', e);
+      }
+  }
+
+  if (code !== 0) {
+    console.error('[CLI] ❌ 更新失败。');
+    if (wasRunning) {
+      console.log('[CLI] 更新失败，尝试恢复启动旧版本服务...');
+      await runCommand('node', [path.join(packageRoot, 'bin', 'cli.js'), 'start'], { cwd: APP_HOME });
+    }
+    process.exit(code);
+  }
+
+  console.log('[CLI] ✅ 更新完成。');
+
+  if (wasRunning) {
+    console.log('[CLI] 正在自动重启服务...');
+    const restartCode = await runCommand('xiaozuoAssistant', ['start'], { cwd: APP_HOME });
+    process.exit(restartCode);
+  }
+}
+
+async function removeApp() {
+  const registry = getRegistry();
+  const targets = ['config.json', 'memories', 'data', 'logs', 'sessions', 'workspace']
+    .map(p => path.join(APP_HOME, 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() {
+    ensureAppHome();
+    ensureDefaultDirs();
+    console.log('📦 Starting data export...');
+    
+    // Generate filename with timestamp
+    const now = new Date();
+    const timestamp = now.toISOString().replace(/[-:T.]/g, '').slice(0, 14); // YYYYMMDDHHmmss
+    const exportFilename = `${EXPORT_FILENAME_PREFIX}-${timestamp}.tar.gz`;
+    
+    const filesToArchive = [];
+    
+    for (const p of DATA_PATHS) {
+        if (fs.existsSync(path.join(APP_HOME, p))) {
+            filesToArchive.push(p);
+            console.log(`   - Found: ${p}`);
+        }
+    }
+
+    if (filesToArchive.length === 0) {
+        console.warn('⚠️ No data files found to export in current directory.');
+        return;
+    }
+
+    const exportPath = path.join(APP_HOME, exportFilename);
+
+    try {
+        await tar.c(
+            {
+                gzip: true,
+                file: exportPath,
+                cwd: APP_HOME
+            },
+            filesToArchive
+        );
+        console.log(`✅ Export successful!`);
+        console.log(`   Backup file: ${exportPath}`);
+        console.log(`   Copy this file to your new machine to import.`);
+    } catch (err) {
+        console.error('❌ Export failed:', err);
+    }
+}
+
+async function importData() {
+    ensureAppHome();
+    
+    // Find backup file
+    let backupFile = getFlagValue('--file');
+    
+    if (!backupFile) {
+        // Try to find the latest backup file in APP_HOME
+        try {
+            const files = fs.readdirSync(APP_HOME)
+                .filter(f => f.startsWith(EXPORT_FILENAME_PREFIX) && f.endsWith('.tar.gz'))
+                .sort()
+                .reverse();
+                
+            if (files.length > 0) {
+                backupFile = path.join(APP_HOME, files[0]);
+                console.log(`[CLI] No file specified, using latest found: ${files[0]}`);
+            } else {
+                // Fallback to old filename for compatibility
+                const oldFile = path.join(APP_HOME, 'xiaozuoAssistant-backup.tar.gz');
+                if (fs.existsSync(oldFile)) {
+                    backupFile = oldFile;
+                    console.log(`[CLI] Using legacy backup file: xiaozuoAssistant-backup.tar.gz`);
+                }
+            }
+        } catch (e) {
+            // ignore
+        }
+    }
+
+    if (!backupFile) {
+        console.error(`❌ No backup file found or specified.`);
+        console.log(`   Use --file <path> to specify a backup file.`);
+        return;
+    }
+    
+    // Resolve absolute path
+    const backupPath = path.resolve(process.cwd(), backupFile);
+    
+    if (!fs.existsSync(backupPath)) {
+        console.error(`❌ Backup file not found: ${backupPath}`);
+        return;
+    }
+
+    console.log(`📦 Starting data import from: ${backupPath}`);
+    console.log('⚠️  Warning: This will overwrite existing data files (config.json, memories, etc.)');
+    
+    // Simple prompt implementation for Node.js
+    process.stdout.write('   Are you sure? (y/N) ');
+    process.stdin.once('data', async (data) => {
+        const answer = data.toString().trim().toLowerCase();
+        if (answer === 'y' || answer === 'yes') {
+            try {
+                // Stop server if running before import
+                const pidFile = getPidFilePath();
+                if (fs.existsSync(pidFile)) {
+                     console.log('[CLI] Stopping server before import...');
+                     await stopServer();
+                }
+
+                await tar.x({
+                    file: backupPath,
+                    cwd: APP_HOME
+                });
+                console.log('✅ Import successful! Data restored.');
+                
+                // Auto-configure workspace path in config.json
+                const configPath = path.join(APP_HOME, 'config.json');
+                if (fs.existsSync(configPath)) {
+                    try {
+                        const configContent = fs.readFileSync(configPath, 'utf-8');
+                        const config = JSON.parse(configContent);
+                        
+                        // Update workspace to current directory
+                        const oldWorkspace = config.workspace;
+                        config.workspace = APP_HOME;
+                        
+                        // Also update System Prompt if it contains the old workspace path
+                        if (config.systemPrompt && typeof config.systemPrompt === 'string') {
+                            if (oldWorkspace && config.systemPrompt.includes(oldWorkspace)) {
+                                config.systemPrompt = config.systemPrompt.replace(oldWorkspace, APP_HOME);
+                            } else if (config.systemPrompt.includes('Current Workspace:')) {
+                                // Fallback regex replacement if exact string match fails
+                                config.systemPrompt = config.systemPrompt.replace(/Current Workspace: .*/, `Current Workspace: ${APP_HOME}`);
+                            }
+                        }
+
+                        // Ensure port is set to production port if it was dev port
+                        if (config.server && config.server.port === DEV_PORT) {
+                            config.server.port = PRODUCTION_PORT;
+                            console.log(`[CLI] Updated port from ${DEV_PORT} to ${PRODUCTION_PORT}`);
+                        }
+
+                        fs.writeFileSync(configPath, JSON.stringify(config, null, 2));
+                        console.log(`✅ Auto-configured workspace path to: ${APP_HOME}`);
+                    } catch (e) {
+                        console.warn('⚠️ Failed to auto-update config.json path:', e);
+                    }
+                }
+
+                console.log('   You can now start the application.');
+            } catch (err) {
+                console.error('❌ Import failed:', err);
+            }
+        } else {
+            console.log('   Import cancelled.');
+        }
+        process.exit(0);
+    });
+}
+
+if (command === 'start') {
+  ensureAppHome();
+  ensureDefaultDirs();
+  ensureDefaultConfig();
+  console.log('Starting xiaozuoAssistant...');
+  
+  const serverPath = path.join(packageRoot, 'dist', 'server', 'index.js');
+  
+  // Ensure logs directory exists in APP_HOME
+  const logDir = path.join(APP_HOME, 'logs');
+  if (!fs.existsSync(logDir)) {
+    try {
+      fs.mkdirSync(logDir, { recursive: true });
+    } catch (e) {
+      console.error(`[CLI] Failed to create logs directory at ${logDir}:`, e);
+      process.exit(1);
+    }
+  }
+
+  const outLog = path.join(logDir, 'stdout.log');
+  const errLog = path.join(logDir, 'stderr.log');
+
+  console.log(`[CLI] Launching server from: ${serverPath}`);
+  console.log(`[CLI] Logs redirected to:`);
+  console.log(`   - Stdout: ${outLog}`);
+  console.log(`   - Stderr: ${errLog}`);
+  console.log(`   - App Logs (Rotated): ${path.join(logDir, 'app.log')}`);
+
+  const port = getPortFromConfig();
+  const pidFile = getPidFilePath();
+
+  if (await isPortOpen(port)) {
+    console.error(`[CLI] 端口 ${port} 已被占用，可能服务已在运行。`);
+    console.error('[CLI] 可先执行：xiaozuoAssistant stop');
+    process.exit(1);
+  }
+
+  if (fs.existsSync(pidFile)) {
+    try {
+      const pid = Number(fs.readFileSync(pidFile, 'utf-8').trim());
+      if (Number.isFinite(pid) && pid > 0 && isProcessRunning(pid)) {
+        console.log(`[CLI] 服务已在运行（PID: ${pid}）。`);
+        console.log(`[CLI] 访问地址: http://localhost:${port}`);
+        process.exit(0);
+      }
+      fs.unlinkSync(pidFile);
+    } catch (e) {
+      // ignore
+    }
+  }
+
+  console.log(`[CLI] Waiting for server to start on port ${port}...`);
+
+  try {
+    const out = fs.openSync(outLog, 'a');
+    const err = fs.openSync(errLog, 'a');
+
+    const child = spawn(process.execPath, [serverPath, ...args.slice(1)], {
+      detached: true, // Allow child to run independently
+      stdio: ['ignore', out, err], // Disconnect stdin, redirect stdout/stderr
+      cwd: APP_HOME, // Run in unified app home
+      env: {
+        ...process.env,
+        NODE_ENV: 'production'
+      }
+    });
+
+    try {
+      fs.writeFileSync(pidFile, String(child.pid));
+    } catch (e) {
+      // ignore
+    }
+
+    let childExited = false;
+    child.on('exit', (code) => {
+      childExited = true;
+      console.error(`\n[CLI] Server process exited unexpectedly with code ${code}.`);
+      console.error(`[CLI] Check logs at ${errLog} or ${outLog}`);
+      process.exit(1);
+    });
+
+    // Health check function
+    const checkHealth = async (port, retries = 30) => {
+        const check = () => new Promise((resolve, reject) => {
+            const req = http.get(`http://localhost:${port}/api/health`, (res) => {
+                if (res.statusCode === 200) {
+                    resolve(true);
+                } else {
+                    reject(new Error(`Status ${res.statusCode}`));
+                }
+                res.resume();
+            });
+            
+            req.on('error', (err) => reject(err));
+            req.setTimeout(500, () => {
+                req.destroy();
+                reject(new Error('Timeout'));
+            });
+        });
+
+        for (let i = 0; i < retries; i++) {
+            if (childExited) throw new Error('Child process exited');
+            try {
+                await check();
+                return true;
+            } catch (e) {
+                await new Promise(r => setTimeout(r, 1000));
+                process.stdout.write('.');
+            }
+        }
+        throw new Error('Health check timed out');
+    };
+
+    // Perform health check
+    await checkHealth(port);
+    console.log(`\n[CLI] ✅ Health check passed!`);
+
+    // Unreference the child so the parent process can exit
+    child.unref();
+
+    console.log(`[CLI] Server running in background.`);
+    console.log(`[CLI] Access the Web UI at: http://localhost:${port}`);
+    process.exit(0);
+  } catch (error) {
+    try { if (fs.existsSync(pidFile)) fs.unlinkSync(pidFile); } catch (e) {}
+    console.error(`\n[CLI] Failed to start server: ${error.message}`);
+    process.exit(1);
+  }
+} else if (command === 'doctor') {
+    ensureAppHome();
+    ensureDefaultDirs();
+    console.log('Running doctor check...');
+    try {
+        const pkgPath = path.join(packageRoot, 'package.json');
+        const pkg = JSON.parse(fs.readFileSync(pkgPath, 'utf-8'));
+        console.log('Package Name:', pkg.name);
+        console.log('Package Version:', pkg.version);
+    } catch (e) {
+        console.log('Package Version: Unknown');
+    }
+    console.log('Package Root:', packageRoot);
+    console.log('Node Version:', process.version);
+    console.log('Doctor check complete.');
+} else if (command === 'version') {
+    ensureAppHome();
+    printVersion();
+} else if (command === 'update') {
+    updateApp();
+} else if (command === 'remove') {
+    removeApp();
+} else if (command === 'export') {
+    exportData();
+} else if (command === 'import') {
+    importData();
+} else if (command === 'stop') {
+    ensureAppHome();
+    stopServer();
+} else {
+    console.log(`Usage: ${binName} <command>`);
+    console.log('Commands:');
+    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('    Options:');
+    console.log('      -cn  Use Chinese npm registry (https://registry.npmmirror.com)');
+    console.log('      -en  Use English npm registry (https://registry.npmjs.org)');
+    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');
+    console.log('');
+    console.log(`Data Dir: ${APP_HOME}`);
+    console.log('  Use --home <path> or env XIAOZUOASSISTANT_HOME to override.');
+}