pi-input-bar 1.0.0 → 1.1.0

package/README.md

@@ -34,6 +34,7 @@ No dependencies, no build step, nothing to configure — the bar appears as soon
 - Current folder and git branch
 - A dot that pulses while the agent is responding (`●`)
 - `provider/model` and thinking effort level (color-coded)
+- If [pi-loop-police](https://github.com/sebaxzero/pi-loop-police) is installed and has caught loops this session, a summary appears after the dot: `⚠ 10 loops (tool 7, think 3)`. No loop-police (or no loops) — nothing shown.
 
 **Bottom border** — what's happening under the hood:
 - **Context usage** — fill level, colored green → yellow → red as it fills
@@ -65,6 +66,7 @@ Persistent configuration lives in `extensions/input-bar.json` next to the instal
 | `ANIM` | `true` | Random working word while streaming |
 | `ICONS` | `true` | Glyphs (⌂ ⎇ ⛁ ⚡) in the bars |
 | `WORDS` | `""` | Comma-separated custom working-word pool |
+| `LOOPS` | `true` | Summary of pi-loop-police detections in the top border (only shown once a loop is caught) |
 
 ## License
 

package/extensions/input-bar.ts

@@ -31,6 +31,7 @@ import {
   composeBottomLine,
   composeLine,
   DEFAULT_WORDS,
+  loopSummary,
   pickWord,
   replaceEdgeChars,
   stripAnsi,
@@ -51,6 +52,7 @@ const DEFAULTS = {
   ANIM: true, // random working word while streaming
   ICONS: true, // glyphs (⌂ ⎇ ⛁ ⚡) in the bars
   WORDS: "", // comma-separated override for the working-word pool
+  LOOPS: true, // loop-police detection summary in the top border (if installed)
 };
 
 const cfg: typeof DEFAULTS = (() => {
@@ -122,6 +124,18 @@ export default function (pi: ExtensionAPI) {
   let ctxRef: ExtensionContext | undefined;
   // FooterDataProvider captured from the setFooter factory; the editor reads it too.
   let footerData: { getGitBranch(): string | null } | undefined;
+  // TUI handle captured from the editor factory, to repaint on bus events.
+  let tuiRef: { requestRender(): void } | undefined;
+
+  // Per-session loop-police detections, keyed by detector name (payload.event).
+  // If pi-loop-police isn't installed the event never fires and the map stays
+  // empty — the bar renders exactly as before.
+  const loopCounts = new Map<string, number>();
+  (pi as any).events?.on?.("loop-police:detection", (data: any) => {
+    const kind = typeof data?.event === "string" ? data.event : "unknown";
+    loopCounts.set(kind, (loopCounts.get(kind) ?? 0) + 1);
+    tuiRef?.requestRender();
+  });
 
   // t/s tracking. The clock starts at the FIRST streamed token, not at
   // message_start — otherwise prompt-processing time (long on local servers)
@@ -170,6 +184,7 @@ export default function (pi: ExtensionAPI) {
       provider: ctx?.model?.provider,
       modelId: ctx?.model?.id,
       effort: ctx?.model?.reasoning ? effort : undefined,
+      loops: cfg.LOOPS ? loopSummary(loopCounts) : undefined,
       icons: cfg.ICONS,
     });
     const line = composeLine(left, right, width, makeColorize(theme, effort), tuiVisibleWidth);
@@ -272,7 +287,10 @@ export default function (pi: ExtensionAPI) {
       // paddingX: 2 keeps a spare space column on each side of the content so
       // the │ sides can be painted without touching text or the cursor cell.
       ctx.ui.setEditorComponent(
-        (tui, theme, keybindings) => new BarEditor(tui, theme, keybindings, { paddingX: 2 }),
+        (tui, theme, keybindings) => {
+          tuiRef = tui;
+          return new BarEditor(tui, theme, keybindings, { paddingX: 2 });
+        },
       );
     } else {
       ctx.ui.setEditorComponent(undefined);
@@ -304,6 +322,7 @@ export default function (pi: ExtensionAPI) {
 
   pi.on("session_start", (_event, ctx) => {
     ctxRef = ctx;
+    loopCounts.clear();
     if (ctx.mode !== "tui") return;
     installBars(ctx);
   });

package/extensions/render.ts

@@ -107,6 +107,40 @@ export function composeLine(
   return paint(l, c) + c("dim", "─".repeat(gap)) + paint(r, c);
 }
 
+/** Short labels for loop-police detector names, grouped by family. */
+export const LOOP_KIND_LABELS: Record<string, string> = {
+  thinking_loop: "think",
+  semantic_loop: "think",
+  output_loop: "out",
+  output_semantic_loop: "out",
+  stagnation: "stag",
+  file_read_loop: "file",
+  file_scan_loop: "file",
+  search_spiral: "search",
+  tool_loop: "tool",
+};
+
+/**
+ * Compact per-session summary of loop-police detections:
+ * "10 loops (tool 7, think 3)" — undefined when there are none, so the bar
+ * renders exactly as without loop-police installed.
+ */
+export function loopSummary(counts: ReadonlyMap<string, number>): string | undefined {
+  let total = 0;
+  const byLabel = new Map<string, number>();
+  for (const [kind, n] of counts) {
+    if (n <= 0) continue;
+    total += n;
+    const label = LOOP_KIND_LABELS[kind] ?? kind;
+    byLabel.set(label, (byLabel.get(label) ?? 0) + n);
+  }
+  if (total === 0) return undefined;
+  const parts = [...byLabel.entries()]
+    .sort((a, b) => b[1] - a[1])
+    .map(([label, n]) => `${label} ${n}`);
+  return `${total} loop${total === 1 ? "" : "s"} (${parts.join(", ")})`;
+}
+
 export interface TopBarData {
   folder: string;
   branch: string | null;
@@ -115,6 +149,8 @@ export interface TopBarData {
   modelId: string | undefined;
   /** Thinking level, or undefined when the model has no reasoning. */
   effort: string | undefined;
+  /** loopSummary() text, or undefined when loop-police reported nothing. */
+  loops: string | undefined;
   icons: boolean;
 }
 
@@ -129,6 +165,9 @@ export function topBarSegments(d: TopBarData): { left: Segment[]; right: Segment
     left.push(sep, { text: d.icons ? "⎇ " : "", color: "success" }, { text: d.branch, color: "success" });
   }
   left.push({ text: d.streaming ? " ● " : " ○ ", color: d.streaming ? "warning" : "dim" });
+  if (d.loops) {
+    left.push({ text: (d.icons ? "⚠ " : "") + d.loops + " ", color: "warning" });
+  }
 
   const right: Segment[] = [];
   if (d.modelId) {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "pi-input-bar",
-  "version": "1.0.0",
+  "version": "1.1.0",
   "description": "Pi extension: decorated input bar — folder/branch/model/effort above the editor, context/token usage and tokens-per-second below, plus random working words while the agent streams.",
   "keywords": ["pi-package", "pi", "pi-coding-agent", "extension", "tui", "statusbar", "footer", "input-bar"],
   "license": "MIT",