mitchell/lmc
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

refactor: move containment logic into function

Browse source
This commit is contained in:
mitchell 2026-04-15 16:32:03 -04:00
parent 1f59cbca7b
commit 8aaa7f6931
1 changed files with 31 additions and 36 deletions
Download patch file Download diff file Expand all files Collapse all files

57
src/render/index.ts
View file

@ -5,63 +5,58 @@ const moveCursor = (x: number, y: number) =>
const clearScreenDown = () => const clearScreenDown = () =>
new Promise<void>((resolve) => process.stdout.clearScreenDown(resolve)); new Promise<void>((resolve) => process.stdout.clearScreenDown(resolve));
let lineCount = 0;
let charsSinceNL = 0;
function resetCounts() {
lineCount = 0;
charsSinceNL = 0;
}
export async function renderStream( export async function renderStream(
stream: AsyncIterableStream<TextStreamPart<ToolSet>>, stream: AsyncIterableStream<TextStreamPart<ToolSet>>,
) { ) {
let lineCount = 0;
let charsSinceNL = 0;
const [numColumns, numRows] = process.stdout.getWindowSize(); const [numColumns, numRows] = process.stdout.getWindowSize();
resetCounts();
for await (const part of stream) { for await (const part of stream) {
if (part.type === "reasoning-start") { if (part.type === "reasoning-start") {
process.stdout.write("\u001b[2m\n"); process.stdout.write("\u001b[2m\n");
} else if (part.type === "reasoning-delta") { } else if (part.type === "reasoning-delta") {
process.stdout.write(part.text); process.stdout.write(part.text);
await checkContainment(part.text, numColumns, numRows);
charsSinceNL += part.text.length;
if (part.text.includes("\n")) {
lineCount += part.text.match(/\n/g)?.length ?? 1;
charsSinceNL = 0;
} else if (charsSinceNL >= numColumns) {
lineCount++;
charsSinceNL = charsSinceNL - numColumns;
}
if (lineCount >= numRows / 2) {
await moveCursor(0, -lineCount);
await clearScreenDown();
lineCount = 0;
charsSinceNL = 0;
}
} else if (part.type === "reasoning-end") { } else if (part.type === "reasoning-end") {
process.stdout.write("\n\u001b[22m"); process.stdout.write("\n\u001b[22m");
await moveCursor(0, -lineCount - 2); await moveCursor(0, -lineCount - 2);
await clearScreenDown(); await clearScreenDown();
lineCount = 0; resetCounts();
charsSinceNL = 0;
} else if (part.type === "text-start") { } else if (part.type === "text-start") {
process.stdout.write("\n"); process.stdout.write("\n");
} else if (part.type === "text-delta") { } else if (part.type === "text-delta") {
process.stdout.write(part.text); process.stdout.write(part.text);
await checkContainment(part.text, numColumns, numRows);
} else if (part.type === "text-end") {
await moveCursor(0, -lineCount - 1);
await clearScreenDown();
resetCounts();
}
}
}
charsSinceNL += part.text.length; async function checkContainment(text: string, columns: number, rows: number) {
if (part.text.includes("\n")) { charsSinceNL += text.length;
lineCount += part.text.match(/\n/g)?.length ?? 1; if (text.includes("\n")) {
lineCount += text.match(/\n/g)?.length ?? 1;
charsSinceNL = 0; charsSinceNL = 0;
} else if (charsSinceNL >= numColumns) { } else if (charsSinceNL >= columns) {
lineCount++; lineCount++;
charsSinceNL = charsSinceNL - numColumns; charsSinceNL = charsSinceNL - columns;
} }
if (lineCount >= numRows / 2) { if (lineCount >= rows / 2) {
await moveCursor(0, -lineCount); await moveCursor(0, -lineCount);
await clearScreenDown(); await clearScreenDown();
lineCount = 0; lineCount = 0;
charsSinceNL = 0; charsSinceNL = 0;
} }
} else if (part.type === "text-end") {
await moveCursor(0, -lineCount - 1);
await clearScreenDown();
}
}
} }