aicodeswitch 1.10.1 → 2.0.1

package/bin/ui.js

@@ -1,128 +1,16 @@
 const { spawn } = require('child_process');
-const path = require('path');
-const fs = require('fs');
 const os = require('os');
 const chalk = require('chalk');
 const ora = require('ora');
-
-const PID_FILE = path.join(os.homedir(), '.aicodeswitch', 'server.pid');
-const LOG_FILE = path.join(os.homedir(), '.aicodeswitch', 'server.log');
-
-// 确保目录存在
-const ensureDir = (filePath) => {
-  const dir = path.dirname(filePath);
-  if (!fs.existsSync(dir)) {
-    fs.mkdirSync(dir, { recursive: true });
-  }
-};
-
-const isServerRunning = () => {
-  if (!fs.existsSync(PID_FILE)) {
-    return false;
-  }
-
-  try {
-    const pid = parseInt(fs.readFileSync(PID_FILE, 'utf-8'), 10);
-    // 检查进程是否存在
-    process.kill(pid, 0);
-    return true;
-  } catch (err) {
-    // 进程不存在,删除过期的 PID 文件
-    fs.unlinkSync(PID_FILE);
-    return false;
-  }
-};
-
-const getServerInfo = () => {
-  // 尝试多个可能的配置文件位置
-  const possiblePaths = [
-    path.join(os.homedir(), '.aicodeswitch', '.env'),
-    path.join(os.homedir(), '.aicodeswitch', 'aicodeswitch.conf')
-  ];
-
-  let host = '127.0.0.1';
-  let port = 4567;
-
-  for (const dotenvPath of possiblePaths) {
-    if (fs.existsSync(dotenvPath)) {
-      const content = fs.readFileSync(dotenvPath, 'utf-8');
-      const hostMatch = content.match(/HOST=(.+)/);
-      const portMatch = content.match(/PORT=(.+)/);
-
-      if (hostMatch) host = hostMatch[1].trim();
-      if (portMatch) port = parseInt(portMatch[1].trim(), 10);
-      break;
-    }
-  }
-
-  return { host, port };
-};
-
-const startServer = async () => {
-  const spinner = ora({
-    text: chalk.cyan('Starting AI Code Switch server...'),
-    color: 'cyan'
-  }).start();
-
-  ensureDir(PID_FILE);
-  ensureDir(LOG_FILE);
-
-  // 找到 main.js 的路径
-  const serverPath = path.join(__dirname, '..', 'dist', 'server', 'main.js');
-
-  if (!fs.existsSync(serverPath)) {
-    spinner.fail(chalk.red('Server file not found!'));
-    console.log(chalk.yellow(`\nPlease run ${chalk.cyan('npm run build')} first.\n`));
-    process.exit(1);
-  }
-
-  // 启动服务器进程 - 完全分离
-  // 打开日志文件用于输出
-  const logFd = fs.openSync(LOG_FILE, 'a');
-
-  const serverProcess = spawn('node', [serverPath], {
-    detached: true,
-    stdio: ['ignore', logFd, logFd]  // 使用文件描述符
-  });
-
-  // 关闭文件描述符(子进程会保持打开)
-  fs.closeSync(logFd);
-
-  // 保存 PID
-  fs.writeFileSync(PID_FILE, serverProcess.pid.toString());
-
-  // 分离进程,让父进程可以退出
-  serverProcess.unref();
-
-  // 等待服务器启动
-  await new Promise(resolve => setTimeout(resolve, 2000));
-
-  // 检查服务器是否成功启动
-  if (fs.existsSync(PID_FILE)) {
-    try {
-      const pid = parseInt(fs.readFileSync(PID_FILE, 'utf-8'), 10);
-      process.kill(pid, 0);
-      spinner.succeed(chalk.green('Server started successfully!'));
-      return true;
-    } catch (err) {
-      spinner.fail(chalk.red('Failed to start server!'));
-      console.log(chalk.yellow(`\nCheck logs: ${chalk.cyan(LOG_FILE)}\n`));
-      return false;
-    }
-  } else {
-    spinner.fail(chalk.red('Failed to start server!'));
-    console.log(chalk.yellow(`\nCheck logs: ${chalk.cyan(LOG_FILE)}\n`));
-    return false;
-  }
-};
+const start = require('./start');
+const { getServerInfo } = require('./utils/get-server');
 
 const openBrowser = (url) => {
-  const platform = os.platform();
   let command;
 
-  if (platform === 'darwin') {
+  if (os.platform() === 'darwin') {
     command = 'open';
-  } else if (platform === 'win32') {
+  } else if (os.platform() === 'win32') {
     command = 'start';
   } else {
     // Linux and others
@@ -138,36 +26,33 @@ const openBrowser = (url) => {
 };
 
 const openUI = async () => {
-  console.log('\n');
-
-  const running = isServerRunning();
+  // 使用 start 命令启动服务器（如果需要）
+  // 使用 silent 和 noExit 选项来控制行为
+  const started = await start({ silent: true, noExit: true });
 
-  if (!running) {
-    console.log(chalk.yellow('⚠ Server is not running, starting server first...\n'));
-    const started = await startServer();
-    if (!started) {
-      console.log(chalk.red('\n✗ Failed to start server, cannot open UI\n'));
-      process.exit(1);
-    }
-  } else {
-    console.log(chalk.green('✓ Server is already running\n'));
+  if (!started) {
+    console.log(chalk.red('\n✗ Failed to start server, cannot open UI\n'));
+    process.exit(1);
   }
 
   const { host, port } = getServerInfo();
   const url = `http://${host}:${port}`;
 
-  console.log(chalk.cyan('🌐 Opening browser...'));
-  console.log(chalk.white('   URL: ') + chalk.cyan.bold(url) + '\n');
+  const spinner = ora({
+    text: chalk.cyan('Opening browser...'),
+    color: 'cyan'
+  }).start();
 
   try {
     openBrowser(url);
-    console.log(chalk.green('✓ Browser opened successfully!\n'));
+    spinner.succeed(chalk.green('Browser opened successfully!'));
+    console.log(chalk.white('   URL: ') + chalk.cyan.bold(url) + '\n');
   } catch (err) {
-    console.log(chalk.yellow('⚠ Failed to open browser automatically'));
-    console.log(chalk.white('  Please open this URL manually: ') + chalk.cyan.bold(url) + '\n');
+    spinner.fail(chalk.red('Failed to open browser automatically'));
+    console.log(chalk.yellow('⚠ Please open this URL manually: ') + chalk.cyan.bold(url) + '\n');
   }
 
   process.exit(0);
 };
 
-module.exports = openUI();
+module.exports = openUI;

package/bin/update.js

@@ -280,4 +280,4 @@ const update = async () => {
   console.log('\n');
 };
 
-module.exports = update();
+module.exports = update;

package/bin/utils/get-server.js

@@ -0,0 +1,58 @@
+const path = require('path');
+const fs = require('fs');
+const os = require('os');
+
+const PID_FILE = path.join(os.homedir(), '.aicodeswitch', 'server.pid');
+
+/**
+ * 判断服务器是否正在运行
+ * @returns
+ */
+const isServerRunning = () => {
+  if (!fs.existsSync(PID_FILE)) {
+    return false;
+  }
+
+  try {
+    const pid = parseInt(fs.readFileSync(PID_FILE, 'utf-8'), 10);
+    // 检查进程是否存在
+    process.kill(pid, 0);
+    return true;
+  } catch (err) {
+    // 进程不存在,删除过期的 PID 文件
+    fs.unlinkSync(PID_FILE);
+    return false;
+  }
+};
+
+/**
+ * 获取服务器信息
+ * @returns
+ */
+const getServerInfo = () => {
+  // 尝试多个可能的配置文件位置
+  const possiblePaths = [
+    path.join(os.homedir(), '.aicodeswitch', '.env'),
+    path.join(os.homedir(), '.aicodeswitch', 'aicodeswitch.conf')
+  ];
+
+  let host = '127.0.0.1';
+  let port = 4567;
+
+  for (const dotenvPath of possiblePaths) {
+    if (fs.existsSync(dotenvPath)) {
+      const content = fs.readFileSync(dotenvPath, 'utf-8');
+      const hostMatch = content.match(/HOST=(.+)/);
+      const portMatch = content.match(/PORT=(.+)/);
+
+      if (hostMatch) host = hostMatch[1].trim();
+      if (portMatch) port = parseInt(portMatch[1].trim(), 10);
+      break;
+    }
+  }
+
+  return { host, port };
+};
+
+module.exports.isServerRunning = isServerRunning;
+module.exports.getServerInfo = getServerInfo;

package/bin/utils/port-utils.js

@@ -0,0 +1,118 @@
+const { exec } = require('child_process');
+const util = require('util');
+const execPromise = util.promisify(exec);
+
+/**
+ * 获取占用指定端口的进程 PID
+ * @param {number} port - 端口号
+ * @returns {Promise<number|null>} 返回 PID 或 null
+ */
+const findPidByPort = async (port) => {
+  const platform = process.platform;
+
+  try {
+    if (platform === 'win32') {
+      // Windows: 使用 netstat
+      const { stdout } = await execPromise(`netstat -ano | findstr :${port}`);
+      const lines = stdout.split('\n').filter(line => line.trim());
+
+      for (const line of lines) {
+        const parts = line.trim().split(/\s+/);
+        // 格式: 协议 本地地址 外部地址 状态 PID
+        if (parts.length >= 5) {
+          const localAddress = parts[1];
+          if (localAddress.includes(`:${port}`)) {
+            const pid = parseInt(parts[parts.length - 1], 10);
+            if (!isNaN(pid)) {
+              return pid;
+            }
+          }
+        }
+      }
+    } else {
+      // macOS 和 Linux: 使用 lsof
+      const { stdout } = await execPromise(`lsof -ti :${port}`);
+      const pidStr = stdout.trim();
+      if (pidStr) {
+        // lsof -ti 可能返回多个 PID，取第一个
+        const pids = pidStr.split('\n').filter(p => p);
+        if (pids.length > 0) {
+          return parseInt(pids[0], 10);
+        }
+      }
+    }
+  } catch (err) {
+    // 命令执行失败，说明端口未被占用
+    return null;
+  }
+
+  return null;
+};
+
+/**
+ * 终止指定 PID 的进程
+ * @param {number} pid - 进程 PID
+ * @returns {Promise<boolean>} 是否成功终止
+ */
+const killProcess = async (pid) => {
+  const platform = process.platform;
+
+  try {
+    if (platform === 'win32') {
+      // Windows: 使用 taskkill
+      await execPromise(`taskkill /F /PID ${pid}`);
+    } else {
+      // macOS 和 Linux: 使用 kill
+      process.kill(pid, 'SIGTERM');
+
+      // 等待进程停止
+      let attempts = 0;
+      const maxAttempts = 10;
+
+      while (attempts < maxAttempts) {
+        try {
+          process.kill(pid, 0);
+          await new Promise(resolve => setTimeout(resolve, 200));
+          attempts++;
+        } catch (err) {
+          // 进程已停止
+          return true;
+        }
+      }
+
+      // 强制终止
+      process.kill(pid, 'SIGKILL');
+    }
+    return true;
+  } catch (err) {
+    // 进程可能已经不存在
+    return false;
+  }
+};
+
+/**
+ * 获取进程信息（用于显示）
+ * @param {number} pid - 进程 PID
+ * @returns {Promise<string>} 进程信息
+ */
+const getProcessInfo = async (pid) => {
+  const platform = process.platform;
+
+  try {
+    if (platform === 'win32') {
+      const { stdout } = await execPromise(`tasklist /FI "PID eq ${pid}" /FO CSV /NH`);
+      return stdout.trim();
+    } else {
+      const { stdout } = await execPromise(`ps -p ${pid} -o comm=`);
+      return stdout.trim();
+    }
+  } catch (err) {
+    return 'Unknown';
+  }
+};
+
+module.exports = {
+  findPidByPort,
+  killProcess,
+  getProcessInfo
+};

package/bin/version.js

@@ -44,4 +44,4 @@ const version = () => {
   process.exit(0);
 };
 
-module.exports = version();
+module.exports = version;