@herbertgao/pi-extensions 2026.8.14 → 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 {
@@ -503,10 +517,13 @@ var FullscreenUiDisposedError = class extends Error {
     this.name = "FullscreenUiDisposedError";
   }
 };
-async function runBtwFullscreen(ctx, run, dependencies = {}) {
-  const createTui = dependencies.createTui ?? ((parent, theme) => createBtwFullscreenTui(
+async function runBtwFullscreen(ctx, run, options = {}, dependencies = {}) {
+  const createTui = dependencies.createTui ?? ((parent, theme, keybindings, fullscreenOptions) => createBtwFullscreenTui(
     parent,
     theme,
+    keybindings,
+    fullscreenOptions.copyOnSelect ?? true,
+    dependencies.manualSelectionCopySupported ?? hasManualSelectionCopyApi(),
     dependencies.openUrl ?? openUrlInBrowser,
     dependencies.copyToClipboard ?? copyToHostClipboard
   ));
@@ -529,7 +546,8 @@ async function runBtwFullscreen(ctx, run, dependencies = {}) {
           }
           done(value);
         },
-        createTui
+        createTui,
+        options
       );
       return host;
     },
@@ -550,7 +568,7 @@ async function runBtwFullscreen(ctx, run, dependencies = {}) {
 var btwInputListeners = /* @__PURE__ */ new WeakMap();
 function dispatchBtwInput(listeners, data) {
   let current = data;
-  for (const group of [listeners.beforeViewport, listeners.regular]) {
+  for (const group of [listeners.beforeAll, listeners.beforeViewport, listeners.regular]) {
     for (const listener of group) {
       const result = listener(current);
       if (result?.consume) return result;
@@ -560,10 +578,14 @@ function dispatchBtwInput(listeners, data) {
   return current === data ? void 0 : { data: current };
 }
 var BtwTuiAltScreen = class extends TuiAltScreen {
+  hasFocusedOverlay() {
+    return this.isOverlayFocused();
+  }
   addInputListener(listener) {
     let listeners = btwInputListeners.get(this);
     if (!listeners) {
       const registeredListeners = {
+        beforeAll: /* @__PURE__ */ new Set(),
         beforeViewport: /* @__PURE__ */ new Set(),
         regular: /* @__PURE__ */ new Set()
       };
@@ -574,6 +596,12 @@ var BtwTuiAltScreen = class extends TuiAltScreen {
     listeners.regular.add(listener);
     return () => listeners.regular.delete(listener);
   }
+  addInputListenerBeforeAll(listener) {
+    const listeners = btwInputListeners.get(this);
+    if (!listeners) return super.addInputListener(listener);
+    listeners.beforeAll.add(listener);
+    return () => listeners.beforeAll.delete(listener);
+  }
   addInputListenerBeforeViewport(listener) {
     const listeners = btwInputListeners.get(this);
     if (!listeners) return super.addInputListener(listener);
@@ -586,26 +614,165 @@ var BtwTuiAltScreen = class extends TuiAltScreen {
       super.removeInputListener(listener);
       return;
     }
+    listeners.beforeAll.delete(listener);
     listeners.beforeViewport.delete(listener);
     listeners.regular.delete(listener);
   }
 };
-function createBtwFullscreenTui(parent, theme, openUrl, copyToClipboard2) {
+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";
+}
+function createBtwFullscreenTui(parent, theme, keybindings, copyOnSelect, manualSelectionCopySupported, openUrl, copyToClipboard2) {
+  if (!copyOnSelect && !manualSelectionCopySupported) {
+    throw new Error(
+      "Manual fullscreen selection copying is unavailable in this Pi version; update Pi or enable automatic selection copying."
+    );
+  }
   const styleSearchMatch = (text) => theme.bg("searchMatchBg", theme.fg("searchMatchText", text));
-  return new BtwTuiAltScreen(parent.terminal, parent.getShowHardwareCursor(), void 0, {
-    mouse: true,
-    searchMatchStyle: (text) => theme.underline(styleSearchMatch(text)),
-    searchCurrentMatchStyle: (text) => theme.bold(theme.inverse(styleSearchMatch(text))),
-    openUrl,
-    copySelection: async (text) => {
-      try {
-        await copyToClipboard2(text);
-        return true;
-      } catch {
-        return false;
+  const fullscreen = new BtwTuiAltScreen(
+    parent.terminal,
+    parent.getShowHardwareCursor(),
+    void 0,
+    {
+      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) => {
+        try {
+          await copyToClipboard2(text);
+          return true;
+        } catch {
+          return false;
+        }
       }
     }
-  });
+  );
+  if (!copyOnSelect) {
+    let isInBracketedPaste = false;
+    fullscreen.addInputListenerBeforeViewport((data) => {
+      const wasInBracketedPaste = isInBracketedPaste;
+      const startsBracketedPaste = data.includes(BRACKETED_PASTE_START);
+      if (startsBracketedPaste) isInBracketedPaste = true;
+      if (isInBracketedPaste && data.includes(BRACKETED_PASTE_END)) {
+        isInBracketedPaste = false;
+      }
+      if (wasInBracketedPaste || startsBracketedPaste || fullscreen.hasFocusedOverlay() || isKeyRelease(data) || !keybindings.matches(data, "app.message.copy")) {
+        return void 0;
+      }
+      if (!fullscreen.hasActiveSelection()) {
+        fullscreen.flash("No selection to copy");
+        return { consume: true };
+      }
+      void fullscreen.copyActiveSelectionToClipboard().catch(() => fullscreen.flash("Copy failed"));
+      return { consume: true };
+    });
+  }
+  return fullscreen;
 }
 function openUrlInBrowser(target) {
   const [command, args] = process.platform === "darwin" ? ["open", [target]] : process.platform === "win32" ? ["rundll32", ["url.dll,FileProtocolHandler", target]] : ["xdg-open", [target]];
@@ -613,7 +780,7 @@ function openUrlInBrowser(target) {
   }).unref();
 }
 var BtwFullscreenHost = class {
-  constructor(parent, theme, keybindings, ctx, run, done, createTui) {
+  constructor(parent, theme, keybindings, ctx, run, done, createTui, options) {
     this.parent = parent;
     this.theme = theme;
     this.keybindings = keybindings;
@@ -621,6 +788,7 @@ var BtwFullscreenHost = class {
     this.run = run;
     this.done = done;
     this.createTui = createTui;
+    this.options = options;
     queueMicrotask(() => void this.start());
   }
   parent;
@@ -630,6 +798,7 @@ var BtwFullscreenHost = class {
   run;
   done;
   createTui;
+  options;
   fullscreen;
   parentOverlay;
   cancelActiveCustom;
@@ -639,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;
@@ -665,16 +836,16 @@ var BtwFullscreenHost = class {
       this.parent.stop({ preserveScreen: true });
       this.parentStopped = true;
       if (this.disposed) throw new FullscreenUiDisposedError();
-      this.fullscreen = this.createTui(this.parent, this.theme);
+      this.fullscreen = this.createTui(this.parent, this.theme, this.keybindings, this.options);
       this.fullscreenCreated = true;
       this.fullscreen.start();
-      const addHardCancelListener = this.fullscreen.addInputListenerBeforeViewport?.bind(this.fullscreen) ?? this.fullscreen.addInputListener.bind(this.fullscreen);
+      const addHardCancelListener = this.fullscreen.addInputListenerBeforeAll?.bind(this.fullscreen) ?? this.fullscreen.addInputListenerBeforeViewport?.bind(this.fullscreen) ?? this.fullscreen.addInputListener.bind(this.fullscreen);
       this.removeHardCancelListener = addHardCancelListener((data) => {
         if (isKeyRelease(data) || !matchesKey2(data, Key2.ctrl("c"))) return void 0;
         try {
           this.hardCancelActiveCustom?.();
         } finally {
-          this.restoreParent();
+          this.queueParentRestore();
         }
         return { consume: true };
       });
@@ -687,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 {
@@ -703,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 {
@@ -712,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;
@@ -1020,6 +1210,7 @@ Use the provided conversation context only as background. Answer the user's side
 
 // src/settings.ts
 var BTW_SETTINGS_FILE = "pi-btw.json";
+var DEFAULT_FULLSCREEN_COPY_ON_SELECT = true;
 var DEFAULT_REMEMBER_THINKING_LEVEL_CHANGES = true;
 var MAX_SETTINGS_BYTES = 64 * 1024;
 var mutationQueues = /* @__PURE__ */ new Map();
@@ -1044,6 +1235,11 @@ function normalizeBtwSettings(value) {
     if (typeof remember !== "boolean") return void 0;
     settings.rememberThinkingLevelChanges = remember;
   }
+  if (Object.hasOwn(value, "fullscreenCopyOnSelect")) {
+    const copyOnSelect = Reflect.get(value, "fullscreenCopyOnSelect");
+    if (typeof copyOnSelect !== "boolean") return void 0;
+    settings.fullscreenCopyOnSelect = copyOnSelect;
+  }
   return settings;
 }
 function parseBtwModelReference(reference) {
@@ -1052,6 +1248,9 @@ function parseBtwModelReference(reference) {
   if (separator <= 0 || separator === reference.length - 1) return void 0;
   return { provider: reference.slice(0, separator), modelId: reference.slice(separator + 1) };
 }
+function effectiveFullscreenCopyOnSelect(settings) {
+  return settings.fullscreenCopyOnSelect ?? DEFAULT_FULLSCREEN_COPY_ON_SELECT;
+}
 function effectiveRememberThinkingLevelChanges(settings) {
   return settings.rememberThinkingLevelChanges ?? DEFAULT_REMEMBER_THINKING_LEVEL_CHANGES;
 }
@@ -1189,6 +1388,10 @@ function applyBtwSettingsPatch(current, patch) {
   if (Object.hasOwn(patch, "rememberThinkingLevelChanges")) {
     updated.rememberThinkingLevelChanges = patch.rememberThinkingLevelChanges;
   }
+  if (Object.hasOwn(patch, "fullscreenCopyOnSelect")) {
+    if (patch.fullscreenCopyOnSelect === void 0) delete updated.fullscreenCopyOnSelect;
+    else updated.fullscreenCopyOnSelect = patch.fullscreenCopyOnSelect;
+  }
   return updated;
 }
 function isSettingsDocument(value) {
@@ -1246,7 +1449,8 @@ async function showBtwCommandMenu(ctx, options) {
         kind: "actions",
         title: "Pi BTW",
         lines: [
-          `Thinking: ${displayThinkingSummary(state.settings)} \xB7 Remember changes: ${displayRememberSummary(state.settings)}`
+          `Thinking: ${displayThinkingSummary(state.settings)} \xB7 Remember changes: ${displayRememberSummary(state.settings)}`,
+          `Copy on select: ${effectiveFullscreenCopyOnSelect(state.settings) ? "On" : "Off"}`
         ],
         items: [
           {
@@ -1272,7 +1476,7 @@ async function showBtwCommandMenu(ctx, options) {
           {
             id: "settings",
             label: "Settings",
-            description: "Choose pi-btw thinking level and fixed-level shortcut memory",
+            description: "Choose thinking, shortcut memory, and selection copying",
             to: state.kind === "invalid" ? "invalid" : "settings"
           }
         ],
@@ -1311,6 +1515,14 @@ async function showBtwCommandMenu(ctx, options) {
             currentValue: effectiveRememberThinkingLevelChanges(state.settings) ? "On" : "Off",
             values: ["On", "Off"],
             action: "set-remember"
+          },
+          {
+            id: "fullscreenCopyOnSelect",
+            label: "Copy selection automatically",
+            description: "Copy mouse selections immediately instead of with the configured copy key.",
+            currentValue: effectiveFullscreenCopyOnSelect(state.settings) ? "On" : "Off",
+            values: ["On", "Off"],
+            action: "set-fullscreen-copy"
           }
         ]
       }),
@@ -1368,6 +1580,21 @@ async function showBtwCommandMenu(ctx, options) {
           if (!signal.aborted) notifySaveFailure(ctx, error);
           return { kind: "rejected" };
         }
+      },
+      "set-fullscreen-copy": async ({ value, signal }) => {
+        if (value !== "On" && value !== "Off") return { kind: "rejected" };
+        try {
+          await updateSettings(
+            { fullscreenCopyOnSelect: value === "On" },
+            { settingsPath, signal }
+          );
+          if (signal.aborted) return { kind: "rejected" };
+          notifySafely(ctx, `Copy selection automatically: ${value}.`, "info");
+          return { kind: "stay" };
+        } catch (error) {
+          if (!signal.aborted) notifySaveFailure(ctx, error);
+          return { kind: "rejected" };
+        }
       }
     }
   });
@@ -2264,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];
@@ -2475,29 +2695,33 @@ function btw(pi, dependencies = {}) {
       }
       const startingTurnCount = state?.thread.turns.length ?? 0;
       try {
-        await runFullscreen(ctx, (fullscreenCtx) => {
-          if (!state) {
-            const createdAt = Date.now();
-            state = {
-              id: `btw-${nextThreadNumber}`,
-              thread: createSideThread(
-                selectedConversationContext ?? buildConversationContext(fullscreenCtx.sessionManager.getBranch())
-              ),
-              thinkingLevel: settings.thinkingLevel ?? pi.getThinkingLevel(),
-              createdAt,
-              updatedAt: createdAt
-            };
-            nextThreadNumber += 1;
-          }
-          return runThread({
-            initialQuestion: question || void 0,
-            selected: resolution.selected,
-            thinkingLevel: state.thinkingLevel,
-            rememberThinkingLevelChanges: !sameAsMainThinkingLevel && effectiveRememberThinkingLevelChanges(settings),
-            state,
-            ctx: fullscreenCtx
-          });
-        });
+        await runFullscreen(
+          ctx,
+          (fullscreenCtx) => {
+            if (!state) {
+              const createdAt = Date.now();
+              state = {
+                id: `btw-${nextThreadNumber}`,
+                thread: createSideThread(
+                  selectedConversationContext ?? buildConversationContext(fullscreenCtx.sessionManager.getBranch())
+                ),
+                thinkingLevel: settings.thinkingLevel ?? pi.getThinkingLevel(),
+                createdAt,
+                updatedAt: createdAt
+              };
+              nextThreadNumber += 1;
+            }
+            return runThread({
+              initialQuestion: question || void 0,
+              selected: resolution.selected,
+              thinkingLevel: state.thinkingLevel,
+              rememberThinkingLevelChanges: !sameAsMainThinkingLevel && effectiveRememberThinkingLevelChanges(settings),
+              state,
+              ctx: fullscreenCtx
+            });
+          },
+          { copyOnSelect: effectiveFullscreenCopyOnSelect(settings) }
+        );
       } finally {
         if (state?.title && state.thread.turns.length > 0) {
           if (state.thread.turns.length > startingTurnCount) {