shariq-pi-extensions 0.2.7 → 0.2.8

package/README.md

@@ -31,6 +31,7 @@ The package contains:
 - Firecrawl search and scraping
 - Git status UI
 - persistent task goals
+- configurable steer, interrupt, or follow-up input behavior
 - dedicated multi-agent orchestration
 - Pi Memory
 - per-response TPS, TTFT, elapsed-time, and output status

package/docs/EXTENSIONS.md

@@ -80,6 +80,10 @@ Adds repository state to Pi's interface. `/lg` opens the local Git view and `/pr
 
 `/copy-all` copies the current conversation in a readable form while omitting tool protocol noise that does not belong in the transcript.
 
+### [Input mode](../extensions/input-mode/README.md)
+
+`/input-mode` chooses what ordinary Enter does while Pi is running: steer before the next model step, interrupt the active run, or wait as a follow-up. The global selection lives in `<agent-dir>/input-mode.json`; explicit Alt+Enter follow-ups and non-interactive inputs retain native behavior. Pi does not expose extension rows in core `/settings`, so this is a dedicated extension settings picker rather than a private TUI patch.
+
 ### [Shell shortcuts](../extensions/shell-shortcuts/README.md)
 
 Adds `/exit` as an alias for Pi's normal quit command. Keep this extension limited to small, low-risk conveniences.

package/extensions/input-mode/README.md

@@ -0,0 +1,13 @@
+# Input mode
+
+Controls what ordinary **Enter** does when the interactive Pi agent is already running:
+
+- `steer` (default) queues the message before the agent's next model step.
+- `interrupt` signals Pi's active abort controller immediately, then preserves the submitted text, images, prompt-template expansion, and normal delivery as the next input.
+- `follow-up` queues the message until the active run finishes.
+
+Use `/input-mode` for the picker or `/input-mode steer|interrupt|follow-up` for direct selection. The global choice is stored with restrictive permissions in `<agent-dir>/input-mode.json`; non-default modes appear in Pi's status area.
+
+Pi does not expose an extension API for adding rows to its built-in `/settings` selector, so this extension owns a dedicated settings command rather than patching private TUI internals. The existing core **Steering mode** and **Follow-up mode** settings remain batching controls (`all` versus `one-at-a-time`).
+
+Explicit Alt+Enter follow-ups, extension-originated messages, commands, idle input, compaction input, and RPC input retain Pi's native behavior. Interrupt cancellation is cooperative: Pi stops model streaming and abort-aware tools, but it cannot undo an external side effect that already completed or force a third-party operation that ignores its abort signal to stop.

package/extensions/input-mode/config.ts

@@ -0,0 +1,46 @@
+import * as fs from "node:fs";
+import * as path from "node:path";
+import { getAgentDir } from "@earendil-works/pi-coding-agent";
+
+export const INPUT_MODES = ["steer", "interrupt", "follow-up"] as const;
+export type InputMode = (typeof INPUT_MODES)[number];
+
+interface InputModeDocument {
+  version: 1;
+  mode: InputMode;
+}
+
+export function inputModePath(): string {
+  return path.join(getAgentDir(), "input-mode.json");
+}
+
+export function isInputMode(value: unknown): value is InputMode {
+  return typeof value === "string" && INPUT_MODES.includes(value as InputMode);
+}
+
+export function loadInputMode(file = inputModePath()): InputMode {
+  try {
+    const document = JSON.parse(fs.readFileSync(file, "utf8")) as Partial<InputModeDocument>;
+    return document.version === 1 && isInputMode(document.mode) ? document.mode : "steer";
+  } catch {
+    return "steer";
+  }
+}
+
+export function saveInputMode(mode: InputMode, file = inputModePath()): void {
+  const directory = path.dirname(file);
+  fs.mkdirSync(directory, { recursive: true, mode: 0o700 });
+  const temporary = `${file}.${process.pid}.${Date.now()}.tmp`;
+  const document: InputModeDocument = { version: 1, mode };
+  try {
+    fs.writeFileSync(temporary, `${JSON.stringify(document, null, 2)}\n`, { mode: 0o600 });
+    fs.renameSync(temporary, file);
+    fs.chmodSync(file, 0o600);
+  } finally {
+    try {
+      fs.rmSync(temporary, { force: true });
+    } catch {
+      // Best-effort cleanup after a failed atomic replacement.
+    }
+  }
+}

package/extensions/input-mode/index.ts

@@ -0,0 +1,104 @@
+import type {
+  ExtensionAPI,
+  ExtensionCommandContext,
+  ExtensionUIContext,
+  InputEvent,
+  InputEventResult,
+} from "@earendil-works/pi-coding-agent";
+import {
+  INPUT_MODES,
+  isInputMode,
+  loadInputMode,
+  saveInputMode,
+  type InputMode,
+} from "./config.ts";
+
+const STATUS_KEY = "input-mode";
+
+interface InputModeExtensionOptions {
+  configFile?: string;
+}
+
+function modelInput(event: InputEvent) {
+  if (!event.images?.length) return event.text;
+  return [{ type: "text" as const, text: event.text }, ...event.images];
+}
+
+export function createInputModeExtension(options: InputModeExtensionOptions = {}) {
+  return (pi: ExtensionAPI) => {
+    let mode = loadInputMode(options.configFile);
+    let ui: ExtensionUIContext | undefined;
+
+    const updateStatus = () => {
+      if (!ui) return;
+      ui.setStatus(STATUS_KEY, mode === "steer" ? undefined : `input: ${mode}`);
+    };
+
+    const selectMode = async (args: string, ctx: ExtensionCommandContext) => {
+      const requested = args.trim().toLowerCase();
+      let selected: InputMode | undefined;
+      if (requested) {
+        if (!isInputMode(requested)) {
+          ctx.ui.notify(`Usage: /input-mode [${INPUT_MODES.join("|")}]`, "warning");
+          return;
+        }
+        selected = requested;
+      } else if (ctx.hasUI) {
+        selected = await ctx.ui.select(
+          `Input behavior while agent is running (current: ${mode})`,
+          [...INPUT_MODES],
+        ) as InputMode | undefined;
+      } else {
+        ctx.ui.notify(`Input mode: ${mode}. Usage: /input-mode [${INPUT_MODES.join("|")}]`, "info");
+        return;
+      }
+      if (!selected) return;
+      mode = selected;
+      saveInputMode(mode, options.configFile);
+      updateStatus();
+      const explanation = mode === "interrupt"
+        ? "new Enter input aborts the active run before it is delivered"
+        : mode === "follow-up"
+          ? "new Enter input waits until the active run finishes"
+          : "new Enter input is injected before the agent's next step";
+      ctx.ui.notify(`Input mode: ${mode} — ${explanation}.`, "info");
+    };
+
+    pi.registerCommand("input-mode", {
+      description: "Choose Enter behavior while the agent runs: steer, interrupt, or follow-up",
+      handler: selectMode,
+    });
+
+    pi.on("session_start", (_event, ctx) => {
+      if (ctx.hasUI) ui = ctx.ui;
+      updateStatus();
+    });
+
+    pi.on("session_shutdown", () => {
+      ui?.setStatus(STATUS_KEY, undefined);
+      ui = undefined;
+    });
+
+    pi.on("input", (event, ctx): InputEventResult => {
+      // Extension-originated results and explicit Alt+Enter follow-ups retain
+      // their requested delivery. Idle input and commands use Pi unchanged.
+      if (event.source !== "interactive" || event.streamingBehavior !== "steer") {
+        return { action: "continue" };
+      }
+      if (mode === "steer") return { action: "continue" };
+      if (mode === "follow-up") {
+        pi.sendUserMessage(modelInput(event), { deliverAs: "followUp" });
+        return { action: "handled" };
+      }
+
+      // Abort is signalled synchronously. Returning continue preserves Pi's
+      // normal template expansion, image handling, history, and queue logic;
+      // the prompt becomes a fresh turn if abort settlement wins the race, or
+      // a steering message consumed immediately after the aborted run.
+      ctx.abort();
+      return { action: "continue" };
+    });
+  };
+}
+
+export default createInputModeExtension();

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "shariq-pi-extensions",
-  "version": "0.2.7",
+  "version": "0.2.8",
   "description": "Cross-platform extension suite for the Pi coding agent.",
   "license": "MIT",
   "author": "Shariq Riaz",
@@ -47,6 +47,7 @@
       "./extensions/firecrawl-web/index.ts",
       "./extensions/git-info/index.ts",
       "./extensions/goal/index.ts",
+      "./extensions/input-mode/index.ts",
       "./extensions/orchestration/index.ts",
       "./extensions/performance-status/index.ts",
       "./extensions/pi-memory/index.ts",