bashrusakh
59ecd86b4b
Reworks the chat and session-sidebar render paths to cut render cascades, memory
churn, and UI jank on large sessions and big session trees. Behavior is preserved;
the changes are about *when* and *how much* the UI re-renders.
## Chat streaming
- Freeze the streaming message's parts in the bulk turn projection during streaming,
and re-inject live parts only in an isolated tail leaf, so a ~60/sec delta stream
no longer re-runs the whole-session projection or re-renders unrelated rows.
session with referential reuse of unchanged turns.
- Memoize message rows with field-aware comparators instead of reference equality.
- Replace the manual child-session polling in the task tool with the live SSE
stream + a one-shot load, removing a fetch/settle state machine.
## History loading & scroll
- Load an initial page fast, then prepend one older page in the background so the
scroll container has headroom and "load older on scroll-up" fires before the user
hits the absolute top.
- Compensate scroll synchronously (in a layout effect, before paint) for prepends —
including background prepends that don't originate from a user scroll — so the
viewport stays stable instead of judder-correcting on the next frame.
## Markdown rendering
- Render markdown synchronously *styled* on first paint (paragraphs, lists, code
cards, tables, inline code) instead of raw escaped text; the async pass then only
upgrades syntax-highlight colors. Eliminates the flash of full-width raw text.
- Load KaTeX CSS eagerly with the main bundle instead of inside the lazy markdown
chunk, avoiding a late stylesheet injection on first render.
## Sidebar
- Hoist per-row recursive tree walks out of row comparators into per-group
precomputed sets/keys; batch live-session lookups into a single map; add a
group-level memo boundary.
- Isolate rename drafts so per-keystroke typing doesn't repaint the row tree.
## Sync layer
- Add a staleness guard so a slow message fetch can't repopulate a session the user
navigated away from.
- Throw on fetch failure for authoritative loaders so a transient blip can't read as
an empty server response.
## Cleanup
- Remove dead code (unused hooks, params, duplicated inline types) surfaced while
reworking the above.
## Known issue
- A rare, purely cosmetic first-paint width flash can still appear on large sessions;
it has no behavioral or data impact and is tracked for a follow-up runtime trace.
90 lines
3.1 KiB
TypeScript
90 lines
3.1 KiB
TypeScript
import { describe, expect, test } from 'bun:test';
|
|
import type { Message, Part } from '@opencode-ai/sdk/v2/client';
|
|
import type { State } from '@/sync/types';
|
|
|
|
import { EMPTY_REVERTED_MESSAGE_DOCK_STATE, buildRevertedMessageDockState } from './revertedMessageDockState';
|
|
|
|
const message = (id: string, role: 'user' | 'assistant'): Message => ({
|
|
id,
|
|
role,
|
|
sessionID: 'ses_1',
|
|
time: { created: 1 },
|
|
} as Message);
|
|
|
|
const textPart = (id: string, text: string): Part => ({
|
|
id,
|
|
type: 'text',
|
|
text,
|
|
} as Part);
|
|
|
|
const state = (partial: Partial<State>): Pick<State, 'session' | 'message' | 'part'> => ({
|
|
session: [],
|
|
message: {},
|
|
part: {},
|
|
...partial,
|
|
});
|
|
|
|
describe('buildRevertedMessageDockState', () => {
|
|
test('returns a shared empty state when the session is not reverted', () => {
|
|
const first = buildRevertedMessageDockState(state({}), 'ses_1');
|
|
const second = buildRevertedMessageDockState(
|
|
state({ part: { assistant_1: [textPart('part_1', 'streaming')] } }),
|
|
'ses_1',
|
|
first,
|
|
);
|
|
|
|
expect(first).toBe(EMPTY_REVERTED_MESSAGE_DOCK_STATE);
|
|
expect(second).toBe(EMPTY_REVERTED_MESSAGE_DOCK_STATE);
|
|
});
|
|
|
|
test('reuses the previous state when unrelated assistant parts change', () => {
|
|
const user = message('user_1', 'user');
|
|
const userParts = [textPart('part_user', 'hello')];
|
|
const first = buildRevertedMessageDockState(
|
|
state({
|
|
session: [{ id: 'ses_1', revert: { messageID: 'user_1' } } as State['session'][number]],
|
|
message: { ses_1: [user, message('assistant_1', 'assistant')] },
|
|
part: { user_1: userParts, assistant_1: [textPart('part_a', 'a')] },
|
|
}),
|
|
'ses_1',
|
|
);
|
|
|
|
const second = buildRevertedMessageDockState(
|
|
state({
|
|
session: [{ id: 'ses_1', revert: { messageID: 'user_1' } } as State['session'][number]],
|
|
message: { ses_1: [user, message('assistant_1', 'assistant')] },
|
|
part: { user_1: userParts, assistant_1: [textPart('part_a2', 'updated')] },
|
|
}),
|
|
'ses_1',
|
|
first,
|
|
);
|
|
|
|
expect(second).toBe(first);
|
|
});
|
|
|
|
test('updates when a reverted user message part changes', () => {
|
|
const user = message('user_1', 'user');
|
|
const first = buildRevertedMessageDockState(
|
|
state({
|
|
session: [{ id: 'ses_1', revert: { messageID: 'user_1' } } as State['session'][number]],
|
|
message: { ses_1: [user] },
|
|
part: { user_1: [textPart('part_user', 'hello')] },
|
|
}),
|
|
'ses_1',
|
|
);
|
|
|
|
const second = buildRevertedMessageDockState(
|
|
state({
|
|
session: [{ id: 'ses_1', revert: { messageID: 'user_1' } } as State['session'][number]],
|
|
message: { ses_1: [user] },
|
|
part: { user_1: [textPart('part_user_updated', 'updated')] },
|
|
}),
|
|
'ses_1',
|
|
first,
|
|
);
|
|
|
|
expect(second).not.toBe(first);
|
|
expect(second.records).toHaveLength(1);
|
|
});
|
|
});
|