@yivan-lab/pretty-please 1.0.0

package/dist/src/config.js

@@ -0,0 +1,203 @@
+import fs from 'fs';
+import path from 'path';
+import os from 'os';
+import readline from 'readline';
+import chalk from 'chalk';
+const CONFIG_DIR = path.join(os.homedir(), '.please');
+const CONFIG_FILE = path.join(CONFIG_DIR, 'config.json');
+const DEFAULT_CONFIG = {
+    apiKey: '',
+    baseUrl: 'https://api.openai.com/v1',
+    model: 'gpt-4-turbo',
+    provider: 'openai', // Mastra provider: openai, anthropic, deepseek, google, groq, mistral, cohere 等
+    shellHook: false, // 是否启用 shell hook 记录终端命令
+    chatHistoryLimit: 10 // chat 对话历史保留轮数
+};
+// 导出配置目录路径
+export { CONFIG_DIR };
+/**
+ * 确保配置目录存在
+ */
+function ensureConfigDir() {
+    if (!fs.existsSync(CONFIG_DIR)) {
+        fs.mkdirSync(CONFIG_DIR, { recursive: true });
+    }
+}
+/**
+ * 读取配置
+ */
+export function getConfig() {
+    ensureConfigDir();
+    if (!fs.existsSync(CONFIG_FILE)) {
+        return { ...DEFAULT_CONFIG };
+    }
+    try {
+        const content = fs.readFileSync(CONFIG_FILE, 'utf-8');
+        return { ...DEFAULT_CONFIG, ...JSON.parse(content) };
+    }
+    catch {
+        return { ...DEFAULT_CONFIG };
+    }
+}
+/**
+ * 保存配置
+ */
+export function saveConfig(config) {
+    ensureConfigDir();
+    fs.writeFileSync(CONFIG_FILE, JSON.stringify(config, null, 2));
+}
+/**
+ * 设置单个配置项
+ */
+export function setConfigValue(key, value) {
+    const config = getConfig();
+    if (!(key in DEFAULT_CONFIG)) {
+        throw new Error(`未知的配置项: ${key}`);
+    }
+    // 处理特殊类型
+    if (key === 'shellHook') {
+        config[key] = value === 'true' || value === true;
+    }
+    else if (key === 'chatHistoryLimit') {
+        const num = parseInt(value, 10);
+        if (isNaN(num) || num < 1) {
+            throw new Error('chatHistoryLimit 必须是大于 0 的整数');
+        }
+        config[key] = num;
+    }
+    else if (key === 'provider') {
+        // 验证 provider 值
+        const validProviders = ['openai', 'anthropic', 'deepseek', 'google', 'groq', 'mistral', 'cohere', 'fireworks', 'together'];
+        if (!validProviders.includes(value)) {
+            throw new Error(`provider 必须是以下之一: ${validProviders.join(', ')}`);
+        }
+        config[key] = value;
+    }
+    else {
+        config[key] = value;
+    }
+    saveConfig(config);
+    return config;
+}
+/**
+ * 检查配置是否有效
+ */
+export function isConfigValid() {
+    const config = getConfig();
+    return config.apiKey && config.apiKey.length > 0;
+}
+/**
+ * 隐藏 API Key 中间部分
+ */
+export function maskApiKey(apiKey) {
+    if (!apiKey || apiKey.length < 10)
+        return apiKey || '(未设置)';
+    return apiKey.slice(0, 6) + '****' + apiKey.slice(-4);
+}
+/**
+ * 显示当前配置
+ */
+export function displayConfig() {
+    const config = getConfig();
+    console.log(chalk.bold('\n当前配置:'));
+    console.log(chalk.gray('━'.repeat(40)));
+    console.log(`  ${chalk.cyan('apiKey')}:           ${maskApiKey(config.apiKey)}`);
+    console.log(`  ${chalk.cyan('baseUrl')}:          ${config.baseUrl}`);
+    console.log(`  ${chalk.cyan('provider')}:         ${config.provider}`);
+    console.log(`  ${chalk.cyan('model')}:            ${config.model}`);
+    console.log(`  ${chalk.cyan('shellHook')}:        ${config.shellHook ? chalk.green('已启用') : chalk.gray('未启用')}`);
+    console.log(`  ${chalk.cyan('chatHistoryLimit')}: ${config.chatHistoryLimit} 轮`);
+    console.log(chalk.gray('━'.repeat(40)));
+    console.log(chalk.gray(`配置文件: ${CONFIG_FILE}\n`));
+}
+/**
+ * 创建 readline 接口
+ */
+function createReadlineInterface() {
+    return readline.createInterface({
+        input: process.stdin,
+        output: process.stdout
+    });
+}
+/**
+ * 异步提问
+ */
+function question(rl, prompt) {
+    return new Promise((resolve) => {
+        rl.question(prompt, (answer) => {
+            resolve(answer);
+        });
+    });
+}
+/**
+ * 交互式配置向导
+ */
+export async function runConfigWizard() {
+    const rl = createReadlineInterface();
+    const config = getConfig();
+    console.log(chalk.bold.hex('#00D9FF')('\n🔧 Pretty Please 配置向导'));
+    console.log(chalk.gray('━'.repeat(50)));
+    console.log(chalk.gray('直接回车使用默认值，输入值后回车确认\n'));
+    try {
+        // 1. Provider
+        const validProviders = ['openai', 'anthropic', 'deepseek', 'google', 'groq', 'mistral', 'cohere', 'fireworks', 'together'];
+        const providerHint = chalk.gray(`(可选: ${validProviders.join(', ')})`);
+        const providerPrompt = `${chalk.cyan('Provider')} ${providerHint}\n${chalk.gray('默认:')} ${chalk.yellow(config.provider)} ${chalk.gray('→')} `;
+        const provider = await question(rl, providerPrompt);
+        if (provider.trim()) {
+            if (!validProviders.includes(provider.trim())) {
+                console.log(chalk.hex('#EF4444')(`\n✗ 无效的 provider，必须是以下之一: ${validProviders.join(', ')}`));
+                console.log();
+                rl.close();
+                return;
+            }
+            config.provider = provider.trim();
+        }
+        // 2. Base URL
+        const baseUrlPrompt = `${chalk.cyan('API Base URL')}\n${chalk.gray('默认:')} ${chalk.yellow(config.baseUrl)} ${chalk.gray('→')} `;
+        const baseUrl = await question(rl, baseUrlPrompt);
+        if (baseUrl.trim()) {
+            config.baseUrl = baseUrl.trim();
+        }
+        // 3. API Key
+        const currentKeyDisplay = config.apiKey ? maskApiKey(config.apiKey) : '(未设置)';
+        const apiKeyPrompt = `${chalk.cyan('API Key')} ${chalk.gray(`(当前: ${currentKeyDisplay})`)}\n${chalk.gray('→')} `;
+        const apiKey = await question(rl, apiKeyPrompt);
+        if (apiKey.trim()) {
+            config.apiKey = apiKey.trim();
+        }
+        // 4. Model
+        const modelPrompt = `${chalk.cyan('Model')}\n${chalk.gray('默认:')} ${chalk.yellow(config.model)} ${chalk.gray('→')} `;
+        const model = await question(rl, modelPrompt);
+        if (model.trim()) {
+            config.model = model.trim();
+        }
+        // 5. Shell Hook
+        const shellHookPrompt = `${chalk.cyan('启用 Shell Hook')} ${chalk.gray('(记录终端命令历史)')}\n${chalk.gray('默认:')} ${chalk.yellow(config.shellHook ? 'true' : 'false')} ${chalk.gray('→')} `;
+        const shellHook = await question(rl, shellHookPrompt);
+        if (shellHook.trim()) {
+            config.shellHook = shellHook.trim() === 'true';
+        }
+        // 6. Chat History Limit
+        const chatHistoryPrompt = `${chalk.cyan('Chat 历史保留轮数')}\n${chalk.gray('默认:')} ${chalk.yellow(config.chatHistoryLimit)} ${chalk.gray('→')} `;
+        const chatHistoryLimit = await question(rl, chatHistoryPrompt);
+        if (chatHistoryLimit.trim()) {
+            const num = parseInt(chatHistoryLimit.trim(), 10);
+            if (!isNaN(num) && num > 0) {
+                config.chatHistoryLimit = num;
+            }
+        }
+        saveConfig(config);
+        console.log('\n' + chalk.gray('━'.repeat(50)));
+        console.log(chalk.hex('#10B981')('✅ 配置已保存'));
+        console.log(chalk.gray(`   ${CONFIG_FILE}`));
+        console.log();
+    }
+    catch (error) {
+        console.log(chalk.hex('#EF4444')(`\n✗ 配置失败: ${error.message}`));
+        console.log();
+    }
+    finally {
+        rl.close();
+    }
+}

package/dist/src/history.d.ts

@@ -0,0 +1,20 @@
+/**
+ * 读取历史记录
+ */
+export function getHistory(): any;
+/**
+ * 添加一条历史记录
+ */
+export function addHistory(record: any): void;
+/**
+ * 清空历史记录
+ */
+export function clearHistory(): void;
+/**
+ * 格式化历史记录供 AI 使用
+ */
+export function formatHistoryForAI(): string;
+/**
+ * 获取历史记录文件路径（供显示用）
+ */
+export function getHistoryFilePath(): string;

package/dist/src/history.js

@@ -0,0 +1,113 @@
+import fs from 'fs';
+import path from 'path';
+import os from 'os';
+const CONFIG_DIR = path.join(os.homedir(), '.please');
+const HISTORY_FILE = path.join(CONFIG_DIR, 'history.json');
+const MAX_HISTORY = 10;
+const MAX_OUTPUT_LENGTH = 500;
+/**
+ * 确保配置目录存在
+ */
+function ensureConfigDir() {
+    if (!fs.existsSync(CONFIG_DIR)) {
+        fs.mkdirSync(CONFIG_DIR, { recursive: true });
+    }
+}
+/**
+ * 读取历史记录
+ */
+export function getHistory() {
+    ensureConfigDir();
+    if (!fs.existsSync(HISTORY_FILE)) {
+        return [];
+    }
+    try {
+        const content = fs.readFileSync(HISTORY_FILE, 'utf-8');
+        return JSON.parse(content);
+    }
+    catch {
+        return [];
+    }
+}
+/**
+ * 保存历史记录
+ */
+function saveHistory(history) {
+    ensureConfigDir();
+    fs.writeFileSync(HISTORY_FILE, JSON.stringify(history, null, 2));
+}
+/**
+ * 添加一条历史记录
+ */
+export function addHistory(record) {
+    const history = getHistory();
+    // 截断输出
+    if (record.output && record.output.length > MAX_OUTPUT_LENGTH) {
+        record.output = record.output.slice(0, MAX_OUTPUT_LENGTH) + '...(截断)';
+    }
+    // 添加时间戳
+    record.timestamp = new Date().toISOString();
+    // 添加到开头
+    history.unshift(record);
+    // 保留最近 N 条
+    if (history.length > MAX_HISTORY) {
+        history.length = MAX_HISTORY;
+    }
+    saveHistory(history);
+}
+/**
+ * 清空历史记录
+ */
+export function clearHistory() {
+    saveHistory([]);
+}
+/**
+ * 格式化历史记录供 AI 使用
+ */
+export function formatHistoryForAI() {
+    const history = getHistory();
+    if (history.length === 0) {
+        return '';
+    }
+    const lines = history.map((item, index) => {
+        const timeAgo = getTimeAgo(item.timestamp);
+        let status;
+        if (item.executed) {
+            status = item.exitCode === 0 ? '✓' : `✗ 退出码:${item.exitCode}`;
+        }
+        else if (item.reason === 'builtin') {
+            status = '(包含 builtin，未执行)';
+        }
+        else {
+            status = '(用户取消执行)';
+        }
+        let line = `${index + 1}. [${timeAgo}] "${item.userPrompt}" → ${item.command} ${status}`;
+        // 如果有输出且命令失败，附加输出摘要
+        if (item.output && item.exitCode !== 0) {
+            line += `\n   输出: ${item.output.split('\n')[0]}`; // 只取第一行
+        }
+        return line;
+    }).reverse(); // 从旧到新排列
+    return `【最近通过 pls 执行的命令】\n${lines.join('\n')}`;
+}
+/**
+ * 计算时间差的友好显示
+ */
+function getTimeAgo(timestamp) {
+    const now = Date.now();
+    const then = new Date(timestamp).getTime();
+    const diff = Math.floor((now - then) / 1000);
+    if (diff < 60)
+        return '刚刚';
+    if (diff < 3600)
+        return `${Math.floor(diff / 60)}分钟前`;
+    if (diff < 86400)
+        return `${Math.floor(diff / 3600)}小时前`;
+    return `${Math.floor(diff / 86400)}天前`;
+}
+/**
+ * 获取历史记录文件路径（供显示用）
+ */
+export function getHistoryFilePath() {
+    return HISTORY_FILE;
+}

package/dist/src/mastra-agent.d.ts

@@ -0,0 +1,7 @@
+import { Agent } from '@mastra/core';
+/**
+ * 创建 Mastra Shell Agent
+ * 根据用户配置的 API Key、Base URL、Provider 和 Model
+ * 使用 ai.js 中的统一提示词
+ */
+export declare function createShellAgent(): Agent<"shell-commander", Record<string, import("@mastra/core").ToolAction<any, any, any, any, import("@mastra/core").ToolExecutionContext<any, any, any>>>, Record<string, import("@mastra/core").Metric>>;

package/dist/src/mastra-agent.js

@@ -0,0 +1,31 @@
+import { Agent } from '@mastra/core';
+import { getConfig } from './config.js';
+import { buildSystemPrompt } from './ai.js';
+import { formatSystemInfo } from './sysinfo.js';
+import { formatHistoryForAI } from './history.js';
+import { formatShellHistoryForAI, getShellHistory } from './shell-hook.js';
+/**
+ * 创建 Mastra Shell Agent
+ * 根据用户配置的 API Key、Base URL、Provider 和 Model
+ * 使用 ai.js 中的统一提示词
+ */
+export function createShellAgent() {
+    const config = getConfig();
+    // 组合 provider/model 格式（Mastra 要求）
+    const modelId = `${config.provider}/${config.model}`;
+    // 构建系统提示词（使用 ai.js 中的统一函数）
+    const sysinfo = formatSystemInfo();
+    const plsHistory = formatHistoryForAI();
+    const shellHistory = formatShellHistoryForAI();
+    const shellHookEnabled = config.shellHook && getShellHistory().length > 0;
+    const systemPrompt = buildSystemPrompt(sysinfo, plsHistory, shellHistory, shellHookEnabled);
+    return new Agent({
+        name: 'shell-commander',
+        instructions: systemPrompt,
+        model: {
+            url: config.baseUrl,
+            id: modelId,
+            apiKey: config.apiKey,
+        },
+    });
+}

package/dist/src/multi-step.d.ts

@@ -0,0 +1,41 @@
+import { z } from 'zod';
+/**
+ * 多步骤命令的 Zod Schema
+ */
+export declare const CommandStepSchema: z.ZodObject<{
+    command: z.ZodString;
+    continue: z.ZodOptional<z.ZodBoolean>;
+    reasoning: z.ZodOptional<z.ZodString>;
+    nextStepHint: z.ZodOptional<z.ZodString>;
+}, "strip", z.ZodTypeAny, {
+    command: string;
+    continue?: boolean | undefined;
+    reasoning?: string | undefined;
+    nextStepHint?: string | undefined;
+}, {
+    command: string;
+    continue?: boolean | undefined;
+    reasoning?: string | undefined;
+    nextStepHint?: string | undefined;
+}>;
+export type CommandStep = z.infer<typeof CommandStepSchema>;
+/**
+ * 执行步骤结果
+ */
+export interface ExecutedStep extends CommandStep {
+    exitCode: number;
+    output: string;
+}
+/**
+ * 生成系统上下文信息（供 Mastra 使用）
+ */
+export declare function getFullSystemPrompt(): string;
+/**
+ * 使用 Mastra 生成多步骤命令
+ */
+export declare function generateMultiStepCommand(userPrompt: string, previousSteps?: ExecutedStep[], options?: {
+    debug?: boolean;
+}): Promise<{
+    stepData: CommandStep;
+    debugInfo?: any;
+}>;

package/dist/src/multi-step.js

@@ -0,0 +1,67 @@
+import { z } from 'zod';
+import { createShellAgent } from './mastra-agent.js';
+import { buildSystemPrompt } from './ai.js';
+import { formatSystemInfo } from './sysinfo.js';
+import { formatHistoryForAI } from './history.js';
+import { formatShellHistoryForAI, getShellHistory } from './shell-hook.js';
+import { getConfig } from './config.js';
+/**
+ * 多步骤命令的 Zod Schema
+ */
+export const CommandStepSchema = z.object({
+    command: z.string(),
+    continue: z.boolean().optional(), // 可选！没有 continue = 单步模式
+    reasoning: z.string().optional(),
+    nextStepHint: z.string().optional(),
+});
+/**
+ * 生成系统上下文信息（供 Mastra 使用）
+ */
+export function getFullSystemPrompt() {
+    const config = getConfig();
+    const sysinfo = formatSystemInfo();
+    const plsHistory = formatHistoryForAI();
+    const shellHistory = formatShellHistoryForAI();
+    const shellHookEnabled = config.shellHook && getShellHistory().length > 0;
+    return buildSystemPrompt(sysinfo, plsHistory, shellHistory, shellHookEnabled);
+}
+/**
+ * 使用 Mastra 生成多步骤命令
+ */
+export async function generateMultiStepCommand(userPrompt, previousSteps = [], options = {}) {
+    const agent = createShellAgent();
+    const fullSystemPrompt = getFullSystemPrompt();
+    // 构建消息数组（string[] 格式）
+    const messages = [userPrompt];
+    // 添加之前步骤的执行结果
+    previousSteps.forEach((step) => {
+        messages.push(JSON.stringify({
+            command: step.command,
+            continue: step.continue,
+            reasoning: step.reasoning,
+            nextStepHint: step.nextStepHint,
+        }));
+        messages.push(`命令已执行\n退出码: ${step.exitCode}\n输出:\n${step.output.slice(0, 500)}`);
+    });
+    // 调用 Mastra Agent 生成结构化输出
+    const response = await agent.generate(messages, {
+        structuredOutput: {
+            schema: CommandStepSchema,
+            jsonPromptInjection: true, // 对于不支持 response_format 的模型使用提示词注入
+        },
+    });
+    const stepData = response.object;
+    // 返回调试信息
+    if (options.debug) {
+        return {
+            stepData,
+            debugInfo: {
+                fullPrompt: fullSystemPrompt,
+                userPrompt,
+                previousStepsCount: previousSteps.length,
+                response: stepData,
+            },
+        };
+    }
+    return { stepData };
+}

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

@@ -0,0 +1,35 @@
+/**
+ * 检测当前 shell 类型
+ */
+export function detectShell(): "unknown" | "zsh" | "bash" | "powershell";
+/**
+ * 获取 shell 配置文件路径
+ */
+export function getShellConfigPath(shellType: any): string | null;
+/**
+ * 安装 shell hook
+ */
+export function installShellHook(): Promise<boolean>;
+/**
+ * 卸载 shell hook
+ */
+export function uninstallShellHook(): boolean;
+/**
+ * 读取 shell 历史记录
+ */
+export function getShellHistory(): any[];
+/**
+ * 格式化 shell 历史供 AI 使用
+ * 对于 pls 命令，会从 pls history 中查找对应的详细信息
+ */
+export function formatShellHistoryForAI(): string;
+/**
+ * 获取 hook 状态
+ */
+export function getHookStatus(): {
+    enabled: any;
+    installed: boolean;
+    shellType: string;
+    configPath: string | null;
+    historyFile: string;
+};