pi-firecode 0.8.1 → 1.0.1

package/session/working-flame.ts

@@ -1,116 +0,0 @@
-/**
- * 火焰 working 效果：工作回合期间编辑器上方居中燃烧的多行火焰 widget。
- *
- * 走 setWidget(aboveEditor)：多行组件的合法槽位（审查活动框同款），由 TUI
- * 排版器独立渲染，与工具输出物理隔离——旧 working-style.ts 的事故是把多行帧
- * 塞进单行 spinner 通道，这里结构上不可能复发。agent_start 挂载、agent_end 撤下，
- * 动画计时器随组件 dispose 清理。高度随终端行数自适应，宽度不够时逐级缩小。
- *
- * 火焰即工作信号：回合或审查占用期隐藏宿主的 「Working...」 文本行。
- * review 模块也写 setWorkingVisible，且它的写入在同一同步调度链里排在占用事件之后；
- * 因此本模块的所有 UI 写入经微任务合并调度，在调度链收尾后作为最终仲裁者落地，
- * 消除「占用期 agent_end 无条件复显 Working 行」一类最后写者竞态。
- */
-import type { ExtensionAPI, ExtensionContext } from "@earendil-works/pi-coding-agent";
-import type { Component, TUI } from "@earendil-works/pi-tui";
-import { FLAME_FRAME_COUNT, flameFrameLines, flameFrameWidth } from "../flame-frames.js";
-
-const WIDGET_KEY = "firecode-working-flame";
-const FRAME_MS = 100;
-const MAX_HEIGHT = 10;
-const MIN_HEIGHT = 3;
-
-/** 高度自适应：约占终端四分之一，钳在 3–10 行；行数未知时给满高。 */
-export function flameHeightFor(terminalRows: number | undefined): number {
-	if (!terminalRows || !Number.isFinite(terminalRows)) return MAX_HEIGHT;
-	return Math.min(MAX_HEIGHT, Math.max(MIN_HEIGHT, Math.floor(terminalRows / 4)));
-}
-
-/** 宽度自适应：装不下就逐级降高（火焰宽随高缩），实在不行才隐藏。 */
-export function flameFitHeight(height: number, width: number): number {
-	for (let fit = height; fit >= MIN_HEIGHT; fit -= 1) {
-		if (flameFrameWidth(fit) <= width) return fit;
-	}
-	return 0;
-}
-
-export function registerWorkingFlame(pi: ExtensionAPI): void {
-	let turnActive = false;
-	let reviewHeld = false;
-	let ui: ExtensionContext["ui"] | undefined;
-	let pending = false;
-
-	const apply = () => {
-		pending = false;
-		if (!ui) return;
-		// Working 行：回合内由大火焰替代，审查占用期由审查活动条替代，两者都不在才复显。
-		if (typeof ui.setWorkingVisible === "function")
-			ui.setWorkingVisible(!turnActive && !reviewHeld);
-		if (typeof ui.setWidget !== "function") return;
-		if (turnActive && !reviewHeld)
-			ui.setWidget(
-				WIDGET_KEY,
-				(tui: TUI) => new WorkingFlame(tui, () => tui.requestRender()),
-				{ placement: "aboveEditor" },
-			);
-		else ui.setWidget(WIDGET_KEY, undefined);
-	};
-	/** 微任务合并：同一调度链内的多个事件只落地一次，且排在 review 的同步写入之后。 */
-	const sync = () => {
-		if (pending) return;
-		pending = true;
-		queueMicrotask(apply);
-	};
-
-	pi.on("agent_start", (_event, ctx) => {
-		if (ctx.mode !== "tui") return;
-		ui = ctx.ui;
-		turnActive = true;
-		sync();
-	});
-	pi.on("agent_end", (_event, ctx) => {
-		if (ctx.mode !== "tui") return;
-		ui = ctx.ui;
-		turnActive = false;
-		sync();
-	});
-	// 审查占用频道（review 模块发布）：审查活跃期活动条自带火焰，大火焰退让避免双火同烧。
-	pi.events?.on?.("herdr:blocked", (data: { active?: boolean } | undefined) => {
-		reviewHeld = Boolean(data?.active);
-		sync();
-	});
-}
-
-/** 居中的多行火焰；素材自带 ANSI 颜色与行尾复位。 */
-class WorkingFlame implements Component {
-	private frame = 0;
-	private readonly timer: ReturnType<typeof setInterval>;
-
-	constructor(
-		private readonly tui: TUI,
-		requestRender: () => void,
-	) {
-		this.timer = setInterval(() => {
-			this.frame += 1;
-			requestRender();
-		}, FRAME_MS);
-		this.timer.unref?.();
-	}
-
-	render(width: number): string[] {
-		const rows = (this.tui as TUI & { terminal?: { rows?: number } }).terminal?.rows;
-		const height = flameFitHeight(flameHeightFor(rows), Math.max(0, width));
-		if (height === 0) return [];
-		const flameWidth = flameFrameWidth(height);
-		const indent = " ".repeat(Math.max(0, Math.floor((width - flameWidth) / 2)));
-		return flameFrameLines(height, this.frame % FLAME_FRAME_COUNT).map(
-			(line) => `${indent}${line}`,
-		);
-	}
-
-	invalidate(): void {}
-
-	dispose(): void {
-		clearInterval(this.timer);
-	}
-}

package/statusbar/index.ts

@@ -1,118 +0,0 @@
-/** 底部两行状态栏：位置/会话名 + 模型/额度/上下文/缓存/速度。 */
-import { basename, join } from "node:path";
-import { type ExtensionAPI, getAgentDir } from "@earendil-works/pi-coding-agent";
-import { formatModelName } from "../format.js";
-import { thinkingColor } from "../theme.js";
-import { fileQuotaCache } from "./quota-cache.js";
-import { registerQuota } from "./quota.js";
-import type { QuotaStatus } from "./quota-parse.js";
-import {
-	alignRight,
-	fitMetadataLine,
-	fitStatusLine,
-	reviewStatus,
-	statusBadges,
-	latestCacheHitPercent,
-	renderCache,
-	renderContext,
-	renderLocation,
-	renderQuota,
-	renderTps,
-} from "./render.js";
-import { type TpsStatus, registerTps } from "./tps.js";
-
-export function registerStatusBar(pi: ExtensionAPI, subsession = false): void {
-	if (subsession) return;
-	let quota: QuotaStatus | undefined;
-	let tpsStatus: TpsStatus | undefined;
-	let requestRender = () => {};
-
-	registerQuota(
-		pi,
-		(status) => {
-			quota = status;
-			requestRender();
-		},
-		fileQuotaCache(join(getAgentDir(), "tmp")),
-	);
-	registerTps(pi, (status) => {
-		tpsStatus = status;
-		requestRender();
-	});
-
-	// 助手消息落定后缓存命中率才可算。
-	pi.on("message_end", (event) => {
-		if (event.message.role === "assistant") requestRender();
-	});
-
-	pi.on("session_start", (_event, ctx) => {
-		ctx.ui.setFooter((tui, theme, footerData) => {
-			requestRender = () => tui.requestRender();
-			const unsubscribe = footerData.onBranchChange(requestRender);
-
-			return {
-				dispose: unsubscribe,
-				invalidate() {},
-				render(width: number): string[] {
-					const statuses = footerData.getExtensionStatuses();
-					const review = reviewStatus(statuses);
-					const model = ctx.model;
-					const thinkingLevel = pi.getThinkingLevel();
-					const modelCore = `${theme.fg("accent", `🧠 ${formatModelName(model?.id)}`)}${
-						model?.reasoning
-							? theme.fg(thinkingColor(thinkingLevel), `/${thinkingLevel}`)
-							: ""
-					}`;
-					const modelText = statuses.has("pi-openai-native-fast")
-						? `${modelCore}${theme.fg("warning", " · ⚡fast")}`
-						: modelCore;
-
-					const location = renderLocation(
-						theme,
-						basename(ctx.sessionManager.getCwd()) || "/",
-						footerData.getGitBranch(),
-					);
-					const sessionName = ctx.sessionManager.getSessionName();
-					const title = sessionName ? theme.fg("dim", `💬 ${sessionName}`) : "";
-
-					const usage = ctx.getContextUsage();
-					const percent = usage?.percent;
-					const contextWindow = usage?.contextWindow ?? model?.contextWindow ?? 0;
-					const separator = ` ${theme.fg("dim", "｜")}`;
-
-					const statusLine = fitStatusLine(
-						{
-							model: modelText,
-							modelCompact: modelCore,
-							quota: quota ? renderQuota(theme, quota) : "",
-							quotaCompact: quota ? renderQuota(theme, quota, true) : "",
-							context: renderContext(theme, percent, contextWindow),
-							contextCompact: renderContext(theme, percent, contextWindow, true),
-							cache: renderCache(
-								theme,
-								latestCacheHitPercent(ctx.sessionManager.getEntries()),
-							),
-							tps: renderTps(theme, tpsStatus),
-						},
-						width,
-						separator,
-					);
-					return [
-						alignRight(
-							fitMetadataLine(location, title, width, separator, statusBadges(statuses, separator)),
-							review,
-							width,
-						),
-						statusLine,
-					];
-				},
-			};
-		});
-	});
-
-	pi.on("thinking_level_select", () => requestRender());
-	pi.on("session_shutdown", (_event, ctx) => {
-		requestRender = () => {};
-		ctx.ui.setFooter(undefined);
-	});
-}

package/statusbar/quota-cache.ts

@@ -1,60 +0,0 @@
-/**
- * 配额结果的跨进程缓存：同时开多个 pi 会话时共享一次抓取，
- * 也共享失败退避，避免离线时每个进程各自重试。
- */
-import { mkdirSync, readFileSync, renameSync, writeFileSync } from "node:fs";
-import { join } from "node:path";
-import type { QuotaWindow } from "./quota-parse.js";
-
-export type QuotaCacheEntry = {
-	/** 抓取成功的窗口；失败时为空数组 */
-	windows: QuotaWindow[];
-	/** 早于此刻不再发起请求（成功用 TTL，失败用退避） */
-	nextAttemptAt: number;
-	/** 连续失败次数，驱动退避档位 */
-	failures: number;
-};
-
-export interface QuotaCache {
-	read(provider: string): QuotaCacheEntry | undefined;
-	write(provider: string, entry: QuotaCacheEntry): void;
-}
-
-function isEntry(value: unknown): value is QuotaCacheEntry {
-	const entry = value as QuotaCacheEntry | undefined;
-	return (
-		!!entry &&
-		typeof entry === "object" &&
-		Array.isArray(entry.windows) &&
-		typeof entry.nextAttemptAt === "number" &&
-		typeof entry.failures === "number"
-	);
-}
-
-/** 每个 provider 一个文件，避免多进程写入互相覆盖。 */
-export function fileQuotaCache(directory: string): QuotaCache {
-	const pathFor = (provider: string) =>
-		join(directory, `firecode-quota-${provider}.json`);
-
-	return {
-		read(provider) {
-			try {
-				const parsed: unknown = JSON.parse(readFileSync(pathFor(provider), "utf8"));
-				return isEntry(parsed) ? parsed : undefined;
-			} catch {
-				return undefined;
-			}
-		},
-		write(provider, entry) {
-			const path = pathFor(provider);
-			const temporary = `${path}.${process.pid}.tmp`;
-			try {
-				mkdirSync(directory, { recursive: true });
-				writeFileSync(temporary, JSON.stringify(entry));
-				renameSync(temporary, path);
-			} catch {
-				// 缓存是加速手段，写失败只意味着下次照常请求
-			}
-		},
-	};
-}

package/statusbar/quota-parse.ts

@@ -1,85 +0,0 @@
-/** 供应商用量响应 → 剩余百分比窗口。响应结构不是公开契约，解析一律容错。 */
-
-type JsonRecord = Record<string, unknown>;
-
-export type QuotaWindow = {
-	label: string;
-	remaining: number;
-};
-
-export type QuotaStatus =
-	| { state: "loading" }
-	| { state: "ready"; windows: QuotaWindow[] }
-	| { state: "unavailable" };
-
-export const record = (value: unknown): JsonRecord | undefined =>
-	typeof value === "object" && value !== null ? (value as JsonRecord) : undefined;
-
-export const finiteNumber = (value: unknown): number | undefined =>
-	typeof value === "number" && Number.isFinite(value) ? value : undefined;
-
-/** 秒级时间戳或 ISO 字符串 → 毫秒时间戳。 */
-export const epochMillis = (value: unknown): number | undefined => {
-	if (typeof value === "number" && Number.isFinite(value)) return value * 1_000;
-	if (typeof value !== "string") return undefined;
-	const parsed = Date.parse(value);
-	return Number.isFinite(parsed) ? parsed : undefined;
-};
-
-const remainingPercent = (used: number) =>
-	Math.round(Math.max(0, Math.min(100, 100 - used)));
-
-function windowFromUsage(
-	label: string,
-	value: unknown,
-	usedKey: string,
-): QuotaWindow | undefined {
-	const used = finiteNumber(record(value)?.[usedKey]);
-	return used === undefined ? undefined : { label, remaining: remainingPercent(used) };
-}
-
-const defined = (window: QuotaWindow | undefined): window is QuotaWindow =>
-	Boolean(window);
-
-export function parseOpenAIQuota(value: unknown): QuotaWindow[] {
-	const rateLimit = record(record(value)?.rate_limit);
-	const primary = record(rateLimit?.primary_window);
-	const secondary = record(rateLimit?.secondary_window);
-	const label = (window: JsonRecord | undefined, fallback: string) => {
-		const seconds = finiteNumber(window?.limit_window_seconds);
-		if (!seconds) return fallback;
-		if (seconds % 86_400 === 0) return `${seconds / 86_400}d`;
-		if (seconds % 3_600 === 0) return `${seconds / 3_600}h`;
-		return fallback;
-	};
-	return [
-		windowFromUsage(label(primary, "5h"), primary, "used_percent"),
-		windowFromUsage(label(secondary, "7d"), secondary, "used_percent"),
-	].filter(defined);
-}
-
-export function parseAnthropicQuota(value: unknown): QuotaWindow[] {
-	const source = record(value);
-	return [
-		windowFromUsage("5h", source?.five_hour, "utilization"),
-		windowFromUsage("7d", source?.seven_day, "utilization"),
-	].filter(defined);
-}
-
-export function parseGrokQuota(
-	monthlyValue: unknown,
-	weeklyValue: unknown,
-): QuotaWindow[] {
-	const monthly = record(record(monthlyValue)?.config);
-	const weekly = record(record(weeklyValue)?.config);
-	const monthlyLimit = finiteNumber(record(monthly?.monthlyLimit)?.val);
-	const monthlyUsed = finiteNumber(record(monthly?.used)?.val);
-	const monthlyWindow =
-		monthlyLimit !== undefined && monthlyLimit > 0 && monthlyUsed !== undefined
-			? { label: "30d", remaining: remainingPercent((monthlyUsed / monthlyLimit) * 100) }
-			: undefined;
-	return [
-		windowFromUsage("7d", weekly, "creditUsagePercent"),
-		monthlyWindow,
-	].filter(defined);
-}

package/statusbar/quota.ts

@@ -1,203 +0,0 @@
-/**
- * 订阅额度抓取：事件驱动（会话启动、切模型、每轮结束），无定时轮询。
- * 新鲜度与失败退避都落在跨进程缓存里，多个 pi 会话共享同一次请求。
- */
-import { existsSync, readFileSync } from "node:fs";
-import { homedir } from "node:os";
-import { join } from "node:path";
-import type { ExtensionAPI, ExtensionContext } from "@earendil-works/pi-coding-agent";
-import type { QuotaCache } from "./quota-cache.js";
-import {
-	type QuotaStatus,
-	type QuotaWindow,
-	epochMillis,
-	parseAnthropicQuota,
-	parseGrokQuota,
-	parseOpenAIQuota,
-	record,
-} from "./quota-parse.js";
-
-// 供应商自家 CLI 使用的只读接口，schema 不是公开契约。
-const OPENAI_USAGE_URL = "https://chatgpt.com/backend-api/wham/usage";
-const ANTHROPIC_USAGE_URL = "https://api.anthropic.com/api/oauth/usage";
-const GROK_BILLING_URL = "https://cli-chat-proxy.grok.com/v1/billing";
-const FRESH_MS = 60_000;
-const REQUEST_TIMEOUT_MS = 3_000;
-/** 连续失败的退避档位，末档为上限 */
-const BACKOFF_MS = [60_000, 120_000, 300_000];
-
-function decodeJwtPayload(token: string): ReturnType<typeof record> {
-	const payload = token.split(".")[1];
-	if (!payload) return undefined;
-	try {
-		return record(JSON.parse(Buffer.from(payload, "base64url").toString("utf8")));
-	} catch {
-		return undefined;
-	}
-}
-
-function openAIAccountId(token: string): string | undefined {
-	const claims = record(decodeJwtPayload(token)?.["https://api.openai.com/auth"]);
-	return typeof claims?.chatgpt_account_id === "string" ? claims.chatgpt_account_id : undefined;
-}
-
-function grokCliToken(): string | undefined {
-	// Grok billing 拒绝 pi 的 OAuth token，只认官方 CLI 登录态。
-	const path = join(homedir(), ".grok", "auth.json");
-	if (!existsSync(path)) return undefined;
-	try {
-		const auth = record(JSON.parse(readFileSync(path, "utf8")));
-		for (const [scope, value] of Object.entries(auth ?? {})) {
-			if (!scope.startsWith("https://auth.x.ai::")) continue;
-			const credential = record(value);
-			const token = credential?.key;
-			const expiresAt = epochMillis(credential?.expires_at);
-			if (typeof token === "string" && token.length > 0 && (!expiresAt || expiresAt > Date.now()))
-				return token;
-		}
-	} catch {
-		return undefined;
-	}
-	return undefined;
-}
-
-async function fetchJson(url: string, headers: Record<string, string>): Promise<unknown> {
-	const response = await fetch(url, {
-		headers,
-		signal: AbortSignal.timeout(REQUEST_TIMEOUT_MS),
-	});
-	if (!response.ok) throw new Error(`Quota request failed: ${response.status}`);
-	return response.json();
-}
-
-type QuotaRequest =
-	| { provider: "openai-codex" | "anthropic"; token: Promise<string | undefined> }
-	| { provider: "xai"; token: string | undefined };
-
-type QuotaProvider = QuotaRequest["provider"];
-
-async function loadQuota(request: QuotaRequest): Promise<QuotaWindow[]> {
-	if (request.provider === "openai-codex") {
-		const token = await request.token;
-		if (!token) return [];
-		const accountId = openAIAccountId(token);
-		return parseOpenAIQuota(
-			await fetchJson(OPENAI_USAGE_URL, {
-				Authorization: `Bearer ${token}`,
-				...(accountId ? { "ChatGPT-Account-Id": accountId } : {}),
-				Accept: "application/json",
-			}),
-		);
-	}
-	if (request.provider === "anthropic") {
-		const token = await request.token;
-		if (!token) return [];
-		return parseAnthropicQuota(
-			await fetchJson(ANTHROPIC_USAGE_URL, {
-				Authorization: `Bearer ${token}`,
-				"anthropic-beta": "oauth-2025-04-20",
-				Accept: "application/json",
-			}),
-		);
-	}
-	const { token } = request;
-	if (!token) return [];
-	const headers = {
-		Authorization: `Bearer ${token}`,
-		"x-xai-token-auth": "xai-grok-cli",
-		Accept: "application/json",
-	};
-	const [monthly, weekly] = await Promise.allSettled([
-		fetchJson(GROK_BILLING_URL, headers),
-		fetchJson(`${GROK_BILLING_URL}?format=credits`, headers),
-	]);
-	if (monthly.status === "rejected" && weekly.status === "rejected") throw monthly.reason;
-	return parseGrokQuota(
-		monthly.status === "fulfilled" ? monthly.value : undefined,
-		weekly.status === "fulfilled" ? weekly.value : undefined,
-	);
-}
-
-const isQuotaProvider = (provider: string | undefined): provider is QuotaProvider =>
-	provider === "openai-codex" || provider === "anthropic" || provider === "xai";
-
-/** 只有订阅制（OAuth 登录）的供应商才有额度可查。 */
-function subscriptionProvider(ctx: ExtensionContext): QuotaProvider | undefined {
-	const model = ctx.model;
-	const provider = model?.provider;
-	if (!isQuotaProvider(provider)) return undefined;
-	if (
-		(provider === "anthropic" || provider === "xai") &&
-		(!model || !ctx.modelRegistry.isUsingOAuth(model))
-	)
-		return undefined;
-	return provider;
-}
-
-function quotaRequest(ctx: ExtensionContext, provider: QuotaProvider): QuotaRequest {
-	if (provider === "xai") return { provider, token: grokCliToken() };
-	return { provider, token: ctx.modelRegistry.getApiKeyForProvider(provider) };
-}
-
-const statusOf = (windows: QuotaWindow[]): QuotaStatus =>
-	windows.length ? { state: "ready", windows } : { state: "unavailable" };
-
-export function registerQuota(
-	pi: ExtensionAPI,
-	update: (status?: QuotaStatus) => void,
-	cache: QuotaCache,
-) {
-	let requestGeneration = 0;
-
-	const refresh = (ctx: ExtensionContext, force = false) => {
-		const provider = subscriptionProvider(ctx);
-		if (!provider) {
-			requestGeneration++;
-			update();
-			return;
-		}
-
-		const now = Date.now();
-		const cached = cache.read(provider);
-		if (cached && now < cached.nextAttemptAt) {
-			requestGeneration++;
-			update(statusOf(cached.windows));
-			return;
-		}
-		if (force) update({ state: "loading" });
-
-		const generation = ++requestGeneration;
-		// 先占住退避窗口，避免同一进程内并发请求叠加。
-		cache.write(provider, {
-			windows: cached?.windows ?? [],
-			nextAttemptAt: now + FRESH_MS,
-			failures: cached?.failures ?? 0,
-		});
-		void loadQuota(quotaRequest(ctx, provider))
-			.then((windows) => {
-				cache.write(provider, {
-					windows,
-					nextAttemptAt: Date.now() + FRESH_MS,
-					failures: 0,
-				});
-				if (generation === requestGeneration) update(statusOf(windows));
-			})
-			.catch(() => {
-				const failures = (cached?.failures ?? 0) + 1;
-				cache.write(provider, {
-					windows: [],
-					nextAttemptAt: Date.now() + BACKOFF_MS[Math.min(failures, BACKOFF_MS.length) - 1],
-					failures,
-				});
-				if (generation === requestGeneration) update({ state: "unavailable" });
-			});
-	};
-
-	pi.on("session_start", (_event, ctx) => refresh(ctx, true));
-	pi.on("model_select", (_event, ctx) => refresh(ctx, true));
-	pi.on("agent_end", (_event, ctx) => refresh(ctx));
-	pi.on("session_shutdown", () => {
-		requestGeneration++;
-		update();
-	});
-}