mocode-ai 0.1.3 → 0.1.5

package/dist/ui/intervention.js

@@ -0,0 +1,318 @@
+import readline from 'node:readline';
+import { stdin, stderr } from 'node:process';
+import { ui } from './theme.js';
+import { displayWidth, truncateDisplay } from './render.js';
+import * as layout from './layout.js';
+import * as mouse from './mouse.js';
+import { Spinner } from './spinner.js';
+const emitter = stdin;
+/** choice 末尾的"自定义输入"项标签(纯 ASCII,宽度安全)。选中后切到 input 子态。 */
+const CUSTOM_LABEL = '其他(自定义输入)';
+/** 弹出介入面板,阻塞直到用户完成选择。 */
+export async function promptIntervention(req) {
+    // 非 TTY 降级:不阻塞,自动选默认。日志走 stderr(不污染 stdout 内容流)。
+    if (!layout.isActive()) {
+        const kind = req.type === 'choice' ? '自动选默认项' : '自动返回空输入';
+        stderr.write(`[介入] ${req.title}(非交互环境,${kind})\n`);
+        if (req.type === 'choice') {
+            return { action: 'selected', value: req.options?.[0] ?? '' };
+        }
+        return { action: 'submitted', value: req.seed ?? '' };
+    }
+    const options = req.type === 'choice' && Array.isArray(req.options)
+        ? req.options.map((o) => String(o)).filter((s) => s.length > 0)
+        : [];
+    // choice 但选项被滤空 → 降级 input(对齐设计文档 §8:ask_human 选项为空数组→input)。
+    const startMode = req.type === 'choice' && options.length > 0 ? 'choice' : 'input';
+    let mode = startMode;
+    // choice:选中下标(0..options.length-1 为各选项,options.length 为"自定义"项)。
+    let selected = 0;
+    // input:可编辑文本 + 光标(UTF-16 码元索引;显示位置由 displayWidth 算,见 paintInput)。
+    let text = req.seed ?? '';
+    let cursor = text.length;
+    // input 是否由 choice 的"自定义"项切入(决定 Esc 是返回 choice 还是取消)。
+    let cameFromChoice = false;
+    let resolved = false;
+    let resolve;
+    // 挂自己监听前快照的现有 keypress 监听(运行态即 onRunningKey),退出时按原序恢复。
+    let savedListeners = [];
+    /** choice 菜单行:标题(bold)+ detail(dim,行数上限保选项可见)+ 空行 + 选项(▸ 选中/dim)。 */
+    function menuLinesChoice() {
+        const g = layout.getGeo();
+        const cols = g.cols;
+        const items = [...options, CUSTOM_LABEL];
+        const optionCount = items.length;
+        // detail 上限:title(1)+detail(D)+空行(1)+options(O) ≤ contentBottom → D ≤ contentBottom-2-O
+        const detailCap = Math.max(0, g.contentBottom - 2 - optionCount);
+        const lines = [];
+        lines.push(`${ui.bold}${truncateDisplay(req.title, cols)}${ui.reset}`);
+        if (req.detail) {
+            const dl = req.detail.split('\n');
+            const shown = dl.slice(0, detailCap);
+            for (const d of shown) {
+                lines.push(`${ui.dim}${truncateDisplay(d, cols)}${ui.reset}`);
+            }
+            if (dl.length > detailCap)
+                lines.push(`${ui.dim}…${ui.reset}`);
+        }
+        lines.push(''); // 分隔空行
+        // 选项开窗:超屏高时以 selected 为中心取窗,保选中项可见。
+        const maxOptRows = Math.max(1, g.contentBottom - lines.length);
+        let start = 0;
+        if (optionCount > maxOptRows) {
+            start = Math.max(0, Math.min(selected - Math.floor(maxOptRows / 2), optionCount - maxOptRows));
+        }
+        const count = Math.min(maxOptRows, optionCount);
+        for (let i = 0; i < count; i++) {
+            const idx = start + i;
+            // 选中项:▸ 与正文均 cyan+bold(去 dim),未选中项保持 dim——选中行整体高亮。
+            const isSel = idx === selected;
+            const color = isSel ? `${ui.cyan}${ui.bold}` : ui.dim;
+            const marker = isSel ? `${ui.cyan}${ui.bold}▸${ui.reset}` : ' ';
+            const body = truncateDisplay(items[idx], cols - 2);
+            lines.push(`${marker} ${color}${body}${ui.reset}`);
+        }
+        return lines;
+    }
+    /** input 菜单行:标题(bold)+ detail(dim)+ 提示行(dim)。可编辑文本在底栏输入框。 */
+    function menuLinesInput() {
+        const g = layout.getGeo();
+        const cols = g.cols;
+        const detailCap = Math.max(0, g.contentBottom - 2); // title(1)+提示(1)
+        const lines = [];
+        lines.push(`${ui.bold}${truncateDisplay(req.title, cols)}${ui.reset}`);
+        if (req.detail) {
+            const dl = req.detail.split('\n');
+            const shown = dl.slice(0, detailCap);
+            for (const d of shown) {
+                lines.push(`${ui.dim}${truncateDisplay(d, cols)}${ui.reset}`);
+            }
+            if (dl.length > detailCap)
+                lines.push(`${ui.dim}…${ui.reset}`);
+        }
+        lines.push(`${ui.dim}Enter 提交 · Esc ${cameFromChoice ? '返回选项' : '取消'} · Ctrl+C 取消${ui.reset}`);
+        return lines;
+    }
+    function redraw() {
+        if (mode === 'choice') {
+            const hint = '↑↓ 选择 · Enter 确认 · 数字键直选 · Esc 取消';
+            layout.paintInput({
+                prompt: '❯ ',
+                lines: [hint],
+                cursorLine: 0,
+                cursorCol: displayWidth(hint),
+                menu: { lines: menuLinesChoice() },
+                caret: false, // 纯导航(非文本输入):不画输入框块状光标,聚焦由选项 ▸ 标记
+            });
+        }
+        else {
+            layout.paintInput({
+                prompt: '❯ ',
+                lines: [text],
+                cursorLine: 0,
+                cursorCol: displayWidth(text.slice(0, cursor)),
+                menu: { lines: menuLinesInput() },
+            });
+        }
+    }
+    function finish(result) {
+        if (resolved)
+            return;
+        resolved = true;
+        cleanup();
+        resolve(result);
+    }
+    /** 退出:摘自己的监听 + 恢复快照监听 + 擦菜单恢复内容区。不 setRawMode(false)/pause stdin(运行态由 repl 接管)。 */
+    function cleanup() {
+        emitter.removeListener('keypress', onKey);
+        for (const l of savedListeners)
+            emitter.on('keypress', l);
+        savedListeners = [];
+        // 先 paintInput(dim,menu:null):擦菜单(用 lastMenuRows)+ 画 dim 占位底栏 + 复位 lastMenuRows=0;
+        // 再处理滚动:面板内可能滚过滚轮(scrollOffset>0)→ resetScroll 回尾让 agent 后续输出可见;
+        //   未滚则 repaintViewport 从 content 缓冲重画内容区(恢复被菜单覆盖的 ● ask_human 行)。
+        // 顺序不可反——resetScroll/repaintViewport 在菜单还在(lastMenuRows>0)时会用内容覆盖菜单区,擦不掉菜单。
+        layout.paintInput({
+            prompt: '❯ ',
+            lines: [''],
+            cursorLine: 0,
+            cursorCol: 0,
+            menu: null,
+            dim: true,
+        });
+        if (layout.isScrolled())
+            layout.resetScroll();
+        else
+            layout.repaintViewport();
+    }
+    function onKey(_str, key) {
+        if (resolved || !key)
+            return;
+        // SGR 鼠标 fragment:readline 拆碎 \x1B[<…M,经 mouse.consumeMouse 重组;
+        // 滚轮 → scrollBy(±5) 滚动查看 agent 输出内容(菜单只靠 ↑↓ 选,滚轮不选菜单);
+        // fragment 期间 suppress 吞掉防砸进 onKeyInput 的可打印插入。菜单经 scrollBy 末尾
+        // repaint()→paintInput(lastView) 重画,不被 viewport 滚动覆盖。
+        const _m = mouse.consumeMouse(key.sequence ?? '');
+        if (_m.suppress) {
+            if (_m.wheel)
+                layout.scrollBy(_m.wheel * 5);
+            return;
+        }
+        // Ctrl+C → 取消(不 reject SIGINT——否则经 executeTool 的 try/catch 变成 tool 错误串)
+        if (key.ctrl && key.name === 'c') {
+            finish({ action: 'cancelled' });
+            return;
+        }
+        if (key.name === 'escape') {
+            if (mode === 'input' && cameFromChoice) {
+                // 由 choice"自定义"项切入的 input:Esc 返回选项(不取消整次提问)
+                mode = 'choice';
+                redraw();
+                return;
+            }
+            finish({ action: 'cancelled' });
+            return;
+        }
+        // PgUp/PgDn:翻页滚动查看 agent 输出内容(与 onRunningKey/prompt.ts 一致;↑↓ 仍归菜单导航,不抢)。
+        if (key.name === 'pageup' || key.name === 'pagedown') {
+            const pageH = layout.getGeo().contentBottom;
+            layout.scrollBy(key.name === 'pageup' ? pageH : -pageH);
+            return;
+        }
+        if (mode === 'choice') {
+            onKeyChoice(key);
+        }
+        else {
+            onKeyInput(key);
+        }
+    }
+    function onKeyChoice(key) {
+        const itemCount = options.length + 1; // +自定义项
+        switch (key.name) {
+            case 'up':
+                selected = (selected - 1 + itemCount) % itemCount;
+                redraw();
+                return;
+            case 'down':
+                selected = (selected + 1) % itemCount;
+                redraw();
+                return;
+            case 'return':
+            case 'enter':
+                if (selected === options.length) {
+                    // 自定义项 → 切 input 子态(空文本起)
+                    mode = 'input';
+                    cameFromChoice = true;
+                    text = '';
+                    cursor = 0;
+                    redraw();
+                }
+                else {
+                    finish({ action: 'selected', value: options[selected] });
+                }
+                return;
+        }
+        // 数字键 1-9 直选对应选项(自定义项不绑定数字)
+        const s = key.sequence ?? '';
+        if (s >= '1' && s <= '9') {
+            const n = Number(s) - 1;
+            if (n < options.length) {
+                finish({ action: 'selected', value: options[n] });
+            }
+        }
+    }
+    function onKeyInput(key) {
+        if (key.name === 'return' || key.name === 'enter') {
+            finish({ action: 'submitted', value: text });
+            return;
+        }
+        if (key.ctrl && key.name === 'a') {
+            cursor = 0;
+            redraw();
+            return;
+        }
+        if (key.ctrl && key.name === 'e') {
+            cursor = text.length;
+            redraw();
+            return;
+        }
+        switch (key.name) {
+            case 'backspace':
+                if (cursor > 0) {
+                    text = text.slice(0, cursor - 1) + text.slice(cursor);
+                    cursor--;
+                    redraw();
+                }
+                return;
+            case 'left':
+                if (cursor > 0) {
+                    cursor--;
+                    redraw();
+                }
+                return;
+            case 'right':
+                if (cursor < text.length) {
+                    cursor++;
+                    redraw();
+                }
+                return;
+            case 'home':
+                cursor = 0;
+                redraw();
+                return;
+            case 'end':
+                cursor = text.length;
+                redraw();
+                return;
+        }
+        // 可打印字符(>= 空格,非 ctrl/meta)→ 插入光标处(\n < ' ' 自动排除,保单行)
+        const s = key.sequence ?? '';
+        if (s && s >= ' ' && !key.ctrl && !key.meta) {
+            text = text.slice(0, cursor) + s + text.slice(cursor);
+            cursor += s.length;
+            redraw();
+        }
+    }
+    return new Promise((res, rej) => {
+        resolve = res;
+        try {
+            // 进入面板:停 spinner(避免 onFrame 覆盖)+ 回尾(若用户正滚动回看)
+            Spinner.pauseCurrent();
+            layout.resetScroll();
+            // 快照现有 keypress 监听(运行态的 onRunningKey)并摘掉,挂自己的 onKey
+            savedListeners = emitter.listeners('keypress').slice();
+            for (const l of savedListeners)
+                emitter.removeListener('keypress', l);
+            readline.emitKeypressEvents(stdin); // 幂等(运行态已挂解析器,防御性再调)
+            // raw 模式:runAgent 期间 repl 已设 true;防御性确保(失败则按键不来,但不崩)
+            try {
+                stdin.setRawMode(true);
+            }
+            catch {
+                // 非 TTY / 不支持:忽略(实际非 TTY 已在上方 isActive 守卫返回)
+            }
+            stdin.resume();
+            emitter.on('keypress', onKey);
+            redraw();
+        }
+        catch (e) {
+            // 进入失败:必须恢复运行态监听,否则 onRunningKey 残留摘除 → 本 turns 的 Ctrl+C/滚动/typeahead 全废
+            try {
+                emitter.removeListener('keypress', onKey);
+            }
+            catch {
+                // 忽略
+            }
+            for (const l of savedListeners) {
+                try {
+                    emitter.on('keypress', l);
+                }
+                catch {
+                    // 忽略
+                }
+            }
+            savedListeners = [];
+            rej(e instanceof Error ? e : new Error(String(e)));
+        }
+    });
+}