codex-slot 0.1.19 → 0.1.21

package/dist/status.js

@@ -8,6 +8,14 @@ const config_1 = require("./config");
 const account_store_1 = require("./account-store");
 const state_1 = require("./state");
 const text_1 = require("./text");
+const TABLE_ANSI = {
+    reset: "\x1b[0m",
+    dim: "\x1b[2m",
+    green: "\x1b[32m",
+    yellow: "\x1b[33m",
+    red: "\x1b[31m",
+    cyan: "\x1b[36m"
+};
 function computeLeftPercent(usedPercent) {
     if (usedPercent === null || usedPercent === undefined || Number.isNaN(usedPercent)) {
         return null;
@@ -30,23 +38,214 @@ function formatReset(unixSeconds) {
     return (0, text_1.formatLocalDateTime)(unixSeconds);
 }
 /**
- * 按给定最大宽度截断单元格文本，优先保证表格整体不换行。
+ * 移除 ANSI 控制序列，获得真实可见文本。
+ *
+ * 业务含义：
+ * 1. 交互界面会对状态列做轻量着色。
+ * 2. 表格宽度计算必须忽略颜色控制符，否则列宽会被错误拉大。
+ *
+ * @param value 可能包含 ANSI 样式的文本。
+ * @returns 去除 ANSI 控制序列后的可见文本。
+ * @throws 无显式抛出。
+ */
+function stripAnsi(value) {
+    return value.replace(/\x1b\[[0-9;]*m/g, "");
+}
+/**
+ * 按需给文本添加 ANSI 样式。
  *
  * @param value 原始文本。
- * @param maxWidth 最大宽度。
+ * @param color ANSI 颜色或样式控制符。
+ * @param styled 是否启用样式。
+ * @returns 启用样式时返回带 ANSI 控制符的文本，否则返回原文。
+ * @throws 无显式抛出。
+ */
+function styleCell(value, color, styled) {
+    if (!styled) {
+        return value;
+    }
+    return `${color}${value}${TABLE_ANSI.reset}`;
+}
+/**
+ * 判断字符是否应按双列宽展示。
+ *
+ * 业务含义：
+ * 1. 终端中中文、全角符号与多数 CJK 字符通常占两个显示列。
+ * 2. 表格列宽若只按字符串长度计算，会导致包含中文括号的账号名错位或过早截断。
+ *
+ * @param codePoint Unicode code point；必须来自单个字符迭代结果。
+ * @returns `true` 表示该字符应按双列宽计算；其他字符返回 `false`。
+ * @throws 无显式抛出。
+ */
+function isWideCodePoint(codePoint) {
+    return (codePoint >= 0x1100 &&
+        (codePoint <= 0x115f ||
+            codePoint === 0x2329 ||
+            codePoint === 0x232a ||
+            (codePoint >= 0x2e80 && codePoint <= 0xa4cf && codePoint !== 0x303f) ||
+            (codePoint >= 0xac00 && codePoint <= 0xd7a3) ||
+            (codePoint >= 0xf900 && codePoint <= 0xfaff) ||
+            (codePoint >= 0xfe10 && codePoint <= 0xfe19) ||
+            (codePoint >= 0xfe30 && codePoint <= 0xfe6f) ||
+            (codePoint >= 0xff00 && codePoint <= 0xff60) ||
+            (codePoint >= 0xffe0 && codePoint <= 0xffe6)));
+}
+/**
+ * 计算文本在常见等宽终端中的显示列宽。
+ *
+ * 业务含义：
+ * 1. 状态表需要根据终端列数分配可用空间。
+ * 2. 账号名可能包含中文日期括号，必须按显示宽度而不是 UTF-16 长度计算。
+ *
+ * @param value 待展示文本；允许为空字符串。
+ * @returns 文本占用的终端显示列数。
+ * @throws 无显式抛出。
+ */
+function getDisplayWidth(value) {
+    const visibleValue = stripAnsi(value);
+    let width = 0;
+    for (const char of visibleValue) {
+        const codePoint = char.codePointAt(0) ?? 0;
+        width += isWideCodePoint(codePoint) ? 2 : 1;
+    }
+    return width;
+}
+/**
+ * 按终端显示列宽截断文本。
+ *
+ * 业务含义：
+ * 1. 优先保证表格整体不换行。
+ * 2. 截断时保留省略号，让用户能看出内容未完全展示。
+ *
+ * @param value 原始文本。
+ * @param maxWidth 最大显示列宽；小于等于 0 时返回空字符串。
  * @returns 截断后的文本；宽度过小时退化为最短可读形式。
+ * @throws 无显式抛出。
  */
 function truncateCell(value, maxWidth) {
     if (maxWidth <= 0) {
         return "";
     }
-    if (value.length <= maxWidth) {
+    if (getDisplayWidth(value) <= maxWidth) {
         return value;
     }
     if (maxWidth <= 2) {
-        return value.slice(0, maxWidth);
+        let output = "";
+        let width = 0;
+        for (const char of value) {
+            const charWidth = getDisplayWidth(char);
+            if (width + charWidth > maxWidth) {
+                break;
+            }
+            output += char;
+            width += charWidth;
+        }
+        return output;
+    }
+    const ellipsis = "…";
+    const targetWidth = maxWidth - getDisplayWidth(ellipsis);
+    let output = "";
+    let width = 0;
+    for (const char of value) {
+        const charWidth = getDisplayWidth(char);
+        if (width + charWidth > targetWidth) {
+            break;
+        }
+        output += char;
+        width += charWidth;
+    }
+    return `${output}${ellipsis}`;
+}
+/**
+ * 按终端显示列宽补齐单元格。
+ *
+ * @param value 已完成截断的单元格文本。
+ * @param width 目标显示列宽。
+ * @returns 右侧补空格后的单元格文本。
+ * @throws 无显式抛出。
+ */
+function padCell(value, width) {
+    return `${value}${" ".repeat(Math.max(0, width - getDisplayWidth(value)))}`;
+}
+/**
+ * 根据账号状态给状态标签选择终端样式。
+ *
+ * 业务含义：
+ * 1. `available` 是调度可用状态，使用绿色强调。
+ * 2. 额度限制和短时熔断需要提醒但不一定是错误，使用黄色。
+ * 3. 工作空间损坏、账号缺失等不可用异常使用红色。
+ * 4. 禁用账号使用弱化样式，减少对可用账号的视觉干扰。
+ *
+ * @param status 已归一化的状态标签。
+ * @param item 单个账号状态。
+ * @param styled 是否启用 ANSI 样式。
+ * @returns 应用于表格状态列的文本。
+ * @throws 无显式抛出。
+ */
+function styleStatusCell(status, item, styled) {
+    if (!styled) {
+        return status;
+    }
+    if (item.isAvailable) {
+        return styleCell(status, TABLE_ANSI.green, styled);
+    }
+    if (!item.enabled) {
+        return styleCell(status, TABLE_ANSI.dim, styled);
+    }
+    if (item.refreshErrorCode || !item.exists) {
+        return styleCell(status, TABLE_ANSI.red, styled);
     }
-    return `${value.slice(0, maxWidth - 1)}…`;
+    if (item.isFiveHourLimited || item.isWeeklyLimited || item.localBlockUntil) {
+        return styleCell(status, TABLE_ANSI.yellow, styled);
+    }
+    return status;
+}
+/**
+ * 对当前自动选中账号的名称做轻量强调。
+ *
+ * @param name 账号展示名称。
+ * @param styled 是否启用 ANSI 样式。
+ * @returns 表格名称列展示文本。
+ * @throws 无显式抛出。
+ */
+function styleNameCell(name, styled) {
+    if (!styled || !name.endsWith("*")) {
+        return name;
+    }
+    return styleCell(name, TABLE_ANSI.cyan, styled);
+}
+/**
+ * 计算紧凑状态表中账号名称列可使用的显示宽度。
+ *
+ * 业务含义：
+ * 1. 窄终端下保留最小可读名称。
+ * 2. 终端变宽时优先把新增空间分配给账号名称，避免固定 12 列导致名字仍被截断。
+ *
+ * @param statuses 待展示账号状态。
+ * @param hasSelector 是否展示选择/启用状态列。
+ * @param maxWidth 当前终端最大显示列宽；无穷大表示不限制。
+ * @param headerWidth 当前账号列标题宽度。
+ * @param planWidth plan 列宽。
+ * @param statusWidth 状态列宽。
+ * @returns 账号名称列的目标显示宽度。
+ * @throws 无显式抛出。
+ */
+function resolveCompactSlotWidth(statuses, hasSelector, maxWidth, headerWidth, planWidth, statusWidth) {
+    const longestNameWidth = Math.max(headerWidth, ...statuses.map((item) => getDisplayWidth(item.name)));
+    if (!Number.isFinite(maxWidth)) {
+        return longestNameWidth;
+    }
+    const fixedColumnWidths = [
+        ...(hasSelector ? [4] : []),
+        planWidth,
+        3,
+        4,
+        statusWidth
+    ];
+    const separatorWidth = 2 * fixedColumnWidths.length;
+    const availableWidth = Math.floor(maxWidth) - fixedColumnWidths.reduce((sum, width) => sum + width, 0) - separatorWidth;
+    const minWidth = maxWidth < 56 ? 8 : 12;
+    return Math.max(Math.min(longestNameWidth, availableWidth), Math.min(minWidth, Math.max(4, availableWidth)));
 }
 /**
  * 生成固定标签宽度的详情行，超出终端宽度时自动截断值部分。
@@ -263,15 +462,17 @@ function renderStatusTable(statuses, options) {
     const selectorColumn = options?.selectorColumn;
     const compact = options?.compact ?? false;
     const maxWidth = options?.maxWidth ?? Number.POSITIVE_INFINITY;
+    const styled = options?.styled ?? false;
     const compactHeader = maxWidth < 68;
-    const compactSlotWidth = maxWidth < 56 ? 8 : 12;
     const compactPlanWidth = maxWidth < 56 ? 4 : 6;
     const compactStatusWidth = maxWidth < 56 ? 12 : 18;
+    const compactSlotHeader = compactHeader ? "ID" : "SLOT";
+    const compactSlotWidth = resolveCompactSlotWidth(statuses, Boolean(selectorColumn), maxWidth, getDisplayWidth(compactSlotHeader), compactPlanWidth, compactStatusWidth);
     const rows = [
         compact
             ? [
                 ...(selectorColumn ? [" "] : []),
-                compactHeader ? "ID" : "SLOT",
+                compactSlotHeader,
                 compactHeader ? "P" : "PLAN",
                 "5H",
                 compactHeader ? "WK" : "WEEK",
@@ -297,27 +498,27 @@ function renderStatusTable(statuses, options) {
         rows.push(compact
             ? [
                 ...(selectorCell ? [selectorCell] : []),
-                truncateCell(item.name, compactSlotWidth),
+                styleNameCell(truncateCell(item.name, compactSlotWidth), styled),
                 truncateCell(item.plan, compactPlanWidth),
                 formatPercent(item.fiveHourLeftPercent),
                 formatPercent(item.weeklyLeftPercent),
-                truncateCell(status, compactStatusWidth)
+                styleStatusCell(truncateCell(status, compactStatusWidth), item, styled)
             ]
             : [
                 ...(selectorCell ? [selectorCell] : []),
-                item.name,
+                styleNameCell(item.name, styled),
                 item.email ?? "-",
                 item.plan,
                 formatPercent(item.fiveHourLeftPercent),
                 formatReset(item.fiveHourResetsAt),
                 formatPercent(item.weeklyLeftPercent),
                 formatReset(item.weeklyResetsAt),
-                status
+                styleStatusCell(status, item, styled)
             ]);
     }
-    const widths = rows[0].map((_, columnIndex) => Math.max(...rows.map((row) => row[columnIndex].length)));
+    const widths = rows[0].map((_, columnIndex) => Math.max(...rows.map((row) => getDisplayWidth(row[columnIndex]))));
     return rows
-        .map((row) => row.map((cell, index) => cell.padEnd(widths[index])).join("  "))
+        .map((row) => row.map((cell, index) => padCell(cell, widths[index])).join("  "))
         .join("\n");
 }
 /**

package/dist/upstream-client.js

@@ -0,0 +1,63 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.buildUpstreamHeaders = buildUpstreamHeaders;
+exports.sendCodexResponsesRequest = sendCodexResponsesRequest;
+const undici_1 = require("undici");
+/**
+ * 构造发往上游的请求头，并移除仅属于本地代理链路的头信息。
+ *
+ * 业务含义：
+ * 1. 本地服务使用独立 api_key 鉴权，转发时必须替换为真实上游 access token。
+ * 2. body 会在本地先读取成 Buffer 以支持失败后切换账号重试，因此这里重算 content-length。
+ *
+ * @param requestHeaders 客户端发到本地服务的原始请求头。
+ * @param accessToken 当前候选账号可用的上游访问令牌。
+ * @param bodyLength 请求体字节长度。
+ * @param accountIdHeader 可选的 ChatGPT 账号标识头。
+ * @returns 可直接传给上游请求的请求头对象。
+ * @throws 无显式抛出。
+ */
+function buildUpstreamHeaders(requestHeaders, accessToken, bodyLength, accountIdHeader) {
+    const headers = {};
+    for (const [headerName, headerValue] of Object.entries(requestHeaders)) {
+        const normalizedName = headerName.toLowerCase();
+        if (headerValue == null ||
+            normalizedName === "authorization" ||
+            normalizedName === "host" ||
+            normalizedName === "connection" ||
+            normalizedName === "content-length") {
+            continue;
+        }
+        headers[normalizedName] = Array.isArray(headerValue)
+            ? headerValue.join(", ")
+            : headerValue;
+    }
+    headers.authorization = `Bearer ${accessToken}`;
+    if (!headers.accept) {
+        headers.accept = "text/event-stream, application/json";
+    }
+    headers["content-length"] = String(bodyLength);
+    headers["user-agent"] = "codex-slot/0.1.1";
+    if (accountIdHeader) {
+        headers["chatgpt-account-id"] = accountIdHeader;
+    }
+    return headers;
+}
+/**
+ * 向 Codex responses 上游发送一次请求。
+ *
+ * 业务含义：
+ * 1. 该 Adapter 隔离 undici 与上游 URL 细节，代理重试服务不直接依赖 HTTP 客户端实现。
+ * 2. 调用方负责决定 access token、账号切换与失败策略。
+ *
+ * @param options 上游请求参数。
+ * @returns undici 上游响应对象。
+ * @throws 当网络层或 undici 请求失败时透传底层异常。
+ */
+async function sendCodexResponsesRequest(options) {
+    return await (0, undici_1.request)(`${options.codexBaseUrl}/responses`, {
+        method: "POST",
+        headers: buildUpstreamHeaders(options.requestHeaders, options.accessToken, options.body.length, options.accountIdHeader),
+        body: options.body
+    });
+}

package/dist/upstream-error-policy.js

@@ -0,0 +1,114 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.resolveBlockWindow = resolveBlockWindow;
+exports.extractErrorCode = extractErrorCode;
+exports.isNetworkUnavailableError = isNetworkUnavailableError;
+exports.buildNetworkUnavailablePayload = buildNetworkUnavailablePayload;
+exports.isUsageLimitErrorText = isUsageLimitErrorText;
+const text_1 = require("./text");
+/**
+ * 根据错误文本与当前账号状态，决定本地禁用时长。
+ *
+ * 业务规则：
+ * 1. 周限制优先，直到周窗口重置时间。
+ * 2. 5 小时额度限制次之，直到 5 小时窗口重置时间。
+ * 3. 未能明确识别时，按 5 分钟临时熔断处理。
+ *
+ * @param picked 当前被选中的账号及状态。
+ * @param errorText 上游返回的错误文本。
+ * @returns 本地禁用窗口与原因。
+ * @throws 无显式抛出。
+ */
+function resolveBlockWindow(picked, errorText) {
+    const lowerText = errorText.toLowerCase();
+    if (lowerText.includes("weekly") ||
+        lowerText.includes("7 day") ||
+        lowerText.includes("7-day") ||
+        picked.status.isWeeklyLimited) {
+        return {
+            until: picked.status.weeklyResetsAt ?? Math.floor(Date.now() / 1000) + 7 * 24 * 60 * 60,
+            reason: "weekly_limited"
+        };
+    }
+    if (lowerText.includes("5 hour") ||
+        lowerText.includes("5-hour") ||
+        lowerText.includes("5h") ||
+        lowerText.includes("usage limit") ||
+        picked.status.isFiveHourLimited) {
+        return {
+            until: picked.status.fiveHourResetsAt ?? Math.floor(Date.now() / 1000) + 5 * 60,
+            reason: "5h_limited"
+        };
+    }
+    return {
+        until: Math.floor(Date.now() / 1000) + 5 * 60,
+        reason: "temporary_5m_limit"
+    };
+}
+/**
+ * 提取错误对象中最接近底层网络层的错误码。
+ *
+ * @param error 捕获到的异常对象。
+ * @returns 错误码；若无法识别则返回 `null`。
+ * @throws 无显式抛出。
+ */
+function extractErrorCode(error) {
+    if (!error || typeof error !== "object") {
+        return null;
+    }
+    const errnoError = error;
+    if (typeof errnoError.code === "string" && errnoError.code.length > 0) {
+        return errnoError.code;
+    }
+    return extractErrorCode(errnoError.cause);
+}
+/**
+ * 判断一次请求失败是否属于本机到上游之间的网络不可达场景。
+ *
+ * @param error 捕获到的异常对象。
+ * @returns `true` 表示网络层异常，不应写入账号熔断；否则返回 `false`。
+ * @throws 无显式抛出。
+ */
+function isNetworkUnavailableError(error) {
+    const errorCode = extractErrorCode(error);
+    return [
+        "ECONNREFUSED",
+        "ECONNRESET",
+        "EHOSTUNREACH",
+        "ENETDOWN",
+        "ENETUNREACH",
+        "ENOTFOUND",
+        "EAI_AGAIN",
+        "ETIMEDOUT",
+        "UND_ERR_CONNECT_TIMEOUT",
+        "UND_ERR_SOCKET"
+    ].includes(errorCode ?? "");
+}
+/**
+ * 将网络层异常转换为统一的响应体，避免误导成“当前没有可用账号”。
+ *
+ * @param accountId 当前尝试的账号标识。
+ * @param error 捕获到的异常对象。
+ * @returns 统一的网络异常响应体。
+ * @throws 无显式抛出。
+ */
+function buildNetworkUnavailablePayload(accountId, error) {
+    const message = error instanceof Error ? error.message : String(error);
+    return {
+        error: {
+            message: (0, text_1.bi)(`网络不可用，账号 ${accountId} 无法连接上游: ${message}`, `Network unavailable. Account ${accountId} cannot reach upstream: ${message}`),
+            type: "network_unavailable"
+        }
+    };
+}
+/**
+ * 判断错误响应文本是否表示上游额度限制。
+ *
+ * @param errorText 上游返回的错误文本。
+ * @returns `true` 表示可按额度限制处理并切换账号。
+ * @throws 无显式抛出。
+ */
+function isUsageLimitErrorText(errorText) {
+    const lowerText = errorText.toLowerCase();
+    return lowerText.includes("usage limit") || lowerText.includes("try again later");
+}

package/dist/usage-sync.js

@@ -12,6 +12,12 @@ const state_1 = require("./state");
 const text_1 = require("./text");
 const USAGE_CACHE_TTL_MS = 60 * 1000;
 const inflightUsageRefreshes = new Map();
+const SHORT_LIVED_ACCOUNT_BLOCK_REASONS = new Set([
+    "request_failed",
+    "upstream_5xx",
+    "temporary_5m_limit",
+    "token_refresh_failed"
+]);
 function normalizeResetAt(value, resetAfterSeconds) {
     if (typeof value === "number" && Number.isFinite(value)) {
         return value;
@@ -21,6 +27,21 @@ function normalizeResetAt(value, resetAfterSeconds) {
     }
     return null;
 }
+/**
+ * 当账号已经成功完成鉴权或额度刷新时，清理与瞬时异常相关的本地熔断。
+ *
+ * 只会移除短期失败类熔断，不会误清理 5 小时或周额度限制。
+ *
+ * @param accountId 账号标识。
+ * @returns 无返回值。
+ */
+function clearShortLivedAccountBlock(accountId) {
+    const block = (0, state_1.getAccountBlock)(accountId);
+    if (!block || !SHORT_LIVED_ACCOUNT_BLOCK_REASONS.has(block.reason)) {
+        return;
+    }
+    (0, state_1.clearAccountBlock)(accountId);
+}
 /**
  * 将额度刷新异常归类为可直接展示在 `status` 表格中的状态码。
  *
@@ -96,6 +117,7 @@ async function refreshAccountTokens(accountId) {
         last_refresh: new Date().toISOString()
     };
     (0, account_store_1.writeAuthFile)(account.codex_home, nextAuth);
+    clearShortLivedAccountBlock(accountId);
     return nextAuth;
 }
 /**
@@ -149,6 +171,7 @@ async function refreshAccountUsage(accountId) {
     };
     (0, state_1.setUsageCache)(result);
     (0, state_1.clearUsageRefreshError)(accountId);
+    clearShortLivedAccountBlock(accountId);
     return result;
 }
 /**

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "codex-slot",
-  "version": "0.1.19",
+  "version": "0.1.21",
   "description": "本地 Codex 多账号切换与状态管理工具",
   "type": "commonjs",
   "main": "dist/cli.js",
@@ -12,8 +12,8 @@
     "dist"
   ],
   "scripts": {
-    "clean": "rm -rf dist",
-    "build": "npm run clean && tsc -p tsconfig.json && chmod +x dist/cli.js dist/serve.js",
+    "clean": "node -e \"require('node:fs').rmSync('dist',{recursive:true,force:true})\"",
+    "build": "npm run clean && tsc -p tsconfig.json && node -e \"const fs=require('node:fs'); for (const file of ['dist/cli.js','dist/serve.js']) { try { fs.chmodSync(file, 0o755); } catch {} }\"",
     "prepublishOnly": "npm run build",
     "dev": "tsx src/cli.ts",
     "check": "tsc --noEmit -p tsconfig.json",