David Marshall
c8d4f22b59
The HTML export feed, sidebar tree, and TUI session tree did not handle the developer role, so plan content injected after /plan approval (and any other developer messages) was hidden from /export and shown as a bare [developer] label in /tree. Renders developer messages in the main feed with a dimmed .developer-message style, labels them in the sidebar tree, counts them in header stats, and shows their content in the TUI tree selector so search matches the body. Closes #753. Co-Authored-By: omp <noreply@oh-my-pi.dev>
66 lines
2.3 KiB
TypeScript
66 lines
2.3 KiB
TypeScript
import { beforeAll, describe, expect, it } from "bun:test";
|
|
import type { AgentMessage } from "@oh-my-pi/pi-agent-core";
|
|
import { TreeSelectorComponent } from "../../../src/modes/components/tree-selector";
|
|
import * as themeModule from "../../../src/modes/theme/theme";
|
|
import type { SessionEntry, SessionTreeNode } from "../../../src/session/session-manager";
|
|
|
|
let counter = 0;
|
|
function makeMessageNode(message: AgentMessage, parentId: string | null = null, label?: string): SessionTreeNode {
|
|
const id = `entry-${counter++}`;
|
|
const entry: SessionEntry = {
|
|
type: "message",
|
|
id,
|
|
parentId,
|
|
timestamp: new Date().toISOString(),
|
|
message,
|
|
};
|
|
return { entry, children: [], label };
|
|
}
|
|
|
|
function render(tree: SessionTreeNode[], width = 120): string {
|
|
const selector = new TreeSelectorComponent(
|
|
tree,
|
|
tree[tree.length - 1]?.entry.id ?? null,
|
|
60,
|
|
() => {},
|
|
() => {},
|
|
);
|
|
return Bun.stripANSI(selector.render(width).join("\n"));
|
|
}
|
|
|
|
describe("TreeSelectorComponent developer message rendering", () => {
|
|
beforeAll(async () => {
|
|
await themeModule.initTheme(false, undefined, undefined, "dark", "light");
|
|
});
|
|
|
|
it("renders developer messages with their content, not just [developer]", () => {
|
|
const planContent = "## Plan\n\n1. Fix the tree selector\n2. Update the HTML export";
|
|
const root = makeMessageNode({ role: "user", content: "/plan", timestamp: 1 });
|
|
const developer = makeMessageNode(
|
|
{ role: "developer", content: [{ type: "text", text: planContent }], timestamp: 2 },
|
|
root.entry.id,
|
|
);
|
|
root.children.push(developer);
|
|
|
|
const rendered = render([root]);
|
|
|
|
expect(rendered).toContain("developer:");
|
|
expect(rendered).toContain("Fix the tree selector");
|
|
expect(rendered).toContain("Update the HTML export");
|
|
expect(rendered).not.toMatch(/^\s*\[developer\]\s*$/m);
|
|
});
|
|
|
|
it("matches developer messages in search (content is searchable)", () => {
|
|
const planContent = "ZZZ_UNIQUE_PLAN_TOKEN approved plan body";
|
|
const root = makeMessageNode({ role: "user", content: "/plan", timestamp: 1 });
|
|
const developer = makeMessageNode(
|
|
{ role: "developer", content: [{ type: "text", text: planContent }], timestamp: 2 },
|
|
root.entry.id,
|
|
);
|
|
root.children.push(developer);
|
|
|
|
const rendered = render([root]);
|
|
expect(rendered).toContain("ZZZ_UNIQUE_PLAN_TOKEN");
|
|
});
|
|
});
|