@buyi1net/pi-toolkit 0.0.1

package/modules/tui/adapter/provider-usage.ts

@@ -0,0 +1,143 @@
+import type { ExtensionContext } from "@earendil-works/pi-coding-agent";
+import { homedir } from "node:os";
+import { join } from "node:path";
+import type {
+	ProviderAccess,
+	ProviderAccessOptions,
+	ProviderQueryAccess,
+} from "../kernel/usage-core.ts";
+import {
+	createProviderAccess,
+	FileUsageSnapshotCache,
+	findProviderQueryConfig,
+	UsageRuntime,
+	type UsageRuntimeState,
+} from "../kernel/usage-node.ts";
+
+type PiModel = NonNullable<ExtensionContext["model"]>;
+type AccessResolver = (
+	ctx: ExtensionContext,
+	model: PiModel | undefined,
+	config?: PiProviderAccessConfig,
+) => Promise<ProviderAccess | null>;
+
+export interface PiProviderAccessConfig {
+	queries?: readonly ProviderQueryAccess[];
+	credentials?: ProviderAccessOptions["credentials"];
+	githubDomain?: string;
+}
+
+const OPENAI_AUTH_CLAIM = "https://api.openai.com/auth";
+
+function extractCodexAccountId(providerId: string, credential: string): string | undefined {
+	if (providerId !== "openai-codex") return undefined;
+	const payload = credential.split(".")[1];
+	if (!payload) return undefined;
+	try {
+		const claims: unknown = JSON.parse(Buffer.from(payload, "base64url").toString("utf8"));
+		if (!claims || typeof claims !== "object") return undefined;
+		const auth = (claims as Record<string, unknown>)[OPENAI_AUTH_CLAIM];
+		if (!auth || typeof auth !== "object") return undefined;
+		const accountId = (auth as Record<string, unknown>).chatgpt_account_id;
+		return typeof accountId === "string" && accountId.trim() ? accountId : undefined;
+	} catch {
+		return undefined;
+	}
+}
+
+export async function resolvePiProviderAccess(
+	ctx: ExtensionContext,
+	model: PiModel | undefined = ctx.model,
+	config: PiProviderAccessConfig = {},
+): Promise<ProviderAccess | null> {
+	if (!model) return null;
+	try {
+		const auth = await ctx.modelRegistry.getApiKeyAndHeaders(model);
+		if (!auth.ok) return null;
+		const credential = auth.apiKey ?? "";
+		const endpoint = auth.baseUrl ?? model.baseUrl;
+		return createProviderAccess({
+			providerId: model.provider,
+			modelId: model.id,
+			endpoint,
+			credential,
+			authKind: ctx.modelRegistry.isUsingOAuth(model) ? "oauth" : "api-key",
+			accountId: extractCodexAccountId(model.provider, credential),
+			githubDomain: config.githubDomain,
+			query: findProviderQueryConfig(endpoint, config.queries ?? []),
+			credentials: config.credentials,
+		});
+	} catch {
+		return null;
+	}
+}
+
+export class PiProviderUsageController {
+	private readonly ctx: ExtensionContext;
+	private readonly runtime: UsageRuntime;
+	private readonly resolveAccess: AccessResolver;
+	private readonly accessConfig: PiProviderAccessConfig;
+	private readonly refreshMs: number;
+	private interval: NodeJS.Timeout | undefined;
+	private activeModelKey = "";
+	private refreshEpoch = 0;
+	private disposed = false;
+
+	constructor(
+		ctx: ExtensionContext,
+		onChange: (state: UsageRuntimeState) => void,
+		options: {
+			refreshMs?: number;
+			resolveAccess?: AccessResolver;
+			accessConfig?: PiProviderAccessConfig;
+			runtime?: UsageRuntime;
+		} = {},
+	) {
+		this.ctx = ctx;
+		this.refreshMs = options.refreshMs ?? 60_000;
+		this.resolveAccess = options.resolveAccess ?? resolvePiProviderAccess;
+		this.accessConfig = options.accessConfig ?? {};
+		this.runtime = options.runtime ?? new UsageRuntime({
+			userAgent: "pi-tui",
+			onChange,
+			cache: new FileUsageSnapshotCache(join(homedir(), ".pi", "agent", "cache", "pi-tui", "usage")),
+		});
+	}
+
+	getState(): UsageRuntimeState {
+		return this.runtime.getState();
+	}
+
+	start(): Promise<void> {
+		if (this.disposed || this.interval) return Promise.resolve();
+		const initialRefresh = this.refresh();
+		this.interval = setInterval(() => void this.refresh(), this.refreshMs);
+		this.interval.unref?.();
+		return initialRefresh;
+	}
+
+	async refresh(model: PiModel | undefined = this.ctx.model): Promise<void> {
+		if (this.disposed) return;
+		const epoch = ++this.refreshEpoch;
+		const modelKey = model ? `${model.provider}\0${model.id}\0${model.baseUrl}` : "";
+		if (modelKey !== this.activeModelKey) {
+			this.activeModelKey = modelKey;
+			// 凭据是异步解析的：身份一变先交运行层隔离旧供应商数据，解析期间不显示旧快照
+			this.runtime.begin(model
+				? { providerId: model.provider, modelId: model.id, endpoint: model.baseUrl }
+				: null);
+		}
+		const access = await this.resolveAccess(this.ctx, model, this.accessConfig);
+		// 解析期间有更新的 refresh 进场（换模型或换账号）：本次结果已过期，由最新一次 refresh 提交
+		if (this.disposed || epoch !== this.refreshEpoch) return;
+		await this.runtime.commit(access);
+	}
+
+	dispose(): void {
+		if (this.disposed) return;
+		this.disposed = true;
+		if (this.interval) clearInterval(this.interval);
+		this.interval = undefined;
+		this.runtime.dispose();
+	}
+}

package/modules/tui/config.ts

@@ -0,0 +1,50 @@
+// pi-tui 模块的配置落点：pi-toolkit.json 的 `modules.tui` 节。
+//
+// 节结构（与迁入前的 pi-tui.json 外观语义一致，只是换了宿主文件）：
+//   modules.tui.appearance.editor / header / footer
+//   modules.tui.status.preset / segments
+//   modules.tui.data.providerRefreshMs / telemetry
+//   modules.tui.advanced.spinner
+// 供应商查询凭据不在这里：按用户裁决留在独立文件 <agentDir>/pi-tui.json，
+// 本模块只读（见 plugin/settings-config.ts），toolkit 永不写它。
+//
+// 写盘走骨架的 saveToolkitConfig：深合并、剥敏感键、0600 原子替换。
+
+import { getAgentDir } from "@earendil-works/pi-coding-agent";
+import { getToolkitConfigPath, saveToolkitConfig } from "../../config.ts";
+import type { ModuleConfigRecord } from "../../module.ts";
+import {
+	parseTuiSection,
+	type LoadedPiTuiConfig,
+	type PiTuiConfig,
+} from "./plugin/settings-config.ts";
+
+/** 模块 id：同时是配置节名与菜单项 id 前缀 */
+export const TUI_MODULE_ID = "tui";
+export const TUI_SECTION_LABEL = `modules.${TUI_MODULE_ID}`;
+
+export type TuiSectionUpdate = {
+	appearance?: Partial<PiTuiConfig["appearance"]>;
+	status?: Partial<PiTuiConfig["status"]>;
+	data?: Partial<Pick<PiTuiConfig["data"], "providerRefreshMs" | "telemetry">>;
+	advanced?: Partial<PiTuiConfig["advanced"]>;
+};
+
+/**
+ * 读 `modules.tui` 节并规范化。非法取值逐条记 warning 并回退默认值，
+ * 菜单展示与 lifecycle 的实际安装都用这一份结果。
+ */
+export function readTuiSection(section: Record<string, unknown> | undefined): LoadedPiTuiConfig {
+	return parseTuiSection(section);
+}
+
+/** 写本节点（深合并，不动其它模块节与语言设置）。 */
+export async function saveTuiSection(
+	update: TuiSectionUpdate,
+	options: { readonly agentDir?: string } = {},
+): Promise<void> {
+	const path = getToolkitConfigPath(options.agentDir ?? getAgentDir());
+	await saveToolkitConfig(path, {
+		modules: { [TUI_MODULE_ID]: update as ModuleConfigRecord },
+	});
+}

package/modules/tui/index.ts

@@ -0,0 +1,120 @@
+// pi-tui 模块：把 pi-tui 的 pi 宿主侧（编辑器边框、Header、Footer、转场、
+// 遥测、状态数据源、主题、配置）装进 pi-toolkit。
+//
+// 迁入原则是搬移优先于重写：`plugin/`、`renderer/`、`status/`、`adapter/` 是原
+// pi-src 的实现（只把对 `packages/usage-*` 的 import 改走 `kernel/` 转发），
+// `themes/` 是随包注册的主题 JSON。`claude-src/` 与 `packages/` 共享内核留在原仓，
+// 继续支撑 claude-line 独立发布线（规格说明决策 2）。
+//
+// 本文件只做装配：
+//   - 15 个事件钩子、常驻 UI（编辑器 / Header / Footer / 工作指示动画）由
+//     registerPiTuiLifecycle 按配置安装；enabled=false 时装配器根本不调用 register
+//   - 外观 / 状态 / 数据 / 高级四个子节读写 pi-toolkit.json 的 `modules.tui` 节
+//   - 供应商查询凭据只读独立文件 <agentDir>/pi-tui.json（用户裁决），永不写回
+//   - 服务句柄 `ui.provider-usage`：供其它模块查询供应商余额/套餐运行态（本单只注册）
+
+import { getAgentDir } from "@earendil-works/pi-coding-agent";
+import { enabledField, type ModuleContext, type ModuleDefinition } from "../../module.ts";
+import { TUI_MODULE_ID, readTuiSection } from "./config.ts";
+import type { UsageRuntimeState } from "./kernel/usage-node.ts";
+import { buildTuiMenuItems, TUI_MENU_ITEM_ID, type TuiMenuRuntime } from "./menu.ts";
+import { registerPiTuiLifecycle, type PiTuiLifecycleHandle } from "./plugin/lifecycle.ts";import type { VisibleScreenOutput } from "./plugin/screen-transition.ts";
+import { loadProviderAccessFile, type LoadedPiTuiConfig } from "./plugin/settings-config.ts";
+
+export { TUI_MODULE_ID } from "./config.ts";
+export { TUI_MENU_ITEM_ID, TuiPanel } from "./menu.ts";
+
+/** 服务句柄名（注册表要求全小写） */
+export const TUI_PROVIDER_SERVICE_NAME = "ui.provider-usage";
+
+/**
+ * 供应商余额/套餐查询服务句柄：与编辑器状态栏同一份运行态数据源。
+ * 本单只注册不做联动；常驻 UI 未安装（enabled=false 或非 tui 会话）时
+ * snapshot() 返回 undefined，refresh() 是空操作。
+ */
+export interface TuiProviderUsageService {
+	readonly id: "tui";
+	snapshot(): UsageRuntimeState | undefined;
+	refresh(): Promise<void>;
+}
+
+export interface TuiModuleOptions {
+	/** pi 的配置目录（pi-toolkit.json 与只读的 pi-tui.json 都在它下面）；默认 getAgentDir() */
+	readonly agentDir?: string;
+	/** 终端的可见屏幕输出；默认 process.stdout */
+	readonly output?: VisibleScreenOutput;
+}
+
+/** 自测钩子：最近一次 register 交回的 lifecycle 把手（生产代码不读） */
+export const __test__: { handle?: PiTuiLifecycleHandle } = {};
+
+/** 本模块认识的配置 = modules.tui 节 + 独立凭据文件里的 providerAccess */
+function loadModuleConfig(context: ModuleContext, agentDir: string): LoadedPiTuiConfig {
+	const section = readTuiSection(context.getConfig());
+	const provider = loadProviderAccessFile(agentDir);
+	return {
+		config: {
+			appearance: section.config.appearance,
+			status: section.config.status,
+			data: {
+				...section.config.data,
+				...(provider.access ? { providerAccess: provider.access } : {}),
+			},
+			advanced: section.config.advanced,
+		},
+		warnings: [...section.warnings, ...provider.warnings],
+	};
+}
+
+function registerTui(
+	context: ModuleContext,
+	options: TuiModuleOptions,
+	runtime: TuiMenuRuntime,
+): void {
+	const agentDir = options.agentDir ?? getAgentDir();
+
+	// 宿主注入：配置完全来自本模块（modules.tui 节 + pi-tui.json 凭据文件），
+	// lifecycle 不再自己找配置文件；保存后由 TuiMenuRuntime.applyUi 触发原子重装。
+	const handle = registerPiTuiLifecycle(context.pi, options.output ?? process.stdout, {
+		agentDir,
+		loadConfig: () => loadModuleConfig(context, agentDir),
+		t: context.t,
+	});
+	__test__.handle = handle;
+
+	runtime.reload = () => context.reloadConfig();
+	runtime.applyUi = () => handle.applyConfig();
+
+	context.services.register(TUI_PROVIDER_SERVICE_NAME, {
+		id: "tui",
+		snapshot: () => handle.getProviderUsage()?.getState(),
+		refresh: async () => {
+			await handle.getProviderUsage()?.refresh();
+		},
+	} satisfies TuiProviderUsageService);
+}
+
+export function createTuiModule(options: TuiModuleOptions = {}): ModuleDefinition {
+	// 菜单写盘后要让状态中枢重新读盘，并让 lifecycle 重装常驻 UI
+	const runtime: TuiMenuRuntime = { reload: async () => {}, applyUi: () => false };
+
+	return {
+		id: TUI_MODULE_ID,
+		labelKey: "module.tui.label",
+		descriptionKey: "module.tui.description",
+		group: "general",
+		// schema 只放总开关；外观/状态/数据/高级是结构化配置（非 schema 键），
+		// 由本模块的菜单与配置层读写。
+		configSchema: {
+			enabled: enabledField("module.tui.enabled.label", "module.tui.enabled.description"),
+		},
+		register(context): void {
+			registerTui(context, options, runtime);
+		},
+		menuItems(context): ReturnType<typeof buildTuiMenuItems> {
+			return buildTuiMenuItems(context, runtime);
+		},
+	};
+}
+
+export const tuiModule: ModuleDefinition = createTuiModule();

package/modules/tui/kernel/pkg/shared/grok-subscription.ts

@@ -0,0 +1,169 @@
+// xAI / Grok 官方订阅额度。grok.com 使用 gRPC-web + protobuf，且未公开 .proto；
+// 此处保留 cc-switch 0b5da51 的防御式字段扫描策略，不把二进制响应当 JSON 猜测。
+
+import type { ProviderUsage } from './provider-contracts.ts';
+
+const ENDPOINT = 'https://grok.com/grok_api_v2.GrokBuildBilling/GetGrokCreditsConfig';
+
+interface Scan {
+  floats: Array<{ path: number[]; value: number; order: number }>;
+  varints: Array<{ path: number[]; value: number }>;
+}
+
+function readVarint(bytes: Uint8Array, state: { index: number }): number | null {
+  let value = 0;
+  let shift = 0;
+  while (state.index < bytes.length && shift < 53) {
+    const byte = bytes[state.index++];
+    value += (byte & 0x7f) * 2 ** shift;
+    if ((byte & 0x80) === 0) return value;
+    shift += 7;
+  }
+  return null;
+}
+
+function scanProtobuf(
+  bytes: Uint8Array,
+  depth: number,
+  path: number[],
+  scan: Scan,
+  order: { value: number },
+): void {
+  const state = { index: 0 };
+  while (state.index < bytes.length) {
+    const start = state.index;
+    const key = readVarint(bytes, state);
+    if (!key) {
+      state.index = start + 1;
+      continue;
+    }
+    const field = Math.floor(key / 8);
+    const wire = key & 7;
+    const fieldPath = [...path, field];
+    if (wire === 0) {
+      const value = readVarint(bytes, state);
+      if (value == null) state.index = start + 1;
+      else scan.varints.push({ path: fieldPath, value });
+    } else if (wire === 1) {
+      state.index += 8;
+    } else if (wire === 2) {
+      const length = readVarint(bytes, state);
+      if (length == null || state.index + length > bytes.length) {
+        state.index = start + 1;
+        continue;
+      }
+      const end = state.index + length;
+      if (depth < 4) scanProtobuf(bytes.subarray(state.index, end), depth + 1, fieldPath, scan, order);
+      state.index = end;
+    } else if (wire === 5) {
+      if (state.index + 4 > bytes.length) break;
+      const view = new DataView(bytes.buffer, bytes.byteOffset + state.index, 4);
+      scan.floats.push({ path: fieldPath, value: view.getFloat32(0, true), order: order.value++ });
+      state.index += 4;
+    } else {
+      state.index = start + 1;
+    }
+  }
+}
+
+function grpcPayloads(bytes: Uint8Array): Uint8Array[] {
+  const payloads: Uint8Array[] = [];
+  let index = 0;
+  while (index < bytes.length) {
+    if (index + 5 > bytes.length) return [];
+    const flags = bytes[index];
+    const length = new DataView(bytes.buffer, bytes.byteOffset + index + 1, 4).getUint32(0, false);
+    const start = index + 5;
+    const end = start + length;
+    if (end > bytes.length) return [];
+    if ((flags & 0x80) === 0) payloads.push(bytes.subarray(start, end));
+    index = end;
+  }
+  return payloads;
+}
+
+function looksLikeProtobuf(bytes: Uint8Array): boolean {
+  const first = bytes[0];
+  return bytes.length > 0 && first >> 3 > 0 && [0, 1, 2, 5].includes(first & 7);
+}
+
+function samePath(left: number[], right: number[]): boolean {
+  return left.length === right.length && left.every((value, index) => value === right[index]);
+}
+
+function tierLabel(resetSeconds: number | null, nowSeconds: number): string {
+  if (resetSeconds != null) {
+    const days = Math.round((resetSeconds - nowSeconds) / 86_400);
+    if (days >= 4 && days <= 12) return '7d';
+    if (days >= 20 && days <= 45) return '30d';
+  }
+  return 'Credits';
+}
+
+export function parseGrokSubscription(
+  bytes: Uint8Array,
+  nowSeconds = Math.floor(Date.now() / 1000),
+): ProviderUsage | null {
+  let payloads = grpcPayloads(bytes);
+  if (!payloads.length && looksLikeProtobuf(bytes)) payloads = [bytes];
+  if (!payloads.length) return null;
+
+  const scan: Scan = { floats: [], varints: [] };
+  for (const payload of payloads) scanProtobuf(payload, 0, [], scan, { value: 0 });
+  const percent = scan.floats
+    .filter((item) => item.path.at(-1) === 1 && Number.isFinite(item.value) && item.value >= 0 && item.value <= 100)
+    .sort((left, right) => left.path.length - right.path.length || left.order - right.order)[0]?.value;
+  const resets = scan.varints
+    .filter((item) => item.value >= 1_700_000_000 && item.value <= 2_100_000_000 && item.value > nowSeconds)
+    .sort((left, right) => {
+      const leftExact = samePath(left.path, [1, 5, 1]) ? 0 : 1;
+      const rightExact = samePath(right.path, [1, 5, 1]) ? 0 : 1;
+      return leftExact - rightExact || left.value - right.value;
+    });
+  const reset = resets[0]?.value ?? null;
+  const hasPeriod = scan.varints.some(
+    (item) =>
+      (item.path[0] === 1 && item.path[1] === 6) ||
+      (samePath(item.path, [1, 8, 1]) && (item.value === 1 || item.value === 2)),
+  );
+  const used = percent ?? (!scan.floats.length && reset != null && hasPeriod ? 0 : null);
+  if (used == null) return null;
+  return {
+    mode: 'subscription',
+    quota: {
+      provider: 'xai',
+      windows: [
+        {
+          label: tierLabel(reset, nowSeconds),
+          remainingPercent: Math.max(0, Math.min(100, 100 - used)),
+          resetMs: reset == null ? null : reset * 1000,
+        },
+      ],
+    },
+  };
+}
+
+export async function fetchGrokSubscription(
+  accessToken: string,
+  request: typeof fetch = fetch,
+  userAgent?: string,
+): Promise<ProviderUsage | null> {
+  if (!accessToken) return null;
+  const response = await request(ENDPOINT, {
+    method: 'POST',
+    headers: {
+      Authorization: `Bearer ${accessToken}`,
+      Origin: 'https://grok.com',
+      Referer: 'https://grok.com/?_s=usage',
+      Accept: '*/*',
+      'Content-Type': 'application/grpc-web+proto',
+      'x-grpc-web': '1',
+      'x-user-agent': 'connect-es/2.1.1',
+      ...(userAgent ? { 'User-Agent': userAgent } : {}),
+    },
+    body: new Uint8Array(5),
+    signal: AbortSignal.timeout(15_000),
+  });
+  if (!response.ok || (response.headers.get('grpc-status') ?? '0') !== '0') return null;
+  return parseGrokSubscription(new Uint8Array(await response.arrayBuffer()));
+}