@bastani/atomic 0.9.16-alpha.2 → 0.9.16-alpha.3

package/dist/builtin/workflows/src/tui/graph-view-input.ts

@@ -1,427 +0,0 @@
-import { GRAPH_SCROLL_STEP_COLS } from "./graph-view-constants.js";
-import { GraphViewRenderer } from "./graph-view-render.js";
-import { APP_ACTION, isKeybindingsLike, type KeybindingsLike, matchesAction } from "./keybindings-adapter.js";
-import { isTerminalLeftMousePress, parseTerminalMouseInput, terminalMouseWheelDirection } from "./mouse-input.js";
-import { defaultResponseFor, handlePromptCardInput } from "./prompt-card.js";
-import { filterStages, type SwitcherState } from "./switcher.js";
-import { Key, matchesKey } from "./text-helpers.js";
-
-interface MouseWheelDelta {
-	cols: number;
-	rows: number;
-}
-
-const GRAPH_SCROLL_WHEEL_LINES = 4;
-/** Raw Ctrl+O is only a compatibility fallback for hosts without keybindings. */
-const TOOL_EXPAND_FALLBACK = "\x0f";
-
-/** Keyboard, mouse, switcher, prompt, and focus navigation handling. */
-export abstract class GraphViewInputController extends GraphViewRenderer {
-	/** Returns true if consumed. */
-	handleInput(data: string): boolean {
-		if (this.toolDetail !== null) return this._handleToolDetailInput(data);
-		if (this._isReturnToMainChatInput(data)) {
-			this._returnToMainChat();
-			return true;
-		}
-		if (this.graphLayout.isScrollbarInput(data)) return this._handleGraphInput(data);
-		if (this.switcherOpen) return this._handleSwitcherInput(data);
-		const wheelDelta = this._mouseWheelDelta(data);
-		if (wheelDelta) return this._handleWheelDelta(wheelDelta);
-		// Stage-local HIL is represented by graph nodes and remains graph-first;
-		// only the legacy run-level prompt card sets `promptState`. Keep that
-		// fallback answerable, but let the scrollbar and wheel controls above
-		// continue to belong to the graph instead of leaking to the transcript.
-		// Printable keys such as "/" belong to the prompt card while legacy
-		// run-level text/editor prompts own input.
-		if (this.promptState) return this._handlePromptInput(data);
-		return this._handleGraphInput(data);
-	}
-
-	private _handleToolDetailInput(data: string): boolean {
-		if (this.graphLayout.handleScrollbarInput(data)) {
-			this.pendingEnsureFocusedVisible = false;
-			return true;
-		}
-		const wheelDelta = this._mouseWheelDelta(data);
-		if (wheelDelta) {
-			if (wheelDelta.rows !== 0) this._scrollGraphVertically(wheelDelta.rows);
-			return true;
-		}
-		if (this._matchesToolExpandInput(data)) {
-			this.toolDetailExpanded = !this.toolDetailExpanded;
-			this.pendingEnsureFocusedVisible = true;
-			return true;
-		}
-		if (this._isReturnToMainChatInput(data) || matchesKey(data, Key.escape)) {
-			this._closeToolDetail();
-			return true;
-		}
-		if (matchesKey(data, Key.ctrl("c"))) {
-			this.onClose?.();
-			return true;
-		}
-		if (matchesKey(data, "h") && this.onHide) {
-			this.onHide();
-			return true;
-		}
-		// A capped payload still renders far more rows than a terminal shows, so
-		// the detail must be readable without a mouse: these keys drive the same
-		// vertical scroll the wheel and scrollbar drive.
-		if (matchesKey(data, Key.down)) {
-			this._scrollGraphVertically(1);
-			return true;
-		}
-		if (matchesKey(data, Key.up)) {
-			this._scrollGraphVertically(-1);
-			return true;
-		}
-		if (matchesKey(data, Key.pageDown)) {
-			this._scrollGraphVertically(this._toolDetailPageRows());
-			return true;
-		}
-		if (matchesKey(data, Key.pageUp)) {
-			this._scrollGraphVertically(-this._toolDetailPageRows());
-			return true;
-		}
-		if (matchesKey(data, Key.home)) {
-			this.pendingEnsureFocusedVisible = false;
-			this.graphLayout.scrollView.scrollToStart();
-			return true;
-		}
-		if (matchesKey(data, Key.end)) {
-			this.pendingEnsureFocusedVisible = false;
-			this.graphLayout.scrollView.scrollToEnd();
-			return true;
-		}
-		if (
-			matchesKey(data, Key.enter) ||
-			matchesKey(data, Key.left) ||
-			matchesKey(data, Key.right) ||
-			matchesKey(data, "/") ||
-			parseTerminalMouseInput(data) !== null
-		)
-			return true;
-		return false;
-	}
-
-	/** One page is the visible body, less a row of overlap for continuity. */
-	private _toolDetailPageRows(): number {
-		return Math.max(1, this.graphLayout.viewportRows - 1);
-	}
-
-	private _promptKeybindings(): KeybindingsLike | undefined {
-		return isKeybindingsLike(this.piKeybindings) ? this.piKeybindings : undefined;
-	}
-
-	private _matchesToolExpandInput(data: string): boolean {
-		const keybindings = this._promptKeybindings();
-		// Do not run a raw fallback beside a host manager: a user remap must win,
-		// and an explicitly unbound action must remain unbound.
-		return keybindings ? matchesAction(keybindings, data, APP_ACTION.toolsExpand) : data === TOOL_EXPAND_FALLBACK;
-	}
-
-	private _isReturnToMainChatInput(data: string): boolean {
-		return matchesKey(data, Key.ctrl("x"));
-	}
-
-	private _returnToMainChat(): void {
-		if (this.onDetach) this.onDetach();
-		else this.onHide?.();
-	}
-
-	private _handlePromptInput(data: string): boolean {
-		const state = this.promptState;
-		if (!state) return false;
-		const action = handlePromptCardInput(data, state, this._promptKeybindings());
-		if (action.kind === "noop") return true;
-		const runId = this.runId;
-		if (!runId) return true;
-		const response = action.kind === "cancel" ? defaultResponseFor(state.prompt) : action.response;
-		this._resolvePrompt(runId, state.prompt.id, response);
-		return true;
-	}
-
-	private _resolvePrompt(runId: string, promptId: string, response: unknown): void {
-		// Clear local state immediately so the card disappears even if the
-		// host doesn't re-emit a store snapshot between resolve and render.
-		this.promptState = null;
-		if (this.onPromptResolve) {
-			this.onPromptResolve(runId, promptId, response);
-			return;
-		}
-		// Fallback path used by callers that wire GraphView directly without
-		// injecting onPromptResolve. Best-effort — if the store rejects (stale
-		// id) we already cleared local state, so we don't try to re-arm.
-		this.store.resolvePendingPrompt(runId, promptId, response);
-	}
-
-	private _handleGraphInput(data: string): boolean {
-		const stageCount = this.cachedLayout.length;
-		if (this.graphLayout.handleScrollbarInput(data)) {
-			this.pendingEnsureFocusedVisible = false;
-			return true;
-		}
-		const wheelDelta = this._mouseWheelDelta(data);
-		if (wheelDelta) return this._handleWheelDelta(wheelDelta);
-
-		const clickedNodeIndex = this._graphNodeIndexForClick(data);
-		if (clickedNodeIndex !== undefined) {
-			if (clickedNodeIndex !== null) {
-				this._setFocusedIndex(clickedNodeIndex);
-				this._activateFocusedNode();
-			}
-			return true;
-		}
-
-		// Vertical-graph navigation: up/down step between depth levels
-		// (col), left/right step between siblings at the same depth (row).
-		// j/k preserved as a flat-order fallback for muscle memory.
-		if (matchesKey(data, Key.down)) return this._moveByDepth(+1);
-		if (matchesKey(data, Key.up)) return this._moveByDepth(-1);
-		if (matchesKey(data, Key.right)) return this._moveBySibling(+1);
-		if (matchesKey(data, Key.left)) return this._moveBySibling(-1);
-		if (matchesKey(data, "j")) {
-			this._setFocusedIndex(Math.min(this.focusedIndex + 1, stageCount - 1));
-			return true;
-		}
-		if (matchesKey(data, "k")) {
-			this._setFocusedIndex(Math.max(this.focusedIndex - 1, 0));
-			return true;
-		}
-		if (matchesKey(data, "g")) {
-			const now = Date.now();
-			if (this._lastGTime != null && now - this._lastGTime < 500) {
-				this._setFocusedIndex(0);
-				this._lastGTime = null;
-			} else {
-				this._lastGTime = now;
-			}
-			return true;
-		}
-		if (matchesKey(data, "/")) {
-			this.switcherOpen = true;
-			this.switcherState = { query: "", selectedIndex: 0 };
-			return true;
-		}
-		if (matchesKey(data, Key.enter)) {
-			this._activateFocusedNode();
-			return true;
-		}
-		if (matchesKey(data, "h") && this.onHide) {
-			this.onHide();
-			return true;
-		}
-		if (matchesKey(data, Key.escape) || matchesKey(data, Key.ctrl("c"))) {
-			this.onClose?.();
-			return true;
-		}
-		return false;
-	}
-
-	private _handleSwitcherInput(data: string): boolean {
-		const stages = this.cachedLayout.map((layoutNode) => layoutNode.stage);
-
-		if (matchesKey(data, Key.escape)) {
-			this.switcherOpen = false;
-			return true;
-		}
-		if (matchesKey(data, Key.enter)) {
-			const filtered = filterStages(stages, this.switcherState.query);
-			const selected = filtered[this.switcherState.selectedIndex];
-			if (selected) {
-				const idx = this.cachedLayout.findIndex((n) => n.stage.id === selected.id);
-				if (idx !== -1) {
-					this._setFocusedIndex(idx);
-					// Selecting from the switcher opens tool detail or preserves the
-					// existing stage attach path; stages without an attach callback
-					// retain their historical no-op behavior.
-					this.switcherOpen = false;
-					if (this.cachedLayout[idx]?.stage.nodeKind === "tool") this._openFocusedToolDetail();
-					else this._attachFocusedStage();
-					return true;
-				}
-			}
-			this.switcherOpen = false;
-			return true;
-		}
-		const wheelEvent = parseTerminalMouseInput(data);
-		const wheelDirection = wheelEvent ? terminalMouseWheelDirection(wheelEvent) : null;
-		if (wheelDirection === "down") return this._moveSwitcherSelection(1);
-		if (wheelDirection === "up") return this._moveSwitcherSelection(-1);
-		if (wheelDirection !== null) return true;
-		if (matchesKey(data, Key.down)) return this._moveSwitcherSelection(1);
-		if (matchesKey(data, Key.up)) return this._moveSwitcherSelection(-1);
-		if (matchesKey(data, Key.backspace)) {
-			this.switcherState = {
-				query: this.switcherState.query.slice(0, -1),
-				selectedIndex: 0,
-			};
-			return true;
-		}
-		if (data.length === 1 && data >= " ") {
-			this.switcherState = {
-				query: this.switcherState.query + data,
-				selectedIndex: 0,
-			};
-			return true;
-		}
-		return false;
-	}
-
-	private _moveSwitcherSelection(step: number): boolean {
-		const stages = this.cachedLayout.map((layoutNode) => layoutNode.stage);
-		const filtered = filterStages(stages, this.switcherState.query);
-		const maxIndex = Math.max(0, filtered.length - 1);
-		this.switcherState = {
-			...this.switcherState,
-			selectedIndex: Math.max(0, Math.min(this.switcherState.selectedIndex + step, maxIndex)),
-		};
-		return true;
-	}
-
-	/**
-	 * Move focus to the nearest node `step` depth-levels away (↑/↓).
-	 * Picks the sibling with the closest `row` to the current node so
-	 * navigation feels spatially continuous in the vertical layout.
-	 */
-	private _moveByDepth(step: number): boolean {
-		const cur = this.cachedLayout[this.focusedIndex];
-		if (!cur) return true;
-		const targetCol = cur.col + step;
-		const candidates = this.cachedLayout.map((n, i) => ({ n, i })).filter(({ n }) => n.col === targetCol);
-		if (candidates.length === 0) return true;
-		let best = candidates[0]!;
-		let bestDist = Math.abs(best.n.row - cur.row);
-		for (const c of candidates) {
-			const d = Math.abs(c.n.row - cur.row);
-			if (d < bestDist) {
-				best = c;
-				bestDist = d;
-			}
-		}
-		this._setFocusedIndex(best.i);
-		return true;
-	}
-
-	/**
-	 * Move focus to the next sibling at the same depth (←/→). Clamps
-	 * at the band edges — no wrap, so the user always knows when they
-	 * hit a boundary.
-	 */
-	private _moveBySibling(step: number): boolean {
-		const cur = this.cachedLayout[this.focusedIndex];
-		if (!cur) return true;
-		const siblings = this.cachedLayout
-			.map((n, i) => ({ n, i }))
-			.filter(({ n }) => n.col === cur.col)
-			.sort((a, b) => a.n.row - b.n.row);
-		const pos = siblings.findIndex(({ i }) => i === this.focusedIndex);
-		if (pos === -1) return true;
-		const next = siblings[pos + step];
-		if (!next) return true;
-		this._setFocusedIndex(next.i);
-		return true;
-	}
-
-	private _activateFocusedNode(): void {
-		// Tool nodes open a read-only snapshot detail; they never enter the
-		// stage-chat attach path even though they share the graph card shape.
-		if (this._openFocusedToolDetail()) return;
-		// Enter and direct node clicks attach the popup interior to the focused
-		// stage. The attach shell swaps in the stage-chat view without remounting
-		// the overlay; without a callback, fall back to the legacy expand/collapse
-		// toggle so non-attach hosts still work.
-		if (this._attachFocusedStage()) return;
-		this.detailsExpanded = !this.detailsExpanded;
-	}
-
-	private _attachFocusedStage(): boolean {
-		if (!this.onStageAttach) return false;
-		const target = this._focusedStageChatTarget();
-		if (!target) return false;
-		this.onStageAttach(target.runId, target.stageId);
-		return true;
-	}
-
-	private _setFocusedIndex(index: number): void {
-		const max = Math.max(0, this.cachedLayout.length - 1);
-		const next = Math.max(0, Math.min(index, max));
-		if (next === this.focusedIndex) return;
-		this.focusedIndex = next;
-		this.pendingEnsureFocusedVisible = true;
-	}
-	private _scrollGraphVertically(deltaRows: number): void {
-		this.pendingEnsureFocusedVisible = false;
-		this.graphLayout.scrollView.scrollBy(deltaRows);
-	}
-
-	private _handleWheelDelta(delta: MouseWheelDelta): boolean {
-		if (delta.rows !== 0) this._scrollGraphVertically(delta.rows);
-		if (delta.cols !== 0) this._scrollGraphHorizontallyBy(delta.cols);
-		return true;
-	}
-
-	private _scrollGraphHorizontallyBy(deltaCols: number): void {
-		this.pendingEnsureFocusedVisible = false;
-		this.graphScrollColOffset = Math.max(0, this.graphScrollColOffset + deltaCols);
-	}
-
-	private _graphNodeIndexForClick(data: string): number | null | undefined {
-		const click = this._sgrLeftMousePress(data);
-		if (!click) return undefined;
-		if (this.mode !== "overlay") return undefined;
-		if (this.cachedLayout.length === 0) return null;
-
-		for (const rect of this.graphNodeHitRects) {
-			if (click.row >= rect.top && click.row < rect.bottom && click.col >= rect.left && click.col < rect.right) {
-				return rect.index;
-			}
-		}
-		return null;
-	}
-	private _sgrLeftMousePress(data: string): { col: number; row: number } | null {
-		const event = parseTerminalMouseInput(data);
-		if (event?.protocol !== "sgr" || !isTerminalLeftMousePress(event)) return null;
-		return { col: event.col, row: event.row };
-	}
-
-	private _mouseWheelDelta(data: string): MouseWheelDelta | null {
-		const event = parseTerminalMouseInput(data);
-		if (!event) return null;
-		const direction = terminalMouseWheelDirection(event);
-		if (direction === "up") return { cols: 0, rows: -GRAPH_SCROLL_WHEEL_LINES };
-		if (direction === "down") return { cols: 0, rows: GRAPH_SCROLL_WHEEL_LINES };
-		if (direction === "left") return { cols: -GRAPH_SCROLL_STEP_COLS, rows: 0 };
-		if (direction === "right") return { cols: GRAPH_SCROLL_STEP_COLS, rows: 0 };
-		return null;
-	}
-
-	get _graphScrollbarGeometry() {
-		return this.graphLayout.scrollbarGeometry;
-	}
-
-	get _toolDetail(): typeof this.toolDetail {
-		return this.toolDetail;
-	}
-	get _toolDetailExpanded(): boolean {
-		return this.toolDetailExpanded;
-	}
-
-	// ---- test seams ----
-	get _focusedIndex(): number {
-		return this.focusedIndex;
-	}
-	get _switcherOpen(): boolean {
-		return this.switcherOpen;
-	}
-	get _switcherState(): SwitcherState {
-		return this.switcherState;
-	}
-	get _graphScrollOffset(): number {
-		return this._graphScrollTop();
-	}
-	get _graphScrollColOffset(): number {
-		return this.graphScrollColOffset;
-	}
-}