@zhushanwen/pi-statusline 0.4.0 → 0.4.1

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@zhushanwen/pi-statusline",
-  "version": "0.4.0",
+  "version": "0.4.1",
   "description": "Pi statusline extension — shows context usage, token speed, and provider quota in the footer.",
   "type": "module",
   "main": "src/index.ts",
@@ -22,12 +22,12 @@
     "index.ts"
   ],
   "dependencies": {
-    "@zhushanwen/pi-quota-providers": "0.4.0"
+    "@zhushanwen/pi-quota-providers": "0.4.1"
   },
   "peerDependencies": {
-    "@mariozechner/pi-coding-agent": "*",
+    "@earendil-works/pi-ai": "*",
     "@earendil-works/pi-tui": "*",
-    "@earendil-works/pi-ai": "*"
+    "@mariozechner/pi-coding-agent": "*"
   },
   "peerDependenciesMeta": {
     "@earendil-works/pi-tui": {
@@ -37,7 +37,12 @@
       "optional": true
     }
   },
+  "devDependencies": {
+    "vitest": "^4.1.8"
+  },
   "scripts": {
-    "typecheck": "npx tsc --noEmit"
+    "typecheck": "npx tsc --noEmit",
+    "test": "vitest run",
+    "test:watch": "vitest"
   }
 }

package/src/__tests__/format.test.ts

@@ -0,0 +1,608 @@
+/**
+ * Statusline 渲染逻辑测试
+ *
+ * 分两部分：
+ * 1. 真实数据验证：用实际 provider normalize 输出验证对齐和格式
+ * 2. Mock 回归测试：用固定数据防止未来格式变化破坏对齐
+ */
+
+import { describe, it, expect } from "vitest";
+import { INFINITE_WIN } from "@zhushanwen/pi-quota-providers";
+import type { QuotaWindow, QuotaProvider, NormalizedQuotaRow } from "@zhushanwen/pi-quota-providers";
+import {
+	formatWinCol,
+	buildTokenPlanLines,
+	buildSearchLine,
+	normalizeRows,
+	formatSpeedPart,
+	splitPath,
+	tailSessionId,
+	fmtResetSec,
+	fmtDuration,
+	fmtTokens,
+	fmtCount,
+	pctColor,
+	plainPallet,
+	plainThemeFg,
+} from "../format.js";
+
+// ── 辅助：从渲染文本中提取 `·` 的位置 ─────────────────
+
+function dotPositions(line: string): number[] {
+	return [...line.matchAll(/·/g)].map((m) => m.index);
+}
+
+// ════════════════════════════════════════════════════════
+// 1. 基础格式化函数
+// ════════════════════════════════════════════════════════
+
+describe("fmtResetSec", () => {
+	it("0 或负数返回空串", () => {
+		expect(fmtResetSec(0)).toBe("");
+		expect(fmtResetSec(-1)).toBe("");
+	});
+
+	it("分钟级：90s → 1m", () => {
+		expect(fmtResetSec(90)).toBe("1m");
+	});
+
+	it("小时级：3661s → 1h1m", () => {
+		expect(fmtResetSec(3661)).toBe("1h1m");
+	});
+
+	it("天级：90061s → 1d1h", () => {
+		expect(fmtResetSec(90061)).toBe("1d1h");
+	});
+});
+
+describe("fmtDuration", () => {
+	it("秒级", () => expect(fmtDuration(5000)).toBe("5s"));
+	it("分钟级", () => expect(fmtDuration(65000)).toBe("1m05s"));
+	it("小时级", () => expect(fmtDuration(3660000)).toBe("1h01m"));
+});
+
+describe("fmtTokens", () => {
+	it("< 1K", () => expect(fmtTokens(999)).toBe("999"));
+	it("1K+", () => expect(fmtTokens(1500)).toBe("1.5K"));
+	it("1M+", () => expect(fmtTokens(1_500_000)).toBe("1.5M"));
+});
+
+describe("fmtCount", () => {
+	it("< 1K", () => expect(fmtCount(500)).toBe("500"));
+	it("1K+", () => expect(fmtCount(1500)).toBe("1.5k"));
+});
+
+// ════════════════════════════════════════════════════════
+// 1b. pctColor — 颜色边界值
+// ════════════════════════════════════════════════════════
+
+describe("pctColor", () => {
+	it("< 40% → success", () => expect(pctColor(0)).toBe("success"));
+	it("39% → success", () => expect(pctColor(39)).toBe("success"));
+	it("40% → accent", () => expect(pctColor(40)).toBe("accent"));
+	it("59% → accent", () => expect(pctColor(59)).toBe("accent"));
+	it("60% → warning", () => expect(pctColor(60)).toBe("warning"));
+	it("79% → warning", () => expect(pctColor(79)).toBe("warning"));
+	it("80% → error", () => expect(pctColor(80)).toBe("error"));
+	it("100% → error", () => expect(pctColor(100)).toBe("error"));
+});
+
+// ════════════════════════════════════════════════════════
+// 1c. 路径工具
+// ════════════════════════════════════════════════════════
+
+describe("splitPath", () => {
+	it("标准绝对路径", () => {
+		expect(splitPath("/Users/foo/project")).toEqual(["Users", "foo", "project"]);
+	});
+
+	it("相对路径", () => {
+		expect(splitPath("foo/bar")).toEqual(["foo", "bar"]);
+	});
+
+	it("空串", () => {
+		expect(splitPath("")).toEqual([]);
+	});
+
+	it("尾部分隔符", () => {
+		expect(splitPath("/foo/bar/")).toEqual(["foo", "bar"]);
+	});
+});
+
+describe("tailSessionId", () => {
+	it("正常路径截取末尾 12 字符", () => {
+		// abc123def456.json → pop → abc123def456.json → slice(-12) → 3def456.json
+		expect(tailSessionId("/path/to/abc123def456.json", 12)).toBe("3def456.json");
+	});
+
+	it("undefined 返回空串", () => {
+		expect(tailSessionId(undefined, 12)).toBe("");
+	});
+
+	it("空串返回空串", () => {
+		expect(tailSessionId("", 12)).toBe("");
+	});
+
+	it("路径短于 n 字符时返回全名", () => {
+		expect(tailSessionId("file.json", 12)).toBe("file.json");
+	});
+});
+
+// ════════════════════════════════════════════════════════
+// 2. formatWinCol — 列对齐核心
+// ════════════════════════════════════════════════════════
+
+describe("formatWinCol", () => {
+	it("有限百分比 + 有 reset", () => {
+		const result = formatWinCol("5h", { pct: 7, resetSec: 120 }, plainPallet, plainThemeFg);
+		expect(result).toBe("5h    7%       2m");
+	});
+
+	it("有限百分比 + 无 reset", () => {
+		const result = formatWinCol("wk", { pct: 42, resetSec: null }, plainPallet, plainThemeFg);
+		// reset 列补 RESET_COL_W 空格
+		expect(result).toBe("wk   42%         ");
+	});
+
+	it("无限（pct=null）", () => {
+		const result = formatWinCol("mh", INFINITE_WIN, plainPallet, plainThemeFg);
+		// ∞ padStart(4) 右对齐 + "  " + "--" padStart(7)
+		expect(result).toBe("mh     ∞       --");
+	});
+
+	it("100% 满格", () => {
+		const result = formatWinCol("5h", { pct: 100, resetSec: 1800 }, plainPallet, plainThemeFg);
+		expect(result).toBe("5h  100%      30m");
+	});
+
+	it("0%", () => {
+		const result = formatWinCol("5h", { pct: 0, resetSec: 5400 }, plainPallet, plainThemeFg);
+		expect(result).toBe("5h    0%    1h30m");
+	});
+
+	it("∞ 列和有限列总字符宽度一致", () => {
+		const infinite = formatWinCol("5h", INFINITE_WIN, plainPallet, plainThemeFg);
+		const finite = formatWinCol("5h", { pct: 7, resetSec: 120 }, plainPallet, plainThemeFg);
+		expect(infinite.length).toBe(finite.length);
+	});
+
+	it("resetSec=0 与 resetSec=null 同宽", () => {
+		const zero = formatWinCol("5h", { pct: 50, resetSec: 0 }, plainPallet, plainThemeFg);
+		const nul = formatWinCol("5h", { pct: 50, resetSec: null }, plainPallet, plainThemeFg);
+		expect(zero.length).toBe(nul.length);
+		expect(zero).toBe(nul); // 都是空格填充
+	});
+
+	it("浮点 pct 四舍五入", () => {
+		const down = formatWinCol("5h", { pct: 7.4, resetSec: 0 }, plainPallet, plainThemeFg);
+		const up = formatWinCol("5h", { pct: 7.5, resetSec: 0 }, plainPallet, plainThemeFg);
+		expect(down).toContain("  7%");
+		expect(up).toContain("  8%");
+	});
+});
+
+// ════════════════════════════════════════════════════════
+// 2b. formatWinCol 边界场景补充
+// ════════════════════════════════════════════════════════
+
+describe("formatWinCol 边界", () => {
+	it("resetSec=0 与 resetSec=null 完全相同（都填空格）", () => {
+		const zero = formatWinCol("5h", { pct: 50, resetSec: 0 }, plainPallet, plainThemeFg);
+		const nul = formatWinCol("5h", { pct: 50, resetSec: null }, plainPallet, plainThemeFg);
+		expect(zero).toBe(nul);
+	});
+
+	it("浮点 pct 四舍五入", () => {
+		const down = formatWinCol("5h", { pct: 7.4, resetSec: 0 }, plainPallet, plainThemeFg);
+		const up = formatWinCol("5h", { pct: 7.5, resetSec: 0 }, plainPallet, plainThemeFg);
+		expect(down).toContain("  7%");
+		expect(up).toContain("  8%");
+	});
+
+	it("负数 pct 原样显示", () => {
+		const result = formatWinCol("5h", { pct: -1, resetSec: 0 }, plainPallet, plainThemeFg);
+		expect(result).toContain("-1%");
+	});
+});
+
+// ════════════════════════════════════════════════════════
+// 3b. buildTokenPlanLines 空数据
+// ════════════════════════════════════════════════════════
+
+describe("buildTokenPlanLines 空数据", () => {
+	it("空 cache 返回空数组", () => {
+		const providers: QuotaProvider[] = [{
+			id: "test", label: "test", category: "token-plan",
+			fetch: async () => null, normalize: () => ({ label: "test", wins: [INFINITE_WIN, INFINITE_WIN, INFINITE_WIN] }),
+		}];
+		expect(buildTokenPlanLines({}, providers, plainPallet, plainThemeFg)).toEqual([]);
+	});
+
+	it("空 providers 返回空数组", () => {
+		expect(buildTokenPlanLines({ test: {} }, [], plainPallet, plainThemeFg)).toEqual([]);
+	});
+
+	it("search-tool 类型的 provider 不出现", () => {
+		const providers: QuotaProvider[] = [{
+			id: "tavily", label: "tavily", category: "search-tool",
+			fetch: async () => null, normalize: () => null,
+		}];
+		expect(buildTokenPlanLines({ tavily: {} }, providers, plainPallet, plainThemeFg)).toEqual([]);
+	});
+
+	it("单行全 ∞ 对齐不变", () => {
+		const providers: QuotaProvider[] = [{
+			id: "test", label: "test-plan", category: "token-plan",
+			fetch: async () => null, normalize: () => ({ label: "t", wins: [INFINITE_WIN, INFINITE_WIN, INFINITE_WIN] }),
+		}];
+		const lines = buildTokenPlanLines({ test: {} }, providers, plainPallet, plainThemeFg);
+		expect(lines).toHaveLength(1);
+		const dots = dotPositions(lines[0]!);
+		expect(dots).toHaveLength(2);
+		expect(dots[0]).toBe(dots[1]! - 20); // 每个 cell 格式一致
+	});
+
+	it("超长 label 会撑宽但同长 label 间对齐", () => {
+		const providers: QuotaProvider[] = [
+			{ id: "a", label: "short", category: "token-plan", fetch: async () => null,
+				normalize: () => ({ label: "s", wins: [{ pct: 10, resetSec: 100 }, { pct: 20, resetSec: 200 }, { pct: 30, resetSec: 300 }] }) },
+			{ id: "b", label: "another-one", category: "token-plan", fetch: async () => null,
+				normalize: () => ({ label: "v", wins: [{ pct: 50, resetSec: 0 }, INFINITE_WIN, INFINITE_WIN] }) },
+		];
+		const lines = buildTokenPlanLines({ a: {}, b: {} }, providers, plainPallet, plainThemeFg);
+		expect(lines).toHaveLength(2);
+		// 两个 label 都 <= 19，padEnd(19) 保证等宽
+		expect(lines[0]!.length).toBe(lines[1]!.length);
+	});
+});
+
+// ════════════════════════════════════════════════════════
+// 3. buildTokenPlanLines — 整行渲染和列对齐
+// ════════════════════════════════════════════════════════
+
+describe("buildTokenPlanLines — 真实数据对齐验证", () => {
+	/** 从截图推断的真实 provider normalize 输出 */
+	const realProviders: QuotaProvider[] = [
+		{
+			id: "zhipu",
+			label: "zhipu-coding-plan",
+			category: "token-plan",
+			fetch: async () => null,
+			normalize(): NormalizedQuotaRow {
+				return {
+					label: "Z.ai-pro",
+					wins: [{ pct: 4, resetSec: 15660 }, INFINITE_WIN, INFINITE_WIN],
+				};
+			},
+		},
+		{
+			id: "opencode-go",
+			label: "opencode-go",
+			category: "token-plan",
+			fetch: async () => null,
+			normalize(): NormalizedQuotaRow {
+				return {
+					label: "opencode-go",
+					wins: [
+						{ pct: 7, resetSec: 120 },
+						{ pct: 42, resetSec: 396000 },
+						{ pct: 71, resetSec: 1872600 },
+					],
+				};
+			},
+		},
+		{
+			id: "kimi-coding",
+			label: "kimi-coding-plan",
+			category: "token-plan",
+			fetch: async () => null,
+			normalize(): NormalizedQuotaRow {
+				return {
+					label: "kimi-coding",
+					wins: [
+						{ pct: 0, resetSec: 2280 },
+						{ pct: 70, resetSec: 52680 },
+						INFINITE_WIN,
+					],
+				};
+			},
+		},
+		{
+			id: "minimax",
+			label: "minimax-token-plan",
+			category: "token-plan",
+			fetch: async () => null,
+			normalize(): NormalizedQuotaRow {
+				return {
+					label: "minimax-token",
+					wins: [
+						{ pct: 98, resetSec: 2220 },
+						{ pct: 10, resetSec: 361440 },
+						INFINITE_WIN,
+					],
+				};
+			},
+		},
+	];
+
+	// 用假 cache 数据（只要 key 存在就行，normalize 不读 raw）
+	const cache: Record<string, unknown> = {
+		zhipu: { dummy: true },
+		"opencode-go": { dummy: true },
+		"kimi-coding": { dummy: true },
+		minimax: { dummy: true },
+	};
+
+	it("所有行使用 providers.json 的 label（非 normalize 返回的 label）", () => {
+		const lines = buildTokenPlanLines(cache, realProviders, plainPallet, plainThemeFg);
+		expect(lines[0]).toMatch(/^zhipu-coding-plan\s/);
+		expect(lines[1]).toMatch(/^opencode-go\s/);
+		expect(lines[2]).toMatch(/^kimi-coding-plan\s/);
+		expect(lines[3]).toMatch(/^minimax-token-plan\s/);
+	});
+
+	it("所有行的 `·` 分隔符位置一致", () => {
+		const lines = buildTokenPlanLines(cache, realProviders, plainPallet, plainThemeFg);
+		const positions = lines.map(dotPositions);
+		// 所有行的 · 位置应该相同
+		for (let i = 1; i < positions.length; i++) {
+			expect(positions[i], `Row ${i} dot positions mismatch`).toEqual(positions[0]);
+		}
+	});
+
+	it("所有行总长度一致（ASCII plain 模式下）", () => {
+		const lines = buildTokenPlanLines(cache, realProviders, plainPallet, plainThemeFg);
+		const lengths = lines.map((l) => l.length);
+		for (let i = 1; i < lengths.length; i++) {
+			expect(lengths[i], `Row ${i} length ${lengths[i]} !== Row 0 length ${lengths[0]}`).toBe(lengths[0]);
+		}
+	});
+
+	it("输出快照", () => {
+		const lines = buildTokenPlanLines(cache, realProviders, plainPallet, plainThemeFg);
+		for (const line of lines) {
+			// 确保无 ANSI 转义
+			expect(line).not.toMatch(/\x1b\[/);
+		}
+		// 肉眼可检查的快照
+		expect(lines).toMatchInlineSnapshot(`
+			[
+			  "zhipu-coding-plan  5h    4%    4h21m · wk     ∞       -- · mh     ∞       --",
+			  "opencode-go        5h    7%       2m · wk   42%    4d14h · mh   71%   21d16h",
+			  "kimi-coding-plan   5h    0%      38m · wk   70%   14h38m · mh     ∞       --",
+			  "minimax-token-plan 5h   98%      37m · wk   10%     4d4h · mh     ∞       --",
+			]
+		`);
+	});
+});
+
+// ════════════════════════════════════════════════════════
+// 4. buildSearchLine — Tavily 显示
+// ════════════════════════════════════════════════════════
+
+describe("buildSearchLine", () => {
+	const searchProviders: QuotaProvider[] = [
+		{
+			id: "tavily",
+			label: "tavily",
+			category: "search-tool",
+			fetch: async () => null,
+			normalize: () => null,
+		},
+	];
+
+	it("有 planUsage/planLimit 时优先使用（API 调用次数）", () => {
+		const cache = {
+			tavily: { planUsage: 892, planLimit: 5000, available: 4, total: 5 },
+		};
+		const result = buildSearchLine(cache, searchProviders, plainPallet, plainThemeFg);
+		expect(result).toContain("892");
+		expect(result).toContain("5000");
+		expect(result).toContain("18%");
+		expect(result).not.toContain("4/5");
+	});
+
+	it("无 planUsage/planLimit 时 fallback 到 available/total（key 数量）", () => {
+		const cache = {
+			tavily: { available: 4, total: 5 },
+		};
+		const result = buildSearchLine(cache, searchProviders, plainPallet, plainThemeFg);
+		expect(result).toContain("4/5");
+		expect(result).toContain("80%");
+	});
+
+	it("total <= 0 时不显示", () => {
+		const cache = {
+			tavily: { planUsage: 0, planLimit: 0, available: 4, total: 0 },
+		};
+		const result = buildSearchLine(cache, searchProviders, plainPallet, plainThemeFg);
+		expect(result).toBe("");
+	});
+
+	it("无数据时不显示", () => {
+		const cache = {};
+		const result = buildSearchLine(cache, searchProviders, plainPallet, plainThemeFg);
+		expect(result).toBe("");
+	});
+
+	it("百分比计算精度（used=1, total=3 → 33%）", () => {
+		const cache = { tavily: { planUsage: 1, planLimit: 3 } };
+		const result = buildSearchLine(cache, searchProviders, plainPallet, plainThemeFg);
+		expect(result).toContain("33%");
+	});
+
+	it("多 search-tool 用 | 分隔", () => {
+		const providers: QuotaProvider[] = [
+			{ id: "tavily", label: "tavily", category: "search-tool", fetch: async () => null, normalize: () => null },
+			{ id: "other", label: "other", category: "search-tool", fetch: async () => null, normalize: () => null },
+		];
+		const cache = {
+				tavily: { planUsage: 100, planLimit: 1000 },
+				other: { planUsage: 50, planLimit: 500 },
+			};
+		const result = buildSearchLine(cache, providers, plainPallet, plainThemeFg);
+		expect(result).toContain(" | ");
+		expect(result).toContain("tavily");
+		expect(result).toContain("other");
+	});
+
+	it("used=0 时仍显示", () => {
+		const cache = { tavily: { planUsage: 0, planLimit: 5000 } };
+		const result = buildSearchLine(cache, searchProviders, plainPallet, plainThemeFg);
+		expect(result).toContain("0/5000");
+		expect(result).toContain("0%");
+	});
+});
+
+// ════════════════════════════════════════════════════════
+// 5. normalizeRows — label 优先级
+// ════════════════════════════════════════════════════════
+
+describe("normalizeRows", () => {
+	it("优先使用 provider.label（来自 providers.json）", () => {
+		const providers: QuotaProvider[] = [{
+			id: "test",
+			label: "configured-label",
+			category: "token-plan",
+			fetch: async () => null,
+			normalize: () => ({ label: "dynamic-label", wins: [INFINITE_WIN, INFINITE_WIN, INFINITE_WIN] }),
+		}];
+		const rows = normalizeRows({ test: {} }, providers);
+		expect(rows[0]?.name).toBe("configured-label");
+	});
+
+	it("provider.label 缺失时 fallback 到 normalize 返回的 label", () => {
+		const providers: QuotaProvider[] = [{
+			id: "test",
+			label: "",
+			category: "token-plan",
+			fetch: async () => null,
+			normalize: () => ({ label: "dynamic-label", wins: [INFINITE_WIN, INFINITE_WIN, INFINITE_WIN] }),
+		}];
+		const rows = normalizeRows({ test: {} }, providers);
+		expect(rows[0]?.name).toBe("dynamic-label");
+	});
+
+	it("normalize 返回 null 时跳过该 provider", () => {
+		const providers: QuotaProvider[] = [{
+			id: "test",
+			label: "test",
+			category: "token-plan",
+			fetch: async () => null,
+			normalize: () => null,
+		}];
+		const rows = normalizeRows({ test: {} }, providers);
+		expect(rows).toHaveLength(0);
+	});
+
+	it("normalize 抛异常时不影响其他 provider", () => {
+		const providers: QuotaProvider[] = [
+			{
+				id: "bad",
+				label: "bad",
+				category: "token-plan",
+				fetch: async () => null,
+				normalize: () => { throw new Error("boom"); },
+			},
+			{
+				id: "good",
+				label: "good",
+				category: "token-plan",
+				fetch: async () => null,
+				normalize: () => ({ label: "good", wins: [INFINITE_WIN, INFINITE_WIN, INFINITE_WIN] }),
+			},
+		];
+		const rows = normalizeRows({ bad: {}, good: {} }, providers);
+		expect(rows).toHaveLength(1);
+		expect(rows[0]?.name).toBe("good");
+	});
+
+	it("search-tool 类型的 provider 被过滤", () => {
+		const providers: QuotaProvider[] = [{
+			id: "tavily", label: "tavily", category: "search-tool",
+			fetch: async () => null,
+			normalize: () => ({ label: "tavily", wins: [INFINITE_WIN, INFINITE_WIN, INFINITE_WIN] }),
+		}];
+		const rows = normalizeRows({ tavily: {} }, providers);
+		expect(rows).toHaveLength(0);
+	});
+
+	it("cache 中无对应 key 时跳过", () => {
+		const providers: QuotaProvider[] = [{
+			id: "test", label: "test", category: "token-plan",
+			fetch: async () => null,
+			normalize: () => ({ label: "test", wins: [INFINITE_WIN, INFINITE_WIN, INFINITE_WIN] }),
+		}];
+		const rows = normalizeRows({ other: {} }, providers);
+		expect(rows).toHaveLength(0);
+	});
+});
+
+// ════════════════════════════════════════════════════════
+// 6. Mock 数据回归测试 — 边界场景
+// ════════════════════════════════════════════════════════
+
+describe("回归：边界百分比和 reset 时间对齐", () => {
+	const p = plainPallet;
+	const fg = plainThemeFg;
+
+	it("pct=0% 和 pct=100% 与 ∞ 对齐", () => {
+		const zero = formatWinCol("5h", { pct: 0, resetSec: 0 }, p, fg);
+		const full = formatWinCol("5h", { pct: 100, resetSec: 0 }, p, fg);
+		const inf = formatWinCol("5h", INFINITE_WIN, p, fg);
+		expect(zero.length).toBe(full.length);
+		expect(zero.length).toBe(inf.length);
+	});
+
+	it("resetSec=null 与有 resetSec 对齐", () => {
+		const noReset = formatWinCol("wk", { pct: 50, resetSec: null }, p, fg);
+		const hasReset = formatWinCol("wk", { pct: 50, resetSec: 3600 }, p, fg);
+		expect(noReset.length).toBe(hasReset.length);
+	});
+
+	it("多行组合对齐：全 ∞ / 全有限 / 混合", () => {
+		const scenarios: Array<{ pct: number | null; resetSec: number | null }> = [
+			{ pct: null, resetSec: null },
+			{ pct: 0, resetSec: 30 },
+			{ pct: 50, resetSec: 3600 },
+			{ pct: 99, resetSec: 86400 },
+			{ pct: 100, resetSec: null },
+		];
+		const results = scenarios.map((w) =>
+			formatWinCol("5h", w as QuotaWindow, p, fg),
+		);
+		const lengths = results.map((r) => r.length);
+		for (let i = 1; i < lengths.length; i++) {
+			expect(lengths[i], `scenario ${i} length mismatch`).toBe(lengths[0]);
+		}
+	});
+});
+
+// ════════════════════════════════════════════════════════
+// 7. formatSpeedPart — buildLine2 速度部分
+// ════════════════════════════════════════════════════════
+
+describe("formatSpeedPart", () => {
+	it("current + day 都有", () => {
+		const result = formatSpeedPart({ current: 127, day: 85 }, plainPallet);
+		expect(result).toBe("│ speed 127t/s · day 85t/s");
+	});
+
+	it("只有 current", () => {
+		const result = formatSpeedPart({ current: 50, day: 0 }, plainPallet);
+		expect(result).toBe("│ speed 50t/s");
+	});
+
+	it("只有 day", () => {
+		const result = formatSpeedPart({ current: 0, day: 30 }, plainPallet);
+		expect(result).toBe("│ speed day 30t/s");
+	});
+
+	it("都为 0 时返回空串", () => {
+		expect(formatSpeedPart({ current: 0, day: 0 }, plainPallet)).toBe("");
+	});
+
+	it("负数不显示", () => {
+		expect(formatSpeedPart({ current: -1, day: -1 }, plainPallet)).toBe("");
+	});
+});

package/src/format.ts

@@ -0,0 +1,227 @@
+/**
+ * Statusline 纯格式化函数
+ *
+ * 从 index.ts 提取的可测试纯函数。
+ * 不依赖 Pi 运行时（ExtensionAPI / Theme），只做数据→字符串转换。
+ */
+
+import type { QuotaWindow, QuotaProvider } from "@zhushanwen/pi-quota-providers";
+
+// ── 时间常量 ───────────────────────────────────────────
+
+const MS_PER_SEC = 1000;
+const SEC_PER_MIN = 60;
+const MIN_PER_HOUR = 60;
+const HOURS_PER_DAY = 24;
+const SEC_PER_HOUR = SEC_PER_MIN * MIN_PER_HOUR;
+const SEC_PER_DAY = SEC_PER_HOUR * HOURS_PER_DAY;
+
+// ── token 数字单位阈值 ─────────────────────────────────
+
+const KILO = 1_000;
+const MILLION = 1_000_000;
+
+// ── 百分比阈值 ─────────────────────────────────────────
+
+const PCT_HIGH = 80;
+const PCT_MED = 60;
+const PCT_LOW = 40;
+const PERCENT_SCALE = 100;
+
+// ── 渲染常量 ───────────────────────────────────────────
+
+/** 标题列宽（按最长 "minimax-token-plan"=18，+1 空格余量） */
+export const TITLE_COL_W = 19;
+/** reset 时间列宽（fmtResetSec 最长 "12d23h"=6 + 1 空格余量） */
+export const RESET_COL_W = 7;
+/** pct 列宽（"100%"=4，但 padStart(3) 给 " 23%"=4） */
+export const PCT_COL_W = 3;
+
+// ── 列定义 ─────────────────────────────────────────────
+
+export const COLS = [
+	{ key: "5h", label: "5h" },
+	{ key: "week", label: "wk" },
+	{ key: "month", label: "mh" },
+] as const;
+
+// ── 格式化函数 ─────────────────────────────────────────
+
+const MIN_PAD = 2;
+
+export function fmtDuration(ms: number): string {
+	const s = Math.floor(ms / MS_PER_SEC);
+	if (s < SEC_PER_MIN) return `${s}s`;
+	const m = Math.floor(s / SEC_PER_MIN);
+	if (m < MIN_PER_HOUR) return `${m}m${String(s % SEC_PER_MIN).padStart(MIN_PAD, "0")}s`;
+	return `${Math.floor(m / MIN_PER_HOUR)}h${String(m % MIN_PER_HOUR).padStart(MIN_PAD, "0")}m`;
+}
+
+export function fmtTokens(n: number): string {
+	if (n >= MILLION) return `${(n / MILLION).toFixed(1)}M`;
+	if (n >= KILO) return `${(n / KILO).toFixed(1)}K`;
+	return `${n}`;
+}
+
+export function fmtResetSec(sec: number): string {
+	if (sec <= 0) return "";
+	const d = Math.floor(sec / SEC_PER_DAY);
+	const h = Math.floor((sec % SEC_PER_DAY) / SEC_PER_HOUR);
+	const m = Math.floor((sec % SEC_PER_HOUR) / SEC_PER_MIN);
+	if (d > 0) return `${d}d${h}h`;
+	if (h > 0) return `${h}h${m}m`;
+	return `${m}m`;
+}
+
+export function fmtCount(n: number): string {
+	return n < KILO ? `${n}` : `${(n / KILO).toFixed(1)}k`;
+}
+
+/** 按百分比返回语义色 token */
+export function pctColor(pct: number): "error" | "warning" | "accent" | "success" {
+	if (pct >= PCT_HIGH) return "error";
+	if (pct >= PCT_MED) return "warning";
+	if (pct >= PCT_LOW) return "accent";
+	return "success";
+}
+
+// ── 速度渲染 ─────────────────────────────────────────
+
+export interface SpeedLike {
+	current: number;
+	day: number;
+}
+
+/** 渲染速度部分：speed 123t/s · day 85t/s（无速度返回空串） */
+export function formatSpeedPart(sp: SpeedLike, p: PlainPallet): string {
+	const parts: string[] = [];
+	if (sp.current > 0) parts.push(`${p.g(`${sp.current}`)}${p.d("t/s")}`);
+	if (sp.day > 0) parts.push(`${p.d("day")} ${p.g(`${sp.day}`)}${p.d("t/s")}`);
+	return parts.length ? `│ ${p.d("speed")} ${parts.join(" · ")}` : "";
+}
+
+// ── 路径工具 ─────────────────────────────────────────
+
+/** 把路径切成段（按系统分隔符） */
+export function splitPath(p: string): string[] {
+	return p.split("/").filter(Boolean);
+}
+
+/** 截取 sessionId 文件名的末尾 N 字符（去路径） */
+export function tailSessionId(filePath: string | undefined, n: number): string {
+	if (!filePath) return "";
+	return filePath.split("/").pop()?.slice(-n) ?? "";
+}
+
+// ── Palette（strip ANSI 的 plain 版本，用于测试） ──────
+
+export interface PlainPallet {
+	d: (s: string) => string;
+	v: (s: string) => string;
+	g: (s: string) => string;
+	w: (s: string) => string;
+	a: (s: string) => string;
+	m: (s: string) => string;
+}
+
+/** 无 ANSI 色码的 palette，返回原始字符串 */
+export const plainPallet: PlainPallet = {
+	d: (s) => s,
+	v: (s) => s,
+	g: (s) => s,
+	w: (s) => s,
+	a: (s) => s,
+	m: (s) => s,
+};
+
+/** 模拟 Theme.fg — 只返回原始文本 */
+export const plainThemeFg = (_token: string, text: string) => text;
+
+// ── 行数据类型 ─────────────────────────────────────────
+
+export interface QuotaRow {
+	name: string;
+	wins: [QuotaWindow, QuotaWindow, QuotaWindow];
+}
+
+// ── 核心渲染函数 ───────────────────────────────────────
+
+/** 缓存数据 → 归一化行（用于 token-plans 显示） */
+export function normalizeRows(
+	cache: Record<string, unknown>,
+	providers: QuotaProvider[],
+): QuotaRow[] {
+	const rows: QuotaRow[] = [];
+	for (const p of providers) {
+		if (p.category !== "token-plan") continue;
+		try {
+			const raw = cache[p.id];
+			if (!raw) continue;
+			const norm = p.normalize(raw);
+			if (!norm) continue;
+			// 优先使用 providers.json 配置的 label，fallback 到 normalize 返回的 label
+			rows.push({ name: p.label || norm.label, wins: norm.wins });
+		} catch {
+			// eslint-disable-next-line taste/no-silent-catch -- 单 provider normalize 失败不应拖垮整个 statusline
+		}
+	}
+	return rows;
+}
+
+/** 渲染搜索工具行 */
+export function buildSearchLine(
+	cache: Record<string, unknown>,
+	providers: QuotaProvider[],
+	p: PlainPallet,
+	themeFg: (token: string, text: string) => string,
+): string {
+	const parts: string[] = [];
+	for (const prov of providers) {
+		if (prov.category !== "search-tool") continue;
+		const raw = cache[prov.id] as Record<string, unknown> | undefined;
+		if (!raw) continue;
+		const used = (raw.planUsage as number) ?? (raw.available as number);
+		const total = (raw.planLimit as number) ?? (raw.total as number);
+		if (used === undefined || !total || total <= 0) continue;
+		const pct = Math.round((used / total) * PERCENT_SCALE);
+		const pctCol = themeFg(pctColor(pct), `${pct}%`);
+		parts.push(`${p.d(prov.label)} ${p.g(`${used}`)}/${p.v(`${total}`)}${p.d("次")} ${pctCol}`);
+	}
+	return parts.join(" | ");
+}
+
+/** 渲染 token-plan 行列表 */
+export function buildTokenPlanLines(
+	cache: Record<string, unknown>,
+	providers: QuotaProvider[],
+	p: PlainPallet,
+	themeFg: (token: string, text: string) => string,
+): string[] {
+	const rows = normalizeRows(cache, providers);
+	return rows.map((row) => {
+		const title = p.d(row.name.padEnd(TITLE_COL_W));
+		const cells = COLS.map((col, i) => {
+			const win = row.wins[i]!;
+			return formatWinCol(col.label, win, p, themeFg);
+		});
+		return title + cells.join(" · ");
+	});
+}
+
+/** 渲染单个窗口列：label pct% [reset]（无 bar） */
+export function formatWinCol(
+	label: string,
+	win: QuotaWindow,
+	p: PlainPallet,
+	themeFg: (token: string, text: string) => string,
+): string {
+	const pctWidth = PCT_COL_W + 1; // "NNN%" = padStart(3) + 1 = 4 chars
+	if (win.pct === null) {
+		// 无限：∞ 右对齐到 pctStr 宽度，reset 用 -- 占位
+		return `${p.d(label)}  ${p.v("∞".padStart(pctWidth))}  ${p.v("--".padStart(RESET_COL_W))}`;
+	}
+	const pctStr = `${String(Math.round(win.pct)).padStart(PCT_COL_W)}%`;
+	const rtRaw = win.resetSec != null && win.resetSec > 0 ? fmtResetSec(win.resetSec) : "";
+	const rtStr = rtRaw ? p.v(rtRaw.padStart(RESET_COL_W)) : " ".repeat(RESET_COL_W);
+	return `${p.d(label)}  ${themeFg(pctColor(win.pct), pctStr)}  ${rtStr}`;
+}

package/src/index.ts

@@ -28,6 +28,7 @@ import {
 	type QuotaProvider,
 } from "@zhushanwen/pi-quota-providers";
 import { registerSetupCommand } from "./setup.js";
+import { formatSpeedPart, splitPath, tailSessionId } from "./format.js";
 
 // ── 本地事件类型 ───────────────────────────────────────
 interface PiMessageEvent {
@@ -38,12 +39,6 @@ interface PiThinkingLevelEvent {
 	level: string;
 }
 
-interface SearchToolRaw {
-	available?: number;
-	used?: number;
-	total?: number;
-}
-
 // ── 时间常量 ───────────────────────────────────────────
 
 const MS_PER_SEC = 1000;
@@ -128,17 +123,6 @@ function pctColor(pct: number): "error" | "warning" | "accent" | "success" {
 	return "success";
 }
 
-/** 把 cwd 切成段，按系统分隔符（macOS/Linux: /；Windows: \） */
-function splitPath(p: string): string[] {
-	return p.split(sep).filter(Boolean);
-}
-
-/** 截取 sessionId 文件名的末尾 N 字符（去路径） */
-function tailSessionId(filePath: string | undefined, n: number): string {
-	if (!filePath) return "";
-	return filePath.split(sep).pop()?.slice(-n) ?? "";
-}
-
 /** 当前 cwd 是否在 git worktree 内（粗略：看 .git 是文件还是目录） */
 function isWorktree(cwd: string): boolean {
 	return existsSync(join(cwd, ".git"));
@@ -331,7 +315,8 @@ function normalizeRows(cache: CacheData, providers: QuotaProvider[]): QuotaRow[]
 			if (!raw) continue;
 			const norm = p.normalize(raw);
 			if (!norm) continue;
-			rows.push({ name: norm.label || p.label, wins: norm.wins });
+			// 优先使用 providers.json 配置的 label，fallback 到 normalize 返回的 label
+			rows.push({ name: p.label || norm.label, wins: norm.wins });
 		// eslint-disable-next-line taste/no-silent-catch -- render 容错：单 provider normalize 失败不应拖垮整个 statusline
 		} catch (e) {
 			console.warn(`[statusline] normalize failed for ${p.id}:`, e);
@@ -361,7 +346,11 @@ function buildLine2(ctx: ExtensionContext, st: StatuslineRuntimeState, p: Pallet
 	const provider = model.provider || "";
 	const modelId = model.id || model.name || "unknown";
 	const tlPart = st.thinkingLevel ? ` ${p.m(`[${st.thinkingLevel}]`)}` : "";
-	return `${p.d(provider)}/${p.a(modelId)}${tlPart}`;
+
+	const speedPart = formatSpeedPart(st.speed, p);
+	const speedPrefix = speedPart ? ` ${speedPart}` : "";
+
+	return `${p.d(provider)}/${p.a(modelId)}${tlPart}${speedPrefix}`;
 }
 
 function buildLine3(
@@ -419,10 +408,11 @@ function buildSearchLine(
 	const parts: string[] = [];
 	for (const prov of providers) {
 		if (prov.category !== "search-tool") continue;
-		const raw = (cache as Record<string, unknown>)[prov.id] as SearchToolRaw | undefined;
+		const raw = (cache as Record<string, unknown>)[prov.id] as Record<string, unknown> | undefined;
 		if (!raw) continue;
-		const used = raw.used ?? raw.available;
-		const total = raw.total;
+		// 优先使用 planUsage/planLimit（API 调用次数），fallback 到 available/total（key 数量）
+		const used = (raw.planUsage as number) ?? (raw.available as number);
+		const total = (raw.planLimit as number) ?? (raw.total as number);
 		if (used === undefined || !total || total <= 0) continue;
 		const pct = Math.round((used / total) * PERCENT_SCALE);
 		const pctCol = theme.fg(pctColor(pct), `${pct}%`);
@@ -450,8 +440,10 @@ function buildTokenPlanLines(
 
 /** 渲染单个窗口列：label pct% [reset]（无 bar） */
 function formatWinCol(label: string, win: QuotaWindow, p: Pallet, theme: Theme): string {
+	const pctWidth = PCT_COL_W + 1; // "NNN%" = padStart(3) + 1 = 4 chars
 	if (win.pct === null) {
-		return `${p.d(label)}  ${p.v("∞")}`;
+		// 无限：∞ 右对齐到 pctStr 宽度，reset 用 -- 占位
+		return `${p.d(label)}  ${p.v("∞".padStart(pctWidth))}  ${p.v("--".padStart(RESET_COL_W))}`;
 	}
 	const pctStr = `${String(Math.round(win.pct)).padStart(PCT_COL_W)}%`;
 	const rtRaw = win.resetSec != null && win.resetSec > 0 ? fmtResetSec(win.resetSec) : "";

package/src/setup.ts

@@ -59,12 +59,7 @@ export function registerSetupCommand(pi: ExtensionAPI): void {
 				missing,
 			});
 
-			try {
-				await ctx.sessionManager.appendEntry("user", prompt);
-			} catch (e) {
-				ctx.ui.notify(`Failed to inject setup prompt: ${(e as Error).message}`, "error");
-				return;
-			}
+			pi.sendUserMessage(prompt);
 			ctx.ui.notify(`Setup wizard started. Will generate: ${missing.join(", ")}`, "info");
 		},
 	});