feat(版本检查): 添加启动时自动检查版本更新功能

huzijie.sea · huzijie.sea · commit 0686fb9f3757 · 2025-12-17T15:23:32.000+08:00
添加版本检查服务 VersionChecker，实现以下功能：
1. 从 npm registry 获取最新版本
2. 本地缓存版本信息避免频繁请求
3. 启动时后台检查并提示更新
4. 提供手动更新命令
diff --git a/src/commands/update.ts b/src/commands/update.ts
@@ -2,8 +2,10 @@
  * Update 命令 - Yargs 版本
  */
 
+import { execSync } from 'child_process';
 import type { CommandModule } from 'yargs';
 import type { UpdateOptions } from '../cli/types.js';
+import { checkVersion } from '../services/VersionChecker.js';
 
 export const updateCommands: CommandModule<{}, UpdateOptions> = {
   command: 'update',
@@ -12,23 +14,38 @@ export const updateCommands: CommandModule<{}, UpdateOptions> = {
     console.log('🔍 Checking for updates...');
 
     try {
-      // 读取当前版本
-      const fs = await import('fs/promises');
-      const path = await import('path');
-      const packageJsonPath = path.join(process.cwd(), 'package.json');
-      const packageJson = JSON.parse(await fs.readFile(packageJsonPath, 'utf-8'));
-      const currentVersion = packageJson.version;
+      const result = await checkVersion(true); // 强制检查，忽略缓存
 
-      console.log(`📦 Current version: ${currentVersion}`);
+      console.log(`📦 Current version: ${result.currentVersion}`);
 
-      // 模拟检查更新（实际项目中应该检查 npm registry 或 GitHub releases）
-      console.log('✅ You are running the latest version of Blade');
+      if (result.error) {
+        console.log(`⚠️  ${result.error}`);
+        return;
+      }
 
-      // 实际实现时可以添加：
-      // 1. 检查 npm registry 的最新版本
-      // 2. 比较版本号
-      // 3. 如果有更新，提示用户或自动更新
-      // 4. 显示更新日志
+      if (result.latestVersion) {
+        console.log(`📦 Latest version:  ${result.latestVersion}`);
+      }
+
+      if (result.hasUpdate && result.latestVersion) {
+        console.log('');
+        console.log(
+          `\x1b[33m⚠️  Update available: ${result.currentVersion} → ${result.latestVersion}\x1b[0m`
+        );
+        console.log('');
+        console.log('🚀 Updating...');
+        try {
+          execSync('npm install -g blade-code@latest', { stdio: 'inherit' });
+          console.log('');
+          console.log('✅ Update complete!');
+        } catch (_errrr) {
+          console.error('❌ Update failed. Please run manually:');
+          console.error('   npm install -g blade-code@latest');
+          process.exit(1);
+        }
+      } else {
+        console.log('✅ You are running the latest version of Blade');
+      }
     } catch (error) {
       console.error(
         `❌ Failed to check for updates: ${error instanceof Error ? error.message : '未知错误'}`
diff --git a/src/services/VersionChecker.ts b/src/services/VersionChecker.ts
@@ -0,0 +1,238 @@
+/**
+ * 版本检查服务
+ *
+ * 启动时检查 npm registry 获取最新版本，提示用户更新
+ */
+
+import * as fs from 'fs/promises';
+import * as path from 'path';
+import { fileURLToPath } from 'url';
+
+// 包名
+const PACKAGE_NAME = 'blade-code';
+
+// 缓存文件路径
+const CACHE_DIR = path.join(
+  process.env.HOME || process.env.USERPROFILE || '.',
+  '.blade'
+);
+const CACHE_FILE = path.join(CACHE_DIR, 'version-cache.json');
+
+// 缓存有效期：24 小时
+const CACHE_TTL = 24 * 60 * 60 * 1000;
+
+// npm registry URL
+const NPM_REGISTRY_URL = `https://registry.npmmirror.com/${PACKAGE_NAME}/latest`;
+
+interface VersionCache {
+  latestVersion: string;
+  checkedAt: number;
+}
+
+interface VersionCheckResult {
+  currentVersion: string;
+  latestVersion: string | null;
+  hasUpdate: boolean;
+  error?: string;
+}
+
+/**
+ * 获取当前安装的版本
+ */
+async function getCurrentVersion(): Promise<string> {
+  try {
+    // 方法1：从环境变量获取（构建时注入）
+    if (process.env.BLADE_VERSION) {
+      return process.env.BLADE_VERSION;
+    }
+
+    // 方法2：从 package.json 获取
+    // 获取当前模块的目录
+    const __filename = fileURLToPath(import.meta.url);
+    const __dirname = path.dirname(__filename);
+
+    // 尝试多个可能的 package.json 位置
+    const possiblePaths = [
+      path.join(__dirname, '..', '..', 'package.json'), // src/services -> root
+      path.join(__dirname, '..', 'package.json'), // dist -> root
+      path.join(process.cwd(), 'package.json'), // 当前工作目录
+    ];
+
+    for (const pkgPath of possiblePaths) {
+      try {
+        const content = await fs.readFile(pkgPath, 'utf-8');
+        const pkg = JSON.parse(content);
+        if (pkg.name === PACKAGE_NAME && pkg.version) {
+          return pkg.version;
+        }
+      } catch {
+        // 继续尝试下一个路径
+      }
+    }
+
+    return 'unknown';
+  } catch {
+    return 'unknown';
+  }
+}
+
+/**
+ * 从缓存读取版本信息
+ */
+async function readCache(): Promise<VersionCache | null> {
+  try {
+    const content = await fs.readFile(CACHE_FILE, 'utf-8');
+    const cache: VersionCache = JSON.parse(content);
+
+    // 检查缓存是否过期
+    if (Date.now() - cache.checkedAt < CACHE_TTL) {
+      return cache;
+    }
+    return null;
+  } catch {
+    return null;
+  }
+}
+
+/**
+ * 写入缓存
+ */
+async function writeCache(cache: VersionCache): Promise<void> {
+  try {
+    await fs.mkdir(CACHE_DIR, { recursive: true });
+    await fs.writeFile(CACHE_FILE, JSON.stringify(cache, null, 2));
+  } catch {
+    // 静默失败
+  }
+}
+
+/**
+ * 从 npm registry 获取最新版本
+ */
+async function fetchLatestVersion(): Promise<string | null> {
+  try {
+    const controller = new AbortController();
+    const timeout = setTimeout(() => controller.abort(), 5000); // 5 秒超时
+
+    const response = await fetch(NPM_REGISTRY_URL, {
+      signal: controller.signal,
+      headers: {
+        Accept: 'application/json',
+      },
+    });
+
+    clearTimeout(timeout);
+
+    if (!response.ok) {
+      return null;
+    }
+
+    const data = await response.json();
+    return data.version || null;
+  } catch {
+    return null;
+  }
+}
+
+/**
+ * 比较版本号
+ * 返回: 1 如果 a > b, -1 如果 a < b, 0 如果相等
+ */
+function compareVersions(a: string, b: string): number {
+  const partsA = a.split('.').map(Number);
+  const partsB = b.split('.').map(Number);
+
+  for (let i = 0; i < Math.max(partsA.length, partsB.length); i++) {
+    const numA = partsA[i] || 0;
+    const numB = partsB[i] || 0;
+
+    if (numA > numB) return 1;
+    if (numA < numB) return -1;
+  }
+
+  return 0;
+}
+
+/**
+ * 检查版本更新
+ *
+ * @param forceCheck - 强制检查（忽略缓存）
+ * @returns 版本检查结果
+ */
+export async function checkVersion(forceCheck = false): Promise<VersionCheckResult> {
+  const currentVersion = await getCurrentVersion();
+
+  // 如果无法获取当前版本，跳过检查
+  if (currentVersion === 'unknown') {
+    return {
+      currentVersion,
+      latestVersion: null,
+      hasUpdate: false,
+      error: 'Unable to determine current version',
+    };
+  }
+
+  // 尝试从缓存读取
+  if (!forceCheck) {
+    const cache = await readCache();
+    if (cache) {
+      return {
+        currentVersion,
+        latestVersion: cache.latestVersion,
+        hasUpdate: compareVersions(cache.latestVersion, currentVersion) > 0,
+      };
+    }
+  }
+
+  // 从 npm 获取最新版本
+  const latestVersion = await fetchLatestVersion();
+
+  if (latestVersion) {
+    // 更新缓存
+    await writeCache({
+      latestVersion,
+      checkedAt: Date.now(),
+    });
+
+    return {
+      currentVersion,
+      latestVersion,
+      hasUpdate: compareVersions(latestVersion, currentVersion) > 0,
+    };
+  }
+
+  return {
+    currentVersion,
+    latestVersion: null,
+    hasUpdate: false,
+    error: 'Unable to check for updates',
+  };
+}
+
+/**
+ * 格式化版本更新提示消息
+ */
+export function formatUpdateMessage(result: VersionCheckResult): string | null {
+  if (!result.hasUpdate || !result.latestVersion) {
+    return null;
+  }
+
+  return (
+    `\x1b[33m⚠️  Update available: ${result.currentVersion} → ${result.latestVersion}\x1b[0m\n` +
+    `   Run \x1b[36mnpm install -g ${PACKAGE_NAME}@latest\x1b[0m to update`
+  );
+}
+
+/**
+ * 启动时版本检查（后台执行，不阻塞）
+ *
+ * @returns Promise<string | null> 更新提示消息，如果没有更新则返回 null
+ */
+export async function checkVersionOnStartup(): Promise<string | null> {
+  try {
+    const result = await checkVersion();
+    return formatUpdateMessage(result);
+  } catch {
+    return null;
+  }
+}
diff --git a/src/ui/App.tsx b/src/ui/App.tsx
@@ -10,6 +10,7 @@ import {
 } from '../config/index.js';
 import { HookManager } from '../hooks/HookManager.js';
 import { Logger } from '../logging/Logger.js';
+import { checkVersionOnStartup } from '../services/VersionChecker.js';
 import { appActions, getState } from '../store/vanilla.js';
 import { BladeInterface } from './components/BladeInterface.js';
 import { ErrorBoundary } from './components/ErrorBoundary.js';
@@ -122,6 +123,15 @@ export const AppWrapper: React.FC<AppProps> = (props) => {
         }
       }
 
+      // 8. 后台检查版本更新（不阻塞启动）
+      checkVersionOnStartup().then((updateMessage) => {
+        if (updateMessage) {
+          console.log('');
+          console.log(updateMessage);
+          console.log('');
+        }
+      });
+
       setIsInitialized(true);
     } catch (error) {
       // 静默失败，使用默认配置
