codex-slot 0.1.0

package/dist/status.js

@@ -0,0 +1,199 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.collectAccountStatuses = collectAccountStatuses;
+exports.renderStatusTable = renderStatusTable;
+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
+    });
+}
+function formatLimitStatus(label, resetAt) {
+    const remaining = formatRemainingDuration(resetAt);
+    if (!remaining) {
+        return label;
+    }
+    return `${label}(${remaining})`;
+}
+function normalizeBlockReason(reason) {
+    if (!reason) {
+        return "blocked";
+    }
+    if (reason === "five_hour_limited") {
+        return "5h_limited";
+    }
+    return reason;
+}
+/**
+ * 将剩余秒数格式化为紧凑的人类可读文本，便于在状态列中展示熔断剩余时间。
+ *
+ * @param unixSeconds 熔断截止时间，Unix 秒时间戳。
+ * @returns 格式化后的剩余时长；当时间为空或已过期时返回 `null`。
+ */
+function formatRemainingDuration(unixSeconds) {
+    if (!unixSeconds) {
+        return null;
+    }
+    const diffSeconds = unixSeconds - Math.floor(Date.now() / 1000);
+    if (diffSeconds <= 0) {
+        return null;
+    }
+    const hours = Math.floor(diffSeconds / 3600);
+    const minutes = Math.floor((diffSeconds % 3600) / 60);
+    const seconds = diffSeconds % 60;
+    if (hours > 0) {
+        return `${hours}h${minutes}m`;
+    }
+    if (minutes > 0) {
+        return `${minutes}m${seconds}s`;
+    }
+    return `${seconds}s`;
+}
+/**
+ * 将本地熔断原因与剩余时间格式化为更直观的状态文本。
+ *
+ * @param reason 熔断原因。
+ * @param until 熔断截止时间，Unix 秒时间戳。
+ * @returns 适合在终端表格中展示的状态文本。
+ */
+function formatBlockedStatus(reason, until) {
+    const label = normalizeBlockReason(reason);
+    const remaining = formatRemainingDuration(until ?? null);
+    if (!remaining) {
+        return label;
+    }
+    return `${label}(${remaining})`;
+}
+/**
+ * 汇总所有受管账号的运行状态，供状态展示与调度复用。
+ *
+ * @returns 所有账号的运行时状态列表。
+ */
+function collectAccountStatuses() {
+    const config = (0, config_1.loadConfig)();
+    return config.accounts.map((account) => {
+        const exists = (0, account_store_1.hasCompleteCodexAuthState)(account.codex_home);
+        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: account.codex_home
+        };
+    });
+}
+/**
+ * 将账号状态渲染为适合终端输出的表格文本。
+ *
+ * @param statuses 待展示的账号状态列表。
+ * @param options 渲染选项；交互模式下可传入选择列配置，将勾选状态与当前光标合并到表格首列。
+ * @returns 可直接打印到终端的表格字符串。
+ */
+function renderStatusTable(statuses, options) {
+    const selectorColumn = options?.selectorColumn;
+    const rows = [
+        [
+            ...(selectorColumn ? [" "] : []),
+            "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 = formatBlockedStatus(item.localBlockReason, item.localBlockUntil);
+            }
+            else if (item.isWeeklyLimited) {
+                status = formatLimitStatus("weekly_limited", item.weeklyResetsAt);
+            }
+            else if (item.isFiveHourLimited) {
+                status = formatLimitStatus("5h_limited", item.fiveHourResetsAt);
+            }
+            else if (item.isAvailable) {
+                status = "available";
+            }
+            else {
+                status = "unknown";
+            }
+        }
+        const selectorCell = selectorColumn
+            ? `${selectorColumn.cursorAccountId === item.id ? ">" : " "}[${selectorColumn.enabledById[item.id] ? "x" : " "}]`
+            : null;
+        rows.push([
+            ...(selectorCell ? [selectorCell] : []),
+            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,189 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.refreshAccountTokens = refreshAccountTokens;
+exports.refreshAccountUsage = refreshAccountUsage;
+exports.isUsageCacheStale = isUsageCacheStale;
+exports.refreshAccountUsageInBackgroundIfNeeded = refreshAccountUsageInBackgroundIfNeeded;
+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");
+const USAGE_CACHE_TTL_MS = 60 * 1000;
+const inflightUsageRefreshes = new Map();
+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;
+}
+/**
+ * 查询单个账号的最新额度信息，并写入 cslot 自己的 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": "codex-slot/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),
+        refreshedAt: new Date().toISOString()
+    };
+    (0, state_1.setUsageCache)(result);
+    return result;
+}
+/**
+ * 判断指定账号的额度缓存是否已经过期。
+ *
+ * @param accountId 账号标识。
+ * @returns `true` 表示不存在缓存或缓存已超过 TTL，需要重新刷新；`false` 表示缓存仍可直接复用。
+ */
+function isUsageCacheStale(accountId) {
+    const usageCache = (0, state_1.getUsageCache)(accountId);
+    if (!usageCache?.refreshedAt) {
+        return true;
+    }
+    const refreshedAt = Date.parse(usageCache.refreshedAt);
+    if (Number.isNaN(refreshedAt)) {
+        return true;
+    }
+    return Date.now() - refreshedAt > USAGE_CACHE_TTL_MS;
+}
+/**
+ * 在不阻塞主请求链路的前提下，按需异步刷新指定账号的额度缓存。
+ *
+ * @param accountId 账号标识。
+ * @returns 无返回值；若缓存仍在 TTL 内或已有刷新任务进行中则直接跳过。
+ */
+function refreshAccountUsageInBackgroundIfNeeded(accountId) {
+    if (!isUsageCacheStale(accountId) || inflightUsageRefreshes.has(accountId)) {
+        return;
+    }
+    // 同一账号同一时刻只保留一个后台刷新任务，避免高并发下重复打远端 usage 接口。
+    const refreshTask = (async () => {
+        try {
+            await refreshAccountUsage(accountId);
+        }
+        catch {
+            // 后台刷新失败时保留旧缓存，由正式转发请求中的错误处理继续兜底。
+        }
+        finally {
+            inflightUsageRefreshes.delete(accountId);
+        }
+    })();
+    inflightUsageRefreshes.set(accountId, refreshTask);
+}
+/**
+ * 批量刷新所有受管账号的额度信息。
+ *
+ * @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,49 @@
+{
+  "name": "codex-slot",
+  "version": "0.1.0",
+  "description": "本地 Codex 多账号切换与状态管理工具",
+  "type": "commonjs",
+  "main": "dist/cli.js",
+  "bin": {
+    "codex-slot": "dist/cli.js",
+    "cslot": "dist/cli.js"
+  },
+  "files": [
+    "dist"
+  ],
+  "scripts": {
+    "clean": "rm -rf dist",
+    "build": "tsc -p tsconfig.json && chmod +x dist/cli.js dist/serve.js",
+    "prepublishOnly": "npm run build",
+    "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/cslot.git"
+  },
+  "homepage": "https://github.com/openxiaobu/cslot#readme",
+  "bugs": {
+    "url": "https://github.com/openxiaobu/cslot/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"
+  }
+}