ccus-cli 0.1.26 → 0.2.2

package/dist/lib/codex-fetcher.js

@@ -0,0 +1,281 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.fetchCodexQuota = fetchCodexQuota;
+exports.readCodexQuotaCacheSync = readCodexQuotaCacheSync;
+exports.resolveCodexQuota = resolveCodexQuota;
+const node_child_process_1 = require("node:child_process");
+const node_fs_1 = __importDefault(require("node:fs"));
+const promises_1 = __importDefault(require("node:fs/promises"));
+const node_os_1 = require("node:os");
+const node_path_1 = require("node:path");
+const debug_1 = require("./debug");
+const paths_1 = require("./paths");
+// app-server 子进程参数：read-only + untrusted，Codex 官方推荐的最低权限拉额度姿态。
+const CODEX_ARGS = ["-s", "read-only", "-a", "untrusted", "app-server"];
+const DEFAULT_TIMEOUT_MS = 10_000;
+// notify 每 turn 触发，额度缓存 5 分钟，命中秒回避免阻塞 Codex 主流程。
+const DEFAULT_TTL_MS = 5 * 60 * 1000;
+function isRecord(value) {
+    return typeof value === "object" && value !== null && !Array.isArray(value);
+}
+function toFiniteNumber(value) {
+    if (typeof value === "number" && Number.isFinite(value)) {
+        return value;
+    }
+    if (typeof value === "string" && value.trim() !== "") {
+        const parsed = Number(value);
+        return Number.isFinite(parsed) ? parsed : null;
+    }
+    return null;
+}
+function clampPercent(value) {
+    if (value === null) {
+        return null;
+    }
+    return Math.min(100, Math.max(0, value));
+}
+/**
+ * 读窗口使用率：驼峰优先（app-server RPC 实际返回 usedPercent），
+ * used_percentage / usedPercentage fallback（对齐 ccus 生态字段名，防御未来协议变更）。
+ */
+function readUsedPercent(window) {
+    if (!isRecord(window)) {
+        return null;
+    }
+    return clampPercent(toFiniteNumber(window.usedPercent ?? window.used_percentage ?? window.usedPercentage));
+}
+/**
+ * 读重置时间：Codex app-server 返回 Unix 秒；< 10^10 视为秒转毫秒，否则按毫秒。
+ */
+function readResetsAtMs(window) {
+    if (!isRecord(window)) {
+        return null;
+    }
+    const raw = window.resetsAt ?? window.reset_at ?? window.resetsAtMs;
+    const n = toFiniteNumber(raw);
+    if (n === null) {
+        return null;
+    }
+    return n < 10_000_000_000 ? n * 1000 : n;
+}
+/**
+ * 解析 account/rateLimits/read 的 result：primary→5h、secondary→weekly（Codex 约定）。
+ * 字段缺失时对应窗口为 null，不阻断另一窗口。
+ */
+function parseRateLimitsResult(result) {
+    const wrapper = isRecord(result) && isRecord(result.rateLimits) ? result.rateLimits : null;
+    if (!wrapper) {
+        return { fiveHour: null, sevenDay: null, resetsAt: null };
+    }
+    const fiveHour = readUsedPercent(wrapper.primary);
+    const sevenDay = readUsedPercent(wrapper.secondary);
+    const resetsAt = readResetsAtMs(wrapper.primary) ?? readResetsAtMs(wrapper.secondary);
+    return { fiveHour, sevenDay, resetsAt };
+}
+function getCodexHome(codexHomePath) {
+    return codexHomePath ?? process.env.CODEX_HOME ?? (0, node_path_1.join)((0, node_os_1.homedir)(), ".codex");
+}
+function buildRpcMessage(id, method, params) {
+    return `${JSON.stringify({ jsonrpc: "2.0", id, method, params: params ?? {} })}\n`;
+}
+function buildRpcNotification(method) {
+    return `${JSON.stringify({ jsonrpc: "2.0", method, params: {} })}\n`;
+}
+// 真实 spawn：Windows 上 codex 多为 codex.cmd，不带 shell 会失败，故 win32 走 shell。
+const defaultSpawn = (command, args, options) => {
+    const merged = process.platform === "win32" ? { ...options, shell: true } : options;
+    return (0, node_child_process_1.spawn)(command, args, merged);
+};
+/**
+ * spawn `codex app-server` 走 JSON-RPC 握手 + `account/rateLimits/read` 拉额度。
+ *
+ * 握手必须 initialize → 收响应 → 发 initialized 通知，否则后续方法被拒为 "Not initialized"。
+ * 全程不抛错：ENOENT → unavailable，超时 / RPC error / 进程退出 → error。
+ */
+async function fetchCodexQuota(options = {}) {
+    const timeoutMs = options.timeoutMs ?? DEFAULT_TIMEOUT_MS;
+    const env = options.env ?? process.env;
+    const spawnFn = options.spawn ?? defaultSpawn;
+    const childEnv = { ...env, CODEX_HOME: getCodexHome(options.codexHomePath) };
+    const empty = (status) => ({
+        fiveHour: null,
+        sevenDay: null,
+        resetsAt: null,
+        status,
+    });
+    return new Promise((resolve) => {
+        let settled = false;
+        let buffer = "";
+        let rpcId = 0;
+        let rateLimitsId = null;
+        let child = null;
+        let timer = null;
+        const finish = (outcome, kill = false) => {
+            if (settled) {
+                return;
+            }
+            settled = true;
+            if (timer) {
+                clearTimeout(timer);
+                timer = null;
+            }
+            // settled 标志已保证 resolve 幂等，无需再手动摘 listener；kill 终止子进程后其 stdio 自然关闭。
+            if (kill && child) {
+                try {
+                    child.kill();
+                }
+                catch {
+                    // kill 失败不影响结果。
+                }
+            }
+            resolve(outcome);
+        };
+        try {
+            child = spawnFn("codex", CODEX_ARGS, {
+                env: childEnv,
+                stdio: ["pipe", "pipe", "pipe"],
+                windowsHide: true,
+            });
+        }
+        catch (error) {
+            (0, debug_1.debugLog)("codex", "spawn threw", error instanceof Error ? error.message : String(error));
+            resolve(empty("error"));
+            return;
+        }
+        timer = setTimeout(() => {
+            (0, debug_1.debugLog)("codex", "rpc timeout", { timeoutMs });
+            finish(empty("error"), true);
+        }, timeoutMs);
+        const send = (message) => {
+            try {
+                child?.stdin.write(message);
+            }
+            catch (error) {
+                (0, debug_1.debugLog)("codex", "stdin write failed", error instanceof Error ? error.message : String(error));
+            }
+        };
+        const initId = (rpcId += 1);
+        send(buildRpcMessage(initId, "initialize", { clientInfo: { name: "ccus", version: "1.0.0" } }));
+        child.stdout.on("data", (chunk) => {
+            buffer += chunk.toString();
+            let newline;
+            while ((newline = buffer.indexOf("\n")) !== -1) {
+                const line = buffer.slice(0, newline).trim();
+                buffer = buffer.slice(newline + 1);
+                if (!line) {
+                    continue;
+                }
+                let message;
+                try {
+                    message = JSON.parse(line);
+                }
+                catch {
+                    continue; // 非 JSON 行（server 日志等）忽略。
+                }
+                if (message.id === undefined) {
+                    continue; // server 通知无 id。
+                }
+                if (message.id === initId) {
+                    // 握手成功，发 initialized 通知后再请求额度。
+                    send(buildRpcNotification("initialized"));
+                    rateLimitsId = rpcId += 1;
+                    send(buildRpcMessage(rateLimitsId, "account/rateLimits/read"));
+                    continue;
+                }
+                if (rateLimitsId !== null && message.id === rateLimitsId) {
+                    if (message.error) {
+                        (0, debug_1.debugLog)("codex", "rateLimits rpc error", message.error.message ?? "unknown");
+                        finish(empty("error"), true);
+                        return;
+                    }
+                    const quota = parseRateLimitsResult(message.result);
+                    (0, debug_1.debugLog)("codex", "quota parsed", quota);
+                    finish({ ...quota, status: "ok" }, true);
+                }
+            }
+        });
+        child.on("error", (err) => {
+            const enoent = err.code === "ENOENT";
+            (0, debug_1.debugLog)("codex", "spawn error", { code: err.code, message: err.message });
+            finish(empty(enoent ? "unavailable" : "error"));
+        });
+        child.on("close", (code) => {
+            (0, debug_1.debugLog)("codex", "app-server exited before quota", { code });
+            finish(empty("error"));
+        });
+    });
+}
+/** 同步读取额度缓存；损坏 / 缺失返回 null。 */
+function readCodexQuotaCacheSync(dataDir) {
+    try {
+        const raw = node_fs_1.default.readFileSync((0, paths_1.getCodexQuotaCachePath)(dataDir), "utf8");
+        const parsed = JSON.parse(raw);
+        if (typeof parsed.fetchedAt !== "string") {
+            return null;
+        }
+        return {
+            fetchedAt: parsed.fetchedAt,
+            fiveHour: typeof parsed.fiveHour === "number" ? parsed.fiveHour : null,
+            sevenDay: typeof parsed.sevenDay === "number" ? parsed.sevenDay : null,
+            resetsAt: typeof parsed.resetsAt === "number" ? parsed.resetsAt : null,
+        };
+    }
+    catch {
+        return null;
+    }
+}
+/** 写入额度缓存；失败静默（不能因缓存写失败影响采集主流程）。 */
+async function writeCodexQuotaCache(dataDir, cache) {
+    try {
+        await promises_1.default.mkdir(dataDir, { recursive: true });
+        await promises_1.default.writeFile((0, paths_1.getCodexQuotaCachePath)(dataDir), `${JSON.stringify(cache, null, 2)}\n`, "utf8");
+    }
+    catch (error) {
+        (0, debug_1.debugLog)("codex", "failed to write quota cache", error instanceof Error ? error.message : String(error));
+    }
+}
+function isCodexCacheFresh(cache, ttlMs, now) {
+    const fetched = Date.parse(cache.fetchedAt);
+    return Number.isFinite(fetched) && now.getTime() - fetched < ttlMs;
+}
+function quotaFromCache(cache) {
+    return { fiveHour: cache.fiveHour, sevenDay: cache.sevenDay, resetsAt: cache.resetsAt };
+}
+function hasAnyQuota(quota) {
+    return quota.fiveHour !== null || quota.sevenDay !== null;
+}
+/**
+ * 缓存优先解析额度：新鲜直接返回；过期拉一次（带超时）；拉取失败 / 无有效数据回退旧缓存，全失败返回 null。
+ *
+ * notify 路径用，全程不抛错、不写 stdout。`options.fetcher` 供测试注入。
+ */
+async function resolveCodexQuota(dataDir, options = {}) {
+    const now = options.now ?? new Date();
+    const ttlMs = options.ttlMs ?? DEFAULT_TTL_MS;
+    const fetcher = options.fetcher ?? fetchCodexQuota;
+    const cached = readCodexQuotaCacheSync(dataDir);
+    if (cached && isCodexCacheFresh(cached, ttlMs, now)) {
+        return quotaFromCache(cached);
+    }
+    try {
+        const outcome = await fetcher(options.fetchOptions);
+        if (outcome.status === "ok" && hasAnyQuota(outcome)) {
+            const quota = {
+                fiveHour: outcome.fiveHour,
+                sevenDay: outcome.sevenDay,
+                resetsAt: outcome.resetsAt,
+            };
+            await writeCodexQuotaCache(dataDir, { ...quota, fetchedAt: now.toISOString() });
+            return quota;
+        }
+        (0, debug_1.debugLog)("codex", "fetch did not yield usable quota", { status: outcome.status });
+    }
+    catch (error) {
+        (0, debug_1.debugLog)("codex", "fetch threw, fallback to cache", error instanceof Error ? error.message : String(error));
+    }
+    return cached ? quotaFromCache(cached) : null;
+}
+//# sourceMappingURL=codex-fetcher.js.map

package/dist/lib/codex-install.js

@@ -0,0 +1,228 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.installCodexNotify = installCodexNotify;
+exports.uninstallCodexNotify = uninstallCodexNotify;
+exports.installCodexHook = installCodexHook;
+exports.uninstallCodexHook = uninstallCodexHook;
+const promises_1 = __importDefault(require("node:fs/promises"));
+const node_path_1 = __importDefault(require("node:path"));
+/** 序列化 notify 数组为 TOML 数组字面量（字符串元素，元素间带空格，符合 TOML 习惯）。 */
+function serializeNotify(notify) {
+    return `[${notify.map((value) => JSON.stringify(value)).join(", ")}]`;
+}
+/**
+ * 从 notify 赋值行的 `=` 之后提取字符串元素（兼容双引号 / 单引号）。
+ * 提取不到任何字符串返回空数组（调用方另判多行数组）。
+ */
+function parseNotifyElements(line) {
+    const eq = line.indexOf("=");
+    if (eq < 0) {
+        return [];
+    }
+    const value = line.slice(eq + 1);
+    return [...value.matchAll(/"((?:[^"\\]|\\.)*)"|'([^']*)'/g)].map((match) => match[1] ?? match[2] ?? "");
+}
+/** 判断一行是否是顶层 notify 赋值（非注释、非 table 头、key 为 notify）。 */
+function isTopLevelNotifyLine(line) {
+    const trimmed = line.trim();
+    if (trimmed.startsWith("#") || trimmed.startsWith("[")) {
+        return false;
+    }
+    return /^\s*notify\s*=/.test(line);
+}
+/** 判断一行是否是 TOML table 头（[section] / [[array-of-table]]）。 */
+function isTableHeader(line) {
+    return /^\s*\[\[?/.test(line);
+}
+/** 顶层区域结束行号：第一个 table 头的索引，没有则为行数。 */
+function findTopLevelEnd(lines) {
+    for (let index = 0; index < lines.length; index += 1) {
+        if (lines[index].trim().startsWith("#")) {
+            continue;
+        }
+        if (isTableHeader(lines[index])) {
+            return index;
+        }
+    }
+    return lines.length;
+}
+/** 读现有 config.toml；缺失返回空文本。 */
+async function readConfig(configPath) {
+    try {
+        const raw = await promises_1.default.readFile(configPath, "utf8");
+        return { raw, existed: true };
+    }
+    catch (error) {
+        if (error.code === "ENOENT") {
+            return { raw: "", existed: false };
+        }
+        throw error;
+    }
+}
+/** 统一以 \n 行尾写回，避免 split/join 产生多余空行。 */
+async function writeConfig(configPath, lines) {
+    await promises_1.default.mkdir(node_path_1.default.dirname(configPath), { recursive: true });
+    const content = `${lines.join("\n").replace(/\n+$/, "")}\n`;
+    await promises_1.default.writeFile(configPath, content, "utf8");
+}
+function arrayEqual(left, right) {
+    return left.length === right.length && left.every((value, index) => value === right[index]);
+}
+/**
+ * 把 notify 写进 Codex config.toml 的顶层。
+ *
+ * 只改顶层 notify 赋值行（table 头之前），保留其它顶层 key、所有 table、注释与格式。
+ * 已等于目标时不动文件；现有 notify 是跨行数组时拒绝改写（避免破坏多行结构）。
+ */
+async function installCodexNotify(configPath, notify) {
+    const { raw, existed } = await readConfig(configPath);
+    const lines = raw.length > 0 ? raw.split(/\r?\n/) : [];
+    const topLevelEnd = findTopLevelEnd(lines);
+    const notifyIdx = lines.slice(0, topLevelEnd).findIndex((line) => isTopLevelNotifyLine(line));
+    if (notifyIdx >= 0) {
+        const line = lines[notifyIdx];
+        const valuePart = line.slice(line.indexOf("=") + 1);
+        // 跨行 notify 数组（行内有 [ 但无闭合 ]）无法安全单行替换，拒绝以免破坏结构。
+        if (valuePart.includes("[") && !valuePart.includes("]")) {
+            throw new Error(`Cannot update multi-line notify array in ${configPath}. Fold it into a single line, then retry.`);
+        }
+        const current = parseNotifyElements(line);
+        const unchanged = arrayEqual(current, notify);
+        if (!unchanged) {
+            lines[notifyIdx] = `notify = ${serializeNotify(notify)}`;
+            await writeConfig(configPath, lines);
+        }
+        return {
+            configPath,
+            notify,
+            previousNotify: current.length > 0 ? current : null,
+            created: !existed,
+            unchanged,
+        };
+    }
+    // 顶层无 notify：顶层有实质 key 时紧跟其后插入（与上方空一行）；顶层为空则插到最前。
+    const hasTopLevelContent = lines
+        .slice(0, topLevelEnd)
+        .some((line) => line.trim() !== "" && !line.trim().startsWith("#"));
+    const notifyLine = `notify = ${serializeNotify(notify)}`;
+    const insertAt = hasTopLevelContent ? topLevelEnd : 0;
+    const needsBlankPrefix = hasTopLevelContent && insertAt > 0 && lines[insertAt - 1].trim() !== "";
+    lines.splice(insertAt, 0, ...(needsBlankPrefix ? ["", notifyLine] : [notifyLine]));
+    await writeConfig(configPath, lines);
+    return { configPath, notify, previousNotify: null, created: !existed, unchanged: false };
+}
+/** 移除 Codex config.toml 顶层的 notify 赋值；不存在或无 notify 时 removed=false。 */
+async function uninstallCodexNotify(configPath) {
+    const { raw, existed } = await readConfig(configPath);
+    if (!existed) {
+        return { configPath, removed: false, previousNotify: null };
+    }
+    const lines = raw.split(/\r?\n/);
+    const topLevelEnd = findTopLevelEnd(lines);
+    const notifyIdx = lines.slice(0, topLevelEnd).findIndex((line) => isTopLevelNotifyLine(line));
+    if (notifyIdx < 0) {
+        return { configPath, removed: false, previousNotify: null };
+    }
+    const previousNotify = parseNotifyElements(lines[notifyIdx]);
+    lines.splice(notifyIdx, 1);
+    await writeConfig(configPath, lines);
+    return {
+        configPath,
+        removed: true,
+        previousNotify: previousNotify.length > 0 ? previousNotify : null,
+    };
+}
+/** Stop hook 超时（秒）：ccus 缓存命中秒回、过期拉额度 ~10s，给足余量。 */
+const CODEX_HOOK_TIMEOUT_SECONDS = 60;
+/** 判断一个 hook 条目是否是 ccus 装的（command 含 `__codex-hook`，兼容带 `--data-dir`）。 */
+function isCcusCodexHookEntry(entry) {
+    if (typeof entry !== "object" || entry === null) {
+        return false;
+    }
+    const command = entry.command;
+    return typeof command === "string" && command.includes("__codex-hook");
+}
+/**
+ * 读 hooks.json；缺失返回空结构，非法 JSON 抛错（不静默覆盖，避免破坏 orca 等外部工具写入的文件）。
+ */
+async function readHooksFile(hooksPath) {
+    try {
+        const raw = await promises_1.default.readFile(hooksPath, "utf8");
+        if (raw.trim() === "") {
+            return { data: {}, existed: true };
+        }
+        const parsed = JSON.parse(raw);
+        if (typeof parsed !== "object" || parsed === null || Array.isArray(parsed)) {
+            throw new Error(`${hooksPath} is not a JSON object`);
+        }
+        return { data: parsed, existed: true };
+    }
+    catch (error) {
+        if (error.code === "ENOENT") {
+            return { data: {}, existed: false };
+        }
+        throw error;
+    }
+}
+/** 以 2 空格缩进 + 末尾换行写回，保持 hooks.json 可读、对 git 友好。 */
+async function writeHooksFile(hooksPath, data) {
+    await promises_1.default.mkdir(node_path_1.default.dirname(hooksPath), { recursive: true });
+    await promises_1.default.writeFile(hooksPath, `${JSON.stringify(data, null, 2)}\n`, "utf8");
+}
+/**
+ * 把 ccus 的 Stop hook 挂进 Codex hooks.json。
+ *
+ * 只往 `hooks.Stop` 追加一条 command hook，保留其它事件、其它 hook（如 orca 的 codex-hook.cmd）、
+ * description 与格式。Stop 已有相同 command 时不动文件（幂等）；Stop 不存在则新建一个分组。
+ * 同一 Stop 事件下多条 hook 由 Codex 并发执行，互不阻塞。
+ */
+async function installCodexHook(hooksPath, command) {
+    const { data, existed } = await readHooksFile(hooksPath);
+    const hooksMap = data.hooks ?? {};
+    const stopGroups = hooksMap.Stop ?? [];
+    const stopExisted = stopGroups.length > 0;
+    for (const group of stopGroups) {
+        if ((group.hooks ?? []).some((entry) => entry.command === command)) {
+            return { hooksPath, command, created: !existed, unchanged: true, stopExisted };
+        }
+    }
+    const ccusHook = { type: "command", command, timeout: CODEX_HOOK_TIMEOUT_SECONDS };
+    if (stopGroups.length > 0) {
+        const first = stopGroups[0];
+        first.hooks = [...(first.hooks ?? []), ccusHook];
+    }
+    else {
+        stopGroups.push({ hooks: [ccusHook] });
+    }
+    hooksMap.Stop = stopGroups;
+    data.hooks = hooksMap;
+    await writeHooksFile(hooksPath, data);
+    return { hooksPath, command, created: !existed, unchanged: false, stopExisted };
+}
+/** 移除 hooks.json 里 ccus 装的 Stop hook（command 含 `__codex-hook`）；不存在时 removed=false。 */
+async function uninstallCodexHook(hooksPath) {
+    const { data, existed } = await readHooksFile(hooksPath);
+    if (!existed) {
+        return { hooksPath, removed: false };
+    }
+    const hooksMap = data.hooks ?? {};
+    const stopGroups = hooksMap.Stop ?? [];
+    let removed = false;
+    for (const group of stopGroups) {
+        const before = group.hooks?.length ?? 0;
+        group.hooks = (group.hooks ?? []).filter((entry) => !isCcusCodexHookEntry(entry));
+        if (group.hooks.length !== before) {
+            removed = true;
+        }
+    }
+    if (removed) {
+        hooksMap.Stop = stopGroups;
+        data.hooks = hooksMap;
+        await writeHooksFile(hooksPath, data);
+    }
+    return { hooksPath, removed };
+}
+//# sourceMappingURL=codex-install.js.map