@esso0428/pi-subagents 0.15.0 → 0.15.1

package/src/ui/markdown-result.ts

@@ -0,0 +1,56 @@
+import { getMarkdownTheme } from "@earendil-works/pi-coding-agent";
+import { Markdown, Text, type Component } from "@earendil-works/pi-tui";
+import type { Theme } from "./agent-widget.js";
+
+const TRUNCATION_MARKER = "… (more output available)";
+
+/**
+ * Render a child result as Markdown while keeping notification output bounded.
+ *
+ * Markdown is rendered at paint time so the TUI's current width controls wrapping.
+ * If the host's Markdown implementation is unavailable or throws, the original
+ * text is rendered through Text instead of making the result disappear.
+ */
+class CappedMarkdownResult implements Component {
+  private readonly text: string;
+  private readonly theme: Theme;
+  private readonly maxLines: number;
+  private markdown: Component | undefined;
+
+  constructor(text: string, theme: Theme, maxLines: number) {
+    this.text = text;
+    this.theme = theme;
+    this.maxLines = Math.max(0, Math.floor(maxLines));
+
+    try {
+      this.markdown = new Markdown(text, 0, 0, getMarkdownTheme());
+    } catch {
+      // A mismatched host Pi version should still leave the child output visible.
+    }
+  }
+
+  invalidate(): void {
+    this.markdown?.invalidate();
+  }
+
+  render(width: number): string[] {
+    let lines: string[];
+    try {
+      lines = this.markdown?.render(width) ?? new Text(this.text, 0, 0).render(width);
+    } catch {
+      lines = new Text(this.text, 0, 0).render(width);
+    }
+
+    if (lines.length <= this.maxLines) return lines;
+    if (this.maxLines === 0) return [];
+
+    return [
+      ...lines.slice(0, this.maxLines - 1),
+      this.theme.fg("muted", TRUNCATION_MARKER),
+    ];
+  }
+}
+
+export function createMarkdownResult(text: string, theme: Theme, maxLines: number): Component {
+  return new CappedMarkdownResult(text, theme, maxLines);
+}

package/test/ui/markdown-result.test.ts

@@ -0,0 +1,45 @@
+import { describe, expect, it } from "vitest";
+import { createMarkdownResult } from "../../src/ui/markdown-result.js";
+
+const theme = {
+  fg: (_name: string, text: string) => text,
+  bold: (text: string) => text,
+};
+
+describe("createMarkdownResult", () => {
+  it("renders headings, lists, inline code, and fenced code", () => {
+    const component = createMarkdownResult(
+      [
+        "# Result",
+        "",
+        "- first",
+        "- `inline()`",
+        "",
+        "```ts",
+        "const answer: number = 42;",
+        "```",
+      ].join("\n"),
+      theme,
+      50,
+    );
+
+    const rendered = component.render(100).join("\n");
+    expect(rendered).toContain("Result");
+    expect(rendered).toContain("first");
+    expect(rendered).toContain("inline()");
+    expect(rendered).toContain("answer");
+  });
+
+  it("caps rendered lines and adds a truncation marker", () => {
+    const component = createMarkdownResult("line\n".repeat(20), theme, 5);
+    const rendered = component.render(80);
+
+    expect(rendered).toHaveLength(5);
+    expect(rendered.at(-1)).toContain("more");
+  });
+
+  it("keeps unlabelled code blocks safe without language guessing", () => {
+    const component = createMarkdownResult("```\nplain text\n```", theme, 20);
+    expect(component.render(80).join("\n")).toContain("plain text");
+  });
+});