aihezu 2.0.1 → 2.2.0

package/bin/aihezu.js

@@ -38,77 +38,95 @@ async function askQuestion(rl, question) {
 }
 
 async function main() {
-  // Removed global rl creation
-
   try {
-    if (args.length === 0 || args[0] === 'help') { // Handle 'aihezu' or 'aihezu help'
+    if (args.length === 0 || args[0] === 'help') {
       showHelp();
       return;
     }
 
     const commandName = args[0];
-    let serviceName = args[1];
-
     if (!commands[commandName]) {
-      console.error(`❌ 未知命��: ${commandName}`);
+      console.error(`❌ 未知命令: ${commandName}`);
       showHelp();
       process.exit(1);
     }
 
+    let serviceName = '';
+    let commandArgs = [];
+
+    // Check if the second argument is a service name or an option/flag
+    if (args.length > 1 && !args[1].startsWith('-')) {
+      serviceName = args[1].toLowerCase();
+      commandArgs = args.slice(2);
+    } else {
+      // No service name provided, or an option was provided instead.
+      // The command arguments will be all arguments after the command itself.
+      commandArgs = args.slice(1);
+    }
+
+    // If service name is not provided, and command is 'install', enter interactive mode.
     if (!serviceName) {
       if (commandName === 'install') {
-        // Init rl ONLY here for selection
         const rl = readline.createInterface({
           input: process.stdin,
           output: process.stdout
         });
 
         try {
-          // Interactive service selection for 'install' command
           console.log('\n请选择要安装的服务:');
           const serviceKeys = Object.keys(services);
+          const claudeIndex = serviceKeys.indexOf('claude');
+
           serviceKeys.forEach((key, index) => {
-            console.log(`  [${index + 1}] ${services[key].displayName}`);
+            const isDefault = (key === 'claude');
+            console.log(`  [${index + 1}] ${services[key].displayName}${isDefault ? ' (默认)' : ''}`);
           });
 
-          let chosenIndex;
+          let chosenIndex = claudeIndex !== -1 ? claudeIndex : 0; // Default to Claude
           let validChoice = false;
+
           while (!validChoice) {
-            const answer = await askQuestion(rl, '请输入选项编号: ');
-            chosenIndex = parseInt(answer, 10) - 1;
-            if (chosenIndex >= 0 && chosenIndex < serviceKeys.length) {
-              serviceName = serviceKeys[chosenIndex];
-              validChoice = true;
+            const answer = await askQuestion(rl, '请输入选项编号 (直接回车使用默认值): ');
+            if (answer === '') {
+              validChoice = true; // Use the default index
             } else {
-              console.log('无效的选择，请输入列表中的数字。');
+              const selectedIdx = parseInt(answer, 10) - 1;
+              if (selectedIdx >= 0 && selectedIdx < serviceKeys.length) {
+                chosenIndex = selectedIdx;
+                validChoice = true;
+              } else {
+                console.log('无效的选择，请输入列表中的数字。');
+              }
             }
           }
+          serviceName = serviceKeys[chosenIndex];
           console.log(`✅ 已选择: ${services[serviceName].displayName}`);
+
         } finally {
-          // Close rl immediately after selection and BEFORE calling the command
           rl.close();
         }
       } else {
-        // For 'clear' or other commands, service name is mandatory
+        // For other commands like 'clear', service name is mandatory.
         console.error(`❌ 请为 '${commandName}' 命令指定一个服务名称。`);
         showHelp();
         process.exit(1);
       }
     }
 
-    const service = services[serviceName.toLowerCase()];
+    const service = services[serviceName];
     if (!service) {
       console.error(`❌ 未知服务: ${serviceName}`);
       console.log('可用服务: ' + Object.keys(services).map(key => services[key].displayName).join(', '));
       process.exit(1);
     }
 
-    // Execute
-    await commands[commandName](service, args.slice(2));
+    // Execute the command with the correct service and arguments
+    await commands[commandName](service, commandArgs);
+    
   } catch (error) {
     console.error('❌ 发生错误:', error);
     process.exit(1);
   }
 }
 
-main();
+main();

package/commands/install.js

@@ -56,7 +56,23 @@ async function installCommand(service, args = []) {
     const apiUrlInput = await askQuestion(rl, '请输入 API 地址 (直接回车使用默认地址): ');
     let apiUrl = apiUrlInput || defaultUrl;
 
+    // Determine service-specific suffix
+    let suffix = '';
+    if (service.name === 'claude') {
+      suffix = '/api';
+    } else if (service.name === 'codex') {
+      suffix = '/openai';
+    } else if (service.name === 'gemini') {
+      suffix = '/gemini';
+    }
+
+    // Append suffix if apiUrl does not already end with it
+    if (suffix && !apiUrl.endsWith(suffix)) {
+      apiUrl = apiUrl.replace(/\/+$/, '') + suffix; // Remove any trailing slash and add suffix
+    }
+
     // Basic URL validation/normalization
+    // This part should handle http/https prefixing.
     if (!/^https?:\/\//i.test(apiUrl)) {
       apiUrl = 'https://' + apiUrl;
     }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "aihezu",
-  "version": "2.0.1",
+  "version": "2.2.0",
   "description": "AI 开发环境配置工具 - 支持 Claude Code, Codex, Google Gemini 的本地化配置、代理设置与缓存清理",
   "main": "bin/aihezu.js",
   "bin": {