pi-ui-extend 0.1.1

package/external/pi-tools-suite/src/usage/lib/openai.ts

@@ -0,0 +1,268 @@
+/**
+ * OpenAI 额度查询模块
+ *
+ * [输入]: OAuth access token
+ * [输出]: 格式化的额度使用情况
+ * [定位]: 被 usage.ts 调用，处理 OpenAI 账号
+ * [同步]: usage.ts, types.ts, utils.ts
+ */
+
+import { type QueryResult, type OpenAIAuthData } from "./types";
+import {
+  formatDuration,
+  createProgressBar,
+  calcRemainPercent,
+  fetchWithTimeout,
+} from "./utils";
+
+// ============================================================================
+// 类型定义
+// ============================================================================
+
+interface RateLimitWindow {
+  used_percent: number;
+  limit_window_seconds: number;
+  reset_after_seconds: number;
+}
+
+interface OpenAIRateLimit {
+  allowed?: boolean;
+  limit_reached: boolean;
+  primary_window: RateLimitWindow;
+  secondary_window: RateLimitWindow | null;
+}
+
+interface OpenAIAdditionalRateLimit {
+  limit_name: string;
+  metered_feature?: string;
+  rate_limit: OpenAIRateLimit | null;
+}
+
+interface OpenAIUsageResponse {
+  plan_type: string;
+  rate_limit: OpenAIRateLimit | null;
+  additional_rate_limits?: OpenAIAdditionalRateLimit[];
+}
+
+// ============================================================================
+// JWT 解析
+// ============================================================================
+
+interface JwtPayload {
+  "https://api.openai.com/profile"?: {
+    email?: string;
+  };
+  "https://api.openai.com/auth"?: {
+    chatgpt_account_id?: string;
+  };
+}
+
+/**
+ * Base64URL 解码
+ */
+function base64UrlDecode(input: string): string {
+  const base64 = input.replace(/-/g, "+").replace(/_/g, "/");
+  const padLen = (4 - (base64.length % 4)) % 4;
+  const padded = base64 + "=".repeat(padLen);
+  return Buffer.from(padded, "base64").toString("utf8");
+}
+
+/**
+ * 解析 JWT token，提取 payload
+ */
+function parseJwt(token: string): JwtPayload | null {
+  try {
+    const parts = token.split(".");
+    if (parts.length !== 3) return null;
+    const payloadJson = base64UrlDecode(parts[1]);
+    return JSON.parse(payloadJson) as JwtPayload;
+  } catch {
+    return null;
+  }
+}
+
+/**
+ * 从 JWT 中提取用户邮箱
+ */
+function getEmailFromJwt(token: string): string | null {
+  const payload = parseJwt(token);
+  return payload?.["https://api.openai.com/profile"]?.email ?? null;
+}
+
+/**
+ * 从 JWT 中提取 ChatGPT 账号 ID（用于团队/组织账号）
+ */
+function getAccountIdFromJwt(token: string): string | null {
+  const payload = parseJwt(token);
+  return payload?.["https://api.openai.com/auth"]?.chatgpt_account_id ?? null;
+}
+
+// ============================================================================
+// 工具函数
+// ============================================================================
+
+/**
+ * 将窗口时长（秒）转换为可读的窗口名称
+ */
+function formatWindowName(seconds: number): string {
+  const days = Math.round(seconds / 86400);
+
+  if (days >= 1) {
+    return `${days}-day limit`;
+  }
+  return `${Math.round(seconds / 3600)}-hour limit`;
+}
+
+/**
+ * 格式化单个窗口的使用情况
+ */
+function formatWindow(window: RateLimitWindow): string[] {
+  const windowName = formatWindowName(window.limit_window_seconds);
+  const remainPercent = calcRemainPercent(window.used_percent);
+  const progressBar = createProgressBar(remainPercent);
+  const resetTime = formatDuration(window.reset_after_seconds);
+
+  return [
+    windowName,
+    `${progressBar} ${remainPercent}% remaining`,
+    `Resets in: ${resetTime}`,
+  ];
+}
+
+/**
+ * 格式化一个完整的速率限制（主窗口 + 次窗口 + 状态）
+ */
+function formatRateLimit(rateLimit: OpenAIRateLimit): string[] {
+  const lines: string[] = [];
+
+  if (rateLimit.primary_window) {
+    lines.push(...formatWindow(rateLimit.primary_window));
+  }
+
+  if (rateLimit.secondary_window) {
+    lines.push("");
+    lines.push(...formatWindow(rateLimit.secondary_window));
+  }
+
+  if (rateLimit.limit_reached) {
+    lines.push("");
+    lines.push("⚠️ Rate limit reached!");
+  }
+
+  return lines;
+}
+
+// ============================================================================
+// API 调用
+// ============================================================================
+
+const OPENAI_USAGE_URL = "https://chatgpt.com/backend-api/wham/usage";
+
+/**
+ * 获取 OpenAI 使用情况
+ */
+async function fetchOpenAIUsage(
+  accessToken: string,
+): Promise<OpenAIUsageResponse> {
+  const headers: Record<string, string> = {
+    Authorization: `Bearer ${accessToken}`,
+    "User-Agent": "OpenCode-Status-Plugin/1.0",
+  };
+
+  const accountId = getAccountIdFromJwt(accessToken);
+  if (accountId) {
+    headers["ChatGPT-Account-Id"] = accountId;
+  }
+
+  const response = await fetchWithTimeout(OPENAI_USAGE_URL, {
+    headers,
+  });
+
+  if (!response.ok) {
+    const errorText = await response.text();
+    throw new Error(`OpenAI API request failed (${response.status}): ${errorText}`);
+  }
+
+  return response.json() as Promise<OpenAIUsageResponse>;
+}
+
+// ============================================================================
+// 格式化输出
+// ============================================================================
+
+/**
+ * 格式化 OpenAI 使用情况
+ */
+function formatOpenAIUsage(
+  data: OpenAIUsageResponse,
+  email: string | null,
+): string {
+  const { plan_type, rate_limit, additional_rate_limits } = data;
+  const lines: string[] = [];
+
+  // 标题行：Account: email (plan)
+  const accountDisplay = email || "unknown";
+  lines.push(`Account:        ${accountDisplay} (${plan_type})`);
+  lines.push("");
+
+  // 默认账号窗口
+  if (rate_limit) {
+    lines.push(...formatRateLimit(rate_limit));
+  }
+
+  // 额外的独立模型/功能限制（例如 GPT-5.3-Codex-Spark）
+  const additionalLimits = additional_rate_limits?.filter((item) => item.rate_limit) ?? [];
+  for (const limit of additionalLimits) {
+    lines.push("");
+    lines.push(`Additional limit: ${limit.limit_name}`);
+    if (limit.metered_feature) {
+      lines.push(`Metered feature:  ${limit.metered_feature}`);
+    }
+    lines.push("");
+    lines.push(...formatRateLimit(limit.rate_limit!));
+  }
+
+  return lines.join("\n");
+}
+
+// ============================================================================
+// 导出接口
+// ============================================================================
+
+export type { OpenAIAuthData };
+
+/**
+ * 查询 OpenAI 账号额度
+ * @param authData OpenAI 认证数据
+ * @returns 查询结果，如果账号不存在或无效返回 null
+ */
+export async function queryOpenAIUsage(
+  authData: OpenAIAuthData | undefined,
+): Promise<QueryResult | null> {
+  // 检查账号是否存在且有效
+  if (!authData || authData.type !== "oauth" || !authData.access) {
+    return null;
+  }
+
+  // 检查 OAuth token 是否过期
+  if (authData.expires && authData.expires < Date.now()) {
+    return {
+      success: false,
+      error: "⚠️ OAuth token expired. Please use an OpenAI model in OpenCode to refresh authorization.",
+    };
+  }
+
+  try {
+    const email = getEmailFromJwt(authData.access);
+    const usage = await fetchOpenAIUsage(authData.access);
+    return {
+      success: true,
+      output: formatOpenAIUsage(usage, email),
+    };
+  } catch (err) {
+    return {
+      success: false,
+      error: err instanceof Error ? err.message : String(err),
+    };
+  }
+}

package/external/pi-tools-suite/src/usage/lib/types.ts

@@ -0,0 +1,114 @@
+/**
+ * 共享类型定义
+ *
+ * [定位]: 被所有平台模块共享使用的类型
+ * [同步]: openai.ts, zhipu.ts, google.ts, usage.ts
+ */
+
+// ============================================================================
+// 查询结果类型
+// ============================================================================
+
+/**
+ * 平台查询结果
+ */
+export interface QueryResult {
+  success: boolean;
+  output?: string;
+  error?: string;
+}
+
+// ============================================================================
+// 认证数据类型
+// ============================================================================
+
+/**
+ * OpenAI OAuth 认证数据
+ */
+export interface OpenAIAuthData {
+  type: string;
+  access?: string;
+  refresh?: string;
+  expires?: number;
+}
+
+/**
+ * 智谱 AI API 认证数据
+ */
+export interface ZhipuAuthData {
+  type: string;
+  key?: string;
+}
+
+/**
+ * GitHub Copilot 认证数据
+ */
+export interface CopilotAuthData {
+  type: string;
+  refresh?: string;
+  access?: string;
+  expires?: number;
+}
+
+/**
+ * Copilot subscription tier
+ * See: https://docs.github.com/en/copilot/about-github-copilot/subscription-plans-for-github-copilot
+ */
+export type CopilotTier = "free" | "pro" | "pro+" | "business" | "enterprise";
+
+/**
+ * Copilot quota token configuration
+ * Stored in ~/.config/opencode/copilot-quota-token.json
+ *
+ * Users can create a fine-grained PAT with "Plan" read permission
+ * to enable quota checking via the public GitHub REST API.
+ */
+export interface CopilotQuotaConfig {
+  /** Fine-grained PAT with "Plan" read permission */
+  token: string;
+  /** GitHub username (for API calls) */
+  username: string;
+  /** Copilot subscription tier (determines monthly quota limit) */
+  tier: CopilotTier;
+}
+
+/**
+ * Antigravity 账号（来自 ~/.config/opencode/antigravity-accounts.json）
+ */
+export interface AntigravityAccount {
+  email?: string;
+  refreshToken: string;
+  projectId?: string;
+  managedProjectId?: string;
+  addedAt: number;
+  lastUsed: number;
+  rateLimitResetTimes?: Record<string, number>;
+}
+
+/**
+ * Antigravity 账号文件
+ */
+export interface AntigravityAccountsFile {
+  version: number;
+  accounts: AntigravityAccount[];
+}
+
+/**
+ * 完整认证数据结构
+ */
+export interface AuthData {
+  openai?: OpenAIAuthData;
+  "zhipuai-coding-plan"?: ZhipuAuthData;
+  "zai-coding-plan"?: ZhipuAuthData;
+  "github-copilot"?: CopilotAuthData;
+}
+
+// ============================================================================
+// 常量配置
+// ============================================================================
+
+/** 高使用率警告阈值（百分比） */
+export const HIGH_USAGE_THRESHOLD = 80;
+
+/** API 请求超时时间（毫秒） */
+export const REQUEST_TIMEOUT_MS = 10000;

package/external/pi-tools-suite/src/usage/lib/utils.ts

@@ -0,0 +1,134 @@
+/**
+ * 共享工具函数
+ *
+ * [定位]: 被所有平台模块共享使用的工具函数
+ * [同步]: openai.ts, zhipu.ts, google.ts
+ */
+
+import { REQUEST_TIMEOUT_MS } from "./types";
+
+// ============================================================================
+// 时间格式化
+// ============================================================================
+
+/**
+ * 将秒数转换为人类可读的时间格式
+ */
+export function formatDuration(seconds: number): string {
+  const days = Math.floor(seconds / 86400);
+  const hours = Math.floor((seconds % 86400) / 3600);
+  const minutes = Math.floor((seconds % 3600) / 60);
+
+  const parts: string[] = [];
+  if (days > 0) parts.push(`${days}d`);
+  if (hours > 0) parts.push(`${hours}h`);
+  if (minutes > 0 || parts.length === 0) parts.push(`${minutes}m`);
+
+  return parts.join(" ");
+}
+
+// ============================================================================
+// 进度条
+// ============================================================================
+
+/**
+ * 生成进度条（实心代表剩余额度）
+ * @param remainPercent 剩余百分比 (0-100)
+ * @param width 进度条宽度（字符数）
+ */
+export function createProgressBar(
+  remainPercent: number,
+  width: number = 30,
+): string {
+  // 确保百分比在有效范围内
+  const safePercent = Math.max(0, Math.min(100, remainPercent));
+  const filled = Math.round((safePercent / 100) * width);
+  const empty = width - filled;
+
+  const filledChar = "█";
+  const emptyChar = "░";
+
+  return filledChar.repeat(filled) + emptyChar.repeat(empty);
+}
+
+// ============================================================================
+// 数值格式化
+// ============================================================================
+
+/**
+ * 计算剩余百分比并取整
+ * @param usedPercent 已使用百分比
+ */
+export function calcRemainPercent(usedPercent: number): number {
+  return Math.round(100 - usedPercent);
+}
+
+/**
+ * 格式化 Token 数量（以百万为单位）
+ */
+export function formatTokens(tokens: number): string {
+  return (tokens / 1000000).toFixed(1) + "M";
+}
+
+// ============================================================================
+// 网络请求
+// ============================================================================
+
+/**
+ * 带超时的 fetch 请求
+ * @param url 请求 URL
+ * @param options fetch 选项
+ * @param timeoutMs 超时时间（毫秒），默认使用全局配置
+ */
+export async function fetchWithTimeout(
+  url: string,
+  options: RequestInit,
+  timeoutMs: number = REQUEST_TIMEOUT_MS,
+): Promise<Response> {
+  const controller = new AbortController();
+  const timeoutId = setTimeout(() => controller.abort(), timeoutMs);
+
+  try {
+    const response = await fetch(url, {
+      ...options,
+      signal: controller.signal,
+    });
+    return response;
+  } catch (err) {
+    if (err instanceof Error && err.name === "AbortError") {
+      throw new Error(`Request timeout (${Math.round(timeoutMs / 1000)}s)`);
+    }
+    throw err;
+  } finally {
+    clearTimeout(timeoutId);
+  }
+}
+
+// ============================================================================
+// 安全计算
+// ============================================================================
+
+/**
+ * 安全获取数组最大值，空数组返回 0
+ */
+export function safeMax(arr: number[]): number {
+  if (arr.length === 0) return 0;
+  return Math.max(...arr);
+}
+
+// ============================================================================
+// 字符串处理
+// ============================================================================
+
+/**
+ * 脱敏显示敏感字符串
+ * 显示前 N 位和后 N 位，中间用 **** 替代
+ * @param str 原始字符串
+ * @param showChars 前后各显示的字符数，默认 4
+ */
+export function maskString(str: string, showChars: number = 4): string {
+  if (str.length <= showChars * 2) {
+    return str;
+  }
+  return `${str.slice(0, showChars)}****${str.slice(-showChars)}`;
+}