@zhushanwen/pi-session-reader 0.13.0 → 0.13.2

package/src/__tests__/summary.test.ts

@@ -1,223 +0,0 @@
-import type { Theme } from '@earendil-works/pi-coding-agent'
-import { describe, expect, it } from 'vitest'
-
-import type { SessionReadParams, ToolResult } from '../tool-handler.js'
-import { buildSessionReadSummary, withResultSummary } from '../summary.js'
-import { renderSessionReadResult } from '../tui/tool-render.js'
-
-// ---- fixtures ----
-
-function params(overrides: Partial<SessionReadParams> & { action: SessionReadParams['action'] }): SessionReadParams {
-  return overrides as SessionReadParams
-}
-
-function toolResult(details: unknown, text = 'body text'): ToolResult {
-  return { content: [{ type: 'text', text }], details }
-}
-
-// ---- buildSessionReadSummary ----
-
-describe('buildSessionReadSummary', () => {
-  it('find → Found N sessions matching "<query>"（截断带后缀）', () => {
-    const r = toolResult({ matches: [{}, {}, {}], truncated: false })
-    expect(buildSessionReadSummary(params({ action: 'find', query: 'login' }), r)).toBe(
-      'Found 3 sessions matching "login"',
-    )
-    const truncated = toolResult({ matches: [{}], truncated: true })
-    expect(buildSessionReadSummary(params({ action: 'find', query: 'login' }), truncated)).toBe(
-      'Found 1 sessions matching "login" (truncated)',
-    )
-  })
-
-  it('find 零匹配 → No sessions matching', () => {
-    const r = toolResult({ matches: [], truncated: false })
-    expect(buildSessionReadSummary(params({ action: 'find', query: 'gone' }), r)).toBe(
-      'No sessions matching "gone"',
-    )
-  })
-
-  it('ambiguous（任意 action 的多匹配消歧）→ N candidates', () => {
-    const r = toolResult({ ambiguous: true, candidates: [{}, {}] })
-    expect(buildSessionReadSummary(params({ action: 'outline', session: 'e6c9' }), r)).toBe(
-      'Ambiguous session "e6c9": 2 candidates',
-    )
-  })
-
-  it('family flat → N relations（parents/forks/subagents/workflows 合计）', () => {
-    const family = {
-      root: { sessionId: 's1' },
-      parents: [{}],
-      forks: [{}, {}],
-      subagents: [{}],
-      workflows: [],
-    }
-    const r = toolResult(family)
-    expect(buildSessionReadSummary(params({ action: 'family', session: 's1' }), r)).toBe(
-      'Family of session s1: 4 relations',
-    )
-  })
-
-  it('family recursive（details.tree）→ Execution tree', () => {
-    const r = toolResult({ tree: { root: {} } })
-    expect(buildSessionReadSummary(params({ action: 'family', session: 's1', recursive: true }), r)).toBe(
-      'Execution tree of session s1',
-    )
-  })
-
-  it('outline → N turns（stats.totalTurns）', () => {
-    const r = toolResult({ turns: [], lines: [], stats: { totalTurns: 32, totalEntries: 90, totalBytes: 1, parsedBytes: 1, skippedLines: 0 }, tokenEstimate: 1 })
-    expect(buildSessionReadSummary(params({ action: 'outline', session: 's1' }), r)).toBe(
-      'Outline of session s1 · 32 turns',
-    )
-  })
-
-  it('expand → T 序号补零 + N entries', () => {
-    const r = toolResult({ turn: { index: 13 }, entries: [{}, {}, {}, {}] })
-    expect(buildSessionReadSummary(params({ action: 'expand', session: 's1', turn: 'T013' }), r)).toBe(
-      'Expanded turn T013 · 4 entries',
-    )
-  })
-
-  it('detail → turns 范围标签 + N entries', () => {
-    const r = toolResult({ turns: 'T013-T015', entries: [{}, {}, {}] })
-    expect(buildSessionReadSummary(params({ action: 'detail', session: 's1', turns: 'T013-T015' }), r)).toBe(
-      'Read turns T013-T015 · 3 entries',
-    )
-  })
-
-  it('search → Found N hits / No hits', () => {
-    const r = toolResult({ hits: [{}, {}], truncated: false })
-    expect(buildSessionReadSummary(params({ action: 'search', session: 's1', pattern: 'fix' }), r)).toBe(
-      'Found 2 hits for "fix"',
-    )
-    const empty = toolResult({ hits: [], truncated: false })
-    expect(buildSessionReadSummary(params({ action: 'search', session: 's1', pattern: 'fix' }), empty)).toBe(
-      'No hits for "fix"',
-    )
-  })
-
-  it('export → format + sizeBytes', () => {
-    const r = toolResult({ path: '/tmp/x.md', sizeBytes: 2048 })
-    expect(buildSessionReadSummary(params({ action: 'export', session: 's1' }), r)).toBe(
-      'Exported outline view · 2048 bytes',
-    )
-    expect(buildSessionReadSummary(params({ action: 'export', session: 's1', format: 'full' }), r)).toBe(
-      'Exported full view · 2048 bytes',
-    )
-  })
-
-  it('extract → count / 零匹配 / 截断 showing', () => {
-    const r = toolResult({ what: 'user-messages', count: 26, shown: 26, truncated: false, items: [] })
-    expect(buildSessionReadSummary(params({ action: 'extract', session: 's1', what: 'user-messages' }), r)).toBe(
-      'Extracted 26 user-messages',
-    )
-    const empty = toolResult({ what: 'commands', count: 0, shown: 0, truncated: false, items: [] })
-    expect(buildSessionReadSummary(params({ action: 'extract', session: 's1', what: 'commands' }), empty)).toBe(
-      'No commands found',
-    )
-    const clipped = toolResult({ what: 'files', count: 500, shown: 120, truncated: true, items: [] })
-    expect(buildSessionReadSummary(params({ action: 'extract', session: 's1', what: 'files' }), clipped)).toBe(
-      'Extracted 500 files (showing 120)',
-    )
-  })
-
-  it('workflow → runs 计数 / 空 / skipped 后缀', () => {
-    expect(buildSessionReadSummary(params({ action: 'workflow', session: 's1' }), toolResult({ runs: [{}, {}], runIds: ['a', 'b'] }))).toBe(
-      'Found 2 workflow runs',
-    )
-    expect(buildSessionReadSummary(params({ action: 'workflow', session: 's1' }), toolResult({ runs: [], runIds: [] }))).toBe(
-      'No workflow runs',
-    )
-    expect(
-      buildSessionReadSummary(
-        params({ action: 'workflow', session: 's1' }),
-        toolResult({ runs: [{}], runIds: ['a'], skippedRuns: [{ runId: 'b', stateFile: 'x', reason: 'snapshot-unreadable' }] }),
-      ),
-    ).toBe('Found 1 workflow runs · 1 skipped')
-  })
-
-  it('result 单条 → session + totalChars（截断后缀）', () => {
-    const r = toolResult({ session: 'sa-abc', sessionId: 'uuid', sessionFile: '/x', totalChars: 8000, truncated: true, text: '...' })
-    expect(buildSessionReadSummary(params({ action: 'result', session: 'sa-abc' }), r)).toBe(
-      'Read final result of sa-abc · 8000 chars (truncated)',
-    )
-  })
-
-  it('result 批量 → count + items 形态识别', () => {
-    const r = toolResult({ count: 3, items: [{}, {}, {}] })
-    expect(buildSessionReadSummary(params({ action: 'result', session: 'a,b,c' }), r)).toBe(
-      'Read 3 final results',
-    )
-  })
-
-  it('doctor → roots 计数', () => {
-    const r = toolResult({ environment: {}, roots: [{ kind: 'default' }, { kind: 'legacy' }] })
-    expect(buildSessionReadSummary(params({ action: 'doctor' }), r)).toBe(
-      'Doctor: 2 roots checked',
-    )
-  })
-
-  it('details 形态漂移（空对象）→ 各 action 兜底不抛错', () => {
-    const empty = toolResult({})
-    expect(buildSessionReadSummary(params({ action: 'outline', session: 's1' }), empty)).toBe(
-      'Outline of session s1',
-    )
-    expect(buildSessionReadSummary(params({ action: 'workflow', session: 's1' }), empty)).toBe(
-      'No workflow runs',
-    )
-    expect(buildSessionReadSummary(params({ action: 'doctor' }), empty)).toBe('Doctor check completed')
-    expect(buildSessionReadSummary(params({ action: 'extract', what: 'files' }), empty)).toBe('No files found')
-  })
-})
-
-// ---- withResultSummary ----
-
-describe('withResultSummary', () => {
-  it('details 追加 summary，现有字段保留', () => {
-    const r = toolResult({ hits: [{}], truncated: false })
-    const out = withResultSummary(params({ action: 'search', pattern: 'fix' }), r)
-    expect(out.details).toEqual({ hits: [{}], truncated: false, summary: 'Found 1 hits for "fix"' })
-  })
-})
-
-// ---- renderSessionReadResult（TUI renderResult）----
-
-const mockTheme = {
-  fg: (_color: string, text: string) => text,
-  bg: (_color: string, text: string) => text,
-  bold: (text: string) => text,
-  italic: (text: string) => text,
-  underline: (text: string) => text,
-  inverse: (text: string) => text,
-  strikethrough: (text: string) => text,
-  getFgAnsi: (_color: string) => '',
-  getBgAnsi: (_color: string) => '',
-  getColorMode: () => 'truecolor' as const,
-  getThinkingBorderColor: () => (text: string) => text,
-  getBashModeBorderColor: () => (text: string) => text,
-} as unknown as Theme
-
-function renderedText(result: unknown, expanded: boolean): string {
-  return renderSessionReadResult(result, { expanded }, mockTheme)
-    .render(400)
-    .map((line) => line.trimEnd())
-    .join('\n')
-}
-
-describe('renderSessionReadResult', () => {
-  it('折叠态 → details.summary 一行', () => {
-    const r = toolResult({ matches: [], truncated: false }, 'multi\nline\nbody')
-    const withSummary = withResultSummary(params({ action: 'find', query: 'login' }), r)
-    expect(renderedText(withSummary, false)).toBe('No sessions matching "login"')
-  })
-
-  it('展开态 → content 纯文本（多行完整）', () => {
-    const r = toolResult({ matches: [] }, 'line1\nline2')
-    expect(renderedText(r, true)).toBe('line1\nline2')
-  })
-
-  it('无 summary（旧 entry）→ 折叠态回退 content 首行', () => {
-    const r = toolResult({}, 'Header line\ndetail body')
-    expect(renderedText(r, false)).toBe('Header line')
-  })
-})

package/src/__tests__/turns.test.ts

@@ -1,144 +0,0 @@
-import { describe, it, expect } from 'vitest'
-import { segmentTurns } from '../core/turns.js'
-import type { Entry } from '@zhushanwen/session-core'
-import { buildTreeView } from '../core/tree.js'
-
-// ---- Entry 构造助手（turns.ts 只消费 type/id/parentId/message.role） ----
-
-function msg(
-  id: string,
-  parentId: string | null,
-  role: 'user' | 'assistant' | 'toolResult',
-): Entry {
-  return { type: 'message', id, parentId, message: { role, content: '' } }
-}
-
-function entry(id: string, parentId: string | null, type: string): Entry {
-  return { type, id, parentId }
-}
-
-describe('segmentTurns', () => {
-  it('1. user 开 turn，后续 assistant/toolResult 并入同一 turn', () => {
-    const entries = [msg('U', null, 'user'), msg('A', 'U', 'assistant'), msg('T', 'A', 'toolResult')]
-    const turns = segmentTurns(entries, new Set(['U', 'A', 'T']))
-
-    expect(turns).toHaveLength(1)
-    expect(turns[0].index).toBe(0)
-    expect(turns[0].isCompaction).toBe(false)
-    expect(turns[0].userEntry?.id).toBe('U')
-    expect(turns[0].entries.map((e) => e.id)).toEqual(['U', 'A', 'T'])
-  })
-
-  it('2. compaction 开新 turn（isCompaction=true，无 userEntry），后续 user 又开一个 turn', () => {
-    const entries = [
-      msg('U', null, 'user'),
-      msg('A', 'U', 'assistant'),
-      entry('C', 'A', 'compaction'),
-      msg('U2', 'C', 'user'),
-    ]
-    const turns = segmentTurns(entries, new Set(['U', 'A', 'C', 'U2']))
-
-    expect(turns).toHaveLength(3)
-    // turn 0：user + assistant
-    expect(turns[0].userEntry?.id).toBe('U')
-    expect(turns[0].isCompaction).toBe(false)
-    expect(turns[0].entries.map((e) => e.id)).toEqual(['U', 'A'])
-    // turn 1：compaction 独立成 turn
-    expect(turns[1].isCompaction).toBe(true)
-    expect(turns[1].userEntry).toBeUndefined()
-    expect(turns[1].entries.map((e) => e.id)).toEqual(['C'])
-    // turn 2：新 user
-    expect(turns[2].userEntry?.id).toBe('U2')
-    expect(turns[2].isCompaction).toBe(false)
-  })
-
-  it('3. model_change / thinking_level_change / custom 并入当前 turn', () => {
-    const entries = [
-      msg('U', null, 'user'),
-      entry('M', 'U', 'model_change'),
-      entry('T', 'M', 'thinking_level_change'),
-      entry('X', 'T', 'custom'),
-    ]
-    const turns = segmentTurns(entries, new Set(['U', 'M', 'T', 'X']))
-
-    expect(turns).toHaveLength(1)
-    expect(turns[0].userEntry?.id).toBe('U')
-    expect(turns[0].entries.map((e) => e.id)).toEqual(['U', 'M', 'T', 'X'])
-  })
-
-  it('4. branch entry（不在 leafSet）不计入', () => {
-    const entries = [
-      msg('U', null, 'user'),
-      msg('A', 'U', 'assistant'),
-      msg('B1', 'U', 'assistant'), // 旁支：不在 leafSet
-      msg('B2', 'B1', 'toolResult'), // 旁支子节点
-    ]
-    const turns = segmentTurns(entries, new Set(['U', 'A']))
-
-    expect(turns).toHaveLength(1)
-    expect(turns[0].entries.map((e) => e.id)).toEqual(['U', 'A'])
-  })
-
-  it('5. 孤儿 assistant（首个 entry 是 assistant 无前置 user）→ 并入 turn 0（preface）', () => {
-    const entries = [msg('O', null, 'assistant'), msg('U', 'O', 'user')]
-    const turns = segmentTurns(entries, new Set(['O', 'U']))
-
-    expect(turns).toHaveLength(2)
-    // turn 0：preface（孤儿 assistant），无 userEntry，非 compaction
-    expect(turns[0].isCompaction).toBe(false)
-    expect(turns[0].userEntry).toBeUndefined()
-    expect(turns[0].entries.map((e) => e.id)).toEqual(['O'])
-    // turn 1：首个 user
-    expect(turns[1].userEntry?.id).toBe('U')
-    expect(turns[1].index).toBe(1)
-  })
-
-  it('6. session header 被忽略（即使在 leafSet 中也不计 turn）', () => {
-    const entries = [
-      entry('S', null, 'session'),
-      msg('U', 'S', 'user'),
-      msg('A', 'U', 'assistant'),
-    ]
-    // S 也在 leafSet，验证规则 1 优先于规则 5
-    const turns = segmentTurns(entries, new Set(['S', 'U', 'A']))
-
-    expect(turns).toHaveLength(1)
-    expect(turns[0].entries.map((e) => e.id)).toEqual(['U', 'A'])
-  })
-
-  it('7. 空 entries → []', () => {
-    expect(segmentTurns([], new Set())).toEqual([])
-  })
-
-  it('startTime 取 turn 首条 entry 的 timestamp', () => {
-    const entries = [
-      { ...msg('U', null, 'user'), timestamp: '2026-05-28T03:17:12.844Z' },
-      msg('A', 'U', 'assistant'),
-    ]
-    const turns = segmentTurns(entries, new Set(['U', 'A']))
-    expect(turns[0].startTime).toBe('2026-05-28T03:17:12.844Z')
-  })
-
-  it('leaf 视图分段结构形态：preface（首 user 前的 model_change）+ user turns + compaction turn', () => {
-    // 合成分段结构形态（1 preface + 1 compaction + 2 user turn）：
-    // 规则 2（compaction 独立成 turn）+ 规则 4（首 user 前的 model_change/thinking_level_change
-    // 成 preface turn，无 userEntry）
-    const entries = [
-      entry('MC', null, 'model_change'),
-      msg('U1', 'MC', 'user'),
-      msg('A1', 'U1', 'assistant'),
-      entry('C1', 'A1', 'compaction'),
-      msg('U2', 'C1', 'user'),
-      msg('A2', 'U2', 'assistant'),
-    ]
-    const tree = buildTreeView(entries)
-    const turns = segmentTurns(entries, new Set(tree.leafPath))
-
-    expect(turns.length).toBe(4)
-    // 首 turn 是 preface（model_change），无 userEntry、非 compaction
-    expect(turns[0].userEntry).toBeUndefined()
-    expect(turns[0].isCompaction).toBe(false)
-    expect(turns.filter((t) => t.isCompaction).length).toBe(1)
-    expect(turns.filter((t) => t.userEntry !== undefined).length).toBe(2)
-  })
-})