@yivan-lab/pretty-please 1.0.0

package/dist/src/shell-hook.js

@@ -0,0 +1,348 @@
+import fs from 'fs';
+import path from 'path';
+import os from 'os';
+import chalk from 'chalk';
+import { CONFIG_DIR, getConfig, setConfigValue } from './config.js';
+import { getHistory } from './history.js';
+const SHELL_HISTORY_FILE = path.join(CONFIG_DIR, 'shell_history.jsonl');
+const MAX_SHELL_HISTORY = 20;
+// Hook 标记，用于识别我们添加的内容
+const HOOK_START_MARKER = '# >>> pretty-please shell hook >>>';
+const HOOK_END_MARKER = '# <<< pretty-please shell hook <<<';
+/**
+ * 检测当前 shell 类型
+ */
+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';
+}
+/**
+ * 获取 shell 配置文件路径
+ */
+export function getShellConfigPath(shellType) {
+    const home = os.homedir();
+    switch (shellType) {
+        case 'zsh':
+            return path.join(home, '.zshrc');
+        case 'bash':
+            // macOS 使用 .bash_profile，Linux 使用 .bashrc
+            if (process.platform === 'darwin') {
+                return path.join(home, '.bash_profile');
+            }
+            return path.join(home, '.bashrc');
+        case 'powershell':
+            // PowerShell profile 路径
+            return path.join(home, 'Documents', 'PowerShell', 'Microsoft.PowerShell_profile.ps1');
+        default:
+            return null;
+    }
+}
+/**
+ * 生成 zsh hook 脚本
+ */
+function generateZshHook() {
+    return `
+${HOOK_START_MARKER}
+# 记录命令到 pretty-please 历史
+__pls_preexec() {
+  __PLS_LAST_CMD="$1"
+  __PLS_CMD_START=$(date +%s)
+}
+
+__pls_precmd() {
+  local exit_code=$?
+  if [[ -n "$__PLS_LAST_CMD" ]]; then
+    local end_time=$(date +%s)
+    local timestamp=$(date -u +"%Y-%m-%dT%H:%M:%SZ")
+    # 转义命令中的特殊字符
+    local escaped_cmd=$(echo "$__PLS_LAST_CMD" | sed 's/\\\\/\\\\\\\\/g; s/"/\\\\"/g')
+    echo "{\\"cmd\\":\\"$escaped_cmd\\",\\"exit\\":$exit_code,\\"time\\":\\"$timestamp\\"}" >> "${CONFIG_DIR}/shell_history.jsonl"
+    # 保持文件不超过 ${MAX_SHELL_HISTORY} 行
+    tail -n ${MAX_SHELL_HISTORY} "${CONFIG_DIR}/shell_history.jsonl" > "${CONFIG_DIR}/shell_history.jsonl.tmp" && mv "${CONFIG_DIR}/shell_history.jsonl.tmp" "${CONFIG_DIR}/shell_history.jsonl"
+    unset __PLS_LAST_CMD
+  fi
+}
+
+autoload -Uz add-zsh-hook
+add-zsh-hook preexec __pls_preexec
+add-zsh-hook precmd __pls_precmd
+${HOOK_END_MARKER}
+`;
+}
+/**
+ * 生成 bash hook 脚本
+ */
+function generateBashHook() {
+    return `
+${HOOK_START_MARKER}
+# 记录命令到 pretty-please 历史
+__pls_prompt_command() {
+  local exit_code=$?
+  local last_cmd=$(history 1 | sed 's/^ *[0-9]* *//')
+  if [[ -n "$last_cmd" && "$last_cmd" != "$__PLS_LAST_CMD" ]]; then
+    __PLS_LAST_CMD="$last_cmd"
+    local timestamp=$(date -u +"%Y-%m-%dT%H:%M:%SZ")
+    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 ${MAX_SHELL_HISTORY} "${CONFIG_DIR}/shell_history.jsonl" > "${CONFIG_DIR}/shell_history.jsonl.tmp" && mv "${CONFIG_DIR}/shell_history.jsonl.tmp" "${CONFIG_DIR}/shell_history.jsonl"
+  fi
+}
+
+if [[ ! "$PROMPT_COMMAND" =~ __pls_prompt_command ]]; then
+  PROMPT_COMMAND="__pls_prompt_command;\${PROMPT_COMMAND}"
+fi
+${HOOK_END_MARKER}
+`;
+}
+/**
+ * 生成 PowerShell hook 脚本
+ */
+function generatePowerShellHook() {
+    return `
+${HOOK_START_MARKER}
+# 记录命令到 pretty-please 历史
+$Global:__PlsLastCmd = ""
+
+function __Pls_RecordCommand {
+    $lastCmd = (Get-History -Count 1).CommandLine
+    if ($lastCmd -and $lastCmd -ne $Global:__PlsLastCmd) {
+        $Global:__PlsLastCmd = $lastCmd
+        $exitCode = $LASTEXITCODE
+        if ($null -eq $exitCode) { $exitCode = 0 }
+        $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
+        # 保持文件不超过 ${MAX_SHELL_HISTORY} 行
+        $content = Get-Content "${CONFIG_DIR}/shell_history.jsonl" -Tail ${MAX_SHELL_HISTORY}
+        $content | Set-Content "${CONFIG_DIR}/shell_history.jsonl"
+    }
+}
+
+if (-not (Get-Variable -Name __PlsPromptBackup -ErrorAction SilentlyContinue)) {
+    $Global:__PlsPromptBackup = $function:prompt
+    function Global:prompt {
+        __Pls_RecordCommand
+        & $Global:__PlsPromptBackup
+    }
+}
+${HOOK_END_MARKER}
+`;
+}
+/**
+ * 生成 hook 脚本
+ */
+function generateHookScript(shellType) {
+    switch (shellType) {
+        case 'zsh':
+            return generateZshHook();
+        case 'bash':
+            return generateBashHook();
+        case 'powershell':
+            return generatePowerShellHook();
+        default:
+            return null;
+    }
+}
+/**
+ * 安装 shell hook
+ */
+export async function installShellHook() {
+    const shellType = detectShell();
+    const configPath = getShellConfigPath(shellType);
+    if (!configPath) {
+        console.log(chalk.red(`❌ 不支持的 shell 类型: ${shellType}`));
+        return false;
+    }
+    const hookScript = generateHookScript(shellType);
+    if (!hookScript) {
+        console.log(chalk.red(`❌ 无法为 ${shellType} 生成 hook 脚本`));
+        return false;
+    }
+    // 检查是否已安装
+    if (fs.existsSync(configPath)) {
+        const content = fs.readFileSync(configPath, 'utf-8');
+        if (content.includes(HOOK_START_MARKER)) {
+            console.log(chalk.yellow('⚠️  Shell hook 已安装，跳过'));
+            setConfigValue('shellHook', true);
+            return true;
+        }
+    }
+    // 确保配置目录存在
+    if (!fs.existsSync(CONFIG_DIR)) {
+        fs.mkdirSync(CONFIG_DIR, { recursive: true });
+    }
+    // 备份原配置文件
+    if (fs.existsSync(configPath)) {
+        const backupPath = configPath + '.pls-backup';
+        fs.copyFileSync(configPath, backupPath);
+        console.log(chalk.gray(`已备份原配置文件到: ${backupPath}`));
+    }
+    // 追加 hook 脚本
+    fs.appendFileSync(configPath, hookScript);
+    // 更新配置
+    setConfigValue('shellHook', true);
+    console.log(chalk.green(`✅ Shell hook 已安装到: ${configPath}`));
+    console.log(chalk.yellow('⚠️  请重启终端或执行以下命令使其生效:'));
+    console.log(chalk.cyan(`   source ${configPath}`));
+    return true;
+}
+/**
+ * 卸载 shell hook
+ */
+export function uninstallShellHook() {
+    const shellType = detectShell();
+    const configPath = getShellConfigPath(shellType);
+    if (!configPath || !fs.existsSync(configPath)) {
+        console.log(chalk.yellow('⚠️  未找到 shell 配置文件'));
+        setConfigValue('shellHook', false);
+        return true;
+    }
+    let content = fs.readFileSync(configPath, 'utf-8');
+    // 移除 hook 脚本
+    const startIndex = content.indexOf(HOOK_START_MARKER);
+    const endIndex = content.indexOf(HOOK_END_MARKER);
+    if (startIndex === -1 || endIndex === -1) {
+        console.log(chalk.yellow('⚠️  未找到已安装的 hook'));
+        setConfigValue('shellHook', false);
+        return true;
+    }
+    // 移除从标记开始到结束的所有内容（包括换行符）
+    const before = content.substring(0, startIndex);
+    const after = content.substring(endIndex + HOOK_END_MARKER.length);
+    content = before + after.replace(/^\n/, '');
+    fs.writeFileSync(configPath, content);
+    setConfigValue('shellHook', false);
+    // 清空 shell 历史文件
+    if (fs.existsSync(SHELL_HISTORY_FILE)) {
+        fs.unlinkSync(SHELL_HISTORY_FILE);
+    }
+    console.log(chalk.green('✅ Shell hook 已卸载'));
+    console.log(chalk.yellow('⚠️  请重启终端使其生效'));
+    return true;
+}
+/**
+ * 读取 shell 历史记录
+ */
+export function getShellHistory() {
+    const config = getConfig();
+    // 如果未启用 shell hook，返回空数组
+    if (!config.shellHook) {
+        return [];
+    }
+    if (!fs.existsSync(SHELL_HISTORY_FILE)) {
+        return [];
+    }
+    try {
+        const content = fs.readFileSync(SHELL_HISTORY_FILE, 'utf-8');
+        const lines = content.trim().split('\n').filter(line => line.trim());
+        return lines.map(line => {
+            try {
+                return JSON.parse(line);
+            }
+            catch {
+                return null;
+            }
+        }).filter(Boolean);
+    }
+    catch {
+        return [];
+    }
+}
+/**
+ * 从 pls history 中查找匹配的记录
+ * @param {string} prompt - pls 命令后面的 prompt 部分
+ * @returns {object|null} 匹配的 pls history 记录
+ */
+function findPlsHistoryMatch(prompt) {
+    const plsHistory = getHistory();
+    // 尝试精确匹配 userPrompt
+    for (const record of plsHistory) {
+        if (record.userPrompt === prompt) {
+            return record;
+        }
+    }
+    // 尝试模糊匹配（处理引号等情况）
+    const normalizedPrompt = prompt.trim().replace(/^["']|["']$/g, '');
+    for (const record of plsHistory) {
+        if (record.userPrompt === normalizedPrompt) {
+            return record;
+        }
+    }
+    return null;
+}
+/**
+ * 格式化 shell 历史供 AI 使用
+ * 对于 pls 命令，会从 pls history 中查找对应的详细信息
+ */
+export function formatShellHistoryForAI() {
+    const history = getShellHistory();
+    if (history.length === 0) {
+        return '';
+    }
+    // pls 的子命令列表（这些不是 AI prompt）
+    const plsSubcommands = ['config', 'history', 'hook', 'help', '--help', '-h', '--version', '-v'];
+    const lines = history.map((item, index) => {
+        const status = item.exit === 0 ? '✓' : `✗ 退出码:${item.exit}`;
+        // 检查是否是 pls 命令
+        const plsMatch = item.cmd.match(/^(pls|please)\s+(.+)$/);
+        if (plsMatch) {
+            let args = plsMatch[2];
+            // 去掉 --debug / -d 选项，获取真正的参数
+            args = args.replace(/^(--debug|-d)\s+/, '');
+            const firstArg = args.split(/\s+/)[0];
+            // 如果是子命令，当作普通命令处理
+            if (plsSubcommands.includes(firstArg)) {
+                return `${index + 1}. ${item.cmd} ${status}`;
+            }
+            // 是 AI prompt，尝试从 pls history 查找详细信息
+            const prompt = args;
+            const plsRecord = findPlsHistoryMatch(prompt);
+            if (plsRecord) {
+                // 找到对应的 pls 记录，展示详细信息
+                if (plsRecord.reason === 'builtin') {
+                    return `${index + 1}. [pls] "${prompt}" → 生成命令: ${plsRecord.command} (包含 builtin，未执行)`;
+                }
+                else if (plsRecord.executed) {
+                    const execStatus = plsRecord.exitCode === 0 ? '✓' : `✗ 退出码:${plsRecord.exitCode}`;
+                    return `${index + 1}. [pls] "${prompt}" → 实际执行: ${plsRecord.command} ${execStatus}`;
+                }
+                else {
+                    return `${index + 1}. [pls] "${prompt}" → 生成命令: ${plsRecord.command} (用户取消执行)`;
+                }
+            }
+            // 找不到记录，只显示原始命令
+            return `${index + 1}. [pls] "${prompt}" ${status}`;
+        }
+        // 普通命令
+        return `${index + 1}. ${item.cmd} ${status}`;
+    });
+    return `【用户终端最近执行的命令】\n${lines.join('\n')}`;
+}
+/**
+ * 获取 hook 状态
+ */
+export function getHookStatus() {
+    const config = getConfig();
+    const shellType = detectShell();
+    const configPath = getShellConfigPath(shellType);
+    let installed = false;
+    if (configPath && fs.existsSync(configPath)) {
+        const content = fs.readFileSync(configPath, 'utf-8');
+        installed = content.includes(HOOK_START_MARKER);
+    }
+    return {
+        enabled: config.shellHook,
+        installed,
+        shellType,
+        configPath,
+        historyFile: SHELL_HISTORY_FILE
+    };
+}

package/dist/src/sysinfo.d.ts

@@ -0,0 +1,15 @@
+/**
+ * 收集系统信息
+ */
+export function collectSystemInfo(): {
+    os: NodeJS.Platform;
+    arch: NodeJS.Architecture;
+    shell: string;
+    packageManager: string;
+    cwd: string;
+    user: string;
+};
+/**
+ * 将系统信息格式化为字符串（供 AI 使用）
+ */
+export function formatSystemInfo(): string;

package/dist/src/sysinfo.js

@@ -0,0 +1,52 @@
+import os from 'os';
+import { execSync } from 'child_process';
+/**
+ * 检测包管理器
+ */
+function detectPackageManager() {
+    const managers = [
+        { name: 'brew', command: 'brew' },
+        { name: 'apt', command: 'apt-get' },
+        { name: 'dnf', command: 'dnf' },
+        { name: 'yum', command: 'yum' },
+        { name: 'pacman', command: 'pacman' },
+        { name: 'zypper', command: 'zypper' },
+        { name: 'apk', command: 'apk' }
+    ];
+    for (const mgr of managers) {
+        try {
+            execSync(`which ${mgr.command}`, { stdio: 'ignore' });
+            return mgr.name;
+        }
+        catch {
+            // 继续检测下一个
+        }
+    }
+    return 'unknown';
+}
+/**
+ * 获取当前 Shell
+ */
+function getCurrentShell() {
+    return process.env.SHELL || 'unknown';
+}
+/**
+ * 收集系统信息
+ */
+export function collectSystemInfo() {
+    return {
+        os: os.platform(),
+        arch: os.arch(),
+        shell: getCurrentShell(),
+        packageManager: detectPackageManager(),
+        cwd: process.cwd(),
+        user: os.userInfo().username
+    };
+}
+/**
+ * 将系统信息格式化为字符串（供 AI 使用）
+ */
+export function formatSystemInfo() {
+    const info = collectSystemInfo();
+    return `OS: ${info.os}, Arch: ${info.arch}, Shell: ${info.shell}, PkgMgr: ${info.packageManager}, CWD: ${info.cwd}`;
+}

package/dist/src/ui/theme.d.ts

@@ -0,0 +1,26 @@
+export declare const theme: {
+    readonly primary: "#00D9FF";
+    readonly secondary: "#A78BFA";
+    readonly accent: "#F472B6";
+    readonly success: "#10B981";
+    readonly error: "#EF4444";
+    readonly warning: "#F59E0B";
+    readonly info: "#3B82F6";
+    readonly text: {
+        readonly primary: "#E5E7EB";
+        readonly secondary: "#9CA3AF";
+        readonly muted: "#6B7280";
+        readonly dim: "#4B5563";
+    };
+    readonly border: "#374151";
+    readonly divider: "#1F2937";
+    readonly code: {
+        readonly background: "#1F2937";
+        readonly text: "#E5E7EB";
+        readonly keyword: "#C678DD";
+        readonly string: "#98C379";
+        readonly function: "#61AFEF";
+        readonly comment: "#5C6370";
+    };
+};
+export type Theme = typeof theme;

package/dist/src/ui/theme.js

@@ -0,0 +1,31 @@
+// 主题配置 - 优雅的配色方案
+export const theme = {
+    // 主色调
+    primary: '#00D9FF', // 青色 - 主要交互元素
+    secondary: '#A78BFA', // 紫色 - 次要元素
+    accent: '#F472B6', // 粉色 - 强调元素
+    // 状态色
+    success: '#10B981', // 绿色 - 成功
+    error: '#EF4444', // 红色 - 错误
+    warning: '#F59E0B', // 橙色 - 警告
+    info: '#3B82F6', // 蓝色 - 信息
+    // 文本色
+    text: {
+        primary: '#E5E7EB', // 主文本
+        secondary: '#9CA3AF', // 次要文本
+        muted: '#6B7280', // 弱化文本
+        dim: '#4B5563', // 暗淡文本
+    },
+    // 边框和分隔
+    border: '#374151', // 边框色
+    divider: '#1F2937', // 分隔线
+    // 代码相关
+    code: {
+        background: '#1F2937',
+        text: '#E5E7EB',
+        keyword: '#C678DD',
+        string: '#98C379',
+        function: '#61AFEF',
+        comment: '#5C6370',
+    }
+};

package/dist/src/utils/console.d.ts

@@ -0,0 +1,44 @@
+/**
+ * 计算字符串的显示宽度（中文占2个宽度）
+ */
+export function getDisplayWidth(str: any): number;
+/**
+ * 绘制命令框（原生版本）
+ */
+export function drawCommandBox(command: any, title?: string): void;
+/**
+ * 格式化耗时
+ */
+export function formatDuration(ms: any): string;
+/**
+ * 输出分隔线
+ */
+export function printSeparator(text?: string, length?: number): void;
+/**
+ * 输出成功消息
+ */
+export function success(message: any): void;
+/**
+ * 输出错误消息
+ */
+export function error(message: any): void;
+/**
+ * 输出警告消息
+ */
+export function warning(message: any): void;
+/**
+ * 输出信息消息
+ */
+export function info(message: any): void;
+/**
+ * 输出灰色文本
+ */
+export function muted(message: any): void;
+/**
+ * 输出标题
+ */
+export function title(message: any): void;
+/**
+ * 输出主色文本
+ */
+export function primary(message: any): void;

package/dist/src/utils/console.js

@@ -0,0 +1,114 @@
+import chalk from 'chalk';
+/**
+ * 原生控制台输出工具函数
+ * 用于不需要 Ink 的场景，避免清屏和性能问题
+ */
+// 主题色
+const colors = {
+    primary: '#00D9FF',
+    secondary: '#A78BFA',
+    accent: '#F472B6',
+    success: '#10B981',
+    error: '#EF4444',
+    warning: '#F59E0B',
+    info: '#3B82F6',
+    muted: '#6B7280',
+};
+/**
+ * 计算字符串的显示宽度（中文占2个宽度）
+ */
+export function getDisplayWidth(str) {
+    let width = 0;
+    for (const char of str) {
+        if (char.match(/[\u4e00-\u9fff\u3400-\u4dbf\uff00-\uffef\u3000-\u303f]/)) {
+            width += 2;
+        }
+        else {
+            width += 1;
+        }
+    }
+    return width;
+}
+/**
+ * 绘制命令框（原生版本）
+ */
+export function drawCommandBox(command, title = '生成命令') {
+    const lines = command.split('\n');
+    const titleWidth = getDisplayWidth(title);
+    const maxContentWidth = Math.max(...lines.map(l => getDisplayWidth(l)));
+    const boxWidth = Math.max(maxContentWidth + 4, titleWidth + 6, 20);
+    const topPadding = boxWidth - titleWidth - 5;
+    const topBorder = '┌─ ' + title + ' ' + '─'.repeat(topPadding) + '┐';
+    const bottomBorder = '└' + '─'.repeat(boxWidth - 2) + '┘';
+    console.log(chalk.hex(colors.warning)(topBorder));
+    for (const line of lines) {
+        const lineWidth = getDisplayWidth(line);
+        const padding = ' '.repeat(boxWidth - lineWidth - 4);
+        console.log(chalk.hex(colors.warning)('│ ') +
+            chalk.hex(colors.primary)(line) +
+            padding +
+            chalk.hex(colors.warning)(' │'));
+    }
+    console.log(chalk.hex(colors.warning)(bottomBorder));
+}
+/**
+ * 格式化耗时
+ */
+export function formatDuration(ms) {
+    if (ms < 1000) {
+        return `${ms}ms`;
+    }
+    return `${(ms / 1000).toFixed(2)}s`;
+}
+/**
+ * 输出分隔线
+ */
+export function printSeparator(text = '输出', length = 38) {
+    const textPart = text ? ` ${text} ` : '';
+    const lineLength = Math.max(0, length - textPart.length);
+    const leftDashes = '─'.repeat(Math.floor(lineLength / 2));
+    const rightDashes = '─'.repeat(Math.ceil(lineLength / 2));
+    console.log(chalk.gray(`${leftDashes}${textPart}${rightDashes}`));
+}
+/**
+ * 输出成功消息
+ */
+export function success(message) {
+    console.log(chalk.hex(colors.success)('✓ ' + message));
+}
+/**
+ * 输出错误消息
+ */
+export function error(message) {
+    console.log(chalk.hex(colors.error)('✗ ' + message));
+}
+/**
+ * 输出警告消息
+ */
+export function warning(message) {
+    console.log(chalk.hex(colors.warning)('⚠️  ' + message));
+}
+/**
+ * 输出信息消息
+ */
+export function info(message) {
+    console.log(chalk.hex(colors.info)(message));
+}
+/**
+ * 输出灰色文本
+ */
+export function muted(message) {
+    console.log(chalk.hex(colors.muted)(message));
+}
+/**
+ * 输出标题
+ */
+export function title(message) {
+    console.log(chalk.bold(message));
+}
+/**
+ * 输出主色文本
+ */
+export function primary(message) {
+    console.log(chalk.hex(colors.primary)(message));
+}