cc-haha/adapters/common/format.ts

/**
 * 消息格式化工具
 */

type AdapterChatState =
  | 'idle'
  | 'thinking'
  | 'streaming'
  | 'tool_executing'
  | 'permission_pending'

type ImStatusSummary = {
  sessionId?: string
  projectName?: string | null
  branch?: string | null
  model?: string | null
  state?: AdapterChatState | null
  verb?: string | null
  pendingPermissionCount?: number
  taskCounts?: {
    total: number
    pending: number
    inProgress: number
    completed: number
  }
}

const IM_HELP_LINES = [
  '/new [项目] / 新会话 — 新建会话或切换项目',
  '/projects / 项目列表 — 查看最近项目',
  '/status / 状态 — 查看当前会话状态',
  '/clear / 清空 — 清空当前会话上下文',
  '/stop / 停止 — 停止当前生成',
  '/help / 帮助 — 显示这份帮助',
  '权限审批：/allow <id>、/always <id>、/deny <id>',
]

/** Split text into chunks that fit within a character limit, respecting paragraph/sentence boundaries. */
export function splitMessage(text: string, limit: number): string[] {
  if (text.length <= limit) return [text]

  const chunks: string[] = []
  let remaining = text

  while (remaining.length > 0) {
    if (remaining.length <= limit) {
      chunks.push(remaining)
      break
    }

    let splitAt = remaining.lastIndexOf('\n\n', limit)
    if (splitAt <= 0) splitAt = remaining.lastIndexOf('\n', limit)
    if (splitAt <= 0) splitAt = remaining.lastIndexOf('. ', limit)
    if (splitAt <= 0) splitAt = remaining.lastIndexOf(' ', limit)
    if (splitAt <= 0) splitAt = limit

    // Include the delimiter for paragraph/sentence breaks
    if (remaining[splitAt] === '\n' || remaining[splitAt] === '.') splitAt += 1

    chunks.push(remaining.slice(0, splitAt).trimEnd())
    remaining = remaining.slice(splitAt).trimStart()
  }

  return chunks
}

/** Format tool use info for display in IM. */
export function formatToolUse(toolName: string, input: unknown): string {
  const inp = (input && typeof input === 'object' ? input : {}) as Record<string, unknown>
  const summary = formatToolSummary(toolName, inp)
  if (summary) return `🔧 ${toolName}  ${summary}`
  const preview = truncateInput(input, 200)
  return `🔧 ${toolName}\n${preview}`
}

/** Generate a concise human-readable summary for common tools. */
function formatToolSummary(tool: string, inp: Record<string, unknown>): string | null {
  switch (tool) {
    case 'Bash': {
      const desc = inp.description as string | undefined
      const cmd = inp.command as string | undefined
      if (desc) return desc
      if (cmd) return truncate(cmd, 120)
      return null
    }
    case 'Read': {
      const fp = inp.file_path as string | undefined
      if (fp) return shortPath(fp)
      return null
    }
    case 'Edit': {
      const fp = inp.file_path as string | undefined
      if (fp) return shortPath(fp)
      return null
    }
    case 'Write': {
      const fp = inp.file_path as string | undefined
      if (fp) return shortPath(fp)
      return null
    }
    case 'Grep': {
      const pat = inp.pattern as string | undefined
      const p = inp.path as string | undefined
      if (pat) return `"${truncate(pat, 60)}"` + (p ? ` in ${shortPath(p)}` : '')
      return null
    }
    case 'Glob': {
      const pat = inp.pattern as string | undefined
      return pat ? `"${pat}"` : null
    }
    case 'Skill': {
      const skill = inp.skill as string | undefined
      return skill || null
    }
    case 'Agent': {
      const desc = inp.description as string | undefined
      return desc || null
    }
    case 'WebFetch': {
      const url = inp.url as string | undefined
      return url ? truncate(url, 120) : null
    }
    case 'WebSearch': {
      const q = inp.query as string | undefined
      return q ? `"${truncate(q, 80)}"` : null
    }
    default:
      return null
  }
}

function shortPath(fp: string): string {
  const parts = fp.split('/')
  return parts.length > 3 ? '…/' + parts.slice(-3).join('/') : fp
}

function truncate(s: string, max: number): string {
  return s.length > max ? s.slice(0, max) + '…' : s
}

/** Format a permission request for display in IM. */
export function formatPermissionRequest(toolName: string, input: unknown, requestId: string): string {
  const preview = truncateInput(input, 300)
  return `🔐 需要权限确认 [${requestId}]\n工具: ${toolName}\n${preview}`
}

/** Truncate tool input to a preview string. */
export function truncateInput(input: unknown, maxLen: number): string {
  try {
    const s = typeof input === 'string' ? input : JSON.stringify(input, null, 2)
    return s.length > maxLen ? s.slice(0, maxLen) + '…' : s
  } catch {
    return '(unserializable)'
  }
}

/** Escape special characters for Telegram MarkdownV2. */
export function escapeMarkdownV2(text: string): string {
  return text.replace(/([_*\[\]()~`>#+\-=|{}.!\\])/g, '\\$1')
}

export function formatImHelp(): string {
  return `可用命令：\n\n${IM_HELP_LINES.join('\n')}`
}

export function formatImStatus(summary: ImStatusSummary | null): string {
  if (!summary?.sessionId) {
    return '当前没有活动会话。\n\n发送 /new 新建会话，或发送 /projects 选择项目。'
  }

  const lines = ['当前会话状态：']

  if (summary.projectName) {
    lines.push(`项目: ${summary.projectName}${summary.branch ? ` (${summary.branch})` : ''}`)
  } else if (summary.branch) {
    lines.push(`分支: ${summary.branch}`)
  }

  lines.push(`会话: ${shortSessionId(summary.sessionId)}`)

  if (summary.model) {
    lines.push(`模型: ${summary.model}`)
  }

  lines.push(`状态: ${formatAdapterChatState(summary.state, summary.verb)}`)

  const pendingPermissionCount = summary.pendingPermissionCount ?? 0
  if (pendingPermissionCount > 0) {
    lines.push(`审批: ${pendingPermissionCount} 个待确认`)
  }

  const taskCounts = summary.taskCounts
  if (taskCounts && taskCounts.total > 0) {
    const taskParts = [`总计 ${taskCounts.total}`]
    if (taskCounts.inProgress > 0) taskParts.push(`进行中 ${taskCounts.inProgress}`)
    if (taskCounts.pending > 0) taskParts.push(`待处理 ${taskCounts.pending}`)
    if (taskCounts.completed > 0) taskParts.push(`已完成 ${taskCounts.completed}`)
    lines.push(`任务: ${taskParts.join(' · ')}`)
  }

  return lines.join('\n')
}

function formatAdapterChatState(
  state: AdapterChatState | null | undefined,
  verb: string | null | undefined,
): string {
  const label = (() => {
    switch (state) {
      case 'thinking':
        return '思考中'
      case 'streaming':
        return '生成中'
      case 'tool_executing':
        return '执行工具中'
      case 'permission_pending':
        return '等待权限确认'
      case 'idle':
      default:
        return '空闲'
    }
  })()

  if (!verb || verb === 'Thinking') return label
  return `${label} (${verb})`
}

function shortSessionId(sessionId: string): string {
  return sessionId.length > 12 ? `${sessionId.slice(0, 8)}…` : sessionId
}