pi-double-esc 1.0.13 → 1.0.15

package/README.md

@@ -101,3 +101,9 @@ npm run lint:dead   # check for unused exports
 ## License
 
 MIT
+
+## Pi 0.99 compatibility (1.0.15)
+
+Preserves single-Escape autocomplete dismissal, restores editor methods/factories and clears debounce timers on shutdown. Non-TUI sessions are untouched; cached render output is not mutated.
+
+Tested with Pi 0.99.0. Host-provided Pi packages are wildcard peers, not bundled dependencies; development uses exact 0.99.0 versions.

package/double-esc.ts

@@ -46,6 +46,7 @@ function decorateDoubleEscapeEditor(
   tui: TUI,
   appTheme: Theme,
   isIdle: () => boolean,
+  onCleanup: (cleanup: () => void) => void,
 ): EditorComponent {
   let escapeState: DoubleEscapeState = createInitialState();
   let debounceTimer: ReturnType<typeof setTimeout> | undefined;
@@ -55,8 +56,18 @@ function decorateDoubleEscapeEditor(
     debounceTimer = undefined;
   };
 
-  const handleInput = editor.handleInput.bind(editor);
-  editor.handleInput = (data: string): void => {
+  let active = true;
+  const originalHandleInput = editor.handleInput;
+  const originalRender = editor.render;
+  const handleInput = originalHandleInput.bind(editor);
+  const wrappedInput = (data: string): void => {
+    const autocomplete = editor as EditorComponent & { isShowingAutocomplete?: () => boolean };
+    if (!active || autocomplete.isShowingAutocomplete?.()) {
+      clearDebounce();
+      escapeState = createInitialState();
+      handleInput(data);
+      return;
+    }
     if (!matchesKey(data, "escape")) {
       if (escapeState.hintActive) {
         escapeState = handleOtherKey(escapeState).state;
@@ -86,7 +97,7 @@ function decorateDoubleEscapeEditor(
 
   const render = editor.render.bind(editor);
   editor.render = (width: number): string[] => {
-    const lines = render(width);
+    const lines = [...render(width)];
     if (!escapeState.hintActive || lines.length === 0) return lines;
 
     const label = " esc again to abort ";
@@ -104,6 +115,15 @@ function decorateDoubleEscapeEditor(
     return lines;
   };
 
+  editor.handleInput = wrappedInput;
+  const wrappedRender = editor.render;
+  onCleanup(() => {
+    active = false;
+    clearDebounce();
+    escapeState = createInitialState();
+    if (editor.handleInput === wrappedInput) editor.handleInput = originalHandleInput;
+    if (editor.render === wrappedRender) editor.render = originalRender;
+  });
   return editor;
 }
 
@@ -117,7 +137,11 @@ type EditorFactory = NonNullable<ReturnType<ExtensionContext["ui"]["getEditorCom
 type DecoratedEditorFactory = EditorFactory & { [DECORATED_FACTORY]?: true };
 
 export default function (pi: ExtensionAPI) {
+  const cleanups = new Set<() => void>();
+  let installed: DecoratedEditorFactory | undefined;
+  let previous: EditorFactory | undefined;
   pi.on("session_start", (_event, ctx) => {
+    if (ctx.mode !== "tui") return;
     const previousEditorFactory = ctx.ui.getEditorComponent() as DecoratedEditorFactory | undefined;
     if (previousEditorFactory?.[DECORATED_FACTORY]) return;
 
@@ -125,10 +149,19 @@ export default function (pi: ExtensionAPI) {
       const editor =
         previousEditorFactory?.(tui, editorTheme, keybindings) ??
         new CustomEditor(tui, editorTheme, keybindings);
-      return decorateDoubleEscapeEditor(editor, tui, ctx.ui.theme, () => ctx.isIdle());
+      return decorateDoubleEscapeEditor(editor, tui, ctx.ui.theme, () => ctx.isIdle(), (cleanup) => cleanups.add(cleanup));
     };
     editorFactory[DECORATED_FACTORY] = true;
 
+    previous = previousEditorFactory;
+    installed = editorFactory;
     ctx.ui.setEditorComponent(editorFactory);
   });
+  pi.on("session_shutdown", (_event, ctx) => {
+    for (const cleanup of cleanups) cleanup();
+    cleanups.clear();
+    if (installed && ctx.ui.getEditorComponent() === installed) ctx.ui.setEditorComponent(previous);
+    installed = undefined;
+    previous = undefined;
+  });
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "pi-double-esc",
-  "version": "1.0.13",
+  "version": "1.0.15",
   "description": "Prevent accidental Escape from aborting the LLM — requires double-press to interrupt while streaming",
   "type": "module",
   "author": "Tom X Nguyen",
@@ -37,8 +37,8 @@
     "lint:dead": "knip --no-gitignore"
   },
   "devDependencies": {
-    "@earendil-works/pi-coding-agent": "0.86.0",
-    "@earendil-works/pi-tui": "0.86.0",
+    "@earendil-works/pi-coding-agent": "0.99.0",
+    "@earendil-works/pi-tui": "0.99.0",
     "@types/node": "25.9.1",
     "@vitest/coverage-v8": "4.1.7",
     "knip": "6.14.1",
@@ -52,5 +52,9 @@
   },
   "overrides": {
     "brace-expansion": "5.0.6"
+  },
+  "peerDependencies": {
+    "@earendil-works/pi-coding-agent": "*",
+    "@earendil-works/pi-tui": "*"
   }
 }

package/vitest.config.ts

@@ -4,6 +4,10 @@ export default defineConfig({
   test: {
     globals: true,
     environment: "node",
+    // pi 0.87's module graph is much heavier to first-evaluate; each test
+    // file's first test pays the cold vi.resetModules() import, which can
+    // exceed the 5s default under load. Give the harness headroom.
+    testTimeout: 20_000,
     include: ["__tests__/**/*.test.ts"],
     exclude: ["node_modules", "dist", ".idea", ".git", ".cache"],
     coverage: {