@ekworp/opencode-mystatus 0.0.0-stage → 1.2.7

package/dist/plugin/lib/utils.js

@@ -0,0 +1,115 @@
+/**
+ * 共享工具函数
+ *
+ * [定位]: 被所有平台模块共享使用的工具函数
+ * [同步]: openai.ts, zhipu.ts, google.ts
+ */
+import { currentLang, t } from "./i18n";
+import { REQUEST_TIMEOUT_MS } from "./types";
+// ============================================================================
+// 时间格式化
+// ============================================================================
+/**
+ * 将秒数转换为人类可读的时间格式
+ */
+export function formatDuration(seconds) {
+    const days = Math.floor(seconds / 86400);
+    const hours = Math.floor((seconds % 86400) / 3600);
+    const minutes = Math.floor((seconds % 3600) / 60);
+    const parts = [];
+    if (days > 0)
+        parts.push(t.days(days));
+    if (hours > 0)
+        parts.push(t.hours(hours));
+    if (minutes > 0 || parts.length === 0)
+        parts.push(t.minutes(minutes));
+    return parts.join(currentLang === "en" ? " " : "");
+}
+// ============================================================================
+// 进度条
+// ============================================================================
+/**
+ * 生成进度条（实心代表剩余额度）
+ * @param remainPercent 剩余百分比 (0-100)
+ * @param width 进度条宽度（字符数）
+ */
+export function createProgressBar(remainPercent, width = 30) {
+    // 确保百分比在有效范围内
+    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) {
+    return Math.round(100 - usedPercent);
+}
+/**
+ * 格式化 Token 数量（以百万为单位）
+ */
+export function formatTokens(tokens) {
+    return `${(tokens / 1000000).toFixed(1)}M`;
+}
+// ============================================================================
+// 网络请求
+// ============================================================================
+/**
+ * 带超时的 fetch 请求
+ * @param url 请求 URL
+ * @param options fetch 选项
+ * @param timeoutMs 超时时间（毫秒），默认使用全局配置
+ */
+export async function fetchWithTimeout(url, options, timeoutMs = REQUEST_TIMEOUT_MS) {
+    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(t.timeoutError(Math.round(timeoutMs / 1000)));
+        }
+        throw err;
+    }
+    finally {
+        clearTimeout(timeoutId);
+    }
+}
+// ============================================================================
+// 安全计算
+// ============================================================================
+/**
+ * 安全获取数组最大值，空数组返回 0
+ */
+export function safeMax(arr) {
+    if (arr.length === 0)
+        return 0;
+    return Math.max(...arr);
+}
+// ============================================================================
+// 字符串处理
+// ============================================================================
+/**
+ * 脱敏显示敏感字符串
+ * 显示前 N 位和后 N 位，中间用 **** 替代
+ * @param str 原始字符串
+ * @param showChars 前后各显示的字符数，默认 4
+ */
+export function maskString(str, showChars = 4) {
+    if (str.length <= showChars * 2) {
+        return str;
+    }
+    return `${str.slice(0, showChars)}****${str.slice(-showChars)}`;
+}

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

@@ -0,0 +1,23 @@
+/**
+ * 智谱 AI 额度查询模块
+ *
+ * [输入]: API Key
+ * [输出]: 格式化的额度使用情况
+ * [定位]: 被 mystatus.ts 调用，处理智谱 AI 账号
+ * [同步]: mystatus.ts, types.ts, utils.ts, i18n.ts
+ */
+import { type QueryResult, type ZhipuAuthData } from "./types";
+export type { ZhipuAuthData };
+/**
+ * 查询智谱 AI 账号额度
+ * @param authData 智谱认证数据
+ * @returns 查询结果，如果账号不存在或无效返回 null
+ */
+export declare function queryZhipuUsage(authData: ZhipuAuthData | undefined): Promise<QueryResult | null>;
+/**
+ * 查询 Z.ai 账号额度
+ * @param authData Z.ai 认证数据
+ * @returns 查询结果，如果账号不存在或无效返回 null
+ */
+export declare function queryZaiUsage(authData: ZhipuAuthData | undefined): Promise<QueryResult | null>;
+//# sourceMappingURL=zhipu.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"zhipu.d.ts","sourceRoot":"","sources":["../../../plugin/lib/zhipu.ts"],"names":[],"mappings":"AAAA;;;;;;;GAOG;AAGH,OAAO,EAEL,KAAK,WAAW,EAChB,KAAK,aAAa,EACnB,MAAM,SAAS,CAAC;AAwKjB,YAAY,EAAE,aAAa,EAAE,CAAC;AAyB9B;;;;GAIG;AACH,wBAAsB,eAAe,CACnC,QAAQ,EAAE,aAAa,GAAG,SAAS,GAClC,OAAO,CAAC,WAAW,GAAG,IAAI,CAAC,CAE7B;AAED;;;;GAIG;AACH,wBAAsB,aAAa,CACjC,QAAQ,EAAE,aAAa,GAAG,SAAS,GAClC,OAAO,CAAC,WAAW,GAAG,IAAI,CAAC,CAE7B"}

package/dist/plugin/lib/zhipu.js

@@ -0,0 +1,134 @@
+/**
+ * 智谱 AI 额度查询模块
+ *
+ * [输入]: API Key
+ * [输出]: 格式化的额度使用情况
+ * [定位]: 被 mystatus.ts 调用，处理智谱 AI 账号
+ * [同步]: mystatus.ts, types.ts, utils.ts, i18n.ts
+ */
+import { t } from "./i18n";
+import { HIGH_USAGE_THRESHOLD, } from "./types";
+import { calcRemainPercent, createProgressBar, fetchWithTimeout, formatDuration, formatTokens, maskString, safeMax, } from "./utils";
+// ============================================================================
+// API 调用
+// ============================================================================
+const ZHIPU_QUOTA_QUERY_URL = "https://bigmodel.cn/api/monitor/usage/quota/limit";
+const ZAI_QUOTA_QUERY_URL = "https://api.z.ai/api/monitor/usage/quota/limit";
+const ZHIPU_CONFIG = {
+    apiUrl: ZHIPU_QUOTA_QUERY_URL,
+    apiError: t.zhipuApiError,
+    accountLabel: t.zhipuAccountName,
+};
+const ZAI_CONFIG = {
+    apiUrl: ZAI_QUOTA_QUERY_URL,
+    apiError: t.zaiApiError,
+    accountLabel: t.zaiAccountName,
+};
+/**
+ * 获取智谱/Z.ai 使用情况
+ */
+async function fetchUsage(apiKey, config) {
+    const response = await fetchWithTimeout(config.apiUrl, {
+        method: "GET",
+        headers: {
+            Authorization: apiKey,
+            "Content-Type": "application/json",
+            "User-Agent": "OpenCode-Status-Plugin/1.0",
+        },
+    });
+    if (!response.ok) {
+        const errorText = await response.text();
+        throw new Error(config.apiError(response.status, errorText));
+    }
+    const data = (await response.json());
+    if (!data.success || data.code !== 200) {
+        throw new Error(config.apiError(data.code, data.msg || "Unknown error"));
+    }
+    return data;
+}
+// ============================================================================
+// 格式化输出
+// ============================================================================
+/**
+ * 格式化智谱 AI 使用情况
+ */
+function formatZhipuUsage(data, apiKey, accountLabel) {
+    const lines = [];
+    const limits = data.data.limits;
+    // 标题行：Account: API Key (Plan) - 显示脱敏后的 key
+    const maskedKey = maskString(apiKey);
+    lines.push(`${t.account}        ${maskedKey} (${accountLabel})`);
+    lines.push("");
+    // 空数组检查
+    if (!limits || limits.length === 0) {
+        lines.push(t.noQuotaData);
+        return lines.join("\n");
+    }
+    // 查找 TOKENS_LIMIT（5小时 Token 限额）
+    const tokensLimit = limits.find((l) => l.type === "TOKENS_LIMIT");
+    if (tokensLimit) {
+        const remainPercent = calcRemainPercent(tokensLimit.percentage);
+        const progressBar = createProgressBar(remainPercent);
+        lines.push(t.zhipuTokensLimit);
+        lines.push(`${progressBar} ${t.remaining(remainPercent)}`);
+        lines.push(`${t.used}: ${formatTokens(tokensLimit.currentValue)} / ${formatTokens(tokensLimit.usage)}`);
+        // 重置时间
+        if (tokensLimit.nextResetTime) {
+            const resetSeconds = Math.max(0, Math.floor((tokensLimit.nextResetTime - Date.now()) / 1000));
+            lines.push(t.resetIn(formatDuration(resetSeconds)));
+        }
+    }
+    // 查找 TIME_LIMIT（MCP 搜索次数）
+    const timeLimit = limits.find((l) => l.type === "TIME_LIMIT");
+    if (timeLimit) {
+        if (tokensLimit)
+            lines.push(""); // 分隔符
+        const remainPercent = calcRemainPercent(timeLimit.percentage);
+        const progressBar = createProgressBar(remainPercent);
+        lines.push(t.zhipuMcpLimit);
+        lines.push(`${progressBar} ${t.remaining(remainPercent)}`);
+        lines.push(`${t.used}: ${timeLimit.currentValue} / ${timeLimit.usage}`);
+    }
+    // 警告：如果任一限额使用率超过阈值
+    const maxPercentage = safeMax(limits.map((l) => l.percentage));
+    if (maxPercentage >= HIGH_USAGE_THRESHOLD) {
+        lines.push("");
+        lines.push(t.limitReached);
+    }
+    return lines.join("\n");
+}
+async function queryUsage(authData, config) {
+    // 检查账号是否存在且有效
+    if (authData?.type !== "api" || !authData.key) {
+        return null;
+    }
+    try {
+        const usage = await fetchUsage(authData.key, config);
+        return {
+            success: true,
+            output: formatZhipuUsage(usage, authData.key, config.accountLabel),
+        };
+    }
+    catch (err) {
+        return {
+            success: false,
+            error: err instanceof Error ? err.message : String(err),
+        };
+    }
+}
+/**
+ * 查询智谱 AI 账号额度
+ * @param authData 智谱认证数据
+ * @returns 查询结果，如果账号不存在或无效返回 null
+ */
+export async function queryZhipuUsage(authData) {
+    return queryUsage(authData, ZHIPU_CONFIG);
+}
+/**
+ * 查询 Z.ai 账号额度
+ * @param authData Z.ai 认证数据
+ * @returns 查询结果，如果账号不存在或无效返回 null
+ */
+export async function queryZaiUsage(authData) {
+    return queryUsage(authData, ZAI_CONFIG);
+}

package/dist/plugin/mystatus.d.ts

@@ -0,0 +1,12 @@
+/**
+ * OpenCode 额度状态查询插件
+ *
+ * [输入]: ~/.local/share/opencode/auth.json 和 ~/.config/opencode/antigravity-accounts.json 中的认证信息
+ * [输出]: 带进度条的额度使用情况展示
+ * [定位]: 通过 mystatus 工具查询各账号额度
+ * [同步]: lib/openai.ts, lib/zhipu.ts, lib/google.ts, lib/types.ts, lib/i18n.ts
+ */
+import { Plugin } from "@opencode/plugin";
+declare const _default: Plugin.Plugin;
+export default _default;
+//# sourceMappingURL=mystatus.d.ts.map

package/dist/plugin/mystatus.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"mystatus.d.ts","sourceRoot":"","sources":["../../plugin/mystatus.ts"],"names":[],"mappings":"AAAA;;;;;;;GAOG;AAKH,OAAO,EAAE,MAAM,EAAE,MAAM,kBAAkB,CAAC;;AAY1C,wBAiFG"}

package/dist/plugin/mystatus.js

@@ -0,0 +1,100 @@
+/**
+ * OpenCode 额度状态查询插件
+ *
+ * [输入]: ~/.local/share/opencode/auth.json 和 ~/.config/opencode/antigravity-accounts.json 中的认证信息
+ * [输出]: 带进度条的额度使用情况展示
+ * [定位]: 通过 mystatus 工具查询各账号额度
+ * [同步]: lib/openai.ts, lib/zhipu.ts, lib/google.ts, lib/types.ts, lib/i18n.ts
+ */
+import { readFile } from "node:fs/promises";
+import { homedir } from "node:os";
+import { join } from "node:path";
+import { Plugin } from "@opencode/plugin";
+import { queryCopilotUsage } from "./lib/copilot";
+import { queryGoogleUsage } from "./lib/google";
+import { t } from "./lib/i18n";
+import { queryOpenAIUsage } from "./lib/openai";
+import { queryZaiUsage, queryZhipuUsage } from "./lib/zhipu";
+// ============================================================================
+// V2 插件导出
+// ============================================================================
+export default Plugin.define({
+    id: "opencode-mystatus",
+    async setup(ctx) {
+        await ctx.tool.transform((editor) => {
+            editor.add({
+                name: "mystatus",
+                description: "Query account quota usage for all configured AI platforms. Returns remaining quota percentages, usage stats, and reset countdowns with visual progress bars. Currently supports OpenAI (ChatGPT/Codex), Zhipu AI, Z.ai, Google Antigravity, and GitHub Copilot.",
+                input: {
+                    type: "object",
+                    properties: {},
+                    additionalProperties: false,
+                },
+                async execute() {
+                    // 1. 读取 auth.json
+                    const authPath = join(homedir(), ".local/share/opencode/auth.json");
+                    let authData;
+                    try {
+                        const content = await readFile(authPath, "utf-8");
+                        authData = JSON.parse(content);
+                    }
+                    catch (err) {
+                        return {
+                            content: t.authError(authPath, err instanceof Error ? err.message : String(err)),
+                        };
+                    }
+                    // 2. 并行查询所有平台（Google 不依赖 authData）
+                    const [openaiResult, zhipuResult, zaiResult, googleResult, copilotResult,] = await Promise.all([
+                        queryOpenAIUsage(authData.openai),
+                        queryZhipuUsage(authData["zhipuai-coding-plan"]),
+                        queryZaiUsage(authData["zai-coding-plan"]),
+                        queryGoogleUsage(),
+                        queryCopilotUsage(authData["github-copilot"]),
+                    ]);
+                    // 3. 收集结果
+                    const results = [];
+                    const errors = [];
+                    // 处理 OpenAI 结果
+                    collectResult(openaiResult, t.openaiTitle, results, errors);
+                    // 处理智谱结果
+                    collectResult(zhipuResult, t.zhipuTitle, results, errors);
+                    // 处理 Z.ai 结果
+                    collectResult(zaiResult, t.zaiTitle, results, errors);
+                    // 处理 Google 结果
+                    collectResult(googleResult, t.googleTitle, results, errors);
+                    // 处理 GitHub Copilot 结果
+                    collectResult(copilotResult, t.copilotTitle, results, errors);
+                    // 4. 汇总输出
+                    if (results.length === 0 && errors.length === 0) {
+                        return { content: t.noAccounts };
+                    }
+                    let output = results.join("\n");
+                    if (errors.length > 0) {
+                        if (output)
+                            output += "\n\n";
+                        output += t.queryFailed + errors.join("\n");
+                    }
+                    return { content: output };
+                },
+            });
+        });
+    },
+});
+/**
+ * 收集查询结果到 results 和 errors 数组
+ * 注意：这是内部函数，不导出，避免被 OpenCode 当作插件加载
+ */
+function collectResult(result, title, results, errors) {
+    if (!result)
+        return;
+    if (result.success && result.output) {
+        if (results.length > 0)
+            results.push(""); // 分隔符
+        results.push(title);
+        results.push("");
+        results.push(result.output);
+    }
+    else if (result.error) {
+        errors.push(result.error);
+    }
+}

package/package.json

@@ -1,6 +1,65 @@
 {
   "name": "@ekworp/opencode-mystatus",
-  "version": "0.0.0-stage",
-  "stub": true,
-  "description": "Temporary package placeholder for staged publishing"
+  "version": "1.2.7",
+  "description": "Check all your AI subscription quotas in one command. Supports OpenAI, Zhipu AI, Z.ai, and Google Antigravity. More platforms coming soon.",
+  "type": "module",
+  "main": "dist/plugin/mystatus.js",
+  "types": "dist/plugin/mystatus.d.ts",
+  "exports": {
+    ".": {
+      "import": "./dist/plugin/mystatus.js",
+      "types": "./dist/plugin/mystatus.d.ts"
+    }
+  },
+  "files": [
+    "dist",
+    "command",
+    "README.md",
+    "LICENSE"
+  ],
+  "keywords": [
+    "opencode",
+    "plugin",
+    "quota",
+    "status",
+    "openai",
+    "zhipu",
+    "zai",
+    "google",
+    "antigravity",
+    "gemini",
+    "claude"
+  ],
+  "author": "javazys",
+  "license": "MIT",
+  "repository": {
+    "type": "git",
+    "url": "git+https://github.com/vbgate/opencode-mystatus.git"
+  },
+  "bugs": {
+    "url": "https://github.com/vbgate/opencode-mystatus/issues"
+  },
+  "homepage": "https://github.com/vbgate/opencode-mystatus#readme",
+  "dependencies": {
+    "@opencode/plugin": "2.0.25"
+  },
+  "devDependencies": {
+    "@biomejs/biome": "^2.5.15",
+    "@types/node": "^22.0.0",
+    "husky": "^9.1.7",
+    "typescript": "^5.8.0"
+  },
+  "peerDependencies": {
+    "@opencode/plugin": "2.0.25"
+  },
+  "engines": {
+    "node": ">=18.0.0"
+  },
+  "scripts": {
+    "build": "tsc",
+    "typecheck": "tsc --noEmit",
+    "lint": "biome check .",
+    "lint:fix": "biome check --write .",
+    "format": "biome format --write ."
+  }
 }