@astrofoundry/pi-astro 0.13.0 → 0.13.1

package/extensions/astro-footer/index.test.ts

@@ -43,6 +43,7 @@ interface FakeCtx {
     ui: {
         notify: ReturnType<typeof vi.fn>;
         setFooter: ReturnType<typeof vi.fn>;
+        setWidget: ReturnType<typeof vi.fn>;
     };
     cwd: string;
     model: { id: string } | undefined;
@@ -56,6 +57,7 @@ function makeCtx(overrides: Partial<FakeCtx> = {}, oauth = false): FakeCtx {
         ui: {
             notify: vi.fn(),
             setFooter: vi.fn(),
+            setWidget: vi.fn(),
         },
         cwd: "/Users/astro/proj",
         model: { id: "anthropic/claude-sonnet-4-6" },
@@ -123,6 +125,55 @@ describe("astro-footer extension", () => {
         expect(pi.commands.has("footer")).toBe(true);
         expect(pi.handlers.session_start).toBeDefined();
         expect(pi.handlers.tool_result).toBeDefined();
+        expect(pi.handlers.before_agent_start).toBeDefined();
+    });
+
+    it("registers a last-prompt widget below the editor on activate", async () => {
+        const { pi } = await install();
+        const ctx = makeCtx();
+        await pi.handlers.session_start({}, ctx);
+        const calls = ctx.ui.setWidget.mock.calls;
+        const lastPromptCall = calls.find((c) => c[0] === "astro-footer-last-prompt");
+        expect(lastPromptCall).toBeDefined();
+        expect(lastPromptCall![2]).toEqual({ placement: "belowEditor" });
+    });
+
+    it("last-prompt widget renders nothing before the first prompt", async () => {
+        const { pi } = await install();
+        const ctx = makeCtx();
+        await pi.handlers.session_start({}, ctx);
+        const widgetCall = ctx.ui.setWidget.mock.calls.find((c) => c[0] === "astro-footer-last-prompt");
+        const factory = widgetCall![1] as (
+            tui: unknown,
+            theme: unknown,
+        ) => { render: (w: number) => string[] };
+        const lines = factory({}, fakeTheme).render(120);
+        expect(lines).toEqual([]);
+    });
+
+    it("last-prompt widget renders the last submitted prompt with arrow prefix", async () => {
+        const { pi } = await install();
+        const ctx = makeCtx();
+        await pi.handlers.session_start({}, ctx);
+        await pi.handlers.before_agent_start({ prompt: "fix the auth middleware token expiry check" }, ctx);
+        const widgetCall = ctx.ui.setWidget.mock.calls.find((c) => c[0] === "astro-footer-last-prompt");
+        const factory = widgetCall![1] as (
+            tui: unknown,
+            theme: unknown,
+        ) => { render: (w: number) => string[] };
+        const lines = factory({}, fakeTheme).render(120);
+        expect(lines).toHaveLength(1);
+        expect(lines[0]).toContain("↳");
+        expect(lines[0]).toContain("fix the auth middleware");
+    });
+
+    it("/footer off clears the last-prompt widget too", async () => {
+        const { pi } = await install();
+        const ctx = makeCtx();
+        await pi.commands.get("footer")!.handler("off", ctx);
+        const widgetCalls = ctx.ui.setWidget.mock.calls;
+        const cleared = widgetCalls.find((c) => c[0] === "astro-footer-last-prompt" && c[1] === undefined);
+        expect(cleared).toBeDefined();
     });
 
     it("session_start activates the footer via ctx.ui.setFooter", async () => {

package/extensions/astro-footer/index.ts

@@ -4,6 +4,7 @@ import type {
     ReadonlyFooterDataProvider,
 } from "@mariozechner/pi-coding-agent";
 import type { AssistantMessage } from "@mariozechner/pi-ai";
+import type { Theme } from "@mariozechner/pi-coding-agent";
 import { truncateToWidth, visibleWidth } from "@mariozechner/pi-tui";
 import { resolvePathMode, type PathMode } from "./format.ts";
 import { GitStatusTracker } from "./git-status.ts";
@@ -27,6 +28,17 @@ import {
 } from "./segments.ts";
 
 const SUBAGENT_STATUS_KEY = "subagents";
+const LAST_PROMPT_WIDGET_KEY = "astro-footer-last-prompt";
+
+function renderLastPromptLine(theme: Theme, lastPrompt: string, width: number): string[] {
+    const prefix = ` ${theme.fg("dim", "↳")} `;
+    const available = width - visibleWidth(prefix);
+    if (available < 10) return [];
+    const collapsed = lastPrompt.replace(/\s+/g, " ").trim();
+    if (collapsed.length === 0) return [];
+    const truncated = truncateToWidth(collapsed, available, "…");
+    return [truncateToWidth(`${prefix}${theme.fg("dim", truncated)}`, width, "…")];
+}
 
 interface UsageTotals {
     input: number;
@@ -162,6 +174,7 @@ export default function astroFooterExtension(pi: ExtensionAPI): void {
     let autoCompact = true;
     let activeRequestRender: (() => void) | null = null;
     let timeTickHandle: ReturnType<typeof setInterval> | null = null;
+    let lastPrompt = "";
 
     const gitTracker = new GitStatusTracker(() => {
         activeRequestRender?.();
@@ -184,6 +197,17 @@ export default function astroFooterExtension(pi: ExtensionAPI): void {
         sessionStartedAt = Date.now();
         autoCompact = isAutoCompactEnabled(ctx.cwd);
         gitTracker.refresh(ctx.cwd, true);
+        ctx.ui.setWidget(
+            LAST_PROMPT_WIDGET_KEY,
+            (_tui, theme) => ({
+                invalidate() {},
+                render(width: number): string[] {
+                    if (!lastPrompt) return [];
+                    return renderLastPromptLine(theme, lastPrompt, width);
+                },
+            }),
+            { placement: "belowEditor" },
+        );
         ctx.ui.setFooter((tui, theme, footerData) => {
             activeRequestRender = () => tui.requestRender();
             startTimeTicker();
@@ -215,9 +239,15 @@ export default function astroFooterExtension(pi: ExtensionAPI): void {
     }
 
     pi.on("session_start", async (_event, ctx) => {
+        lastPrompt = "";
         if (enabled) activate(ctx);
     });
 
+    pi.on("before_agent_start", async (event) => {
+        lastPrompt = event.prompt ?? "";
+        activeRequestRender?.();
+    });
+
     pi.on("tool_result", async (event, ctx) => {
         if (!enabled) return;
         if (event.toolName === "edit" || event.toolName === "write" || event.toolName === "bash") {
@@ -256,6 +286,7 @@ export default function astroFooterExtension(pi: ExtensionAPI): void {
                 enabled = false;
                 stopTimeTicker();
                 ctx.ui.setFooter(undefined);
+                ctx.ui.setWidget(LAST_PROMPT_WIDGET_KEY, undefined);
                 ctx.ui.notify("astro-footer: off (default footer restored)", "info");
                 return;
             }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@astrofoundry/pi-astro",
-  "version": "0.13.0",
+  "version": "0.13.1",
   "description": "Personal pi customizations (extensions, skills, prompts, themes) for the pi coding agent.",
   "keywords": [
     "pi-package"