@bachi/pi-coder 1.0.0

package/extensions/prompt-editor/render.test.ts

@@ -0,0 +1,283 @@
+/**
+ * Tests for prompt-editor.ts — `!` bash 模式在**真编辑器**上的渲染契约。
+ *
+ * Run with:  node --test clients/pi/extensions/prompt-editor/render.test.ts
+ *
+ * pi 自己的扩展加载器真的加载 `../prompt-editor.ts`（语法、import 面、session_start 注册都在
+ * 覆盖范围内），假的只有编辑器外面的一切：tui 只要 `terminal.rows` 与 `requestRender()`，
+ * 主题的 `borderColor` 是恒等函数，keybindings 的 `matches` 一律 false。于是
+ * 「gutter 换成 `!`、正文里输入的 `!` 不再显示、光标进不到被摘掉的那一列、点选跨过那一列、
+ * 退格删掉 `!` 就退出模式」这条链能被逐行 / 逐光标列断言。纯逻辑另在 `bash-prompt.test.ts`。
+ */
+
+import assert from "node:assert/strict";
+import fs from "node:fs";
+import os from "node:os";
+import path from "node:path";
+import test from "node:test";
+import { fileURLToPath, pathToFileURL } from "node:url";
+
+const EXTENSION_PATH = path.join(
+	path.dirname(fileURLToPath(import.meta.url)),
+	"..",
+	"prompt-editor.ts",
+);
+const SKIP = "找不到本机 pi 的库入口（装过 pi 才有）";
+
+/**
+ * pi 的库入口（非 CLI）：bundle 是 `pi` 实际跑的形态，dist 是 node 构建形态。
+ *
+ * 先从 `pi` 可执行文件反查**真正的安装位置**（pnpm/npm 的 shim 脚本里留有
+ * `# cmd-shim-target=<绝对路径>`；npm 在 Unix 上则是符号链接，两种都试），再退回
+ * `~/.pi/agent/npm` 那份副本 —— 后者是**扩展包的安装根**，`pi update --extensions` 会用
+ * `--config.auto-install-peers=false` 把自动装进去的 `@earendil-works/pi-*` 同伴包剪掉，
+ * 于是那份副本只剩空壳（文件都在、import 报 `ERR_MODULE_NOT_FOUND`）。
+ *
+ * 判定方式是**能不能真 import**，不是路径存不存在 —— 只有真 import 一次才分得清空壳。
+ * 全都不行就返回 undefined，调用方整体 skip（不假装通过）。
+ */
+async function findPiLibraryEntry(): Promise<string | undefined> {
+	const candidates: string[] = [];
+	if (process.env.PI_TEST_PI_ENTRY) candidates.push(process.env.PI_TEST_PI_ENTRY);
+
+	for (const dir of (process.env.PATH ?? "").split(path.delimiter)) {
+		if (!dir) continue;
+		const shimPath = path.join(dir, "pi");
+		try {
+			// npm 的 shim 是符号链接
+			const real = fs.realpathSync(shimPath);
+			if (real !== shimPath) candidates.push(path.join(path.dirname(real), "index.js"));
+		} catch {
+			// 不是符号链接 / 不存在：看下面的 shim 脚本
+		}
+		try {
+			const match = /^# cmd-shim-target=(.+)$/m.exec(fs.readFileSync(shimPath, "utf8"));
+			if (match?.[1]) candidates.push(path.join(path.dirname(match[1].trim()), "index.js"));
+		} catch {
+			// 读不到这个 shim：跳过
+		}
+	}
+
+	const packageDir = path.join(os.homedir(), ".pi/agent/npm/node_modules/@earendil-works/pi-coding-agent");
+	candidates.push(path.join(packageDir, "dist/bundle/index.js"), path.join(packageDir, "dist/index.js"));
+
+	for (const candidate of candidates) {
+		if (!fs.existsSync(candidate)) continue;
+		try {
+			await import(pathToFileURL(candidate).href);
+			return candidate;
+		} catch {
+			// 空壳副本：换下一个候选
+		}
+	}
+	return undefined;
+}
+
+const piEntry = await findPiLibraryEntry();
+const skip = piEntry === undefined ? SKIP : false;
+
+type Handler = (event: unknown, ctx: unknown) => Promise<unknown> | unknown;
+
+function createTestBus(): {
+	emit(channel: string, data: unknown): void;
+	on(channel: string, handler: (data: unknown) => void): () => void;
+} {
+	const handlers = new Map<string, Set<(data: unknown) => void>>();
+	return {
+		on(channel, handler) {
+			const set = handlers.get(channel) ?? new Set<(data: unknown) => void>();
+			handlers.set(channel, set);
+			set.add(handler);
+			return () => {
+				set.delete(handler);
+			};
+		},
+		emit(channel, data) {
+			for (const handler of [...(handlers.get(channel) ?? [])]) handler(data);
+		},
+	};
+}
+
+interface EditorLike {
+	setText(text: string): void;
+	getText(): string;
+	getCursor(): { line: number; col: number };
+	handleInput(data: string): void;
+	handleMouse(event: Record<string, unknown>): unknown;
+	render(width: number): string[];
+}
+
+function makeWorkspace(): { agentDir: string; projectDir: string; cleanup: () => void } {
+	const root = fs.mkdtempSync(path.join(os.tmpdir(), "pi-prompt-editor-"));
+	const agentDir = path.join(root, "agent");
+	const projectDir = path.join(root, "project");
+	fs.mkdirSync(agentDir);
+	fs.mkdirSync(projectDir);
+	return {
+		agentDir,
+		projectDir,
+		cleanup: () => fs.rmSync(root, { recursive: true, force: true }),
+	};
+}
+
+/**
+ * 用 pi 自己的加载器加载扩展，再调用它注册的 session_start，抓出编辑器工厂并实例化。
+ * 编辑器外面全部是假的：`paddingX: 0` 的 Editor 只需要 `tui.terminal.rows`，
+ * 主题只需要一个恒等的 `borderColor`，keybindings 只需要一个永远不命中的 `matches`
+ * （Editor 自己的键位走 pi-tui 的全局默认表，所以 Ctrl+A / 退格都是真按真绑定的行为）。
+ */
+async function mountEditor(): Promise<EditorLike> {
+	const workspace = makeWorkspace();
+	try {
+		const pi = (await import(pathToFileURL(piEntry as string).href)) as {
+			discoverAndLoadExtensions: (
+				configuredPaths: string[],
+				cwd: string,
+				agentDir?: string,
+				eventBus?: unknown,
+			) => Promise<{
+				extensions: Array<{ handlers: Map<string, Handler[]> }>;
+				errors: Array<{ path: string; error: string }>;
+			}>;
+		};
+		const loaded = await pi.discoverAndLoadExtensions(
+			[EXTENSION_PATH],
+			workspace.projectDir,
+			workspace.agentDir,
+			createTestBus(),
+		);
+		assert.deepEqual(loaded.errors, [], "pi 的扩展加载器不应该报错");
+		const extension = loaded.extensions[0];
+		assert.ok(extension, "扩展应该被加载");
+		const sessionStart = extension.handlers.get("session_start")?.[0];
+		assert.ok(sessionStart, "扩展应该注册 session_start");
+
+		let factory: ((tui: unknown, theme: unknown, keybindings: unknown) => EditorLike) | undefined;
+		const ctx = {
+			hasUI: true,
+			cwd: workspace.projectDir,
+			ui: {
+				setEditorComponent(next: typeof factory) {
+					factory = next;
+				},
+			},
+		};
+		await sessionStart({}, ctx);
+		assert.ok(factory, "session_start 应该注册编辑器工厂");
+
+		return factory(
+			{ terminal: { rows: 40 }, requestRender() {} },
+			{ borderColor: (text: string) => text },
+			{ matches: () => false },
+		);
+	} finally {
+		workspace.cleanup();
+	}
+}
+
+/** 去掉 CSI 与 APC（CURSOR_MARKER）转义，剩下纯文本。 */
+function plainLines(lines: string[]): string[] {
+	return lines.map((line) =>
+		line
+			.replace(/\x1b\[[0-9;:?]*[ -/]*[@-~]/g, "")
+			.replace(/\x1b_[^\x07\x1b]*(?:\x07|\x1b\\)/g, ""),
+	);
+}
+
+/** Editor.handleMouse 要的完整事件形状；x / y 是组件内坐标。 */
+function mouseClick(x: number, y: number): Record<string, unknown> {
+	return {
+		type: "click",
+		button: "left",
+		x,
+		y,
+		screenX: x,
+		screenY: y,
+		width: 40,
+		height: 40,
+		shift: false,
+		alt: false,
+		ctrl: false,
+		clickCount: 1,
+	};
+}
+
+test("普通输入：gutter 是 ❯，正文原样", { skip, timeout: 30_000 }, async () => {
+	const editor = await mountEditor();
+	editor.setText("hello");
+
+	const lines = plainLines(editor.render(40));
+	assert.ok(lines[1]?.startsWith("❯ hello"), `第一行正文应是 "❯ hello"：${JSON.stringify(lines[1])}`);
+	assert.equal(lines[1]?.length, 40, "每行都要满宽（gutter + contentWidth）");
+	assert.equal(lines[0]?.length, 40, "上边框也要满宽");
+});
+
+test("`!` 开头：gutter 换成 !，正文里输入的 ! 不再显示", { skip, timeout: 30_000 }, async () => {
+	const editor = await mountEditor();
+	editor.setText("!ls -la");
+
+	const lines = plainLines(editor.render(40));
+	assert.ok(lines[1]?.startsWith("! ls -la"), `gutter 应是 "! " 且正文紧接其后：${JSON.stringify(lines[1])}`);
+	assert.equal(lines[1]?.includes("!ls"), false, "正文里不能再出现那个 `!`");
+	assert.equal(lines[1]?.length, 40, "摘掉一列后要补回满宽");
+	assert.equal(lines[0]?.length, 40, "上边框不受影响");
+});
+
+test("`!!`（不进上下文）只藏第一个 !，第二个留在正文里可见", { skip, timeout: 30_000 }, async () => {
+	const editor = await mountEditor();
+	editor.setText("!!ls");
+
+	const lines = plainLines(editor.render(40));
+	assert.ok(lines[1]?.startsWith("! !ls"), `第二个 "!" 要留在正文里：${JSON.stringify(lines[1])}`);
+	assert.equal(lines[1]?.length, 40);
+});
+
+test("正文只剩 `!` 时按退格退出模式：gutter 自己变回 ❯", { skip, timeout: 30_000 }, async () => {
+	const editor = await mountEditor();
+	editor.setText("!");
+	assert.ok(plainLines(editor.render(40))[1]?.startsWith("! "), "先进入 bash 模式");
+
+	editor.handleInput("\x7f"); // backspace
+	assert.equal(editor.getText(), "", "退格删掉了那个 `!`");
+	assert.deepEqual(editor.getCursor(), { line: 0, col: 0 });
+	assert.ok(
+		plainLines(editor.render(40))[1]?.startsWith("❯ "),
+		"退出模式后 gutter 自然回到 ❯",
+	);
+});
+
+test("Ctrl+A 把光标挡在隐藏的 `!` 之后：继续打字不会顶掉 bash 模式", { skip, timeout: 30_000 }, async () => {
+	const editor = await mountEditor();
+	editor.setText("!ls");
+
+	editor.handleInput("\x01"); // ctrl+a
+	assert.deepEqual(editor.getCursor(), { line: 0, col: 1 }, "光标不该落到被摘掉的那一列");
+
+	editor.handleInput("c");
+	assert.equal(editor.getText(), "!cls", "插入点在第 1 列之后，`!` 还在最前面");
+	assert.ok(plainLines(editor.render(40))[1]?.startsWith("! cls"), "仍然是 bash 模式的画法");
+});
+
+test("普通正文的 Ctrl+A 不受影响：光标可以到第 0 列", { skip, timeout: 30_000 }, async () => {
+	const editor = await mountEditor();
+	editor.setText("ls");
+
+	editor.handleInput("\x01"); // ctrl+a
+	assert.deepEqual(editor.getCursor(), { line: 0, col: 0 });
+});
+
+test("点选：正文左移一列后坐标仍然对得上（点 gutter 也会被挡回第 1 列）", { skip, timeout: 30_000 }, async () => {
+	const editor = await mountEditor();
+	editor.setText("!ls");
+	editor.render(40); // handleMouse 依赖最近一次 render 的宽度与行数
+
+	// 可见正文：gutter 占 x=0/1，`l` 在 x=2，`s` 在 x=3
+	editor.handleMouse(mouseClick(3, 1));
+	assert.deepEqual(editor.getCursor(), { line: 0, col: 2 }, "点在 `s` 上 → 光标在 `s` 之前");
+
+	editor.handleMouse(mouseClick(2, 1));
+	assert.deepEqual(editor.getCursor(), { line: 0, col: 1 }, "点在 `l` 上 → 光标在 `l` 之前");
+
+	editor.handleMouse(mouseClick(0, 1));
+	assert.deepEqual(editor.getCursor(), { line: 0, col: 1 }, "点 gutter（隐藏列）被挡回第 1 列");
+});

package/extensions/prompt-editor.ts

@@ -0,0 +1,212 @@
+/**
+ * 输入框视觉层微调：① 类似 Claude Code 的 "❯" 提示符（`!` bash 模式换成 "!"）；
+ * ② 补全列表与 statusline 之间留一行空行。
+ *
+ * ① 提示符实现方式：继承 CustomEditor，把真正的编辑区收窄 gutter 列（super.render(width - gutter)），
+ * 再把 gutter 补回每一行的左侧：
+ *   - 上/下边框：由 renderTopBorder / renderBottomBorder 自己补满，保持边框满宽
+ *   - 第一行文本（未滚动时）→ "❯ "，bash 模式（正文以 `!` 开头）→ "! "
+ *   - 其余文本行 / 自动补全行 → 等宽空格
+ * 光标位置由行内的 CURSOR_MARKER 决定，跟着文本一起右移，所以硬件光标 / IME 定位不受影响；
+ * 鼠标点击坐标同样左移 gutter 列后再交给父类（bash 模式再多算一列隐藏的 `!`），点击定位保持准确。
+ *
+ * ② 补全列表与 statusline 的空行：Editor.render() 把补全列表渲染在**编辑器底边框之后**
+ * （editor.js：renderBottomBorder 之后才 push autocompleteList.render()），而 statusline 紧跟
+ * 在编辑器下方，所以输入 "/xxx" 时补全列表会紧贴 statusline。这里在补全列表真的渲染了的时候
+ * 追加一行空白行（只在有补全列表时补，静态布局不变）。判据用公开的 isShowingAutocomplete()
+ * （= autocompleteState !== null），而 Editor 里 state 与 autocompleteList 总是一起设置 / 一起清除
+ * （clearAutocompleteUi），所以它与“列表真的渲染了”严格同步，不必碰 private 的
+ * renderedAutocompleteHeight。空白行按 pi 自己的惯例补满宽度（Editor 每行都 pad 到
+ * contentWidth），避免差分渲染器只写变化前缀时留残影。
+ * 鼠标不受影响：空白行在补全列表之下（超出 autocompleteStartRow + renderedAutocompleteHeight），
+ * 落到 Editor.handleMouse 的“超出可见行”分支；Container 的鼠标分派用 render().length 算子组件
+ * 高度，所以高度 +1 后下方组件的坐标仍然对得上。
+ *
+ * 为什么不开第三个编辑器包装器扩展：`ctx.ui.setEditorComponent()` 是**替换**而非叠加，
+ * prompt-editor 的工厂会直接丢掉之前的包装（README 里 folder-history 就是被这个顺序坑过）。
+ * 所以这个改动放在已经拥有编辑器子类的本文件里，不新增任何加载顺序约束。
+ *
+ * Editor 的 theme / scrollOffset / renderedAutocompleteHeight 都是 private，
+ * 所以这里只用 protected 的 renderTopBorder/renderBottomBorder 拿到需要的信息。
+ *
+ * ③ 补全列表左移：Editor 把补全列表渲染在底边框之后，PromptEditor 又给每一行补了 gutter
+ * 列，于是命令词落在第 gutter+2 列（"→ " 前缀占 2 列）。这里让底边框之后的补全行少补
+ * shift 列空格（默认 shift = 1，即整个列表含选择箭头左移一列）。
+ * 渲染器对每条变化行先写 \x1b[2K 整行清除，补全行变短不会留残影；SelectList.handleMouse
+ * 只用 event.y 定位，左移不影响鼠标点选。
+ *
+ * ④ `!` bash 模式（对齐 Claude Code）：输入框第一个字符是 `!` 时，gutter 的 `❯` 换成 `!`，
+ * 正文里那个 `!` 不再显示 —— 于是正文看起来就是命令本身，而光标/IME/点选都还在它原来那一列。
+ * 模式判定、Enter 提交（`text.startsWith("!")`）、↑ 历史、Esc 清空全部沿用 pi 自己的链路
+ * （`interactive-mode.js` 的 `onChange` 是 `text.trimStart().startsWith("!")`，边框颜色用的就是它），
+ * 这里只把渲染层对齐过去：判定与摘除在 `./prompt-editor/bash-prompt.ts`（纯逻辑，可单测）。
+ * 三处细节：
+ *   - 正文一个字符都不动（隐藏只是画法），所以 pi 那边的颜色/提交/历史不需要任何配合；
+ *   - 隐藏掉一列后，正文整体左移一列 —— 渲染时把首行摘掉一个 `!` 再补一个空格回满宽，
+ *     鼠标点选要多算一列，光标如果停在那一列（Ctrl+A / 方向键 / 点 gutter）会被挡回第 1 列：
+ *     放进去的话反显光标会落在被摘掉的空列上，而在那儿打字会把 `x!ls` 写进正文，
+ *     pi 当场判定「不是 bash 模式了」；
+ *   - 退出模式不需要额外代码：正文为空时按回退键删掉 `!`（或 Enter 提交 / Esc 清空），
+ *     pi 那边 `isBashMode` 变回 false、边框颜色照旧逻辑回落，这里下一帧就画回 `❯`。
+ *
+ * 可用环境变量：
+ *   PI_EDITOR_PROMPT             提示符字符，默认 "❯"（bash 模式的 "!" 不跟着变）
+ *   PI_EDITOR_AUTOCOMPLETE_GAP   设为 off 关闭补全列表下方的空白行
+ *   PI_EDITOR_AUTOCOMPLETE_SHIFT 补全列表左移列数，0..gutter，默认 1；
+ *                                设 0 恢复旧行为，设 2（= gutter）则命令词对齐光标起始列
+ */
+
+import {
+	CustomEditor,
+	type ExtensionAPI,
+	type KeybindingsManager,
+} from "@earendil-works/pi-coding-agent";
+import type { EditorTheme, TUI, TuiMouseEvent, TuiMouseEventResult } from "@earendil-works/pi-tui";
+import { visibleWidth } from "@earendil-works/pi-tui";
+import { BASH_MARKER, dropLeadingBang, resolveBashPrompt } from "./prompt-editor/bash-prompt.ts";
+
+const PROMPT_CHAR = process.env.PI_EDITOR_PROMPT ?? "❯";
+const AUTOCOMPLETE_GAP = (process.env.PI_EDITOR_AUTOCOMPLETE_GAP ?? "").toLowerCase() !== "off";
+
+/** 补全列表左移列数覆盖值；未设置时默认左移 1 列。 */
+const AUTOCOMPLETE_SHIFT_OVERRIDE = (() => {
+	const raw = process.env.PI_EDITOR_AUTOCOMPLETE_SHIFT;
+	if (raw === undefined || raw.trim() === "") return undefined;
+	const parsed = Number.parseInt(raw, 10);
+	return Number.isFinite(parsed) ? parsed : undefined;
+})();
+
+class PromptEditor extends CustomEditor {
+	private readonly gutter: number;
+	/** 最近一次 renderBottomBorder 生成的整行，用来在 render() 里认出下边框 */
+	private bottomBorderLine = "";
+	/** 上边框带的 "↑ N more" 说明当前视图是滚动过的 */
+	private scrolled = false;
+
+	constructor(tui: TUI, theme: EditorTheme, keybindings: KeybindingsManager) {
+		super(tui, theme, keybindings, { paddingX: 0 });
+		this.gutter = visibleWidth(PROMPT_CHAR) + 1; // 提示符 + 一个空格
+	}
+
+	private get gutterFill(): string {
+		return this.borderColor("─".repeat(this.gutter));
+	}
+
+	/**
+	 * 提示符字形铺满 gutter 列（`❯` / `!` 都是 1 列，但 `PI_EDITOR_PROMPT` 可能是宽字符）。
+	 * 只把字形染色、补齐的空格不染 —— 和原来的 `${borderColor("❯")} ` 逐字一致。
+	 */
+	private promptCell(char: string): string {
+		const pad = Math.max(0, this.gutter - visibleWidth(char));
+		return this.borderColor(char) + " ".repeat(pad);
+	}
+
+	/**
+	 * 把光标挡在隐藏的 `!` 之外：Claude Code 里光标永远进不到模式标记上。
+	 * 放它进去有两个后果 —— 反显光标落在被摘掉的那一空列上（看起来光标消失），
+	 * 以及在那儿打字会把 `x!ls` 写进正文，pi 当场判定「不是 bash 模式了」。
+	 * 光标列住在 Editor private 的 state 里、没有公开 setter，所以直接调它自己的
+	 * setCursorCol(1)（顺带清掉 sticky 列）；pi 以后改名时这里退化成「光标停在 0 列」，
+	 * 只影响这个边角行为，不影响渲染。
+	 */
+	private keepCursorOffBang(): void {
+		if (!resolveBashPrompt(this.getText()).hidden) return;
+		const cursor = this.getCursor();
+		if (cursor.line !== 0 || cursor.col !== 0) return;
+
+		const internals = this as unknown as {
+			setCursorCol?: (col: number) => void;
+			state?: { cursorCol: number };
+		};
+		if (typeof internals.setCursorCol === "function") internals.setCursorCol(1);
+		else if (internals.state) internals.state.cursorCol = 1;
+	}
+
+	protected renderTopBorder(width: number, hiddenLineCount: number): string {
+		this.scrolled = hiddenLineCount > 0;
+		return this.gutterFill + super.renderTopBorder(width, hiddenLineCount);
+	}
+
+	protected renderBottomBorder(width: number, hiddenLineCount: number): string {
+		this.bottomBorderLine = this.gutterFill + super.renderBottomBorder(width, hiddenLineCount);
+		return this.bottomBorderLine;
+	}
+
+	render(width: number): string[] {
+		const gutter = this.gutter;
+		const lines = super.render(Math.max(1, width - gutter));
+		if (lines.length === 0) return lines;
+
+		// pi 的 bash 模式只由正文文本决定，不额外维护状态：每帧重算，永远不会和边框颜色脱节
+		const bash = resolveBashPrompt(this.getText());
+
+		const blank = " ".repeat(gutter);
+		// 滚动后第一可见行已经不是逻辑首行，不再显示提示符
+		const showPrompt = !this.scrolled;
+		// EditorTheme 只暴露 borderColor，所以提示符沿用边框色（bash 模式下就是 bashMode 色）
+		const prompt = this.promptCell(bash.active ? BASH_MARKER : PROMPT_CHAR);
+		// 要摘 `!` 的那一行正好就是显示提示符的那一行
+		const hideBang = bash.hidden && showPrompt;
+
+		// 补全列表左移：默认 1 列（含选择箭头一起左移）。
+		const shift = Math.max(0, Math.min(gutter, AUTOCOMPLETE_SHIFT_OVERRIDE ?? 1));
+		const autocompleteBlank = " ".repeat(gutter - shift);
+
+		let inAutocomplete = false;
+		const mapped = lines.map((line, index) => {
+			// 上边框已经在上面补满宽度
+			if (index === 0) return line;
+			// 下边框已补满宽度；其后的行都是补全列表行
+			if (line === this.bottomBorderLine) {
+				inAutocomplete = true;
+				return line;
+			}
+			if (inAutocomplete) return autocompleteBlank + line;
+			if (index === 1 && showPrompt) {
+				if (!hideBang) return prompt + line;
+				// 摘掉正文那个 `!` 后整行少 1 列，补一个空格回满宽（Editor 每行都铺满 contentWidth，
+				// 少一列会在差分渲染器下留残影）
+				const dropped = dropLeadingBang(line);
+				return prompt + dropped.text + (dropped.dropped ? " " : "");
+			}
+			return blank + line;
+		});
+
+		// 补全列表渲染在编辑器底边框之后，紧接着就是 statusline，视觉上拥挤；
+		// 只在补全列表真的存在时补一行空白行，静态布局不变。
+		// 按 pi 自己的惯例补满宽度（其余行也都是 gutter + contentWidth = width）。
+		if (AUTOCOMPLETE_GAP && this.isShowingAutocomplete()) {
+			mapped.push(" ".repeat(Math.max(0, width)));
+		}
+
+		return mapped;
+	}
+
+	handleInput(data: string): void {
+		super.handleInput(data);
+		// 任何一条能改光标/正文的路径（方向键、Home/Ctrl+A、↑ 历史、退格、粘贴、undo…）都从这里过
+		this.keepCursorOffBang();
+	}
+
+	handleMouse(event: TuiMouseEvent): TuiMouseEventResult | undefined {
+		// bash 模式下正文整体左移了 1 列（隐藏的 `!`），点选要往右多算 1 列才对得上原文。
+		// 补全列表的 x 不参与命中（SelectList 只用 y），多这 1 列不影响它。
+		const bangHitShift = resolveBashPrompt(this.getText()).hidden ? 1 : 0;
+		const result = super.handleMouse({
+			...event,
+			x: Math.max(0, event.x - this.gutter + bangHitShift),
+			width: Math.max(1, event.width - this.gutter),
+		});
+		this.keepCursorOffBang();
+		return result;
+	}
+}
+
+export default function (pi: ExtensionAPI) {
+	pi.on("session_start", (_event, ctx) => {
+		if (!ctx.hasUI) return;
+		ctx.ui.setEditorComponent(
+			(tui, theme, keybindings) => new PromptEditor(tui, theme, keybindings),
+		);
+	});
+}