fnva 0.0.10 → 0.0.12

package/README.md

@@ -28,6 +28,17 @@ yarn global add fnva
 
 # 使用 pnpm
 pnpm add -g fnva
+
+function fnva {
+  if ($args.Count -ge 2 -and ($args[0] -eq "java" -or $args[0] -eq "llm" -or $args[0] -eq "cc") -and $args[1] -eq "use") {
+      $tempFile = "$env:TEMP\fnva_script_$(Get-Random).ps1"
+      & node bin\fnva.js $args | Out-File -FilePath $tempFile -Encoding UTF8
+      & $tempFile
+      Remove-Item $tempFile -ErrorAction SilentlyContinue
+  } else {
+      & node bin\fnva.js $args
+  }
+}
 ```
 
 ### 方式二：从 Releases 下载二进制文件

package/bin/fnva.js

@@ -38,6 +38,11 @@ function platformBinaryPath(platformOverride) {
 }
 
 function buildBinaryPath() {
+  // 如果设置了 FNVA_SKIP_NATIVE，跳过原生二进制查找
+  if (process.env.FNVA_SKIP_NATIVE === '1') {
+    return null;
+  }
+
   const platform = resolvePlatform();
   const binaryCandidates = [];
 
@@ -103,6 +108,43 @@ function hasSessionFlag(args) {
   return args.includes('--session');
 }
 
+function hasApplyFlag(args) {
+  return args.includes('--apply');
+}
+
+function hasAutoExecuteFlag(args) {
+  return args.includes('--auto');
+}
+
+function removeAutoFlag(args) {
+  const index = args.indexOf('--auto');
+  if (index > -1) {
+    return args.slice(0, index).concat(args.slice(index + 1));
+  }
+  return args;
+}
+
+function createTempScriptFile(script, envType, envName) {
+  try {
+    const os = require('os');
+    const fs = require('fs');
+    const path = require('path');
+
+    const tempDir = os.tmpdir();
+    const scriptFile = path.join(tempDir, `fnva_${envType}_${envName}_${Date.now()}.ps1`);
+
+    fs.writeFileSync(scriptFile, script, 'utf8');
+
+    console.log('');
+    console.log('💡 环境已切换到当前进程。要在新的 PowerShell 窗口中使用此环境，运行：');
+    console.log(`   ${scriptFile}`);
+    console.log('   或者: fnva', envType, 'use', envName, '--auto');
+
+  } catch (error) {
+    console.warn('⚠️  无法创建临时脚本文件:', error.message);
+  }
+}
+
 function parseEnvironmentScript(scriptContent) {
   if (!scriptContent || scriptContent.trim() === '') {
     return {};
@@ -120,10 +162,25 @@ function parseEnvironmentScript(scriptContent) {
     const trimmedLine = line.trim();
 
     // 解析 PowerShell 环境变量设置
-    if (trimmedLine.startsWith('$env:')) {
-      const match = trimmedLine.match(/\$env:(\w+)\s*=\s*"([^"]*)"/);
+    if (trimmedLine.includes('$env:')) {
+      // 匹配 $env:VARNAME = "value" 格式
+      let match = trimmedLine.match(/\$env:(\w+)\s*=\s*"([^"]*)"/);
+      if (match) {
+        envVars[match[1]] = match[2];
+        continue;
+      }
+
+      // 匹配 $env:VARNAME = 'value' 格式
+      match = trimmedLine.match(/\$env:(\w+)\s*=\s*'([^']*)'/);
       if (match) {
         envVars[match[1]] = match[2];
+        continue;
+      }
+
+      // 匹配 $env:VARNAME = value 格式（不带引号）
+      match = trimmedLine.match(/\$env:(\w+)\s*=\s*([^;]+)/);
+      if (match) {
+        envVars[match[1]] = match[2].trim().replace(/['"]/g, '');
       }
     }
 
@@ -134,12 +191,6 @@ function parseEnvironmentScript(scriptContent) {
         envVars[match[1]] = match[2];
       }
     }
-
-    // 解析不带引号的环境变量设置
-    const unquotedMatch = trimmedLine.match(/\$env:(\w+)\s*=\s*([^;]+)/);
-    if (unquotedMatch) {
-      envVars[unquotedMatch[1]] = unquotedMatch[2].trim();
-    }
   }
 
   return envVars;
@@ -167,20 +218,169 @@ function displaySuccessMessage(envType, envName, envVars) {
   }
 }
 
+function generateSimpleScript(envVars, envType, envName) {
+  const lines = [];
+
+  if (process.platform === 'win32') {
+    // Windows PowerShell
+    lines.push(`Write-Host "Switched to ${envType} environment: ${envName}" -ForegroundColor Green`);
+
+    if (envVars.JAVA_HOME) {
+      lines.push(`$env:JAVA_HOME = "${envVars.JAVA_HOME}"`);
+      // 对于 PATH，我们需要智能处理：移除旧的 Java 路径，添加新的
+      lines.push(`# Remove existing Java paths from PATH`);
+      lines.push(`$pathParts = $env:PATH -split ';'`);
+      lines.push(`$cleanPath = @()`);
+      lines.push(`foreach ($part in $pathParts) {`);
+      lines.push(`    if ($part -notmatch 'java' -and $part -notmatch 'jdk') {`);
+      lines.push(`        $cleanPath += $part`);
+      lines.push(`    }`);
+      lines.push(`}`);
+      lines.push(`$env:PATH = "${envVars.JAVA_HOME}\\bin;" + ($cleanPath -join ';')`);
+      lines.push(`Write-Host "JAVA_HOME: $env:JAVA_HOME" -ForegroundColor Yellow`);
+    }
+
+    if (envVars.ANTHROPIC_AUTH_TOKEN) {
+      lines.push(`$env:ANTHROPIC_AUTH_TOKEN = "${envVars.ANTHROPIC_AUTH_TOKEN}"`);
+      lines.push(`Write-Host "ANTHROPIC_AUTH_TOKEN: [已设置]" -ForegroundColor Yellow`);
+    }
+
+    if (envVars.OPENAI_API_KEY) {
+      lines.push(`$env:OPENAI_API_KEY = "${envVars.OPENAI_API_KEY}"`);
+      lines.push(`Write-Host "OPENAI_API_KEY: [已设置]" -ForegroundColor Yellow`);
+    }
+  } else {
+    // Unix-like systems
+    lines.push(`echo "Switched to ${envType} environment: ${envName}"`);
+
+    if (envVars.JAVA_HOME) {
+      lines.push(`export JAVA_HOME="${envVars.JAVA_HOME}"`);
+      // 对于 PATH，我们也需要智能处理
+      lines.push(`# Remove existing Java paths from PATH`);
+      lines.push(`echo $PATH | tr ':' '\\n' | grep -v java | grep -v jdk | tr '\\n' ':' | sed 's/:$//' > /tmp/clean_path`);
+      lines.push(`export PATH="${envVars.JAVA_HOME}/bin:$(cat /tmp/clean_path)"`);
+      lines.push(`rm -f /tmp/clean_path`);
+      lines.push(`echo "JAVA_HOME: $JAVA_HOME"`);
+    }
+
+    if (envVars.ANTHROPIC_AUTH_TOKEN) {
+      lines.push(`export ANTHROPIC_AUTH_TOKEN="${envVars.ANTHROPIC_AUTH_TOKEN}"`);
+      lines.push(`echo "ANTHROPIC_AUTH_TOKEN: [已设置]"`);
+    }
+
+    if (envVars.OPENAI_API_KEY) {
+      lines.push(`export OPENAI_API_KEY="${envVars.OPENAI_API_KEY}"`);
+      lines.push(`echo "OPENAI_API_KEY: [已设置]"`);
+    }
+  }
+
+  return lines.join('\n');
+}
+
+function handleNodeOnlyMode(args) {
+  const fs = require('fs');
+  const path = require('path');
+  const os = require('os');
+
+  // 简单的命令处理
+  if (args.length === 0) {
+    console.log('fnva - 环境管理工具 (Node.js 模式)');
+    console.log('');
+    console.log('支持的命令:');
+    console.log('  java list     - 列出 Java 环境');
+    console.log('  java use <n>  - 切换 Java 环境');
+    console.log('');
+    console.log('注意: Node.js 模式功能有限，建议使用原生二进制版本。');
+    return;
+  }
+
+  if (args[0] === 'java') {
+    const homeDir = os.homedir();
+    const fnvaDir = path.join(homeDir, '.fnva', 'java-packages');
+
+    if (args[1] === 'list') {
+      if (!fs.existsSync(fnvaDir)) {
+        console.log('No Java environments found');
+        return;
+      }
+
+      const versions = fs.readdirSync(fnvaDir, { withFileTypes: true })
+        .filter(dirent => dirent.isDirectory())
+        .map(dirent => dirent.name)
+        .sort();
+
+      if (versions.length === 0) {
+        console.log('No Java environments found');
+      } else {
+        console.log('Available java environments:');
+        versions.forEach(version => {
+          const jdkPath = path.join(fnvaDir, version);
+          if (fs.existsSync(path.join(jdkPath, 'release'))) {
+            try {
+              const releaseContent = fs.readFileSync(path.join(jdkPath, 'release'), 'utf8');
+              const versionMatch = releaseContent.match(/JAVA_VERSION="(.+)"/);
+              const javaVersion = versionMatch ? versionMatch[1].replace(/"/g, '') : 'Unknown';
+              console.log(`  ${version}: Java ${javaVersion} (${jdkPath})`);
+            } catch (e) {
+              console.log(`  ${version}: (${jdkPath})`);
+            }
+          }
+        });
+      }
+    } else if (args[1] === 'use' && args[2]) {
+      const version = args[2];
+      const jdkPath = path.join(fnvaDir, version);
+
+      if (!fs.existsSync(jdkPath)) {
+        console.error(`Java environment '${version}' not found`);
+        process.exit(1);
+      }
+
+      // 生成环境切换脚本
+      const envVars = {
+        JAVA_HOME: jdkPath
+      };
+
+      const script = generateSimpleScript(envVars, 'java', version);
+      console.log(script);
+    } else {
+      console.error('Usage: fnva java <list|use <version>>');
+      process.exit(1);
+    }
+  } else {
+    console.error(`Command '${args[0]}' not supported in Node.js mode`);
+    process.exit(1);
+  }
+}
+
 function run() {
   const binaryPath = buildBinaryPath();
 
   if (!binaryPath) {
+    if (process.env.FNVA_SKIP_NATIVE === '1') {
+      // 纯 Node.js 模式 - 实现基本的环境切换功能
+      const args = process.argv.slice(2);
+      handleNodeOnlyMode(args);
+      return;
+    }
+
     console.error('Error: fnva native binary not found.');
     console.error('');
     console.error("Please either:");
     console.error("  1) Run 'npm run build' (or 'npm run build:all') to produce platform binaries,");
     console.error("  2) Install a release package that includes the platforms directory, or");
     console.error("  3) Set FNVA_NATIVE_PATH to the full path of an existing fnva executable.");
+    console.error("  4) Set FNVA_SKIP_NATIVE=1 to use Node.js mode (limited functionality).");
     process.exit(1);
   }
 
-  const args = process.argv.slice(2);
+  let args = process.argv.slice(2);
+
+  // 如果设置了 FNVA_AUTO_EXECUTE，则为环境切换命令启用自动执行
+  if (process.env.FNVA_AUTO_EXECUTE === '1' && isEnvironmentSwitchCommand(args) && !hasSessionFlag(args)) {
+    // 添加 --auto 标志来启用自动执行
+    args = args.concat('--auto');
+  }
   const isSwitchCommand = isEnvironmentSwitchCommand(args);
 
   if (isSwitchCommand) {
@@ -248,13 +448,45 @@ function run() {
             console.log(`📝 Script was: ${script}`);
           }
         } else {
-          console.log(`✅ Switched to ${envType} environment: ${envName}`);
-          if (process.stdout.isTTY) {
-            console.log('');
-            console.log('💡 在当前会话应用环境：');
-            console.log(`  fnva ${envType} use ${envName} --shell powershell | Invoke-Expression`);
+          // 检查是否使用了 --apply 参数
+          if (hasApplyFlag(args)) {
+            // 直接应用环境变量到当前进程
+            const envVars = parseEnvironmentScript(script);
+            applyEnvironmentVariables(envVars);
+            displaySuccessMessage(envType, envName, envVars);
           } else {
-            process.stdout.write(script);
+            // 在 Windows 中，智能处理环境设置
+            const envVars = parseEnvironmentScript(script);
+            const simpleScript = generateSimpleScript(envVars, envType, envName);
+
+            // 尝试自动执行（如果可能）
+            if (process.env.FNVA_AUTO_EXECUTE === '1') {
+              const os = require('os');
+              const fs = require('fs');
+              const path = require('path');
+              const { spawn } = require('child_process');
+
+              try {
+                const tempFile = path.join(os.tmpdir(), `fnva_auto_${Date.now()}.ps1`);
+                fs.writeFileSync(tempFile, simpleScript, 'utf8');
+
+                // 使用 PowerShell 执行脚本
+                spawn('powershell', ['-ExecutionPolicy', 'Bypass', '-File', tempFile], {
+                  stdio: 'inherit',
+                  shell: false
+                }).on('exit', () => {
+                  try { fs.unlinkSync(tempFile); } catch (_) {}
+                });
+
+                console.log('✅ 环境已自动切换');
+                return;
+              } catch (error) {
+                console.warn('⚠️  自动执行失败，回退到脚本输出');
+              }
+            }
+
+            // 默认输出脚本
+            process.stdout.write(simpleScript);
           }
         }
       } else {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "fnva",
-  "version": "0.0.10",
+  "version": "0.0.12",
   "description": "跨平台环境切换工具，支持 Java 和 LLM 环境配置",
   "author": "protagonistss",
   "license": "MIT",