@selesai/code 0.1.0

package/dist/extensions/pi-powerline-footer/tests/jump-shortcuts.test.ts

@@ -0,0 +1,213 @@
+import test from "node:test";
+import assert from "node:assert/strict";
+import { readFileSync } from "node:fs";
+import { KEYBINDINGS } from "../node_modules/@earendil-works/pi-coding-agent/dist/core/keybindings.js";
+import {
+  isSupportedSuperShortcut,
+  matchesConfiguredShortcut,
+  shortcutConflictKey,
+} from "../shortcuts.ts";
+
+const source = readFileSync(new URL("../index.ts", import.meta.url), "utf-8");
+
+const powerlineShortcutKeys = new Set([
+  "stashHistory",
+  "copyEditor",
+  "cutEditor",
+  "jumpPreviousUserMessage",
+  "jumpNextUserMessage",
+  "jumpPreviousLlmMessage",
+  "jumpNextLlmMessage",
+  "jumpChatBottom",
+  "scrollChatUp",
+  "scrollChatDown",
+  "editorStart",
+  "editorEnd",
+]);
+
+function normalizeShortcut(shortcut: string): string {
+  const parts = shortcut.trim().toLowerCase().split("+");
+  if (parts.length <= 1) return parts[0] ?? "";
+
+  const modifierRank = new Map(["ctrl", "alt", "super", "shift"].map((modifier, index) => [modifier, index]));
+  return [
+    ...parts.slice(0, -1).sort((a, b) => (modifierRank.get(a) ?? 99) - (modifierRank.get(b) ?? 99)),
+    parts[parts.length - 1],
+  ].join("+");
+}
+
+function powerlineDefaults(): Map<string, string> {
+  const defaults = new Map<string, string>();
+  for (const match of source.matchAll(/^  ([a-zA-Z0-9]+): "([^"]+)",?$/gm)) {
+    const key = match[1];
+    const value = match[2];
+    if (key && value && powerlineShortcutKeys.has(key)) {
+      defaults.set(key, value);
+    }
+  }
+  return defaults;
+}
+
+test("chat jump shortcuts are configurable and route through fixed editor scrolling", () => {
+  const defaults = powerlineDefaults();
+  assert.equal(defaults.get("jumpPreviousUserMessage"), "ctrl+shift+u");
+  assert.equal(defaults.get("jumpNextUserMessage"), "ctrl+shift+i");
+  assert.equal(defaults.get("jumpPreviousLlmMessage"), "ctrl+alt+,");
+  assert.equal(defaults.get("jumpNextLlmMessage"), "ctrl+alt+.");
+  assert.equal(defaults.get("jumpChatBottom"), "ctrl+shift+g");
+  assert.equal(defaults.get("scrollChatUp"), "super+up");
+  assert.equal(defaults.get("scrollChatDown"), "super+down");
+  assert.equal(defaults.get("editorStart"), "super+shift+up");
+  assert.equal(defaults.get("editorEnd"), "super+shift+down");
+  assert.match(source, /const CHAT_JUMP_SHORTCUTS:/);
+  assert.match(source, /shortcutKey: "jumpPreviousUserMessage"/);
+  assert.match(source, /shortcutKey: "jumpNextUserMessage"/);
+  assert.match(source, /shortcutKey: "jumpPreviousLlmMessage"/);
+  assert.match(source, /shortcutKey: "jumpNextLlmMessage"/);
+  assert.match(source, /shortcutKey: "jumpChatBottom"/);
+  assert.match(source, /pi\.registerShortcut\(resolvedShortcuts\[shortcutKey\]/);
+  assert.match(source, /function collectChatMessageStartLines\(role: ChatJumpRole\): number\[\]/);
+  assert.match(source, /componentName === "UserMessageComponent"/);
+  assert.match(source, /componentName === "SkillInvocationMessageComponent"/);
+  assert.match(source, /componentName === "AssistantMessageComponent"/);
+  assert.match(source, /fixedEditorCompositor\.jumpToPreviousRootTarget\(targets\)/);
+  assert.match(source, /fixedEditorCompositor\.jumpToNextRootTarget\(targets\)/);
+  assert.match(source, /fixedEditorCompositor\.jumpToRootBottom\(\)/);
+  assert.match(source, /function getChatJumpShortcutAction\(data: string\): ChatJumpShortcutAction \| null/);
+  assert.match(source, /let resolvedShortcuts = resolveShortcutConfig\(startupSettings\)/);
+  assert.match(source, /resolvedShortcuts = resolveShortcutConfig\(settings\)/);
+  assert.match(source, /keyboardScrollShortcuts: \{\n\s+up: resolvedShortcuts\.scrollChatUp,\n\s+down: resolvedShortcuts\.scrollChatDown,/);
+  assert.match(source, /editorBoundaryShortcuts: \{\n\s+start: resolvedShortcuts\.editorStart,\n\s+end: resolvedShortcuts\.editorEnd,/);
+  assert.match(source, /modifier === "cmd" \|\| modifier === "command" \? "super" : modifier/);
+  assert.match(source, /shortcutUsesSuper\(normalizedShortcut\) && !isSupportedSuperShortcut\(normalizedShortcut\)/);
+  assert.match(source, /jumpToChatMessage\(ctx, action\.action\.role, action\.action\.direction\)/);
+});
+
+test("super shortcut matching rejects plain keys and unsupported command aliases", () => {
+  assert.equal(matchesConfiguredShortcut("c", "super+c"), false);
+  assert.equal(matchesConfiguredShortcut("X", "super+shift+x"), false);
+  assert.equal(matchesConfiguredShortcut("\x1b[A", "super+up"), false);
+  assert.equal(matchesConfiguredShortcut("\x1b[1;9A", "super+up"), true);
+  assert.equal(matchesConfiguredShortcut("\x1b[1;10A", "super+shift+up"), true);
+  assert.equal(matchesConfiguredShortcut("\x1b[122;9u", "super+z"), false);
+  assert.equal(isSupportedSuperShortcut("super+c"), false);
+  assert.equal(isSupportedSuperShortcut("super+shift+x"), false);
+  assert.equal(isSupportedSuperShortcut("super+z"), false);
+  assert.equal(isSupportedSuperShortcut("super+up"), true);
+  assert.equal(shortcutConflictKey("super+home"), "super+up");
+  assert.equal(shortcutConflictKey("super+end"), "super+down");
+  assert.equal(shortcutConflictKey("super+shift+home"), "super+shift+up");
+  assert.equal(shortcutConflictKey("super+shift+end"), "super+shift+down");
+});
+
+test("editor submits follow the fixed chat viewport to bottom", () => {
+  assert.match(source, /function followSubmittedEditorToBottom\(\): void/);
+  assert.match(source, /onEditorSubmit: \(\) => followSubmittedEditorToBottom\(\)/);
+  assert.match(source, /Object\.defineProperty\(editor, "onSubmit"/);
+  assert.match(source, /followSubmittedEditorToBottom\(\);\n\s+handler\(text\);/);
+  assert.match(source, /keybindings\.matches\(data, "app\.message\.followUp"\)/);
+});
+
+test("thinking level changes invalidate powerline status rendering", () => {
+  assert.match(source, /let currentThinkingLevel: string \| null = null/);
+  assert.match(source, /pi\.on\("thinking_level_select", async \(event, ctx\) => \{\n\s+currentCtx = ctx;\n\s+currentThinkingLevel = getThinkingLevelFn\?\.\(\) \?\? \(typeof event\.level === "string" \? event\.level : null\);\n\s+requestImmediateStatusRender\(\{ deferDuringTyping: false \}\);\n\s+\}\);/);
+  assert.match(source, /if \(e\.type === "thinking_level_change" && typeof e\.thinkingLevel === "string"\) \{\n\s+thinkingLevelFromSession = e\.thinkingLevel;/);
+  assert.match(source, /const thinkingLevel = currentThinkingLevel \?\? thinkingLevelFromSession \?\? getThinkingLevelFn\?\.\(\) \?\? "off"/);
+});
+
+test("context usage changes repaint from live streaming message usage", () => {
+  assert.match(source, /const CONTEXT_STATUS_RENDER_MS = 250/);
+  assert.match(source, /function getUsageTokenTotal\(usage: SessionAssistantUsage\): number/);
+  assert.match(source, /const totalTokens = "totalTokens" in usage && typeof usage\.totalTokens === "number"/);
+  assert.match(source, /return totalTokens \|\| usage\.input \+ usage\.output \+ usage\.cacheRead \+ usage\.cacheWrite/);
+  assert.match(source, /let liveAssistantUsage: SessionAssistantUsage \| null = null/);
+  assert.doesNotMatch(source, /const requestContextStatusRender/);
+  assert.match(source, /lastUserPrompt = "";\n\s+isStreaming = false;\n\s+liveAssistantUsage = null;\n\s+stashedEditorText = null;/);
+  assert.match(source, /pi\.on\("agent_start", async \(_event, ctx\) => \{\n\s+isStreaming = true;\n\s+liveAssistantUsage = null;\n\s+onVibeAgentStart\(\);\n\s+dismissWelcome\(ctx\);\n\s+currentCtx = ctx;\n\s+\}\);/);
+  assert.match(source, /pi\.on\("message_update", async \(event, ctx\) => \{\n\s+if \(isSessionAssistantMessage\(event\.message\)\n\s+&& event\.message\.stopReason !== "error"\n\s+&& event\.message\.stopReason !== "aborted"\n\s+&& getUsageTokenTotal\(event\.message\.usage\) > 0\) \{\n\s+liveAssistantUsage = event\.message\.usage;\n\s+currentCtx = ctx;\n\s+layoutDirty = true;\n\s+statusRenderScheduler\.schedule\(CONTEXT_STATUS_RENDER_MS\);\n\s+\}\n\s+\}\);/);
+  assert.match(source, /pi\.on\("message_end", async \(event, ctx\) => \{\n\s+currentCtx = ctx;\n\s+if \(isSessionAssistantMessage\(event\.message\)\) \{\n\s+if \(event\.message\.stopReason === "error" \|\| event\.message\.stopReason === "aborted"\) \{\n\s+liveAssistantUsage = null;\n\s+\} else if \(getUsageTokenTotal\(event\.message\.usage\) > 0\) \{\n\s+liveAssistantUsage = event\.message\.usage;\n\s+\}\n\s+\}\n\s+requestImmediateStatusRender\(\{ deferDuringTyping: false \}\);\n\s+\}\);/);
+  assert.match(source, /pi\.on\("agent_end", async \(_event, ctx\) => \{\n\s+isStreaming = false;\n\s+liveAssistantUsage = null;\n\s+currentCtx = ctx;/);
+  assert.match(source, /pi\.on\("session_tree", async \(_event, ctx\) => \{\n\s+currentCtx = ctx;\n\s+currentThinkingLevel = null;\n\s+liveAssistantUsage = null;\n\s+requestImmediateStatusRender\(\{ deferDuringTyping: false \}\);\n\s+\}\);/);
+  assert.match(source, /if \(getUsageTokenTotal\(m\.usage\) > 0\) \{\n\s+lastAssistant = m;\n\s+\}/);
+  assert.match(source, /const coreContextUsage = isStreaming && liveAssistantUsage \? null : readCoreContextUsage\(ctx\)/);
+  assert.match(source, /const contextTokens = coreContextUsage\?\.contextTokens \?\? \(latestUsage \? getUsageTokenTotal\(latestUsage\) : 0\)/);
+});
+
+test("extension status changes invalidate powerline status rendering", () => {
+  assert.match(source, /let forceNextLayoutRecompute = false/);
+  assert.match(source, /let restoreFooterStatusRepaintHook: \(\(\) => void\) \| null = null/);
+  assert.match(source, /const requestImmediateStatusRender = \(options: \{ deferDuringTyping\?: boolean \} = \{\}\) => \{/);
+  assert.match(source, /if \(options\.deferDuringTyping !== false && Date\.now\(\) - lastEditorInputAt < EDITOR_STATUS_DEFER_MS\) \{\n\s+statusRenderScheduler\.schedule\(\);\n\s+return;\n\s+\}/);
+  assert.match(source, /forceNextLayoutRecompute = true;\n\s+statusRenderScheduler\.cancel\(\);\n\s+statusRenderScheduler\.schedule\(0\);/);
+  assert.match(source, /const installFooterStatusRepaintHook = \(footerData: ReadonlyFooterDataProvider\) => \{/);
+  assert.match(source, /setExtensionStatus\?: \(key: string, text: string \| undefined\) => void/);
+  assert.match(source, /const setExtensionStatusAndRepaint = function setExtensionStatusAndRepaint/);
+  assert.match(source, /originalSetExtensionStatus\.call\(this, key, text\);\n\s+requestImmediateStatusRender\(\);/);
+  assert.match(source, /installFooterStatusRepaintHook\(footerData\);/);
+  assert.match(source, /if \(writableFooterData\.setExtensionStatus === setExtensionStatusAndRepaint\) \{\n\s+writableFooterData\.setExtensionStatus = originalSetExtensionStatus;/);
+  assert.match(source, /if \(clearExtensionStatusesAndRepaint && writableFooterData\.clearExtensionStatuses === clearExtensionStatusesAndRepaint\)/);
+  assert.match(source, /restoreFooterStatusRepaintHook\?\.\(\);\n\s+restoreFooterStatusRepaintHook = null;/);
+});
+
+test("fixed editor captures Pi status messages with the editor cluster", () => {
+  assert.match(source, /let fixedStatusContainer: any = null/);
+  assert.match(source, /const statusContainerCandidate = tuiChildren\[editorContainerMatch\.index - 2\] \?\? null/);
+  assert.match(source, /fixedStatusContainer = statusContainerCandidate && typeof statusContainerCandidate\.render === "function"/);
+  assert.match(source, /compositor\.renderHidden\(fixedStatusContainer, width\)\.filter\(\(line\) => visibleWidth\(line\) > 0\)/);
+  assert.match(source, /statusLines: \[\.\.\.aboveWidgetLines, \.\.\.renderPowerlineStatusLines\(width\), \.\.\.statusContainerLines\]/);
+  assert.match(source, /if \(fixedStatusContainer\?\.render\) compositor\.hideRenderable\(fixedStatusContainer\)/);
+  assert.match(source, /fixedStatusContainer = null/);
+});
+
+test("shutdown cleanup resets terminal modes even before compositor install", () => {
+  assert.match(source, /import \{ emergencyTerminalModeReset, TerminalSplitCompositor \}/);
+  assert.match(source, /const hadCompositor = fixedEditorCompositor !== null/);
+  assert.match(source, /if \(!hadCompositor && options\?\.resetExtendedKeyboardModes\)/);
+  assert.match(source, /process\.stdout\.write\(emergencyTerminalModeReset\(\)\)/);
+});
+
+test("powerline shortcut defaults do not claim reserved Pi shortcuts", () => {
+  const reservedKeys = new Map<string, string>();
+  for (const [id, definition] of Object.entries(KEYBINDINGS)) {
+    const keys = definition.defaultKeys === undefined
+      ? []
+      : Array.isArray(definition.defaultKeys)
+        ? definition.defaultKeys
+        : [definition.defaultKeys];
+    for (const key of keys) {
+      reservedKeys.set(normalizeShortcut(key), id);
+    }
+  }
+
+  for (const [name, shortcut] of powerlineDefaults()) {
+    const conflict = reservedKeys.get(normalizeShortcut(shortcut));
+    assert.equal(conflict, undefined, `${name} default ${shortcut} conflicts with ${conflict}`);
+  }
+});
+
+test("powerline fallback routing rejects reserved Pi shortcut defaults", () => {
+  assert.doesNotMatch(source, /KeybindingsManager/);
+  assert.match(source, /TUI_KEYBINDINGS/);
+  assert.match(source, /const APP_RESERVED_SHORTCUTS = \[/);
+  assert.match(source, /"alt\+enter"/);
+  assert.match(source, /"alt\+up"/);
+  assert.match(source, /"alt\+down"/);
+  assert.match(source, /"ctrl\+s"/);
+  assert.match(source, /"shift\+l"/);
+  assert.match(source, /for \(const definition of Object\.values\(TUI_KEYBINDINGS\)\)/);
+  assert.doesNotMatch(source, /RESERVED_TUI_KEYBINDING_IDS/);
+  assert.match(source, /const EXTRA_RESERVED_SHORTCUTS = \["alt\+s"\] as const/);
+  assert.match(source, /const SHORTCUT_MODIFIER_ORDER = \["ctrl", "alt", "super", "shift"\] as const/);
+  assert.match(source, /const SHORTCUT_MODIFIERS = new Set\(SHORTCUT_MODIFIER_ORDER\)/);
+  assert.match(source, /modifierRank\.get\(a\)/);
+  assert.match(source, /configuredToggleShortcut && !reservedShortcuts\(\)\.has\(shortcutUsageKey\(configuredToggleShortcut\)\)/);
+});
+
+test("powerline shortcuts have terminal-input fallback routing", () => {
+  assert.match(source, /function getPowerlineShortcutAction\(data: string\): PowerlineShortcutAction \| null/);
+  assert.match(source, /matchesConfiguredShortcut\(data, resolvedShortcuts\.stashHistory\)/);
+  assert.match(source, /matchesConfiguredShortcut\(data, resolvedShortcuts\.copyEditor\)/);
+  assert.match(source, /matchesConfiguredShortcut\(data, resolvedShortcuts\.cutEditor\)/);
+  assert.match(source, /matchesConfiguredShortcut\(data, bashModeSettings\.toggleShortcut\)/);
+  assert.match(source, /runPowerlineShortcut\(ctx, powerlineShortcutAction\)/);
+});

package/dist/extensions/pi-powerline-footer/tests/stash-shortcut.test.ts

@@ -0,0 +1,32 @@
+import test from "node:test";
+import assert from "node:assert/strict";
+import { readFileSync } from "node:fs";
+
+const source = readFileSync(new URL("../index.ts", import.meta.url), "utf-8");
+
+test("stash shortcut supports macOS Option+S character input", () => {
+  assert.match(source, /isKeyRelease, matchesKey, type AutocompleteProvider/);
+  assert.match(source, /pi\.registerShortcut\("alt\+s"/);
+  assert.match(source, /function isStashShortcutInput\(data: string\): boolean/);
+  assert.match(source, /if \(isKeyRelease\(data\)\) return false;/);
+  assert.match(source, /data === "ß"/);
+  assert.match(source, /data === "\\x1bs"/);
+  assert.match(source, /data === "\\x1bS"/);
+  assert.match(source, /27;3;115/);
+  assert.match(source, /matchesKey\(data, "alt\+s"\)/);
+  assert.match(source, /ctx\.ui\.onTerminalInput\(\(data: string\) =>/);
+  assert.match(source, /if \(isStashShortcutInput\(data\)\)/);
+  assert.match(source, /return \{ consume: true \};/);
+  assert.match(source, /function stashOrRestoreEditorText\(ctx: any\): void/);
+  assert.match(source, /if \(isStashShortcutInput\(data\)\)/);
+  assert.match(source, /stashOrRestoreEditorText\(ctx\);/);
+  assert.match(source, /function isPromptHistoryShortcutInput\(data: string\): boolean/);
+  assert.match(source, /if \(isKeyRelease\(data\)\) return null;/);
+  assert.match(source, /matchesConfiguredShortcut\(data, resolvedShortcuts\.stashHistory\)/);
+  assert.doesNotMatch(source, /data === "\\x1b\\b"/);
+  assert.doesNotMatch(source, /data === "\\x1b\\x7f"/);
+  assert.match(source, /104\(\?:/);
+  assert.match(source, /27;7;104/);
+  assert.match(source, /return \{ kind: "stashHistory" \};/);
+  assert.match(source, /void openStashHistory\(ctx\);/);
+});

package/dist/extensions/pi-powerline-footer/tests/thinking-segment.test.ts

@@ -0,0 +1,61 @@
+import test from "node:test";
+import assert from "node:assert/strict";
+import { renderSegment } from "../segments.ts";
+import type { ColorScheme, SegmentContext, ThemeLike } from "../types.ts";
+
+function hexAnsi(hex: `#${string}`): string {
+  const value = hex.slice(1);
+  const r = parseInt(value.slice(0, 2), 16);
+  const g = parseInt(value.slice(2, 4), 16);
+  const b = parseInt(value.slice(4, 6), 16);
+  return `\x1b[38;2;${r};${g};${b}m`;
+}
+
+function createSegmentContext(thinkingLevel: string, colors: ColorScheme): SegmentContext {
+  return {
+    model: undefined,
+    thinkingLevel,
+    sessionId: undefined,
+    usageStats: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0, cost: 0 },
+    contextPercent: 0,
+    contextWindow: 0,
+    autoCompactEnabled: true,
+    customCompactionEnabled: false,
+    usingSubscription: false,
+    sessionStartTime: Date.now(),
+    shellModeActive: false,
+    shellRunning: false,
+    shellName: null,
+    shellCwd: null,
+    git: { branch: null, staged: 0, unstaged: 0, untracked: 0 },
+    extensionStatuses: new Map(),
+    hiddenExtensionStatusKeys: new Set(),
+    customItemsById: new Map(),
+    options: {},
+    theme: {
+      fg() {
+        throw new Error("unexpected theme color lookup in thinking segment test");
+      },
+    } satisfies ThemeLike,
+    colors,
+  };
+}
+
+test("thinking segment uses per-level colors for off through medium", () => {
+  const colors: ColorScheme = {
+    thinking: "#111111",
+    thinkingMinimal: "#222222",
+    thinkingLow: "#333333",
+    thinkingMedium: "#444444",
+  };
+
+  const off = renderSegment("thinking", createSegmentContext("off", colors));
+  const minimal = renderSegment("thinking", createSegmentContext("minimal", colors));
+  const low = renderSegment("thinking", createSegmentContext("low", colors));
+  const medium = renderSegment("thinking", createSegmentContext("medium", colors));
+
+  assert.equal(off.content, `${hexAnsi("#111111")}think:off\x1b[0m`);
+  assert.equal(minimal.content, `${hexAnsi("#222222")}think:min\x1b[0m`);
+  assert.equal(low.content, `${hexAnsi("#333333")}think:low\x1b[0m`);
+  assert.equal(medium.content, `${hexAnsi("#444444")}think:med\x1b[0m`);
+});

package/dist/extensions/pi-powerline-footer/tests/working-vibes.test.ts

@@ -0,0 +1,226 @@
+import test from "node:test";
+import assert from "node:assert/strict";
+import { existsSync, mkdtempSync, mkdirSync, readFileSync, rmSync, symlinkSync } from "node:fs";
+import { tmpdir } from "node:os";
+import { join } from "node:path";
+
+const FAUX_PROVIDER_PATH = new URL("../node_modules/@earendil-works/pi-ai/dist/providers/faux.js", import.meta.url).href;
+
+function ensurePiModuleLinks(): { cleanup: () => void } {
+  const nodeModulesDir = join(process.cwd(), "node_modules", "@earendil-works");
+  mkdirSync(nodeModulesDir, { recursive: true });
+  const links = [
+    {
+      link: join(nodeModulesDir, "pi-coding-agent"),
+      target: "/opt/homebrew/lib/node_modules/@earendil-works/pi-coding-agent",
+    },
+    {
+      link: join(nodeModulesDir, "pi-ai"),
+      target: "/opt/homebrew/lib/node_modules/@earendil-works/pi-coding-agent/node_modules/@earendil-works/pi-ai",
+    },
+  ];
+
+  const createdLinks: string[] = [];
+  for (const { link, target } of links) {
+    if (!existsSync(link)) {
+      symlinkSync(target, link);
+      createdLinks.push(link);
+    }
+  }
+
+  return {
+    cleanup() {
+      for (const link of createdLinks.reverse()) {
+        if (existsSync(link)) {
+          rmSync(link, { recursive: true, force: true });
+        }
+      }
+    },
+  };
+}
+
+test("generateVibesBatch includes a system prompt so faux providers can return text", async () => {
+  const links = ensurePiModuleLinks();
+  const home = mkdtempSync(join(tmpdir(), "powerline-vibes-home-"));
+  const previousHome = process.env.HOME;
+  process.env.HOME = home;
+
+  try {
+    const { fauxAssistantMessage, registerFauxProvider } = await import(FAUX_PROVIDER_PATH);
+    const { generateVibesBatch, initVibeManager, setVibeModel } = await import("../working-vibes.ts");
+
+    const registration = registerFauxProvider({
+      provider: "test-provider",
+      models: [{ id: "test-model" }],
+    });
+
+    try {
+      const model = registration.getModel("test-model");
+      assert.ok(model);
+
+      registration.setResponses([
+        (context) => {
+          assert.match(context.systemPrompt ?? "", /loading messages/i);
+          return fauxAssistantMessage("Engaging warp drive...\nRunning diagnostics...");
+        },
+      ]);
+
+      initVibeManager({
+        modelRegistry: {
+          find(provider: string, modelId: string) {
+            return provider === "test-provider" && modelId === "test-model" ? model : undefined;
+          },
+          async getApiKeyAndHeaders() {
+            return { ok: true, apiKey: "test-key", headers: {} };
+          },
+        },
+      });
+
+      assert.equal(setVibeModel("test-provider/test-model"), true);
+
+      const result = await generateVibesBatch("star trek", 2);
+
+      assert.equal(result.success, true);
+      assert.equal(result.count, 2);
+      assert.equal(existsSync(result.filePath), true);
+      assert.deepEqual(readFileSync(result.filePath, "utf8").trim().split("\n"), [
+        "Engaging warp drive...",
+        "Running diagnostics...",
+      ]);
+    } finally {
+      registration.unregister();
+    }
+  } finally {
+    if (previousHome === undefined) {
+      delete process.env.HOME;
+    } else {
+      process.env.HOME = previousHome;
+    }
+    rmSync(home, { recursive: true, force: true });
+    links.cleanup();
+  }
+});
+
+test("on-demand vibe generation includes a system prompt for providers that require instructions", async () => {
+  const links = ensurePiModuleLinks();
+  const home = mkdtempSync(join(tmpdir(), "powerline-vibes-home-"));
+  const previousHome = process.env.HOME;
+  process.env.HOME = home;
+
+  try {
+    const { fauxAssistantMessage, registerFauxProvider } = await import(FAUX_PROVIDER_PATH);
+    const { initVibeManager, onVibeAgentStart, onVibeBeforeAgentStart, setVibeModel, setVibeTheme } = await import("../working-vibes.ts");
+
+    const registration = registerFauxProvider({
+      provider: "test-provider",
+      models: [{ id: "test-model" }],
+    });
+
+    try {
+      const model = registration.getModel("test-model");
+      assert.ok(model);
+
+      registration.setResponses([
+        (context) => {
+          assert.match(context.systemPrompt ?? "", /loading messages/i);
+          return fauxAssistantMessage("Engaging warp drive...");
+        },
+      ]);
+
+      initVibeManager({
+        modelRegistry: {
+          find(provider: string, modelId: string) {
+            return provider === "test-provider" && modelId === "test-model" ? model : undefined;
+          },
+          async getApiKeyAndHeaders() {
+            return { ok: true, apiKey: "test-key", headers: {} };
+          },
+        },
+      });
+
+      assert.equal(setVibeTheme("star trek"), true);
+      assert.equal(setVibeModel("test-provider/test-model"), true);
+
+      const updates: Array<string | undefined> = [];
+      onVibeAgentStart();
+      onVibeBeforeAgentStart("fix a bug", (message) => {
+        updates.push(message);
+      });
+
+      const start = Date.now();
+      while (!updates.includes("Engaging warp drive...") && Date.now() - start < 1000) {
+        await new Promise((resolve) => setTimeout(resolve, 10));
+      }
+
+      assert.equal(updates[0], "Channeling star trek...");
+      assert.ok(updates.includes("Engaging warp drive..."));
+    } finally {
+      registration.unregister();
+    }
+  } finally {
+    if (previousHome === undefined) {
+      delete process.env.HOME;
+    } else {
+      process.env.HOME = previousHome;
+    }
+    rmSync(home, { recursive: true, force: true });
+    links.cleanup();
+  }
+});
+
+test("generateVibesBatch preserves provider errors instead of reporting an empty response", async () => {
+  const links = ensurePiModuleLinks();
+  const home = mkdtempSync(join(tmpdir(), "powerline-vibes-home-"));
+  const previousHome = process.env.HOME;
+  process.env.HOME = home;
+
+  try {
+    const { fauxAssistantMessage, registerFauxProvider } = await import(FAUX_PROVIDER_PATH);
+    const { generateVibesBatch, initVibeManager, setVibeModel } = await import("../working-vibes.ts");
+
+    const registration = registerFauxProvider({
+      provider: "test-provider",
+      models: [{ id: "test-model" }],
+    });
+
+    try {
+      const model = registration.getModel("test-model");
+      assert.ok(model);
+
+      registration.setResponses([
+        fauxAssistantMessage([], {
+          stopReason: "error",
+          errorMessage: "Instructions are required",
+        }),
+      ]);
+
+      initVibeManager({
+        modelRegistry: {
+          find(provider: string, modelId: string) {
+            return provider === "test-provider" && modelId === "test-model" ? model : undefined;
+          },
+          async getApiKeyAndHeaders() {
+            return { ok: true, apiKey: "test-key", headers: {} };
+          },
+        },
+      });
+
+      assert.equal(setVibeModel("test-provider/test-model"), true);
+
+      const result = await generateVibesBatch("noir", 2);
+
+      assert.equal(result.success, false);
+      assert.equal(result.error, "Instructions are required");
+    } finally {
+      registration.unregister();
+    }
+  } finally {
+    if (previousHome === undefined) {
+      delete process.env.HOME;
+    } else {
+      process.env.HOME = previousHome;
+    }
+    rmSync(home, { recursive: true, force: true });
+    links.cleanup();
+  }
+});

package/dist/extensions/pi-powerline-footer/theme.example.json

@@ -0,0 +1,24 @@
+{
+  "colors": {
+    "model": "#d787af",
+    "shellMode": "accent",
+    "path": "#00afaf",
+    "gitDirty": "warning",
+    "gitClean": "success",
+    "thinking": "thinkingOff",
+    "thinkingMinimal": "thinkingMinimal",
+    "thinkingLow": "thinkingLow",
+    "thinkingMedium": "thinkingMedium",
+    "context": "dim",
+    "contextWarn": "warning",
+    "contextError": "error",
+    "cost": "text",
+    "tokens": "muted",
+    "separator": "dim",
+    "border": "borderMuted"
+  },
+  "icons": {
+    "auto": "↯",
+    "warning": ""
+  }
+}

package/dist/extensions/pi-powerline-footer/theme.json

@@ -0,0 +1,12 @@
+{
+  "colors": {
+    "context": "#00c853",
+    "contextWarn": "#ff9800",
+    "contextError": "#f44336"
+  },
+  "icons": {
+    "folder": "",
+    "context": "",
+    "auto": ""
+  }
+}