@yivan-lab/pretty-please 1.3.1 → 1.4.0

package/bin/pls.tsx

@@ -1240,7 +1240,7 @@ program
   .argument('[prompt...]', '自然语言描述你想执行的操作')
   .option('-d, --debug', '显示调试信息（系统信息、完整 prompt 等）')
   .option('-r, --remote [name]', '在远程服务器上执行（不指定则使用默认服务器）')
-  .action((promptArgs, options) => {
+  .action(async (promptArgs, options) => {
     // 智能处理 -r 参数：如果 -r 后面的值不是已注册的服务器名，把它当作 prompt 的一部分
     if (typeof options.remote === 'string' && !getRemote(options.remote)) {
       // "查看当前目录" 不是服务器名，放回 prompt
@@ -1248,12 +1248,32 @@ program
       options.remote = true  // 改为使用默认服务器
     }
 
+    let prompt = ''
+
     if (promptArgs.length === 0) {
-      program.help()
-      return
-    }
+      // 无参数时：尝试自动检测上一条失败的命令
+      const { getLastNonPlsCommand } = await import('../src/shell-hook.js')
+      const lastCmd = getLastNonPlsCommand()
+
+      if (lastCmd && lastCmd.exit !== 0) {
+        // 找到了失败的命令，自动生成 prompt
+        prompt = `上一条命令「${lastCmd.cmd}」执行失败，退出码：${lastCmd.exit}。请生成正确的命令。`
+
+        if (options.debug) {
+          console.log('')
+          console2.muted(`自动检测到失败命令: ${lastCmd.cmd} (退出码: ${lastCmd.exit})`)
+          console2.muted(`生成 prompt: ${prompt}`)
+        }
 
-    let prompt = promptArgs.join(' ')
+        // 继续执行命令生成流程（不 return）
+      } else {
+        // 没有失败的命令，显示帮助
+        program.help()
+        return
+      }
+    } else {
+      prompt = promptArgs.join(' ')
+    }
 
     if (!prompt.trim()) {
       console.log('')

package/dist/bin/pls.js

@@ -1030,18 +1030,37 @@ program
     .argument('[prompt...]', '自然语言描述你想执行的操作')
     .option('-d, --debug', '显示调试信息（系统信息、完整 prompt 等）')
     .option('-r, --remote [name]', '在远程服务器上执行（不指定则使用默认服务器）')
-    .action((promptArgs, options) => {
+    .action(async (promptArgs, options) => {
     // 智能处理 -r 参数：如果 -r 后面的值不是已注册的服务器名，把它当作 prompt 的一部分
     if (typeof options.remote === 'string' && !getRemote(options.remote)) {
         // "查看当前目录" 不是服务器名，放回 prompt
         promptArgs.unshift(options.remote);
         options.remote = true; // 改为使用默认服务器
     }
+    let prompt = '';
     if (promptArgs.length === 0) {
-        program.help();
-        return;
+        // 无参数时：尝试自动检测上一条失败的命令
+        const { getLastNonPlsCommand } = await import('../src/shell-hook.js');
+        const lastCmd = getLastNonPlsCommand();
+        if (lastCmd && lastCmd.exit !== 0) {
+            // 找到了失败的命令，自动生成 prompt
+            prompt = `上一条命令「${lastCmd.cmd}」执行失败，退出码：${lastCmd.exit}。请生成正确的命令。`;
+            if (options.debug) {
+                console.log('');
+                console2.muted(`自动检测到失败命令: ${lastCmd.cmd} (退出码: ${lastCmd.exit})`);
+                console2.muted(`生成 prompt: ${prompt}`);
+            }
+            // 继续执行命令生成流程（不 return）
+        }
+        else {
+            // 没有失败的命令，显示帮助
+            program.help();
+            return;
+        }
+    }
+    else {
+        prompt = promptArgs.join(' ');
     }
-    let prompt = promptArgs.join(' ');
     if (!prompt.trim()) {
         console.log('');
         console2.error('请提供你想执行的操作描述');

package/dist/package.json

@@ -1,6 +1,6 @@
 {
     "name": "@yivan-lab/pretty-please",
-    "version": "1.3.1",
+    "version": "1.4.0",
     "description": "AI 驱动的命令行工具，将自然语言转换为可执行的 Shell 命令",
     "type": "module",
     "bin": {

package/dist/src/mastra-chat.js

@@ -3,7 +3,6 @@ import { getConfig } from './config.js';
 import { CHAT_SYSTEM_PROMPT, buildChatUserContext } from './prompts.js';
 import { formatSystemInfo } from './sysinfo.js';
 import { formatHistoryForAI } from './history.js';
-import { formatShellHistoryForAI, getShellHistory } from './shell-hook.js';
 import { getChatHistory, addChatMessage } from './chat-history.js';
 /**
  * 创建 Mastra Chat Agent（使用静态系统提示词）
@@ -50,8 +49,10 @@ export async function chatWithMastra(prompt, options = {}) {
     // 3. 构建最新消息（动态上下文 + 用户问题）
     const sysinfo = formatSystemInfo();
     const plsHistory = formatHistoryForAI();
-    const shellHistory = formatShellHistoryForAI();
-    const shellHookEnabled = config.shellHook && getShellHistory().length > 0;
+    // 使用统一的历史获取接口（自动降级到系统历史）
+    const { formatShellHistoryForAIWithFallback } = await import('./shell-hook.js');
+    const shellHistory = formatShellHistoryForAIWithFallback();
+    const shellHookEnabled = !!shellHistory; // 如果有 shell 历史就视为启用
     const latestUserContext = buildChatUserContext(prompt, sysinfo, plsHistory, shellHistory, shellHookEnabled);
     messages.push(latestUserContext);
     // 4. 发送给 AI（流式或非流式）

package/dist/src/multi-step.js

@@ -3,8 +3,6 @@ import { createShellAgent } from './mastra-agent.js';
 import { SHELL_COMMAND_SYSTEM_PROMPT, buildUserContextPrompt } from './prompts.js';
 import { formatSystemInfo } from './sysinfo.js';
 import { formatHistoryForAI } from './history.js';
-import { formatShellHistoryForAI, getShellHistory } from './shell-hook.js';
-import { getConfig } from './config.js';
 import { formatRemoteHistoryForAI, formatRemoteShellHistoryForAI } from './remote-history.js';
 import { formatRemoteSysInfoForAI } from './remote.js';
 /**
@@ -47,10 +45,11 @@ export async function generateMultiStepCommand(userPrompt, previousSteps = [], o
     else {
         // 本地执行：格式化本地系统信息和历史
         sysinfoStr = formatSystemInfo();
-        const config = getConfig();
         const plsHistory = formatHistoryForAI();
-        const shellHistory = formatShellHistoryForAI();
-        historyStr = (config.shellHook && getShellHistory().length > 0) ? shellHistory : plsHistory;
+        // 使用统一的历史获取接口（自动降级到系统历史）
+        const { formatShellHistoryForAIWithFallback } = await import('./shell-hook.js');
+        const shellHistory = formatShellHistoryForAIWithFallback();
+        historyStr = shellHistory || plsHistory; // 优先使用 shell 历史，降级到 pls 历史
     }
     // 构建包含所有动态数据的 User Prompt（XML 格式）
     const userContextPrompt = buildUserContextPrompt(userPrompt, sysinfoStr, historyStr, previousSteps);

package/dist/src/shell-hook.d.ts

@@ -59,6 +59,25 @@ export declare function reinstallHookForLimitChange(oldLimit: number, newLimit:
  * 清空 shell 历史
  */
 export declare function clearShellHistory(): void;
+/**
+ * 获取 shell 历史（统一接口）
+ * 优先使用 shell hook，降级到系统历史文件
+ *
+ * 这是推荐的历史获取方式，会自动选择最佳来源：
+ * 1. 优先：shell hook（有退出码，最准确）
+ * 2. 降级：系统历史文件（无退出码，兼容未安装 hook 的情况）
+ */
+export declare function getShellHistoryWithFallback(): ShellHistoryItem[];
+/**
+ * 获取最近一条非 pls 的命令（统一接口）
+ * 自动选择最佳历史来源
+ */
+export declare function getLastNonPlsCommand(): ShellHistoryItem | null;
+/**
+ * 格式化 shell 历史供 AI 使用（统一接口）
+ * 自动选择最佳历史来源
+ */
+export declare function formatShellHistoryForAIWithFallback(): string;
 /**
  * 检测远程服务器的 shell 类型
  */

package/dist/src/shell-hook.js

@@ -479,6 +479,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/system-history.d.ts

@@ -0,0 +1,13 @@
+import type { ShellHistoryItem } from './shell-hook.js';
+/**
+ * 直接读取系统 shell 历史文件（类似 thefuck）
+ * 用于没有安装 shell hook 的情况
+ *
+ * 限制：系统历史文件不记录退出码，所以 exit 字段都是 0
+ */
+export declare function getSystemShellHistory(): ShellHistoryItem[];
+/**
+ * 从系统历史中获取最近一条命令
+ * 排除 pls 命令本身
+ */
+export declare function getLastCommandFromSystem(): ShellHistoryItem | null;

package/dist/src/system-history.js

@@ -0,0 +1,105 @@
+import fs from 'fs';
+import path from 'path';
+import os from 'os';
+import { getConfig } from './config.js';
+/**
+ * 直接读取系统 shell 历史文件（类似 thefuck）
+ * 用于没有安装 shell hook 的情况
+ *
+ * 限制：系统历史文件不记录退出码，所以 exit 字段都是 0
+ */
+export function getSystemShellHistory() {
+    const shell = process.env.SHELL || '';
+    const home = os.homedir();
+    let historyFile;
+    let parser;
+    if (shell.includes('zsh')) {
+        // zsh 历史文件
+        historyFile = process.env.HISTFILE || path.join(home, '.zsh_history');
+        parser = parseZshHistoryLine;
+    }
+    else if (shell.includes('bash')) {
+        // bash 历史文件
+        historyFile = process.env.HISTFILE || path.join(home, '.bash_history');
+        parser = parseBashHistoryLine;
+    }
+    else {
+        // 不支持的 shell
+        return [];
+    }
+    if (!fs.existsSync(historyFile)) {
+        return [];
+    }
+    try {
+        const content = fs.readFileSync(historyFile, 'utf-8');
+        const lines = content.trim().split('\n');
+        const limit = getConfig().shellHistoryLimit || 10;
+        // 只取最后 N 条
+        const recentLines = lines.slice(-limit);
+        return recentLines
+            .map(line => parser(line))
+            .filter((item) => item !== null);
+    }
+    catch {
+        return [];
+    }
+}
+/**
+ * 解析 zsh 历史行
+ * 格式: ": 1234567890:0;ls -la"
+ * 或者: "ls -la" (简单格式)
+ */
+function parseZshHistoryLine(line) {
+    // 扩展格式: ": timestamp:duration;command"
+    const extendedMatch = line.match(/^:\s*(\d+):\d+;(.+)$/);
+    if (extendedMatch) {
+        const timestamp = parseInt(extendedMatch[1]);
+        const cmd = extendedMatch[2].trim();
+        return {
+            cmd,
+            exit: 0, // 系统历史文件不记录退出码
+            time: new Date(timestamp * 1000).toISOString(),
+        };
+    }
+    // 简单格式
+    const cmd = line.trim();
+    if (cmd) {
+        return {
+            cmd,
+            exit: 0,
+            time: new Date().toISOString(),
+        };
+    }
+    return null;
+}
+/**
+ * 解析 bash 历史行
+ * 格式: "ls -la"
+ * bash 历史文件默认不记录时间戳
+ */
+function parseBashHistoryLine(line) {
+    const cmd = line.trim();
+    if (cmd) {
+        return {
+            cmd,
+            exit: 0, // 系统历史文件不记录退出码
+            time: new Date().toISOString(),
+        };
+    }
+    return null;
+}
+/**
+ * 从系统历史中获取最近一条命令
+ * 排除 pls 命令本身
+ */
+export function getLastCommandFromSystem() {
+    const history = getSystemShellHistory();
+    // 从后往前找第一条非 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;
+}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@yivan-lab/pretty-please",
-  "version": "1.3.1",
+  "version": "1.4.0",
   "description": "AI 驱动的命令行工具，将自然语言转换为可执行的 Shell 命令",
   "type": "module",
   "bin": {

package/src/mastra-chat.ts

@@ -73,8 +73,10 @@ export async function chatWithMastra(
   // 3. 构建最新消息（动态上下文 + 用户问题）
   const sysinfo = formatSystemInfo()
   const plsHistory = formatHistoryForAI()
-  const shellHistory = formatShellHistoryForAI()
-  const shellHookEnabled = config.shellHook && getShellHistory().length > 0
+  // 使用统一的历史获取接口（自动降级到系统历史）
+  const { formatShellHistoryForAIWithFallback } = await import('./shell-hook.js')
+  const shellHistory = formatShellHistoryForAIWithFallback()
+  const shellHookEnabled = !!shellHistory  // 如果有 shell 历史就视为启用
 
   const latestUserContext = buildChatUserContext(
     prompt,

package/src/multi-step.ts

@@ -75,10 +75,11 @@ export async function generateMultiStepCommand(
   } else {
     // 本地执行：格式化本地系统信息和历史
     sysinfoStr = formatSystemInfo()
-    const config = getConfig()
     const plsHistory = formatHistoryForAI()
-    const shellHistory = formatShellHistoryForAI()
-    historyStr = (config.shellHook && getShellHistory().length > 0) ? shellHistory : plsHistory
+    // 使用统一的历史获取接口（自动降级到系统历史）
+    const { formatShellHistoryForAIWithFallback } = await import('./shell-hook.js')
+    const shellHistory = formatShellHistoryForAIWithFallback()
+    historyStr = shellHistory || plsHistory  // 优先使用 shell 历史，降级到 pls 历史
   }
 
   // 构建包含所有动态数据的 User Prompt（XML 格式）

package/src/shell-hook.ts

@@ -573,6 +573,82 @@ export function clearShellHistory(): void {
   console.log('')
 }
 
+// ================== 统一历史获取 ==================
+
+/**
+ * 获取 shell 历史（统一接口）
+ * 优先使用 shell hook，降级到系统历史文件
+ *
+ * 这是推荐的历史获取方式，会自动选择最佳来源：
+ * 1. 优先：shell hook（有退出码，最准确）
+ * 2. 降级：系统历史文件（无退出码，兼容未安装 hook 的情况）
+ */
+export function getShellHistoryWithFallback(): ShellHistoryItem[] {
+  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(): ShellHistoryItem | null {
+  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(): string {
+  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 ==================
 
 /**

package/src/system-history.ts

@@ -0,0 +1,117 @@
+import fs from 'fs'
+import path from 'path'
+import os from 'os'
+import { getConfig } from './config.js'
+import type { ShellHistoryItem } from './shell-hook.js'
+
+/**
+ * 直接读取系统 shell 历史文件（类似 thefuck）
+ * 用于没有安装 shell hook 的情况
+ *
+ * 限制：系统历史文件不记录退出码，所以 exit 字段都是 0
+ */
+export function getSystemShellHistory(): ShellHistoryItem[] {
+  const shell = process.env.SHELL || ''
+  const home = os.homedir()
+
+  let historyFile: string
+  let parser: (line: string) => ShellHistoryItem | null
+
+  if (shell.includes('zsh')) {
+    // zsh 历史文件
+    historyFile = process.env.HISTFILE || path.join(home, '.zsh_history')
+    parser = parseZshHistoryLine
+  } else if (shell.includes('bash')) {
+    // bash 历史文件
+    historyFile = process.env.HISTFILE || path.join(home, '.bash_history')
+    parser = parseBashHistoryLine
+  } else {
+    // 不支持的 shell
+    return []
+  }
+
+  if (!fs.existsSync(historyFile)) {
+    return []
+  }
+
+  try {
+    const content = fs.readFileSync(historyFile, 'utf-8')
+    const lines = content.trim().split('\n')
+    const limit = getConfig().shellHistoryLimit || 10
+
+    // 只取最后 N 条
+    const recentLines = lines.slice(-limit)
+
+    return recentLines
+      .map(line => parser(line))
+      .filter((item): item is ShellHistoryItem => item !== null)
+  } catch {
+    return []
+  }
+}
+
+/**
+ * 解析 zsh 历史行
+ * 格式: ": 1234567890:0;ls -la"
+ * 或者: "ls -la" (简单格式)
+ */
+function parseZshHistoryLine(line: string): ShellHistoryItem | null {
+  // 扩展格式: ": timestamp:duration;command"
+  const extendedMatch = line.match(/^:\s*(\d+):\d+;(.+)$/)
+  if (extendedMatch) {
+    const timestamp = parseInt(extendedMatch[1])
+    const cmd = extendedMatch[2].trim()
+    return {
+      cmd,
+      exit: 0,  // 系统历史文件不记录退出码
+      time: new Date(timestamp * 1000).toISOString(),
+    }
+  }
+
+  // 简单格式
+  const cmd = line.trim()
+  if (cmd) {
+    return {
+      cmd,
+      exit: 0,
+      time: new Date().toISOString(),
+    }
+  }
+
+  return null
+}
+
+/**
+ * 解析 bash 历史行
+ * 格式: "ls -la"
+ * bash 历史文件默认不记录时间戳
+ */
+function parseBashHistoryLine(line: string): ShellHistoryItem | null {
+  const cmd = line.trim()
+  if (cmd) {
+    return {
+      cmd,
+      exit: 0,  // 系统历史文件不记录退出码
+      time: new Date().toISOString(),
+    }
+  }
+  return null
+}
+
+/**
+ * 从系统历史中获取最近一条命令
+ * 排除 pls 命令本身
+ */
+export function getLastCommandFromSystem(): ShellHistoryItem | null {
+  const history = getSystemShellHistory()
+
+  // 从后往前找第一条非 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
+}