pi-firecode 0.8.1 → 1.0.1

package/statusbar/render.ts

@@ -1,181 +0,0 @@
-/** 状态栏的纯渲染与布局：给定数据和宽度产出字符串，不触碰会话状态。 */
-import type { ThemeColor } from "@earendil-works/pi-coding-agent";
-import { visibleWidth } from "@earendil-works/pi-tui";
-import { clip, formatDuration, formatTokens, oneLine } from "../format.js";
-import { cacheColor, contextColor, quotaColor } from "../theme.js";
-import type { QuotaStatus } from "./quota-parse.js";
-import type { TpsStatus } from "./tps.js";
-
-export type StatusLineParts = {
-	model: string;
-	modelCompact: string;
-	quota: string;
-	quotaCompact: string;
-	context: string;
-	contextCompact: string;
-	cache: string;
-	tps: string;
-};
-
-type ForegroundTheme = {
-	fg(color: ThemeColor, text: string): string;
-};
-
-/** 最近一条助手消息的 cacheRead / (input + cacheRead + cacheWrite)。 */
-export function latestCacheHitPercent(
-	entries: ReadonlyArray<{
-		type?: string;
-		message?: {
-			role?: string;
-			usage?: { input?: number; cacheRead?: number; cacheWrite?: number };
-		};
-	}>,
-): number | undefined {
-	for (let index = entries.length - 1; index >= 0; index--) {
-		const entry = entries[index];
-		if (entry?.type !== "message" || entry.message?.role !== "assistant") continue;
-		const usage = entry.message.usage;
-		if (!usage) continue;
-		const input = usage.input ?? 0;
-		const cacheRead = usage.cacheRead ?? 0;
-		const cacheWrite = usage.cacheWrite ?? 0;
-		const promptTokens = input + cacheRead + cacheWrite;
-		if (promptTokens <= 0 || (cacheRead === 0 && cacheWrite === 0)) return undefined;
-		return (cacheRead / promptTokens) * 100;
-	}
-	return undefined;
-}
-
-export function renderLocation(
-	theme: ForegroundTheme,
-	directory: string,
-	branch: string | null | undefined,
-): string {
-	return [theme.fg("dim", `📍${directory}`), branch ? theme.fg("dim", `🌿 ${branch}`) : ""]
-		.filter(Boolean)
-		.join(" · ");
-}
-
-export function renderQuota(
-	theme: ForegroundTheme,
-	status: QuotaStatus,
-	compact = false,
-): string {
-	if (status.state === "loading") return theme.fg("dim", "🔋 …");
-	if (status.state === "unavailable") return theme.fg("dim", "🔋 —");
-	const windows = compact
-		? [
-				status.windows.reduce((tightest, window) =>
-					window.remaining < tightest.remaining ? window : tightest,
-				),
-			]
-		: status.windows;
-	return `${theme.fg("dim", "🔋 ")}${windows
-		.map((window) =>
-			theme.fg(quotaColor(window.remaining), `${window.label} ${window.remaining}%`),
-		)
-		.join(theme.fg("dim", "/"))}`;
-}
-
-export function renderContext(
-	theme: ForegroundTheme,
-	percent: number | null | undefined,
-	contextWindow: number,
-	compact = false,
-): string {
-	const percentText = percent == null ? "?" : `${percent.toFixed(1)}%`;
-	return `${theme.fg("dim", "📦 ")}${theme.fg(contextColor(percent), percentText)}${
-		compact ? "" : theme.fg("dim", `/${formatTokens(contextWindow)}`)
-	}`;
-}
-
-export function renderCache(theme: ForegroundTheme, percent: number | undefined): string {
-	if (percent === undefined) return "";
-	return `${theme.fg("dim", "♻️ ")}${theme.fg(cacheColor(percent), `${Math.round(percent)}%`)}`;
-}
-
-export function renderTps(theme: ForegroundTheme, status: TpsStatus | undefined): string {
-	if (!status) return "";
-	if (status.phase === "live") {
-		return theme.fg(
-			"success",
-			`↗ ${status.tokensPerSecond ? `${status.tokensPerSecond}t/s` : "…"}`,
-		);
-	}
-	const elapsed = formatDuration(status.elapsedSeconds * 1_000);
-	const duration = theme.fg("dim", `⏱ ${elapsed}${status.tokensPerSecond ? " · " : ""}`);
-	return status.tokensPerSecond
-		? `${duration}${theme.fg("success", `↗ ${status.tokensPerSecond}t/s`)}`
-		: duration;
-}
-
-/** /fire-review 广播的审查进度，右对齐挂在首行末尾。 */
-export function reviewStatus(statuses: ReadonlyMap<string, string>): string {
-	return oneLine(statuses.get("fire-review") ?? "");
-}
-
-/** 首行模块状态由各模块的 setStatus/onChange 驱动，状态栏只负责组合。 */
-export function statusBadges(statuses: ReadonlyMap<string, string>, separator: string): string {
-	return [statuses.get("master"), statuses.get("watcher")].filter(Boolean).join(separator);
-}
-
-export function alignRight(left: string, right: string, width: number): string {
-	if (!right) return left;
-	const padding = width - visibleWidth(left) - visibleWidth(right);
-	return padding >= 2 ? `${left}${" ".repeat(padding)}${right}` : left;
-}
-
-export function fitMetadataLine(
-	location: string,
-	title: string,
-	width: number,
-	separator: string,
-	badge = "",
-): string {
-	if (width <= 0) return "";
-	// badge（如指挥官态）整段取舍：放不下就丢，不截半个，再走会话名截断阶梯。
-	if (badge) {
-		const full = [location, title, badge].filter(Boolean).join(separator);
-		if (visibleWidth(full) <= width) return full;
-	}
-	if (!title) return clip(location, width, "end", visibleWidth(location) > width ? "…" : "");
-	const full = `${location}${separator}${title}`;
-	if (visibleWidth(full) <= width) return full;
-	const titleWidth = width - visibleWidth(location) - visibleWidth(separator);
-	if (titleWidth > 0) {
-		const fittedTitle = clip(title, titleWidth, "end");
-		if (fittedTitle) return `${location}${separator}${fittedTitle}`;
-	}
-	return clip(location, width, "end");
-}
-
-const joinParts = (parts: string[], separator: string) => parts.filter(Boolean).join(separator);
-
-export function fitStatusLine(
-	parts: StatusLineParts,
-	width: number,
-	separator: string,
-): string {
-	if (width <= 0) return "";
-	// joinParts 会丢掉空的 quota/cache/tps。顺序：model → quota? → context → cache → tps。
-	const candidates = [
-		[parts.model, parts.quota, parts.context, parts.cache, parts.tps],
-		[parts.model, parts.quota, parts.context, parts.cache],
-		[parts.modelCompact, parts.quota, parts.context, parts.cache],
-		[parts.modelCompact, parts.quotaCompact, parts.context, parts.cache],
-		[parts.modelCompact, parts.context, parts.cache],
-		[parts.modelCompact, parts.context],
-		[parts.modelCompact, parts.contextCompact],
-	].map((candidate) => joinParts(candidate, separator));
-	for (const candidate of candidates) {
-		if (visibleWidth(candidate) <= width) return candidate;
-	}
-
-	const context = parts.contextCompact || parts.context;
-	const contextWidth = visibleWidth(context);
-	const modelBudget = width - contextWidth - visibleWidth(separator);
-	if (modelBudget > 0 && contextWidth <= width) {
-		return `${clip(parts.modelCompact, modelBudget, "end")}${separator}${context}`;
-	}
-	return clip(context, width, "end", "");
-}

package/statusbar/tps.ts

@@ -1,87 +0,0 @@
-/** Tracks one provider response from request start through final output usage. */
-import { performance } from "node:perf_hooks";
-import type { ExtensionAPI } from "@earendil-works/pi-coding-agent";
-
-const TOKEN_CHARS = 4;
-const MIN_SAMPLE_MS = 1_000;
-const UPDATE_INTERVAL_MS = 250;
-
-export type TpsStatus =
-	| { phase: "live"; tokensPerSecond?: number }
-	| { phase: "complete"; elapsedSeconds: number; tokensPerSecond?: number };
-
-function deltaText(event: { type: string; delta?: unknown }): string | undefined {
-	if (event.type !== "text_delta" && event.type !== "thinking_delta" && event.type !== "toolcall_delta") {
-		return undefined;
-	}
-	return typeof event.delta === "string" ? event.delta : undefined;
-}
-
-export function registerTps(
-	pi: ExtensionAPI,
-	update: (status?: TpsStatus) => void,
-	now = (): number => performance.now(),
-): void {
-	let requestStartedAt: number | undefined;
-	let firstDeltaAt: number | undefined;
-	let lastDeltaAt: number | undefined;
-	let lastUpdateAt: number | undefined;
-	let estimatedTokens = 0;
-
-	function resetMeasurement(): void {
-		requestStartedAt = undefined;
-		firstDeltaAt = undefined;
-		lastDeltaAt = undefined;
-		lastUpdateAt = undefined;
-		estimatedTokens = 0;
-	}
-
-	function clear(): void {
-		resetMeasurement();
-		update();
-	}
-
-	pi.on("session_start", clear);
-	pi.on("model_select", clear);
-	pi.on("before_provider_request", () => {
-		resetMeasurement();
-		requestStartedAt = now();
-	});
-	pi.on("message_update", (event) => {
-		if (event.message.role !== "assistant") return;
-		const delta = deltaText(event.assistantMessageEvent);
-		if (delta === undefined) return;
-
-		const current = now();
-		lastDeltaAt = current;
-		estimatedTokens += Math.max(0, delta.length / TOKEN_CHARS);
-		if (firstDeltaAt === undefined) {
-			firstDeltaAt = current;
-			lastUpdateAt = current;
-			update({ phase: "live" });
-			return;
-		}
-
-		const elapsed = current - firstDeltaAt;
-		if (elapsed < MIN_SAMPLE_MS || current - (lastUpdateAt ?? 0) < UPDATE_INTERVAL_MS) return;
-		lastUpdateAt = current;
-		const officialTokens = event.message.usage.output;
-		const tokens = officialTokens > 0 ? officialTokens : Math.round(estimatedTokens);
-		update({ phase: "live", tokensPerSecond: Math.round((tokens * 1_000) / elapsed) });
-	});
-	pi.on("message_end", (event) => {
-		if (event.message.role !== "assistant" || firstDeltaAt === undefined || lastDeltaAt === undefined) return;
-		const current = now();
-		const generationMs = lastDeltaAt - firstDeltaAt;
-		const outputTokens = event.message.usage.output;
-		update({
-			phase: "complete",
-			elapsedSeconds: Math.max(0, (current - (requestStartedAt ?? firstDeltaAt)) / 1_000),
-			...(generationMs > 0 && outputTokens > 0
-				? { tokensPerSecond: Math.round((outputTokens * 1_000) / generationMs) }
-				: {}),
-		});
-		resetMeasurement();
-	});
-	pi.on("session_shutdown", clear);
-}

package/theme.ts

@@ -1,84 +0,0 @@
-/** FireCode 品牌色与数值分级配色：所有阈值到颜色的映射集中在此。 */
-import type { ExtensionAPI, ThemeColor } from "@earendil-works/pi-coding-agent";
-
-export const ANSI = {
-	reset: "\x1b[0m",
-	bold: "\x1b[1m",
-} as const;
-
-const rgb = (red: number, green: number, blue: number) =>
-	`\x1b[38;2;${red};${green};${blue}m`;
-
-/** 火焰渐变：横幅按行取色，从底部暗红烧到芯部白金。 */
-export const FLAME = {
-	red: rgb(255, 47, 32),
-	orange: rgb(255, 119, 28),
-	gold: rgb(255, 195, 61),
-	white: rgb(255, 239, 184),
-	textTop: rgb(255, 112, 38),
-	textMid: rgb(255, 184, 62),
-	textCore: rgb(255, 236, 170),
-} as const;
-
-/** 阈值表：首个满足 `value >= at` 的档位胜出，表按 at 降序书写。 */
-type Threshold = { at: number; color: ThemeColor };
-
-function pick(value: number, thresholds: readonly Threshold[]): ThemeColor {
-	for (const threshold of thresholds) {
-		if (value >= threshold.at) return threshold.color;
-	}
-	return thresholds[thresholds.length - 1].color;
-}
-
-const CONTEXT_FILL = [
-	{ at: 75, color: "error" },
-	{ at: 50, color: "warning" },
-	{ at: 0, color: "success" },
-] as const satisfies readonly Threshold[];
-
-/** 缓存命中越高越好，与上下文填充方向相反。 */
-const CACHE_HIT = [
-	{ at: 90, color: "success" },
-	{ at: 50, color: "muted" },
-	{ at: 20, color: "warning" },
-	{ at: 0, color: "error" },
-] as const satisfies readonly Threshold[];
-
-const QUOTA_REMAINING = [
-	{ at: 51, color: "success" },
-	{ at: 26, color: "warning" },
-	{ at: 0, color: "error" },
-] as const satisfies readonly Threshold[];
-
-const RESULT_SIZE = [
-	{ at: 50_000, color: "error" },
-	{ at: 10_000, color: "warning" },
-	{ at: 0, color: "dim" },
-] as const satisfies readonly Threshold[];
-
-export const contextColor = (percent: number | null | undefined): ThemeColor =>
-	percent == null ? "muted" : pick(percent, CONTEXT_FILL);
-
-export const cacheColor = (percent: number): ThemeColor =>
-	pick(percent, CACHE_HIT);
-
-export const quotaColor = (remaining: number): ThemeColor =>
-	pick(remaining, QUOTA_REMAINING);
-
-export const sizeColor = (chars: number): ThemeColor =>
-	pick(chars, RESULT_SIZE);
-
-type ThinkingLevel = ReturnType<ExtensionAPI["getThinkingLevel"]>;
-
-const THINKING: Record<ThinkingLevel, ThemeColor> = {
-	off: "thinkingOff",
-	minimal: "thinkingMinimal",
-	low: "thinkingLow",
-	medium: "thinkingMedium",
-	high: "thinkingHigh",
-	xhigh: "thinkingXhigh",
-	max: "thinkingMax",
-};
-
-export const thinkingColor = (level: ThinkingLevel): ThemeColor =>
-	THINKING[level];

package/tools/grouping.ts

@@ -1,233 +0,0 @@
-/**
- * 连续工具行的视觉合并与第三方工具通用单行兜底：
- * 1. 原生内置或声明了 self 的工具：保留自渲染，去除相邻空行形成轨道；
- * 2. 未知第三方或默认宽框工具：自动收口为通用 ToolLine 单行紧凑轨道与全宽背景。
- */
-import type { ExtensionUIContext, Theme } from "@earendil-works/pi-coding-agent";
-import { ToolLine } from "./line.js";
-import { genericArgsParts } from "./parts.js";
-
-const GROUP_PATCH = Symbol.for("pi.firecode.group-patch");
-const REQUEST_RENDER_PATCH = Symbol.for("pi.firecode.request-render-patch");
-const GLOBAL_STATE = Symbol.for("pi.firecode.tools-state");
-const THEME_KEY = Symbol.for("@earendil-works/pi-coding-agent:theme");
-
-function getTheme(): Theme {
-	const globalTheme = (globalThis as any)[THEME_KEY];
-	if (globalTheme && typeof globalTheme.fg === "function") return globalTheme;
-	return {
-		fg: (_color: string, text: string) => text,
-		bg: (_color: string, text: string) => text,
-		bold: (text: string) => text,
-	} as unknown as Theme;
-}
-
-type RenderFunction = (width: number) => string[];
-type RequestRenderFunction = (force?: boolean) => unknown;
-type ToolRow = { toolName: unknown; setExpanded: unknown; render: RenderFunction };
-type ToolRowPatch = { original: RenderFunction; patched: RenderFunction };
-type ToolRowPrototype = ToolRow & { [GROUP_PATCH]?: ToolRowPatch };
-type RequestRenderPatch = { original: RequestRenderFunction; patched: RequestRenderFunction };
-type ContainerLike = { children: unknown[] };
-type TuiLike = {
-	requestRender: RequestRenderFunction;
-	children?: unknown[];
-	[REQUEST_RENDER_PATCH]?: RequestRenderPatch;
-};
-type GlobalState = {
-	tui?: TuiLike;
-	chat?: ContainerLike;
-	indexedChildren?: unknown[];
-	indexedLength?: number;
-	joinedRows?: WeakSet<object>;
-	patchedPrototype?: ToolRowPrototype;
-	patchErrorReported?: boolean;
-	decorated: Set<string>;
-};
-
-// 扩展热重载会重新执行模块，patch 状态必须挂在全局才能正确卸载。
-const runtimeGlobal = globalThis as typeof globalThis & { [GLOBAL_STATE]?: GlobalState };
-const globalState = (runtimeGlobal[GLOBAL_STATE] ??= { decorated: new Set<string>() });
-
-function isToolRow(value: unknown): value is ToolRow {
-	if (!value || typeof value !== "object") return false;
-	const row = value as ToolRow;
-	return typeof row.render === "function" && typeof row.setExpanded === "function" && "toolName" in row;
-}
-
-function isDecoratedToolRow(value: unknown): value is ToolRow {
-	return isToolRow(value);
-}
-
-export function renderFallbackToolRow(row: any, width: number, customTheme?: Theme): string[] {
-	const isError = row.result?.isError ?? false;
-	const isPartial = row.isPartial ?? false;
-	const toolName = typeof row.toolName === "string" ? row.toolName : "tool";
-	const activeTheme = customTheme ?? getTheme();
-	const toolLine = new ToolLine({
-		label: toolName,
-		value: genericArgsParts(row.args),
-		clip: "end",
-		theme: activeTheme,
-		ctx: {
-			state: row.rendererState ?? {},
-			cwd: row.cwd ?? process.cwd(),
-			toolCallId: row.toolCallId ?? "",
-			isPartial,
-			isError,
-			expanded: row.expanded ?? false,
-		},
-	});
-	const lines = ["", ...toolLine.render(width)];
-	if (row.expanded && row.result) {
-		const textContent = Array.isArray(row.result.content)
-			? row.result.content
-					.map((c: any) => (c?.type === "text" && typeof c.text === "string" ? c.text : ""))
-					.filter(Boolean)
-					.join("\n")
-			: "";
-		if (textContent) {
-			const outputLines = textContent.split("\n").map((l: string) => activeTheme.fg("toolOutput", l));
-			lines.push("", ...outputLines);
-		}
-	}
-	return lines;
-}
-
-function findChatContainer(value: unknown, seen = new Set<unknown>()): ContainerLike | undefined {
-	if (!value || typeof value !== "object" || seen.has(value)) return undefined;
-	seen.add(value);
-	const children = (value as ContainerLike).children;
-	if (!Array.isArray(children)) return undefined;
-	if (children.some(isToolRow)) return value as ContainerLike;
-	for (const child of children) {
-		const found = findChatContainer(child, seen);
-		if (found) return found;
-	}
-	return undefined;
-}
-
-/** 增量标记"紧跟另一条装饰行"的行，聊天记录只增不改，可以从上次位置续算。 */
-function indexJoinedRows(children: unknown[]): void {
-	const canAppend =
-		globalState.indexedChildren === children &&
-		globalState.indexedLength !== undefined &&
-		globalState.indexedLength <= children.length &&
-		globalState.joinedRows !== undefined;
-	const joinedRows =
-		canAppend && globalState.joinedRows ? globalState.joinedRows : new WeakSet<object>();
-	const start = canAppend ? Math.max(1, globalState.indexedLength ?? 1) : 1;
-	for (let index = start; index < children.length; index++) {
-		const current = children[index];
-		if (isDecoratedToolRow(current) && isDecoratedToolRow(children[index - 1]))
-			joinedRows.add(current);
-	}
-	globalState.indexedChildren = children;
-	globalState.indexedLength = children.length;
-	globalState.joinedRows = joinedRows;
-}
-
-function followsToolRow(row: ToolRow): boolean {
-	if (!globalState.chat) globalState.chat = findChatContainer(globalState.tui);
-	const children = globalState.chat?.children;
-	if (!children) return false;
-	if (globalState.indexedChildren !== children || globalState.indexedLength !== children.length)
-		indexJoinedRows(children);
-	return globalState.joinedRows?.has(row) ?? false;
-}
-
-/** 走 pi 原生渲染（会消费工具自带 renderCall）的工具；其余走通用兜底行。
- * subagents 自带中文 ToolLine 渲染（master/index.ts），不列入会被兜底行遮掉。 */
-export const FIRECODE_TOOLS = new Set(["read", "bash", "edit", "write", "find", "grep", "ls", "subagents"]);
-
-function patchToolRows(): void {
-	if (globalState.patchedPrototype) return;
-	const chat = globalState.chat ?? findChatContainer(globalState.tui);
-	if (!chat) return;
-	globalState.chat = chat;
-	const row = chat.children.find(isToolRow);
-	if (!row) return;
-	const prototype = Object.getPrototypeOf(row) as ToolRowPrototype;
-	if (prototype[GROUP_PATCH]) {
-		globalState.patchedPrototype = prototype;
-		return;
-	}
-	const original = prototype.render;
-	const patched: RenderFunction = function (this: ToolRow, width: number): string[] {
-		const toolName = typeof this.toolName === "string" ? this.toolName : "";
-		const isFirecodeTool = FIRECODE_TOOLS.has(toolName);
-		const isExpanded = Boolean((this as any).expanded);
-		const lines = isFirecodeTool || isExpanded
-			? original.call(this, width)
-			: renderFallbackToolRow(this, width);
-		return followsToolRow(this) && lines[0] === "" ? lines.slice(1) : lines;
-	};
-	Object.defineProperty(prototype, GROUP_PATCH, {
-		configurable: true,
-		value: { original, patched } satisfies ToolRowPatch,
-	});
-	prototype.render = patched;
-	globalState.patchedPrototype = prototype;
-}
-
-export function installGroupPatch(ui: ExtensionUIContext, decorated: Set<string>): void {
-	globalState.decorated = decorated;
-	globalState.patchErrorReported = false;
-	// 借一次 widget 生命周期拿到 tui 实例，拿到即注销。
-	ui.setWidget("firecode-tui-capture", (tui) => {
-		const patchableTui = tui as TuiLike;
-		globalState.tui = patchableTui;
-		globalState.chat = undefined;
-		globalState.indexedChildren = undefined;
-		globalState.joinedRows = undefined;
-		if (!patchableTui[REQUEST_RENDER_PATCH]) {
-			const original = patchableTui.requestRender;
-			const patched: RequestRenderFunction = (force) => {
-				try {
-					patchToolRows();
-				} catch (error) {
-					if (!globalState.patchErrorReported) {
-						globalState.patchErrorReported = true;
-						ui.notify(
-							`工具轨道连接失败：${error instanceof Error ? error.message : String(error)}`,
-							"warning",
-						);
-					}
-				}
-				return original.call(patchableTui, force);
-			};
-			Object.defineProperty(patchableTui, REQUEST_RENDER_PATCH, {
-				configurable: true,
-				value: { original, patched } satisfies RequestRenderPatch,
-			});
-			patchableTui.requestRender = patched;
-		}
-		patchToolRows();
-		return { render: () => [], invalidate: () => {} };
-	});
-	ui.setWidget("firecode-tui-capture", undefined);
-}
-
-export function uninstallGroupPatch(): void {
-	const tui = globalState.tui;
-	const requestPatch = tui?.[REQUEST_RENDER_PATCH];
-	if (tui && requestPatch && tui.requestRender === requestPatch.patched) {
-		tui.requestRender = requestPatch.original;
-		delete tui[REQUEST_RENDER_PATCH];
-	}
-
-	const prototype = globalState.patchedPrototype;
-	const rowPatch = prototype?.[GROUP_PATCH];
-	if (prototype && rowPatch && prototype.render === rowPatch.patched) {
-		prototype.render = rowPatch.original;
-		delete prototype[GROUP_PATCH];
-	}
-
-	globalState.tui = undefined;
-	globalState.chat = undefined;
-	globalState.indexedChildren = undefined;
-	globalState.indexedLength = undefined;
-	globalState.joinedRows = undefined;
-	globalState.patchedPrototype = undefined;
-	globalState.patchErrorReported = false;
-}