@iflow-ai/iflow-cli 0.2.1 → 0.2.2

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@iflow-ai/iflow-cli",
-  "version": "0.2.1",
+  "version": "0.2.2",
   "engines": {
     "node": ">=20.0.0"
   },

package/scripts/postinstallidea.js

@@ -9,6 +9,7 @@ import { join, dirname, resolve } from 'path';
 import { fileURLToPath } from 'url';
 import os from 'os';
 import https from 'https';
+import http from 'http'; // 添加HTTP模块导入
 import { createHash } from 'crypto';
 import { pipeline } from 'stream/promises';
 import { Extract } from 'unzipper';
@@ -17,8 +18,20 @@ import { Extract } from 'unzipper';
 const PLUGIN_URL = 'https://cloud.iflow.cn/iflow-cli/iflow-idea-0.0.1.zip';
 const PLUGIN_DIR_NAME = 'iflow-idea';
 const TEMP_DIR_NAME = 'tstar-cli-idea-plugin';
-const LOG_PREFIX = '[IDEA Extension]';
-const IDEA_PATTERNS = ['IntelliJIdea', 'IdeaIC'];
+const LOG_PREFIX = '[JetBrains Extension]';
+// 扩展支持的JetBrains IDE产品
+const IDE_PATTERNS = [
+  'IntelliJIdea', 'IdeaIC',  // IntelliJ IDEA
+  'PyCharm', 'PyCharmCE',    // PyCharm
+  'GoLand',                  // GoLand
+  'WebStorm',                // WebStorm
+  'PhpStorm',                // PhpStorm
+  'CLion',                   // CLion
+  'RubyMine',                // RubyMine
+  'DataGrip',                // DataGrip
+  'Rider',                   // Rider
+  'AndroidStudio'            // Android Studio
+];
 
 // 获取当前脚本的目录
 const __filename = fileURLToPath(import.meta.url);
@@ -45,8 +58,8 @@ function isExecutable(filePath) {
   }
 }
 
-// 获取 IntelliJ IDEA 插件目录的跨平台函数
-function getIdeaPluginsDirectory() {
+// 获取 JetBrains IDE 插件目录的跨平台函数
+function getJetBrainsPluginsDirectory() {
   const platform = os.platform();
   const homedir = os.homedir();
 
@@ -62,23 +75,24 @@ function getIdeaPluginsDirectory() {
   }
 }
 
-// 查找最新版本的 IntelliJ IDEA 目录
-function findLatestIdeaDirectory() {
-  const jetbrainsDir = getIdeaPluginsDirectory();
+// 查找最新版本的 JetBrains IDE 目录
+function findLatestJetBrainsDirectories() {
+  const jetbrainsDir = getJetBrainsPluginsDirectory();
 
   if (!existsSync(jetbrainsDir)) {
     console.warn(`${LOG_PREFIX} JetBrains directory not found:`, jetbrainsDir);
-    return null;
+    return [];
   }
 
   try {
+    // 获取所有匹配的IDE目录，并按修改时间排序
     const dirs = readdirSync(jetbrainsDir)
-      .filter(dir => IDEA_PATTERNS.some(pattern => dir.startsWith(pattern)))
+      .filter(dir => IDE_PATTERNS.some(pattern => dir.startsWith(pattern)))
       .map(dir => {
         const fullPath = join(jetbrainsDir, dir);
         try {
           const stats = statSync(fullPath);
-          return { path: fullPath, mtime: stats.mtime };
+          return { path: fullPath, mtime: stats.mtime, name: dir };
         } catch (error) {
           console.warn(`${LOG_PREFIX} Cannot access directory ${fullPath}:`, error.message);
           return null;
@@ -88,14 +102,23 @@ function findLatestIdeaDirectory() {
       .sort((a, b) => b.mtime - a.mtime);
 
     if (dirs.length === 0) {
-      console.warn(`${LOG_PREFIX} No IntelliJ IDEA installation found`);
-      return null;
+      console.warn(`${LOG_PREFIX} No JetBrains IDE installation found`);
+      return [];
     }
 
-    return dirs[0].path;
+    // 按IDE类型分组，每种IDE只保留最新版本
+    const latestByType = {};
+    dirs.forEach(dir => {
+      const ideType = IDE_PATTERNS.find(pattern => dir.name.startsWith(pattern));
+      if (ideType && !latestByType[ideType]) {
+        latestByType[ideType] = dir;
+      }
+    });
+
+    return Object.values(latestByType);
   } catch (error) {
-    console.error(`${LOG_PREFIX} Error finding IntelliJ IDEA directory:`, error.message);
-    return null;
+    console.error(`${LOG_PREFIX} Error finding JetBrains IDE directories:`, error.message);
+    return [];
   }
 }
 
@@ -104,7 +127,10 @@ async function downloadPlugin(url, destPath) {
   return new Promise((resolve, reject) => {
     const file = createWriteStream(destPath);
 
-    https.get(url, (response) => {
+    // 根据URL协议选择适当的模块
+    const protocol = url.startsWith('https:') ? https : http;
+
+    protocol.get(url, (response) => {
       if (response.statusCode !== 200) {
         reject(new Error(`Failed to download plugin: ${response.statusCode}`));
         return;
@@ -145,41 +171,16 @@ async function extractPlugin(zipFilePath, destDir) {
   });
 }
 
-// 安装 IDEA 插件
-async function installIdeaExtension() {
+// 安装 JetBrains IDE 插件
+async function installJetBrainsExtension() {
   try {
-    // 查找 IDEA 目录
-    const ideaDir = findLatestIdeaDirectory();
-    if (!ideaDir) {
-      console.warn(`${LOG_PREFIX} Could not find IntelliJ IDEA installation, skipping plugin installation`);
+    // 查找所有支持的 JetBrains IDE 目录
+    const ideDirs = findLatestJetBrainsDirectories();
+    if (ideDirs.length === 0) {
+      console.warn(`${LOG_PREFIX} Could not find any JetBrains IDE installation, skipping plugin installation`);
       return;
     }
 
-    console.info(`${LOG_PREFIX} Found IntelliJ IDEA at: ${ideaDir}`);
-
-    // 创建插件目录路径
-    const pluginsDir = join(ideaDir, 'plugins');
-    if (!existsSync(pluginsDir)) {
-      try {
-        mkdirSync(pluginsDir, { recursive: true });
-      } catch (error) {
-        console.error(`${LOG_PREFIX} Failed to create plugins directory: ${pluginsDir}`, error.message);
-        return;
-      }
-    }
-
-    // 检查并删除已存在的插件目录
-    const existingPluginDir = join(pluginsDir, PLUGIN_DIR_NAME);
-    if (existsSync(existingPluginDir)) {
-      try {
-        console.info(`${LOG_PREFIX} Removing existing plugin directory: ${existingPluginDir}`);
-        rmSync(existingPluginDir, { recursive: true, force: true });
-      } catch (error) {
-        console.error(`${LOG_PREFIX} Failed to remove existing plugin directory: ${existingPluginDir}`, error.message);
-        // 继续安装，即使删除失败
-      }
-    }
-
     // 创建临时下载目录
     const tempDir = join(os.tmpdir(), TEMP_DIR_NAME);
     if (!existsSync(tempDir)) {
@@ -206,13 +207,44 @@ async function installIdeaExtension() {
       const fileHash = await calculateFileHash(pluginFilePath);
       console.info(`${LOG_PREFIX} Plugin file hash: ${fileHash}`);
 
-      // 解压插件到插件目录
-      console.info(`${LOG_PREFIX} Extracting plugin to: ${pluginsDir}`);
+      // 为每个找到的IDE安装插件
+      for (const ideDir of ideDirs) {
+        console.info(`${LOG_PREFIX} Found JetBrains IDE at: ${ideDir.path}`);
+
+        // 创建插件目录路径
+        const pluginsDir = join(ideDir.path, 'plugins');
+        if (!existsSync(pluginsDir)) {
+          try {
+            mkdirSync(pluginsDir, { recursive: true });
+          } catch (error) {
+            console.error(`${LOG_PREFIX} Failed to create plugins directory: ${pluginsDir}`, error.message);
+            continue; // 跳过此IDE，继续下一个
+          }
+        }
 
-      // 直接解压到插件目录，不创建额外的子目录
-      await extractPlugin(pluginFilePath, pluginsDir);
+        // 检查并删除已存在的插件目录
+        const existingPluginDir = join(pluginsDir, PLUGIN_DIR_NAME);
+        if (existsSync(existingPluginDir)) {
+          try {
+            console.info(`${LOG_PREFIX} Removing existing plugin directory: ${existingPluginDir}`);
+            rmSync(existingPluginDir, { recursive: true, force: true });
+          } catch (error) {
+            console.error(`${LOG_PREFIX} Failed to remove existing plugin directory: ${existingPluginDir}`, error.message);
+            // 继续安装，即使删除失败
+          }
+        }
 
-      console.info(`${LOG_PREFIX} ✅ Plugin installed successfully to: ${pluginsDir}`);
+        // 解压插件到插件目录
+        console.info(`${LOG_PREFIX} Extracting plugin to: ${pluginsDir}`);
+
+        try {
+          // 直接解压到插件目录，不创建额外的子目录
+          await extractPlugin(pluginFilePath, pluginsDir);
+          console.info(`${LOG_PREFIX} ✅ Plugin installed successfully to: ${pluginsDir}`);
+        } catch (error) {
+          console.error(`${LOG_PREFIX} Failed to install plugin for ${ideDir.name}:`, error.message);
+        }
+      }
 
       // 清理临时文件
       try {
@@ -226,14 +258,14 @@ async function installIdeaExtension() {
       console.debug(`${LOG_PREFIX} Stack trace:`, error.stack);
     }
   } catch (error) {
-    console.error(`${LOG_PREFIX} Unexpected error during IntelliJ IDEA extension installation:`, error.message);
+    console.error(`${LOG_PREFIX} Unexpected error during JetBrains extension installation:`, error.message);
     console.debug(`${LOG_PREFIX} Stack trace:`, error.stack);
   }
 }
 
 // 只在非 CI 环境执行
 if (!process.env.CI) {
-  installIdeaExtension().catch(error => {
+  installJetBrainsExtension().catch(error => {
     console.error(`${LOG_PREFIX} Installation failed:`, error.message);
   });
 }