@zhushanwen/pi-statusline 0.1.3 → 0.4.0

package/README.md

@@ -1,38 +1,192 @@
 # statusline
 
-Pi 自定义状态栏 — 显示上下文用量、Token 速度、Provider 套餐额度。
+Pi 自定义状态栏 — 显示上下文用量、Token 流量、Provider 套餐额度、搜索工具额度。
 
-## 功能
+## 状态栏布局
 
-- **Line 1**：目录/仓库名 · 分支 │ session-name │ provider : model [thinking level]
-- **Line 2**：上下文用量 │ Token 速度（当前 + 日累计）│ 搜索配额
-- **Line 3-5**：套餐用量（Z.ai-pro / opencode-go / kimi-coding 等），进度条可视化
-- **Line 6**：时间 · 费用 · session ID
+```
+Line 1: 父目录/子目录 · ⎇ branch │ worktree
+Line 2: provider/model [thinking level]
+Line 3: ctx 45.2K/200K 23% │ from 13:25 · run 34m40s · last 12s │ ↑↓ 128.3k/8.5k │ 253b75.jsonl
+Line 4: tavily 234/1000次 23% | anysearch 250/500次 50%
+Line 5+: zhipu-coding-plan   5h  23%  4h11m  · wk   ∞  · mh   ∞
+        opencode-go         5h  45%  2h35m  · wk  12%  3d2h  · mh  78%  4d5h
+        kimi-coding-plan    5h  32%  1h42m  · wk  45%  2d8h  · mh   ∞
+        minimax-token-plan  5h  10%  4h55m  · wk   8%  5d1h  · mh  15%  12d
+```
+
+| 行 | 内容 | 说明 |
+|---|------|------|
+| 1 | 目录 + 分支 + worktree | 仓库路径显示倒数两级；worktree 文字标识 |
+| 2 | `provider/model [thinking]` | 完整 provider/model；thinking level 灰显 |
+| 3 | 上下文 + 时间 + 流量 + 会话 ID | `ctx` 百分比按区间配色（绿/黄/红）；`from` 启动时刻；`run` 运行时长；`last` 距上次 LLM 响应；`↑↓` 累计 input/output token；最后是 session 文件后缀 |
+| 4 | 搜索工具额度 | 多个工具用 ` \| ` 分隔；格式 `{label} {used}/{total}次 {pct}%` |
+| 5+ | token-plans 套餐 | 3 列：5h / wk / mh；去进度条纯文本；`∞` 表示无限；reset 时间右对齐 |
 
 ## 安装
 
 ```bash
-# symlink 方式（开发推荐）
-ln -s /path/to/xyz-pi-extensions-workspace/main/packages/statusline \
+# npm 方式（唯一正式方式）
+pi install npm:@zhushanwen/pi-statusline
+
+# 本地开发（symlink）
+ln -s /path/to/xyz-pi-extensions-workspace/main/extensions/statusline \
       ~/.pi/agent/extensions/statusline
+```
 
-# npm 方式（正式）
-pi install npm:@zhushanwen/pi-statusline
+## 配置
+
+扩展通过**声明式 JSON 配置**管理 provider 和凭证。首次使用需要运行：
+
+```bash
+/setup-statusline
+```
+
+命令行为：
+- 配置文件都存在 → 加载并打印审查摘要
+- 缺失 → 注入 LLM prompt，让 LLM 生成 demo 文件
+  - `providers.json` 默认启用所有内置 provider（用户后续可禁用）
+  - `secrets.json` 默认所有凭证用 `${ENV_VAR}` 引用（不写明文）
+  - 支持中英文（基于 `Intl.DateTimeFormat().resolvedOptions().locale`）
+
+### 配置文件位置
+
+| 文件 | 路径 | 作用 |
+|------|------|------|
+| providers.json | `~/.pi/agent/config/providers.json` | provider 声明 |
+| secrets.json | `~/.pi/agent/config/secrets.json` | 凭证（明文或 env 引用） |
+
+路径通过 `getAgentDir()` 派生，**不写绝对路径**。
+
+### providers.json schema
+
+```json
+{
+  "token-plans": [
+    {
+      "id": "zhipu",
+      "label": "zhipu-coding-plan",
+      "enabled": true,
+      "fetcher": "zhipu"
+    }
+  ],
+  "search-tools": [
+    {
+      "id": "tavily",
+      "label": "tavily",
+      "enabled": true,
+      "fetcher": "tavily"
+    }
+  ]
+}
+```
+
+| 字段 | 必填 | 说明 |
+|------|------|------|
+| `id` | ✓ | 在 cache 中的 key |
+| `label` | ✓ | 状态栏显示名 |
+| `enabled` | ✓ | `false` 跳过该 provider（保留配置便于回滚） |
+| `fetcher` | ✓ | 内置 fetcher ID（见下方支持列表） |
+
+### secrets.json schema
+
+```json
+{
+  "zhipu": {
+    "token": "${ZAI_AUTH_TOKEN}"
+  },
+  "tavily": {
+    "apiKey": "tvly-plain-text-token-here"
+  }
+}
+```
+
+- 每个 provider 是一个 section
+- value 字符串匹配 `^\$\{[A-Z_][A-Z0-9_]*\}$` → 当作环境变量引用，从 `process.env` 取
+- 环境变量不存在 → 静默返回空串（该 provider 拉不到数据）
+- 其它值 → 原样使用
+
+## 内置 Provider
+
+| fetcher | 类别 | 周期 | 说明 |
+|---------|------|------|------|
+| `zhipu` | token-plan | 5h | 智谱 GLM Coding |
+| `opencode-go` | token-plan | 5h/wk/mh | Go API |
+| `kimi-coding` | token-plan | 5h/wk/mh | Kimi |
+| `minimax` | token-plan | 5h/wk/mh | MiniMax |
+| `tavily` | search-tool | 次数 | 搜索 API |
+
+- **token-plan**：按 3 窗口（5h / wk / mh）显示用量 + reset 时间
+- **search-tool**：按 `used/total次` 显示搜索配额，多个工具用 `|` 分隔
+
+## 添加新 Provider
+
+三步走，**statusline 代码零修改**：
+
+### 1. 实现 fetcher
+
+在 `shared/quota-providers/src/providers/xxx.ts`：
+
+```typescript
+import type { QuotaProvider, NormalizedQuotaRow } from "./types.js";
+import { INFINITE_WIN } from "./types.js";
+
+export interface XxxData {
+	// 你的原始数据结构
+	pct: number;
+	resetSec: number;
+}
+
+async function fetchXxx(): Promise<XxxData | null> {
+	// 从 API 拉数据；失败/无凭证返回 null
+	return null;
+}
+
+export const xxxProvider: QuotaProvider<XxxData> = {
+	id: "xxx",
+	label: "xxx-plan",
+	category: "token-plan",  // 或 "search-tool"
+	fetch: fetchXxx,
+	normalize(raw): NormalizedQuotaRow | null {
+		return {
+			label: "xxx-plan",
+			wins: [
+				{ pct: raw.pct, resetSec: raw.resetSec },
+				INFINITE_WIN,
+				INFINITE_WIN,
+			],
+		};
+	},
+};
 ```
 
-## 使用
+### 2. 在 registry.ts 注册
 
-安装后自动生效，Pi 底部状态栏自动显示信息。
+`shared/quota-providers/src/registry.ts` 的 `FETCHERS` 和 `NORMALIZERS` 表加一行：
 
-## 支持的 Provider
+```typescript
+const FETCHERS: Record<string, Fetcher> = {
+	// ...
+	"xxx": xxxProvider.fetch as Fetcher,
+};
 
-| Provider | 额度类型 |
-|----------|---------|
-| Z.ai-pro (智谱) | 5h 重置周期 |
-| opencode-go (Go API) | 周期/周/月额度 |
-| kimi-coding (Kimi) | 周期额度 |
-| minimax | 周期额度 |
-| tavily | 搜索次数 |
+const NORMALIZERS: Record<string, Normalize> = {
+	// ...
+	"xxx": xxxProvider.normalize as Normalize,
+};
+```
+
+### 3. 用户在 providers.json 启用
+
+```json
+{
+  "token-plans": [
+    { "id": "xxx", "label": "xxx-plan", "enabled": true, "fetcher": "xxx" }
+  ]
+}
+```
+
+完事。状态栏下次渲染自动出现新行。
 
 ## 文件结构
 
@@ -40,14 +194,37 @@ pi install npm:@zhushanwen/pi-statusline
 statusline/
 ├── index.ts
 └── src/
-    ├── index.ts          # 入口 — Footer 渲染
-    ├── cache.ts          # 数据缓存 + Token 速度追踪
+    ├── index.ts            # 入口 — Footer 渲染 + 状态机
+    ├── setup.ts            # /setup-statusline 命令
+    └── setup-prompts.ts    # i18n prompt 模板
+
+shared/quota-providers/    # workspace 共享包
+├── index.ts
+└── src/
+    ├── cache.ts            # TTL 缓存 + Token 速度追踪
+    ├── config.ts           # providers.json 加载器
+    ├── secrets.ts          # secrets.json 加载器
+    ├── paths.ts            # 路径工具（getAgentDir）
+    ├── registry.ts         # 运行时 provider 构建
     └── providers/
-        ├── index.ts      # Provider 注册
-        ├── types.ts      # 额度类型定义
-        ├── zhipu.ts      # 智谱
-        ├── opencode-go.ts# Go API
-        ├── kimi-coding.ts# Kimi
-        ├── minimax.ts    # MiniMax
-        └── tavily.ts     # 搜索
+        ├── index.ts        # Provider 注册表
+        ├── types.ts        # QuotaProvider 接口
+        ├── zhipu.ts
+        ├── opencode-go.ts
+        ├── kimi-coding.ts
+        ├── minimax.ts
+        └── tavily.ts
 ```
+
+## 性能 / 缓存
+
+- provider 数据通过 `cache.ts` 缓存，TTL 5 分钟
+- `triggerUpdate()` 在 `session_start` / `message_end` 触发，但实际请求受 TTL/2 节流
+- `fetch` 失败 / 无凭证 → 保留旧值（Promise.allSettled 模式）
+- Token 速度按模型分别存到 `~/.pi/agent/token-stats/<model>.json`，30 天滚动窗口
+
+## 调试
+
+- `npx tsc --noEmit` — 类型检查
+- 修改 `providers.json` 后无需重启，下次 render 自动 reload
+- provider 加载失败会在 console.warn（`unknown fetcher: xxx`）

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@zhushanwen/pi-statusline",
-  "version": "0.1.3",
+  "version": "0.4.0",
   "description": "Pi statusline extension — shows context usage, token speed, and provider quota in the footer.",
   "type": "module",
   "main": "src/index.ts",
@@ -22,7 +22,7 @@
     "index.ts"
   ],
   "dependencies": {
-    "@zhushanwen/pi-quota-providers": "0.1.2"
+    "@zhushanwen/pi-quota-providers": "0.4.0"
   },
   "peerDependencies": {
     "@mariozechner/pi-coding-agent": "*",

package/src/index.ts

@@ -2,153 +2,151 @@
  * Pi Statusline — 自定义状态栏
  *
  * 布局：
- * Line 1: 目录/仓库名 · 分支 │ session-name │ provider : model [thinking level]
- * Line 2: ctx │ speed current+t/s day+t/s │ tavily
- * Line 3-5: 套餐用量（统一列对齐）
- *   Z.ai-pro      5h  XXX% [bar] ZzHh · wk  ∞ · mh  ∞  reset ZhZm
- *   opencode-go   5h  XXX% [bar] ZhZm · wk XXX% [bar] Zdh · mh XXX% [bar] Zdh
- *   kimi-coding   5h  XXX% [bar] ZhZm · wk XXX% [bar] Zdh · mh  ∞
- * Line 6: 时间 · 费用 · 会话ID
+ * Line 1: 父目录/子目录 · ⎇ branch │ worktree
+ * Line 2: provider/model [thinking level]
+ * Line 3: ctx X/Y 23% │ from · run · last │ ↑↓ in/out │ <sessionId>
+ * Line 4: search-tool 行（tavily 234/1000次 23% | anysearch 250/500次 50%）
+ * Line 5+: token-plans 行（去 bar，列对齐）
+ *
+ * 配置：通过 ~/.pi/agent/config/{providers,secrets}.json 声明式管理
  */
 
+import { existsSync } from "node:fs";
+import { join, sep } from "node:path";
 import type { AssistantMessage } from "@mariozechner/pi-ai";
 import type { ExtensionAPI, ExtensionContext, ReadonlyFooterDataProvider, Theme } from "@mariozechner/pi-coding-agent";
 import { truncateToWidth } from "@mariozechner/pi-tui";
 
-// ── 本地事件类型 ───────────────────────────────────────
-// Pi SDK 将 ContextEvent 定义为 any，此处用具体接口替代
-interface PiMessageEvent {
-  message: { role: string } & Record<string, unknown>;
-}
-
-interface PiThinkingLevelEvent {
-  level: string;
-}
 import {
 	readCache,
 	triggerUpdate,
 	trackSpeed,
-	PROVIDERS,
+	buildRuntimeProviders,
 	type CacheData,
 	type SpeedData,
 	type QuotaWindow,
+	type QuotaProvider,
 } from "@zhushanwen/pi-quota-providers";
-// ── 常量 ───────────────────────────────────────────────
+import { registerSetupCommand } from "./setup.js";
 
-const SEP = "│";
-const DOT = "·";
-const WIDE_THRESHOLD = 100;
-const RUN_UPDATE_MS = 5000;
-
-/** 标题列宽（按最长 "opencode-go"=11, +4 空格余量） */
-const TITLE_COL_W = 15;
-
-// ── Bar rendering (no ANSI) ──────────────────────────
-
-/** Render a usage bar with Unicode block chars and semantic fg tokens. */
-function barSegment(pct: number, theme: Theme, w = 6): string {
-	const p = Math.max(0, Math.min(100, Math.round(pct)));
-	const filled = Math.floor((p * w) / 100);
-	const fillToken =
-		p >= 80 ? "error"
-			: p >= 60 ? "warning"
-			: p >= 40 ? "accent"
-			: "success";
-	return theme.fg(fillToken, "█".repeat(filled)) + theme.fg("muted", "░".repeat(w - filled));
+// ── 本地事件类型 ───────────────────────────────────────
+interface PiMessageEvent {
+	message: { role: string } & Record<string, unknown>;
 }
 
-/** 构建一个窗口列，所有单元格 data 区域固定可见宽度。 */
-function winCol(
-	label: string,
-	pct: number | null,
-	resetSec: number | null,
-	wide: boolean,
-	d: (s: string) => string,
-	v: (s: string) => string,
-	theme: Theme,
-): string {
-	const l = d(label);
-	if (pct === null) {
-		const dataW = wide ? 20 : 12;
-		return `${l}  ${v(padCenter("∞", dataW))}`;
-	}
-	const pctStr = `${String(Math.round(pct)).padStart(3)}%`;
-	const rtRaw = resetSec && resetSec > 0 ? fmtResetSec(resetSec) : "";
-	const rtStr = rtRaw.padEnd(6);
-
-	if (wide) {
-		const bar = barSegment(pct, theme, 6);
-		return `${l}  ${v(pctStr)}  ${bar}  ${v(rtStr)}`;
-	}
-	return `${l}  ${v(pctStr)}  ${v(rtStr)}`;
+interface PiThinkingLevelEvent {
+	level: string;
 }
 
-/** 将 str 居中到 width 宽度，用空格填充 */
-function padCenter(str: string, width: number): string {
-	const pad = width - str.length;
-	if (pad <= 0) return str.slice(0, width);
-	const left = Math.floor(pad / 2);
-	const right = pad - left;
-	return " ".repeat(left) + str + " ".repeat(right);
+interface SearchToolRaw {
+	available?: number;
+	used?: number;
+	total?: number;
 }
 
+// ── 时间常量 ───────────────────────────────────────────
+
+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;
+
+// ── 渲染常量 ───────────────────────────────────────────
+
+const SEP = "│";
+const DOT = "·";
+const RUN_UPDATE_MS = 5000;
+/** 标题列宽（按最长 "minimax-token-plan"=18，+1 空格余量） */
+const TITLE_COL_W = 19;
+/** reset 时间列宽（fmtResetSec 最长 "12d23h"=6 + 1 空格余量） */
+const RESET_COL_W = 7;
+/** pct 列宽（"100%"=4，但 padStart(3) 给 " 23%"=4） */
+const PCT_COL_W = 3;
+/** sessionId 截取末尾字符数 */
+const SESSION_ID_TAIL = 12;
+/** 路径展示的层数（cwd 倒数 N 段） */
+const DIR_DEPTH = 2;
+/** 分/秒 pad 宽度 */
+const MIN_PAD = 2;
+/** bogus replay 阈值：output > 50 tokens 但 duration < 100ms 视为重放，跳过速度统计 */
+const BOGUS_OUTPUT_THRESHOLD = 50;
+const BOGUS_DURATION_THRESHOLD_MS = 100;
+
+// ── 阈值常量 ───────────────────────────────────────────
+
+/** token 数字单位阈值 */
+const KILO = 1_000;
+const MILLION = 1_000_000;
+
+/** pct 颜色分档 */
+const PCT_HIGH = 80;
+const PCT_MED = 60;
+const PCT_LOW = 40;
+/** 百分比标度 */
+const PERCENT_SCALE = 100;
+
+/** contextWindow fallback */
+const DEFAULT_CONTEXT_WINDOW = 128_000;
+
+// ── 工具函数 ───────────────────────────────────────────
+
 function fmtDuration(ms: number): string {
-	const s = Math.floor(ms / 1000);
-	if (s < 60) return `${s}s`;
-	const m = Math.floor(s / 60);
-	if (m < 60) return `${m}m${String(s % 60).padStart(2, "0")}s`;
-	return `${Math.floor(m / 60)}h${String(m % 60).padStart(2, "0")}m`;
+	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`;
 }
 
 function fmtTokens(n: number): string {
-	if (n >= 1_000_000) return `${(n / 1_000_000).toFixed(1)}M`;
-	if (n >= 1_000) return `${(n / 1_000).toFixed(1)}K`;
+	if (n >= MILLION) return `${(n / MILLION).toFixed(1)}M`;
+	if (n >= KILO) return `${(n / KILO).toFixed(1)}K`;
 	return `${n}`;
 }
 
 function fmtResetSec(sec: number): string {
 	if (sec <= 0) return "";
-	const d = Math.floor(sec / 86400);
-	const h = Math.floor((sec % 86400) / 3600);
-	const m = Math.floor((sec % 3600) / 60);
+	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`;
 }
 
-// ── 归一化的套餐窗口列 ─────────────────────────────────
+function fmtCount(n: number): string {
+	return n < KILO ? `${n}` : `${(n / KILO).toFixed(1)}k`;
+}
 
-interface QuotaRow {
-	name: string;
-	wins: [QuotaWindow, QuotaWindow, QuotaWindow];
+/** 按百分比返回语义色 token */
+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";
 }
 
-const COLS = [
-	{ key: "5h", label: "5h" },
-	{ key: "week", label: "wk" },
-	{ key: "month", label: "mh" },
-] as const;
+/** 把 cwd 切成段，按系统分隔符（macOS/Linux: /；Windows: \） */
+function splitPath(p: string): string[] {
+	return p.split(sep).filter(Boolean);
+}
 
-/** 将缓存数据归一化为对齐的行数组（走注册表）。 */
-function normalizeRows(cache: CacheData): QuotaRow[] {
-	const rows: QuotaRow[] = [];
-	for (const p of PROVIDERS) {
-		try {
-			const raw = (cache as Record<string, unknown>)[p.id];
-			if (!raw) continue;
-			const norm = p.normalize(raw);
-			if (!norm) continue;
-			rows.push({ name: norm.label || p.label, wins: norm.wins });
-		} catch {
-			// 单个 provider normalize 失败不影响其他 provider 显示
-		}
-	}
-	return rows;
+/** 截取 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"));
 }
 
 // ── 状态 ───────────────────────────────────────────────
 
-interface State {
+interface StatuslineRuntimeState {
 	sessionStart: number;
 	lastLlmTime: number;
 	assistantStart: number;
@@ -162,14 +160,10 @@ interface State {
 	usedPct: number;
 	contextTokens: number;
 	contextWindow: number;
-	treeTokens: number;
-	treeId: string;
 }
 
-// ── 扩展入口 ───────────────────────────────────────────
-
-export default function (pi: ExtensionAPI) {
-	const state: State = {
+function makeInitialState(): StatuslineRuntimeState {
+	return {
 		sessionStart: 0,
 		lastLlmTime: 0,
 		assistantStart: 0,
@@ -183,22 +177,28 @@ export default function (pi: ExtensionAPI) {
 		usedPct: 0,
 		contextTokens: 0,
 		contextWindow: 0,
-		treeTokens: 0,
-		treeId: "",
 	};
+}
 
+// ── 扩展入口 ───────────────────────────────────────────
+
+export default function statuslineExtension(pi: ExtensionAPI) {
+	registerSetupCommand(pi);
+	registerSessionLifecycle(pi);
+}
+
+function registerSessionLifecycle(pi: ExtensionAPI): void {
+	const state: StatuslineRuntimeState = makeInitialState();
 	let tui: { requestRender(): void } | null = null;
 
 	pi.on("session_start", async (_event: unknown, ctx: ExtensionContext) => {
-		state.sessionStart = Date.now();
-		state.lastLlmTime = 0;
-		state.speed = { current: 0, day: 0, d7: 0, d30: 0 };
-		state.isAgentBusy = false;
-		state.thinkingLevel = pi.getThinkingLevel();
+		Object.assign(state, makeInitialState(), {
+			sessionStart: Date.now(),
+			thinkingLevel: pi.getThinkingLevel(),
+		});
 		refreshTotals(state, ctx);
 
-		// eslint-disable-next-line @typescript-eslint/no-explicit-any — SDK ExtensionContext.ui 类型缺失 setFooter
-		(ctx.ui as any).setFooter((t: { requestRender(): void }, theme: Theme, footerData: ReadonlyFooterDataProvider) => {
+		ctx.ui.setFooter((t: { requestRender(): void }, theme: Theme, footerData: ReadonlyFooterDataProvider) => {
 			tui = t;
 			const unsub = footerData.onBranchChange(() => t.requestRender());
 			return {
@@ -213,49 +213,47 @@ export default function (pi: ExtensionAPI) {
 		triggerUpdate();
 	});
 
-	pi.on("message_start", async (event: PiMessageEvent) => {
+	pi.on("message_start", (event: PiMessageEvent) => {
 		if (event.message.role === "assistant") {
 			state.assistantStart = Date.now();
 			state.isAgentBusy = true;
 		}
 	});
 
-	pi.on("message_end", async (event: PiMessageEvent, ctx: ExtensionContext) => {
-		if (event.message.role === "assistant") {
-			const msg = event.message as AssistantMessage;
-			if (!msg.usage) return;
-			const dur = state.assistantStart ? Date.now() - state.assistantStart : 0;
-			state.lastLlmTime = Date.now();
-			const isBogusReplay = msg.usage.output > 50 && dur < 100;
-			if (isBogusReplay) {
-				state.speed = { current: 0, day: 0, d7: 0, d30: 0 };
-				return;
-			}
-			state.speed = trackSpeed(msg.usage.output, dur, ctx.model?.id ?? "");
-			state.totalInp += msg.usage.input;
-			state.totalOut += msg.usage.output;
-			state.totalCost += msg.usage.cost.total;
-			refreshContextUsage(state, ctx);
-			tui?.requestRender();
-			triggerUpdate();
+	pi.on("message_end", (event: PiMessageEvent, ctx: ExtensionContext) => {
+		if (event.message.role !== "assistant") return;
+		const msg = event.message as AssistantMessage;
+		if (!msg.usage) return;
+		const dur = state.assistantStart ? Date.now() - state.assistantStart : 0;
+		state.lastLlmTime = Date.now();
+		if (msg.usage.output > BOGUS_OUTPUT_THRESHOLD && dur < BOGUS_DURATION_THRESHOLD_MS) {
+			state.speed = { current: 0, day: 0, d7: 0, d30: 0 };
+			return;
 		}
+		state.speed = trackSpeed(msg.usage.output, dur, ctx.model?.id ?? "");
+		state.totalInp += msg.usage.input;
+		state.totalOut += msg.usage.output;
+		state.totalCost += msg.usage.cost.total;
+		refreshContextUsage(state, ctx);
+		tui?.requestRender();
+		triggerUpdate();
 	});
 
-	pi.on("turn_end", async () => {
+	pi.on("turn_end", () => {
 		state.isAgentBusy = false;
 		state.lastRunUpdate = Date.now();
 		tui?.requestRender();
 	});
-	pi.on("agent_end", async () => {
+	pi.on("agent_end", () => {
 		state.isAgentBusy = false;
 		state.lastRunUpdate = Date.now();
 		tui?.requestRender();
 	});
-	pi.on("model_select", async () => {
+	pi.on("model_select", () => {
 		state.thinkingLevel = pi.getThinkingLevel();
 		tui?.requestRender();
 	});
-	pi.on("thinking_level_select", async (event: PiThinkingLevelEvent) => {
+	pi.on("thinking_level_select", (event: PiThinkingLevelEvent) => {
 		state.thinkingLevel = event.level;
 		if (!state.isAgentBusy) tui?.requestRender();
 	});
@@ -263,7 +261,7 @@ export default function (pi: ExtensionAPI) {
 
 // ── 数据刷新 ───────────────────────────────────────────
 
-function refreshTotals(st: State, ctx: ExtensionContext): void {
+function refreshTotals(st: StatuslineRuntimeState, ctx: ExtensionContext): void {
 	let inp = 0, out = 0, cost = 0;
 	for (const e of ctx.sessionManager.getBranch()) {
 		if (e.type === "message" && e.message.role === "assistant") {
@@ -280,187 +278,208 @@ function refreshTotals(st: State, ctx: ExtensionContext): void {
 	refreshContextUsage(st, ctx);
 }
 
-function refreshContextUsage(st: State, ctx: ExtensionContext): void {
+function refreshContextUsage(st: StatuslineRuntimeState, ctx: ExtensionContext): void {
 	const usage = ctx.getContextUsage();
 	if (!usage || usage.tokens === null) return;
-	const contextWindow = usage.contextWindow || 128_000;
+	const contextWindow = usage.contextWindow || DEFAULT_CONTEXT_WINDOW;
 	st.contextTokens = usage.tokens;
 	st.contextWindow = contextWindow;
-	st.usedPct = Math.min(Math.round((usage.tokens / contextWindow) * 100), 100);
-	refreshTreeTokens(st, ctx);
+	st.usedPct = Math.min(Math.round((usage.tokens / contextWindow) * PERCENT_SCALE), PERCENT_SCALE);
 }
 
-/** 从 session entries 中读取最新 ic-compact-tree 的 totalTokens */
-function refreshTreeTokens(st: State, ctx: ExtensionContext): void {
-	let latestTokens: number | undefined;
-	let latestTreeId: string | undefined;
-	for (const e of ctx.sessionManager.getEntries()) {
-		if (e.type === "custom" && (e as { customType: string }).customType === "ic-compact-tree") {
-			const data = (e as { data?: { totalTokens?: number; treeId?: string } }).data;
-			if (data?.totalTokens != null) latestTokens = data.totalTokens;
-			if (data?.treeId != null) latestTreeId = data.treeId;
-		}
-	}
-	st.treeTokens = latestTokens ?? 0;
-	st.treeId = latestTreeId ?? "";
+// ── 渲染 ───────────────────────────────────────────────
+
+interface QuotaRow {
+	name: string;
+	wins: [QuotaWindow, QuotaWindow, QuotaWindow];
 }
 
-// ── 渲染 ───────────────────────────────────────────────
+const COLS = [
+	{ key: "5h", label: "5h" },
+	{ key: "week", label: "wk" },
+	{ key: "month", label: "mh" },
+] as const;
 
-function buildLines(
-	ctx: ExtensionContext,
-	theme: Theme,
-	fd: ReadonlyFooterDataProvider,
-	width: number,
-	st: State,
-): string[] {
-	const cache = readCache();
-	const wide = width >= WIDE_THRESHOLD;
+type Pallet = {
+	d: (s: string) => string;
+	v: (s: string) => string;
+	g: (s: string) => string;
+	w: (s: string) => string;
+	a: (s: string) => string;
+	m: (s: string) => string;
+};
 
+function makePalette(theme: Theme): Pallet {
 	const fg = (c: string, t: string) => theme.fg(c, t);
-	const d = (s: string) => fg("dim", s);
-	const v = (s: string) => fg("text", s);
-	const g = (s: string) => fg("success", s);
-	const w = (s: string) => fg("warning", s);
-	const a = (s: string) => fg("accent", s);
-	const m = (s: string) => fg("muted", s);
-
-	const lines: string[] = [];
-
-	// ═══════════════════════════════════════════════════
-	// Line 1: 目录/仓库 · 分支 │ session-name │ provider : model [thinking]
-	// ═══════════════════════════════════════════════════
-	const branch = fd.getGitBranch();
-	const cwd = ctx.cwd || "";
+	return {
+		d: (s) => fg("dim", s),
+		v: (s) => fg("text", s),
+		g: (s) => fg("success", s),
+		w: (s) => fg("warning", s),
+		a: (s) => fg("accent", s),
+		m: (s) => fg("muted", s),
+	};
+}
 
-	const idParts: string[] = [];
-	if (branch) {
-		const segs = cwd.split("/").filter(Boolean);
-		const repoName = segs.slice(-2).join("/");
-		if (repoName) idParts.push(a(repoName));
-		idParts.push(`⎇ ${g(branch)}`);
-	} else {
-		const segs = cwd.split("/").filter(Boolean);
-		const last2 = segs.slice(-2).join("/");
-		if (last2) idParts.push(a(last2));
+/** 缓存数据 → 归一化行（用于 token-plans 显示） */
+function normalizeRows(cache: CacheData, providers: QuotaProvider[]): QuotaRow[] {
+	const rows: QuotaRow[] = [];
+	for (const p of providers) {
+		if (p.category !== "token-plan") continue;
+		try {
+			const raw = (cache as Record<string, unknown>)[p.id];
+			if (!raw) continue;
+			const norm = p.normalize(raw);
+			if (!norm) continue;
+			rows.push({ name: norm.label || p.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);
+		}
 	}
+	return rows;
+}
 
-	let line1 = idParts.join(` ${DOT} `);
+// ── 5 个独立行渲染函数 ─────────────────────────────────
 
-	// Session name (set by /name command)
-	const sessionName = ctx.sessionManager.getSessionName();
-	if (sessionName) {
-		line1 += ` ${SEP} ${a(sessionName)}`;
-	}
+function buildLine1(ctx: ExtensionContext, fd: ReadonlyFooterDataProvider, p: Pallet): string {
+	const branch = fd.getGitBranch();
+	const cwd = ctx.cwd || "";
+	const segs = splitPath(cwd);
+	const dirLabel = segs.slice(-DIR_DEPTH).join(sep) || cwd;
+	const inWt = isWorktree(cwd);
+
+	const parts: string[] = [p.a(dirLabel)];
+	if (branch) parts.push(`⎇ ${p.g(branch)}`);
+	if (inWt) parts.push(p.d("worktree"));
+	return parts.join(` ${DOT} `);
+}
 
+function buildLine2(ctx: ExtensionContext, st: StatuslineRuntimeState, p: Pallet): string {
 	const model = ctx.model;
-	if (model) {
-		const provider = model.provider || "";
-		const provShort = provider.includes("/")
-			? provider.split("/").pop()!
-			: provider;
-		const modelId = model.id || model.name || "unknown";
-		const tlPart = st.thinkingLevel ? ` ${m(`[${st.thinkingLevel}]`)}` : "";
-		line1 += ` ${SEP} ${d(provShort)} : ${a(modelId)}${tlPart}`;
-	}
-	if (line1) lines.push(line1);
-
-	// ═══════════════════════════════════════════════════
-	// Line 2: ctx │ speed current+t/s day+t/s │ tavily
-	// ═══════════════════════════════════════════════════
-	const ctxSizeStr =
-		st.contextWindow > 0
-			? `${d("ctx")} ${v(fmtTokens(st.contextTokens))}/${v(fmtTokens(st.contextWindow))}`
-			: `${d("ctx")} ${v(`${st.usedPct}%`)}`;
-	const ctxBarStr = wide
-		? `${barSegment(st.usedPct, theme)} ${v(`${st.usedPct}%`)}`
-		: `${v(`${st.usedPct}%`)}`;
-
-	const line2Parts: string[] = [`${ctxSizeStr} ${ctxBarStr}`];
-
-	// tree-ctx：格式和 ctx 相同，始终展示
-	if (st.contextWindow > 0) {
-		const treePctRaw = (st.treeTokens / st.contextWindow) * 100;
-		const treePct = Math.min(Math.round(treePctRaw), 100);
-		const treeDisplayPct = treePct === 0 && st.treeTokens > 0 ? "<1" : `${treePct}`;
-		const treeSizeStr = `${d("tree")} ${v(fmtTokens(st.treeTokens))}/${v(fmtTokens(st.contextWindow))}`;
-		const treeBarStr = wide
-			? `${barSegment(treePct || 1, theme)} ${v(`${treeDisplayPct}%`)}`
-			: `${v(`${treeDisplayPct}%`)}`;
-		line2Parts.push(`${treeSizeStr} ${treeBarStr}`);
-	}
+	if (!model) return "";
+	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 sp: string[] = [];
-	if (st.speed.current > 0)
-		sp.push(`${g(`${st.speed.current}`)}${d("t/s")}`);
-	if (st.speed.day > 0)
-		sp.push(`${d("day")} ${g(`${st.speed.day}`)}${d("t/s")}`);
-	if (sp.length)
-		line2Parts.push(`${d("speed")} ${sp.join(` ${DOT} `)}`);
-
-	const tv = cache["tavily"] as { available: number; total: number } | undefined;
-	if (tv)
-		line2Parts.push(`${d("tavily")} ${g(`${tv.available}`)}/${v(`${tv.total}`)}`);
-
-	lines.push(line2Parts.join(` ${SEP} `));
-
-	// ═══════════════════════════════════════════════════
-	// Line 3+: 套餐用量（归一化 → 统一列渲染）
-	// ═══════════════════════════════════════════════════
-	const rows = normalizeRows(cache);
-	for (const row of rows) {
-		const title = d(row.name.padEnd(TITLE_COL_W));
-		const cells = COLS.map((col, i) => {
-			const win = row.wins[i]!;
-			return winCol(col.label, win.pct, win.resetSec, wide, d, v, theme);
-		});
-		lines.push(title + cells.join(` ${DOT} `));
-	}
+function buildLine3(
+	ctx: ExtensionContext,
+	st: StatuslineRuntimeState,
+	p: Pallet,
+	theme: Theme,
+): string {
+	const ctxPct = st.usedPct;
+	const ctxPctCol = theme.fg(pctColor(ctxPct), `${ctxPct}%`);
+	const ctxStr = st.contextWindow > 0
+		? `${p.d("ctx")} ${p.v(fmtTokens(st.contextTokens))}/${p.v(fmtTokens(st.contextWindow))} ${ctxPctCol}`
+		: `${p.d("ctx")} ${ctxPctCol}`;
 
-	// ═══════════════════════════════════════════════════
-	// 末行: 时间 · 费用 · 会话ID
-	// ═══════════════════════════════════════════════════
 	const tp: string[] = [];
 	if (st.sessionStart) {
 		const from = new Date(st.sessionStart);
-		tp.push(
-			`${d("from")} ${g(`${from.getHours()}:${String(from.getMinutes()).padStart(2, "0")}`)}`,
-		);
+		tp.push(`${p.d("from")} ${p.g(`${from.getHours()}:${String(from.getMinutes()).padStart(MIN_PAD, "0")}`)}`);
 	}
-	const shouldRefreshRun =
-		!st.isAgentBusy &&
-		(st.lastRunUpdate === 0 || Date.now() - st.lastRunUpdate >= RUN_UPDATE_MS);
-	if (shouldRefreshRun) st.lastRunUpdate = Date.now();
-	const displayRunMs = st.lastRunUpdate
-		? st.lastRunUpdate - st.sessionStart
-		: st.sessionStart ? Date.now() - st.sessionStart : 0;
-	if (displayRunMs > 0) tp.push(`${d("run")} ${g(fmtDuration(displayRunMs))}`);
+	if (shouldRefreshRun(st)) st.lastRunUpdate = Date.now();
+	const displayRunMs = computeRunMs(st);
+	if (displayRunMs > 0) tp.push(`${p.d("run")} ${p.g(fmtDuration(displayRunMs))}`);
 	if (st.lastLlmTime) {
-		const ago = Math.floor((Date.now() - st.lastLlmTime) / 1000);
-		tp.push(
-			`${d("last")} ${w(ago < 60 ? `${ago}s` : `${Math.floor(ago / 60)}m${ago % 60}s`)}`,
-		);
+		const ago = Math.floor((Date.now() - st.lastLlmTime) / MS_PER_SEC);
+		tp.push(`${p.d("last")} ${p.w(ago < SEC_PER_MIN ? `${ago}s` : `${Math.floor(ago / SEC_PER_MIN)}m${ago % SEC_PER_MIN}s`)}`);
 	}
 
-	const sid =
-		ctx.sessionManager
-			.getSessionFile()
-			?.split("/")
-			.pop()
-			?.slice(-12) || "";
+	const sid = tailSessionId(ctx.sessionManager.getSessionFile(), SESSION_ID_TAIL);
 
-	const info: string[] = [];
-	if (tp.length) info.push(tp.join(` ${DOT} `));
-	if (st.totalCost > 0) info.push(`${d("cost")} ${w(`$${st.totalCost.toFixed(3)}`)}`);
+	const parts: string[] = [ctxStr];
+	if (tp.length) parts.push(tp.join(` ${DOT} `));
 	if (st.totalInp > 0 || st.totalOut > 0) {
-		const fmt = (n: number) => (n < 1000 ? `${n}` : `${(n / 1000).toFixed(1)}k`);
-		info.push(`${d("↑↓")} ${v(fmt(st.totalInp))}/${v(fmt(st.totalOut))}`);
+		parts.push(`${p.d("↑↓")} ${p.v(fmtCount(st.totalInp))}/${p.v(fmtCount(st.totalOut))}`);
 	}
-	const treeSid = st.treeId ? st.treeId.replace(/^tree_/, "").slice(-8) : "";
-	if (treeSid) info.push(`${d("tree")} ${m(treeSid)}`);
-	if (sid) info.push(m(sid));
+	if (sid) parts.push(p.m(sid));
+	return parts.join(` ${SEP} `);
+}
+
+function shouldRefreshRun(st: StatuslineRuntimeState): boolean {
+	return !st.isAgentBusy && (st.lastRunUpdate === 0 || Date.now() - st.lastRunUpdate >= RUN_UPDATE_MS);
+}
+
+function computeRunMs(st: StatuslineRuntimeState): number {
+	if (st.lastRunUpdate) return st.lastRunUpdate - st.sessionStart;
+	if (st.sessionStart) return Date.now() - st.sessionStart;
+	return 0;
+}
+
+function buildSearchLine(
+	cache: CacheData,
+	providers: QuotaProvider[],
+	p: Pallet,
+	theme: Theme,
+): string {
+	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;
+		if (!raw) continue;
+		const used = raw.used ?? raw.available;
+		const total = raw.total;
+		if (used === undefined || !total || total <= 0) continue;
+		const pct = Math.round((used / total) * PERCENT_SCALE);
+		const pctCol = theme.fg(pctColor(pct), `${pct}%`);
+		parts.push(`${p.d(prov.label)} ${p.g(`${used}`)}/${p.v(`${total}`)}${p.d("次")} ${pctCol}`);
+	}
+	return parts.join(" | ");
+}
 
-	if (info.length) lines.push(info.join(` ${SEP} `));
+function buildTokenPlanLines(
+	cache: CacheData,
+	providers: QuotaProvider[],
+	p: Pallet,
+	theme: Theme,
+): 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, theme);
+		});
+		return title + cells.join(` ${DOT} `);
+	});
+}
 
-	return lines.map((l) => truncateToWidth(l, width));
+/** 渲染单个窗口列：label pct% [reset]（无 bar） */
+function formatWinCol(label: string, win: QuotaWindow, p: Pallet, theme: Theme): string {
+	if (win.pct === null) {
+		return `${p.d(label)}  ${p.v("∞")}`;
+	}
+	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)}  ${theme.fg(pctColor(win.pct), pctStr)}  ${rtStr}`;
+}
+
+function buildLines(
+	ctx: ExtensionContext,
+	theme: Theme,
+	fd: ReadonlyFooterDataProvider,
+	width: number,
+	st: StatuslineRuntimeState,
+): string[] {
+	const cache = readCache();
+	const providers = buildRuntimeProviders();
+	const palette = makePalette(theme);
+
+	const lines: string[] = [
+		buildLine1(ctx, fd, palette),
+		buildLine2(ctx, st, palette),
+		buildLine3(ctx, st, palette, theme),
+		buildSearchLine(cache, providers, palette, theme),
+		...buildTokenPlanLines(cache, providers, palette, theme),
+	];
+
+	// 过滤空行（line2/line3 在某些状态下可能空，line4 没搜索工具时空）
+	return lines
+		.filter((l) => l.length > 0)
+		.map((l) => truncateToWidth(l, width));
 }

package/src/setup-prompts.ts

@@ -0,0 +1,146 @@
+/**
+ * setup-statusline 命令的 i18n prompt 模板
+ *
+ * 中英文切换：基于 Intl.DateTimeFormat().resolvedOptions().locale
+ * 失败/非 zh locale → 英文
+ *
+ * demo 模板：providers.json 的结构由 quota-providers.PROVIDERS 动态生成
+ *   （新增 provider 自动出现，无需改本文件）
+ * secrets.json 的 env var 映射在下面 DEFAULT_SECRETS 集中维护
+ *   （用户编辑后会自动覆盖默认值）
+ */
+
+import { PROVIDERS } from "@zhushanwen/pi-quota-providers";
+
+export type Locale = "zh" | "en";
+
+export function detectLocale(): Locale {
+	try {
+		const locale = Intl.DateTimeFormat().resolvedOptions().locale;
+		return /^zh/i.test(locale) ? "zh" : "en";
+	} catch {
+		return "en";
+	}
+}
+
+export interface SetupPromptArgs {
+	configDir: string;
+	providersPath: string;
+	secretsPath: string;
+	missing: string[];
+}
+
+/** provider.id → 默认 secret 字段（key=value 形式，value 是 env var 占位符） */
+const DEFAULT_SECRETS: Record<string, Record<string, string>> = {
+	zhipu: { token: "${ZAI_AUTH_TOKEN}" },
+	"opencode-go": { token: "${OPENCODE_GO_TOKEN}" },
+	"kimi-coding": { token: "${KIMI_AUTH_TOKEN}" },
+	minimax: { token: "${MINIMAX_TOKEN}" },
+	tavily: { apiKey: "${TAVILY_API_KEY}" },
+};
+
+/** 从 PROVIDERS 动态生成 providers.json demo（自动跟随新增 provider） */
+function buildDemoProvidersJson(): string {
+	// 按 category 分组
+	const groups: Record<string, string[]> = { "token-plans": [], "search-tools": [] };
+	for (const p of PROVIDERS) {
+		const key = p.category === "search-tool" ? "search-tools" : "token-plans";
+		groups[key]!.push(JSON.stringify({
+			id: p.id,
+			label: p.label,
+			enabled: true,
+			fetcher: p.id,
+		}));
+	}
+
+	const lines: string[] = ["{"];
+	const sections: string[] = [];
+	for (const [key, items] of Object.entries(groups)) {
+		if (items.length === 0) continue;
+		sections.push(`  "${key}": [\n${items.map((s) => `    ${s}`).join(",\n")}\n  ]`);
+	}
+	lines.push(sections.join(",\n"));
+	lines.push("}");
+	return lines.join("\n");
+}
+
+/** 从 DEFAULT_SECRETS 动态生成 secrets.json demo（仅含 PROVIDERS 实际支持的） */
+function buildDemoSecretsJson(): string {
+	const lines: string[] = ["{"];
+	const sections: string[] = [];
+	for (const p of PROVIDERS) {
+		const fields = DEFAULT_SECRETS[p.id];
+		if (!fields) continue;
+		const fieldStr = Object.entries(fields)
+			.map(([k, v]) => `    "${k}": "${v}"`)
+			.join(",\n");
+		sections.push(`  "${p.id}": {\n${fieldStr}\n  }`);
+	}
+	lines.push(sections.join(",\n"));
+	lines.push("}");
+	return lines.join("\n");
+}
+
+const T = {
+	zh: {
+		title: "# statusline 配置初始化",
+		missing: (m: string[]) => `缺失文件：${m.join(", ")}`,
+		task: "请帮我创建 demo 配置文件。",
+		constraint1: "secrets.json 凭证请用 \\${ENV_VAR} 环境变量引用形式",
+		constraint2: "providers.json 默认启用所有 provider（用户可后续编辑禁用）",
+		constraint3: "创建完成后告诉用户可以编辑这两个文件",
+		pathsHeader: "## 写入路径",
+		providersHeader: "## providers.json demo",
+		secretsHeader: "secrets.json demo",
+		completeHeader: "## 完成后",
+		completeTask: "读两个文件，输出最终内容给用户确认。",
+	},
+	en: {
+		title: "# statusline setup",
+		missing: (m: string[]) => `Missing files: ${m.join(", ")}`,
+		task: "Generate demo config files for me.",
+		constraint1: "Use \\${ENV_VAR} format in secrets.json for credentials",
+		constraint2: "Enable all providers in providers.json by default (user can disable later)",
+		constraint3: "After creating, tell the user they can edit these files",
+		pathsHeader: "## Target paths",
+		providersHeader: "## providers.json demo",
+		secretsHeader: "## secrets.json demo",
+		completeHeader: "## After writing",
+		completeTask: "Read both files and show the final content to the user for confirmation.",
+	},
+} as const;
+
+/** 生成让 LLM 写 demo 文件的引导 prompt */
+export function buildGenerateDemoPrompt(args: SetupPromptArgs): string {
+	const t = T[detectLocale()];
+	return [
+		t.title,
+		"",
+		t.missing(args.missing),
+		"",
+		"## 任务",
+		t.task,
+		"",
+		"## 重要约束",
+		`1. ${t.constraint1}`,
+		`2. ${t.constraint2}`,
+		`3. ${t.constraint3}`,
+		"",
+		t.pathsHeader,
+		`- providers.json: \`${args.providersPath}\``,
+		`- secrets.json: \`${args.secretsPath}\``,
+		"",
+		t.providersHeader,
+		"```json",
+		buildDemoProvidersJson(),
+		"```",
+		"",
+		t.secretsHeader,
+		"```json",
+		buildDemoSecretsJson(),
+		"```",
+		"",
+		t.completeHeader,
+		t.completeTask,
+	].join("\n");
+}

package/src/setup.ts

@@ -0,0 +1,102 @@
+/**
+ * setup-statusline 命令
+ *
+ * 行为：
+ * 1. 检查 ~/.pi/agent/config/{providers,secrets}.json 是否存在
+ * 2. 都存在 → 加载并显示审查摘要
+ * 3. 缺失 → 注入 prompt 让 LLM 生成 demo 文件
+ *
+ * 故意不做：交互式 wizard（让 LLM 处理）、chmod 校验、自动迁移老路径
+ */
+
+import { existsSync, mkdirSync } from "node:fs";
+import type { ExtensionAPI } from "@mariozechner/pi-coding-agent";
+import type { ExtensionCommandContext } from "@mariozechner/pi-coding-agent";
+import {
+	getConfigDir,
+	getProvidersConfigPath,
+	getSecretsPath,
+	loadProvidersConfig,
+	loadSecrets,
+} from "@zhushanwen/pi-quota-providers";
+import { buildGenerateDemoPrompt } from "./setup-prompts.js";
+
+export function registerSetupCommand(pi: ExtensionAPI): void {
+	pi.registerCommand("setup-statusline", {
+		description: "生成 statusline 的 providers.json + secrets.json demo（LLM 引导）",
+		handler: async (_args: string, ctx: ExtensionCommandContext) => {
+			const configDir = getConfigDir();
+			const providersPath = getProvidersConfigPath();
+			const secretsPath = getSecretsPath();
+
+			try {
+				mkdirSync(configDir, { recursive: true });
+			} catch (e) {
+				ctx.ui.notify(`Failed to create ${configDir}: ${(e as Error).message}`, "error");
+				return;
+			}
+
+			const hasProviders = existsSync(providersPath);
+			const hasSecrets = existsSync(secretsPath);
+
+			// 都存在 → 审查模式
+			if (hasProviders && hasSecrets) {
+				const cfg = loadProvidersConfig();
+				const secrets = loadSecrets();
+				ctx.ui.notify(formatReviewSummary(cfg, secrets), "info");
+				return;
+			}
+
+			// 缺失 → 让 LLM 写 demo
+			const missing: string[] = [];
+			if (!hasProviders) missing.push("providers.json");
+			if (!hasSecrets) missing.push("secrets.json");
+
+			const prompt = buildGenerateDemoPrompt({
+				configDir,
+				providersPath,
+				secretsPath,
+				missing,
+			});
+
+			try {
+				await ctx.sessionManager.appendEntry("user", prompt);
+			} catch (e) {
+				ctx.ui.notify(`Failed to inject setup prompt: ${(e as Error).message}`, "error");
+				return;
+			}
+			ctx.ui.notify(`Setup wizard started. Will generate: ${missing.join(", ")}`, "info");
+		},
+	});
+}
+
+function formatReviewSummary(
+	cfg: ReturnType<typeof loadProvidersConfig>,
+	secrets: ReturnType<typeof loadSecrets>,
+): string {
+	const lines: string[] = ["=== statusline config ==="];
+
+	lines.push("token-plans:");
+	if (cfg["token-plans"].length === 0) {
+		lines.push("  (none)");
+	} else {
+		for (const p of cfg["token-plans"]) {
+			const mark = p.enabled ? "✓" : "✗";
+			const tag = secrets[p.id] ? "[secret ok]" : "[no secret]";
+			lines.push(`  ${mark} ${p.label} (${p.id}) ${tag}`);
+		}
+	}
+
+	lines.push("search-tools:");
+	if (cfg["search-tools"].length === 0) {
+		lines.push("  (none)");
+	} else {
+		for (const p of cfg["search-tools"]) {
+			const mark = p.enabled ? "✓" : "✗";
+			const tag = secrets[p.id] ? "[secret ok]" : "[no secret]";
+			lines.push(`  ${mark} ${p.label} (${p.id}) ${tag}`);
+		}
+	}
+
+	return lines.join("\n");
+}