@yivan-lab/pretty-please 1.3.1 → 1.5.0

package/dist/src/shell-hook.js

@@ -5,6 +5,7 @@ import chalk from 'chalk';
 import { CONFIG_DIR, getConfig, setConfigValue } from './config.js';
 import { getHistory } from './history.js';
 import { getCurrentTheme } from './ui/theme.js';
+import { detectShell as platformDetectShell, } from './utils/platform.js';
 // 获取主题颜色
 function getColors() {
     const theme = getCurrentTheme();
@@ -20,19 +21,36 @@ const SHELL_HISTORY_FILE = path.join(CONFIG_DIR, 'shell_history.jsonl');
 // Hook 标记，用于识别我们添加的内容
 const HOOK_START_MARKER = '# >>> pretty-please shell hook >>>';
 const HOOK_END_MARKER = '# <<< pretty-please shell hook <<<';
+/**
+ * 将 platform 模块的 ShellType 转换为本地 ShellType
+ */
+function toLocalShellType(platformShell) {
+    switch (platformShell) {
+        case 'zsh':
+            return 'zsh';
+        case 'bash':
+            return 'bash';
+        case 'powershell5':
+        case 'powershell7':
+            return 'powershell';
+        case 'cmd':
+        case 'fish':
+        case 'unknown':
+        default:
+            return 'unknown';
+    }
+}
 /**
  * 检测当前 shell 类型
+ * 使用 platform 模块进行跨平台检测
  */
 export function detectShell() {
-    const shell = process.env.SHELL || '';
-    if (shell.includes('zsh'))
-        return 'zsh';
-    if (shell.includes('bash'))
-        return 'bash';
-    // Windows PowerShell
-    if (process.platform === 'win32')
-        return 'powershell';
-    return 'unknown';
+    const platformShell = platformDetectShell();
+    // CMD 不支持 Hook，提示用户
+    if (platformShell === 'cmd') {
+        return 'unknown';
+    }
+    return toLocalShellType(platformShell);
 }
 /**
  * 获取 shell 配置文件路径
@@ -55,6 +73,35 @@ export function getShellConfigPath(shellType) {
             return null;
     }
 }
+/**
+ * 生成命令统计的公共 shell 函数
+ * 用于 zsh 和 bash，避免代码重复
+ */
+function generateStatFunction() {
+    return `
+# 统计命令频率（公共函数）
+__pls_record_stat() {
+  local cmd_name="$1"
+  local stats_file="${CONFIG_DIR}/command_stats.txt"
+
+  # 确保文件存在
+  touch "$stats_file"
+
+  # 更新统计（纯 shell 实现，不依赖 jq）
+  if grep -q "^$cmd_name=" "$stats_file" 2>/dev/null; then
+    # 命令已存在，次数 +1
+    local count=$(grep "^$cmd_name=" "$stats_file" | cut -d= -f2)
+    count=$((count + 1))
+    # macOS 和 Linux 的 sed -i 不同，使用临时文件
+    sed "s/^$cmd_name=.*/$cmd_name=$count/" "$stats_file" > "$stats_file.tmp" 2>/dev/null
+    mv "$stats_file.tmp" "$stats_file" 2>/dev/null
+  else
+    # 新命令，追加
+    echo "$cmd_name=1" >> "$stats_file"
+  fi
+}
+`;
+}
 /**
  * 生成 zsh hook 脚本
  */
@@ -64,6 +111,7 @@ function generateZshHook() {
     return `
 ${HOOK_START_MARKER}
 # 记录命令到 pretty-please 历史
+${generateStatFunction()}
 __pls_preexec() {
   __PLS_LAST_CMD="$1"
   __PLS_CMD_START=$(date +%s)
@@ -79,6 +127,11 @@ __pls_precmd() {
     echo "{\\"cmd\\":\\"$escaped_cmd\\",\\"exit\\":$exit_code,\\"time\\":\\"$timestamp\\"}" >> "${CONFIG_DIR}/shell_history.jsonl"
     # 保持文件不超过 ${limit} 行（从配置读取）
     tail -n ${limit} "${CONFIG_DIR}/shell_history.jsonl" > "${CONFIG_DIR}/shell_history.jsonl.tmp" && mv "${CONFIG_DIR}/shell_history.jsonl.tmp" "${CONFIG_DIR}/shell_history.jsonl"
+
+    # 统计命令频率
+    local cmd_name=$(echo "$__PLS_LAST_CMD" | awk '{print $1}')
+    __pls_record_stat "$cmd_name"
+
     unset __PLS_LAST_CMD
   fi
 }
@@ -98,6 +151,7 @@ function generateBashHook() {
     return `
 ${HOOK_START_MARKER}
 # 记录命令到 pretty-please 历史
+${generateStatFunction()}
 __pls_prompt_command() {
   local exit_code=$?
   local last_cmd=$(history 1 | sed 's/^ *[0-9]* *//')
@@ -107,6 +161,10 @@ __pls_prompt_command() {
     local escaped_cmd=$(echo "$last_cmd" | sed 's/\\\\/\\\\\\\\/g; s/"/\\\\"/g')
     echo "{\\"cmd\\":\\"$escaped_cmd\\",\\"exit\\":$exit_code,\\"time\\":\\"$timestamp\\"}" >> "${CONFIG_DIR}/shell_history.jsonl"
     tail -n ${limit} "${CONFIG_DIR}/shell_history.jsonl" > "${CONFIG_DIR}/shell_history.jsonl.tmp" && mv "${CONFIG_DIR}/shell_history.jsonl.tmp" "${CONFIG_DIR}/shell_history.jsonl"
+
+    # 统计命令频率
+    local cmd_name=$(echo "$last_cmd" | awk '{print $1}')
+    __pls_record_stat "$cmd_name"
   fi
 }
 
@@ -118,15 +176,26 @@ ${HOOK_END_MARKER}
 }
 /**
  * 生成 PowerShell hook 脚本
+ * 使用 PowerShell 原生路径处理，避免跨平台路径问题
  */
 function generatePowerShellHook() {
     const config = getConfig();
     const limit = config.shellHistoryLimit || 10; // 从配置读取
+    // 使用 PowerShell 原生路径变量，而不是嵌入 Node.js 路径
     return `
 ${HOOK_START_MARKER}
 # 记录命令到 pretty-please 历史
+# 使用 PowerShell 原生路径
+$Global:__PlsDir = Join-Path $env:USERPROFILE ".please"
+$Global:__PlsHistoryFile = Join-Path $Global:__PlsDir "shell_history.jsonl"
+$Global:__PlsStatsFile = Join-Path $Global:__PlsDir "command_stats.txt"
 $Global:__PlsLastCmd = ""
 
+# 确保目录存在
+if (-not (Test-Path $Global:__PlsDir)) {
+    New-Item -Path $Global:__PlsDir -ItemType Directory -Force | Out-Null
+}
+
 function __Pls_RecordCommand {
     $lastCmd = (Get-History -Count 1).CommandLine
     if ($lastCmd -and $lastCmd -ne $Global:__PlsLastCmd) {
@@ -136,10 +205,38 @@ function __Pls_RecordCommand {
         $timestamp = (Get-Date).ToUniversalTime().ToString("yyyy-MM-ddTHH:mm:ssZ")
         $escapedCmd = $lastCmd -replace '\\\\', '\\\\\\\\' -replace '"', '\\\\"'
         $json = "{\`"cmd\`":\`"$escapedCmd\`",\`"exit\`":$exitCode,\`"time\`":\`"$timestamp\`"}"
-        Add-Content -Path "${CONFIG_DIR}/shell_history.jsonl" -Value $json
+        Add-Content -Path $Global:__PlsHistoryFile -Value $json
         # 保持文件不超过 ${limit} 行（从配置读取）
-        $content = Get-Content "${CONFIG_DIR}/shell_history.jsonl" -Tail ${limit}
-        $content | Set-Content "${CONFIG_DIR}/shell_history.jsonl"
+        $content = Get-Content $Global:__PlsHistoryFile -Tail ${limit} -ErrorAction SilentlyContinue
+        if ($content) {
+            $content | Set-Content $Global:__PlsHistoryFile
+        }
+
+        # 统计命令频率
+        $cmdName = $lastCmd -split ' ' | Select-Object -First 1
+
+        # 确保统计文件存在
+        if (-not (Test-Path $Global:__PlsStatsFile)) {
+            New-Item -Path $Global:__PlsStatsFile -ItemType File -Force | Out-Null
+        }
+
+        # 更新统计
+        $stats = Get-Content $Global:__PlsStatsFile -ErrorAction SilentlyContinue
+        $found = $false
+        $newStats = @()
+        foreach ($line in $stats) {
+            if ($line -match "^$cmdName=(\\d+)$") {
+                $count = [int]$matches[1] + 1
+                $newStats += "$cmdName=$count"
+                $found = $true
+            } else {
+                $newStats += $line
+            }
+        }
+        if (-not $found) {
+            $newStats += "$cmdName=1"
+        }
+        $newStats | Set-Content $Global:__PlsStatsFile
     }
 }
 
@@ -177,6 +274,18 @@ export async function installShellHook() {
     const colors = getColors();
     if (!configPath) {
         console.log(chalk.hex(colors.error)(`❌ 不支持的 shell 类型: ${shellType}`));
+        // CMD 特殊提示
+        if (shellType === 'unknown') {
+            const platformShell = platformDetectShell();
+            if (platformShell === 'cmd') {
+                console.log('');
+                console.log(chalk.hex(colors.warning)('⚠️  CMD 不支持 Shell Hook 功能'));
+                console.log(chalk.hex(colors.secondary)('建议使用 PowerShell 获得完整体验：'));
+                console.log(chalk.hex(colors.secondary)('  1. 按 Win 键搜索 "PowerShell"'));
+                console.log(chalk.hex(colors.secondary)('  2. 在 PowerShell 中运行 pls hook install'));
+                console.log('');
+            }
+        }
         return false;
     }
     const hookScript = generateHookScript(shellType);
@@ -434,39 +543,60 @@ export function displayShellHistory() {
     console.log('');
 }
 /**
- * 当 shellHistoryLimit 变化时，自动重装 Hook
- * 返回是否成功重装
+ * 重新安装 Shell Hook（通用函数）
+ * 用于版本升级、配置变更等场景
+ *
+ * @param options.silent 是否静默模式（不输出日志）
+ * @param options.reason 重装原因（用于日志显示）
+ * @returns 是否成功重装
  */
-export async function reinstallHookForLimitChange(oldLimit, newLimit) {
+export async function reinstallShellHook(options) {
     const config = getConfig();
     // 只有在 hook 已启用时才重装
     if (!config.shellHook) {
         return false;
     }
-    // 值没有变化，不需要重装
-    if (oldLimit === newLimit) {
-        return false;
-    }
     const colors = getColors();
-    console.log('');
-    console.log(chalk.hex(colors.primary)(`检测到 shellHistoryLimit 变化 (${oldLimit} → ${newLimit})`));
-    console.log(chalk.hex(colors.primary)('正在更新 Shell Hook...'));
+    const { silent = false, reason } = options || {};
+    if (!silent) {
+        console.log('');
+        if (reason) {
+            console.log(chalk.hex(colors.primary)(reason));
+        }
+        console.log(chalk.hex(colors.primary)('正在更新 Shell Hook...'));
+    }
+    // 卸载旧版本，安装新版本
     uninstallShellHook();
     await installShellHook();
-    console.log('');
-    console.log(chalk.hex(colors.warning)('⚠️  请重启终端或运行以下命令使新配置生效:'));
-    const shellType = detectShell();
-    let configFile = '~/.zshrc';
-    if (shellType === 'bash') {
-        configFile = process.platform === 'darwin' ? '~/.bash_profile' : '~/.bashrc';
-    }
-    else if (shellType === 'powershell') {
-        configFile = '~/Documents/PowerShell/Microsoft.PowerShell_profile.ps1';
+    if (!silent) {
+        console.log('');
+        console.log(chalk.hex(colors.warning)('⚠️  请重启终端或运行以下命令使新配置生效:'));
+        const shellType = detectShell();
+        let configFile = '~/.zshrc';
+        if (shellType === 'bash') {
+            configFile = process.platform === 'darwin' ? '~/.bash_profile' : '~/.bashrc';
+        }
+        else if (shellType === 'powershell') {
+            configFile = '~/Documents/PowerShell/Microsoft.PowerShell_profile.ps1';
+        }
+        console.log(chalk.gray(`  source ${configFile}`));
+        console.log('');
     }
-    console.log(chalk.gray(`  source ${configFile}`));
-    console.log('');
     return true;
 }
+/**
+ * 当 shellHistoryLimit 变化时，自动重装 Hook
+ * 返回是否成功重装
+ */
+export async function reinstallHookForLimitChange(oldLimit, newLimit) {
+    // 值没有变化，不需要重装
+    if (oldLimit === newLimit) {
+        return false;
+    }
+    return reinstallShellHook({
+        reason: `检测到 shellHistoryLimit 变化 (${oldLimit} → ${newLimit})`,
+    });
+}
 /**
  * 清空 shell 历史
  */
@@ -479,6 +609,69 @@ export function clearShellHistory() {
     console.log(chalk.hex(colors.success)('✓ Shell 历史已清空'));
     console.log('');
 }
+// ================== 统一历史获取 ==================
+/**
+ * 获取 shell 历史（统一接口）
+ * 优先使用 shell hook，降级到系统历史文件
+ *
+ * 这是推荐的历史获取方式，会自动选择最佳来源：
+ * 1. 优先：shell hook（有退出码，最准确）
+ * 2. 降级：系统历史文件（无退出码，兼容未安装 hook 的情况）
+ */
+export function getShellHistoryWithFallback() {
+    const config = getConfig();
+    // 优先使用 shell hook
+    if (config.shellHook) {
+        const history = getShellHistory();
+        if (history.length > 0) {
+            return history;
+        }
+    }
+    // 降级到系统历史文件
+    const { getSystemShellHistory } = require('./system-history.js');
+    return getSystemShellHistory();
+}
+/**
+ * 获取最近一条非 pls 的命令（统一接口）
+ * 自动选择最佳历史来源
+ */
+export function getLastNonPlsCommand() {
+    const history = getShellHistoryWithFallback();
+    // 从后往前找第一条非 pls 命令
+    for (let i = history.length - 1; i >= 0; i--) {
+        const item = history[i];
+        if (!item.cmd.startsWith('pls') && !item.cmd.startsWith('please')) {
+            return item;
+        }
+    }
+    return null;
+}
+/**
+ * 格式化 shell 历史供 AI 使用（统一接口）
+ * 自动选择最佳历史来源
+ */
+export function formatShellHistoryForAIWithFallback() {
+    const config = getConfig();
+    // 如果启用了 shell hook 且有记录，使用 hook 历史（包含详细信息）
+    if (config.shellHook) {
+        const hookHistory = getShellHistory();
+        if (hookHistory.length > 0) {
+            return formatShellHistoryForAI();
+        }
+    }
+    // 降级到系统历史
+    const { getSystemShellHistory } = require('./system-history.js');
+    const history = getSystemShellHistory();
+    if (history.length === 0) {
+        return '';
+    }
+    // 格式化系统历史（简单格式，无详细信息）
+    const lines = history.map((item, index) => {
+        const status = item.exit === 0 ? '✓' : `✗ 退出码:${item.exit}`;
+        return `${index + 1}. ${item.cmd} ${status}`;
+    });
+    return `【用户终端最近执行的命令（来自系统历史）】\n${lines.join('\n')}`;
+}
 // ================== 远程 Shell Hook ==================
 /**
  * 生成远程 zsh hook 脚本

package/dist/src/sysinfo.d.ts

@@ -1,19 +1,48 @@
+import { type ProjectContext } from './project-context.js';
 /**
- * 系统信息
+ * 静态系统信息（缓存 7 天）
  */
-export interface SystemInfo {
-    os: NodeJS.Platform;
+export interface StaticSystemInfo {
+    os: string;
     arch: string;
     shell: string;
-    packageManager: string;
-    cwd: string;
     user: string;
+    systemPackageManager: string;
+    availableCommands: string[];
+}
+/**
+ * 动态系统信息（每次实时获取）
+ */
+export interface DynamicSystemInfo {
+    cwd: string;
+    project: ProjectContext | null;
+}
+/**
+ * 完整系统信息
+ */
+export interface SystemInfo extends StaticSystemInfo, DynamicSystemInfo {
 }
 /**
- * 收集系统信息
+ * 获取静态系统信息（带缓存）
+ */
+export declare function getStaticSystemInfo(): StaticSystemInfo;
+/**
+ * 获取动态系统信息（每次实时）
+ */
+export declare function getDynamicSystemInfo(): Promise<DynamicSystemInfo>;
+/**
+ * 获取完整系统信息（主接口）
+ */
+export declare function getSystemInfo(): Promise<SystemInfo>;
+/**
+ * 格式化系统信息为字符串（供 AI 使用）
+ */
+export declare function formatSystemInfo(info: SystemInfo): string;
+/**
+ * 显示系统信息（CLI 美化版）
  */
-export declare function collectSystemInfo(): SystemInfo;
+export declare function displaySystemInfo(info: SystemInfo): void;
 /**
- * 将系统信息格式化为字符串（供 AI 使用）
+ * 强制刷新缓存
  */
-export declare function formatSystemInfo(): string;
+export declare function refreshSystemCache(): void;