src/types/message.ts

import type { UUID } from 'crypto'
import type {
  ContentBlockParam,
  MessageParam,
} from '@anthropic-ai/sdk/resources/messages.mjs'

export type MessageOrigin = 'user' | 'assistant' | 'system'

export type Message = {
  role: 'user' | 'assistant' | 'system'
  content: ContentBlockParam[]
  uuid: string
  timestamp: number
  origin?: MessageOrigin
}

export type UserMessage = {
  role: 'user'
  content: ContentBlockParam[]
  uuid: string
  timestamp: number
  origin?: MessageOrigin
}

export type AssistantMessage = {
  role: 'assistant'
  content: ContentBlockParam[]
  uuid: string
  timestamp: number
  origin?: MessageOrigin
  model?: string
}

export type NormalizedMessage =
  | NormalizedUserMessage
  | NormalizedAssistantMessage

export type NormalizedUserMessage = {
  type: 'user'
  message: UserMessage
  uuid: string
  timestamp: number
  origin?: MessageOrigin
  toolUseResult?: { stdout?: string; stderr?: string }
}

export type NormalizedAssistantMessage = {
  type: 'assistant'
  message: AssistantMessage
  uuid: string
  timestamp: number
  origin?: MessageOrigin
  model?: string
}

export type SystemMessage = {
  type: 'system'
  subtype: string
  uuid: string
  timestamp: number
  level?: string
  text?: string
}

export type SystemInformationalMessage = SystemMessage & {
  subtype: 'informational'
  text: string
}

export type SystemAPIErrorMessage = SystemMessage & {
  subtype: 'api_error'
  error: string
}

export type StopHookInfo = {
  command: string
  durationMs?: number
  hookName?: string
}

export type SystemStopHookSummaryMessage = SystemMessage & {
  subtype: 'stop_hook_summary'
  hookLabel: string
  hookCount: number
  totalDurationMs?: number
  hookInfos: StopHookInfo[]
}

export type SystemAgentsKilledMessage = SystemMessage & {
  subtype: 'agents_killed'
}

export type SystemApiMetricsMessage = SystemMessage & {
  subtype: 'api_metrics'
}

export type SystemAwaySummaryMessage = SystemMessage & {
  subtype: 'away_summary'
}

export type SystemBridgeStatusMessage = SystemMessage & {
  subtype: 'bridge_status'
}

export type SystemCompactBoundaryMessage = SystemMessage & {
  subtype: 'compact_boundary'
}

export type SystemLocalCommandMessage = SystemMessage & {
  subtype: 'local_command'
}

export type SystemMemorySavedMessage = SystemMessage & {
  subtype: 'memory_saved'
}

export type SystemMicrocompactBoundaryMessage = SystemMessage & {
  subtype: 'microcompact_boundary'
}

export type SystemPermissionRetryMessage = SystemMessage & {
  subtype: 'permission_retry'
}

export type SystemScheduledTaskFireMessage = SystemMessage & {
  subtype: 'scheduled_task_fire'
}

export type SystemTurnDurationMessage = SystemMessage & {
  subtype: 'turn_duration'
}

export type ProgressMessage = {
  type: 'progress'
  uuid: string
  timestamp: number
  data?: {
    type: string
    phase?: string
    toolName?: string
    toolInput?: unknown
    elapsedTimeSeconds?: number
    totalLines?: number
  }
}

export type HookResultMessage = {
  type: 'hook_result'
  uuid: string
  timestamp: number
  hookName: string
}

export type Attachment = {
  type: string
  memories?: { path: string; content: string; mtimeMs: number }[]
}

export type AttachmentMessage = {
  type: 'attachment'
  attachment: Attachment
  uuid: string
  timestamp: number
}

export type GroupedToolUseMessage = {
  type: 'grouped_tool_use'
  toolName: string
  messages: NormalizedAssistantMessage[]
  uuid: string
  timestamp: number
  displayMessage: NormalizedAssistantMessage
}

export type CollapsibleMessage =
  | NormalizedAssistantMessage
  | GroupedToolUseMessage
  | NormalizedUserMessage

export type RenderableMessage =
  | NormalizedUserMessage
  | NormalizedAssistantMessage
  | SystemMessage
  | AttachmentMessage
  | GroupedToolUseMessage
  | ProgressMessage

export type TombstoneMessage = SystemMessage & {
  subtype: 'tombstone'
}

export type ToolUseSummaryMessage = {
  type: 'tool_use_summary'
  uuid: string
  timestamp: number
}

export type CollapsedReadSearchGroup = {
  type: 'collapsed_read_search'
  searchCount: number
  readCount: number
  listCount: number
  replCount: number
  memorySearchCount: number
  memoryReadCount: number
  memoryWriteCount: number
  readFilePaths: string[]
  searchArgs: string[]
  latestDisplayHint?: string
  messages: CollapsibleMessage[]
  displayMessage: CollapsibleMessage
  uuid: string
  timestamp: number
  mcpCallCount?: number
  mcpServerNames?: string[]
  bashCount?: number
  gitOpBashCount?: number
  commits?: { sha: string; kind: string }[]
  pushes?: { branch: string }[]
  branches?: { ref: string; action: string }[]
  prs?: { number: number; url?: string; action: string }[]
  hookTotalMs?: number
  hookCount?: number
  hookInfos?: StopHookInfo[]
  relevantMemories?: { path: string; content: string; mtimeMs: number }[]
  teamMemorySearchCount?: number
  teamMemoryReadCount?: number
  teamMemoryWriteCount?: number
  webFetchCount?: number
  webFetchURLs?: string[]
}

export type PartialCompactDirection = 'forward' | 'backward'

export type RequestStartEvent = {
  type: 'request_start'
  timestamp: number
}

export type StreamEvent =
  | { type: 'message_start'; message: MessageParam }
  | { type: 'content_block_start'; content_block: ContentBlockParam }
  | { type: 'content_block_delta'; delta: unknown }
  | { type: 'content_block_stop' }
  | { type: 'message_delta'; usage: unknown }
  | { type: 'message_stop' }