@tduyng/pi-subagent 0.0.0-stage → 0.0.2

package/src/subagent/types.ts

@@ -0,0 +1,109 @@
+import type { Model } from '@earendil-works/pi-ai'
+
+import type { JsonPayload } from './structured.js'
+
+export const RUNS_DIR = 'subagents'
+export const AGENTS_DIR = 'agents'
+export const EMIT_INTERVAL_MS = 250
+export const MAX_TASKS = 8
+export const MAX_CONCURRENT = 4
+export const DEFAULT_MAX_TURNS = 50
+export const THINKING_LEVELS = ['off', 'minimal', 'low', 'medium', 'high', 'xhigh', 'max'] as const
+export const WIDGET_KEY = 'subagent-widget'
+export const MAX_WIDGET_ROWS = 8
+export const MAX_REGISTRY = 50
+export const MAX_TOUCHED = 12
+export const FINISHED_RETENTION_MS = 5 * 60_000
+export const FAILED_RETENTION_MS = 30 * 60_000
+
+export type ThinkingLevel = (typeof THINKING_LEVELS)[number]
+export const DEFAULT_THINKING: ThinkingLevel = 'medium'
+export type RunStatus = 'queued' | 'running' | 'completed' | 'failed'
+
+export interface AgentProfile {
+    name: string
+    description: string
+    filePath: string
+    systemPrompt: string
+    systemPromptMode: 'append' | 'replace'
+    thinking?: ThinkingLevel
+    model?: string
+    tools: string[]
+    disallowedTools: string[]
+    inheritProjectContext: boolean
+    inheritSkills: boolean
+    maxExecutionTimeMs?: number
+    maxTurns?: number
+    color?: string
+}
+
+export interface UsageTotals {
+    input: number
+    output: number
+    cost: number
+}
+
+export interface TaskDetails {
+    id: string
+    index: number
+    label: string
+    agent?: string
+    status: RunStatus
+    cwd: string
+    provider: string
+    model: string
+    thinking: string
+    activity?: string
+    turnCount?: number
+    toolCallCount?: number
+    touched?: string[]
+    usage?: UsageTotals
+    output?: string
+    partial?: boolean
+    hasSchema?: boolean
+    maxTurns?: number
+    structured?: JsonPayload
+    error?: string
+    sessionFile?: string
+    worktree?: string
+    changes?: string[]
+    color?: string
+    forkContext?: boolean
+    startedAt?: number
+    finishedAt?: number
+}
+
+export interface RunDetails {
+    status: RunStatus
+    runId: string
+    startedAt: number
+    finishedAt?: number
+    tasks: TaskDetails[]
+}
+
+export interface TaskSpec {
+    index: number
+    id: string
+    label: string
+    task: string
+    agent?: AgentProfile
+    cwd: string
+    dir: string
+    thinking: ThinkingLevel
+    model?: Model<any>
+    timeoutMs?: number
+    maxTurns: number
+    isolation?: 'worktree'
+    forkContext?: boolean
+    schema?: JsonPayload
+}
+
+export interface ToolCallArguments {
+    command?: string
+    path?: string
+    file_path?: string
+    url?: string
+    query?: string
+    pattern?: string
+    content?: string
+}

package/src/subagent/viewer.ts

@@ -0,0 +1,184 @@
+import type { AgentSession, Theme } from '@earendil-works/pi-coding-agent'
+import {
+    isKeyRelease,
+    matchesKey,
+    truncateToWidth,
+    wrapTextWithAnsi,
+    type Component,
+    type TUI
+} from '@earendil-works/pi-tui'
+
+import { MAX_WIDGET_ROWS, type TaskDetails } from './types.js'
+import { roster } from './state.js'
+import { buildTranscriptLines, formatDuration, fmtTokens, statusIcon, truncate } from './format.js'
+
+export function widgetLines(theme: Theme, highlightIndex: number | null): string[] {
+    const all = roster()
+    const items = all.slice(0, MAX_WIDGET_ROWS)
+    if (items.length === 0) return []
+
+    const validColors = new Set<string>([
+        'accent',
+        'success',
+        'error',
+        'warning',
+        'muted',
+        'dim',
+        'text',
+        'toolTitle',
+        'toolOutput',
+        'mdHeading',
+        'mdLink'
+    ])
+
+    const lines: string[] = []
+
+    const running = items.filter((t) => t.status === 'running' || t.status === 'queued').length
+    const done = items.filter((t) => t.status === 'completed').length
+    const failed = items.filter((t) => t.status === 'failed').length
+    const parts = []
+    if (running > 0) parts.push(`${running} running`)
+    if (done > 0) parts.push(`${done} done`)
+    if (failed > 0) parts.push(theme.fg('error', `${failed} failed`))
+    if (all.length > MAX_WIDGET_ROWS) parts.push(`+${all.length - MAX_WIDGET_ROWS} more`)
+    lines.push(theme.fg('dim', `${all.length} agent${all.length !== 1 ? 's' : ''} (${parts.join(', ')})`))
+
+    items.forEach((task, i) => {
+        const marker = i === highlightIndex ? theme.fg('accent', theme.bold('▶ ')) : '  '
+        const duration = formatDuration(task.startedAt, task.finishedAt)
+
+        const tokenInfo = task.usage ? ` ${fmtTokens(task.usage.output)}↓` : ''
+        const costInfo = task.usage && task.usage.cost > 0 ? ` $${task.usage.cost.toFixed(3)}` : ''
+        const turnInfo = task.turnCount !== undefined && task.turnCount > 0 ? ` t${task.turnCount}` : ''
+
+        const modeTag = task.forkContext ? theme.fg('accent', ' fork') : task.worktree ? theme.fg('accent', ' wt') : ''
+
+        const wtInfo =
+            task.worktree && task.status !== 'running'
+                ? task.changes?.length
+                    ? ` (${task.changes.length} changed)`
+                    : ' (clean)'
+                : ''
+
+        const agentColor = (
+            validColors.has(task.color ?? '') ? task.color : 'dim'
+        ) as import('@earendil-works/pi-coding-agent').ThemeColor
+        const name = task.agent ? `${task.label} ${theme.fg(agentColor, `· ${task.agent}`)}` : task.label
+
+        const tag = theme.fg('dim', `${duration ?? ''}${turnInfo}${tokenInfo}${costInfo}`)
+        const errSnippet =
+            task.status === 'failed' && task.error
+                ? theme.fg('error', ` ${truncate(task.error.split('\n')[0]?.split('. ')[0]?.trim() ?? task.error, 60)}`)
+                : ''
+        lines.push(
+            `${marker}${statusIcon(task.status, theme)} ${name}${modeTag} ${tag}${wtInfo ? theme.fg('dim', wtInfo) : ''}${errSnippet}`
+        )
+        if (task.status === 'running' && task.activity) lines.push(`    ${theme.fg('dim', task.activity)}`)
+    })
+    if (highlightIndex !== null) lines.push(theme.fg('muted', '↑/↓ navigate · enter view · d dismiss · esc close'))
+    return lines
+}
+
+const CHROME_LINES = 4
+const MIN_VIEWPORT = 5
+
+export class SubagentViewer implements Component {
+    private raw: string[] = []
+    private scroll = 0
+    private renderedBodyLength = 0
+    private autoScroll = true
+    private unsubscribe: () => void
+    private tui: TUI
+    private theme: Theme
+    private session: AgentSession
+    private task: TaskDetails
+    private done: (result: void) => void
+
+    constructor(tui: TUI, theme: Theme, session: AgentSession, task: TaskDetails, done: (result: void) => void) {
+        this.tui = tui
+        this.theme = theme
+        this.session = session
+        this.task = task
+        this.done = done
+        this.rebuild()
+        this.unsubscribe = session.subscribe((event) => {
+            if (
+                event.type === 'message_end' ||
+                event.type === 'tool_execution_start' ||
+                event.type === 'compaction_end' ||
+                event.type === 'entry_appended'
+            ) {
+                this.rebuild()
+                this.tui.requestRender()
+            }
+        })
+    }
+
+    private rebuild(): void {
+        this.raw = buildTranscriptLines(this.session.messages, this.theme)
+        if (this.raw.length === 0 && this.task.error) this.raw = [this.theme.fg('error', this.task.error)]
+    }
+
+    private viewportHeight(): number {
+        const rows = this.tui.terminal.rows || 24
+        return Math.max(MIN_VIEWPORT, Math.floor(rows * 0.7) - CHROME_LINES)
+    }
+
+    private wrapped(width: number): string[] {
+        return this.raw.flatMap((line) => (line ? wrapTextWithAnsi(line, width) : ['']))
+    }
+
+    render(width: number): string[] {
+        const theme = this.theme
+        const inner = Math.max(10, width - 2)
+        const body = this.wrapped(inner)
+        this.renderedBodyLength = body.length
+        const height = this.viewportHeight()
+        if (this.autoScroll) this.scroll = Math.max(0, body.length - height)
+        this.scroll = Math.min(this.scroll, Math.max(0, body.length - height))
+
+        const task = this.task
+        const duration = formatDuration(task.startedAt, task.finishedAt)
+        const header = truncateToWidth(
+            `${statusIcon(task.status, theme)} ${theme.bold(task.label)} ${theme.fg('dim', `${task.provider}/${task.model} (${task.thinking})${duration ? ` · ${duration}` : ''}`)}`,
+            width
+        )
+        const footer = theme.fg('muted', '↑/↓ scroll · pgup/pgdn page · esc close')
+
+        const visible = body.slice(this.scroll, this.scroll + height)
+        while (visible.length < height) visible.push('')
+
+        return [header, ''.padEnd(inner, '─'), ...visible, ''.padEnd(inner, '─'), footer]
+    }
+
+    handleInput(data: string): void {
+        if (isKeyRelease(data)) return
+        if (matchesKey(data, 'escape') || matchesKey(data, 'q')) {
+            this.unsubscribe()
+            this.done()
+            return
+        }
+        const height = this.viewportHeight()
+        const maxScroll = Math.max(0, this.renderedBodyLength - height)
+        if (matchesKey(data, 'up') || matchesKey(data, 'k')) {
+            this.autoScroll = false
+            this.scroll = Math.max(0, this.scroll - 1)
+        } else if (matchesKey(data, 'down') || matchesKey(data, 'j')) {
+            this.scroll = Math.min(maxScroll, this.scroll + 1)
+            this.autoScroll = this.scroll >= maxScroll
+        } else if (matchesKey(data, 'pageUp')) {
+            this.autoScroll = false
+            this.scroll = Math.max(0, this.scroll - height)
+        } else if (matchesKey(data, 'pageDown')) {
+            this.scroll = Math.min(maxScroll, this.scroll + height)
+            this.autoScroll = this.scroll >= maxScroll
+        } else {
+            return
+        }
+        this.tui.requestRender()
+    }
+
+    invalidate(): void {
+        this.rebuild()
+    }
+}