@yivan-lab/pretty-please 1.1.0 → 1.3.0

package/src/alias.ts

@@ -0,0 +1,301 @@
+import chalk from 'chalk'
+import { getConfig, saveConfig, type AliasConfig } from './config.js'
+import { getCurrentTheme } from './ui/theme.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,
+  }
+}
+
+/**
+ * 别名解析结果
+ */
+export interface AliasResolveResult {
+  resolved: boolean
+  prompt: string
+  aliasName?: string
+  originalInput?: string
+}
+
+/**
+ * 获取所有别名
+ */
+export function getAliases(): Record<string, AliasConfig> {
+  const config = getConfig()
+  return config.aliases || {}
+}
+
+/**
+ * 添加别名
+ * @param name 别名名称
+ * @param prompt 对应的 prompt
+ * @param description 可选描述
+ * @param reservedCommands 保留的子命令列表（动态传入）
+ */
+export function addAlias(
+  name: string,
+  prompt: string,
+  description?: string,
+  reservedCommands: string[] = []
+): void {
+  // 验证别名名称
+  if (!name || !name.trim()) {
+    throw new Error('别名名称不能为空')
+  }
+
+  // 移除可能的 @ 前缀
+  const aliasName = name.startsWith('@') ? name.slice(1) : name
+
+  // 验证别名名称格式（只允许字母、数字、下划线、连字符）
+  if (!/^[a-zA-Z0-9_-]+$/.test(aliasName)) {
+    throw new Error('别名名称只能包含字母、数字、下划线和连字符')
+  }
+
+  // 检查是否与保留命令冲突
+  if (reservedCommands.includes(aliasName)) {
+    throw new Error(`"${aliasName}" 是保留的子命令，不能用作别名`)
+  }
+
+  // 验证 prompt
+  if (!prompt || !prompt.trim()) {
+    throw new Error('prompt 不能为空')
+  }
+
+  const config = getConfig()
+  if (!config.aliases) {
+    config.aliases = {}
+  }
+
+  config.aliases[aliasName] = {
+    prompt: prompt.trim(),
+    description: description?.trim(),
+  }
+
+  saveConfig(config)
+}
+
+/**
+ * 删除别名
+ */
+export function removeAlias(name: string): boolean {
+  // 移除可能的 @ 前缀
+  const aliasName = name.startsWith('@') ? name.slice(1) : name
+
+  const config = getConfig()
+  if (!config.aliases || !config.aliases[aliasName]) {
+    return false
+  }
+
+  delete config.aliases[aliasName]
+  saveConfig(config)
+  return true
+}
+
+/**
+ * 解析参数模板
+ * 支持格式：{{param}} 或 {{param:default}}
+ */
+function parseTemplateParams(prompt: string): string[] {
+  const regex = /\{\{([^}:]+)(?::[^}]*)?\}\}/g
+  const params: string[] = []
+  let match
+
+  while ((match = regex.exec(prompt)) !== null) {
+    if (!params.includes(match[1])) {
+      params.push(match[1])
+    }
+  }
+
+  return params
+}
+
+/**
+ * 替换模板参数
+ * @param prompt 原始 prompt（可能包含模板参数）
+ * @param args 用户提供的参数（key=value 或 --key=value 格式）
+ */
+function replaceTemplateParams(prompt: string, args: string[]): string {
+  // 解析用户参数
+  const userParams: Record<string, string> = {}
+
+  for (const arg of args) {
+    // 支持 --key=value 或 key=value 格式
+    const cleanArg = arg.startsWith('--') ? arg.slice(2) : arg
+    const eqIndex = cleanArg.indexOf('=')
+    if (eqIndex > 0) {
+      const key = cleanArg.slice(0, eqIndex)
+      const value = cleanArg.slice(eqIndex + 1)
+      userParams[key] = value
+    }
+  }
+
+  // 替换模板参数
+  let result = prompt
+
+  // 匹配 {{param}} 或 {{param:default}}
+  result = result.replace(/\{\{([^}:]+)(?::([^}]*))?\}\}/g, (match, param, defaultValue) => {
+    if (userParams[param] !== undefined) {
+      return userParams[param]
+    }
+    if (defaultValue !== undefined) {
+      return defaultValue
+    }
+    // 没有提供值也没有默认值，保留原样（后面会报错或让用户补充）
+    return match
+  })
+
+  return result
+}
+
+/**
+ * 检查是否还有未替换的模板参数
+ */
+function hasUnresolvedParams(prompt: string): string[] {
+  const regex = /\{\{([^}:]+)\}\}/g
+  const unresolved: string[] = []
+  let match
+
+  while ((match = regex.exec(prompt)) !== null) {
+    unresolved.push(match[1])
+  }
+
+  return unresolved
+}
+
+/**
+ * 解析别名
+ * 支持 `pls disk` 和 `pls @disk` 两种格式
+ * @param input 用户输入（可能是别名或普通 prompt）
+ * @returns 解析结果
+ */
+export function resolveAlias(input: string): AliasResolveResult {
+  const parts = input.trim().split(/\s+/)
+  if (parts.length === 0) {
+    return { resolved: false, prompt: input }
+  }
+
+  let aliasName = parts[0]
+  const restArgs = parts.slice(1)
+
+  // 支持 @ 前缀
+  if (aliasName.startsWith('@')) {
+    aliasName = aliasName.slice(1)
+  }
+
+  const aliases = getAliases()
+  const aliasConfig = aliases[aliasName]
+
+  if (!aliasConfig) {
+    return { resolved: false, prompt: input }
+  }
+
+  // 检查是否有模板参数
+  const templateParams = parseTemplateParams(aliasConfig.prompt)
+
+  let resolvedPrompt: string
+
+  if (templateParams.length > 0) {
+    // 有模板参数，进行替换
+    resolvedPrompt = replaceTemplateParams(aliasConfig.prompt, restArgs)
+
+    // 检查是否还有未替换的必填参数
+    const unresolved = hasUnresolvedParams(resolvedPrompt)
+    if (unresolved.length > 0) {
+      throw new Error(`别名 "${aliasName}" 缺少必填参数: ${unresolved.join(', ')}`)
+    }
+
+    // 过滤掉已用于参数替换的 args，剩余的追加到 prompt
+    const usedArgs = restArgs.filter((arg) => {
+      const cleanArg = arg.startsWith('--') ? arg.slice(2) : arg
+      return cleanArg.includes('=')
+    })
+    const extraArgs = restArgs.filter((arg) => !usedArgs.includes(arg))
+
+    if (extraArgs.length > 0) {
+      resolvedPrompt = `${resolvedPrompt} ${extraArgs.join(' ')}`
+    }
+  } else {
+    // 没有模板参数，直接追加额外内容
+    if (restArgs.length > 0) {
+      resolvedPrompt = `${aliasConfig.prompt} ${restArgs.join(' ')}`
+    } else {
+      resolvedPrompt = aliasConfig.prompt
+    }
+  }
+
+  return {
+    resolved: true,
+    prompt: resolvedPrompt,
+    aliasName,
+    originalInput: input,
+  }
+}
+
+/**
+ * 显示所有别名
+ */
+export function displayAliases(): void {
+  const aliases = getAliases()
+  const colors = getColors()
+  const aliasNames = Object.keys(aliases)
+
+  console.log('')
+
+  if (aliasNames.length === 0) {
+    console.log(chalk.gray('  暂无别名'))
+    console.log('')
+    console.log(chalk.gray('  使用 pls alias add <name> "<prompt>" 添加别名'))
+    console.log('')
+    return
+  }
+
+  console.log(chalk.bold('命令别名:'))
+  console.log(chalk.gray('━'.repeat(50)))
+
+  for (const name of aliasNames) {
+    const alias = aliases[name]
+    const params = parseTemplateParams(alias.prompt)
+
+    // 别名名称
+    let line = `  ${chalk.hex(colors.primary)(name)}`
+
+    // 如果有参数，显示参数
+    if (params.length > 0) {
+      line += chalk.gray(` <${params.join('> <')}>`)
+    }
+
+    console.log(line)
+
+    // prompt 内容
+    console.log(`    ${chalk.gray('→')} ${alias.prompt}`)
+
+    // 描述
+    if (alias.description) {
+      console.log(`    ${chalk.gray(alias.description)}`)
+    }
+
+    console.log('')
+  }
+
+  console.log(chalk.gray('━'.repeat(50)))
+  console.log(chalk.gray('使用: pls <alias> 或 pls @<alias>'))
+  console.log('')
+}
+
+/**
+ * 获取别名的参数信息（用于帮助显示）
+ */
+export function getAliasParams(aliasName: string): string[] {
+  const aliases = getAliases()
+  const alias = aliases[aliasName]
+  if (!alias) return []
+  return parseTemplateParams(alias.prompt)
+}

package/src/chat-history.ts

@@ -3,6 +3,15 @@ import path from 'path'
 import os from 'os'
 import chalk from 'chalk'
 import { getConfig } from './config.js'
+import { getCurrentTheme } from './ui/theme.js'
+
+// 获取主题颜色
+function getColors() {
+  const theme = getCurrentTheme()
+  return {
+    primary: theme.primary,
+  }
+}
 
 const CONFIG_DIR = path.join(os.homedir(), '.please')
 const CHAT_HISTORY_FILE = path.join(CONFIG_DIR, 'chat_history.json')
@@ -104,6 +113,7 @@ export function getChatRoundCount(): number {
 export function displayChatHistory(): void {
   const history = getChatHistory()
   const config = getConfig()
+  const colors = getColors()
 
   if (history.length === 0) {
     console.log('\n' + chalk.gray('暂无对话历史'))
@@ -120,7 +130,7 @@ export function displayChatHistory(): void {
 
   userMessages.forEach((msg, index) => {
     const num = index + 1
-    console.log(`  ${chalk.cyan(num.toString().padStart(2, ' '))}. ${msg.content}`)
+    console.log(`  ${chalk.hex(colors.primary)(num.toString().padStart(2, ' '))}. ${msg.content}`)
   })
 
   console.log(chalk.gray('━'.repeat(50)))

package/src/components/Chat.tsx

@@ -3,8 +3,13 @@ import { Box, Text } from 'ink'
 import Spinner from 'ink-spinner'
 import { MarkdownDisplay } from './MarkdownDisplay.js'
 import { chatWithMastra } from '../mastra-chat.js'
-import { getChatRoundCount } from '../chat-history.js'
-import { theme } from '../ui/theme.js'
+import { getChatRoundCount, getChatHistory } from '../chat-history.js'
+import { getCurrentTheme } from '../ui/theme.js'
+import { formatSystemInfo } from '../sysinfo.js'
+import { formatHistoryForAI } from '../history.js'
+import { formatShellHistoryForAI, getShellHistory } from '../shell-hook.js'
+import { getConfig } from '../config.js'
+import { CHAT_SYSTEM_PROMPT, buildChatUserContext } from '../prompts.js'
 
 interface ChatProps {
   prompt: string
@@ -19,7 +24,7 @@ interface DebugInfo {
   sysinfo: string
   model: string
   systemPrompt: string
-  userPrompt: string
+  userContext: string
   chatHistory: any[]
 }
 
@@ -28,12 +33,40 @@ interface DebugInfo {
  * 使用正常渲染，完成后保持最后一帧在终端
  */
 export function Chat({ prompt, debug, showRoundCount, onComplete }: ChatProps) {
+  const theme = getCurrentTheme()
   const [status, setStatus] = useState<Status>('thinking')
   const [content, setContent] = useState('')
   const [duration, setDuration] = useState(0)
-  const [debugInfo, setDebugInfo] = useState<DebugInfo | null>(null)
   const [roundCount] = useState(getChatRoundCount())
 
+  // Debug 信息：直接在 useState 初始化时计算（同步）
+  const [debugInfo] = useState<DebugInfo | null>(() => {
+    if (!debug) return null
+
+    const config = getConfig()
+    const sysinfo = formatSystemInfo()
+    const plsHistory = formatHistoryForAI()
+    const shellHistory = formatShellHistoryForAI()
+    const shellHookEnabled = config.shellHook && getShellHistory().length > 0
+    const chatHistory = getChatHistory()
+
+    const userContext = buildChatUserContext(
+      prompt,
+      sysinfo,
+      plsHistory,
+      shellHistory,
+      shellHookEnabled
+    )
+
+    return {
+      sysinfo,
+      model: config.model,
+      systemPrompt: CHAT_SYSTEM_PROMPT,
+      userContext,
+      chatHistory,
+    }
+  })
+
   useEffect(() => {
     const startTime = Date.now()
 
@@ -44,17 +77,13 @@ export function Chat({ prompt, debug, showRoundCount, onComplete }: ChatProps) {
     }
 
     // 调用 AI
-    chatWithMastra(prompt, { debug: debug || false, onChunk })
+    chatWithMastra(prompt, { debug: false, onChunk })  // 不需要 AI 返回 debug
       .then((result) => {
         const endTime = Date.now()
         setDuration(endTime - startTime)
         setStatus('done')
 
-        if (debug && typeof result === 'object' && 'debug' in result && result.debug) {
-          setDebugInfo(result.debug)
-        }
-
-        setTimeout(onComplete, 100)
+        setTimeout(onComplete, debug ? 500 : 100)
       })
       .catch((error: any) => {
         setStatus('error')
@@ -65,6 +94,38 @@ export function Chat({ prompt, debug, showRoundCount, onComplete }: ChatProps) {
 
   return (
     <Box flexDirection="column">
+      {/* 调试信息 - 放在最前面 */}
+      {debugInfo && (
+        <Box flexDirection="column" marginBottom={1}>
+          <Text color={theme.accent} bold>━━━ 调试信息 ━━━</Text>
+          <Text color={theme.text.secondary}>模型: {debugInfo.model}</Text>
+          <Text color={theme.text.secondary}>对话历史轮数: {Math.floor(debugInfo.chatHistory.length / 2)}</Text>
+
+          {/* 历史对话（只显示用户问题） */}
+          {debugInfo.chatHistory.length > 0 && (
+            <Box flexDirection="column" marginTop={1}>
+              <Text color={theme.text.secondary}>历史对话（用户问题）:</Text>
+              {debugInfo.chatHistory
+                .filter((msg) => msg.role === 'user')
+                .slice(-5)  // 最多显示最近 5 条
+                .map((msg, idx) => (
+                  <Text key={idx} color={theme.text.muted}>
+                    {idx + 1}. {msg.content.substring(0, 50)}{msg.content.length > 50 ? '...' : ''}
+                  </Text>
+                ))}
+            </Box>
+          )}
+
+          {/* User Context */}
+          <Box flexDirection="column" marginTop={1}>
+            <Text color={theme.text.secondary}>User Context (最新消息):</Text>
+            <Text color={theme.text.muted}>{debugInfo.userContext.substring(0, 500)}...</Text>
+          </Box>
+
+          <Text color={theme.accent}>━━━━━━━━━━━━━━━━</Text>
+        </Box>
+      )}
+
       {/* 显示对话轮数 */}
       {showRoundCount && roundCount > 0 && (
         <Box marginBottom={1}>
@@ -101,22 +162,6 @@ export function Chat({ prompt, debug, showRoundCount, onComplete }: ChatProps) {
           <Text color={theme.text.secondary}>({(duration / 1000).toFixed(2)}s)</Text>
         </Box>
       )}
-
-      {/* 调试信息 */}
-      {debugInfo && (
-        <Box flexDirection="column" marginY={1}>
-          <Text color={theme.accent}>━━━ 调试信息 ━━━</Text>
-          <Text color={theme.text.secondary}>系统信息: {debugInfo.sysinfo}</Text>
-          <Text color={theme.text.secondary}>模型: {debugInfo.model}</Text>
-          <Text color={theme.text.secondary}>
-            对话历史轮数: {Math.floor(debugInfo.chatHistory.length / 2)}
-          </Text>
-          <Text color={theme.text.secondary}>System Prompt:</Text>
-          <Text dimColor>{debugInfo.systemPrompt}</Text>
-          <Text color={theme.text.secondary}>User Prompt: {debugInfo.userPrompt}</Text>
-          <Text color={theme.accent}>━━━━━━━━━━━━━━━━</Text>
-        </Box>
-      )}
     </Box>
   )
 }

package/src/components/CodeColorizer.tsx

@@ -2,23 +2,27 @@ import React from 'react'
 import { Text, Box } from 'ink'
 import { common, createLowlight } from 'lowlight'
 import type { Root, Element, Text as HastText, ElementContent, RootContent } from 'hast'
-import { theme } from '../ui/theme.js'
+import { getCurrentTheme, type Theme } from '../ui/theme.js'
 
 // 创建 lowlight 实例
 const lowlight = createLowlight(common)
 
-// 语法高亮颜色映射
-const syntaxColors: Record<string, string> = {
-  'hljs-keyword': theme.code.keyword,
-  'hljs-string': theme.code.string,
-  'hljs-function': theme.code.function,
-  'hljs-comment': theme.code.comment,
-  'hljs-number': theme.primary,
-  'hljs-built_in': theme.secondary,
-  'hljs-title': theme.accent,
-  'hljs-variable': theme.text.primary,
-  'hljs-type': theme.info,
-  'hljs-operator': theme.text.secondary,
+/**
+ * 获取语法高亮颜色映射
+ */
+function getSyntaxColors(theme: Theme): Record<string, string> {
+  return {
+    'hljs-keyword': theme.code.keyword,
+    'hljs-string': theme.code.string,
+    'hljs-function': theme.code.function,
+    'hljs-comment': theme.code.comment,
+    'hljs-number': theme.primary,
+    'hljs-built_in': theme.secondary,
+    'hljs-title': theme.accent,
+    'hljs-variable': theme.text.primary,
+    'hljs-type': theme.info,
+    'hljs-operator': theme.text.secondary,
+  }
 }
 
 /**
@@ -26,7 +30,9 @@ const syntaxColors: Record<string, string> = {
  */
 function renderHastNode(
   node: Root | Element | HastText | RootContent,
-  inheritedColor: string | undefined
+  inheritedColor: string | undefined,
+  syntaxColors: Record<string, string>,
+  theme: Theme
 ): React.ReactNode {
   if (node.type === 'text') {
     const color = inheritedColor || theme.code.text
@@ -51,7 +57,7 @@ function renderHastNode(
     // 递归渲染子节点
     const children = node.children?.map((child: ElementContent, index: number) => (
       <React.Fragment key={index}>
-        {renderHastNode(child, colorToPassDown)}
+        {renderHastNode(child, colorToPassDown, syntaxColors, theme)}
       </React.Fragment>
     ))
 
@@ -65,7 +71,7 @@ function renderHastNode(
 
     return node.children?.map((child: RootContent, index: number) => (
       <React.Fragment key={index}>
-        {renderHastNode(child, inheritedColor)}
+        {renderHastNode(child, inheritedColor, syntaxColors, theme)}
       </React.Fragment>
     ))
   }
@@ -76,13 +82,13 @@ function renderHastNode(
 /**
  * 高亮并渲染一行代码
  */
-function highlightLine(line: string, language: string | null): React.ReactNode {
+function highlightLine(line: string, language: string | null, syntaxColors: Record<string, string>, theme: Theme): React.ReactNode {
   try {
     const highlighted = !language || !lowlight.registered(language)
       ? lowlight.highlightAuto(line)
       : lowlight.highlight(language, line)
 
-    const rendered = renderHastNode(highlighted, undefined)
+    const rendered = renderHastNode(highlighted, undefined, syntaxColors, theme)
     return rendered !== null ? rendered : line
   } catch {
     return line
@@ -99,12 +105,14 @@ interface ColorizeCodeProps {
  * 代码高亮组件
  */
 function ColorizeCodeInternal({ code, language = null, showLineNumbers = false }: ColorizeCodeProps) {
+  const theme = getCurrentTheme()
+  const syntaxColors = getSyntaxColors(theme)
   const codeToHighlight = code.replace(/\n$/, '')
   const lines = codeToHighlight.split('\n')
   const padWidth = String(lines.length).length
 
   const renderedLines = lines.map((line, index) => {
-    const contentToRender = highlightLine(line, language)
+    const contentToRender = highlightLine(line, language, syntaxColors, theme)
 
     return (
       <Box key={index} minHeight={1}>

package/src/components/CommandBox.tsx

@@ -1,7 +1,7 @@
 import React from 'react'
 import { Box, Text } from 'ink'
-import { theme } from '../ui/theme.js'
-import { getDisplayWidth } from '../utils/console.js'
+import { getCurrentTheme } from '../ui/theme.js'
+import { getDisplayWidth, wrapText } from '../utils/console.js'
 
 interface CommandBoxProps {
   command: string
@@ -12,14 +12,32 @@ interface CommandBoxProps {
  * CommandBox 组件 - 显示带边框和标题的命令框
  */
 export const CommandBox: React.FC<CommandBoxProps> = ({ command, title = '生成命令' }) => {
-  const lines = command.split('\n')
+  const theme = getCurrentTheme()
+
+  // 获取终端宽度，限制最大宽度
+  const termWidth = process.stdout.columns || 80
   const titleWidth = getDisplayWidth(title)
-  const maxContentWidth = Math.max(...lines.map(l => getDisplayWidth(l)))
-  const boxWidth = Math.max(maxContentWidth + 4, titleWidth + 6, 20)
+
+  // 计算最大内容宽度（终端宽度 - 边框和内边距）
+  const maxContentWidth = termWidth - 6 // 减去 '│ ' 和 ' │' 以及一些余量
+
+  // 处理命令换行
+  const originalLines = command.split('\n')
+  const wrappedLines: string[] = []
+  for (const line of originalLines) {
+    wrappedLines.push(...wrapText(line, maxContentWidth))
+  }
+
+  // 计算实际使用的宽度
+  const actualMaxWidth = Math.max(
+    ...wrappedLines.map((l) => getDisplayWidth(l)),
+    titleWidth
+  )
+  const boxWidth = Math.min(actualMaxWidth + 4, termWidth - 2)
 
   // 顶部边框：┌─ 生成命令 ─────┐
   const topPadding = boxWidth - titleWidth - 5
-  const topBorder = '┌─ ' + title + ' ' + '─'.repeat(topPadding) + '┐'
+  const topBorder = '┌─ ' + title + ' ' + '─'.repeat(Math.max(0, topPadding)) + '┐'
 
   // 底部边框
   const bottomBorder = '└' + '─'.repeat(boxWidth - 2) + '┘'
@@ -27,9 +45,9 @@ export const CommandBox: React.FC<CommandBoxProps> = ({ command, title = '生成
   return (
     <Box flexDirection="column" marginY={1}>
       <Text color={theme.warning}>{topBorder}</Text>
-      {lines.map((line, index) => {
+      {wrappedLines.map((line, index) => {
         const lineWidth = getDisplayWidth(line)
-        const padding = ' '.repeat(boxWidth - lineWidth - 4)
+        const padding = ' '.repeat(Math.max(0, boxWidth - lineWidth - 4))
         return (
           <Text key={index}>
             <Text color={theme.warning}>│ </Text>

package/src/components/ConfirmationPrompt.tsx

@@ -1,6 +1,6 @@
 import React from 'react'
 import { Text, useInput } from 'ink'
-import { theme } from '../ui/theme.js'
+import { getCurrentTheme } from '../ui/theme.js'
 
 interface ConfirmationPromptProps {
   prompt: string
@@ -19,6 +19,7 @@ export const ConfirmationPrompt: React.FC<ConfirmationPromptProps> = ({
   onCancel,
   onEdit,
 }) => {
+  const theme = getCurrentTheme()
   useInput((input, key) => {
     if (key.return) {
       // 回车键

package/src/components/Duration.tsx

@@ -1,6 +1,6 @@
 import React from 'react'
 import { Text } from 'ink'
-import { theme } from '../ui/theme.js'
+import { getCurrentTheme } from '../ui/theme.js'
 
 interface DurationProps {
   ms: number
@@ -20,5 +20,6 @@ function formatDuration(ms: number): string {
  * Duration 组件 - 显示耗时
  */
 export const Duration: React.FC<DurationProps> = ({ ms }) => {
+  const theme = getCurrentTheme()
   return <Text color={theme.text.secondary}>({formatDuration(ms)})</Text>
 }

package/src/components/InlineRenderer.tsx

@@ -1,6 +1,6 @@
 import React from 'react'
 import { Text } from 'ink'
-import { theme } from '../ui/theme.js'
+import { getCurrentTheme } from '../ui/theme.js'
 
 interface RenderInlineProps {
   text: string
@@ -12,6 +12,7 @@ interface RenderInlineProps {
  * 处理 **粗体**、*斜体*、`代码`、~~删除线~~、<u>下划线</u>、链接
  */
 function RenderInlineInternal({ text, defaultColor }: RenderInlineProps) {
+  const theme = getCurrentTheme()
   const baseColor = defaultColor || theme.text.primary
 
   // 快速路径：纯文本无 markdown