@yivan-lab/pretty-please 1.0.0 → 1.2.0

package/bin/pls.tsx

@@ -1,6 +1,4 @@
-#!/usr/bin/env node
-import React from 'react'
-import { render } from 'ink'
+#!/usr/bin/env tsx
 import { Command } from 'commander'
 import { fileURLToPath } from 'url'
 import { dirname, join } from 'path'
@@ -8,12 +6,14 @@ import { exec } from 'child_process'
 import fs from 'fs'
 import os from 'os'
 import chalk from 'chalk'
-import { CommandGenerator } from '../src/components/CommandGenerator.js'
-import { MultiStepCommandGenerator } from '../src/components/MultiStepCommandGenerator.js'
-import { Chat } from '../src/components/Chat.js'
+// React 和 Ink 懒加载（只在需要 UI 时加载）
+// import React from 'react'
+// import { render } from 'ink'
+// import { MultiStepCommandGenerator } from '../src/components/MultiStepCommandGenerator.js'
+// import { Chat } from '../src/components/Chat.js'
 import { isConfigValid, setConfigValue, getConfig, maskApiKey } from '../src/config.js'
 import { clearHistory, addHistory, getHistory, getHistoryFilePath } from '../src/history.js'
-import { clearChatHistory, getChatRoundCount, getChatHistoryFilePath } from '../src/chat-history.js'
+import { clearChatHistory, getChatRoundCount, getChatHistoryFilePath, displayChatHistory } from '../src/chat-history.js'
 import { type ExecutedStep } from '../src/multi-step.js'
 import {
   installShellHook,
@@ -21,55 +21,146 @@ import {
   getHookStatus,
   detectShell,
   getShellConfigPath,
+  displayShellHistory,
+  clearShellHistory,
 } from '../src/shell-hook.js'
+import {
+  checkForUpdates,
+  showUpdateNotice,
+  performUpgrade,
+} from '../src/upgrade.js'
+import { getCurrentTheme } from '../src/ui/theme.js'
+import {
+  addAlias,
+  removeAlias,
+  displayAliases,
+  resolveAlias,
+} from '../src/alias.js'
+import {
+  addRemote,
+  removeRemote,
+  displayRemotes,
+  getRemote,
+  testRemoteConnection,
+  sshExec,
+  collectRemoteSysInfo,
+  setRemoteWorkDir,
+  getRemoteWorkDir,
+  generateBatchRemoteCommands,
+  executeBatchRemoteCommands,
+} from '../src/remote.js'
+import {
+  addRemoteHistory,
+  displayRemoteHistory,
+  clearRemoteHistory,
+  fetchRemoteShellHistory,
+  displayRemoteShellHistory,
+  clearRemoteShellHistory,
+} from '../src/remote-history.js'
+import {
+  detectRemoteShell,
+  getRemoteShellConfigPath,
+  installRemoteShellHook,
+  uninstallRemoteShellHook,
+  getRemoteHookStatus,
+} from '../src/shell-hook.js'
+
+// 获取主题颜色的辅助函数
+function getThemeColors() {
+  const theme = getCurrentTheme()
+  return {
+    primary: theme.primary,
+    success: theme.success,
+    error: theme.error,
+    warning: theme.warning,
+    info: theme.info,
+    muted: theme.text.muted,
+    secondary: theme.text.secondary,
+  }
+}
 import * as console2 from '../src/utils/console.js'
+// 导入 package.json（Bun 会自动打包进二进制）
+import packageJson from '../package.json' with { type: 'json' }
 
-// 获取 package.json 版本
+// 保留这些用于其他可能的用途
 const __filename = fileURLToPath(import.meta.url)
 const __dirname = dirname(__filename)
-const packageJson = JSON.parse(fs.readFileSync(join(__dirname, '../package.json'), 'utf-8'))
 
 const program = new Command()
 
+
+// 启动时异步检查更新（不阻塞主流程）
+let updateCheckResult: { hasUpdate: boolean; latestVersion: string | null } | null = null
+const isUpgradeCommand = process.argv.includes('upgrade')
+
+// 延迟更新检查到命令解析后（减少启动时间）
+// 非 upgrade 命令时才检查更新
+if (!isUpgradeCommand) {
+  // 延迟 100ms 开始检查，避免影响简单命令的响应速度
+  setTimeout(() => {
+    checkForUpdates(packageJson.version)
+      .then((result) => {
+        updateCheckResult = result
+      })
+      .catch(() => {
+        // 静默失败
+      })
+  }, 100)
+}
+
+// 程序退出时显示更新提示
+process.on('beforeExit', () => {
+  if (updateCheckResult?.hasUpdate && updateCheckResult.latestVersion && !isUpgradeCommand) {
+    showUpdateNotice(packageJson.version, updateCheckResult.latestVersion)
+  }
+})
+
 /**
  * 执行命令（原生版本）
  */
-function executeCommand(command: string, prompt: string): Promise<{ exitCode: number; output: string }> {
+function executeCommand(command: string): Promise<{ exitCode: number; output: string; stdout: string; stderr: string }> {
   return new Promise((resolve) => {
-    let output = ''
+    let stdout = ''
+    let stderr = ''
     let hasOutput = false
 
     console.log('') // 空行
-    console2.printSeparator('输出')
 
-    const child = exec(command)
+    // 计算命令框宽度，让分隔线长度一致
+    const lines = command.split('\n')
+    const maxContentWidth = Math.max(...lines.map(l => console2.getDisplayWidth(l)))
+    const boxWidth = Math.max(maxContentWidth + 4, console2.getDisplayWidth('生成命令') + 6, 20)
+    console2.printSeparator('输出', boxWidth)
+
+    // 使用 bash 并启用 pipefail，确保管道中任何命令失败都能正确返回非零退出码
+    const child = exec(`set -o pipefail; ${command}`, { shell: '/bin/bash' })
 
     child.stdout?.on('data', (data) => {
-      output += data
+      stdout += data
       hasOutput = true
       process.stdout.write(data)
     })
 
     child.stderr?.on('data', (data) => {
-      output += data
+      stderr += data
       hasOutput = true
       process.stderr.write(data)
     })
 
     child.on('close', (code) => {
       if (hasOutput) {
-        console2.printSeparator('')
+        console2.printSeparator('', boxWidth)
       }
-      resolve({ exitCode: code || 0, output })
+      resolve({ exitCode: code || 0, output: stdout + stderr, stdout, stderr })
     })
 
     child.on('error', (err) => {
       if (!hasOutput) {
-        console2.printSeparator('')
+        console2.printSeparator('', boxWidth)
       }
       console2.error(err.message)
-      console2.printSeparator('')
-      resolve({ exitCode: 1, output: err.message })
+      console2.printSeparator('', boxWidth)
+      resolve({ exitCode: 1, output: err.message, stdout: '', stderr: err.message })
     })
   })
 }
@@ -80,6 +171,7 @@ program
   .description('AI 驱动的命令行工具，将自然语言转换为可执行的 Shell 命令')
   .version(packageJson.version, '-v, --version', '显示版本号')
   .helpOption('-h, --help', '显示帮助信息')
+  .allowUnknownOption(true)  // 允许未知选项（用于别名参数传递）
 
 // config 子命令
 const configCmd = program.command('config').description('管理配置')
@@ -94,18 +186,30 @@ configCmd
 
     console.log('')
     console2.title('当前配置:')
-    console2.muted('━'.repeat(40))
-    console.log(`  ${chalk.hex('#00D9FF')('apiKey')}:           ${maskApiKey(config.apiKey)}`)
-    console.log(`  ${chalk.hex('#00D9FF')('baseUrl')}:          ${config.baseUrl}`)
-    console.log(`  ${chalk.hex('#00D9FF')('provider')}:         ${config.provider}`)
-    console.log(`  ${chalk.hex('#00D9FF')('model')}:            ${config.model}`)
+    console2.muted('━'.repeat(50))
+    console.log(`  ${chalk.hex(getThemeColors().primary)('apiKey')}:              ${maskApiKey(config.apiKey)}`)
+    console.log(`  ${chalk.hex(getThemeColors().primary)('baseUrl')}:             ${config.baseUrl}`)
+    console.log(`  ${chalk.hex(getThemeColors().primary)('provider')}:            ${config.provider}`)
+    console.log(`  ${chalk.hex(getThemeColors().primary)('model')}:               ${config.model}`)
     console.log(
-      `  ${chalk.hex('#00D9FF')('shellHook')}:        ${
-        config.shellHook ? chalk.hex('#10B981')('已启用') : chalk.gray('未启用')
+      `  ${chalk.hex(getThemeColors().primary)('shellHook')}:           ${
+        config.shellHook ? chalk.hex(getThemeColors().success)('已启用') : chalk.gray('未启用')
       }`
     )
-    console.log(`  ${chalk.hex('#00D9FF')('chatHistoryLimit')}: ${config.chatHistoryLimit} 轮`)
-    console2.muted('━'.repeat(40))
+    console.log(
+      `  ${chalk.hex(getThemeColors().primary)('editMode')}:            ${
+        config.editMode === 'auto' ? chalk.hex(getThemeColors().primary)('auto (自动编辑)') : chalk.gray('manual (按E编辑)')
+      }`
+    )
+    console.log(`  ${chalk.hex(getThemeColors().primary)('chatHistoryLimit')}:    ${config.chatHistoryLimit} 轮`)
+    console.log(`  ${chalk.hex(getThemeColors().primary)('commandHistoryLimit')}: ${config.commandHistoryLimit} 条`)
+    console.log(`  ${chalk.hex(getThemeColors().primary)('shellHistoryLimit')}:   ${config.shellHistoryLimit} 条`)
+    console.log(
+      `  ${chalk.hex(getThemeColors().primary)('theme')}:               ${
+        config.theme === 'dark' ? chalk.hex(getThemeColors().primary)('dark (深色)') : chalk.hex(getThemeColors().primary)('light (浅色)')
+      }`
+    )
+    console2.muted('━'.repeat(50))
     console2.muted(`配置文件: ${CONFIG_FILE}`)
     console.log('')
   })
@@ -133,6 +237,75 @@ configCmd.action(async () => {
   await runConfigWizard()
 })
 
+// theme 子命令
+const themeCmd = program.command('theme').description('管理主题')
+
+themeCmd
+  .command('list')
+  .description('查看所有可用主题')
+  .action(async () => {
+    const { themes } = await import('../src/ui/theme.js')
+    const config = getConfig()
+    const currentTheme = config.theme || 'dark'
+
+    console.log('')
+    console2.title('🎨 可用主题:')
+    console2.muted('━'.repeat(50))
+
+    Object.keys(themes).forEach((themeName) => {
+      const isCurrent = themeName === currentTheme
+      const prefix = isCurrent ? '●' : '○'
+      const label = themeName === 'dark' ? 'dark (深色)' : 'light (浅色)'
+      const color = themeName === 'dark' ? '#00D9FF' : '#0284C7'
+
+      if (isCurrent) {
+        console.log(`  ${chalk.hex(color)(prefix)} ${chalk.hex(color).bold(label)} ${chalk.gray('(当前)')}`)
+      } else {
+        console.log(`  ${chalk.gray(prefix)} ${label}`)
+      }
+    })
+
+    console2.muted('━'.repeat(50))
+    console.log('')
+  })
+
+themeCmd
+  .argument('[name]', '主题名称 (dark, light)')
+  .description('切换主题')
+  .action((name?: string) => {
+    if (!name) {
+      // 显示当前主题
+      const config = getConfig()
+      const currentTheme = config.theme || 'dark'
+      const label = currentTheme === 'dark' ? 'dark (深色)' : 'light (浅色)'
+      const color = currentTheme === 'dark' ? '#00D9FF' : '#0284C7'
+
+      console.log('')
+      console.log(`当前主题: ${chalk.hex(color).bold(label)}`)
+      console.log('')
+      console2.muted('使用 pls theme list 查看所有主题')
+      console2.muted('使用 pls theme <name> 切换主题')
+      console.log('')
+      return
+    }
+
+    // 切换主题
+    try {
+      setConfigValue('theme', name)
+      const label = name === 'dark' ? 'dark (深色)' : 'light (浅色)'
+      const color = name === 'dark' ? '#00D9FF' : '#0284C7'
+
+      console.log('')
+      console2.success(`已切换到 ${chalk.hex(color).bold(label)} 主题`)
+      console.log('')
+    } catch (error: any) {
+      console.log('')
+      console2.error(error.message)
+      console.log('')
+      process.exit(1)
+    }
+  })
+
 // history 子命令
 const historyCmd = program.command('history').description('查看或管理命令历史')
 
@@ -156,12 +329,20 @@ historyCmd
     history.forEach((item: any, index: number) => {
       const status = item.executed
         ? item.exitCode === 0
-          ? chalk.hex('#10B981')('✓')
-          : chalk.hex('#EF4444')(`✗ 退出码:${item.exitCode}`)
+          ? chalk.hex(getThemeColors().success)('✓')
+          : chalk.hex(getThemeColors().error)(`✗ 退出码:${item.exitCode}`)
         : chalk.gray('(未执行)')
 
-      console.log(`\n${chalk.gray(`${index + 1}.`)} ${chalk.hex('#00D9FF')(item.userPrompt)}`)
-      console.log(`   ${chalk.dim('→')} ${item.command} ${status}`)
+      console.log(`\n${chalk.gray(`${index + 1}.`)} ${chalk.hex(getThemeColors().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(getThemeColors().warning)('(已修改)')}`)
+      } else {
+        console.log(`   ${chalk.dim('→')} ${item.command} ${status}`)
+      }
+
       console.log(`   ${chalk.gray(item.timestamp)}`)
     })
 
@@ -180,7 +361,38 @@ historyCmd
     console.log('')
   })
 
-// 默认 history 命令（显示历史）
+// history chat 子命令
+const historyChatCmd = historyCmd.command('chat').description('查看或管理对话历史')
+
+historyChatCmd.action(() => {
+  displayChatHistory()
+})
+
+historyChatCmd
+  .command('clear')
+  .description('清空对话历史')
+  .action(() => {
+    clearChatHistory()
+    console.log('')
+    console2.success('对话历史已清空')
+    console.log('')
+  })
+
+// history shell 子命令
+const historyShellCmd = historyCmd.command('shell').description('查看或管理 Shell 历史')
+
+historyShellCmd.action(() => {
+  displayShellHistory()
+})
+
+historyShellCmd
+  .command('clear')
+  .description('清空 Shell 历史')
+  .action(() => {
+    clearShellHistory()
+  })
+
+// 默认 history 命令（显示命令历史）
 historyCmd.action(() => {
   const history = getHistory()
 
@@ -198,12 +410,20 @@ historyCmd.action(() => {
   history.forEach((item: any, index: number) => {
     const status = item.executed
       ? item.exitCode === 0
-        ? chalk.hex('#10B981')('✓')
-        : chalk.hex('#EF4444')(`✗ 退出码:${item.exitCode}`)
+        ? chalk.hex(getThemeColors().success)('✓')
+        : chalk.hex(getThemeColors().error)(`✗ 退出码:${item.exitCode}`)
       : chalk.gray('(未执行)')
 
-    console.log(`\n${chalk.gray(`${index + 1}.`)} ${chalk.hex('#00D9FF')(item.userPrompt)}`)
-    console.log(`   ${chalk.dim('→')} ${item.command} ${status}`)
+    console.log(`\n${chalk.gray(`${index + 1}.`)} ${chalk.hex(getThemeColors().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(getThemeColors().warning)('(已修改)')}`)
+    } else {
+      console.log(`   ${chalk.dim('→')} ${item.command} ${status}`)
+    }
+
     console.log(`   ${chalk.gray(item.timestamp)}`)
   })
 
@@ -268,19 +488,19 @@ hookCmd
     console.log('')
     console2.title('📊 Shell Hook 状态')
     console2.muted('━'.repeat(40))
-    console.log(`  ${chalk.hex('#00D9FF')('Shell 类型')}: ${status.shellType}`)
-    console.log(`  ${chalk.hex('#00D9FF')('配置文件')}:   ${status.configPath || '未知'}`)
+    console.log(`  ${chalk.hex(getThemeColors().primary)('Shell 类型')}: ${status.shellType}`)
+    console.log(`  ${chalk.hex(getThemeColors().primary)('配置文件')}:   ${status.configPath || '未知'}`)
     console.log(
-      `  ${chalk.hex('#00D9FF')('已安装')}:     ${
-        status.installed ? chalk.hex('#10B981')('是') : chalk.gray('否')
+      `  ${chalk.hex(getThemeColors().primary)('已安装')}:     ${
+        status.installed ? chalk.hex(getThemeColors().success)('是') : chalk.gray('否')
       }`
     )
     console.log(
-      `  ${chalk.hex('#00D9FF')('已启用')}:     ${
-        status.enabled ? chalk.hex('#10B981')('是') : chalk.gray('否')
+      `  ${chalk.hex(getThemeColors().primary)('已启用')}:     ${
+        status.enabled ? chalk.hex(getThemeColors().success)('是') : chalk.gray('否')
       }`
     )
-    console.log(`  ${chalk.hex('#00D9FF')('历史文件')}:   ${status.historyFile}`)
+    console.log(`  ${chalk.hex(getThemeColors().primary)('历史文件')}:   ${status.historyFile}`)
     console2.muted('━'.repeat(40))
 
     if (!status.installed) {
@@ -297,19 +517,19 @@ hookCmd.action(() => {
   console.log('')
   console2.title('📊 Shell Hook 状态')
   console2.muted('━'.repeat(40))
-  console.log(`  ${chalk.hex('#00D9FF')('Shell 类型')}: ${status.shellType}`)
-  console.log(`  ${chalk.hex('#00D9FF')('配置文件')}:   ${status.configPath || '未知'}`)
+  console.log(`  ${chalk.hex(getThemeColors().primary)('Shell 类型')}: ${status.shellType}`)
+  console.log(`  ${chalk.hex(getThemeColors().primary)('配置文件')}:   ${status.configPath || '未知'}`)
   console.log(
-    `  ${chalk.hex('#00D9FF')('已安装')}:     ${
-      status.installed ? chalk.hex('#10B981')('是') : chalk.gray('否')
+    `  ${chalk.hex(getThemeColors().primary)('已安装')}:     ${
+      status.installed ? chalk.hex(getThemeColors().success)('是') : chalk.gray('否')
     }`
   )
   console.log(
-    `  ${chalk.hex('#00D9FF')('已启用')}:     ${
-      status.enabled ? chalk.hex('#10B981')('是') : chalk.gray('否')
+    `  ${chalk.hex(getThemeColors().primary)('已启用')}:     ${
+      status.enabled ? chalk.hex(getThemeColors().success)('是') : chalk.gray('否')
     }`
   )
-  console.log(`  ${chalk.hex('#00D9FF')('历史文件')}:   ${status.historyFile}`)
+  console.log(`  ${chalk.hex(getThemeColors().primary)('历史文件')}:   ${status.historyFile}`)
   console2.muted('━'.repeat(40))
 
   if (!status.installed) {
@@ -319,6 +539,429 @@ hookCmd.action(() => {
   console.log('')
 })
 
+// upgrade 子命令
+program
+  .command('upgrade')
+  .description('升级到最新版本')
+  .action(async () => {
+    const success = await performUpgrade(packageJson.version)
+    process.exit(success ? 0 : 1)
+  })
+
+// alias 子命令
+const aliasCmd = program.command('alias').description('管理命令别名')
+
+// 获取所有子命令名称（用于检测冲突）
+function getReservedCommands(): string[] {
+  return program.commands.map((cmd) => cmd.name())
+}
+
+aliasCmd
+  .command('list')
+  .description('列出所有别名')
+  .action(() => {
+    displayAliases()
+  })
+
+aliasCmd
+  .command('add <name> <prompt>')
+  .description('添加别名（prompt 支持 {{param}} 或 {{param:default}} 参数模板）')
+  .option('-d, --description <desc>', '别名描述')
+  .action((name, prompt, options) => {
+    try {
+      addAlias(name, prompt, options.description, getReservedCommands())
+      console.log('')
+      console2.success(`已添加别名: ${name}`)
+      console.log(`  ${chalk.gray('→')} ${prompt}`)
+      console.log('')
+    } catch (error: any) {
+      console.log('')
+      console2.error(error.message)
+      console.log('')
+      process.exit(1)
+    }
+  })
+
+aliasCmd
+  .command('remove <name>')
+  .description('删除别名')
+  .action((name) => {
+    const removed = removeAlias(name)
+    console.log('')
+    if (removed) {
+      console2.success(`已删除别名: ${name}`)
+    } else {
+      console2.error(`别名不存在: ${name}`)
+    }
+    console.log('')
+  })
+
+// 默认 alias 命令（显示列表）
+aliasCmd.action(() => {
+  displayAliases()
+})
+
+// remote 子命令
+const remoteCmd = program.command('remote').description('管理远程服务器')
+
+remoteCmd
+  .command('list')
+  .description('列出所有远程服务器')
+  .action(() => {
+    displayRemotes()
+  })
+
+remoteCmd
+  .command('add <name> <host>')
+  .description('添加远程服务器（格式: user@host 或 user@host:port）')
+  .option('-k, --key <path>', 'SSH 私钥路径')
+  .option('-p, --password', '使用密码认证（每次执行时输入）')
+  .action((name, host, options) => {
+    try {
+      addRemote(name, host, { key: options.key, password: options.password })
+      console.log('')
+      console2.success(`已添加远程服务器: ${name}`)
+      console.log(`  ${chalk.gray('→')} ${host}`)
+      if (options.key) {
+        console.log(`  ${chalk.gray('密钥:')} ${options.key}`)
+      }
+      if (options.password) {
+        console.log(`  ${chalk.gray('认证:')} 密码（每次执行时输入）`)
+      }
+      console.log('')
+    } catch (error: any) {
+      console.log('')
+      console2.error(error.message)
+      console.log('')
+      process.exit(1)
+    }
+  })
+
+remoteCmd
+  .command('remove <name>')
+  .description('删除远程服务器')
+  .action((name) => {
+    const removed = removeRemote(name)
+    console.log('')
+    if (removed) {
+      console2.success(`已删除远程服务器: ${name}`)
+    } else {
+      console2.error(`远程服务器不存在: ${name}`)
+    }
+    console.log('')
+  })
+
+remoteCmd
+  .command('test <name>')
+  .description('测试远程服务器连接')
+  .action(async (name) => {
+    const remote = getRemote(name)
+    if (!remote) {
+      console.log('')
+      console2.error(`远程服务器不存在: ${name}`)
+      console.log('')
+      process.exit(1)
+    }
+
+    console.log('')
+    console2.info(`正在测试连接 ${name} (${remote.user}@${remote.host}:${remote.port})...`)
+
+    const result = await testRemoteConnection(name)
+    console.log(`  ${result.message}`)
+
+    if (result.success) {
+      // 采集系统信息
+      console2.info('正在采集系统信息...')
+      try {
+        const sysInfo = await collectRemoteSysInfo(name, true)
+        console.log(`  ${chalk.gray('系统:')} ${sysInfo.os} ${sysInfo.osVersion}`)
+        console.log(`  ${chalk.gray('Shell:')} ${sysInfo.shell}`)
+        console.log(`  ${chalk.gray('主机名:')} ${sysInfo.hostname}`)
+      } catch (error: any) {
+        console2.warning(`无法采集系统信息: ${error.message}`)
+      }
+    }
+    console.log('')
+  })
+
+// remote hook 子命令
+const remoteHookCmd = remoteCmd.command('hook').description('管理远程服务器 Shell Hook')
+
+remoteHookCmd
+  .command('install <name>')
+  .description('在远程服务器安装 Shell Hook')
+  .action(async (name) => {
+    const remote = getRemote(name)
+    if (!remote) {
+      console.log('')
+      console2.error(`远程服务器不存在: ${name}`)
+      console.log('')
+      process.exit(1)
+    }
+
+    console.log('')
+    console2.title('🔧 远程 Shell Hook 安装')
+    console2.muted('━'.repeat(40))
+    console2.info(`目标服务器: ${name} (${remote.user}@${remote.host})`)
+
+    try {
+      // 检测远程 shell 类型
+      const sshExecFn = async (cmd: string) => {
+        const result = await sshExec(name, cmd, { timeout: 30000 })
+        return { stdout: result.stdout, exitCode: result.exitCode }
+      }
+
+      const shellType = await detectRemoteShell(sshExecFn)
+      const configPath = getRemoteShellConfigPath(shellType)
+      console2.muted(`检测到 Shell: ${shellType}`)
+      console2.muted(`配置文件: ${configPath}`)
+      console.log('')
+
+      const result = await installRemoteShellHook(sshExecFn, shellType)
+      console.log(`  ${result.message}`)
+
+      if (result.success) {
+        console.log('')
+        console2.warning('⚠️  请在远程服务器重启终端或执行:')
+        console2.info(`   source ${configPath}`)
+      }
+    } catch (error: any) {
+      console2.error(`安装失败: ${error.message}`)
+    }
+    console.log('')
+  })
+
+remoteHookCmd
+  .command('uninstall <name>')
+  .description('从远程服务器卸载 Shell Hook')
+  .action(async (name) => {
+    const remote = getRemote(name)
+    if (!remote) {
+      console.log('')
+      console2.error(`远程服务器不存在: ${name}`)
+      console.log('')
+      process.exit(1)
+    }
+
+    console.log('')
+    console2.info(`正在从 ${name} 卸载 Shell Hook...`)
+
+    try {
+      const sshExecFn = async (cmd: string) => {
+        const result = await sshExec(name, cmd, { timeout: 30000 })
+        return { stdout: result.stdout, exitCode: result.exitCode }
+      }
+
+      const shellType = await detectRemoteShell(sshExecFn)
+      const result = await uninstallRemoteShellHook(sshExecFn, shellType)
+      console.log(`  ${result.message}`)
+
+      if (result.success) {
+        console.log('')
+        console2.warning('⚠️  请在远程服务器重启终端使其生效')
+      }
+    } catch (error: any) {
+      console2.error(`卸载失败: ${error.message}`)
+    }
+    console.log('')
+  })
+
+remoteHookCmd
+  .command('status <name>')
+  .description('查看远程服务器 Shell Hook 状态')
+  .action(async (name) => {
+    const remote = getRemote(name)
+    if (!remote) {
+      console.log('')
+      console2.error(`远程服务器不存在: ${name}`)
+      console.log('')
+      process.exit(1)
+    }
+
+    console.log('')
+    console2.info(`正在检查 ${name} 的 Hook 状态...`)
+
+    try {
+      const sshExecFn = async (cmd: string) => {
+        const result = await sshExec(name, cmd, { timeout: 30000 })
+        return { stdout: result.stdout, exitCode: result.exitCode }
+      }
+
+      const status = await getRemoteHookStatus(sshExecFn)
+
+      console.log('')
+      console2.title(`📊 远程 Shell Hook 状态 - ${name}`)
+      console2.muted('━'.repeat(40))
+      console.log(`  ${chalk.hex(getThemeColors().primary)('Shell 类型')}: ${status.shellType}`)
+      console.log(`  ${chalk.hex(getThemeColors().primary)('配置文件')}:   ${status.configPath}`)
+      console.log(
+        `  ${chalk.hex(getThemeColors().primary)('已安装')}:     ${
+          status.installed ? chalk.hex(getThemeColors().success)('是') : chalk.gray('否')
+        }`
+      )
+      console2.muted('━'.repeat(40))
+
+      if (!status.installed) {
+        console.log('')
+        console2.muted(`提示: 运行 pls remote hook install ${name} 安装 Shell Hook`)
+      }
+    } catch (error: any) {
+      console2.error(`检查失败: ${error.message}`)
+    }
+    console.log('')
+  })
+
+// remote history 子命令
+const remoteHistoryCmd = remoteCmd.command('history').description('管理远程服务器历史记录')
+
+remoteHistoryCmd
+  .command('show <name>')
+  .description('显示远程服务器命令历史')
+  .action((name) => {
+    displayRemoteHistory(name)
+  })
+
+remoteHistoryCmd
+  .command('clear <name>')
+  .description('清空远程服务器命令历史')
+  .action((name) => {
+    clearRemoteHistory(name)
+    console.log('')
+    console2.success(`已清空服务器 "${name}" 的命令历史`)
+    console.log('')
+  })
+
+remoteHistoryCmd
+  .command('shell <name>')
+  .description('显示远程服务器 Shell 历史')
+  .action(async (name) => {
+    await displayRemoteShellHistory(name)
+  })
+
+remoteHistoryCmd
+  .command('shell-clear <name>')
+  .description('清空远程服务器 Shell 历史')
+  .action(async (name) => {
+    await clearRemoteShellHistory(name)
+  })
+
+// remote default 子命令
+remoteCmd
+  .command('default [name]')
+  .description('设置或查看默认远程服务器')
+  .option('-c, --clear', '清除默认服务器设置')
+  .action((name?: string, options?: { clear?: boolean }) => {
+    const config = getConfig()
+
+    // 清除默认
+    if (options?.clear) {
+      if (config.defaultRemote) {
+        setConfigValue('defaultRemote', '')
+        console.log('')
+        console2.success('已清除默认远程服务器')
+        console.log('')
+      } else {
+        console.log('')
+        console2.muted('当前没有设置默认远程服务器')
+        console.log('')
+      }
+      return
+    }
+
+    // 查看默认
+    if (!name) {
+      console.log('')
+      if (config.defaultRemote) {
+        const remote = getRemote(config.defaultRemote)
+        if (remote) {
+          console.log(`默认远程服务器: ${chalk.hex(getThemeColors().primary)(config.defaultRemote)}`)
+          console.log(`  ${chalk.gray('→')} ${remote.user}@${remote.host}:${remote.port}`)
+        } else {
+          console2.warning(`默认服务器 "${config.defaultRemote}" 不存在，建议清除设置`)
+          console2.muted('运行 pls remote default --clear 清除')
+        }
+      } else {
+        console2.muted('当前没有设置默认远程服务器')
+        console2.muted('使用 pls remote default <name> 设置默认服务器')
+      }
+      console.log('')
+      return
+    }
+
+    // 设置默认
+    const remote = getRemote(name)
+    if (!remote) {
+      console.log('')
+      console2.error(`远程服务器不存在: ${name}`)
+      console2.muted('使用 pls remote list 查看所有服务器')
+      console.log('')
+      process.exit(1)
+    }
+
+    setConfigValue('defaultRemote', name)
+    console.log('')
+    console2.success(`已设置默认远程服务器: ${name}`)
+    console.log(`  ${chalk.gray('→')} ${remote.user}@${remote.host}:${remote.port}`)
+    console2.muted('现在可以使用 pls -r <prompt> 直接在该服务器执行')
+    console.log('')
+  })
+
+// remote workdir 子命令
+remoteCmd
+  .command('workdir <name> [path]')
+  .description('设置或查看远程服务器的工作目录')
+  .option('-c, --clear', '清除工作目录设置')
+  .action((name: string, workdirPath?: string, options?: { clear?: boolean }) => {
+    const remote = getRemote(name)
+    if (!remote) {
+      console.log('')
+      console2.error(`远程服务器不存在: ${name}`)
+      console.log('')
+      process.exit(1)
+    }
+
+    // 清除工作目录
+    if (options?.clear) {
+      if (remote.workDir) {
+        setRemoteWorkDir(name, '-')
+        console.log('')
+        console2.success(`已清除 ${name} 的工作目录设置`)
+        console.log('')
+      } else {
+        console.log('')
+        console2.muted(`${name} 没有设置工作目录`)
+        console.log('')
+      }
+      return
+    }
+
+    // 查看工作目录
+    if (!workdirPath) {
+      console.log('')
+      if (remote.workDir) {
+        console.log(`${chalk.hex(getThemeColors().primary)(name)} 的工作目录:`)
+        console.log(`  ${chalk.gray('→')} ${remote.workDir}`)
+      } else {
+        console2.muted(`${name} 没有设置工作目录`)
+        console2.muted(`使用 pls remote workdir ${name} <path> 设置工作目录`)
+      }
+      console.log('')
+      return
+    }
+
+    // 设置工作目录
+    setRemoteWorkDir(name, workdirPath)
+    console.log('')
+    console2.success(`已设置 ${name} 的工作目录: ${workdirPath}`)
+    console2.muted('现在在该服务器执行的命令会自动切换到此目录')
+    console.log('')
+  })
+
+// 默认 remote 命令（显示列表）
+remoteCmd.action(() => {
+  displayRemotes()
+})
+
 // chat 子命令
 const chatCmd = program.command('chat').description('AI 对话模式，问答、讲解命令')
 
@@ -347,8 +990,8 @@ chatCmd
       console.log('')
       console2.title('💬 AI 对话模式')
       console2.muted('━'.repeat(40))
-      console.log(`  ${chalk.hex('#00D9FF')('当前对话轮数')}: ${roundCount}`)
-      console.log(`  ${chalk.hex('#00D9FF')('历史文件')}:     ${historyFile}`)
+      console.log(`  ${chalk.hex(getThemeColors().primary)('当前对话轮数')}: ${roundCount}`)
+      console.log(`  ${chalk.hex(getThemeColors().primary)('历史文件')}:     ${historyFile}`)
       console2.muted('━'.repeat(40))
       console.log('')
       console2.muted('用法:')
@@ -367,28 +1010,42 @@ chatCmd
       process.exit(1)
     }
 
-    // 使用 Ink 渲染对话（Chat 适合用 Ink 流式输出）
-    render(
-      <Chat
-        prompt={prompt}
-        debug={options.debug}
-        showRoundCount={true}
-        onComplete={() => process.exit(0)}
-      />
-    )
+    // 懒加载 Chat 组件（避免启动时加载 React/Ink）
+    ;(async () => {
+      const React = await import('react')
+      const { render } = await import('ink')
+      const { Chat } = await import('../src/components/Chat.js')
+
+      render(
+        React.createElement(Chat, {
+          prompt,
+          debug: options.debug,
+          showRoundCount: true,
+          onComplete: () => process.exit(0),
+        })
+      )
+    })()
   })
 
 // 默认命令（执行 prompt）
 program
   .argument('[prompt...]', '自然语言描述你想执行的操作')
   .option('-d, --debug', '显示调试信息（系统信息、完整 prompt 等）')
+  .option('-r, --remote [name]', '在远程服务器上执行（不指定则使用默认服务器）')
   .action((promptArgs, options) => {
+    // 智能处理 -r 参数：如果 -r 后面的值不是已注册的服务器名，把它当作 prompt 的一部分
+    if (typeof options.remote === 'string' && !getRemote(options.remote)) {
+      // "查看当前目录" 不是服务器名，放回 prompt
+      promptArgs.unshift(options.remote)
+      options.remote = true  // 改为使用默认服务器
+    }
+
     if (promptArgs.length === 0) {
       program.help()
       return
     }
 
-    const prompt = promptArgs.join(' ')
+    let prompt = promptArgs.join(' ')
 
     if (!prompt.trim()) {
       console.log('')
@@ -398,6 +1055,23 @@ program
       process.exit(1)
     }
 
+    // 尝试解析别名（支持 pls disk 和 pls @disk 两种格式）
+    try {
+      const aliasResult = resolveAlias(prompt)
+      if (aliasResult.resolved) {
+        prompt = aliasResult.prompt
+        if (options.debug) {
+          console.log('')
+          console2.muted(`别名解析: ${aliasResult.aliasName} → ${prompt}`)
+        }
+      }
+    } catch (error: any) {
+      console.log('')
+      console2.error(error.message)
+      console.log('')
+      process.exit(1)
+    }
+
     // 检查配置
     if (!isConfigValid()) {
       console.log('')
@@ -407,26 +1081,215 @@ program
       process.exit(1)
     }
 
-    // 使用多步骤命令生成器（统一处理单步和多步）
+    // 解析远程服务器名称
+    // options.remote 可能是：
+    // - undefined: 没有使用 -r
+    // - true: 使用了 -r 但没有指定名称（使用默认）
+    // - string: 使用了 -r 并指定了名称（支持逗号分隔的多个服务器）
+    let remoteName: string | undefined
+    let remoteNames: string[] | undefined  // 批量执行时的服务器列表
+    if (options.remote !== undefined) {
+      if (options.remote === true) {
+        // 使用默认服务器
+        const config = getConfig()
+        if (!config.defaultRemote) {
+          console.log('')
+          console2.error('未设置默认远程服务器')
+          console2.muted('使用 pls remote default <name> 设置默认服务器')
+          console2.muted('或使用 pls -r <name> <prompt> 指定服务器')
+          console.log('')
+          process.exit(1)
+        }
+        remoteName = config.defaultRemote
+      } else {
+        // 检查是否为批量执行（逗号分隔的服务器名）
+        if (options.remote.includes(',')) {
+          remoteNames = options.remote.split(',').map(s => s.trim()).filter(s => s.length > 0)
+
+          // 验证所有服务器是否存在
+          const invalidServers = remoteNames!.filter(name => !getRemote(name))
+          if (invalidServers.length > 0) {
+            console.log('')
+            console2.error(`以下服务器不存在: ${invalidServers.join(', ')}`)
+            console2.muted('使用 pls remote list 查看所有服务器')
+            console2.muted('使用 pls remote add <name> <user@host> 添加服务器')
+            console.log('')
+            process.exit(1)
+          }
+        } else {
+          remoteName = options.remote
+
+          // 检查服务器是否存在
+          const remote = getRemote(remoteName!)
+          if (!remote) {
+            console.log('')
+            console2.error(`远程服务器不存在: ${remoteName}`)
+            console2.muted('使用 pls remote add <name> <user@host> 添加服务器')
+            console.log('')
+            process.exit(1)
+          }
+        }
+      }
+    }
+
+    // 懒加载 MultiStepCommandGenerator 组件（避免启动时加载 React/Ink）
     ;(async () => {
+      // 批量远程执行模式
+      if (remoteNames && remoteNames.length > 0) {
+        console.log('')
+        console2.info(`正在为 ${remoteNames.length} 台服务器生成命令...`)
+        console.log('')
+
+        try {
+          // 1. 并发生成命令
+          const commands = await generateBatchRemoteCommands(remoteNames, prompt, { debug: options.debug })
+
+          // 2. 显示生成的命令
+          console2.success('✓ 命令生成完成\n')
+          const theme = getCurrentTheme()
+          commands.forEach(({ server, command, sysInfo }) => {
+            console.log(chalk.hex(theme.primary)(`${server}`) + chalk.gray(` (${sysInfo.os}):`))
+            console.log(chalk.hex(theme.secondary)(`  ${command}`))
+          })
+          console.log('')
+
+          // 3. 询问用户确认
+          const readline = await import('readline')
+          const rl = readline.createInterface({
+            input: process.stdin,
+            output: process.stdout,
+          })
+
+          const confirmed = await new Promise<boolean>((resolve) => {
+            console.log(chalk.gray(`将在 ${remoteNames!.length} 台服务器执行以上命令`))
+            rl.question(chalk.gray('执行？ [回车执行 / Ctrl+C 取消] '), (answer) => {
+              rl.close()
+              resolve(true)
+            })
+          })
+
+          if (!confirmed) {
+            console.log('')
+            console2.muted('已取消执行')
+            console.log('')
+            process.exit(0)
+          }
+
+          // 4. 并发执行
+          console.log('')
+          console2.info('正在执行...')
+          const results = await executeBatchRemoteCommands(commands)
+
+          // 5. 显示执行结果摘要
+          console.log('')
+          console2.info('执行完成:\n')
+          results.forEach(({ server, exitCode }) => {
+            const icon = exitCode === 0 ? '✓' : '✗'
+            const color = exitCode === 0 ? theme.success : theme.error
+            console.log(`  ${chalk.hex(color)(icon)} ${server} ${chalk.gray(`(退出码: ${exitCode})`)}`)
+          })
+
+          // 6. 显示每个服务器的详细输出
+          console.log('')
+          results.forEach(({ server, output }) => {
+            console.log(chalk.hex(theme.primary)(`─── ${server} ───`))
+            console.log(output || chalk.gray('(无输出)'))
+          })
+
+          // 7. 记录到历史
+          results.forEach(({ server, command, exitCode, output }) => {
+            addRemoteHistory(server, {
+              userPrompt: prompt,
+              command,
+              aiGeneratedCommand: command,  // 批量执行无编辑功能
+              userModified: false,
+              executed: true,
+              exitCode,
+              output,
+            })
+          })
+
+          // 8. 根据结果决定退出码
+          const allSuccess = results.every(r => r.exitCode === 0)
+          const allFailed = results.every(r => r.exitCode !== 0)
+          if (allFailed) {
+            process.exit(2)  // 全部失败
+          } else if (!allSuccess) {
+            process.exit(1)  // 部分失败
+          }
+          process.exit(0)  // 全部成功
+        } catch (error: any) {
+          console.log('')
+          console2.error(`批量执行失败: ${error.message}`)
+          console.log('')
+          process.exit(1)
+        }
+        return
+      }
+
+      // 单服务器执行模式
+      const React = await import('react')
+      const { render } = await import('ink')
+      const { MultiStepCommandGenerator } = await import('../src/components/MultiStepCommandGenerator.js')
+
+      // 如果是远程模式，先获取远程上下文
+      let remoteContext: {
+        name: string
+        sysInfo: Awaited<ReturnType<typeof collectRemoteSysInfo>>
+        shellHistory: Awaited<ReturnType<typeof fetchRemoteShellHistory>>
+      } | null = null
+
+      if (remoteName) {
+        console.log('')
+        console2.info(`正在连接远程服务器 ${remoteName}...`)
+
+        try {
+          // 采集系统信息（使用缓存）
+          const sysInfo = await collectRemoteSysInfo(remoteName)
+          if (options.debug) {
+            console2.muted(`系统: ${sysInfo.os} ${sysInfo.osVersion} (${sysInfo.shell})`)
+          }
+
+          // 获取远程 shell 历史
+          const shellHistory = await fetchRemoteShellHistory(remoteName)
+          if (options.debug && shellHistory.length > 0) {
+            console2.muted(`Shell 历史: ${shellHistory.length} 条`)
+          }
+
+          remoteContext = { name: remoteName, sysInfo, shellHistory }
+          console2.success(`已连接到 ${remoteName}`)
+        } catch (error: any) {
+          console2.error(`无法连接到 ${remoteName}: ${error.message}`)
+          console.log('')
+          process.exit(1)
+        }
+      }
+
       const executedSteps: ExecutedStep[] = []
       let currentStepNumber = 1
+      let lastStepFailed = false // 跟踪上一步是否失败
 
       while (true) {
         let stepResult: any = null
 
         // 使用 Ink 渲染命令生成
         const { waitUntilExit, unmount } = render(
-          <MultiStepCommandGenerator
-            prompt={prompt}
-            debug={options.debug}
-            previousSteps={executedSteps}
-            currentStepNumber={currentStepNumber}
-            onStepComplete={(res) => {
+          React.createElement(MultiStepCommandGenerator, {
+            prompt,
+            debug: options.debug,
+            previousSteps: executedSteps,
+            currentStepNumber,
+            remoteContext: remoteContext ? {
+              name: remoteContext.name,
+              sysInfo: remoteContext.sysInfo,
+              shellHistory: remoteContext.shellHistory,
+            } : undefined,
+            isRemote: !!remoteName,  // 远程执行时不检测 builtin
+            onStepComplete: (res: any) => {
               stepResult = res
               unmount()
-            }}
-          />
+            },
+          })
         )
 
         await waitUntilExit()
@@ -441,23 +1304,88 @@ program
         }
 
         if (stepResult.hasBuiltin) {
-          addHistory({
-            userPrompt: currentStepNumber === 1 ? prompt : `[步骤${currentStepNumber}] ${prompt}`,
-            command: stepResult.command,
-            executed: false,
-            exitCode: null,
-            output: '',
-            reason: 'builtin',
-          })
+          // 远程模式记录到远程历史
+          if (remoteName) {
+            addRemoteHistory(remoteName, {
+              userPrompt: currentStepNumber === 1 ? prompt : `[步骤${currentStepNumber}] ${prompt}`,
+              command: stepResult.command,
+              aiGeneratedCommand: stepResult.aiGeneratedCommand,
+              userModified: stepResult.userModified || false,
+              executed: false,
+              exitCode: null,
+              output: '',
+              reason: 'builtin',
+            })
+          } else {
+            addHistory({
+              userPrompt: currentStepNumber === 1 ? prompt : `[步骤${currentStepNumber}] ${prompt}`,
+              command: stepResult.command,
+              aiGeneratedCommand: stepResult.aiGeneratedCommand, // AI 原始命令
+              userModified: stepResult.userModified || false,
+              executed: false,
+              exitCode: null,
+              output: '',
+              reason: 'builtin',
+            })
+          }
           process.exit(0)
         }
 
         if (stepResult.confirmed) {
-          // 执行命令
+          // 如果命令为空，说明 AI 决定放弃
+          if (!stepResult.command || stepResult.command.trim() === '') {
+            console.log('')
+            if (stepResult.reasoning) {
+              console2.info(`💡 AI 分析: ${stepResult.reasoning}`)
+            }
+            console2.muted('❌ AI 决定停止尝试，任务失败')
+            console.log('')
+            process.exit(1)
+          }
+
+          // 特殊处理：如果上一步失败，且 AI 决定放弃（continue: false），直接显示原因并退出
+          if (
+            lastStepFailed &&
+            stepResult.needsContinue === false &&
+            stepResult.command.startsWith('echo')
+          ) {
+            console.log('')
+            if (stepResult.reasoning) {
+              console2.info(`💡 AI 分析: ${stepResult.reasoning}`)
+            }
+            console2.muted('❌ AI 决定停止尝试，任务失败')
+            console.log('')
+            process.exit(1)
+          }
+
+          // 执行命令（本地或远程）
           const execStart = Date.now()
-          const { exitCode, output } = await executeCommand(stepResult.command, prompt)
+          let exitCode: number
+          let output: string
+          let stdout: string
+
+          if (remoteName) {
+            // 远程执行
+            const result = await executeRemoteCommand(remoteName, stepResult.command)
+            exitCode = result.exitCode
+            output = result.output
+            stdout = result.stdout
+          } else {
+            // 本地执行
+            const result = await executeCommand(stepResult.command)
+            exitCode = result.exitCode
+            output = result.output
+            stdout = result.stdout
+          }
           const execDuration = Date.now() - execStart
 
+          // 判断命令是否成功
+          // 退出码 141 = 128 + 13 (SIGPIPE)，是管道正常关闭时的信号
+          // 例如：ps aux | head -3，head 读完 3 行就关闭管道，ps 收到 SIGPIPE
+          // 但如果退出码是 141 且没有 stdout 输出，说明可能是真正的错误
+          const isSigpipeWithOutput = exitCode === 141 && stdout.trim().length > 0
+          const isSuccess = exitCode === 0 || isSigpipeWithOutput
+
           // 保存到执行历史
           const executedStep: ExecutedStep = {
             command: stepResult.command,
@@ -469,19 +1397,34 @@ program
           }
           executedSteps.push(executedStep)
 
-          // 记录到 pls 历史
-          addHistory({
-            userPrompt:
-              currentStepNumber === 1 ? prompt : `[步骤${currentStepNumber}] ${stepResult.reasoning || prompt}`,
-            command: stepResult.command,
-            executed: true,
-            exitCode,
-            output,
-          })
+          // 记录到 pls 历史（远程模式记录到远程历史）
+          if (remoteName) {
+            addRemoteHistory(remoteName, {
+              userPrompt:
+                currentStepNumber === 1 ? prompt : `[步骤${currentStepNumber}] ${stepResult.reasoning || prompt}`,
+              command: stepResult.command,
+              aiGeneratedCommand: stepResult.aiGeneratedCommand,
+              userModified: stepResult.userModified || false,
+              executed: true,
+              exitCode,
+              output,
+            })
+          } else {
+            addHistory({
+              userPrompt:
+                currentStepNumber === 1 ? prompt : `[步骤${currentStepNumber}] ${stepResult.reasoning || prompt}`,
+              command: stepResult.command,
+              aiGeneratedCommand: stepResult.aiGeneratedCommand, // AI 原始命令
+              userModified: stepResult.userModified || false,
+              executed: true,
+              exitCode,
+              output,
+            })
+          }
 
           // 显示结果
           console.log('')
-          if (exitCode === 0) {
+          if (isSuccess) {
             if (currentStepNumber === 1 && stepResult.needsContinue !== true) {
               // 单步命令
               console2.success(`执行完成 ${console2.formatDuration(execDuration)}`)
@@ -489,14 +1432,19 @@ program
               // 多步命令
               console2.success(`步骤 ${currentStepNumber} 执行完成 ${console2.formatDuration(execDuration)}`)
             }
+            lastStepFailed = false
           } else {
-            // 执行失败，但不立即退出，让 AI 分析错误并调整策略
+            // 执行失败，标记状态
             console2.error(
               `步骤 ${currentStepNumber} 执行失败，退出码: ${exitCode} ${console2.formatDuration(execDuration)}`
             )
             console.log('')
             console2.warning('正在请 AI 分析错误并调整策略...')
-            // 不退出，继续循环，AI 会收到错误信息
+            lastStepFailed = true
+            // 继续循环，让 AI 分析错误
+            console.log('')
+            currentStepNumber++
+            continue
           }
 
           // 判断是否继续
@@ -511,30 +1459,111 @@ program
 
           console.log('')
           currentStepNumber++
+        } else if (!stepResult.confirmed && !stepResult.cancelled) {
+          // AI 返回了结果但没有确认（空命令的情况）
+          if (lastStepFailed && stepResult.reasoning) {
+            console.log('')
+            console2.info(`💡 AI 分析: ${stepResult.reasoning}`)
+            console2.muted('❌ AI 决定停止尝试，任务失败')
+            console.log('')
+            process.exit(1)
+          }
+          // 其他情况也退出
+          console.log('')
+          console2.muted('任务结束')
+          console.log('')
+          process.exit(0)
         }
       }
     })()
   })
 
+/**
+ * 执行远程命令
+ * 如果设置了工作目录，自动添加 cd 前缀
+ */
+async function executeRemoteCommand(
+  remoteName: string,
+  command: string
+): Promise<{ exitCode: number; output: string; stdout: string; stderr: string }> {
+  let stdout = ''
+  let stderr = ''
+
+  // 如果有工作目录，自动添加 cd 前缀
+  const workDir = getRemoteWorkDir(remoteName)
+  const actualCommand = workDir ? `cd ${workDir} && ${command}` : command
+
+  console.log('') // 空行
+
+  // 计算命令框宽度，让分隔线长度一致
+  const lines = command.split('\n')
+  const maxContentWidth = Math.max(...lines.map(l => console2.getDisplayWidth(l)))
+  const boxWidth = Math.max(maxContentWidth + 4, console2.getDisplayWidth('生成命令') + 6, 20)
+  console2.printSeparator(`远程输出 (${remoteName})`, boxWidth)
+
+  try {
+    const result = await sshExec(remoteName, actualCommand, {
+      onStdout: (data) => {
+        stdout += data
+        process.stdout.write(data)
+      },
+      onStderr: (data) => {
+        stderr += data
+        process.stderr.write(data)
+      },
+    })
+
+    if (stdout || stderr) {
+      console2.printSeparator('', boxWidth)
+    }
+
+    return {
+      exitCode: result.exitCode,
+      output: stdout + stderr,
+      stdout,
+      stderr,
+    }
+  } catch (error: any) {
+    console2.printSeparator('', boxWidth)
+    console2.error(error.message)
+    return {
+      exitCode: 1,
+      output: error.message,
+      stdout: '',
+      stderr: error.message,
+    }
+  }
+}
+
 // 自定义帮助信息
 program.addHelpText(
   'after',
   `
 ${chalk.bold('示例:')}
-  ${chalk.hex('#00D9FF')('pls 安装 git')}                    让 AI 生成安装 git 的命令
-  ${chalk.hex('#00D9FF')('pls 查找大于 100MB 的文件')}        查找大文件
-  ${chalk.hex('#00D9FF')('pls 删除刚才创建的文件')}          AI 会参考历史记录
-  ${chalk.hex('#00D9FF')('pls --debug 压缩 logs 目录')}      显示调试信息
-  ${chalk.hex('#00D9FF')('pls -m 删除当前目录的空文件夹')}    多步骤模式（AI 自动规划）
-  ${chalk.hex('#00D9FF')('pls chat tar 命令怎么用')}         AI 对话模式
-  ${chalk.hex('#00D9FF')('pls chat clear')}                 清空对话历史
-  ${chalk.hex('#00D9FF')('pls history')}                    查看 pls 命令历史
-  ${chalk.hex('#00D9FF')('pls history clear')}              清空历史记录
-  ${chalk.hex('#00D9FF')('pls hook')}                       查看 shell hook 状态
-  ${chalk.hex('#00D9FF')('pls hook install')}               安装 shell hook（增强功能）
-  ${chalk.hex('#00D9FF')('pls hook uninstall')}             卸载 shell hook
-  ${chalk.hex('#00D9FF')('pls config')}                     交互式配置
-  ${chalk.hex('#00D9FF')('pls config list')}                查看当前配置
+  ${chalk.hex(getThemeColors().primary)('pls 安装 git')}                    让 AI 生成安装 git 的命令
+  ${chalk.hex(getThemeColors().primary)('pls 查找大于 100MB 的文件')}        查找大文件
+  ${chalk.hex(getThemeColors().primary)('pls 删除刚才创建的文件')}          AI 会参考历史记录
+  ${chalk.hex(getThemeColors().primary)('pls --debug 压缩 logs 目录')}      显示调试信息
+  ${chalk.hex(getThemeColors().primary)('pls chat tar 命令怎么用')}         AI 对话模式
+  ${chalk.hex(getThemeColors().primary)('pls chat clear')}                 清空对话历史
+  ${chalk.hex(getThemeColors().primary)('pls history')}                    查看 pls 命令历史
+  ${chalk.hex(getThemeColors().primary)('pls history clear')}              清空历史记录
+  ${chalk.hex(getThemeColors().primary)('pls alias')}                      查看命令别名
+  ${chalk.hex(getThemeColors().primary)('pls alias add disk "查看磁盘"')}   添加别名
+  ${chalk.hex(getThemeColors().primary)('pls disk')}                       使用别名（等同于 pls @disk）
+  ${chalk.hex(getThemeColors().primary)('pls hook')}                       查看 shell hook 状态
+  ${chalk.hex(getThemeColors().primary)('pls hook install')}               安装 shell hook（增强功能）
+  ${chalk.hex(getThemeColors().primary)('pls hook uninstall')}             卸载 shell hook
+  ${chalk.hex(getThemeColors().primary)('pls upgrade')}                    升级到最新版本
+  ${chalk.hex(getThemeColors().primary)('pls config')}                     交互式配置
+  ${chalk.hex(getThemeColors().primary)('pls config list')}                查看当前配置
+
+${chalk.bold('远程执行:')}
+  ${chalk.hex(getThemeColors().primary)('pls remote')}                     查看远程服务器列表
+  ${chalk.hex(getThemeColors().primary)('pls remote add myserver root@1.2.3.4')}  添加服务器
+  ${chalk.hex(getThemeColors().primary)('pls remote test myserver')}       测试连接
+  ${chalk.hex(getThemeColors().primary)('pls -r myserver 查看磁盘')}       在远程服务器执行
+  ${chalk.hex(getThemeColors().primary)('pls remote hook install myserver')}  安装远程 Shell Hook
 `
 )