@yivan-lab/pretty-please 1.0.0 → 1.2.0

package/dist/src/remote-history.js

@@ -0,0 +1,315 @@
+/**
+ * 远程服务器历史管理模块
+ * 管理每个远程服务器的命令历史
+ */
+import fs from 'fs';
+import path from 'path';
+import chalk from 'chalk';
+import { CONFIG_DIR, getConfig } from './config.js';
+import { getCurrentTheme } from './ui/theme.js';
+import { sshExec, getRemote } from './remote.js';
+// 获取主题颜色
+function getColors() {
+    const theme = getCurrentTheme();
+    return {
+        primary: theme.primary,
+        secondary: theme.secondary,
+        success: theme.success,
+        error: theme.error,
+        warning: theme.warning,
+        muted: theme.text.muted,
+    };
+}
+// 远程服务器数据目录
+const REMOTES_DIR = path.join(CONFIG_DIR, 'remotes');
+// ================== 命令历史管理 ==================
+/**
+ * 获取远程服务器历史文件路径
+ */
+function getRemoteHistoryPath(name) {
+    return path.join(REMOTES_DIR, name, 'history.json');
+}
+/**
+ * 获取远程服务器命令历史
+ */
+export function getRemoteHistory(name) {
+    const historyPath = getRemoteHistoryPath(name);
+    if (!fs.existsSync(historyPath)) {
+        return [];
+    }
+    try {
+        const content = fs.readFileSync(historyPath, 'utf-8');
+        return JSON.parse(content);
+    }
+    catch {
+        return [];
+    }
+}
+/**
+ * 添加远程命令历史记录
+ */
+export function addRemoteHistory(name, record) {
+    const config = getConfig();
+    const historyPath = getRemoteHistoryPath(name);
+    // 确保目录存在
+    const dir = path.dirname(historyPath);
+    if (!fs.existsSync(dir)) {
+        fs.mkdirSync(dir, { recursive: true });
+    }
+    let history = getRemoteHistory(name);
+    // 添加新记录
+    history.push({
+        ...record,
+        timestamp: new Date().toISOString(),
+    });
+    // 限制历史数量
+    const limit = config.commandHistoryLimit || 10;
+    if (history.length > limit) {
+        history = history.slice(-limit);
+    }
+    fs.writeFileSync(historyPath, JSON.stringify(history, null, 2));
+}
+/**
+ * 清空远程命令历史
+ */
+export function clearRemoteHistory(name) {
+    const historyPath = getRemoteHistoryPath(name);
+    if (fs.existsSync(historyPath)) {
+        fs.unlinkSync(historyPath);
+    }
+}
+/**
+ * 格式化远程命令历史供 AI 使用
+ */
+export function formatRemoteHistoryForAI(name) {
+    const history = getRemoteHistory(name);
+    if (history.length === 0) {
+        return '';
+    }
+    const lines = history.map((record, index) => {
+        let status = '';
+        if (record.reason === 'builtin') {
+            status = '(包含 builtin，未执行)';
+        }
+        else if (record.executed) {
+            status = record.exitCode === 0 ? '✓' : `✗ 退出码:${record.exitCode}`;
+        }
+        else {
+            status = '(用户取消执行)';
+        }
+        // 显示用户修改信息
+        if (record.userModified && record.aiGeneratedCommand) {
+            return `${index + 1}. "${record.userPrompt}" → AI 生成: ${record.aiGeneratedCommand} / 用户修改为: ${record.command} ${status}`;
+        }
+        else {
+            return `${index + 1}. "${record.userPrompt}" → ${record.command} ${status}`;
+        }
+    });
+    return `【该服务器最近通过 pls 执行的命令】\n${lines.join('\n')}`;
+}
+/**
+ * 显示远程命令历史
+ */
+export function displayRemoteHistory(name) {
+    const remote = getRemote(name);
+    const history = getRemoteHistory(name);
+    const colors = getColors();
+    if (!remote) {
+        console.log('');
+        console.log(chalk.hex(colors.error)(`✗ 服务器 "${name}" 不存在`));
+        console.log('');
+        return;
+    }
+    console.log('');
+    if (history.length === 0) {
+        console.log(chalk.gray(`  服务器 "${name}" 暂无命令历史`));
+        console.log('');
+        return;
+    }
+    console.log(chalk.bold(`📜 服务器 "${name}" 命令历史:`));
+    console.log(chalk.gray('━'.repeat(50)));
+    history.forEach((item, index) => {
+        const status = item.executed
+            ? item.exitCode === 0
+                ? chalk.hex(colors.success)('✓')
+                : chalk.hex(colors.error)(`✗ 退出码:${item.exitCode}`)
+            : chalk.gray('(未执行)');
+        console.log(`\n${chalk.gray(`${index + 1}.`)} ${chalk.hex(colors.primary)(item.userPrompt)}`);
+        // 显示用户修改信息
+        if (item.userModified && item.aiGeneratedCommand) {
+            console.log(`   ${chalk.dim('AI 生成:')} ${chalk.gray(item.aiGeneratedCommand)}`);
+            console.log(`   ${chalk.dim('用户修改为:')} ${item.command} ${status} ${chalk.hex(colors.warning)('(已修改)')}`);
+        }
+        else {
+            console.log(`   ${chalk.dim('→')} ${item.command} ${status}`);
+        }
+        console.log(`   ${chalk.gray(item.timestamp)}`);
+    });
+    console.log('');
+    console.log(chalk.gray('━'.repeat(50)));
+    console.log(chalk.gray(`历史文件: ${getRemoteHistoryPath(name)}`));
+    console.log('');
+}
+// ================== Shell 历史管理 ==================
+// 远程 shell 历史的本地缓存文件
+function getRemoteShellHistoryPath(name) {
+    return path.join(REMOTES_DIR, name, 'shell_history.jsonl');
+}
+/**
+ * 从远程服务器读取 shell 历史
+ * 读取远程 ~/.please/shell_history.jsonl
+ */
+export async function fetchRemoteShellHistory(name) {
+    const config = getConfig();
+    const limit = config.shellHistoryLimit || 15;
+    try {
+        // 读取远程 shell 历史文件
+        const result = await sshExec(name, `tail -n ${limit} ~/.please/shell_history.jsonl 2>/dev/null || echo ""`, {
+            timeout: 10000,
+        });
+        if (result.exitCode !== 0 || !result.stdout.trim()) {
+            return [];
+        }
+        const lines = result.stdout.trim().split('\n').filter(line => line.trim());
+        const items = [];
+        for (const line of lines) {
+            try {
+                const item = JSON.parse(line);
+                items.push(item);
+            }
+            catch {
+                // 跳过无效行
+            }
+        }
+        // 缓存到本地
+        saveRemoteShellHistoryCache(name, items);
+        return items;
+    }
+    catch {
+        // 如果无法连接，尝试返回缓存
+        return getRemoteShellHistoryCache(name);
+    }
+}
+/**
+ * 保存远程 shell 历史缓存到本地
+ */
+function saveRemoteShellHistoryCache(name, items) {
+    const cachePath = getRemoteShellHistoryPath(name);
+    // 确保目录存在
+    const dir = path.dirname(cachePath);
+    if (!fs.existsSync(dir)) {
+        fs.mkdirSync(dir, { recursive: true });
+    }
+    const content = items.map(item => JSON.stringify(item)).join('\n');
+    fs.writeFileSync(cachePath, content);
+}
+/**
+ * 获取本地缓存的远程 shell 历史
+ */
+function getRemoteShellHistoryCache(name) {
+    const cachePath = getRemoteShellHistoryPath(name);
+    if (!fs.existsSync(cachePath)) {
+        return [];
+    }
+    try {
+        const content = fs.readFileSync(cachePath, '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((item) => item !== null);
+    }
+    catch {
+        return [];
+    }
+}
+/**
+ * 格式化远程 shell 历史供 AI 使用
+ */
+export function formatRemoteShellHistoryForAI(items) {
+    if (items.length === 0) {
+        return '';
+    }
+    const lines = items.map((item, index) => {
+        const status = item.exit === 0 ? '✓' : `✗ 退出码:${item.exit}`;
+        return `${index + 1}. ${item.cmd} ${status}`;
+    });
+    return `【该服务器终端最近执行的命令】\n${lines.join('\n')}`;
+}
+/**
+ * 显示远程 shell 历史
+ */
+export async function displayRemoteShellHistory(name) {
+    const remote = getRemote(name);
+    const colors = getColors();
+    if (!remote) {
+        console.log('');
+        console.log(chalk.hex(colors.error)(`✗ 服务器 "${name}" 不存在`));
+        console.log('');
+        return;
+    }
+    console.log('');
+    console.log(chalk.gray(`正在从 ${name} 读取 shell 历史...`));
+    try {
+        const history = await fetchRemoteShellHistory(name);
+        if (history.length === 0) {
+            console.log('');
+            console.log(chalk.gray(`  服务器 "${name}" 暂无 shell 历史`));
+            console.log(chalk.gray('  请先安装远程 hook: pls remote hook install ' + name));
+            console.log('');
+            return;
+        }
+        console.log('');
+        console.log(chalk.bold(`终端历史 - ${name}（最近 ${history.length} 条）:`));
+        console.log(chalk.gray('━'.repeat(50)));
+        history.forEach((item, index) => {
+            const num = index + 1;
+            const status = item.exit === 0 ? chalk.hex(colors.success)('✓') : chalk.hex(colors.error)(`✗ (${item.exit})`);
+            console.log(`  ${chalk.hex(colors.primary)(num.toString().padStart(2, ' '))}. ${item.cmd} ${status}`);
+        });
+        console.log(chalk.gray('━'.repeat(50)));
+        console.log(chalk.gray(`远程文件: ~/.please/shell_history.jsonl`));
+        console.log('');
+    }
+    catch (error) {
+        const message = error instanceof Error ? error.message : String(error);
+        console.log('');
+        console.log(chalk.hex(colors.error)(`✗ 无法读取远程 shell 历史: ${message}`));
+        console.log('');
+    }
+}
+/**
+ * 清空远程 shell 历史
+ */
+export async function clearRemoteShellHistory(name) {
+    const remote = getRemote(name);
+    const colors = getColors();
+    if (!remote) {
+        console.log('');
+        console.log(chalk.hex(colors.error)(`✗ 服务器 "${name}" 不存在`));
+        console.log('');
+        return;
+    }
+    try {
+        // 清空远程文件
+        await sshExec(name, 'rm -f ~/.please/shell_history.jsonl', { timeout: 10000 });
+        // 清空本地缓存
+        const cachePath = getRemoteShellHistoryPath(name);
+        if (fs.existsSync(cachePath)) {
+            fs.unlinkSync(cachePath);
+        }
+        console.log('');
+        console.log(chalk.hex(colors.success)(`✓ 服务器 "${name}" 的 shell 历史已清空`));
+        console.log('');
+    }
+    catch (error) {
+        const message = error instanceof Error ? error.message : String(error);
+        console.log('');
+        console.log(chalk.hex(colors.error)(`✗ 无法清空远程 shell 历史: ${message}`));
+        console.log('');
+    }
+}

package/dist/src/remote.d.ts

@@ -0,0 +1,113 @@
+/**
+ * 远程执行器模块
+ * 通过 SSH 在远程服务器上执行命令
+ */
+import { type RemoteConfig, type RemoteSysInfo } from './config.js';
+/**
+ * 关闭 ControlMaster 连接
+ */
+export declare function closeControlMaster(name: string): Promise<void>;
+/**
+ * 获取所有远程服务器配置
+ */
+export declare function getRemotes(): Record<string, RemoteConfig>;
+/**
+ * 获取单个远程服务器配置
+ */
+export declare function getRemote(name: string): RemoteConfig | null;
+/**
+ * 添加远程服务器
+ */
+export declare function addRemote(name: string, hostStr: string, options?: {
+    key?: string;
+    password?: boolean;
+}): void;
+/**
+ * 删除远程服务器
+ */
+export declare function removeRemote(name: string): boolean;
+/**
+ * 显示所有远程服务器
+ */
+export declare function displayRemotes(): void;
+/**
+ * 设置远程服务器工作目录
+ */
+export declare function setRemoteWorkDir(name: string, workDir: string): void;
+/**
+ * 获取远程服务器工作目录
+ */
+export declare function getRemoteWorkDir(name: string): string | undefined;
+/**
+ * SSH 执行选项
+ */
+export interface SSHExecOptions {
+    timeout?: number;
+    stdin?: string;
+    onStdout?: (data: string) => void;
+    onStderr?: (data: string) => void;
+}
+/**
+ * SSH 执行结果
+ */
+export interface SSHExecResult {
+    exitCode: number;
+    stdout: string;
+    stderr: string;
+    output: string;
+}
+/**
+ * 通过 SSH 执行命令
+ * 使用 ControlMaster 实现连接复用，密码认证只需输入一次
+ */
+export declare function sshExec(name: string, command: string, options?: SSHExecOptions): Promise<SSHExecResult>;
+/**
+ * 测试远程连接
+ */
+export declare function testRemoteConnection(name: string): Promise<{
+    success: boolean;
+    message: string;
+}>;
+/**
+ * 获取缓存的远程系统信息
+ */
+export declare function getRemoteSysInfo(name: string): RemoteSysInfo | null;
+/**
+ * 采集远程系统信息
+ */
+export declare function collectRemoteSysInfo(name: string, force?: boolean): Promise<RemoteSysInfo>;
+/**
+ * 格式化远程系统信息供 AI 使用
+ */
+export declare function formatRemoteSysInfoForAI(name: string, sysInfo: RemoteSysInfo): string;
+/**
+ * 批量远程执行结果
+ */
+export interface BatchRemoteResult {
+    server: string;
+    command: string;
+    exitCode: number;
+    stdout: string;
+    stderr: string;
+    output: string;
+    sysInfo: RemoteSysInfo;
+}
+/**
+ * 批量远程执行命令
+ * 每个服务器单独生成命令，支持异构环境
+ */
+export declare function generateBatchRemoteCommands(serverNames: string[], userPrompt: string, options?: {
+    debug?: boolean;
+}): Promise<Array<{
+    server: string;
+    command: string;
+    sysInfo: RemoteSysInfo;
+}>>;
+/**
+ * 执行批量远程命令
+ */
+export declare function executeBatchRemoteCommands(commands: Array<{
+    server: string;
+    command: string;
+    sysInfo: RemoteSysInfo;
+}>): Promise<BatchRemoteResult[]>;