@zigai/pi-mode 0.1.2

package/README.md

@@ -0,0 +1,21 @@
+# Pi Mode
+
+This Pi extension adds prompt modes for model and thinking-level switching.
+
+## Install
+
+```sh
+pi install git:github.com/zigai/pi-ux-tweaks
+```
+
+## Features
+
+- Adds `/mode` for selecting and configuring prompt modes.
+- Adds `Ctrl+Shift+M` to select a mode.
+- Adds `Ctrl+Space` to cycle modes.
+- Can show the current mode in the prompt editor border when enabled.
+- Colors the prompt editor border from the active mode or thinking level.
+
+Modes can store a provider, model, thinking level, and optional color. Project-local modes live in `.pi/modes.json` when present; otherwise global modes live in `~/.pi/agent/modes.json`.
+
+By default, Pi Mode does not print the mode name in the editor border. To opt in, toggle `Show mode name` from `/mode` → `Configure modes…`, or set `"modeShowName": true` in `~/.pi/agent/settings.json`.

package/package.json

@@ -0,0 +1,42 @@
+{
+  "name": "@zigai/pi-mode",
+  "version": "0.1.2",
+  "description": "Pi package that adds prompt modes for model and thinking-level switching.",
+  "keywords": [
+    "mode",
+    "pi",
+    "pi-coding-agent",
+    "pi-extension",
+    "pi-package",
+    "pi-ux"
+  ],
+  "homepage": "https://github.com/zigai/pi-ux-tweaks/tree/main/packages/pi-mode#readme",
+  "bugs": {
+    "url": "https://github.com/zigai/pi-ux-tweaks/issues"
+  },
+  "license": "MIT",
+  "author": "zigai",
+  "repository": {
+    "type": "git",
+    "url": "git+https://github.com/zigai/pi-ux-tweaks.git",
+    "directory": "packages/pi-mode"
+  },
+  "files": [
+    "src",
+    "README.md"
+  ],
+  "type": "module",
+  "publishConfig": {
+    "access": "public"
+  },
+  "peerDependencies": {
+    "@earendil-works/pi-agent-core": "*",
+    "@earendil-works/pi-ai": "*",
+    "@earendil-works/pi-coding-agent": "*"
+  },
+  "pi": {
+    "extensions": [
+      "./src/index.ts"
+    ]
+  }
+}

package/src/constants.ts

@@ -0,0 +1,20 @@
+import type { ThinkingLevel } from "@earendil-works/pi-agent-core";
+
+export const DEFAULT_MODE_ORDER = ["default"] as const;
+export const CUSTOM_MODE_NAME = "custom" as const;
+
+export const MODE_UI_CONFIGURE = "Configure modes…";
+export const MODE_UI_ADD = "Add mode…";
+export const MODE_UI_SHOW_NAME_ON = "Show mode name: on";
+export const MODE_UI_SHOW_NAME_OFF = "Show mode name: off";
+export const MODE_UI_BACK = "Back";
+
+export const ALL_THINKING_LEVELS: ThinkingLevel[] = [
+    "off",
+    "minimal",
+    "low",
+    "medium",
+    "high",
+    "xhigh",
+];
+export const THINKING_UNSET_LABEL = "(don't change)";

package/src/editor.ts

@@ -0,0 +1,100 @@
+import {
+    CustomEditor,
+    type ExtensionAPI,
+    type ExtensionContext,
+} from "@earendil-works/pi-coding-agent";
+import { getCurrentMode, getModeBorderColor, setRequestEditorRender } from "./mode-state.ts";
+import { shouldShowModeName } from "./settings.ts";
+
+const ANSI_SGR = new RegExp(`${String.fromCharCode(27)}\\[[0-9;]*m`, "g");
+const stripAnsi = (value: string) => value.replace(ANSI_SGR, "");
+const MODE_FACTORY_BASE = Symbol.for("zigai.pi-mode.editor-factory-base");
+
+type EditorFactory = NonNullable<ReturnType<ExtensionContext["ui"]["getEditorComponent"]>>;
+
+type EditorLike = CustomEditor & {
+    borderColor: (text: string) => string;
+    getText(): string;
+    render(width: number): string[];
+};
+
+type WrappedEditorFactory = EditorFactory & {
+    [MODE_FACTORY_BASE]?: EditorFactory | undefined;
+};
+
+function modeLabelLine(lines: string[], width: number, editor: EditorLike): string[] {
+    if (!shouldShowModeName()) return lines;
+
+    const mode = getCurrentMode();
+    if (mode.length === 0) return lines;
+
+    const topPlain = stripAnsi(lines[0] ?? "");
+    const scrollPrefixMatch = /^(─── ↑ \d+ more )/.exec(topPlain);
+    const prefix = scrollPrefixMatch?.[1] ?? "──";
+
+    let label = mode;
+    let labelLeftSpace = " ";
+    if (prefix.endsWith(" ")) {
+        labelLeftSpace = "";
+    }
+    const labelRightSpace = " ";
+    const minRightBorder = 1;
+    const maxLabelLen = Math.max(
+        0,
+        width - prefix.length - labelLeftSpace.length - labelRightSpace.length - minRightBorder,
+    );
+    if (maxLabelLen <= 0) return lines;
+    if (label.length > maxLabelLen) label = label.slice(0, maxLabelLen);
+
+    const labelChunk = `${labelLeftSpace}${label}${labelRightSpace}`;
+    const remaining = width - prefix.length - labelChunk.length;
+    if (remaining < 0) return lines;
+
+    const right = "─".repeat(Math.max(0, remaining));
+    const borderColor = editor.borderColor;
+    lines[0] = borderColor(prefix) + borderColor(labelChunk) + borderColor(right);
+    return lines;
+}
+
+function enhanceEditor(
+    editor: EditorLike,
+    pi: ExtensionAPI,
+    ctx: ExtensionContext,
+    requestRender: () => void,
+): EditorLike {
+    const originalRender = editor.render.bind(editor);
+    editor.render = (width: number) => modeLabelLine(originalRender(width), width, editor);
+
+    const borderColor = (text: string) => {
+        const isBashMode = editor.getText().trimStart().startsWith("!");
+        if (isBashMode) {
+            return ctx.ui.theme.getBashModeBorderColor()(text);
+        }
+        return getModeBorderColor(ctx, pi, getCurrentMode())(text);
+    };
+
+    Object.defineProperty(editor, "borderColor", {
+        get: () => borderColor,
+        set: () => {},
+        configurable: true,
+        enumerable: true,
+    });
+
+    setRequestEditorRender(requestRender);
+    return editor;
+}
+
+export function applyModeEditor(pi: ExtensionAPI, ctx: ExtensionContext): void {
+    if (!ctx.hasUI) return;
+
+    const existing = ctx.ui.getEditorComponent() as WrappedEditorFactory | undefined;
+    const baseFactory = existing?.[MODE_FACTORY_BASE] ?? existing;
+    const factory = ((tui, theme, keybindings) => {
+        const editor = (baseFactory?.(tui, theme, keybindings) ??
+            new CustomEditor(tui, theme, keybindings)) as EditorLike;
+        return enhanceEditor(editor, pi, ctx, () => tui.requestRender());
+    }) as WrappedEditorFactory;
+    factory[MODE_FACTORY_BASE] = baseFactory;
+
+    ctx.ui.setEditorComponent(factory);
+}

package/src/index.ts

@@ -0,0 +1,41 @@
+import type { ExtensionAPI } from "@earendil-works/pi-coding-agent";
+import { applyModeEditor } from "./editor.ts";
+import {
+    cycleMode,
+    handleModeCommand,
+    handleModelSelect,
+    handleSessionActivated,
+    selectModeUI,
+} from "./mode-state.ts";
+
+export default function (pi: ExtensionAPI) {
+    pi.registerCommand("mode", {
+        description: "Select prompt mode",
+        handler: async (args, ctx) => {
+            await handleModeCommand(pi, ctx, args);
+        },
+    });
+
+    pi.registerShortcut("ctrl+shift+m", {
+        description: "Select prompt mode",
+        handler: async (ctx) => {
+            await selectModeUI(pi, ctx);
+        },
+    });
+
+    pi.registerShortcut("ctrl+space", {
+        description: "Cycle prompt mode",
+        handler: async (ctx) => {
+            await cycleMode(pi, ctx, 1);
+        },
+    });
+
+    pi.on("session_start", async (_event, ctx) => {
+        await handleSessionActivated(pi, ctx);
+        applyModeEditor(pi, ctx);
+    });
+
+    pi.on("model_select", async (event, ctx) => {
+        await handleModelSelect(pi, ctx, event);
+    });
+}