@ekworp/opencode-mystatus 0.0.0-stage → 1.2.7

package/dist/plugin/lib/copilot.js

@@ -0,0 +1,346 @@
+/**
+ * GitHub Copilot Premium Requests Quota Module
+ *
+ * [Input]: GitHub token from ~/.local/share/opencode/auth.json (github-copilot provider)
+ * [Output]: Formatted quota usage information with progress bars
+ * [Location]: Called by mystatus.ts to handle GitHub Copilot accounts
+ * [Sync]: mystatus.ts, types.ts, utils.ts, i18n.ts
+ *
+ * [Updated]: Jan 2026 - Handle new OpenCode official partnership auth flow
+ * The new OAuth tokens (gho_) need to be exchanged for Copilot session tokens
+ * before calling the internal quota API.
+ */
+import * as fs from "node:fs";
+import * as os from "node:os";
+import * as path from "node:path";
+import { t } from "./i18n";
+import { createProgressBar, fetchWithTimeout } from "./utils";
+// ============================================================================
+// Constants
+// ============================================================================
+const GITHUB_API_BASE_URL = "https://api.github.com";
+// Config file path for user's fine-grained PAT
+const COPILOT_QUOTA_CONFIG_PATH = path.join(os.homedir(), ".config", "opencode", "copilot-quota-token.json");
+// Updated to match current VS Code Copilot extension version
+const COPILOT_VERSION = "0.35.0";
+const EDITOR_VERSION = "vscode/1.107.0";
+const EDITOR_PLUGIN_VERSION = `copilot-chat/${COPILOT_VERSION}`;
+const USER_AGENT = `GitHubCopilotChat/${COPILOT_VERSION}`;
+// Headers matching opencode-copilot-auth plugin (required for token exchange)
+const COPILOT_HEADERS = {
+    "User-Agent": USER_AGENT,
+    "Editor-Version": EDITOR_VERSION,
+    "Editor-Plugin-Version": EDITOR_PLUGIN_VERSION,
+    "Copilot-Integration-Id": "vscode-chat",
+};
+// ============================================================================
+// Token Exchange (New auth flow for official OpenCode partnership)
+// ============================================================================
+/**
+ * Read optional Copilot quota config from user's config file
+ * Returns null if file doesn't exist or is invalid
+ */
+function readQuotaConfig() {
+    try {
+        if (!fs.existsSync(COPILOT_QUOTA_CONFIG_PATH)) {
+            return null;
+        }
+        const content = fs.readFileSync(COPILOT_QUOTA_CONFIG_PATH, "utf-8");
+        const config = JSON.parse(content);
+        // Validate required fields
+        if (!config.token || !config.username || !config.tier) {
+            return null;
+        }
+        // Validate tier is valid
+        const validTiers = [
+            "free",
+            "pro",
+            "pro+",
+            "business",
+            "enterprise",
+        ];
+        if (!validTiers.includes(config.tier)) {
+            return null;
+        }
+        return config;
+    }
+    catch {
+        return null;
+    }
+}
+/**
+ * Fetch quota using the public GitHub REST API
+ * Requires a fine-grained PAT with "Plan" read permission
+ */
+async function fetchPublicBillingUsage(config) {
+    const response = await fetchWithTimeout(`${GITHUB_API_BASE_URL}/users/${config.username}/settings/billing/premium_request/usage`, {
+        headers: {
+            Accept: "application/vnd.github+json",
+            Authorization: `Bearer ${config.token}`,
+            "X-GitHub-Api-Version": "2022-11-28",
+        },
+    });
+    if (!response.ok) {
+        const errorText = await response.text();
+        throw new Error(t.copilotApiError(response.status, errorText));
+    }
+    return response.json();
+}
+/**
+ * Exchange OAuth token for a Copilot session token
+ * Required for the new OpenCode official partnership auth flow (Jan 2026+)
+ */
+async function exchangeForCopilotToken(oauthToken) {
+    try {
+        const response = await fetchWithTimeout(`${GITHUB_API_BASE_URL}/copilot_internal/v2/token`, {
+            headers: {
+                Accept: "application/json",
+                Authorization: `Bearer ${oauthToken}`,
+                ...COPILOT_HEADERS,
+            },
+        });
+        if (!response.ok) {
+            // Token exchange failed - might be old token format or API change
+            return null;
+        }
+        const tokenData = await response.json();
+        return tokenData.token;
+    }
+    catch {
+        return null;
+    }
+}
+// ============================================================================
+// API Call
+// ============================================================================
+/**
+ * Build headers for GitHub API requests (quota endpoint)
+ */
+function buildGitHubHeaders(token) {
+    return {
+        "Content-Type": "application/json",
+        Accept: "application/json",
+        Authorization: `Bearer ${token}`,
+        ...COPILOT_HEADERS,
+    };
+}
+/**
+ * Build headers for legacy token format
+ */
+function buildLegacyHeaders(token) {
+    return {
+        "Content-Type": "application/json",
+        Accept: "application/json",
+        Authorization: `token ${token}`,
+        ...COPILOT_HEADERS,
+    };
+}
+/**
+ * Fetch GitHub Copilot usage data
+ * Tries multiple authentication methods to handle both old and new token formats
+ */
+async function fetchCopilotUsage(authData) {
+    // Use refresh token as the OAuth token (required)
+    // In new auth flow, access === refresh (both are the OAuth token)
+    const oauthToken = authData.refresh || authData.access;
+    if (!oauthToken) {
+        throw new Error("No OAuth token found in auth data");
+    }
+    const cachedAccessToken = authData.access;
+    const tokenExpiry = authData.expires || 0;
+    // Strategy 1: If we have a valid cached access token (from previous exchange), use it
+    if (cachedAccessToken &&
+        cachedAccessToken !== oauthToken &&
+        tokenExpiry > Date.now()) {
+        const response = await fetchWithTimeout(`${GITHUB_API_BASE_URL}/copilot_internal/user`, { headers: buildGitHubHeaders(cachedAccessToken) });
+        if (response.ok) {
+            return response.json();
+        }
+    }
+    // Strategy 2: Try direct call with OAuth token (works with older token formats)
+    const directResponse = await fetchWithTimeout(`${GITHUB_API_BASE_URL}/copilot_internal/user`, { headers: buildLegacyHeaders(oauthToken) });
+    if (directResponse.ok) {
+        return directResponse.json();
+    }
+    // Strategy 3: Exchange OAuth token for Copilot session token (new auth flow)
+    const copilotToken = await exchangeForCopilotToken(oauthToken);
+    if (copilotToken) {
+        const exchangedResponse = await fetchWithTimeout(`${GITHUB_API_BASE_URL}/copilot_internal/user`, { headers: buildGitHubHeaders(copilotToken) });
+        if (exchangedResponse.ok) {
+            return exchangedResponse.json();
+        }
+        const errorText = await exchangedResponse.text();
+        throw new Error(t.copilotApiError(exchangedResponse.status, errorText));
+    }
+    // All strategies failed - likely due to OpenCode's OAuth token lacking copilot scope
+    // The new OpenCode partnership uses a different OAuth client that doesn't grant
+    // access to the /copilot_internal/* endpoints
+    throw new Error(`${t.copilotQuotaUnavailable}\n\n${t.copilotQuotaWorkaround}`);
+}
+// ============================================================================
+// Formatting
+// ============================================================================
+/**
+ * Format a single quota line
+ */
+function formatQuotaLine(name, quota, width = 20) {
+    if (!quota)
+        return "";
+    if (quota.unlimited) {
+        return `${name.padEnd(14)} Unlimited`;
+    }
+    const total = quota.entitlement;
+    const used = total - quota.remaining;
+    const percentRemaining = Math.round(quota.percent_remaining);
+    const progressBar = createProgressBar(percentRemaining, width);
+    return `${name.padEnd(14)} ${progressBar} ${percentRemaining}% (${used}/${total})`;
+}
+/**
+ * Calculate days until reset
+ */
+function getResetCountdown(resetDate) {
+    const reset = new Date(resetDate);
+    const now = new Date();
+    const diffMs = reset.getTime() - now.getTime();
+    if (diffMs <= 0)
+        return t.resetsSoon;
+    const days = Math.floor(diffMs / (1000 * 60 * 60 * 24));
+    const hours = Math.floor((diffMs % (1000 * 60 * 60 * 24)) / (1000 * 60 * 60));
+    if (days > 0) {
+        return `${days}d ${hours}h`;
+    }
+    return `${hours}h`;
+}
+/**
+ * Format GitHub Copilot usage information
+ */
+function formatCopilotUsage(data) {
+    const lines = [];
+    // Account info
+    lines.push(`${t.account}        GitHub Copilot (${data.copilot_plan})`);
+    lines.push("");
+    // Premium requests (main quota)
+    const premium = data.quota_snapshots.premium_interactions;
+    if (premium) {
+        const premiumLine = formatQuotaLine(t.premiumRequests, premium);
+        if (premiumLine)
+            lines.push(premiumLine);
+        // Show overage info if applicable
+        if (premium.overage_count > 0) {
+            lines.push(`${t.overage}: ${premium.overage_count} ${t.overageRequests}`);
+        }
+    }
+    // Chat quota (if separate)
+    const chat = data.quota_snapshots.chat;
+    if (chat && !chat.unlimited) {
+        const chatLine = formatQuotaLine(t.chatQuota, chat);
+        if (chatLine)
+            lines.push(chatLine);
+    }
+    // Completions quota (if separate)
+    const completions = data.quota_snapshots.completions;
+    if (completions && !completions.unlimited) {
+        const completionsLine = formatQuotaLine(t.completionsQuota, completions);
+        if (completionsLine)
+            lines.push(completionsLine);
+    }
+    // Reset date
+    lines.push("");
+    const resetCountdown = getResetCountdown(data.quota_reset_date);
+    lines.push(`${t.quotaResets}: ${resetCountdown} (${data.quota_reset_date})`);
+    return lines.join("\n");
+}
+// Copilot plan limits (premium requests per month)
+// Source: https://docs.github.com/en/copilot/about-github-copilot/subscription-plans-for-github-copilot
+const COPILOT_PLAN_LIMITS = {
+    free: 50, // Copilot Free: 50 premium requests/month
+    pro: 300, // Copilot Pro: 300 premium requests/month
+    "pro+": 1500, // Copilot Pro+: 1500 premium requests/month
+    business: 300, // Copilot Business: 300 premium requests/month
+    enterprise: 1000, // Copilot Enterprise: 1000 premium requests/month
+};
+/**
+ * Format public billing API response
+ * Different structure from internal API - aggregates usage items
+ * Uses grossQuantity (total requests made) since netQuantity shows post-discount amount
+ */
+function formatPublicBillingUsage(data, tier) {
+    const lines = [];
+    // Account info
+    lines.push(`${t.account}        GitHub Copilot (@${data.user})`);
+    lines.push("");
+    // Aggregate all premium request usage (sum grossQuantity across all models)
+    const premiumItems = data.usageItems.filter((item) => item.sku === "Copilot Premium Request" || item.sku.includes("Premium"));
+    const totalUsed = premiumItems.reduce((sum, item) => sum + item.grossQuantity, 0);
+    // Get limit from tier
+    const limit = COPILOT_PLAN_LIMITS[tier];
+    const remaining = Math.max(0, limit - totalUsed);
+    const percentRemaining = Math.round((remaining / limit) * 100);
+    const progressBar = createProgressBar(percentRemaining, 20);
+    lines.push(`${t.premiumRequests.padEnd(14)} ${progressBar} ${percentRemaining}% (${totalUsed}/${limit})`);
+    // Show model breakdown
+    const modelItems = data.usageItems.filter((item) => item.model && item.grossQuantity > 0);
+    if (modelItems.length > 0) {
+        lines.push("");
+        lines.push(t.modelBreakdown || "Model breakdown:");
+        // Sort by usage descending
+        const sortedItems = [...modelItems].sort((a, b) => b.grossQuantity - a.grossQuantity);
+        for (const item of sortedItems.slice(0, 5)) {
+            // Show top 5 models
+            lines.push(`  ${item.model}: ${item.grossQuantity} ${item.unitType}`);
+        }
+    }
+    // Time period info
+    lines.push("");
+    const period = data.timePeriod;
+    const periodStr = period.month
+        ? `${period.year}-${String(period.month).padStart(2, "0")}`
+        : `${period.year}`;
+    lines.push(`${t.billingPeriod || "Period"}: ${periodStr}`);
+    return lines.join("\n");
+}
+/**
+ * Query GitHub Copilot account quota
+ * @param authData GitHub Copilot authentication data (optional if using PAT config)
+ * @returns Query result, null if no account configured
+ */
+export async function queryCopilotUsage(authData) {
+    // Strategy 1: Try public billing API with user's fine-grained PAT
+    const quotaConfig = readQuotaConfig();
+    if (quotaConfig) {
+        try {
+            const billingUsage = await fetchPublicBillingUsage(quotaConfig);
+            return {
+                success: true,
+                output: formatPublicBillingUsage(billingUsage, quotaConfig.tier),
+            };
+        }
+        catch (err) {
+            // PAT config exists but failed - report the error
+            return {
+                success: false,
+                error: err instanceof Error ? err.message : String(err),
+            };
+        }
+    }
+    // Strategy 2: Try internal API with OAuth token (legacy, may not work with new OpenCode auth)
+    // Check if account exists and has a refresh token (the GitHub OAuth token)
+    if (authData?.type !== "oauth" || !authData.refresh) {
+        // No Copilot account is configured. Do not query GitHub or report an
+        // error for platforms the user has not enabled.
+        return null;
+    }
+    try {
+        const usage = await fetchCopilotUsage(authData);
+        return {
+            success: true,
+            output: formatCopilotUsage(usage),
+        };
+    }
+    catch (err) {
+        return {
+            success: false,
+            error: err instanceof Error ? err.message : String(err),
+        };
+    }
+}

package/dist/plugin/lib/google.d.ts

@@ -0,0 +1,15 @@
+/**
+ * Google Cloud 额度查询模块
+ *
+ * [输入]: ~/.config/opencode/antigravity-accounts.json 中的账号信息
+ * [输出]: 格式化的额度使用情况（按重置时间自动分组）
+ * [定位]: 被 mystatus.ts 调用，处理 Google Cloud 账号
+ * [同步]: mystatus.ts, types.ts, utils.ts, i18n.ts
+ */
+import { type QueryResult } from "./types";
+/**
+ * 查询所有 Antigravity 账号的额度
+ * @returns 查询结果
+ */
+export declare function queryGoogleUsage(): Promise<QueryResult | null>;
+//# sourceMappingURL=google.d.ts.map

package/dist/plugin/lib/google.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"google.d.ts","sourceRoot":"","sources":["../../../plugin/lib/google.ts"],"names":[],"mappings":"AAAA;;;;;;;GAOG;AAOH,OAAO,EAIL,KAAK,WAAW,EACjB,MAAM,SAAS,CAAC;AAsRjB;;;GAGG;AACH,wBAAsB,gBAAgB,IAAI,OAAO,CAAC,WAAW,GAAG,IAAI,CAAC,CA2EpE"}

package/dist/plugin/lib/google.js

@@ -0,0 +1,265 @@
+/**
+ * Google Cloud 额度查询模块
+ *
+ * [输入]: ~/.config/opencode/antigravity-accounts.json 中的账号信息
+ * [输出]: 格式化的额度使用情况（按重置时间自动分组）
+ * [定位]: 被 mystatus.ts 调用，处理 Google Cloud 账号
+ * [同步]: mystatus.ts, types.ts, utils.ts, i18n.ts
+ */
+import { readFile } from "node:fs/promises";
+import { homedir } from "node:os";
+import { join } from "node:path";
+import { currentLang, t } from "./i18n";
+import { HIGH_USAGE_THRESHOLD, } from "./types";
+import { createProgressBar, fetchWithTimeout, safeMax } from "./utils";
+// ============================================================================
+// 常量
+// ============================================================================
+const GOOGLE_QUOTA_API_URL = "https://cloudcode-pa.googleapis.com/v1internal:fetchAvailableModels";
+const USER_AGENT = "antigravity/1.11.9 windows/amd64";
+// 需要显示的 4 个模型配置
+const MODELS_TO_DISPLAY = [
+    { key: "gemini-3-pro-high", altKey: "gemini-3-pro-low", display: "G3 Pro" },
+    { key: "gemini-3-pro-image", display: "G3 Image" },
+    { key: "gemini-3-flash", display: "G3 Flash" },
+    {
+        key: "claude-opus-4-5-thinking",
+        altKey: "claude-opus-4-5",
+        display: "Claude",
+    },
+];
+// 获取 Antigravity 账号文件路径
+function getAntigravityAccountsPath() {
+    const home = homedir();
+    const configDir = process.platform === "win32"
+        ? process.env.APPDATA || join(home, "AppData", "Roaming")
+        : join(home, ".config");
+    return join(configDir, "opencode", "antigravity-accounts.json");
+}
+const GOOGLE_TOKEN_REFRESH_URL = "https://oauth2.googleapis.com/token";
+// OAuth credentials from opencode-antigravity-auth project
+// Source: https://github.com/NoeFabris/opencode-antigravity-auth
+const GOOGLE_CLIENT_ID = "1071006060591-tmhssin2h21lcre235vtolojh4g403ep.apps.googleusercontent.com";
+const GOOGLE_CLIENT_SECRET = "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";
+// ============================================================================
+// 工具函数
+// ============================================================================
+/**
+ * 格式化重置时间为简短显示（如 "4h 59m"）
+ */
+function formatResetTimeShort(isoTime) {
+    if (!isoTime)
+        return "-";
+    try {
+        const resetDate = new Date(isoTime);
+        const now = new Date();
+        const diffMs = resetDate.getTime() - now.getTime();
+        if (diffMs <= 0)
+            return currentLang === "zh" ? "已重置" : "reset";
+        const diffMinutes = Math.floor(diffMs / 60000);
+        const days = Math.floor(diffMinutes / 1440);
+        const hours = Math.floor((diffMinutes % 1440) / 60);
+        const minutes = diffMinutes % 60;
+        if (days > 0) {
+            return `${days}d ${hours}h`;
+        }
+        return `${hours}h ${minutes}m`;
+    }
+    catch {
+        return "-";
+    }
+}
+/**
+ * 从 API 响应中提取 4 个模型的额度信息
+ */
+function extractModelQuotas(data) {
+    const quotas = [];
+    for (const modelConfig of MODELS_TO_DISPLAY) {
+        let modelInfo = data.models[modelConfig.key];
+        // 如果主 key 没有数据，尝试 altKey
+        if (!modelInfo && modelConfig.altKey) {
+            modelInfo = data.models[modelConfig.altKey];
+        }
+        // 只要模型存在就显示（即使没有 quotaInfo 或额度为 0）
+        if (modelInfo) {
+            const remainingFraction = modelInfo.quotaInfo?.remainingFraction ?? 0;
+            quotas.push({
+                displayName: modelConfig.display,
+                remainPercent: Math.round(remainingFraction * 100),
+                resetTimeDisplay: formatResetTimeShort(modelInfo.quotaInfo?.resetTime || ""),
+            });
+        }
+    }
+    return quotas;
+}
+/**
+ * 刷新 Google access token
+ */
+async function refreshAccessToken(refreshToken) {
+    const params = new URLSearchParams({
+        client_id: GOOGLE_CLIENT_ID,
+        client_secret: GOOGLE_CLIENT_SECRET,
+        refresh_token: refreshToken,
+        grant_type: "refresh_token",
+    });
+    const response = await fetch(GOOGLE_TOKEN_REFRESH_URL, {
+        method: "POST",
+        headers: { "Content-Type": "application/x-www-form-urlencoded" },
+        body: params,
+    });
+    if (!response.ok) {
+        const errorText = await response.text();
+        throw new Error(t.googleApiError(response.status, errorText));
+    }
+    return response.json();
+}
+// ============================================================================
+// API 调用
+// ============================================================================
+/**
+ * 获取 Google Cloud 使用情况
+ */
+async function fetchGoogleUsage(accessToken, projectId) {
+    const response = await fetchWithTimeout(GOOGLE_QUOTA_API_URL, {
+        method: "POST",
+        headers: {
+            "Content-Type": "application/json",
+            Authorization: `Bearer ${accessToken}`,
+            "User-Agent": USER_AGENT,
+        },
+        body: JSON.stringify({ project: projectId }),
+    });
+    if (!response.ok) {
+        const errorText = await response.text();
+        throw new Error(t.googleApiError(response.status, errorText));
+    }
+    return response.json();
+}
+/**
+ * 查询单个账号的额度
+ */
+async function fetchAccountQuota(account) {
+    try {
+        // 刷新 access token
+        const { access_token } = await refreshAccessToken(account.refreshToken);
+        // 使用 projectId 或 managedProjectId
+        const projectId = account.projectId || account.managedProjectId;
+        if (!projectId) {
+            return { success: false, error: t.googleNoProjectId };
+        }
+        // 查询额度
+        const data = await fetchGoogleUsage(access_token, projectId);
+        // 提取 4 个模型的额度
+        const models = extractModelQuotas(data);
+        if (models.length === 0) {
+            return { success: true, models: undefined, maxUsage: 0 };
+        }
+        // 计算最大使用率
+        const maxUsage = safeMax(models.map((m) => 100 - m.remainPercent));
+        return { success: true, models, maxUsage };
+    }
+    catch (err) {
+        return {
+            success: false,
+            error: err instanceof Error ? err.message : String(err),
+        };
+    }
+}
+// ============================================================================
+// 格式化输出
+// ============================================================================
+/**
+ * 格式化单个账号的额度（4 个模型分别显示）
+ */
+function formatAccountQuota(quotaInfo) {
+    const lines = [];
+    // 标题行
+    lines.push(`### ${quotaInfo.email}`);
+    if (quotaInfo.models.length === 0) {
+        lines.push("");
+        lines.push(t.noQuotaData);
+        return lines.join("\n");
+    }
+    lines.push("");
+    // 每个模型一行：模型名 | 重置时间 | 百分比
+    for (const model of quotaInfo.models) {
+        const progressBar = createProgressBar(model.remainPercent, 20);
+        lines.push(`${model.displayName.padEnd(10)} ${model.resetTimeDisplay.padEnd(10)} ${progressBar} ${model.remainPercent}%`);
+    }
+    // 警告
+    if (quotaInfo.maxUsage >= HIGH_USAGE_THRESHOLD) {
+        lines.push("");
+        lines.push(t.limitReached);
+    }
+    return lines.join("\n");
+}
+// ============================================================================
+// 导出接口
+// ============================================================================
+/**
+ * 查询所有 Antigravity 账号的额度
+ * @returns 查询结果
+ */
+export async function queryGoogleUsage() {
+    try {
+        // 读取账号文件
+        const accountsPath = getAntigravityAccountsPath();
+        let content;
+        try {
+            content = await readFile(accountsPath, "utf-8");
+        }
+        catch (err) {
+            // No Antigravity configuration means Google is not enabled.
+            if (err &&
+                typeof err === "object" &&
+                "code" in err &&
+                err.code === "ENOENT") {
+                return null;
+            }
+            throw err;
+        }
+        const file = JSON.parse(content);
+        if (!file.accounts || file.accounts.length === 0) {
+            return null;
+        }
+        // 过滤掉没有邮箱的账号
+        const validAccounts = file.accounts.filter((account) => account.email);
+        if (validAccounts.length === 0) {
+            return null;
+        }
+        // 并行查询所有账号
+        const results = await Promise.all(validAccounts.map((account) => fetchAccountQuota(account).then((result) => ({ account, result }))));
+        // 收集输出
+        const outputs = [];
+        for (const { account, result } of results) {
+            if (!result.success) {
+                outputs.push(`${account.email || t.unknown}: ${result.error}`);
+            }
+            else if (result.models && result.models.length > 0) {
+                const quotaInfo = {
+                    email: account.email || t.unknown,
+                    models: result.models,
+                    maxUsage: result.maxUsage || 0,
+                };
+                outputs.push(formatAccountQuota(quotaInfo));
+            }
+        }
+        // 如果没有符合条件的账号
+        if (outputs.length === 0) {
+            return {
+                success: true,
+                output: t.noQuotaData,
+            };
+        }
+        return {
+            success: true,
+            output: outputs.join("\n\n"),
+        };
+    }
+    catch (err) {
+        return {
+            success: false,
+            error: err instanceof Error ? err.message : String(err),
+        };
+    }
+}