fnva 0.0.23 → 0.0.25

package/lib/encoding-utils.js

@@ -0,0 +1,152 @@
+/**
+ * 编码处理工具模块
+ * 用于统一处理跨平台字符编码问题
+ */
+
+const fs = require('fs');
+const path = require('path');
+
+/**
+ * 编码处理工具类
+ */
+class EncodingUtils {
+  /**
+   * 设置Windows控制台编码为UTF-8
+   */
+  static setWindowsConsoleEncoding() {
+    if (process.platform === 'win32') {
+      try {
+        // 尝试设置控制台编码为UTF-8
+        const { execSync } = require('child_process');
+        execSync('chcp 65001 > nul 2>&1', { stdio: 'ignore' });
+
+        // 设置Node.js输出编码
+        if (process.stdout._handle && process.stdout._handle.setEncoding) {
+          process.stdout._handle.setEncoding('utf8');
+        }
+        if (process.stderr._handle && process.stderr._handle.setEncoding) {
+          process.stderr._handle.setEncoding('utf8');
+        }
+      } catch (error) {
+        // 静默忽略错误，避免影响正常功能
+      }
+    }
+  }
+
+  /**
+   * 为文件内容添加适当的编码标记
+   * @param {string} content - 文件内容
+   * @param {string} filePath - 文件路径（用于确定文件类型）
+   * @returns {string} - 处理后的内容
+   */
+  static addEncodingSignature(content, filePath) {
+    // 为Windows PowerShell脚本添加BOM
+    if (process.platform === 'win32' && this.isPowerShellScript(filePath)) {
+      return '\ufeff' + content;
+    }
+    return content;
+  }
+
+  /**
+   * 检查文件是否为PowerShell脚本
+   * @param {string} filePath - 文件路径
+   * @returns {boolean} - 是否为PowerShell脚本
+   */
+  static isPowerShellScript(filePath) {
+    const ext = path.extname(filePath).toLowerCase();
+    return ext === '.ps1';
+  }
+
+  /**
+   * 安全地写入文件，自动处理编码
+   * @param {string} filePath - 文件路径
+   * @param {string} content - 文件内容
+   * @param {string} encoding - 编码格式，默认为utf8
+   */
+  static writeFileWithEncoding(filePath, content, encoding = 'utf8') {
+    const processedContent = this.addEncodingSignature(content, filePath);
+    fs.writeFileSync(filePath, processedContent, encoding);
+  }
+
+  /**
+   * 生成PowerShell编码设置脚本
+   * @returns {string} - PowerShell编码设置代码
+   */
+  static generatePowerShellEncodingSetup() {
+    return [
+      '# 设置UTF-8编码以正确显示中文',
+      '[Console]::OutputEncoding = [System.Text.Encoding]::UTF8',
+      '$OutputEncoding = [System.Console]::OutputEncoding',
+      ''
+    ].join('\n');
+  }
+
+  /**
+   * 检测系统默认编码
+   * @returns {string} - 系统编码名称
+   */
+  static detectSystemEncoding() {
+    if (process.platform === 'win32') {
+      try {
+        const { execSync } = require('child_process');
+        const result = execSync('chcp', { encoding: 'utf8' });
+        const match = result.match(/活动代码页: (\d+)/);
+        return match ? `cp${match[1]}` : 'utf8';
+      } catch (e) {
+        return 'cp936'; // Windows中文默认编码
+      }
+    } else {
+      return 'utf8'; // Unix-like系统默认UTF-8
+    }
+  }
+
+  /**
+   * 创建临时PowerShell脚本文件
+   * @param {string} content - 脚本内容
+   * @param {string} prefix - 文件名前缀
+   * @returns {string} - 临时文件路径
+   */
+  static createTempPowerShellScript(content, prefix = 'fnva') {
+    const os = require('os');
+    const path = require('path');
+
+    const tempDir = os.tmpdir();
+    const timestamp = Date.now();
+    const scriptFile = path.join(tempDir, `${prefix}_${timestamp}.ps1`);
+
+    // 在脚本开头添加编码设置
+    const encodingSetup = this.generatePowerShellEncodingSetup();
+    const fullContent = encodingSetup + content;
+
+    this.writeFileWithEncoding(scriptFile, fullContent);
+
+    return scriptFile;
+  }
+
+  /**
+   * 安全地执行PowerShell脚本，确保编码正确
+   * @param {string} scriptPath - 脚本路径
+   * @param {Array} args - 传递给PowerShell的参数
+   * @param {Object} options - 执行选项
+   * @returns {Object} - 执行结果
+   */
+  static executePowerShellScript(scriptPath, args = [], options = {}) {
+    const { spawn } = require('child_process');
+
+    const defaultOptions = {
+      stdio: 'inherit',
+      shell: false,
+      ...options
+    };
+
+    const psArgs = [
+      '-ExecutionPolicy', 'Bypass',
+      '-File', scriptPath,
+      ...args
+    ];
+
+    return spawn('powershell', psArgs, defaultOptions);
+  }
+}
+
+module.exports = EncodingUtils;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "fnva",
-  "version": "0.0.23",
+  "version": "0.0.25",
   "description": "跨平台环境切换工具，支持 Java 和 LLM 环境配置",
   "author": "protagonistss",
   "license": "MIT",
@@ -14,10 +14,12 @@
     "uninstall-shell": "node scripts/uninstall-shell-integration.js",
     "build": "scripts/build-local.sh",
     "build:all": "scripts/build-all.sh",
-    "check-permissions": "node scripts/check-permissions.js"
+    "check-permissions": "node scripts/check-permissions.js",
+    "preuninstall": "node scripts/uninstall-shell-integration.js"
   },
   "files": [
     "bin/",
+    "lib/",
     "platforms/",
     "scripts/",
     "README.md",
@@ -48,3 +50,4 @@
     "node": ">=14.0.0"
   }
 }
+

package/scripts/install-shell-integration.js

@@ -77,11 +77,11 @@ fnva() {
         local temp_file=\$(mktemp)
         chmod +x "\$temp_file"
 
-        FNVA_AUTO_MODE=1 fnva "\$@" > "\$temp_file"
+        FNVA_AUTO_MODE=1 command fnva "\$@" > "\$temp_file"
         source "\$temp_file"
         rm -f "\$temp_file"
     else
-        FNVA_AUTO_MODE=1 fnva "\$@"
+        FNVA_AUTO_MODE=1 command fnva "\$@"
     fi
 }
 `;
@@ -94,11 +94,11 @@ function fnva
     if test (count \$argv) -ge 2; and string match -q -r "^(java|llm|cc)\$" \$argv[1]; and test \$argv[2] = "use"
         set temp_file (mktemp)
         chmod +x \$temp_file
-        env FNVA_AUTO_MODE=1 fnva \$argv > \$temp_file
+        env FNVA_AUTO_MODE=1 command fnva \$argv > \$temp_file
         source \$temp_file
         rm -f \$temp_file
     else
-        env FNVA_AUTO_MODE=1 fnva \$argv
+        env FNVA_AUTO_MODE=1 command fnva \$argv
     end
 end
 `;
@@ -242,4 +242,4 @@ module.exports = {
   getShellFunction,
   isFunctionInstalled,
   installShellIntegration
-};
+};

package/scripts/uninstall-shell-integration.js

@@ -11,126 +11,121 @@ function detectShell() {
   return process.env.SHELL?.split('/').pop() || 'bash';
 }
 
-function getShellConfigPath(shell) {
+function getShellConfigPaths(shell) {
   switch (shell) {
     case 'powershell':
-      return path.join(process.env.USERPROFILE || os.homedir(), 'Documents', 'WindowsPowerShell', 'Microsoft.PowerShell_profile.ps1');
+      return [path.join(process.env.USERPROFILE || os.homedir(), 'Documents', 'WindowsPowerShell', 'Microsoft.PowerShell_profile.ps1')];
     case 'bash':
-      return path.join(os.homedir(), '.bashrc');
+      return [path.join(os.homedir(), '.bashrc')];
     case 'zsh':
-      return path.join(os.homedir(), '.zshrc');
+      return [
+        path.join(os.homedir(), '.zshrc'),
+        path.join(os.homedir(), '.oh-my-zsh', 'custom', '.zshrc'),
+      ];
     case 'fish':
-      return path.join(os.homedir(), '.config', 'fish', 'config.fish');
+      return [path.join(os.homedir(), '.config', 'fish', 'config.fish')];
     default:
-      return null;
+      return [];
   }
 }
 
-function removeShellIntegration(configPath, shell) {
-  if (!fs.existsSync(configPath)) {
-    console.log(`⚠️  配置文件不存在: ${configPath}`);
-    return false;
-  }
-
-  try {
-    let content = fs.readFileSync(configPath, 'utf8');
-    const originalContent = content;
+function cleanConfigFile(cfgPath) {
+  let content = fs.readFileSync(cfgPath, 'utf8');
+  const originalContent = content;
 
-    // 方法1: 查找标记，精确删除整个函数块
-    const marker = '# fnva 自动化函数 - 由 npm 安装自动添加';
-    const startIndex = content.indexOf(marker);
+  const marker = '# fnva 自动化函数 - 用 npm 安装自动添加';
+  const startIndex = content.indexOf(marker);
 
-    if (startIndex !== -1) {
-      // 找到标记前的换行符
-      const beforeMarker = content.substring(0, startIndex).trimEnd();
+  if (startIndex !== -1) {
+    const beforeMarker = content.substring(0, startIndex).trimEnd();
+    const afterMarker = content.substring(startIndex);
+    const lines = afterMarker.split('\n');
 
-      // 从标记开始查找完整的函数
-      const afterMarker = content.substring(startIndex);
-      const lines = afterMarker.split('\n');
+    let functionEndIndex = -1;
+    let braceCount = 0;
+    let foundFunction = false;
 
-      let functionEndIndex = -1;
-      let braceCount = 0;
-      let foundFunction = false;
+    for (let i = 0; i < lines.length; i++) {
+      const line = lines[i];
+      if (line.includes('function fnva') || line.includes('fnva(')) {
+        foundFunction = true;
+      }
 
-      for (let i = 0; i < lines.length; i++) {
-        const line = lines[i];
-        if (line.includes('function fnva') || line.includes('fnva(')) {
-          foundFunction = true;
+      if (foundFunction) {
+        for (const char of line) {
+          if (char === '{') braceCount++;
+          if (char === '}') braceCount--;
         }
-
-        if (foundFunction) {
-          // 计算大括号
-          for (const char of line) {
-            if (char === '{') braceCount++;
-            if (char === '}') braceCount--;
-          }
-
-          // 当大括号平衡时，函数结束
-          if (braceCount === 0) {
-            functionEndIndex = i + 1;
-            break;
-          }
+        if (braceCount === 0) {
+          functionEndIndex = i + 1;
+          break;
         }
       }
-
-      if (functionEndIndex !== -1) {
-        // 重建内容
-        const afterFunction = lines.slice(functionEndIndex).join('\n');
-        content = beforeMarker + '\n' + afterFunction;
-      } else {
-        console.log('⚠️  无法确定函数结束位置');
-        return false;
-      }
     }
 
-    // 方法2: 如果没找到标记，使用正则表达式清理任何 fnva 相关内容
-    if (content === originalContent) {
-      // 使用正则表达式删除任何包含 fnva 的行和相关的环境变量处理
-      content = content
-        // 删除标记到函数结束的所有内容
-        .replace(/# fnva 自动化函数 - 由 npm 安装自动添加[\s\S]*?(?=\n\S|\n$)/g, '')
-        // 删除剩余的 fnva 相关行
-        .replace(/.*fnva.*\n?/g, '')
-        // 删除 FNVAAUTOMODE 相关行
-        .replace(/.*FNVAAUTOMODE.*\n?/g, '')
-        // 删除 cmd.exe 调用 fnva 的行
-        .replace(/.*cmd\.exe.*fnva.*\n?/g, '')
-        // 清理多余的空行
-        .replace(/\n{3,}/g, '\n\n')
-        .trim() + '\n';
+    if (functionEndIndex !== -1) {
+      const afterFunction = lines.slice(functionEndIndex).join('\n');
+      content = beforeMarker + '\n' + afterFunction;
     }
+  }
+
+  // 正则兜底：移除残留 fnva 片段
+  if (content === originalContent) {
+    content = content
+      .replace(/# fnva 自动化函数 - 用 npm 安装自动添加[\s\S]*?(?=\n\S|\n$)/g, '')
+      .replace(/.*fnva.*\n?/g, '')
+      .replace(/.*FNVAAUTOMODE.*\n?/g, '')
+      .replace(/.*cmd\.exe.*fnva.*\n?/g, '')
+      .replace(/\n{3,}/g, '\n\n')
+      .trim() + '\n';
+  }
+
+  if (content !== originalContent) {
+    fs.writeFileSync(cfgPath, content);
+    console.log(`✅ fnva shell 集成已从 ${cfgPath} 移除`);
+    return true;
+  }
+
+  console.log(`⚠️  未在 ${cfgPath} 找到需要清理的内容`);
+  return false;
+}
 
-    // 如果内容有变化，写入文件
-    if (content !== originalContent) {
-      fs.writeFileSync(configPath, content);
-      console.log(`✅ fnva shell 集成已从 ${configPath} 移除`);
-      return true;
-    } else {
-      console.log('⚠️  未找到需要清理的内容');
-      return false;
+function removeShellIntegration(configPath, shell) {
+  const paths = getShellConfigPaths(shell);
+  if (configPath) paths.unshift(configPath); // 兼容传入单一路径
+
+  let removedAny = false;
+  for (const cfgPath of paths) {
+    if (!cfgPath || !fs.existsSync(cfgPath)) continue;
+    try {
+      const removed = cleanConfigFile(cfgPath);
+      removedAny = removedAny || removed;
+    } catch (error) {
+      console.log(`❌ 移除失败 (${cfgPath}): ${error.message}`);
     }
-  } catch (error) {
-    console.log(`❌ 移除失败: ${error.message}`);
-    return false;
   }
+
+  if (!removedAny) {
+    console.log('⚠️  未找到可清理的 shell 配置文件或未匹配到 fnva 片段');
+  }
+  return removedAny;
 }
 
 function main() {
-  console.log('🔧 fnva shell 集成卸载器');
+  console.log('🧹 fnva shell 集成卸载');
 
   const shell = detectShell();
-  const configPath = getShellConfigPath(shell);
+  const paths = getShellConfigPaths(shell);
 
-  if (!configPath) {
-    console.log(`❌ 不支持的 shell: ${shell}`);
+  if (paths.length === 0) {
+    console.log(`⚠️  不支持的 shell: ${shell}`);
     return;
   }
 
-  const success = removeShellIntegration(configPath, shell);
+  const success = removeShellIntegration(null, shell);
 
   if (success) {
     console.log('🔄 请重新加载你的 shell 配置:');
-
     switch (shell) {
       case 'powershell':
         console.log('   . $PROFILE');
@@ -154,6 +149,6 @@ if (require.main === module) {
 
 module.exports = {
   detectShell,
-  getShellConfigPath,
-  removeShellIntegration
-};
+  getShellConfigPaths,
+  removeShellIntegration,
+};