pi-minimal 0.2.2 → 0.3.0

package/test/tools.test.ts

@@ -1,7 +1,7 @@
 import assert from "node:assert/strict"
 import test from "node:test"
 import type { ExtensionAPI } from "@earendil-works/pi-coding-agent"
-import { compactRenderers, installToolRenderers, isToolKind } from "../src/tools.js"
+import { codemodeRenderers, compactRenderers, installToolRenderers, isToolKind, stopLiveWatch } from "../src/tools.js"
 
 test("isToolKind recognises the built-in tools only", () => {
 	for (const name of ["bash", "read", "edit", "write", "grep", "find", "ls"]) assert.equal(isToolKind(name), true)
@@ -19,7 +19,9 @@ test("installToolRenderers registers a resolver at load time that only claims bu
 	installToolRenderers(pi)
 	assert.ok(resolver, "resolver registered synchronously, before session_start")
 	const fallback = { renderShell: "default" }
-	assert.equal(resolver!("codemode", () => fallback), fallback)
+	assert.equal(resolver!("mcp__foo__bar", () => fallback), fallback)
+	const codemode = resolver!("codemode", () => fallback) as { renderShell?: string }
+	assert.equal(codemode.renderShell, "self")
 	const mine = resolver!("bash", () => fallback) as { renderShell?: string }
 	assert.equal(mine.renderShell, "self")
 	assert.equal(typeof (mine as { renderCall?: unknown }).renderCall, "function")
@@ -53,3 +55,71 @@ test("compactRenderers render a one-line row with the outcome once the result la
 	const [line] = row.render(60)
 	assert.match(line, /✓ bash\s+ls\s+2 lines · 1.2s$/)
 })
+
+function codemodeContext(extra: Record<string, unknown> = {}) {
+	return {
+		args: { code: "await tools.bash({command:'ls'})\ntext(1)" },
+		toolCallId: "c1",
+		invalidate() {},
+		state: {} as Record<string, unknown>,
+		isPartial: false,
+		isError: false,
+		durationMs: 1200,
+		...extra,
+	}
+}
+
+const plainTheme = { fg: (_c: string, s: string) => s } as never
+const codemodeResult = (calls: unknown[], output = "a\nb") => ({
+	content: [
+		{ type: "text", text: "Script completed\nWall time 1.2 seconds\nOutput:\n" },
+		{ type: "text", text: output },
+	],
+	details: { calls },
+})
+
+test("codemode renders one parent row plus one child row per nested call, no script", () => {
+	const r = codemodeRenderers()
+	const context = codemodeContext()
+	const calls = [
+		{ name: "plan_tracker", args: '{"action":"update","index":6,"status":"complete"}', status: "ok", durationMs: 309 },
+		{ name: "bash", args: '{"command":"cd ~/r && sed -n 1,2p a.md"}', status: "ok", durationMs: 636 },
+	]
+	const body = r.renderResult!(codemodeResult(calls) as never, { expanded: false, isPartial: false } as never, plainTheme, context as never)
+	const lines = r.renderCall!(context.args as never, plainTheme, context as never).render(100)
+	assert.equal(lines.length, 3)
+	assert.match(lines[0], /✓ code\s+2 calls · plan_tracker, bash\s+2 lines · 1.2s$/)
+	assert.match(lines[1], /✓ plan_tracker action=update index=6 status=complete\s+0.3s$/)
+	assert.match(lines[2], /✓ bash\s+cd ~\/r && sed -n 1,2p a.md\s+0.6s$/)
+	assert.ok(lines.every((line) => !line.includes("await")), "script hidden when collapsed")
+	assert.deepEqual(body.render(100), [])
+	for (const line of lines) assert.ok(line.length <= 100)
+})
+
+test("codemode shows the first failure under the row, and script + output when expanded", () => {
+	const r = codemodeRenderers()
+	const context = codemodeContext({ durationMs: 274_000 })
+	const calls = [{ name: "bash", args: '{"command":"npm test"}', status: "error", durationMs: 274_000, error: "fail 2" }]
+	const collapsed = r.renderResult!(codemodeResult(calls, "TEST_EXIT=1") as never, { expanded: false, isPartial: false } as never, plainTheme, context as never)
+	const [row, child] = r.renderCall!(context.args as never, plainTheme, context as never).render(100)
+	assert.match(row, /✗ code\s+1 call · bash\s+1 failed · 4m 34s$/)
+	assert.match(child, /✗ bash npm test\s+failed · 4m 34s$/)
+	assert.match(collapsed.render(100).join("\n"), /│ fail 2/)
+
+	const open = r.renderResult!(codemodeResult(calls, "TEST_EXIT=1") as never, { expanded: true, isPartial: false } as never, plainTheme, { ...context, expanded: true } as never)
+	const text = open.render(100).join("\n")
+	assert.match(text, /\/\/ script · 2 lines/)
+	assert.match(text, /await tools.bash/)
+	assert.match(text, /\/\/ output\n.*TEST_EXIT=1/)
+})
+
+test("codemode shows nested calls live while the script is still running", () => {
+	const r = codemodeRenderers()
+	const context = codemodeContext({ isPartial: true, durationMs: undefined })
+	r.renderResult!(codemodeResult([{ name: "bash", args: '{"command":"sleep 5"}', status: "running" }], "") as never, { expanded: false, isPartial: true } as never, plainTheme, context as never)
+	const lines = r.renderCall!(context.args as never, plainTheme, context as never).render(80)
+	assert.match(lines[0], /^\s+[◐◓] code\s+1 call · bash/)
+	assert.match(lines[1], /[◐◓] bash sleep 5/)
+	assert.equal((context.state as { done?: boolean }).done, undefined)
+	stopLiveWatch()
+})

package/themes/minimal-dark.json

@@ -7,8 +7,8 @@
 		"panel": "okhsl(230 5% 17%)",
 		"text": "okhsl(234 3% 90%)",
 		"muted": "okhsl(229 6% 64%)",
-		"dim": "okhsl(229 7% 48%)",
-		"faint": "okhsl(230 6% 34%)",
+		"dim": "okhsl(229 7% 56%)",
+		"faint": "okhsl(230 6% 36%)",
 		"violet": "okhsl(295 50% 67%)",
 		"violetSoft": "okhsl(295 45% 74%)",
 		"blue": "okhsl(232 54% 67%)",
@@ -29,7 +29,7 @@
 		"muted": "muted",
 		"dim": "dim",
 		"text": "text",
-		"thinkingText": "okhsl(229 6% 52%)",
+		"thinkingText": "okhsl(229 6% 58%)",
 		"selectedBg": "userBg",
 		"scrollbarTrack": "faint",
 		"scrollbarThumb": "dim",
@@ -40,8 +40,8 @@
 		"customMessageBg": "okhsl(295 30% 20%)",
 		"customMessageText": "muted",
 		"customMessageLabel": "violet",
-		"toolPendingBg": "panel",
-		"toolSuccessBg": "panel",
+		"toolPendingBg": "bg",
+		"toolSuccessBg": "bg",
 		"toolErrorBg": "errorBg",
 		"toolTitle": "text",
 		"toolOutput": "dim",
@@ -60,10 +60,10 @@
 		"toolDiffContext": "dim",
 		"syntaxComment": "dim",
 		"syntaxKeyword": "blue",
-		"syntaxFunction": "yellow",
+		"syntaxFunction": "okhsl(265 45% 76%)",
 		"syntaxVariable": "okhsl(202 58% 70%)",
 		"syntaxString": "okhsl(52 67% 70%)",
-		"syntaxNumber": "green",
+		"syntaxNumber": "okhsl(180 40% 68%)",
 		"syntaxType": "violet",
 		"syntaxOperator": "muted",
 		"syntaxPunctuation": "muted",