@yivan-lab/pretty-please 1.3.1 → 1.5.0

package/src/config.ts

@@ -12,7 +12,8 @@ function getColors() {
     primary: theme.primary,
     secondary: theme.secondary,
     success: theme.success,
-    error: theme.error
+    error: theme.error,
+    warning: theme.warning,
   }
 }
 
@@ -82,11 +83,13 @@ export interface Config {
   chatHistoryLimit: number
   commandHistoryLimit: number
   shellHistoryLimit: number
+  userPreferencesTopK: number          // 用户偏好显示的命令数量（默认 20）
   editMode: EditMode
   theme: ThemeName
   aliases: Record<string, AliasConfig>
   remotes: Record<string, RemoteConfig>  // 远程服务器配置
   defaultRemote?: string                  // 默认远程服务器名称
+  systemCacheExpireDays?: number          // 系统信息缓存过期天数（默认 7 天）
 }
 
 /**
@@ -101,11 +104,13 @@ const DEFAULT_CONFIG: Config = {
   chatHistoryLimit: 5,
   commandHistoryLimit: 5,
   shellHistoryLimit: 10,
+  userPreferencesTopK: 20,  // 默认显示 Top 20
   editMode: 'manual',
   theme: 'dark',
   aliases: {},
   remotes: {},
   defaultRemote: '',
+  systemCacheExpireDays: 7,
 }
 
 /**
@@ -169,7 +174,7 @@ export function setConfigValue(key: string, value: string | boolean | number): C
   // 处理特殊类型
   if (key === 'shellHook') {
     config.shellHook = value === 'true' || value === true
-  } else if (key === 'chatHistoryLimit' || key === 'commandHistoryLimit' || key === 'shellHistoryLimit') {
+  } else if (key === 'chatHistoryLimit' || key === 'commandHistoryLimit' || key === 'shellHistoryLimit' || key === 'userPreferencesTopK' || key === 'systemCacheExpireDays') {
     const num = typeof value === 'number' ? value : parseInt(String(value), 10)
     if (isNaN(num) || num < 1) {
       throw new Error(`${key} 必须是大于 0 的整数`)
@@ -246,6 +251,10 @@ export function displayConfig(): void {
   console.log(`  ${chalk.hex(colors.primary)('chatHistoryLimit')}:    ${config.chatHistoryLimit} 轮`)
   console.log(`  ${chalk.hex(colors.primary)('commandHistoryLimit')}: ${config.commandHistoryLimit} 条`)
   console.log(`  ${chalk.hex(colors.primary)('shellHistoryLimit')}:   ${config.shellHistoryLimit} 条`)
+  console.log(`  ${chalk.hex(colors.primary)('userPreferencesTopK')}: ${config.userPreferencesTopK} 个`)
+  if (config.systemCacheExpireDays !== undefined) {
+    console.log(`  ${chalk.hex(colors.primary)('systemCacheExpireDays')}: ${config.systemCacheExpireDays} 天`)
+  }
 
   // 动态显示主题信息
   const themeMetadata = getAllThemeMetadata().find((m) => m.name === config.theme)
@@ -354,6 +363,8 @@ export async function runConfigWizard(): Promise<void> {
       const num = parseInt(chatHistoryLimit.trim(), 10)
       if (!isNaN(num) && num > 0) {
         config.chatHistoryLimit = num
+      } else {
+        console.log(chalk.hex(colors.warning)('  ⚠️  输入无效，保持原值'))
       }
     }
 
@@ -364,6 +375,8 @@ export async function runConfigWizard(): Promise<void> {
       const num = parseInt(commandHistoryLimit.trim(), 10)
       if (!isNaN(num) && num > 0) {
         config.commandHistoryLimit = num
+      } else {
+        console.log(chalk.hex(colors.warning)('  ⚠️  输入无效，保持原值'))
       }
     }
 
@@ -375,6 +388,20 @@ export async function runConfigWizard(): Promise<void> {
       const num = parseInt(shellHistoryLimit.trim(), 10)
       if (!isNaN(num) && num > 0) {
         config.shellHistoryLimit = num
+      } else {
+        console.log(chalk.hex(colors.warning)('  ⚠️  输入无效，保持原值'))
+      }
+    }
+
+    // 10. User Preferences Top K
+    const userPrefsPrompt = `${chalk.hex(colors.primary)('用户偏好显示命令数')}\n${chalk.gray('默认:')} ${chalk.hex(colors.secondary)(config.userPreferencesTopK)} ${chalk.gray('→')} `
+    const userPrefsTopK = await question(rl, userPrefsPrompt)
+    if (userPrefsTopK.trim()) {
+      const num = parseInt(userPrefsTopK.trim(), 10)
+      if (!isNaN(num) && num > 0) {
+        config.userPreferencesTopK = num
+      } else {
+        console.log(chalk.hex(colors.warning)('  ⚠️  输入无效，保持原值'))
       }
     }
 

package/src/mastra-chat.ts

@@ -1,7 +1,7 @@
 import { Agent } from '@mastra/core'
 import { getConfig } from './config.js'
 import { CHAT_SYSTEM_PROMPT, buildChatUserContext } from './prompts.js'
-import { formatSystemInfo } from './sysinfo.js'
+import { formatSystemInfo, getSystemInfo } from './sysinfo.js'
 import { formatHistoryForAI } from './history.js'
 import { formatShellHistoryForAI, getShellHistory } from './shell-hook.js'
 import { getChatHistory, addChatMessage } from './chat-history.js'
@@ -71,17 +71,24 @@ export async function chatWithMastra(
   }
 
   // 3. 构建最新消息（动态上下文 + 用户问题）
-  const sysinfo = formatSystemInfo()
+  const sysinfo = formatSystemInfo(await getSystemInfo())
   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 { formatUserPreferences } = await import('./user-preferences.js')
+  const userPreferencesStr = formatUserPreferences()
 
   const latestUserContext = buildChatUserContext(
     prompt,
     sysinfo,
     plsHistory,
     shellHistory,
-    shellHookEnabled
+    shellHookEnabled,
+    userPreferencesStr
   )
 
   messages.push(latestUserContext)

package/src/multi-step.ts

@@ -1,7 +1,7 @@
 import { z } from 'zod'
 import { createShellAgent } from './mastra-agent.js'
 import { SHELL_COMMAND_SYSTEM_PROMPT, buildUserContextPrompt } from './prompts.js'
-import { formatSystemInfo } from './sysinfo.js'
+import { formatSystemInfo, getSystemInfo } from './sysinfo.js'
 import { formatHistoryForAI } from './history.js'
 import { formatShellHistoryForAI, getShellHistory } from './shell-hook.js'
 import { getConfig, type RemoteSysInfo } from './config.js'
@@ -74,18 +74,24 @@ export async function generateMultiStepCommand(
     historyStr = options.remoteContext.shellHistory.length > 0 ? shellHistory : plsHistory
   } else {
     // 本地执行：格式化本地系统信息和历史
-    sysinfoStr = formatSystemInfo()
-    const config = getConfig()
+    sysinfoStr = formatSystemInfo(await getSystemInfo())
     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 历史
   }
 
+  // 获取用户偏好
+  const { formatUserPreferences } = await import('./user-preferences.js')
+  const userPreferencesStr = formatUserPreferences()
+
   // 构建包含所有动态数据的 User Prompt（XML 格式）
   const userContextPrompt = buildUserContextPrompt(
     userPrompt,
     sysinfoStr,
     historyStr,
+    userPreferencesStr,
     previousSteps
   )
 

package/src/project-context.ts

@@ -0,0 +1,191 @@
+import fs from 'fs'
+import path from 'path'
+import { detect as detectPM } from 'detect-package-manager'
+import { execSync } from 'child_process'
+
+/**
+ * 项目上下文信息
+ */
+export interface ProjectContext {
+  types: string[]           // 项目类型 ['nodejs', 'docker', 'git']
+  packageManager?: string   // 包管理器 'pnpm' | 'yarn' | 'bun' | 'npm' | 'uv' | 'poetry' | 'cargo'
+  git?: {
+    branch: string          // Git 分支名
+    status: 'clean' | 'dirty' | 'unknown'
+  }
+  scripts?: string[]        // 可用脚本（如 package.json scripts）
+}
+
+/**
+ * 检测项目上下文（优化版：< 30ms）
+ * 若未识别到任何特征，返回 null
+ */
+export async function detectProjectContext(cwd: string): Promise<ProjectContext | null> {
+  const types: string[] = []
+  let packageManager: string | undefined
+  let git: ProjectContext['git'] | undefined
+  let scripts: string[] | undefined
+
+  try {
+    // 1. 快速读取目录文件列表（同步，< 5ms）
+    const files = fs.readdirSync(cwd)
+
+    // 2. 检测项目类型（纯文件检查，< 10ms）
+    if (files.includes('package.json')) {
+      types.push('nodejs')
+
+      // 检测 Node.js 包管理器
+      try {
+        packageManager = await detectPM({ cwd })
+      } catch {
+        // 降级：根据 lock 文件推断
+        if (files.includes('pnpm-lock.yaml')) packageManager = 'pnpm'
+        else if (files.includes('yarn.lock')) packageManager = 'yarn'
+        else if (files.includes('bun.lockb')) packageManager = 'bun'
+        else packageManager = 'npm'
+      }
+
+      // 读取 package.json scripts（最多 5 个）
+      try {
+        const pkg = JSON.parse(fs.readFileSync(path.join(cwd, 'package.json'), 'utf-8'))
+        if (pkg.scripts) {
+          scripts = Object.keys(pkg.scripts).slice(0, 5)
+        }
+      } catch {
+        // 忽略读取错误
+      }
+    }
+
+    // Python
+    if (files.some(f => ['pyproject.toml', 'requirements.txt', 'Pipfile', 'uv.lock', 'poetry.lock'].includes(f))) {
+      types.push('python')
+      if (files.includes('uv.lock')) packageManager = 'uv'
+      else if (files.includes('poetry.lock')) packageManager = 'poetry'
+      else if (files.includes('Pipfile')) packageManager = 'pipenv'
+      else packageManager = 'pip'
+    }
+
+    // Rust
+    if (files.includes('Cargo.toml')) {
+      types.push('rust')
+      packageManager = 'cargo'
+    }
+
+    // Go
+    if (files.includes('go.mod')) {
+      types.push('go')
+    }
+
+    // Docker
+    if (files.some(f => f.includes('docker') || f === 'Dockerfile' || f === 'compose.yaml')) {
+      types.push('docker')
+    }
+
+    // Makefile
+    if (files.includes('Makefile')) {
+      types.push('make')
+    }
+
+    // 3. Git 检测优化（< 20ms）
+    if (files.includes('.git') || fs.existsSync(path.join(cwd, '.git'))) {
+      types.push('git')
+
+      try {
+        // 读取 .git/HEAD 文件获取分支（最快，< 5ms）
+        const headFile = fs.readFileSync(path.join(cwd, '.git/HEAD'), 'utf-8')
+        const refMatch = headFile.match(/ref: refs\/heads\/(.+)/)
+
+        if (refMatch) {
+          // 正常分支
+          git = {
+            branch: refMatch[1].trim(),
+            status: quickGitStatus(cwd),
+          }
+        } else {
+          // detached HEAD 状态（直接 commit hash）
+          const shortHash = headFile.trim().substring(0, 7)
+          git = {
+            branch: `HEAD (${shortHash})`,
+            status: quickGitStatus(cwd),
+          }
+        }
+      } catch {
+        // Git 检测失败，只标记类型，不提供详细信息
+      }
+    }
+
+  } catch (error) {
+    // 检测失败，返回空上下文
+  }
+
+  const hasContext =
+    types.length > 0 ||
+    !!packageManager ||
+    !!git ||
+    (scripts && scripts.length > 0)
+
+  if (!hasContext) {
+    return null
+  }
+
+  return { types, packageManager, git, scripts }
+}
+
+/**
+ * Git status 快速检测（带超时）
+ */
+function quickGitStatus(cwd: string): 'clean' | 'dirty' | 'unknown' {
+  try {
+    // 使用 git status --porcelain，带超时保护
+    const result = execSync('git status --porcelain', {
+      cwd,
+      stdio: 'pipe',
+      timeout: 100,  // 100ms 超时
+      encoding: 'utf-8',
+    })
+    return result.trim() === '' ? 'clean' : 'dirty'
+  } catch {
+    // 超时或失败，返回 unknown
+    return 'unknown'
+  }
+}
+
+/**
+ * 格式化项目上下文为字符串（供 AI 使用）
+ * 格式：当前项目: nodejs+docker+git | pnpm | Git分支: main (有改动) | 脚本: dev, build, test
+ */
+export function formatProjectContext(project: ProjectContext): string {
+  if (project.types.length === 0) {
+    return ''
+  }
+
+  const parts: string[] = []
+
+  // 项目类型
+  parts.push(`当前项目: ${project.types.join('+')}`)
+
+  // 包管理器
+  if (project.packageManager) {
+    parts.push(` | ${project.packageManager}`)
+  }
+
+  // Git 信息
+  if (project.git) {
+    let statusDesc = ''
+    if (project.git.status === 'clean') {
+      statusDesc = '干净'
+    } else if (project.git.status === 'dirty') {
+      statusDesc = '有改动'
+    } else {
+      statusDesc = '未知'
+    }
+    parts.push(` | Git分支: ${project.git.branch} (${statusDesc})`)
+  }
+
+  // 可用脚本
+  if (project.scripts && project.scripts.length > 0) {
+    parts.push(` | 脚本: ${project.scripts.join(', ')}`)
+  }
+
+  return parts.join('')
+}

package/src/prompts.ts

@@ -14,8 +14,11 @@ export const SHELL_COMMAND_SYSTEM_PROMPT = `你是一个专业的 shell 脚本
 
 ### 📋 输入数据格式说明
 你会收到以下 XML 标签包裹的上下文信息：
-- <system_info>：用户的操作系统、Shell 类型、当前目录、包管理器等环境信息
+- <system_info>：用户的操作系统、Shell 类型、当前目录、包管理器、可用工具等环境信息
 - <command_history>：用户最近执行的命令历史（用于理解上下文引用，如"刚才的文件"、"上一个命令"）
+- <user_preferences>：**用户的命令使用偏好**（格式：命令名(使用次数)），帮助你了解用户习惯
+  - 例如：git(234), eza(156) 表示用户经常使用 git 和 eza 命令
+  - 生成命令时可参考偏好，但最终应结合任务需求和 <system_info> 综合判断
 - <execution_log>：**多步骤任务的关键信息**，记录了之前步骤的命令、退出码和输出结果
   - 如果存在此标签，说明这是一个多步骤任务
   - 必须检查每个 <step> 中的 <exit_code>，0=成功，非0=失败
@@ -29,6 +32,7 @@ export const SHELL_COMMAND_SYSTEM_PROMPT = `你是一个专业的 shell 脚本
 4. 根据 <system_info> 中的信息选择合适的命令（如包管理器）
 5. 如果用户引用了之前的操作（如"刚才的"、"上一个"），请参考 <command_history>
 6. 绝对不要输出 pls 或 please 命令！
+7. **建议优先使用标准命令**（ls/find/grep/cat/ps）以确保兼容性和输出捕获，而不是使用 eza/bat/delta 等现代工具，除非用户明确要求使用相关工具，或者用户特别偏好使用相关工具。
 
 ### 📤 输出格式 - 非常重要
 
@@ -189,6 +193,7 @@ export function buildUserContextPrompt(
   userRequest: string,
   sysInfoStr: string,
   historyStr: string,
+  userPreferencesStr: string,
   executedSteps: Array<{ command: string; exitCode: number; output: string }>
 ): string {
   const parts: string[] = []
@@ -205,7 +210,14 @@ export function buildUserContextPrompt(
     parts.push(`</command_history>`)
   }
 
-  // 3. 执行日志（多步骤的核心，紧凑 XML 结构）
+  // 3. 用户偏好（如果有）
+  if (userPreferencesStr && userPreferencesStr.trim()) {
+    parts.push(`<user_preferences>`)
+    parts.push(userPreferencesStr)
+    parts.push(`</user_preferences>`)
+  }
+
+  // 4. 执行日志（多步骤的核心，紧凑 XML 结构）
   if (executedSteps && executedSteps.length > 0) {
     parts.push(`<execution_log>`)
     executedSteps.forEach((step, i) => {
@@ -230,7 +242,7 @@ export function buildUserContextPrompt(
     parts.push(`⚠️ 注意：请检查 <execution_log> 中最后一步的 <exit_code>。如果非 0，请分析 <output> 并修复命令。`)
   }
 
-  // 4. 用户需求
+  // 5. 用户需求
   parts.push(`<user_request>`)
   parts.push(userRequest)
   parts.push(`</user_request>`)
@@ -251,6 +263,7 @@ export const CHAT_SYSTEM_PROMPT = `你是一个命令行专家助手，帮助用
 - <system_info>：用户的操作系统、Shell 类型、当前目录等环境信息
 - <command_history>：用户最近通过 pls 执行的命令（用于理解上下文引用）
 - <shell_history>：用户最近在终端执行的所有命令（如果启用了 Shell Hook）
+- <user_preferences>：用户的命令使用偏好（命令名(使用次数)），帮助你了解用户习惯
 - <user_question>：用户的具体问题
 
 ### 🎯 你的能力
@@ -283,7 +296,8 @@ export function buildChatUserContext(
   sysInfoStr: string,
   plsHistory: string,
   shellHistory: string,
-  shellHookEnabled: boolean
+  shellHookEnabled: boolean,
+  userPreferencesStr?: string
 ): string {
   const parts: string[] = []
 
@@ -303,7 +317,14 @@ export function buildChatUserContext(
     parts.push('</command_history>')
   }
 
-  // 3. 用户问题
+  // 3. 用户偏好（如果有）
+  if (userPreferencesStr && userPreferencesStr.trim()) {
+    parts.push('<user_preferences>')
+    parts.push(userPreferencesStr)
+    parts.push('</user_preferences>')
+  }
+
+  // 4. 用户问题
   parts.push('<user_question>')
   parts.push(userQuestion)
   parts.push('</user_question>')