codex-slot 0.1.0

package/dist/config.js

@@ -0,0 +1,205 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.getCodexSwHome = getCodexSwHome;
+exports.getConfigPath = getConfigPath;
+exports.getPidPath = getPidPath;
+exports.getServiceLogPath = getServiceLogPath;
+exports.expandHome = expandHome;
+exports.loadConfig = loadConfig;
+exports.saveConfig = saveConfig;
+exports.getManagedHome = getManagedHome;
+exports.upsertAccount = upsertAccount;
+const node_fs_1 = __importDefault(require("node:fs"));
+const node_os_1 = __importDefault(require("node:os"));
+const node_path_1 = __importDefault(require("node:path"));
+const yaml_1 = __importDefault(require("yaml"));
+const zod_1 = require("zod");
+const managedAccountSchema = zod_1.z.object({
+    id: zod_1.z.string().min(1),
+    name: zod_1.z.string().min(1),
+    codex_home: zod_1.z.string().min(1),
+    email: zod_1.z.string().email().optional(),
+    enabled: zod_1.z.boolean().default(true),
+    imported_at: zod_1.z.string().optional()
+});
+const configSchema = zod_1.z.object({
+    version: zod_1.z.number().int().default(1),
+    server: zod_1.z
+        .object({
+        host: zod_1.z.string().default("127.0.0.1"),
+        port: zod_1.z.number().int().default(4389),
+        api_key: zod_1.z.string().default("cslot-defaultkey"),
+        body_limit_mb: zod_1.z.number().positive().default(512)
+    })
+        .default({
+        host: "127.0.0.1",
+        port: 4389,
+        api_key: "cslot-defaultkey",
+        body_limit_mb: 512
+    }),
+    upstream: zod_1.z
+        .object({
+        codex_base_url: zod_1.z.string().default("https://chatgpt.com/backend-api/codex"),
+        auth_base_url: zod_1.z.string().default("https://auth.openai.com"),
+        oauth_client_id: zod_1.z.string().default("app_EMoamEEZ73f0CkXaXp7hrann")
+    })
+        .default({
+        codex_base_url: "https://chatgpt.com/backend-api/codex",
+        auth_base_url: "https://auth.openai.com",
+        oauth_client_id: "app_EMoamEEZ73f0CkXaXp7hrann"
+    }),
+    accounts: zod_1.z.array(managedAccountSchema).default([])
+});
+/**
+ * 返回 cslot 的根目录，并确保基础目录结构存在。
+ *
+ * @returns cslot 根目录绝对路径。
+ * @throws 当目录无法创建时抛出文件系统错误。
+ */
+function getCodexSwHome() {
+    const home = node_path_1.default.join(node_os_1.default.homedir(), ".cslot");
+    const legacyHome = node_path_1.default.join(node_os_1.default.homedir(), ".codexsw");
+    if (!node_fs_1.default.existsSync(home) && node_fs_1.default.existsSync(legacyHome)) {
+        node_fs_1.default.cpSync(legacyHome, home, { recursive: true });
+    }
+    // 先创建 cslot 根目录，后续命令统一基于该目录读写状态。
+    node_fs_1.default.mkdirSync(home, { recursive: true });
+    node_fs_1.default.mkdirSync(node_path_1.default.join(home, "homes"), { recursive: true });
+    node_fs_1.default.mkdirSync(node_path_1.default.join(home, "logs"), { recursive: true });
+    return home;
+}
+/**
+ * 返回 cslot 配置文件路径。
+ *
+ * @returns 配置文件绝对路径。
+ */
+function getConfigPath() {
+    return node_path_1.default.join(getCodexSwHome(), "config.yaml");
+}
+/**
+ * 返回后台服务 PID 文件路径。
+ *
+ * @returns PID 文件绝对路径。
+ */
+function getPidPath() {
+    return node_path_1.default.join(getCodexSwHome(), "cslot.pid");
+}
+/**
+ * 返回后台服务日志文件路径。
+ *
+ * @returns 日志文件绝对路径。
+ */
+function getServiceLogPath() {
+    return node_path_1.default.join(getCodexSwHome(), "logs", "service.log");
+}
+/**
+ * 将路径中的 `~` 展开为当前用户家目录。
+ *
+ * @param input 原始路径，允许以 `~` 开头。
+ * @returns 展开后的绝对或原始路径。
+ */
+function expandHome(input) {
+    if (input === "~") {
+        return node_os_1.default.homedir();
+    }
+    if (input.startsWith("~/")) {
+        return node_path_1.default.join(node_os_1.default.homedir(), input.slice(2));
+    }
+    return input;
+}
+/**
+ * 读取 cslot 配置；若配置不存在则返回默认配置。
+ *
+ * @returns 经过 schema 校验后的配置对象。
+ * @throws 当配置存在但内容非法时抛出错误。
+ */
+function loadConfig() {
+    const configPath = getConfigPath();
+    const legacyConfigPath = node_path_1.default.join(node_os_1.default.homedir(), ".codexsw", "config.yaml");
+    if (!node_fs_1.default.existsSync(configPath)) {
+        const defaultConfig = {
+            version: 1,
+            server: {
+                host: "127.0.0.1",
+                port: 4389,
+                api_key: "cslot-defaultkey",
+                body_limit_mb: 512
+            },
+            upstream: {
+                codex_base_url: "https://chatgpt.com/backend-api/codex",
+                auth_base_url: "https://auth.openai.com",
+                oauth_client_id: "app_EMoamEEZ73f0CkXaXp7hrann"
+            },
+            accounts: []
+        };
+        saveConfig(defaultConfig);
+        return defaultConfig;
+    }
+    const raw = node_fs_1.default.readFileSync(configPath, "utf8");
+    const parsed = raw.trim() ? yaml_1.default.parse(raw) : {};
+    const normalized = configSchema.parse(parsed);
+    let changed = JSON.stringify(parsed) !== JSON.stringify(normalized);
+    if (normalized.accounts.length === 0 &&
+        node_fs_1.default.existsSync(legacyConfigPath)) {
+        const legacyRaw = node_fs_1.default.readFileSync(legacyConfigPath, "utf8");
+        const legacyParsed = legacyRaw.trim() ? yaml_1.default.parse(legacyRaw) : {};
+        const legacyConfig = configSchema.parse(legacyParsed);
+        if (legacyConfig.accounts.length > 0) {
+            normalized.accounts = legacyConfig.accounts;
+            changed = true;
+        }
+    }
+    // 兼容历史默认值，统一迁移到新的简短本地 key。
+    if (normalized.server.api_key === "local-only-key" ||
+        normalized.server.api_key === "codexsw-defaultkey") {
+        normalized.server.api_key = "cslot-defaultkey";
+        changed = true;
+    }
+    // 当旧配置缺少新字段时，将补全后的配置回写，便于用户直接编辑查看。
+    if (changed) {
+        saveConfig(normalized);
+    }
+    return normalized;
+}
+/**
+ * 持久化 cslot 配置文件。
+ *
+ * @param config 待写入的配置对象。
+ * @returns 无返回值。
+ * @throws 当配置写入失败时抛出文件系统错误。
+ */
+function saveConfig(config) {
+    const configPath = getConfigPath();
+    const text = yaml_1.default.stringify(config);
+    node_fs_1.default.writeFileSync(configPath, text, "utf8");
+}
+/**
+ * 根据账号标识生成其独立的 HOME 目录。
+ *
+ * @param accountId 账号标识，仅用于本地目录名。
+ * @returns 该账号对应的 HOME 目录绝对路径。
+ */
+function getManagedHome(accountId) {
+    return node_path_1.default.join(getCodexSwHome(), "homes", accountId);
+}
+/**
+ * 将账号追加到配置中；若已存在相同 id 则覆盖更新。
+ *
+ * @param account 待写入的账号配置。
+ * @returns 更新后的完整配置对象。
+ */
+function upsertAccount(account) {
+    const config = loadConfig();
+    const index = config.accounts.findIndex((item) => item.id === account.id);
+    if (index >= 0) {
+        config.accounts[index] = account;
+    }
+    else {
+        config.accounts.push(account);
+    }
+    saveConfig(config);
+    return config;
+}

package/dist/login.js

@@ -0,0 +1,45 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.loginManagedAccount = loginManagedAccount;
+const node_child_process_1 = require("node:child_process");
+const node_fs_1 = __importDefault(require("node:fs"));
+const account_store_1 = require("./account-store");
+const config_1 = require("./config");
+/**
+ * 使用独立 HOME 目录拉起官方 `codex login`，完成单账号录入。
+ *
+ * @param accountId 本地账号标识。
+ * @returns Promise，成功时返回导入后的 HOME 目录。
+ * @throws 当 `codex login` 执行失败时抛出错误。
+ */
+async function loginManagedAccount(accountId) {
+    const managedHome = (0, config_1.getManagedHome)(accountId);
+    node_fs_1.default.mkdirSync(managedHome, { recursive: true });
+    return await new Promise((resolve, reject) => {
+        const child = (0, node_child_process_1.spawn)("codex", ["login"], {
+            env: {
+                ...process.env,
+                HOME: managedHome
+            },
+            stdio: "inherit"
+        });
+        child.on("exit", (code) => {
+            if (code === 0) {
+                if (!(0, account_store_1.hasCompleteCodexAuthState)(managedHome)) {
+                    reject(new Error("codex login 已退出，但未检测到完整登录态，请重新登录"));
+                    return;
+                }
+                (0, account_store_1.registerManagedAccount)(accountId, managedHome);
+                resolve(managedHome);
+                return;
+            }
+            reject(new Error(`codex login 失败，退出码: ${code ?? "unknown"}`));
+        });
+        child.on("error", (error) => {
+            reject(error);
+        });
+    });
+}

package/dist/scheduler.js

@@ -0,0 +1,97 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.pickBestAccount = pickBestAccount;
+exports.listCandidateAccounts = listCandidateAccounts;
+const config_1 = require("./config");
+const status_1 = require("./status");
+function nextResetWeight(resetAt) {
+    if (!resetAt) {
+        return Number.MAX_SAFE_INTEGER;
+    }
+    const diff = resetAt * 1000 - Date.now();
+    return diff > 0 ? diff : Number.MAX_SAFE_INTEGER;
+}
+/**
+ * 判断账号当前是否仅命中可忽略的短期本地熔断。
+ *
+ * 这类熔断通常由瞬时网络抖动、上游 5xx 或短暂 token 刷新失败触发，
+ * 当系统只剩一个可调度账号时，不应因此立刻把它排除掉。
+ *
+ * @param status 账号运行时状态。
+ * @returns `true` 表示仅存在可回退的短期本地熔断；否则返回 `false`。
+ */
+function isSoftLocalBlocked(status) {
+    if (!status.localBlockUntil || status.localBlockUntil * 1000 <= Date.now()) {
+        return false;
+    }
+    return [
+        "request_failed",
+        "upstream_5xx",
+        "temporary_5m_limit",
+        "token_refresh_failed"
+    ].includes(status.localBlockReason ?? "");
+}
+/**
+ * 对未命中额度限制的候选账号按剩余额度与重置时间排序。
+ *
+ * @param statuses 待排序的账号状态列表。
+ * @returns 排序后的账号状态列表，优先返回更适合尝试的账号。
+ */
+function rankEligibleStatuses(statuses) {
+    return [...statuses].sort((left, right) => {
+        const fiveHourDiff = (right.fiveHourLeftPercent ?? -1) - (left.fiveHourLeftPercent ?? -1);
+        if (fiveHourDiff !== 0) {
+            return fiveHourDiff;
+        }
+        const weeklyDiff = (right.weeklyLeftPercent ?? -1) - (left.weeklyLeftPercent ?? -1);
+        if (weeklyDiff !== 0) {
+            return weeklyDiff;
+        }
+        return nextResetWeight(left.fiveHourResetsAt) - nextResetWeight(right.fiveHourResetsAt);
+    });
+}
+/**
+ * 选择当前最适合激活的账号。
+ *
+ * 业务规则：
+ * 1. 仅在账号启用且存在凭据时参与调度。
+ * 2. 优先选择当前 5 小时和周窗口都未受限的账号。
+ * 3. 在多个可用账号间，优先选择 5 小时剩余额度更高的账号。
+ *
+ * @returns 调度结果；若没有可用账号则返回 `null`。
+ */
+function pickBestAccount() {
+    return listCandidateAccounts()[0] ?? null;
+}
+/**
+ * 返回按优先级排序后的可用账号列表，供代理重试链路使用。
+ *
+ * @returns 候选账号列表，已按优先级从高到低排序。
+ */
+function listCandidateAccounts() {
+    const config = (0, config_1.loadConfig)();
+    const statuses = (0, status_1.collectAccountStatuses)();
+    const accountMap = new Map(config.accounts.map((item) => [item.id, item]));
+    const eligible = statuses.filter((item) => item.enabled && item.exists && !item.isFiveHourLimited && !item.isWeeklyLimited);
+    const available = rankEligibleStatuses(statuses.filter((item) => item.isAvailable));
+    const ranked = available.length > 0 ? available : [];
+    // 当所有未限额账号都只命中短期本地熔断时，仍允许继续兜底尝试，避免把网络抖动误判成“无可用账号”。
+    if (ranked.length === 0 && eligible.length > 0 && eligible.every(isSoftLocalBlocked)) {
+        ranked.push(...rankEligibleStatuses(eligible));
+    }
+    return ranked
+        .map((winner) => {
+        const account = accountMap.get(winner.id);
+        if (!account) {
+            return null;
+        }
+        return {
+            account,
+            status: winner,
+            reason: winner.isAvailable
+                ? "优先选择 5 小时窗口剩余额度最高且当前可用的账号"
+                : "当前仅剩一个可调度账号，忽略短期本地熔断后继续兜底尝试"
+        };
+    })
+        .filter((item) => item !== null);
+}

package/dist/serve.js

@@ -0,0 +1,24 @@
+#!/usr/bin/env node
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+const config_1 = require("./config");
+const server_1 = require("./server");
+/**
+ * 后台服务进程入口。
+ *
+ * @returns Promise，无返回值。
+ * @throws 当端口参数非法或服务启动失败时抛出异常。
+ */
+async function main() {
+    const config = (0, config_1.loadConfig)();
+    const portArgIndex = process.argv.findIndex((item) => item === "--port");
+    const port = portArgIndex >= 0 && process.argv[portArgIndex + 1]
+        ? Number(process.argv[portArgIndex + 1])
+        : config.server.port;
+    await (0, server_1.startServer)(port);
+}
+void main().catch((error) => {
+    const message = error instanceof Error ? error.message : String(error);
+    console.error(`cslot service 启动失败: ${message}`);
+    process.exit(1);
+});