@agentunion/kite 1.0.6 → 1.2.0

package/cli.js

@@ -1,70 +1,172 @@
 #!/usr/bin/env node
 /**
- * Kite CLI — Node.js entry point.
- * Generates KITE_TOKEN, spawns the Python launcher, forwards output.
+ * Kite CLI — installer + launcher.
+ *
+ * First run: copies framework code to ~/.kite/versions/{version}/
+ * Subsequent runs: directly starts the installed version.
+ *
+ * Directory structure:
+ *   ~/.kite/
+ *     versions/{ver}/   — framework code (per-version, replaceable)
+ *     modules/          — shared modules (all versions share, user-writable)
+ *     data/             — runtime data (all versions share)
  *
  * Usage:
- *   kite                     # after npm install -g / npm link
+ *   kite                     # start current version
  *   kite --cwd /some/path    # override working directory
- *   node cli.js              # direct execution
- *
- * Environment variables passed to Python:
- *   KITE_HOME  — Kite framework root (where core/, extensions/ live)
- *   KITE_CWD   — directory where user executed the kite command
+ *   kite --use 1.0.6         # use a specific version
  */
-const { spawn } = require('child_process');
+const { spawn, execSync } = require('child_process');
 const crypto = require('crypto');
 const fs = require('fs');
 const path = require('path');
 
-// Parse CLI args
+// Timestamped console.log with delta + color
+const _origLog = console.log.bind(console);
+const _origErr = console.error.bind(console);
+let _lastTs = Date.now();
+const DIM = '\x1b[90m', GREEN = '\x1b[32m', RED = '\x1b[91m', RST = '\x1b[0m';
+
+function _prefix() {
+  const now = Date.now();
+  const delta = now - _lastTs;
+  _lastTs = now;
+  const d = new Date();
+  const ts = d.toTimeString().slice(0, 8) + '.' + String(d.getMilliseconds()).padStart(3, '0');
+  const ds = delta < 1000 ? `+${delta}ms` : `+${(delta / 1000).toFixed(1)}s`;
+  const color = delta >= 5000 ? RED : delta >= 1000 ? GREEN : DIM;
+  return `${ts} ${color}${ds.padStart(8)}${RST} `;
+}
+console.log = (...args) => _origLog(_prefix(), ...args);
+console.error = (...args) => _origErr(_prefix(), ...args);
+
+// ── Paths ──
+const packageDir = fs.realpathSync(__dirname);
+const packageJson = JSON.parse(fs.readFileSync(path.join(packageDir, 'package.json'), 'utf-8'));
+const version = packageJson.version;
+const homeDir = process.env.HOME || process.env.USERPROFILE;
+const kiteHome = path.join(homeDir, '.kite');
+const modulesDir = path.join(kiteHome, 'modules');
+const dataDir = path.join(kiteHome, 'data');
+
+// ── Parse CLI args ──
 const args = process.argv.slice(2);
 const pythonArgs = [];
 let cwdOverride = null;
+let useVersion = version;
 
 for (let i = 0; i < args.length; i++) {
   if (args[i] === '--cwd' && args[i + 1]) {
     cwdOverride = path.resolve(args[++i]);
+  } else if (args[i] === '--use' && args[i + 1]) {
+    useVersion = args[++i];
   } else {
     pythonArgs.push(args[i]);
   }
 }
 
-const token = crypto.randomBytes(32).toString('hex');
-const kiteHome = fs.realpathSync(__dirname);
+const versionDir = path.join(kiteHome, 'versions', useVersion);
+
+// ── Install: copy framework to ~/.kite/versions/{ver}/ if needed ──
+
+function copyDirSync(src, dest) {
+  fs.mkdirSync(dest, { recursive: true });
+  for (const entry of fs.readdirSync(src, { withFileTypes: true })) {
+    const srcPath = path.join(src, entry.name);
+    const destPath = path.join(dest, entry.name);
+    if (entry.isDirectory()) {
+      // Skip directories that shouldn't be copied
+      if (['__pycache__', 'node_modules', '.git', 'data'].includes(entry.name)) continue;
+      copyDirSync(srcPath, destPath);
+    } else {
+      fs.copyFileSync(srcPath, destPath);
+    }
+  }
+}
+
+function install() {
+  console.log(`[kite] 安装版本 ${useVersion} 到 ${versionDir}`);
+
+  // Create shared directories
+  fs.mkdirSync(modulesDir, { recursive: true });
+  fs.mkdirSync(dataDir, { recursive: true });
+
+  // Copy framework code from npm package
+  copyDirSync(packageDir, versionDir);
+
+  // Patch launcher module.md: add shared modules path to discovery
+  patchLauncherDiscovery();
+
+  console.log(`[kite] 版本 ${useVersion} 已安装`);
+}
+
+function patchLauncherDiscovery() {
+  const mdPath = path.join(versionDir, 'core', 'launcher', 'module.md');
+  if (!fs.existsSync(mdPath)) return;
+
+  let content = fs.readFileSync(mdPath, 'utf-8');
+
+  // Already patched?
+  if (content.includes('shared_modules')) return;
+
+  // Insert shared_modules discovery entry before the closing ---
+  const entry = [
+    '  shared_modules:',
+    '    type: scan_dir',
+    `    path: "${modulesDir.replace(/\\/g, '/')}"`,
+    '    enabled: true',
+  ].join('\n');
+
+  // Insert after the last discovery entry, before ---
+  content = content.replace(
+    /(discovery:\n(?:[\s\S]*?))(---)/,
+    `$1${entry}\n$2`
+  );
+
+  fs.writeFileSync(mdPath, content, 'utf-8');
+  console.log(`[kite] 已补丁启动器发现路径: ${modulesDir}`);
+}
+
+// ── Main: check version, install if needed, start ──
+
+if (!fs.existsSync(path.join(versionDir, 'main.py'))) {
+  install();
+}
+
 const kiteCwd = cwdOverride || process.cwd();
 
-console.log(`[main] KITE_TOKEN generated (${token.length} chars)`);
-console.log(`[main] KITE_HOME: ${kiteHome}`);
-console.log(`[main] KITE_CWD:  ${kiteCwd}`);
+console.log(`[kite] 版本:     ${useVersion}`);
+console.log(`[kite] 项目:     ${versionDir}`);
+console.log(`[kite] 工作目录: ${kiteCwd}`);
 
-// Spawn Python using package mode: python -m Kite
-// PYTHONPATH points to Kite's parent so "import Kite" / "python -m Kite" works
-const kiteParent = path.dirname(kiteHome);
 const env = {
   ...process.env,
-  KITE_HOME: kiteHome,
+  KITE_PROJECT: versionDir,
   KITE_CWD: kiteCwd,
-  PYTHONPATH: kiteParent + (process.env.PYTHONPATH ? path.delimiter + process.env.PYTHONPATH : ''),
+  KITE_WORKSPACE: path.join(kiteHome, 'workspace'),
+  KITE_DATA: dataDir,
+  KITE_MODULES: modulesDir,
+  KITE_REPO: path.join(kiteHome, 'repo'),
+  KITE_VERSION: useVersion,
+  KITE_ENV: 'production',
   PYTHONUTF8: '1',
 };
 
 const python = process.platform === 'win32' ? 'python' : 'python3';
 const child = spawn(
   python,
-  [path.join(kiteHome, 'main.py'), ...pythonArgs],
-  { cwd: kiteHome, env, stdio: 'inherit' }
+  [path.join(versionDir, 'main.py'), ...pythonArgs],
+  { cwd: versionDir, env, stdio: 'inherit' }
 );
 
 child.on('error', err => {
-  console.error(`[main] Failed to start Python: ${err.message}`);
-  console.error(`[main] Make sure '${python}' is in PATH and dependencies are installed`);
+  console.error(`[kite] 启动 Python 失败: ${err.message}`);
+  console.error(`[kite] 请确认 '${python}' 在 PATH 中且依赖已安装`);
   process.exit(1);
 });
 
 child.on('exit', code => process.exit(code ?? 0));
 
-// Forward signals for graceful shutdown
 for (const sig of ['SIGINT', 'SIGTERM']) {
   process.on(sig, () => child.kill(sig));
 }