@openxiaobu/codexl 0.1.0

package/dist/status.js

@@ -0,0 +1,134 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.collectAccountStatuses = collectAccountStatuses;
+exports.renderStatusTable = renderStatusTable;
+const node_fs_1 = __importDefault(require("node:fs"));
+const node_path_1 = __importDefault(require("node:path"));
+const config_1 = require("./config");
+const account_store_1 = require("./account-store");
+const state_1 = require("./state");
+function computeLeftPercent(usedPercent) {
+    if (usedPercent === null || usedPercent === undefined || Number.isNaN(usedPercent)) {
+        return null;
+    }
+    return Math.max(0, Math.min(100, 100 - usedPercent));
+}
+function isLimited(usedPercent, resetsAt) {
+    if (usedPercent === null || usedPercent < 100) {
+        return false;
+    }
+    if (!resetsAt) {
+        return true;
+    }
+    return resetsAt * 1000 > Date.now();
+}
+function formatPercent(value) {
+    return value === null ? "-" : `${value}%`;
+}
+function formatReset(unixSeconds) {
+    if (!unixSeconds) {
+        return "-";
+    }
+    return new Date(unixSeconds * 1000).toLocaleString("zh-CN", {
+        hour12: false
+    });
+}
+/**
+ * 汇总所有受管账号的运行状态，供状态展示与调度复用。
+ *
+ * @returns 所有账号的运行时状态列表。
+ */
+function collectAccountStatuses() {
+    const config = (0, config_1.loadConfig)();
+    return config.accounts.map((account) => {
+        const codexDir = (0, account_store_1.getCodexDataDir)(account.codex_home);
+        const registryPath = node_path_1.default.join(codexDir, "accounts", "registry.json");
+        const exists = node_fs_1.default.existsSync(registryPath);
+        const primary = exists ? (0, account_store_1.resolvePrimaryRegistryAccount)(account.codex_home) : null;
+        const usageCache = (0, state_1.getUsageCache)(account.id);
+        const activeEmail = usageCache?.email ?? primary?.email ?? account.email;
+        const fiveHourUsed = usageCache?.fiveHourUsedPercent ?? null;
+        const fiveHourReset = usageCache?.fiveHourResetAt ?? null;
+        const weeklyUsed = usageCache?.weeklyUsedPercent ?? null;
+        const weeklyReset = usageCache?.weeklyResetAt ?? null;
+        const fiveHourLeftPercent = computeLeftPercent(fiveHourUsed);
+        const weeklyLeftPercent = computeLeftPercent(weeklyUsed);
+        const isFiveHourLimited = isLimited(fiveHourUsed, fiveHourReset);
+        const isWeeklyLimited = isLimited(weeklyUsed, weeklyReset);
+        const localBlock = (0, state_1.getAccountBlock)(account.id);
+        const localBlocked = localBlock?.until != null ? localBlock.until * 1000 > Date.now() : false;
+        return {
+            id: account.id,
+            name: account.name,
+            email: activeEmail,
+            enabled: account.enabled,
+            exists,
+            plan: usageCache?.plan ?? primary?.plan ?? "-",
+            fiveHourLeftPercent,
+            fiveHourResetsAt: fiveHourReset,
+            weeklyLeftPercent,
+            weeklyResetsAt: weeklyReset,
+            isFiveHourLimited,
+            isWeeklyLimited,
+            localBlockReason: localBlock?.reason,
+            localBlockUntil: localBlock?.until ?? null,
+            isAvailable: account.enabled &&
+                exists &&
+                !isFiveHourLimited &&
+                !isWeeklyLimited &&
+                !localBlocked,
+            sourcePath: codexDir
+        };
+    });
+}
+/**
+ * 将账号状态渲染为适合终端输出的表格文本。
+ *
+ * @param statuses 待展示的账号状态列表。
+ * @returns 可直接打印到终端的表格字符串。
+ */
+function renderStatusTable(statuses) {
+    const rows = [
+        ["NAME", "EMAIL", "PLAN", "5H_LEFT", "5H_RESET", "WEEK_LEFT", "WEEK_RESET", "STATUS"]
+    ];
+    for (const item of statuses) {
+        let status = "missing";
+        if (item.exists) {
+            if (!item.enabled) {
+                status = "disabled";
+            }
+            else if (item.localBlockUntil && item.localBlockUntil * 1000 > Date.now()) {
+                status = item.localBlockReason ?? "blocked";
+            }
+            else if (item.isWeeklyLimited) {
+                status = "weekly_limited";
+            }
+            else if (item.isFiveHourLimited) {
+                status = "cooldown";
+            }
+            else if (item.isAvailable) {
+                status = "available";
+            }
+            else {
+                status = "unknown";
+            }
+        }
+        rows.push([
+            item.name,
+            item.email ?? "-",
+            item.plan,
+            formatPercent(item.fiveHourLeftPercent),
+            formatReset(item.fiveHourResetsAt),
+            formatPercent(item.weeklyLeftPercent),
+            formatReset(item.weeklyResetsAt),
+            status
+        ]);
+    }
+    const widths = rows[0].map((_, columnIndex) => Math.max(...rows.map((row) => row[columnIndex].length)));
+    return rows
+        .map((row) => row.map((cell, index) => cell.padEnd(widths[index])).join("  "))
+        .join("\n");
+}

package/dist/types.js

@@ -0,0 +1,2 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });

package/dist/usage-sync.js

@@ -0,0 +1,143 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.refreshAccountTokens = refreshAccountTokens;
+exports.refreshAccountUsage = refreshAccountUsage;
+exports.refreshAllAccountUsage = refreshAllAccountUsage;
+const undici_1 = require("undici");
+const account_store_1 = require("./account-store");
+const config_1 = require("./config");
+const state_1 = require("./state");
+function normalizeResetAt(value, resetAfterSeconds) {
+    if (typeof value === "number" && Number.isFinite(value)) {
+        return value;
+    }
+    if (typeof resetAfterSeconds === "number" && Number.isFinite(resetAfterSeconds)) {
+        return Math.floor(Date.now() / 1000) + resetAfterSeconds;
+    }
+    return null;
+}
+/**
+ * 使用 refresh token 刷新指定账号的 access token，并回写到账号目录。
+ *
+ * @param accountId 账号标识。
+ * @returns 最新认证信息。
+ * @throws 当账号不存在、缺少 refresh_token 或刷新失败时抛出错误。
+ */
+async function refreshAccountTokens(accountId) {
+    const config = (0, config_1.loadConfig)();
+    const account = (0, account_store_1.findManagedAccount)(accountId);
+    if (!account) {
+        throw new Error(`未找到账号 ${accountId}`);
+    }
+    const auth = (0, account_store_1.readAuthFile)(account.codex_home);
+    const refreshToken = auth?.tokens?.refresh_token;
+    if (!refreshToken) {
+        throw new Error(`账号 ${accountId} 缺少 refresh_token`);
+    }
+    const response = await (0, undici_1.request)(`${config.upstream.auth_base_url}/oauth/token`, {
+        method: "POST",
+        headers: {
+            "content-type": "application/x-www-form-urlencoded",
+            accept: "application/json"
+        },
+        body: new URLSearchParams({
+            grant_type: "refresh_token",
+            refresh_token: refreshToken,
+            client_id: config.upstream.oauth_client_id
+        }).toString()
+    });
+    if (response.statusCode >= 400) {
+        const errorText = await response.body.text();
+        throw new Error(`刷新 token 失败: HTTP ${response.statusCode} ${errorText}`);
+    }
+    const payload = (await response.body.json());
+    const nextAuth = {
+        ...(auth ?? {}),
+        auth_mode: "chatgpt",
+        OPENAI_API_KEY: null,
+        tokens: {
+            ...(auth?.tokens ?? {}),
+            access_token: payload.access_token ?? auth?.tokens?.access_token,
+            refresh_token: payload.refresh_token ?? auth?.tokens?.refresh_token,
+            id_token: payload.id_token ?? auth?.tokens?.id_token,
+            account_id: auth?.tokens?.account_id
+        },
+        last_refresh: new Date().toISOString()
+    };
+    (0, account_store_1.writeAuthFile)(account.codex_home, nextAuth);
+    return nextAuth;
+}
+/**
+ * 查询单个账号的最新额度信息，并写入 codexl 自己的 usage 缓存。
+ *
+ * @param accountId 账号标识。
+ * @returns 刷新后的额度摘要。
+ * @throws 当账号不存在、未登录或远端请求失败时抛出错误。
+ */
+async function refreshAccountUsage(accountId) {
+    const account = (0, account_store_1.findManagedAccount)(accountId);
+    if (!account) {
+        throw new Error(`未找到账号 ${accountId}`);
+    }
+    const auth = (0, account_store_1.readAuthFile)(account.codex_home);
+    const accessToken = auth?.tokens?.access_token;
+    const accountIdHeader = auth?.tokens?.account_id;
+    if (!accessToken) {
+        throw new Error(`账号 ${accountId} 缺少 access_token`);
+    }
+    const response = await (0, undici_1.request)("https://chatgpt.com/backend-api/wham/usage", {
+        method: "GET",
+        headers: {
+            authorization: `Bearer ${accessToken}`,
+            accept: "application/json",
+            "user-agent": "codexl/0.1.0",
+            ...(accountIdHeader ? { "chatgpt-account-id": accountIdHeader } : {})
+        }
+    });
+    if (response.statusCode === 401) {
+        await refreshAccountTokens(accountId);
+        return await refreshAccountUsage(accountId);
+    }
+    if (response.statusCode >= 400) {
+        const errorText = await response.body.text();
+        throw new Error(`刷新额度失败: HTTP ${response.statusCode} ${errorText}`);
+    }
+    const payload = (await response.body.json());
+    const primary = (0, account_store_1.resolvePrimaryRegistryAccount)(account.codex_home);
+    const email = primary?.email ?? account.email ?? undefined;
+    const plan = payload.plan_type ?? primary?.plan ?? "-";
+    const result = {
+        accountId: account.id,
+        email,
+        plan,
+        fiveHourUsedPercent: payload.rate_limit?.primary_window?.used_percent ?? null,
+        fiveHourResetAt: normalizeResetAt(payload.rate_limit?.primary_window?.reset_at, payload.rate_limit?.primary_window?.reset_after_seconds),
+        weeklyUsedPercent: payload.rate_limit?.secondary_window?.used_percent ?? null,
+        weeklyResetAt: normalizeResetAt(payload.rate_limit?.secondary_window?.reset_at, payload.rate_limit?.secondary_window?.reset_after_seconds)
+    };
+    (0, state_1.setUsageCache)(result);
+    return result;
+}
+/**
+ * 批量刷新所有受管账号的额度信息。
+ *
+ * @returns 每个账号对应的刷新结果列表。
+ */
+async function refreshAllAccountUsage() {
+    const config = (0, config_1.loadConfig)();
+    const results = [];
+    for (const account of config.accounts) {
+        if (!account.enabled) {
+            continue;
+        }
+        try {
+            const result = await refreshAccountUsage(account.id);
+            results.push(result);
+        }
+        catch (error) {
+            const message = error instanceof Error ? error.message : String(error);
+            console.error(`[refresh] ${account.id} 失败: ${message}`);
+        }
+    }
+    return results;
+}

package/package.json

@@ -0,0 +1,47 @@
+{
+  "name": "@openxiaobu/codexl",
+  "version": "0.1.0",
+  "description": "本地 Codex 多账号切换与状态管理工具",
+  "type": "commonjs",
+  "main": "dist/cli.js",
+  "bin": {
+    "codexl": "dist/cli.js"
+  },
+  "files": [
+    "dist"
+  ],
+  "scripts": {
+    "clean": "rm -rf dist",
+    "build": "tsc -p tsconfig.json && chmod +x dist/cli.js dist/serve.js",
+    "dev": "tsx src/cli.ts",
+    "check": "tsc --noEmit -p tsconfig.json"
+  },
+  "keywords": [
+    "codex",
+    "chatgpt",
+    "switcher",
+    "cli"
+  ],
+  "author": "bk",
+  "license": "MIT",
+  "repository": {
+    "type": "git",
+    "url": "git+https://github.com/openxiaobu/codexl.git"
+  },
+  "homepage": "https://github.com/openxiaobu/codexl#readme",
+  "bugs": {
+    "url": "https://github.com/openxiaobu/codexl/issues"
+  },
+  "dependencies": {
+    "commander": "^14.0.3",
+    "fastify": "^5.8.2",
+    "undici": "^7.22.0",
+    "yaml": "^2.8.2",
+    "zod": "^4.3.6"
+  },
+  "devDependencies": {
+    "@types/node": "^25.4.0",
+    "tsx": "^4.21.0",
+    "typescript": "^5.9.3"
+  }
+}