@agentunion/kite 1.4.0 → 1.5.0

package/scripts/env-manager.js

@@ -0,0 +1,328 @@
+/**
+ * Kite 环境管理模块
+ *
+ * 负责在 Node.js 层检查和准备 Python 环境
+ */
+
+const fs = require('fs');
+const path = require('path');
+const os = require('os');
+const { execSync, spawnSync } = require('child_process');
+
+// 颜色输出
+const colors = {
+  reset: '\x1b[0m',
+  green: '\x1b[32m',
+  yellow: '\x1b[33m',
+  red: '\x1b[31m',
+  cyan: '\x1b[36m'
+};
+
+function log(msg, color = 'reset') {
+  console.log(`${colors[color]}[Kite Env]${colors.reset} ${msg}`);
+}
+
+function error(msg) {
+  console.error(`${colors.red}[Kite Env Error]${colors.reset} ${msg}`);
+}
+
+/**
+ * 获取 KITE_DATA 路径
+ */
+function getKiteDataPath() {
+  return path.join(os.homedir(), '.kite');
+}
+
+/**
+ * 获取环境状态文件路径
+ */
+function getEnvStatusPath() {
+  return path.join(getKiteDataPath(), 'env_status.json');
+}
+
+/**
+ * 读取环境状态
+ */
+function loadEnvStatus() {
+  const statusFile = getEnvStatusPath();
+
+  if (!fs.existsSync(statusFile)) {
+    return null;
+  }
+
+  try {
+    const data = fs.readFileSync(statusFile, 'utf-8');
+    return JSON.parse(data);
+  } catch (e) {
+    error(`读取环境状态失败: ${e.message}`);
+    return null;
+  }
+}
+
+/**
+ * 检查是否需要准备 Python
+ */
+function needsPythonSetup(versionDir) {
+  const pythonVersionFile = path.join(versionDir, 'python_version.json');
+
+  if (!fs.existsSync(pythonVersionFile)) {
+    return false; // 没有版本要求文件，跳过
+  }
+
+  try {
+    const versionData = JSON.parse(fs.readFileSync(pythonVersionFile, 'utf-8'));
+    const requiredVersion = versionData.version;
+
+    // 检测系统 Python
+    const pythonCommands = ['python3', 'python'];
+
+    for (const cmd of pythonCommands) {
+      try {
+        const version = execSync(`${cmd} --version`, {
+          encoding: 'utf8',
+          stdio: ['pipe', 'pipe', 'ignore']
+        }).trim();
+
+        const match = version.match(/Python (\d+\.\d+\.\d+)/);
+        if (match && match[1] === requiredVersion) {
+          return false; // 找到匹配的 Python 版本
+        }
+      } catch (e) {
+        // 命令不存在，继续
+      }
+    }
+
+    return true; // 没有找到匹配的 Python 版本
+  } catch (e) {
+    error(`检查 Python 版本失败: ${e.message}`);
+    return true; // 出错时保守处理，执行安装
+  }
+}
+
+/**
+ * 准备 Python 环境
+ */
+function setupPython(versionDir) {
+  log('准备 Python 环境...', 'cyan');
+
+  const setupScript = path.join(versionDir, 'scripts', 'setup-python-env.js');
+
+  if (!fs.existsSync(setupScript)) {
+    error('setup-python-env.js 不存在');
+    return false;
+  }
+
+  try {
+    execSync(`node "${setupScript}"`, {
+      stdio: 'inherit',
+      cwd: versionDir
+    });
+    return true;
+  } catch (e) {
+    error('Python 环境准备失败');
+    return false;
+  }
+}
+
+/**
+ * 检查是否需要准备虚拟环境
+ */
+function needsVenvSetup(status) {
+  // 1. 状态文件不存在
+  if (!status) {
+    return true;
+  }
+
+  // 2. 上次检查失败
+  if (status.status !== 'success') {
+    return true;
+  }
+
+  // 3. 虚拟环境不存在
+  const venvPath = path.join(getKiteDataPath(), 'venv');
+  if (!fs.existsSync(venvPath)) {
+    return true;
+  }
+
+  return false;
+}
+
+/**
+ * 检查是否需要安装依赖
+ */
+function needsDepsInstall(status) {
+  // 1. 状态文件不存在
+  if (!status) {
+    return true;
+  }
+
+  // 2. 上次检查失败
+  if (status.status !== 'success') {
+    return true;
+  }
+
+  // 3. 依赖未安装
+  if (!status.dependencies || !status.dependencies.installed) {
+    return true;
+  }
+
+  return false;
+}
+
+/**
+ * 准备虚拟环境
+ */
+function setupVenv(versionDir) {
+  log('准备虚拟环境...', 'cyan');
+
+  const pythonCmd = process.platform === 'win32' ? 'python' : 'python3';
+
+  try {
+    const result = spawnSync(
+      pythonCmd,
+      ['-m', 'kite_cli', 'venv-setup'],
+      {
+        cwd: versionDir,
+        stdio: 'inherit',
+        encoding: 'utf-8',
+        env: {
+          ...process.env,
+          PYTHONPATH: versionDir
+        }
+      }
+    );
+
+    return result.status === 0;
+  } catch (e) {
+    error('虚拟环境准备失败');
+    return false;
+  }
+}
+
+/**
+ * 安装依赖
+ */
+function installDeps(versionDir) {
+  log('安装依赖库...', 'cyan');
+
+  const pythonCmd = process.platform === 'win32' ? 'python' : 'python3';
+
+  try {
+    const result = spawnSync(
+      pythonCmd,
+      ['-m', 'kite_cli', 'deps-install'],
+      {
+        cwd: versionDir,
+        stdio: 'inherit',
+        encoding: 'utf-8',
+        env: {
+          ...process.env,
+          PYTHONPATH: versionDir
+        }
+      }
+    );
+
+    return result.status === 0;
+  } catch (e) {
+    error('依赖安装失败');
+    return false;
+  }
+}
+
+/**
+ * 完整的环境检查和准备流程（三步）
+ */
+function ensureEnvironment(versionDir) {
+  log('检查环境...', 'cyan');
+
+  // 1. 读取环境状态
+  const status = loadEnvStatus();
+
+  // 步骤 1: 检查是否需要准备 Python
+  if (needsPythonSetup(versionDir)) {
+    log('步骤 1/3: 准备 Python 环境', 'yellow');
+
+    if (!setupPython(versionDir)) {
+      error('Python 环境准备失败，无法启动 Kite');
+      markEnvFailed('python_setup_failed');
+      return false;
+    }
+
+    log('Python 环境准备完成', 'green');
+  } else {
+    log('步骤 1/3: Python 环境已就绪', 'green');
+  }
+
+  // 步骤 2: 检查是否需要准备虚拟环境
+  if (needsVenvSetup(status)) {
+    log('步骤 2/3: 准备虚拟环境', 'yellow');
+
+    if (!setupVenv(versionDir)) {
+      error('虚拟环境准备失败，无法启动 Kite');
+      markEnvFailed('venv_setup_failed');
+      return false;
+    }
+
+    log('虚拟环境准备完成', 'green');
+  } else {
+    log('步骤 2/3: 虚拟环境已就绪', 'green');
+  }
+
+  // 步骤 3: 检查是否需要安装依赖
+  if (needsDepsInstall(status)) {
+    log('步骤 3/3: 安装依赖库', 'yellow');
+
+    if (!installDeps(versionDir)) {
+      error('依赖安装失败，无法启动 Kite');
+      markEnvFailed('deps_install_failed');
+      return false;
+    }
+
+    log('依赖安装完成', 'green');
+  } else {
+    log('步骤 3/3: 依赖库已就绪', 'green');
+  }
+
+  log('环境准备完成', 'green');
+  return true;
+}
+
+/**
+ * 标记环境失败（用于异常处理）
+ */
+function markEnvFailed(reason) {
+  const statusFile = getEnvStatusPath();
+  const statusData = {
+    version: '1.0',
+    last_check: new Date().toISOString(),
+    status: 'failed',
+    error: {
+      type: reason,
+      message: '环境准备失败',
+      timestamp: new Date().toISOString()
+    }
+  };
+
+  try {
+    const kiteData = getKiteDataPath();
+    if (!fs.existsSync(kiteData)) {
+      fs.mkdirSync(kiteData, { recursive: true });
+    }
+
+    fs.writeFileSync(statusFile, JSON.stringify(statusData, null, 2));
+  } catch (e) {
+    error(`无法写入状态文件: ${e.message}`);
+  }
+}
+
+module.exports = {
+  ensureEnvironment,
+  loadEnvStatus,
+  needsPythonSetup,
+  needsVenvSetup,
+  needsDepsInstall,
+  setupPython,
+  setupVenv,
+  installDeps,
+  markEnvFailed
+};

package/scripts/python-env.js

@@ -0,0 +1,79 @@
+/**
+ * Python 环境工具模块
+ *
+ * 提供获取虚拟环境 Python 路径的工具函数
+ */
+
+const fs = require('fs');
+const path = require('path');
+const os = require('os');
+
+/**
+ * 获取 KITE_DATA 路径
+ */
+function getKiteDataPath() {
+  return path.join(os.homedir(), '.kite');
+}
+
+/**
+ * 获取虚拟环境路径
+ */
+function getVenvPath() {
+  return path.join(getKiteDataPath(), 'venv');
+}
+
+/**
+ * 获取虚拟环境的 Python 可执行文件路径
+ */
+function getVenvPython() {
+  const venvPath = getVenvPath();
+
+  if (process.platform === 'win32') {
+    return path.join(venvPath, 'Scripts', 'python.exe');
+  } else {
+    return path.join(venvPath, 'bin', 'python');
+  }
+}
+
+/**
+ * 获取虚拟环境的 pip 可执行文件路径
+ */
+function getVenvPip() {
+  const venvPath = getVenvPath();
+
+  if (process.platform === 'win32') {
+    return path.join(venvPath, 'Scripts', 'pip.exe');
+  } else {
+    return path.join(venvPath, 'bin', 'pip');
+  }
+}
+
+/**
+ * 检查虚拟环境是否存在
+ */
+function venvExists() {
+  const pythonPath = getVenvPython();
+  return fs.existsSync(pythonPath);
+}
+
+/**
+ * 检查 Python 环境是否就绪
+ */
+function isPythonReady() {
+  if (!venvExists()) {
+    return false;
+  }
+
+  // 检查 pip 是否存在
+  const pipPath = getVenvPip();
+  return fs.existsSync(pipPath);
+}
+
+module.exports = {
+  getKiteDataPath,
+  getVenvPath,
+  getVenvPython,
+  getVenvPip,
+  venvExists,
+  isPythonReady
+};