@herbertgao/pi-extensions 2026.9.6 → 2026.9.8

package/node_modules/@narumitw/pi-btw/src/bring-to-main.ts

@@ -1,632 +1,579 @@
 import type { KeybindingsManager, Theme } from "@earendil-works/pi-coding-agent";
-import {
-	type Component,
-	Key,
-	matchesKey,
-	type TUI,
-	truncateToWidth,
-	visibleWidth,
-} from "@earendil-works/pi-tui";
+import { type Component, Key, matchesKey, type TUI, truncateToWidth, visibleWidth } from "@earendil-works/pi-tui";
 import type { SideThreadTurn } from "./side-thread.js";
 
 const RESERVED_APP_ROWS = 3;
 const GRAPHEME_SEGMENTER = new Intl.Segmenter(undefined, { granularity: "grapheme" });
 
 export interface BtwBringToMainSegment {
-	role: "user" | "assistant";
-	text: string;
+  role: "user" | "assistant";
+  text: string;
 }
 
 export interface BtwBringToMainSummary {
-	lines: number;
-	messages: number;
-	tokens: number;
+  lines: number;
+  messages: number;
+  tokens: number;
 }
 
 export interface BtwSelectionLine {
-	role: BtwBringToMainSegment["role"];
-	text: string;
+  role: BtwBringToMainSegment["role"];
+  text: string;
 }
 
 export interface BtwTextPosition {
-	line: number;
-	column: number;
+  line: number;
+  column: number;
 }
 
 export interface BtwTextRangeSelectorState {
-	cursor: BtwTextPosition;
-	anchor?: BtwTextPosition;
-	lineAnchor?: number;
-	preferredColumn: number;
-	scrollOffset: number;
-	horizontalOffset: number;
+  cursor: BtwTextPosition;
+  anchor?: BtwTextPosition;
+  lineAnchor?: number;
+  preferredColumn: number;
+  scrollOffset: number;
+  horizontalOffset: number;
 }
 
 export type BtwQuickBringToMainScope =
-	| { kind: "latest" }
-	| { kind: "from"; answeredTurnIndex: number }
-	| { kind: "entire" };
+  | { kind: "latest" }
+  | { kind: "from"; answeredTurnIndex: number }
+  | { kind: "entire" };
 
 export type BtwTextRangeSelectorAction =
-	| { kind: "confirm"; segments: BtwBringToMainSegment[] }
-	| { kind: "back" }
-	| { kind: "close" };
-
-export function getAnsweredTurns(
-	turns: readonly SideThreadTurn[],
-): Array<Extract<SideThreadTurn, { kind: "answered" }>> {
-	return turns.filter(
-		(turn): turn is Extract<SideThreadTurn, { kind: "answered" }> => turn.kind === "answered",
-	);
+  | { kind: "confirm"; segments: BtwBringToMainSegment[] }
+  | { kind: "back" }
+  | { kind: "close" };
+
+export function getAnsweredTurns(turns: readonly SideThreadTurn[]): Extract<SideThreadTurn, { kind: "answered" }>[] {
+  return turns.filter((turn): turn is Extract<SideThreadTurn, { kind: "answered" }> => turn.kind === "answered");
 }
 
 export function buildQuickBringToMainSegments(
-	turns: readonly SideThreadTurn[],
-	scope: BtwQuickBringToMainScope,
+  turns: readonly SideThreadTurn[],
+  scope: BtwQuickBringToMainScope,
 ): BtwBringToMainSegment[] {
-	const answered = getAnsweredTurns(turns);
-	const selected =
-		scope.kind === "latest"
-			? answered.slice(-1)
-			: scope.kind === "from"
-				? answered.slice(Math.max(0, scope.answeredTurnIndex))
-				: answered;
-	return selected.flatMap((turn) => [
-		{ role: "user" as const, text: turn.question },
-		{ role: "assistant" as const, text: turn.answer },
-	]);
+  const answered = getAnsweredTurns(turns);
+  const selected =
+    scope.kind === "latest"
+      ? answered.slice(-1)
+      : scope.kind === "from"
+        ? answered.slice(Math.max(0, scope.answeredTurnIndex))
+        : answered;
+  return selected.flatMap((turn) => [
+    { role: "user" as const, text: turn.question },
+    { role: "assistant" as const, text: turn.answer },
+  ]);
 }
 
 export function buildBtwSelectionLines(turns: readonly SideThreadTurn[]): BtwSelectionLine[] {
-	return buildQuickBringToMainSegments(turns, { kind: "entire" }).flatMap((segment) =>
-		segment.text.split("\n").map((text) => ({ role: segment.role, text })),
-	);
+  return buildQuickBringToMainSegments(turns, { kind: "entire" }).flatMap((segment) =>
+    segment.text.split("\n").map((text) => ({ role: segment.role, text })),
+  );
 }
 
 export function segmentsFromLineRange(
-	lines: readonly BtwSelectionLine[],
-	anchor: number,
-	cursor: number,
+  lines: readonly BtwSelectionLine[],
+  anchor: number,
+  cursor: number,
 ): BtwBringToMainSegment[] {
-	if (lines.length === 0) return [];
-	const start = Math.max(0, Math.min(anchor, cursor, lines.length - 1));
-	const end = Math.max(0, Math.min(Math.max(anchor, cursor), lines.length - 1));
-	const segments: BtwBringToMainSegment[] = [];
-	for (const line of lines.slice(start, end + 1)) {
-		const previous = segments.at(-1);
-		if (previous?.role === line.role) {
-			previous.text += `\n${line.text}`;
-		} else {
-			segments.push({ role: line.role, text: line.text });
-		}
-	}
-	return segments;
+  if (lines.length === 0) return [];
+  const start = Math.max(0, Math.min(anchor, cursor, lines.length - 1));
+  const end = Math.max(0, Math.min(Math.max(anchor, cursor), lines.length - 1));
+  const segments: BtwBringToMainSegment[] = [];
+  for (const line of lines.slice(start, end + 1)) {
+    const previous = segments.at(-1);
+    if (previous?.role === line.role) {
+      previous.text += `\n${line.text}`;
+    } else {
+      segments.push({ role: line.role, text: line.text });
+    }
+  }
+  return segments;
 }
 
 export function segmentsFromTextRange(
-	lines: readonly BtwSelectionLine[],
-	anchor: BtwTextPosition,
-	cursor: BtwTextPosition,
+  lines: readonly BtwSelectionLine[],
+  anchor: BtwTextPosition,
+  cursor: BtwTextPosition,
 ): BtwBringToMainSegment[] {
-	if (lines.length === 0) return [];
-	const first = clampTextPosition(lines, anchor);
-	const second = clampTextPosition(lines, cursor);
-	const [start, end] = compareTextPositions(first, second) <= 0 ? [first, second] : [second, first];
-	if (compareTextPositions(start, end) === 0) return [];
-
-	const segments: BtwBringToMainSegment[] = [];
-	for (let lineIndex = start.line; lineIndex <= end.line; lineIndex += 1) {
-		const line = lines[lineIndex];
-		if (!line) continue;
-		const characters = splitGraphemes(line.text);
-		const from = lineIndex === start.line ? start.column : 0;
-		const to = lineIndex === end.line ? end.column : characters.length;
-		const text = characters.slice(from, to).join("");
-		if (text) {
-			const previous = segments.at(-1);
-			if (previous?.role === line.role) previous.text += text;
-			else segments.push({ role: line.role, text });
-		}
-		const crossesSameRoleLine = lineIndex < end.line && lines[lineIndex + 1]?.role === line.role;
-		if (crossesSameRoleLine) {
-			const current = segments.at(-1);
-			if (current?.role === line.role) current.text += "\n";
-			else segments.push({ role: line.role, text: "\n" });
-		}
-	}
-	return segments;
+  if (lines.length === 0) return [];
+  const first = clampTextPosition(lines, anchor);
+  const second = clampTextPosition(lines, cursor);
+  const [start, end] = compareTextPositions(first, second) <= 0 ? [first, second] : [second, first];
+  if (compareTextPositions(start, end) === 0) return [];
+
+  const segments: BtwBringToMainSegment[] = [];
+  for (let lineIndex = start.line; lineIndex <= end.line; lineIndex += 1) {
+    const line = lines[lineIndex];
+    if (!line) continue;
+    const characters = splitGraphemes(line.text);
+    const from = lineIndex === start.line ? start.column : 0;
+    const to = lineIndex === end.line ? end.column : characters.length;
+    const text = characters.slice(from, to).join("");
+    if (text) {
+      const previous = segments.at(-1);
+      if (previous?.role === line.role) previous.text += text;
+      else segments.push({ role: line.role, text });
+    }
+    const crossesSameRoleLine = lineIndex < end.line && lines[lineIndex + 1]?.role === line.role;
+    if (crossesSameRoleLine) {
+      const current = segments.at(-1);
+      if (current?.role === line.role) current.text += "\n";
+      else segments.push({ role: line.role, text: "\n" });
+    }
+  }
+  return segments;
 }
 
 export function estimateBringToMainTokens(segments: readonly BtwBringToMainSegment[]): number {
-	return Math.ceil(
-		Buffer.byteLength(segments.map((segment) => segment.text).join("\n"), "utf8") / 4,
-	);
+  return Math.ceil(Buffer.byteLength(segments.map((segment) => segment.text).join("\n"), "utf8") / 4);
 }
 
-export function summarizeBringToMain(
-	segments: readonly BtwBringToMainSegment[],
-): BtwBringToMainSummary {
-	return {
-		lines: segments.reduce((count, segment) => count + segment.text.split("\n").length, 0),
-		messages: segments.length,
-		tokens: estimateBringToMainTokens(segments),
-	};
+export function summarizeBringToMain(segments: readonly BtwBringToMainSegment[]): BtwBringToMainSummary {
+  return {
+    lines: segments.reduce((count, segment) => count + segment.text.split("\n").length, 0),
+    messages: segments.length,
+    tokens: estimateBringToMainTokens(segments),
+  };
 }
 
 export function formatBtwBringToMain(segments: readonly BtwBringToMainSegment[]): string {
-	const body = segments
-		.map(
-			(segment) =>
-				`${segment.role === "user" ? "User" : "Assistant"}:\n${escapeBringToMainText(segment.text)}`,
-		)
-		.join("\n\n");
-	return [
-		"The following context was brought back from a /btw side discussion.",
-		"Treat it as discussion context, not as work already completed.",
-		"",
-		"<btw_context>",
-		body,
-		"</btw_context>",
-	].join("\n");
+  const body = segments
+    .map((segment) => `${segment.role === "user" ? "User" : "Assistant"}:\n${escapeBringToMainText(segment.text)}`)
+    .join("\n\n");
+  return [
+    "The following context was brought back from a /btw side discussion.",
+    "Treat it as discussion context, not as work already completed.",
+    "",
+    "<btw_context>",
+    body,
+    "</btw_context>",
+  ].join("\n");
 }
 
 export class BtwTextRangeSelector implements Component {
-	private readonly lines: BtwSelectionLine[];
-	private cursor: BtwTextPosition = { line: 0, column: 0 };
-	private anchor: BtwTextPosition | undefined;
-	private lineAnchor: number | undefined;
-	private preferredColumn = 0;
-	private scrollOffset = 0;
-	private horizontalOffset = 0;
-	private warning: string | undefined;
-	private finished = false;
-
-	constructor(
-		private readonly tui: TUI,
-		private readonly theme: Theme,
-		private readonly keybindings: KeybindingsManager,
-		turns: readonly SideThreadTurn[],
-		private readonly onAction: (action: BtwTextRangeSelectorAction) => void,
-		initialState?: BtwTextRangeSelectorState,
-	) {
-		this.lines = buildBtwSelectionLines(turns);
-		if (initialState) {
-			this.cursor = clampTextPosition(this.lines, initialState.cursor);
-			this.anchor = initialState.anchor
-				? clampTextPosition(this.lines, initialState.anchor)
-				: undefined;
-			this.lineAnchor =
-				initialState.lineAnchor === undefined
-					? undefined
-					: Math.max(0, Math.min(this.lines.length - 1, initialState.lineAnchor));
-			this.preferredColumn = Math.max(0, initialState.preferredColumn);
-			this.scrollOffset = Math.max(0, initialState.scrollOffset);
-			this.horizontalOffset = Math.max(0, initialState.horizontalOffset);
-		}
-	}
-
-	getState(): BtwTextRangeSelectorState {
-		return {
-			cursor: { ...this.cursor },
-			anchor: this.anchor ? { ...this.anchor } : undefined,
-			lineAnchor: this.lineAnchor,
-			preferredColumn: this.preferredColumn,
-			scrollOffset: this.scrollOffset,
-			horizontalOffset: this.horizontalOffset,
-		};
-	}
-
-	render(width: number): string[] {
-		const safeWidth = Math.max(1, width);
-		const availableRows = Math.max(1, this.tui.terminal.rows - RESERVED_APP_ROWS);
-		const showStatus = availableRows >= 4;
-		const showFooter = availableRows >= 3;
-		const viewportHeight = Math.max(
-			1,
-			availableRows - 1 - (showStatus ? 1 : 0) - (showFooter ? 1 : 0),
-		);
-		this.keepCursorVisible(viewportHeight);
-		const textWidth = Math.max(1, safeWidth - visibleWidth("●> Assistant │ "));
-		this.keepCursorHorizontallyVisible(textWidth);
-		const range = this.getSelectionRange();
-		const lineRange = this.getLineSelectionRange();
-		const visible = this.lines.slice(this.scrollOffset, this.scrollOffset + viewportHeight);
-		const rows = visible.map((line, visibleIndex) => {
-			const lineIndex = this.scrollOffset + visibleIndex;
-			const role = line.role === "user" ? "User" : "Assistant";
-			const lineSelected = lineRange
-				? lineIndex >= lineRange.start && lineIndex <= lineRange.end
-				: false;
-			const prefix = `${lineSelected ? "●" : " "}${lineIndex === this.cursor.line ? ">" : " "} ${role.padEnd(9)} │ `;
-			const text = this.renderTextLine(line, lineIndex, range, lineSelected);
-			return truncateToWidth(
-				lineIndex === this.cursor.line ? this.theme.fg("accent", prefix) + text : prefix + text,
-				safeWidth,
-				"",
-			);
-		});
-		const selected = this.getSelectedSegments();
-		const summary = summarizeBringToMain(selected);
-		const status =
-			selected.length === 0
-				? "Selected: none"
-				: `Selected: ${summary.lines} ${summary.lines === 1 ? "line" : "lines"} · ${summary.messages} ${summary.messages === 1 ? "message" : "messages"} · ~${summary.tokens} ${summary.tokens === 1 ? "token" : "tokens"}`;
-		const confirm = confirmKeyLabel(this.keybindings);
-		const back = keybindingLabel(this.keybindings, "tui.select.cancel", ["ctrl+c"]);
-		const vertical = `${keybindingLabel(this.keybindings, "tui.select.up")}/${keybindingLabel(this.keybindings, "tui.select.down")}`;
-		const confirmUsesSpace = this.keybindings.matches(" ", "tui.select.confirm");
-		const detailedFooter = this.warning
-			? `${this.warning} • ${confirmUsesSpace ? "Shift+Arrows select" : "Space lines • Shift+Arrows text"} • ${back} back • Ctrl+C close`
-			: this.lineAnchor !== undefined
-				? `${confirmUsesSpace ? `${vertical} extend lines` : `Space clear • ${vertical} extend lines`} • Shift+Arrows text • ${confirm} bring • ${back} back • Ctrl+C close`
-				: `Shift+Arrows select • Arrows move${confirmUsesSpace ? "" : " • Space lines"} • ${confirm} bring • ${back} back • Ctrl+C close`;
-		const criticalFooter = `${confirm} bring • ${back} back • Ctrl+C close`;
-		const footer = visibleWidth(detailedFooter) <= safeWidth ? detailedFooter : criticalFooter;
-		return fitRows(
-			[
-				truncateToWidth(
-					this.theme.fg("accent", this.theme.bold("Select text to bring to main")),
-					safeWidth,
-					"",
-				),
-				...(showStatus ? [truncateToWidth(this.theme.fg("muted", status), safeWidth, "")] : []),
-				...rows,
-				...(showFooter
-					? [
-							truncateToWidth(
-								this.theme.fg(this.warning ? "warning" : "muted", footer),
-								safeWidth,
-								"",
-							),
-						]
-					: []),
-			],
-			availableRows,
-		);
-	}
-
-	handleInput(data: string): void {
-		if (this.finished) return;
-		if (matchesKey(data, Key.ctrl("c"))) {
-			this.finish({ kind: "close" });
-			return;
-		}
-		if (this.keybindings.matches(data, "tui.select.cancel")) {
-			this.finish({ kind: "back" });
-			return;
-		}
-		if (matchesConfirm(data, this.keybindings)) {
-			if (this.lines.length > 0) {
-				const segments = this.getSelectedSegments();
-				if (segments.length === 0) {
-					this.warning = "Select text first";
-					this.tui.requestRender();
-				} else {
-					this.finish({ kind: "confirm", segments });
-				}
-			}
-			return;
-		}
-		if (matchesKey(data, Key.space)) {
-			this.anchor = undefined;
-			this.lineAnchor = this.lineAnchor === undefined ? this.cursor.line : undefined;
-			this.afterMove();
-			return;
-		}
-		if (matchesKey(data, Key.shift("left"))) {
-			this.moveHorizontal(-1, true);
-			return;
-		}
-		if (matchesKey(data, Key.shift("right"))) {
-			this.moveHorizontal(1, true);
-			return;
-		}
-		if (matchesKey(data, Key.shift("up"))) {
-			this.moveVertical(-1, true);
-			return;
-		}
-		if (matchesKey(data, Key.shift("down"))) {
-			this.moveVertical(1, true);
-			return;
-		}
-		if (matchesKey(data, Key.left)) {
-			this.moveHorizontal(-1, false);
-			return;
-		}
-		if (matchesKey(data, Key.right)) {
-			this.moveHorizontal(1, false);
-			return;
-		}
-		if (this.keybindings.matches(data, "tui.select.up")) {
-			this.moveVertical(-1, false);
-			return;
-		}
-		if (this.keybindings.matches(data, "tui.select.down")) {
-			this.moveVertical(1, false);
-			return;
-		}
-		if (this.keybindings.matches(data, "tui.select.pageUp")) {
-			this.moveVertical(-10, false);
-			return;
-		}
-		if (this.keybindings.matches(data, "tui.select.pageDown")) {
-			this.moveVertical(10, false);
-			return;
-		}
-	}
-
-	invalidate(): void {}
-
-	private renderTextLine(
-		line: BtwSelectionLine,
-		lineIndex: number,
-		range: { start: BtwTextPosition; end: BtwTextPosition } | undefined,
-		lineSelected: boolean,
-	): string {
-		const characters = splitGraphemes(line.text);
-		let rendered = this.horizontalOffset > 0 ? this.theme.fg("muted", "…") : "";
-		let buffer = "";
-		let bufferSelected = false;
-		const flush = () => {
-			if (!buffer) return;
-			rendered += bufferSelected
-				? this.theme.bg("selectedBg", this.theme.fg("text", buffer))
-				: buffer;
-			buffer = "";
-		};
-		for (let column = this.horizontalOffset; column <= characters.length; column += 1) {
-			if (range && lineIndex === range.start.line && column === range.start.column) {
-				flush();
-				rendered += this.theme.fg("accent", "[");
-			}
-			if (lineIndex === this.cursor.line && column === this.cursor.column) {
-				flush();
-				rendered += this.theme.fg("accent", "│");
-			}
-			if (range && lineIndex === range.end.line && column === range.end.column) {
-				flush();
-				rendered += this.theme.fg("accent", "]");
-			}
-			const character = characters[column];
-			if (character === undefined) continue;
-			const selected =
-				lineSelected || (range ? positionFallsInside(lineIndex, column, range) : false);
-			if (buffer && selected !== bufferSelected) flush();
-			bufferSelected = selected;
-			buffer += escapeTerminalControls(character);
-		}
-		flush();
-		return rendered;
-	}
-
-	private getSelectionRange(): { start: BtwTextPosition; end: BtwTextPosition } | undefined {
-		if (!this.anchor || compareTextPositions(this.anchor, this.cursor) === 0) return undefined;
-		return compareTextPositions(this.anchor, this.cursor) < 0
-			? { start: this.anchor, end: this.cursor }
-			: { start: this.cursor, end: this.anchor };
-	}
-
-	private getLineSelectionRange(): { start: number; end: number } | undefined {
-		return this.lineAnchor === undefined
-			? undefined
-			: {
-					start: Math.min(this.lineAnchor, this.cursor.line),
-					end: Math.max(this.lineAnchor, this.cursor.line),
-				};
-	}
-
-	private getSelectedSegments(): BtwBringToMainSegment[] {
-		if (this.lineAnchor !== undefined) {
-			return segmentsFromLineRange(this.lines, this.lineAnchor, this.cursor.line);
-		}
-		return this.anchor ? segmentsFromTextRange(this.lines, this.anchor, this.cursor) : [];
-	}
-
-	private moveHorizontal(delta: -1 | 1, extend: boolean): void {
-		if (this.lines.length === 0) return;
-		if (!extend) this.lineAnchor = undefined;
-		if (!extend && this.anchor) {
-			const range = this.getSelectionRange();
-			if (range) this.cursor = delta < 0 ? range.start : range.end;
-			this.anchor = undefined;
-			this.preferredColumn = this.cursor.column;
-			this.afterMove();
-			return;
-		}
-		this.beginOrClearSelection(extend);
-		const line = this.lines[this.cursor.line];
-		const length = line ? splitGraphemes(line.text).length : 0;
-		if (delta < 0) {
-			if (this.cursor.column > 0) this.cursor = { ...this.cursor, column: this.cursor.column - 1 };
-			else if (this.cursor.line > 0) {
-				const previousLine = this.lines[this.cursor.line - 1];
-				this.cursor = {
-					line: this.cursor.line - 1,
-					column: previousLine ? splitGraphemes(previousLine.text).length : 0,
-				};
-			}
-		} else if (this.cursor.column < length) {
-			this.cursor = { ...this.cursor, column: this.cursor.column + 1 };
-		} else if (this.cursor.line < this.lines.length - 1) {
-			this.cursor = { line: this.cursor.line + 1, column: 0 };
-		}
-		this.preferredColumn = this.cursor.column;
-		this.afterMove();
-	}
-
-	private moveVertical(delta: number, extend: boolean): void {
-		if (this.lines.length === 0) return;
-		if (extend || this.lineAnchor === undefined) this.beginOrClearSelection(extend);
-		const line = Math.max(0, Math.min(this.lines.length - 1, this.cursor.line + delta));
-		const target = this.lines[line];
-		this.cursor = {
-			line,
-			column: Math.min(this.preferredColumn, target ? splitGraphemes(target.text).length : 0),
-		};
-		this.afterMove();
-	}
-
-	private beginOrClearSelection(extend: boolean): void {
-		if (extend) this.lineAnchor = undefined;
-		if (extend && !this.anchor) this.anchor = { ...this.cursor };
-		if (!extend) this.anchor = undefined;
-	}
-
-	private afterMove(): void {
-		this.warning = undefined;
-		this.tui.requestRender();
-	}
-
-	private keepCursorVisible(height: number): void {
-		if (height <= 0) return;
-		if (this.cursor.line < this.scrollOffset) this.scrollOffset = this.cursor.line;
-		if (this.cursor.line >= this.scrollOffset + height) {
-			this.scrollOffset = this.cursor.line - height + 1;
-		}
-	}
-
-	private keepCursorHorizontallyVisible(width: number): void {
-		const characters = splitGraphemes(this.lines[this.cursor.line]?.text ?? "");
-		const displayWidths = characters.map((character) =>
-			visibleWidth(escapeTerminalControls(character)),
-		);
-		const currentWidth = displayWidths[this.cursor.column] ?? 0;
-		let usedWidth = 1 + Math.min(currentWidth, Math.max(0, width - 1));
-		let offset = this.cursor.column;
-		for (let index = this.cursor.column - 1; index >= 0; index -= 1) {
-			const nextWidth = usedWidth + (displayWidths[index] ?? 0) + (index > 0 ? 1 : 0);
-			if (nextWidth > width) break;
-			usedWidth += displayWidths[index] ?? 0;
-			offset = index;
-		}
-		this.horizontalOffset = offset;
-	}
-
-	private finish(action: BtwTextRangeSelectorAction): void {
-		if (this.finished) return;
-		this.finished = true;
-		this.onAction(action);
-	}
+  private readonly lines: BtwSelectionLine[];
+  private cursor: BtwTextPosition = { line: 0, column: 0 };
+  private anchor: BtwTextPosition | undefined;
+  private lineAnchor: number | undefined;
+  private preferredColumn = 0;
+  private scrollOffset = 0;
+  private horizontalOffset = 0;
+  private warning: string | undefined;
+  private finished = false;
+
+  constructor(
+    private readonly tui: TUI,
+    private readonly theme: Theme,
+    private readonly keybindings: KeybindingsManager,
+    turns: readonly SideThreadTurn[],
+    private readonly onAction: (action: BtwTextRangeSelectorAction) => void,
+    initialState?: BtwTextRangeSelectorState,
+  ) {
+    this.lines = buildBtwSelectionLines(turns);
+    if (initialState) {
+      this.cursor = clampTextPosition(this.lines, initialState.cursor);
+      this.anchor = initialState.anchor ? clampTextPosition(this.lines, initialState.anchor) : undefined;
+      this.lineAnchor =
+        initialState.lineAnchor === undefined
+          ? undefined
+          : Math.max(0, Math.min(this.lines.length - 1, initialState.lineAnchor));
+      this.preferredColumn = Math.max(0, initialState.preferredColumn);
+      this.scrollOffset = Math.max(0, initialState.scrollOffset);
+      this.horizontalOffset = Math.max(0, initialState.horizontalOffset);
+    }
+  }
+
+  getState(): BtwTextRangeSelectorState {
+    return {
+      cursor: { ...this.cursor },
+      anchor: this.anchor ? { ...this.anchor } : undefined,
+      lineAnchor: this.lineAnchor,
+      preferredColumn: this.preferredColumn,
+      scrollOffset: this.scrollOffset,
+      horizontalOffset: this.horizontalOffset,
+    };
+  }
+
+  render(width: number): string[] {
+    const safeWidth = Math.max(1, width);
+    const availableRows = Math.max(1, this.tui.terminal.rows - RESERVED_APP_ROWS);
+    const showStatus = availableRows >= 4;
+    const showFooter = availableRows >= 3;
+    const viewportHeight = Math.max(1, availableRows - 1 - (showStatus ? 1 : 0) - (showFooter ? 1 : 0));
+    this.keepCursorVisible(viewportHeight);
+    const textWidth = Math.max(1, safeWidth - visibleWidth("●> Assistant │ "));
+    this.keepCursorHorizontallyVisible(textWidth);
+    const range = this.getSelectionRange();
+    const lineRange = this.getLineSelectionRange();
+    const visible = this.lines.slice(this.scrollOffset, this.scrollOffset + viewportHeight);
+    const rows = visible.map((line, visibleIndex) => {
+      const lineIndex = this.scrollOffset + visibleIndex;
+      const role = line.role === "user" ? "User" : "Assistant";
+      const lineSelected = lineRange ? lineIndex >= lineRange.start && lineIndex <= lineRange.end : false;
+      const prefix = `${lineSelected ? "●" : " "}${lineIndex === this.cursor.line ? ">" : " "} ${role.padEnd(9)} │ `;
+      const text = this.renderTextLine(line, lineIndex, range, lineSelected);
+      return truncateToWidth(
+        lineIndex === this.cursor.line ? this.theme.fg("accent", prefix) + text : prefix + text,
+        safeWidth,
+        "",
+      );
+    });
+    const selected = this.getSelectedSegments();
+    const summary = summarizeBringToMain(selected);
+    const status =
+      selected.length === 0
+        ? "Selected: none"
+        : `Selected: ${summary.lines} ${summary.lines === 1 ? "line" : "lines"} · ${summary.messages} ${summary.messages === 1 ? "message" : "messages"} · ~${summary.tokens} ${summary.tokens === 1 ? "token" : "tokens"}`;
+    const confirm = confirmKeyLabel(this.keybindings);
+    const back = keybindingLabel(this.keybindings, "tui.select.cancel", ["ctrl+c"]);
+    const vertical = `${keybindingLabel(this.keybindings, "tui.select.up")}/${keybindingLabel(this.keybindings, "tui.select.down")}`;
+    const confirmUsesSpace = this.keybindings.matches(" ", "tui.select.confirm");
+    const detailedFooter = this.warning
+      ? `${this.warning} • ${confirmUsesSpace ? "Shift+Arrows select" : "Space lines • Shift+Arrows text"} • ${back} back • Ctrl+C close`
+      : this.lineAnchor !== undefined
+        ? `${confirmUsesSpace ? `${vertical} extend lines` : `Space clear • ${vertical} extend lines`} • Shift+Arrows text • ${confirm} bring • ${back} back • Ctrl+C close`
+        : `Shift+Arrows select • Arrows move${confirmUsesSpace ? "" : " • Space lines"} • ${confirm} bring • ${back} back • Ctrl+C close`;
+    const criticalFooter = `${confirm} bring • ${back} back • Ctrl+C close`;
+    const footer = visibleWidth(detailedFooter) <= safeWidth ? detailedFooter : criticalFooter;
+    return fitRows(
+      [
+        truncateToWidth(this.theme.fg("accent", this.theme.bold("Select text to bring to main")), safeWidth, ""),
+        ...(showStatus ? [truncateToWidth(this.theme.fg("muted", status), safeWidth, "")] : []),
+        ...rows,
+        ...(showFooter
+          ? [truncateToWidth(this.theme.fg(this.warning ? "warning" : "muted", footer), safeWidth, "")]
+          : []),
+      ],
+      availableRows,
+    );
+  }
+
+  handleInput(data: string): void {
+    if (this.finished) return;
+    if (matchesKey(data, Key.ctrl("c"))) {
+      this.finish({ kind: "close" });
+      return;
+    }
+    if (this.keybindings.matches(data, "tui.select.cancel")) {
+      this.finish({ kind: "back" });
+      return;
+    }
+    if (matchesConfirm(data, this.keybindings)) {
+      if (this.lines.length > 0) {
+        const segments = this.getSelectedSegments();
+        if (segments.length === 0) {
+          this.warning = "Select text first";
+          this.tui.requestRender();
+        } else {
+          this.finish({ kind: "confirm", segments });
+        }
+      }
+      return;
+    }
+    if (matchesKey(data, Key.space)) {
+      this.anchor = undefined;
+      this.lineAnchor = this.lineAnchor === undefined ? this.cursor.line : undefined;
+      this.afterMove();
+      return;
+    }
+    if (matchesKey(data, Key.shift("left"))) {
+      this.moveHorizontal(-1, true);
+      return;
+    }
+    if (matchesKey(data, Key.shift("right"))) {
+      this.moveHorizontal(1, true);
+      return;
+    }
+    if (matchesKey(data, Key.shift("up"))) {
+      this.moveVertical(-1, true);
+      return;
+    }
+    if (matchesKey(data, Key.shift("down"))) {
+      this.moveVertical(1, true);
+      return;
+    }
+    if (matchesKey(data, Key.left)) {
+      this.moveHorizontal(-1, false);
+      return;
+    }
+    if (matchesKey(data, Key.right)) {
+      this.moveHorizontal(1, false);
+      return;
+    }
+    if (this.keybindings.matches(data, "tui.select.up")) {
+      this.moveVertical(-1, false);
+      return;
+    }
+    if (this.keybindings.matches(data, "tui.select.down")) {
+      this.moveVertical(1, false);
+      return;
+    }
+    if (this.keybindings.matches(data, "tui.select.pageUp")) {
+      this.moveVertical(-10, false);
+      return;
+    }
+    if (this.keybindings.matches(data, "tui.select.pageDown")) {
+      this.moveVertical(10, false);
+      return;
+    }
+  }
+
+  invalidate(): void {}
+
+  private renderTextLine(
+    line: BtwSelectionLine,
+    lineIndex: number,
+    range: { start: BtwTextPosition; end: BtwTextPosition } | undefined,
+    lineSelected: boolean,
+  ): string {
+    const characters = splitGraphemes(line.text);
+    let rendered = this.horizontalOffset > 0 ? this.theme.fg("muted", "…") : "";
+    let buffer = "";
+    let bufferSelected = false;
+    const flush = () => {
+      if (!buffer) return;
+      rendered += bufferSelected ? this.theme.bg("selectedBg", this.theme.fg("text", buffer)) : buffer;
+      buffer = "";
+    };
+    for (let column = this.horizontalOffset; column <= characters.length; column += 1) {
+      if (range && lineIndex === range.start.line && column === range.start.column) {
+        flush();
+        rendered += this.theme.fg("accent", "[");
+      }
+      if (lineIndex === this.cursor.line && column === this.cursor.column) {
+        flush();
+        rendered += this.theme.fg("accent", "│");
+      }
+      if (range && lineIndex === range.end.line && column === range.end.column) {
+        flush();
+        rendered += this.theme.fg("accent", "]");
+      }
+      const character = characters[column];
+      if (character === undefined) continue;
+      const selected = lineSelected || (range ? positionFallsInside(lineIndex, column, range) : false);
+      if (buffer && selected !== bufferSelected) flush();
+      bufferSelected = selected;
+      buffer += escapeTerminalControls(character);
+    }
+    flush();
+    return rendered;
+  }
+
+  private getSelectionRange(): { start: BtwTextPosition; end: BtwTextPosition } | undefined {
+    if (!this.anchor || compareTextPositions(this.anchor, this.cursor) === 0) return undefined;
+    return compareTextPositions(this.anchor, this.cursor) < 0
+      ? { start: this.anchor, end: this.cursor }
+      : { start: this.cursor, end: this.anchor };
+  }
+
+  private getLineSelectionRange(): { start: number; end: number } | undefined {
+    return this.lineAnchor === undefined
+      ? undefined
+      : {
+          start: Math.min(this.lineAnchor, this.cursor.line),
+          end: Math.max(this.lineAnchor, this.cursor.line),
+        };
+  }
+
+  private getSelectedSegments(): BtwBringToMainSegment[] {
+    if (this.lineAnchor !== undefined) {
+      return segmentsFromLineRange(this.lines, this.lineAnchor, this.cursor.line);
+    }
+    return this.anchor ? segmentsFromTextRange(this.lines, this.anchor, this.cursor) : [];
+  }
+
+  private moveHorizontal(delta: -1 | 1, extend: boolean): void {
+    if (this.lines.length === 0) return;
+    if (!extend) this.lineAnchor = undefined;
+    if (!extend && this.anchor) {
+      const range = this.getSelectionRange();
+      if (range) this.cursor = delta < 0 ? range.start : range.end;
+      this.anchor = undefined;
+      this.preferredColumn = this.cursor.column;
+      this.afterMove();
+      return;
+    }
+    this.beginOrClearSelection(extend);
+    const line = this.lines[this.cursor.line];
+    const length = line ? splitGraphemes(line.text).length : 0;
+    if (delta < 0) {
+      if (this.cursor.column > 0) this.cursor = { ...this.cursor, column: this.cursor.column - 1 };
+      else if (this.cursor.line > 0) {
+        const previousLine = this.lines[this.cursor.line - 1];
+        this.cursor = {
+          line: this.cursor.line - 1,
+          column: previousLine ? splitGraphemes(previousLine.text).length : 0,
+        };
+      }
+    } else if (this.cursor.column < length) {
+      this.cursor = { ...this.cursor, column: this.cursor.column + 1 };
+    } else if (this.cursor.line < this.lines.length - 1) {
+      this.cursor = { line: this.cursor.line + 1, column: 0 };
+    }
+    this.preferredColumn = this.cursor.column;
+    this.afterMove();
+  }
+
+  private moveVertical(delta: number, extend: boolean): void {
+    if (this.lines.length === 0) return;
+    if (extend || this.lineAnchor === undefined) this.beginOrClearSelection(extend);
+    const line = Math.max(0, Math.min(this.lines.length - 1, this.cursor.line + delta));
+    const target = this.lines[line];
+    this.cursor = {
+      line,
+      column: Math.min(this.preferredColumn, target ? splitGraphemes(target.text).length : 0),
+    };
+    this.afterMove();
+  }
+
+  private beginOrClearSelection(extend: boolean): void {
+    if (extend) this.lineAnchor = undefined;
+    if (extend && !this.anchor) this.anchor = { ...this.cursor };
+    if (!extend) this.anchor = undefined;
+  }
+
+  private afterMove(): void {
+    this.warning = undefined;
+    this.tui.requestRender();
+  }
+
+  private keepCursorVisible(height: number): void {
+    if (height <= 0) return;
+    if (this.cursor.line < this.scrollOffset) this.scrollOffset = this.cursor.line;
+    if (this.cursor.line >= this.scrollOffset + height) {
+      this.scrollOffset = this.cursor.line - height + 1;
+    }
+  }
+
+  private keepCursorHorizontallyVisible(width: number): void {
+    const characters = splitGraphemes(this.lines[this.cursor.line]?.text ?? "");
+    const displayWidths = characters.map((character) => visibleWidth(escapeTerminalControls(character)));
+    const currentWidth = displayWidths[this.cursor.column] ?? 0;
+    let usedWidth = 1 + Math.min(currentWidth, Math.max(0, width - 1));
+    let offset = this.cursor.column;
+    for (let index = this.cursor.column - 1; index >= 0; index -= 1) {
+      const nextWidth = usedWidth + (displayWidths[index] ?? 0) + (index > 0 ? 1 : 0);
+      if (nextWidth > width) break;
+      usedWidth += displayWidths[index] ?? 0;
+      offset = index;
+    }
+    this.horizontalOffset = offset;
+  }
+
+  private finish(action: BtwTextRangeSelectorAction): void {
+    if (this.finished) return;
+    this.finished = true;
+    this.onAction(action);
+  }
 }
 
-function clampTextPosition(
-	lines: readonly BtwSelectionLine[],
-	position: BtwTextPosition,
-): BtwTextPosition {
-	const line = Math.max(0, Math.min(lines.length - 1, position.line));
-	const text = lines[line]?.text ?? "";
-	return {
-		line,
-		column: Math.max(0, Math.min(splitGraphemes(text).length, position.column)),
-	};
+function clampTextPosition(lines: readonly BtwSelectionLine[], position: BtwTextPosition): BtwTextPosition {
+  const line = Math.max(0, Math.min(lines.length - 1, position.line));
+  const text = lines[line]?.text ?? "";
+  return {
+    line,
+    column: Math.max(0, Math.min(splitGraphemes(text).length, position.column)),
+  };
 }
 
 function confirmKeyLabel(keybindings: KeybindingsManager): string {
-	return keybindingLabel(keybindings, "tui.select.confirm", ["ctrl+c"], "enter");
+  return keybindingLabel(keybindings, "tui.select.confirm", ["ctrl+c"], "enter");
 }
 
 function matchesConfirm(data: string, keybindings: KeybindingsManager): boolean {
-	if (matchesKey(data, Key.ctrl("c"))) return false;
-	const hasUsableBinding = keybindings
-		.getKeys("tui.select.confirm")
-		.map(String)
-		.some((key) => key.toLowerCase() !== "ctrl+c");
-	return (
-		keybindings.matches(data, "tui.select.confirm") ||
-		(!hasUsableBinding && matchesKey(data, Key.enter))
-	);
+  if (matchesKey(data, Key.ctrl("c"))) return false;
+  const hasUsableBinding = keybindings
+    .getKeys("tui.select.confirm")
+    .map(String)
+    .some((key) => key.toLowerCase() !== "ctrl+c");
+  return keybindings.matches(data, "tui.select.confirm") || (!hasUsableBinding && matchesKey(data, Key.enter));
 }
 
 function keybindingLabel(
-	keybindings: KeybindingsManager,
-	keybinding:
-		| "tui.select.confirm"
-		| "tui.select.cancel"
-		| "tui.select.up"
-		| "tui.select.down"
-		| "tui.select.pageUp"
-		| "tui.select.pageDown",
-	excluded: readonly string[] = [],
-	fallback?: string,
+  keybindings: KeybindingsManager,
+  keybinding:
+    | "tui.select.confirm"
+    | "tui.select.cancel"
+    | "tui.select.up"
+    | "tui.select.down"
+    | "tui.select.pageUp"
+    | "tui.select.pageDown",
+  excluded: readonly string[] = [],
+  fallback?: string,
 ): string {
-	const key = keybindings
-		.getKeys(keybinding)
-		.map(String)
-		.find((candidate) => !excluded.includes(candidate.toLowerCase()));
-	return formatKeyLabel(key ?? fallback ?? keybinding);
+  const key = keybindings
+    .getKeys(keybinding)
+    .map(String)
+    .find((candidate) => !excluded.includes(candidate.toLowerCase()));
+  return formatKeyLabel(key ?? fallback ?? keybinding);
 }
 
 function formatKeyLabel(key: string): string {
-	return key
-		.split("+")
-		.map((part) => {
-			const lower = part.toLowerCase();
-			if (lower === "ctrl") return "Ctrl";
-			if (lower === "alt") return "Alt";
-			if (lower === "shift") return "Shift";
-			if (lower === "escape" || lower === "esc") return "Esc";
-			if (lower === "enter" || lower === "return") return "Enter";
-			if (lower === "pageup") return "PgUp";
-			if (lower === "pagedown") return "PgDn";
-			return part.length === 1
-				? part.toUpperCase()
-				: `${part[0]?.toUpperCase() ?? ""}${part.slice(1)}`;
-		})
-		.join("+");
+  return key
+    .split("+")
+    .map((part) => {
+      const lower = part.toLowerCase();
+      if (lower === "ctrl") return "Ctrl";
+      if (lower === "alt") return "Alt";
+      if (lower === "shift") return "Shift";
+      if (lower === "escape" || lower === "esc") return "Esc";
+      if (lower === "enter" || lower === "return") return "Enter";
+      if (lower === "pageup") return "PgUp";
+      if (lower === "pagedown") return "PgDn";
+      return part.length === 1 ? part.toUpperCase() : `${part[0]?.toUpperCase() ?? ""}${part.slice(1)}`;
+    })
+    .join("+");
 }
 
 function splitGraphemes(text: string): string[] {
-	return [...GRAPHEME_SEGMENTER.segment(text)].map(({ segment }) => segment);
+  return [...GRAPHEME_SEGMENTER.segment(text)].map(({ segment }) => segment);
 }
 
 function compareTextPositions(first: BtwTextPosition, second: BtwTextPosition): number {
-	return first.line === second.line ? first.column - second.column : first.line - second.line;
+  return first.line === second.line ? first.column - second.column : first.line - second.line;
 }
 
 function positionFallsInside(
-	line: number,
-	column: number,
-	range: { start: BtwTextPosition; end: BtwTextPosition },
+  line: number,
+  column: number,
+  range: { start: BtwTextPosition; end: BtwTextPosition },
 ): boolean {
-	const position = { line, column };
-	return (
-		compareTextPositions(position, range.start) >= 0 &&
-		compareTextPositions(position, range.end) < 0
-	);
+  const position = { line, column };
+  return compareTextPositions(position, range.start) >= 0 && compareTextPositions(position, range.end) < 0;
 }
 
 function fitRows(rows: string[], availableRows: number): string[] {
-	if (rows.length <= availableRows) return rows;
-	if (availableRows <= 1) return rows.slice(0, 1);
-	return [rows[0] ?? "", ...rows.slice(rows.length - availableRows + 1)];
+  if (rows.length <= availableRows) return rows;
+  if (availableRows <= 1) return rows.slice(0, 1);
+  return [rows[0] ?? "", ...rows.slice(rows.length - availableRows + 1)];
 }
 
 function escapeBringToMainText(text: string): string {
-	return [...text]
-		.map((character) => {
-			if (character === "\n") return character;
-			if (character === "\t") return "    ";
-			const code = character.charCodeAt(0);
-			if (code <= 31 || (code >= 127 && code <= 159)) {
-				return `\\x${code.toString(16).padStart(2, "0")}`;
-			}
-			return character;
-		})
-		.join("")
-		.replace(/<btw_context(?=[ \t\r\n>])/g, "&lt;btw_context")
-		.replace(/<\/btw_context[ \t\r\n]*>/g, (terminator) =>
-			terminator.replaceAll("<", "&lt;").replaceAll(">", "&gt;"),
-		);
+  return [...text]
+    .map((character) => {
+      if (character === "\n") return character;
+      if (character === "\t") return "    ";
+      const code = character.charCodeAt(0);
+      if (code <= 31 || (code >= 127 && code <= 159)) {
+        return `\\x${code.toString(16).padStart(2, "0")}`;
+      }
+      return character;
+    })
+    .join("")
+    .replace(/<btw_context(?=[ \t\r\n>])/g, "&lt;btw_context")
+    .replace(/<\/btw_context[ \t\r\n]*>/g, (terminator) => terminator.replaceAll("<", "&lt;").replaceAll(">", "&gt;"));
 }
 
 function escapeTerminalControls(text: string): string {
-	return [...text]
-		.map((character) => {
-			const code = character.charCodeAt(0);
-			if (code <= 31 || (code >= 127 && code <= 159)) {
-				return `\\x${code.toString(16).padStart(2, "0")}`;
-			}
-			return character;
-		})
-		.join("");
+  return [...text]
+    .map((character) => {
+      const code = character.charCodeAt(0);
+      if (code <= 31 || (code >= 127 && code <= 159)) {
+        return `\\x${code.toString(16).padStart(2, "0")}`;
+      }
+      return character;
+    })
+    .join("");
 }