@yivan-lab/pretty-please 1.4.0 → 1.5.1

package/dist/src/utils/platform.js

@@ -0,0 +1,331 @@
+import { execSync } from 'child_process';
+import path from 'path';
+import os from 'os';
+// ================== 平台检测 ==================
+/**
+ * 是否为 Windows 平台
+ */
+export function isWindows() {
+    return process.platform === 'win32';
+}
+/**
+ * 是否为 macOS 平台
+ */
+export function isMacOS() {
+    return process.platform === 'darwin';
+}
+/**
+ * 是否为 Linux 平台
+ */
+export function isLinux() {
+    return process.platform === 'linux';
+}
+// ================== Shell 检测 ==================
+/**
+ * 检测 Windows Shell 类型
+ */
+function detectWindowsShell() {
+    // 1. 检查 PSModulePath 判断 PowerShell 版本
+    const psModulePath = process.env.PSModulePath || '';
+    if (psModulePath) {
+        // PowerShell 7+ 的 PSModulePath 包含 "PowerShell\7" 或 "PowerShell/7"
+        if (/PowerShell[\/\\]7/i.test(psModulePath)) {
+            return 'powershell7';
+        }
+        // Windows PowerShell 5.x 的 PSModulePath 包含 "WindowsPowerShell"
+        if (/WindowsPowerShell/i.test(psModulePath)) {
+            return 'powershell5';
+        }
+    }
+    // 2. 检查 PROMPT 环境变量（CMD 特有）
+    // 注意：在 PowerShell 中运行 CMD 时也可能有 PROMPT
+    if (process.env.PROMPT && !psModulePath) {
+        return 'cmd';
+    }
+    // 3. 尝试检测 pwsh 是否可用
+    if (commandExists('pwsh')) {
+        return 'powershell7';
+    }
+    // 4. 默认 PowerShell 5（Windows 内置）
+    return 'powershell5';
+}
+/**
+ * 检测 Unix Shell 类型
+ */
+function detectUnixShell() {
+    const shell = process.env.SHELL || '';
+    if (shell.includes('zsh'))
+        return 'zsh';
+    if (shell.includes('bash'))
+        return 'bash';
+    if (shell.includes('fish'))
+        return 'fish';
+    return 'unknown';
+}
+/**
+ * 检测当前 Shell 类型
+ */
+export function detectShell() {
+    if (isWindows()) {
+        return detectWindowsShell();
+    }
+    return detectUnixShell();
+}
+// ================== Shell 能力 ==================
+/**
+ * 获取 Shell 配置文件路径
+ */
+function getShellConfigPath(shell) {
+    const home = os.homedir();
+    switch (shell) {
+        case 'zsh':
+            return path.join(home, '.zshrc');
+        case 'bash':
+            // macOS 使用 .bash_profile，Linux 使用 .bashrc
+            return isMacOS()
+                ? path.join(home, '.bash_profile')
+                : path.join(home, '.bashrc');
+        case 'fish':
+            return path.join(home, '.config', 'fish', 'config.fish');
+        case 'powershell5':
+            return path.join(home, 'Documents', 'WindowsPowerShell', 'Microsoft.PowerShell_profile.ps1');
+        case 'powershell7':
+            return path.join(home, 'Documents', 'PowerShell', 'Microsoft.PowerShell_profile.ps1');
+        case 'cmd':
+            return null; // CMD 不支持配置文件
+        default:
+            return null;
+    }
+}
+/**
+ * 获取 Shell 历史文件路径
+ */
+function getShellHistoryPath(shell) {
+    const home = os.homedir();
+    switch (shell) {
+        case 'zsh':
+            return process.env.HISTFILE || path.join(home, '.zsh_history');
+        case 'bash':
+            return process.env.HISTFILE || path.join(home, '.bash_history');
+        case 'fish':
+            return path.join(home, '.local', 'share', 'fish', 'fish_history');
+        case 'powershell5':
+        case 'powershell7':
+            // PowerShell 历史文件位置（PSReadLine）
+            const appData = process.env.APPDATA || path.join(home, 'AppData', 'Roaming');
+            return path.join(appData, 'Microsoft', 'Windows', 'PowerShell', 'PSReadLine', 'ConsoleHost_history.txt');
+        case 'cmd':
+            return null; // CMD 不持久化历史
+        default:
+            return null;
+    }
+}
+/**
+ * 获取 Shell 可执行文件
+ */
+function getShellExecutable(shell) {
+    switch (shell) {
+        case 'zsh':
+            return process.env.SHELL || '/bin/zsh';
+        case 'bash':
+            return process.env.SHELL || '/bin/bash';
+        case 'fish':
+            return process.env.SHELL || '/usr/bin/fish';
+        case 'powershell5':
+            return 'powershell.exe';
+        case 'powershell7':
+            return 'pwsh.exe';
+        case 'cmd':
+            return process.env.COMSPEC || 'cmd.exe';
+        default:
+            return isWindows() ? 'powershell.exe' : '/bin/sh';
+    }
+}
+/**
+ * 获取 Shell 显示名称
+ */
+function getShellDisplayName(shell) {
+    switch (shell) {
+        case 'zsh': return 'Zsh';
+        case 'bash': return 'Bash';
+        case 'fish': return 'Fish';
+        case 'cmd': return 'CMD';
+        case 'powershell5': return 'PowerShell 5.x';
+        case 'powershell7': return 'PowerShell 7+';
+        default: return 'Unknown';
+    }
+}
+/**
+ * 获取 Shell 能力信息
+ */
+export function getShellCapabilities(shell) {
+    return {
+        supportsHook: shell !== 'cmd' && shell !== 'unknown',
+        supportsHistory: shell !== 'cmd' && shell !== 'unknown',
+        configPath: getShellConfigPath(shell),
+        historyPath: getShellHistoryPath(shell),
+        executable: getShellExecutable(shell),
+        displayName: getShellDisplayName(shell),
+    };
+}
+// ================== 命令检测 ==================
+/**
+ * 检测命令是否存在（跨平台）
+ */
+export function commandExists(command) {
+    try {
+        if (isWindows()) {
+            // Windows: 使用 where 命令
+            execSync(`where ${command}`, { stdio: 'ignore' });
+        }
+        else {
+            // Unix: 使用 command -v（比 which 更可靠）
+            execSync(`command -v ${command}`, { stdio: 'ignore', shell: '/bin/sh' });
+        }
+        return true;
+    }
+    catch {
+        return false;
+    }
+}
+/**
+ * 批量检测命令是否存在（优化性能）
+ * @returns 返回存在的命令列表
+ */
+export function batchCommandExists(commands) {
+    if (commands.length === 0)
+        return [];
+    const available = [];
+    if (isWindows()) {
+        // Windows: 使用 PowerShell 批量检测
+        const batchSize = 20;
+        for (let i = 0; i < commands.length; i += batchSize) {
+            const batch = commands.slice(i, i + batchSize);
+            try {
+                // 使用 Get-Command 批量检测
+                const script = batch
+                    .map(cmd => `if(Get-Command ${cmd} -ErrorAction SilentlyContinue){Write-Output ${cmd}}`)
+                    .join(';');
+                const result = execSync(`powershell -NoProfile -Command "${script}"`, {
+                    encoding: 'utf-8',
+                    stdio: ['pipe', 'pipe', 'ignore'],
+                    timeout: 5000,
+                });
+                available.push(...result.trim().split(/\r?\n/).filter(Boolean));
+            }
+            catch {
+                // 这批失败，逐个检测
+                for (const cmd of batch) {
+                    if (commandExists(cmd)) {
+                        available.push(cmd);
+                    }
+                }
+            }
+        }
+    }
+    else {
+        // Unix: 使用 shell 批量检测
+        const batchSize = 20;
+        for (let i = 0; i < commands.length; i += batchSize) {
+            const batch = commands.slice(i, i + batchSize);
+            const script = `(${batch
+                .map(cmd => `command -v ${cmd} >/dev/null 2>&1 && echo ${cmd}`)
+                .join('; ')}) 2>/dev/null || true`;
+            try {
+                const result = execSync(script, {
+                    encoding: 'utf-8',
+                    stdio: ['pipe', 'pipe', 'ignore'],
+                    timeout: 500,
+                });
+                available.push(...result.trim().split('\n').filter(Boolean));
+            }
+            catch {
+                // 这批失败，跳过
+            }
+        }
+    }
+    return available;
+}
+/**
+ * 构建命令执行配置
+ * 处理不同 Shell 的语法差异
+ */
+export function buildShellExecConfig(command, shell) {
+    const currentShell = shell || detectShell();
+    const executable = getShellExecutable(currentShell);
+    switch (currentShell) {
+        case 'bash':
+            return {
+                shell: executable,
+                args: ['-c', `set -o pipefail; ${command}`],
+                command: `set -o pipefail; ${command}`,
+            };
+        case 'zsh':
+            return {
+                shell: executable,
+                args: ['-c', `setopt pipefail; ${command}`],
+                command: `setopt pipefail; ${command}`,
+            };
+        case 'fish':
+            // Fish 不需要特殊处理 pipefail
+            return {
+                shell: executable,
+                args: ['-c', command],
+                command,
+            };
+        case 'powershell5':
+        case 'powershell7':
+            // PowerShell: 使用 $ErrorActionPreference 处理错误
+            // -NoProfile 加快启动速度
+            // -Command 执行命令
+            return {
+                shell: executable,
+                args: ['-NoProfile', '-Command', command],
+                command,
+            };
+        case 'cmd':
+            return {
+                shell: executable,
+                args: ['/c', command],
+                command,
+            };
+        default:
+            // 默认使用 sh
+            return {
+                shell: isWindows() ? 'powershell.exe' : '/bin/sh',
+                args: [isWindows() ? '-Command' : '-c', command],
+                command,
+            };
+    }
+}
+/**
+ * 获取默认 Shell（用于交互式执行）
+ */
+export function getDefaultShell() {
+    if (isWindows()) {
+        const shell = detectWindowsShell();
+        return getShellExecutable(shell);
+    }
+    return process.env.SHELL || '/bin/bash';
+}
+// ================== 路径处理 ==================
+/**
+ * 获取 pls 配置目录
+ * 统一使用 ~/.please
+ */
+export function getConfigDir() {
+    return path.join(os.homedir(), '.please');
+}
+/**
+ * 将路径转换为当前平台格式
+ */
+export function normalizePath(p) {
+    return path.normalize(p);
+}
+/**
+ * 获取用于 PowerShell 脚本中的路径
+ * 使用 $env:USERPROFILE 而不是硬编码路径
+ */
+export function getPowerShellConfigDir() {
+    return '$env:USERPROFILE\\.please';
+}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@yivan-lab/pretty-please",
-  "version": "1.4.0",
+  "version": "1.5.1",
   "description": "AI 驱动的命令行工具，将自然语言转换为可执行的 Shell 命令",
   "type": "module",
   "bin": {
@@ -11,6 +11,10 @@
     "dev": "tsx bin/pls.tsx",
     "build": "tsc && node scripts/postbuild.js",
     "start": "node dist/bin/pls.js",
+    "test": "vitest run",
+    "test:watch": "vitest",
+    "test:ui": "vitest --ui",
+    "test:coverage": "vitest run --coverage",
     "link:dev": "mkdir -p ~/.local/bin && ln -sf \"$(pwd)/bin/pls.tsx\" ~/.local/bin/pls-dev && echo '✅ pls-dev 已链接到 ~/.local/bin/pls-dev'",
     "unlink:dev": "rm -f ~/.local/bin/pls-dev && echo '✅ pls-dev 已移除'",
     "prepublishOnly": "pnpm build"
@@ -55,6 +59,7 @@
     "chalk": "^5.6.2",
     "cli-highlight": "^2.1.11",
     "commander": "^14.0.2",
+    "detect-package-manager": "^3.0.2",
     "ink": "^6.5.1",
     "ink-box": "^2.0.0",
     "ink-markdown": "^1.0.4",
@@ -73,8 +78,11 @@
     "@types/hast": "^3.0.4",
     "@types/node": "^25.0.2",
     "@types/react": "^19.2.7",
+    "@vitest/coverage-v8": "^4.0.16",
+    "@vitest/ui": "^4.0.16",
     "react-devtools-core": "^7.0.1",
     "tsx": "^4.21.0",
-    "typescript": "^5.9.3"
+    "typescript": "^5.9.3",
+    "vitest": "^4.0.16"
   }
 }