@herbertgao/pi-extensions 2026.8.15 → 2026.9.0

package/node_modules/@narumitw/pi-btw/dist/index.ts

@@ -482,8 +482,10 @@ import {
 } from "@earendil-works/pi-coding-agent";
 import {
   isKeyRelease,
+  isKittyProtocolActive,
   Key as Key2,
   matchesKey as matchesKey2,
+  parseKey,
   TuiAltScreen,
   truncateToWidth as truncateToWidth2
 } from "@earendil-works/pi-tui";
@@ -495,6 +497,18 @@ function sanitizeSingleLine(text) {
     return code > 31 && (code < 127 || code > 159);
   }).join("").replace(/ +/gu, " ").trim();
 }
+function formatKeyLabel2(key) {
+  const sanitized = sanitizeSingleLine(key);
+  if (!sanitized) return "";
+  return sanitized.split("+").map((part) => {
+    const lower = part.toLowerCase();
+    if (lower === "shift") return "Shift";
+    if (lower === "ctrl") return "Ctrl";
+    if (lower === "alt") return "Alt";
+    if (lower === "super") return "Super";
+    return part.length === 1 ? part.toUpperCase() : `${part[0]?.toUpperCase() ?? ""}${part.slice(1)}`;
+  }).join("+");
+}
 
 // src/fullscreen-ui.ts
 var FullscreenUiDisposedError = class extends Error {
@@ -607,6 +621,88 @@ var BtwTuiAltScreen = class extends TuiAltScreen {
 };
 var BRACKETED_PASTE_START = "\x1B[200~";
 var BRACKETED_PASTE_END = "\x1B[201~";
+var ALT_SCREEN_ACTIONS_BEFORE_BOTTOM = [
+  "tui.altScreen.search",
+  "tui.altScreen.searchNext",
+  "tui.altScreen.searchPrevious",
+  "tui.altScreen.searchClose",
+  "tui.altScreen.pageUp",
+  "tui.altScreen.pageDown",
+  "tui.altScreen.halfPageUp",
+  "tui.altScreen.halfPageDown",
+  "tui.altScreen.lineUp",
+  "tui.altScreen.lineDown",
+  "tui.altScreen.previousPrompt",
+  "tui.altScreen.nextPrompt",
+  "tui.altScreen.top"
+];
+var KEY_MODIFIER_ORDER = ["shift", "ctrl", "alt", "super"];
+var MATCHABLE_SPECIAL_KEYS = /* @__PURE__ */ new Set([
+  "space",
+  "tab",
+  "enter",
+  "backspace",
+  "delete",
+  "insert",
+  "home",
+  "end",
+  "pageup",
+  "pagedown",
+  "up",
+  "down",
+  "left",
+  "right"
+]);
+var MATCHABLE_SYMBOL_KEYS = new Set("`-=[]\\;',./!@#$%^&*()_+|~{}:<>?");
+function normalizedKeyId(key) {
+  const parts = key.toLowerCase().split("+");
+  const base = parts.at(-1);
+  if (!base) return "";
+  const normalizedBase = base === "esc" ? "escape" : base === "return" ? "enter" : base;
+  const modifiers = KEY_MODIFIER_ORDER.filter((modifier) => parts.includes(modifier));
+  return [...modifiers, normalizedBase].join("+");
+}
+function formatEffectiveKeyLabel(key) {
+  const parts = key.split("+");
+  const base = parts.at(-1);
+  if (base === "pageup") parts[parts.length - 1] = "pageUp";
+  if (base === "pagedown") parts[parts.length - 1] = "pageDown";
+  return formatKeyLabel2(parts.join("+"));
+}
+function canMatchKeyInput(key) {
+  const parts = key.split("+");
+  const base = parts.at(-1) ?? "";
+  const modifiers = parts.slice(0, -1);
+  if (base === "escape") return modifiers.length === 0;
+  if (base === "clear") {
+    return modifiers.length === 0 || modifiers.length === 1 && (modifiers[0] === "shift" || modifiers[0] === "ctrl");
+  }
+  if (/^f(?:[1-9]|1[0-2])$/u.test(base)) return modifiers.length === 0;
+  return MATCHABLE_SPECIAL_KEYS.has(base) || base.length === 1 && (/^[a-z0-9]$/u.test(base) || MATCHABLE_SYMBOL_KEYS.has(base));
+}
+function rawCtrlInput(base) {
+  if (base.length !== 1) return void 0;
+  const rawBase = base === "-" ? "_" : base;
+  if (!"abcdefghijklmnopqrstuvwxyz[\\]_".includes(rawBase)) return void 0;
+  return String.fromCharCode(rawBase.charCodeAt(0) & 31);
+}
+function legacyRawInput(key) {
+  const parts = key.split("+");
+  const base = parts.at(-1) ?? "";
+  if (parts.length === 2 && parts[0] === "ctrl") return rawCtrlInput(base);
+  if (isKittyProtocolActive()) return void 0;
+  if (parts.length === 2 && parts[0] === "alt" && base.length === 1) return `\x1B${base}`;
+  if (parts.length === 3 && parts[0] === "ctrl" && parts[1] === "alt") {
+    const input = rawCtrlInput(base);
+    return input ? `\x1B${input}` : void 0;
+  }
+  return void 0;
+}
+function keyInputIdentity(key) {
+  const identity = normalizedKeyId(key);
+  const input = legacyRawInput(identity);
+  return input ? normalizedKeyId(parseKey(input) ?? identity) : identity;
+}
 function hasManualSelectionCopyApi() {
   return typeof TuiAltScreen.prototype.hasActiveSelection === "function" && typeof TuiAltScreen.prototype.copyActiveSelectionToClipboard === "function";
 }
@@ -625,6 +721,25 @@ function createBtwFullscreenTui(parent, theme, keybindings, copyOnSelect, manual
       mouse: true,
       copyOnSelect,
       searchMatchStyle: (text) => theme.underline(styleSearchMatch(text)),
+      scrollToEndIndicator: () => {
+        const unavailableKeyIdentities = /* @__PURE__ */ new Set([keyInputIdentity(Key2.ctrl("c"))]);
+        for (const action of ALT_SCREEN_ACTIONS_BEFORE_BOTTOM) {
+          for (const actionKey of keybindings.getKeys(action)) {
+            unavailableKeyIdentities.add(keyInputIdentity(String(actionKey)));
+          }
+        }
+        if (!copyOnSelect) {
+          for (const copyKey of keybindings.getKeys("app.message.copy")) {
+            unavailableKeyIdentities.add(keyInputIdentity(String(copyKey)));
+          }
+        }
+        const key = keybindings.getKeys("tui.altScreen.bottom").map((candidate) => keyInputIdentity(String(candidate))).find(
+          (identity) => identity && canMatchKeyInput(identity) && !unavailableKeyIdentities.has(identity) && formatEffectiveKeyLabel(identity)
+        );
+        const label = theme.fg("text", " \u2193 Jump to latest message");
+        const shortcut = key ? theme.fg("muted", ` \xB7 ${formatEffectiveKeyLabel(key)}`) : "";
+        return theme.bg("selectedBg", `${label}${shortcut} `);
+      },
       searchCurrentMatchStyle: (text) => theme.bold(theme.inverse(styleSearchMatch(text))),
       openUrl,
       copySelection: async (text) => {
@@ -693,9 +808,11 @@ var BtwFullscreenHost = class {
   disposed = false;
   finished = false;
   parentStopped = false;
-  parentRestarted = false;
+  parentRestoreAttempted = false;
   fullscreenCreated = false;
   fullscreenStopped = false;
+  parentRestoreQueued = false;
+  parentRestorePromise;
   cleanupError;
   setParentOverlay(overlay) {
     this.parentOverlay = overlay;
@@ -728,7 +845,7 @@ var BtwFullscreenHost = class {
         try {
           this.hardCancelActiveCustom?.();
         } finally {
-          this.restoreParent();
+          this.queueParentRestore();
         }
         return { consume: true };
       });
@@ -741,14 +858,33 @@ var BtwFullscreenHost = class {
     } catch (error) {
       this.cleanupError ??= error;
     }
-    this.restoreParent();
+    if (this.parentRestorePromise) await this.parentRestorePromise;
+    else this.restoreParent();
     if (this.cleanupError !== void 0) outcome = { kind: "failed", error: this.cleanupError };
     this.finished = true;
     this.done(outcome);
   }
+  queueParentRestore() {
+    if (this.parentRestoreQueued || this.parentRestoreAttempted) return;
+    this.parentRestoreQueued = true;
+    this.parentRestorePromise = Promise.resolve().then(async () => {
+      try {
+        await this.fullscreen?.terminal.drainInput?.();
+      } catch (error) {
+        this.cleanupError ??= error;
+      }
+      this.parentRestoreQueued = false;
+      this.restoreParent();
+    });
+  }
   restoreParent() {
-    this.removeHardCancelListener?.();
+    const removeHardCancelListener = this.removeHardCancelListener;
     this.removeHardCancelListener = void 0;
+    try {
+      removeHardCancelListener?.();
+    } catch (error) {
+      this.cleanupError ??= error;
+    }
     if (this.fullscreenCreated && !this.fullscreenStopped) {
       this.fullscreenStopped = true;
       try {
@@ -757,7 +893,7 @@ var BtwFullscreenHost = class {
         this.cleanupError ??= error;
       }
     }
-    if (!this.parentStopped || this.parentRestarted) return;
+    if (!this.parentStopped || this.parentRestoreAttempted) return;
     const parentOverlay = this.parentOverlay;
     this.parentOverlay = void 0;
     try {
@@ -766,8 +902,8 @@ var BtwFullscreenHost = class {
       this.cleanupError ??= error;
     }
     try {
+      this.parentRestoreAttempted = true;
       this.parent.start();
-      this.parentRestarted = true;
       this.parent.renderNow(false);
     } catch (error) {
       this.cleanupError ??= error;
@@ -2355,14 +2491,7 @@ function renderSideThreadHeader(width, theme, thinkingLevel) {
   return theme.fg("muted", `${title}${"\u2500".repeat(ruleWidth)}`);
 }
 function thinkingKeyLabel(keybindings) {
-  const key = sanitizeSingleLine(String(keybindings.getKeys("app.thinking.cycle")[0] ?? "shift+tab")) || "Shift+Tab";
-  return key.split("+").map((part) => {
-    const lower = part.toLowerCase();
-    if (lower === "shift") return "Shift";
-    if (lower === "ctrl") return "Ctrl";
-    if (lower === "alt") return "Alt";
-    return part.length === 1 ? part.toUpperCase() : `${part[0]?.toUpperCase() ?? ""}${part.slice(1)}`;
-  }).join("+");
+  return formatKeyLabel2(String(keybindings.getKeys("app.thinking.cycle")[0] ?? "shift+tab")) || "Shift+Tab";
 }
 function fitComposerLayout(header, contentLines, footer, editorLines, availableRows, statusLines = []) {
   const lines = [header, ...contentLines, ...statusLines, footer, ...editorLines];