mirror of
https://github.com/Ark0N/Codeman.git
synced 2026-10-04 22:49:41 +02:00
Codeman maintainer
f0138fbe47
One absolutely-addressed line per row, each closed with an erase-to-end, so nothing scrolls and a repaint cannot leave the previous frame's tail behind. The caller wraps the result in synchronized-output brackets; that is an IO decision and stays out of the renderer. Color is passed in rather than detected. chalk's detection is right for the one-shot CLI but would make a frame non-deterministic, so the palette is raw SGR in the same semantic roles cli-style uses, and `color: false` emits nothing but the cursor addressing, the session's own colors in the preview included. Co-Authored-By: Claude Fable 5 <noreply@anthropic.com>
404 lines
16 KiB
TypeScript
404 lines
16 KiB
TypeScript
/**
|
||
* @fileoverview Unit tests for the frame renderer.
|
||
*
|
||
* The structural expectations below are full frames with the escapes stripped,
|
||
* which is what makes a layout regression readable in a diff; the escape
|
||
* sequences themselves are asserted separately, including the promise that
|
||
* NO_COLOR leaves nothing but cursor addressing behind.
|
||
*/
|
||
import { describe, it, expect } from 'vitest';
|
||
import { stripStyles, toDisplayLines, visibleWidth } from '../../src/tui/tui-ansi.js';
|
||
import { computeLayout, needsBanner } from '../../src/tui/tui-layout.js';
|
||
import { createTuiModel, type TuiModelStore } from '../../src/tui/tui-model.js';
|
||
import {
|
||
detectGlyphTier,
|
||
formatElapsed,
|
||
formatTokens,
|
||
renderFrame,
|
||
rowLabel,
|
||
type TuiRenderOptions,
|
||
} from '../../src/tui/tui-render.js';
|
||
|
||
const NOW = 1_700_000_000_000;
|
||
|
||
const PLAIN: TuiRenderOptions = { color: false, glyphs: 'unicode', tick: 4, now: NOW };
|
||
|
||
function fixture(): TuiModelStore {
|
||
const model = createTuiModel();
|
||
model.setHeader({ hostname: 'tnode', version: '1.19.0', planUsage: '5h 32% wk 61%' });
|
||
model.replaceSessions([
|
||
{
|
||
sessionId: 'aaa1',
|
||
name: 'w4-api-refactor',
|
||
mode: 'claude',
|
||
status: 'busy',
|
||
isWorking: true,
|
||
workingDir: '/home/dev/api',
|
||
createdAt: NOW - 9_000_000,
|
||
lastActivityAt: NOW - 1_000,
|
||
lastSubmitAt: NOW - 134_000,
|
||
inputTokens: 9_000,
|
||
outputTokens: 3_300,
|
||
sources: ['live', 'persisted'],
|
||
},
|
||
{
|
||
sessionId: 'bbb2',
|
||
name: 'w6-docs',
|
||
mode: 'claude',
|
||
status: 'idle',
|
||
workingDir: '/home/dev/docs',
|
||
createdAt: NOW - 8_000_000,
|
||
lastActivityAt: NOW - 660_000,
|
||
sources: ['live'],
|
||
},
|
||
{
|
||
sessionId: 'ccc3',
|
||
name: 'w1-codeman',
|
||
mode: 'claude',
|
||
status: 'busy',
|
||
isWorking: true,
|
||
workingDir: '/home/dev/codeman',
|
||
createdAt: NOW - 10_000_000,
|
||
lastActivityAt: NOW,
|
||
lastSubmitAt: NOW - 1_020_000,
|
||
inputTokens: 40_000,
|
||
outputTokens: 5_200,
|
||
sources: ['live'],
|
||
},
|
||
{
|
||
sessionId: 'ddd4',
|
||
name: 'w2-gallery',
|
||
mode: 'codex',
|
||
status: 'idle',
|
||
workingDir: '/home/dev/gallery',
|
||
createdAt: NOW - 6_000_000,
|
||
lastActivityAt: NOW - 7_200_000,
|
||
sources: ['live'],
|
||
},
|
||
{
|
||
sessionId: 'eee5',
|
||
firstPrompt: 'fix the release script',
|
||
workingDir: '/home/dev/api',
|
||
lastActivityAt: NOW - 3 * 86_400_000,
|
||
sources: ['history'],
|
||
},
|
||
]);
|
||
model.setApprovals([
|
||
{ id: 'bbb2:1', sessionId: 'bbb2', sessionName: 'w6-docs', kind: 'idle', createdAt: NOW - 680_000 },
|
||
{
|
||
id: 'aaa1:2',
|
||
sessionId: 'aaa1',
|
||
sessionName: 'w4-api-refactor',
|
||
kind: 'permission',
|
||
createdAt: NOW - 120_000,
|
||
toolName: 'Bash',
|
||
},
|
||
]);
|
||
model.select('aaa1');
|
||
model.setPreview({
|
||
sessionId: 'aaa1',
|
||
lines: toDisplayLines('\x1b[32mActualizing...\x1b[0m (2m 14s)\nrunning tests\n\x1b[31mwarning\x1b[0m here\n'),
|
||
});
|
||
return model;
|
||
}
|
||
|
||
function render(model: TuiModelStore, cols: number, rows: number, opts: Partial<TuiRenderOptions> = {}): string {
|
||
const layout = computeLayout(cols, rows, { banner: needsBanner(model.connection) });
|
||
return renderFrame(model, layout, { ...PLAIN, ...opts });
|
||
}
|
||
|
||
/** The frame as visible lines: escapes stripped, trailing padding trimmed. */
|
||
function frameLines(frame: string): string[] {
|
||
return frame
|
||
.split(/\x1b\[\d+;1H/)
|
||
.slice(1)
|
||
.map((part) => stripStyles(part).trimEnd());
|
||
}
|
||
|
||
describe('renderFrame structure', () => {
|
||
it('paints the wide layout at 100x30', () => {
|
||
expect(frameLines(render(fixture(), 100, 30))).toEqual([
|
||
' codeman tnode · v1.19.0 · 4 sessions · 5h 32% wk 61% ? help q quit',
|
||
' NEEDS YOU ─────────────────────────│ w4-api-refactor · claude · /home/dev/api',
|
||
' 1 w6-docs ✋ 11m│ Actualizing... (2m 14s)',
|
||
'▶ 2 w4-api-refactor ⚠ 2m 12.3k│ running tests',
|
||
' WORKING ───────────────────────────│ warning here',
|
||
' 3 w1-codeman ✻ 17m 45.2k│',
|
||
' IDLE ──────────────────────────────│',
|
||
' 4 w2-gallery codex ○ 2h│',
|
||
' RECENT ────────────────────────────│',
|
||
' 5 fix the release script ✔ 3d│',
|
||
' │',
|
||
' │',
|
||
' │',
|
||
' │',
|
||
' │',
|
||
' │',
|
||
' │',
|
||
' │',
|
||
' │',
|
||
' │',
|
||
' │',
|
||
' │',
|
||
' │',
|
||
' │',
|
||
' │',
|
||
' │',
|
||
' │',
|
||
' │',
|
||
' │',
|
||
' ↑↓ select · ⏎ attach · 1-9 jump · y/n answer · p prompt · n new · x kill · / search · g digest · ?',
|
||
]);
|
||
});
|
||
|
||
it('paints the narrow two-line layout at 44x20', () => {
|
||
expect(frameLines(render(fixture(), 44, 20))).toEqual([
|
||
' codeman tnode · v1.19.0 · 4 sessions · 5h',
|
||
' NEEDS YOU ─────────────────────────────────',
|
||
' 1 w6-docs ✋ 11m',
|
||
' /home/dev/docs',
|
||
'▶ 2 w4-api-refactor ⚠ 2m',
|
||
' /home/dev/api · 12.3k',
|
||
' WORKING ───────────────────────────────────',
|
||
' 3 w1-codeman ✻ 17m',
|
||
' /home/dev/codeman · 45.2k',
|
||
' IDLE ──────────────────────────────────────',
|
||
' 4 w2-gallery codex ○ 2h',
|
||
' /home/dev/gallery · codex',
|
||
' RECENT ────────────────────────────────────',
|
||
' 5 fix the release script ✔ 3d',
|
||
' /home/dev/api',
|
||
'',
|
||
'',
|
||
'',
|
||
'',
|
||
' ↑↓ select · ⏎ attach · 1-9 jump · y/n answe',
|
||
]);
|
||
});
|
||
|
||
it('addresses every line absolutely and erases its tail', () => {
|
||
const frame = render(fixture(), 100, 30);
|
||
const addresses = [...frame.matchAll(/\x1b\[(\d+);1H/g)].map((match) => Number(match[1]));
|
||
expect(addresses).toEqual(Array.from({ length: 30 }, (_, i) => i + 1));
|
||
expect(frame.split('\x1b[K')).toHaveLength(31);
|
||
expect(frame).not.toContain('\n');
|
||
});
|
||
|
||
it('never lets a line exceed the terminal width', () => {
|
||
for (const [cols, rows] of [
|
||
[100, 30],
|
||
[44, 20],
|
||
[72, 8],
|
||
[30, 6],
|
||
] as const) {
|
||
for (const line of frameLines(render(fixture(), cols, rows))) {
|
||
expect(visibleWidth(line)).toBeLessThanOrEqual(cols);
|
||
}
|
||
}
|
||
});
|
||
|
||
it('is deterministic for identical inputs', () => {
|
||
expect(render(fixture(), 100, 30)).toBe(render(fixture(), 100, 30));
|
||
});
|
||
|
||
it('degrades to a header-only frame on a 5x5 terminal without throwing', () => {
|
||
expect(() => render(fixture(), 5, 5)).not.toThrow();
|
||
expect(frameLines(render(fixture(), 5, 5))).toHaveLength(5);
|
||
});
|
||
});
|
||
|
||
describe('color', () => {
|
||
it('paints states and chrome when color is on', () => {
|
||
const model = fixture();
|
||
model.select('eee5');
|
||
const frame = render(model, 100, 30, { color: true });
|
||
expect(frame).toContain('\x1b[32m✻');
|
||
expect(frame).toContain('\x1b[31m⚠');
|
||
expect(frame).toContain('\x1b[33m✋');
|
||
expect(frame).toContain('\x1b[1mcodeman');
|
||
});
|
||
|
||
it('paints the selected row as one inverse block with no styling inside it', () => {
|
||
// An inner reset would punch a hole in the highlight, so the selected row
|
||
// is built unpainted and wrapped instead.
|
||
const frame = render(fixture(), 100, 30, { color: true });
|
||
const highlighted = frame.split('\x1b[7m')[1]?.split('\x1b[0m')[0] ?? '';
|
||
expect(highlighted).toContain('w4-api-refactor');
|
||
expect(highlighted).toContain('⚠');
|
||
expect(frame).not.toContain('\x1b[31m⚠');
|
||
});
|
||
|
||
it('emits nothing but cursor addressing when color is off', () => {
|
||
const frame = render(fixture(), 100, 30, { color: false });
|
||
const withoutAddressing = frame.replace(/\x1b\[\d+;1H/g, '').replace(/\x1b\[K/g, '');
|
||
expect(withoutAddressing).not.toContain('\x1b');
|
||
});
|
||
|
||
it('strips the session own colors out of the preview under NO_COLOR', () => {
|
||
const model = fixture();
|
||
model.setPreview({ sessionId: 'aaa1', lines: toDisplayLines('\x1b[31mred tail\x1b[0m') });
|
||
expect(render(model, 100, 30, { color: false })).not.toContain('\x1b[31m');
|
||
expect(render(model, 100, 30, { color: true })).toContain('\x1b[31m');
|
||
});
|
||
});
|
||
|
||
describe('glyph tiers', () => {
|
||
it('falls back to bracketed ASCII tokens', () => {
|
||
const lines = frameLines(render(fixture(), 100, 30, { glyphs: 'ascii' }));
|
||
const list = lines.map((line) => line.split('|')[0]);
|
||
expect(list[2]).toContain('[w]');
|
||
expect(list[3]).toContain('[!]');
|
||
expect(list[5]).toContain('[*]');
|
||
expect(list[7]).toContain('[-]');
|
||
expect(list[9]).toContain('[v]');
|
||
expect(list[3].startsWith('>')).toBe(true);
|
||
expect(lines.join('')).not.toContain('✻');
|
||
expect(lines.join('')).not.toContain('─');
|
||
});
|
||
|
||
it('animates the working glyph with the tick', () => {
|
||
const model = fixture();
|
||
const frames = [0, 1, 2, 3, 4, 5].map((tick) => frameLines(render(model, 100, 30, { tick }))[5]);
|
||
expect(frames[0]).toContain('·');
|
||
expect(frames[1]).toContain('✢');
|
||
expect(frames[2]).toContain('✳');
|
||
expect(frames[3]).toContain('∗');
|
||
expect(frames[4]).toContain('✻');
|
||
expect(frames[5]).toContain('✽');
|
||
expect(new Set(frames).size).toBe(6);
|
||
});
|
||
|
||
it('detects a tier from the environment', () => {
|
||
expect(detectGlyphTier({ TERM: 'xterm-256color', LANG: 'en_US.UTF-8' })).toBe('unicode');
|
||
expect(detectGlyphTier({ TERM: 'xterm-kitty', LANG: 'en_US.UTF-8' })).toBe('nerd');
|
||
expect(detectGlyphTier({ TERM: 'xterm-256color', TERM_PROGRAM: 'iTerm.app', LANG: 'en_US.UTF-8' })).toBe('nerd');
|
||
expect(detectGlyphTier({ TERM: 'xterm-256color', LANG: 'C' })).toBe('ascii');
|
||
expect(detectGlyphTier({ TERM: 'dumb' })).toBe('ascii');
|
||
expect(detectGlyphTier({})).toBe('ascii');
|
||
expect(detectGlyphTier({ TERM: 'xterm-kitty', LANG: 'en_US.UTF-8', CODEMAN_TUI_GLYPHS: 'ascii' })).toBe('ascii');
|
||
});
|
||
});
|
||
|
||
describe('overlays', () => {
|
||
it('draws the help box over the body', () => {
|
||
const model = fixture();
|
||
model.setMode('help');
|
||
const lines = frameLines(render(model, 100, 30));
|
||
expect(lines.join('\n')).toContain('┌ Keys ');
|
||
expect(lines.some((line) => line.includes('attach'))).toBe(true);
|
||
expect(lines[lines.length - 1]).toContain('close');
|
||
});
|
||
|
||
it('draws the typed confirmation for a kill', () => {
|
||
const model = fixture();
|
||
model.beginConfirmKill(model.rows()[0]);
|
||
model.setConfirmInput('w6-d');
|
||
const text = frameLines(render(model, 100, 30)).join('\n');
|
||
expect(text).toContain('Kill w6-docs?');
|
||
expect(text).toContain('Type the name to confirm:');
|
||
expect(text).toContain('w6-d_');
|
||
expect(text).toContain('esc cancel');
|
||
});
|
||
|
||
it('draws a message box', () => {
|
||
const model = fixture();
|
||
model.setMessage({ text: 'session refused to start: tmux is not installed', tone: 'err' });
|
||
const text = frameLines(render(model, 100, 30)).join('\n');
|
||
expect(text).toContain('Error');
|
||
expect(text).toContain('tmux is not installed');
|
||
});
|
||
|
||
it('skips the overlay when the body is too small to hold a box', () => {
|
||
const model = fixture();
|
||
model.setMode('help');
|
||
expect(frameLines(render(model, 100, 4)).join('\n')).not.toContain('Keys');
|
||
});
|
||
});
|
||
|
||
describe('connection states', () => {
|
||
it('banners a degraded server and says the preview is gone', () => {
|
||
const model = fixture();
|
||
model.setConnection('degraded');
|
||
const lines = frameLines(render(model, 100, 30));
|
||
expect(lines[1]).toContain('server not running: attach only');
|
||
expect(lines.join('\n')).toContain('preview unavailable while the server is down');
|
||
});
|
||
|
||
it('banners reconnecting and down differently', () => {
|
||
const model = fixture();
|
||
model.setConnection('reconnecting');
|
||
expect(frameLines(render(model, 100, 30))[1]).toContain('reconnecting');
|
||
model.setConnection('down');
|
||
expect(frameLines(render(model, 100, 30))[1]).toContain('server unreachable');
|
||
});
|
||
});
|
||
|
||
describe('empty and partial states', () => {
|
||
it('shows the empty hint when there are no sessions', () => {
|
||
const model = createTuiModel();
|
||
const lines = frameLines(render(model, 100, 30));
|
||
expect(lines.join('\n')).toContain('No sessions. n to start one, q to quit.');
|
||
expect(lines[1]).not.toContain('NEEDS YOU');
|
||
});
|
||
|
||
it('says the preview is still loading when it belongs to another session', () => {
|
||
const model = fixture();
|
||
model.setPreview({ sessionId: 'bbb2', lines: ['other session'] });
|
||
const text = frameLines(render(model, 100, 30)).join('\n');
|
||
expect(text).toContain('loading preview…');
|
||
expect(text).not.toContain('other session');
|
||
});
|
||
|
||
it('surfaces a preview error instead of a stale tail', () => {
|
||
const model = fixture();
|
||
model.setPreview({ sessionId: 'aaa1', lines: [], error: 'terminal capture failed' });
|
||
expect(frameLines(render(model, 100, 30)).join('\n')).toContain('terminal capture failed');
|
||
});
|
||
|
||
it('scrolls the list so the selected row stays visible', () => {
|
||
const model = createTuiModel();
|
||
model.replaceSessions(
|
||
Array.from({ length: 30 }, (_, i) => ({
|
||
sessionId: `s${String(i).padStart(2, '0')}`,
|
||
name: `session-${String(i).padStart(2, '0')}`,
|
||
sources: ['live'],
|
||
status: 'idle',
|
||
lastActivityAt: NOW - i * 1000,
|
||
}))
|
||
);
|
||
model.select('s29');
|
||
const text = frameLines(render(model, 100, 12)).join('\n');
|
||
expect(text).toContain('session-29');
|
||
expect(text).not.toContain('session-00');
|
||
});
|
||
});
|
||
|
||
describe('formatting helpers', () => {
|
||
it('formats elapsed time compactly', () => {
|
||
expect(formatElapsed(0)).toBe('0s');
|
||
expect(formatElapsed(45_000)).toBe('45s');
|
||
expect(formatElapsed(11 * 60_000)).toBe('11m');
|
||
expect(formatElapsed(2 * 3_600_000)).toBe('2h');
|
||
expect(formatElapsed(3 * 86_400_000)).toBe('3d');
|
||
expect(formatElapsed(-1)).toBe('');
|
||
expect(formatElapsed(Number.NaN)).toBe('');
|
||
});
|
||
|
||
it('formats token counts compactly', () => {
|
||
expect(formatTokens(0)).toBe('');
|
||
expect(formatTokens(842)).toBe('842');
|
||
expect(formatTokens(45_200)).toBe('45.2k');
|
||
expect(formatTokens(45_000)).toBe('45k');
|
||
expect(formatTokens(1_200_000)).toBe('1.2M');
|
||
});
|
||
|
||
it('names a row the way the web history list does', () => {
|
||
expect(rowLabel({ sessionId: 'abcdef12', name: 'w1', sources: [] })).toBe('w1');
|
||
expect(rowLabel({ sessionId: 'abcdef12', firstPrompt: 'do the thing', sources: [] })).toBe('do the thing');
|
||
expect(rowLabel({ sessionId: 'abcdef12', firstPrompt: '(no content)', workingDir: '/a/b/case', sources: [] })).toBe(
|
||
'case'
|
||
);
|
||
expect(rowLabel({ sessionId: 'abcdef1234', sources: [] })).toBe('abcdef12');
|
||
});
|
||
});
|