ccus-cli 0.1.5 → 0.1.8

package/dist/lib/aggregate.js

@@ -9,8 +9,11 @@ exports.buildAggregatedDailyRows = buildAggregatedDailyRows;
 exports.buildAggregatedWeeklyRows = buildAggregatedWeeklyRows;
 const promises_1 = __importDefault(require("node:fs/promises"));
 const node_path_1 = __importDefault(require("node:path"));
+const node_util_1 = require("node:util");
+const node_zlib_1 = require("node:zlib");
 const payload_1 = require("./payload");
 const time_1 = require("./time");
+const gunzipAsync = (0, node_util_1.promisify)(node_zlib_1.gunzip);
 function isRecord(value) {
     return typeof value === "object" && value !== null && !Array.isArray(value);
 }
@@ -59,6 +62,10 @@ function weekKey(date) {
 function toPersonKey(gitUserEmail, gitUserName) {
     return (0, time_1.extractGitEmailAccount)(gitUserEmail) ?? gitUserName ?? "unknown";
 }
+/** bundle 文件可以是明文 `.json`，也可以是 `ccus export` 默认输出的 gzip 压缩 `.json.gz`。 */
+function isBundleFileName(name) {
+    return name.endsWith(".json") || name.endsWith(".json.gz");
+}
 async function collectBundleJsonFiles(directoryPath) {
     const entries = await promises_1.default.readdir(directoryPath, { withFileTypes: true });
     const nested = await Promise.all(entries.map(async (entry) => {
@@ -66,10 +73,19 @@ async function collectBundleJsonFiles(directoryPath) {
         if (entry.isDirectory()) {
             return collectBundleJsonFiles(fullPath);
         }
-        return entry.isFile() && entry.name.endsWith(".json") ? [fullPath] : [];
+        return entry.isFile() && isBundleFileName(entry.name) ? [fullPath] : [];
     }));
     return nested.flat();
 }
+/** 读取 bundle 文件内容：`.gz` 结尾的先 gunzip 再按 UTF-8 解码，其它按明文读取。 */
+async function readBundleFileContent(filePath) {
+    if (filePath.endsWith(".gz")) {
+        const compressed = await promises_1.default.readFile(filePath);
+        const decompressed = await gunzipAsync(compressed);
+        return decompressed.toString("utf8");
+    }
+    return promises_1.default.readFile(filePath, "utf8");
+}
 /** 读取目录里的 export bundle json 文件。 */
 async function loadWeeklyExportBundles(inputDir) {
     const files = await collectBundleJsonFiles(inputDir);
@@ -77,7 +93,7 @@ async function loadWeeklyExportBundles(inputDir) {
     const invalidFiles = [];
     for (const filePath of files) {
         try {
-            const content = await promises_1.default.readFile(filePath, "utf8");
+            const content = await readBundleFileContent(filePath);
             const parsed = JSON.parse(content);
             if (isWeeklyExportBundle(parsed)) {
                 bundles.push({ filePath, bundle: parsed });

package/dist/lib/export.js

@@ -13,9 +13,13 @@ exports.buildSummaryCsv = buildSummaryCsv;
 exports.buildAggregatedDailyCsv = buildAggregatedDailyCsv;
 exports.buildAggregatedWeeklyCsv = buildAggregatedWeeklyCsv;
 exports.writeTextFile = writeTextFile;
+exports.writeGzipFile = writeGzipFile;
 const promises_1 = __importDefault(require("node:fs/promises"));
 const node_path_1 = __importDefault(require("node:path"));
+const node_util_1 = require("node:util");
+const node_zlib_1 = require("node:zlib");
 const time_1 = require("./time");
+const gzipAsync = (0, node_util_1.promisify)(node_zlib_1.gzip);
 /** CSV 字符串字段统一做转义，避免逗号和引号破坏列结构。 */
 function quoteCsv(value) {
     return `"${value.replaceAll("\"", "\"\"")}"`;
@@ -128,9 +132,9 @@ function buildRawJsonl(events) {
 function buildWeeklySummaryJson(summary) {
     return `${JSON.stringify(summary, null, 2)}\n`;
 }
-/** 默认导出把原始事件与周汇总一起打包成一个 JSON 文件。 */
+/** 默认导出把原始事件与周汇总一起打包成一个 JSON 文件，用紧凑序列化避免缩进空白撑大体积。 */
 function buildWeeklyExportBundleJson(bundle) {
-    return `${JSON.stringify(bundle, null, 2)}\n`;
+    return `${JSON.stringify(bundle)}\n`;
 }
 /** 按天汇总 usage 数据，生成 summary 模式的中间结果。 */
 function buildSummaryRows(events) {
@@ -272,4 +276,15 @@ async function writeTextFile(outputPath, content) {
     await promises_1.default.mkdir(node_path_1.default.dirname(outputPath), { recursive: true });
     await promises_1.default.writeFile(outputPath, content, "utf8");
 }
+/**
+ * 把内容 gzip 压缩后写文件，父目录不存在时自动创建。
+ *
+ * 仅压缩存储/传输层，写入的字节解压后与 `writeTextFile` 完全一致，
+ * 不改变 bundle 字段集合，也不影响 schemaVersion。
+ */
+async function writeGzipFile(outputPath, content) {
+    await promises_1.default.mkdir(node_path_1.default.dirname(outputPath), { recursive: true });
+    const compressed = await gzipAsync(content);
+    await promises_1.default.writeFile(outputPath, compressed);
+}
 //# sourceMappingURL=export.js.map

package/dist/lib/payload.js

@@ -6,6 +6,8 @@ Object.defineProperty(exports, "__esModule", { value: true });
 exports.readSessionId = readSessionId;
 exports.extractWorkspaceDir = extractWorkspaceDir;
 exports.parseStatuslinePayload = parseStatuslinePayload;
+exports.resolveCtxTier = resolveCtxTier;
+exports.resolveCtxRedThresholds = resolveCtxRedThresholds;
 exports.formatStatusLine = formatStatusLine;
 exports.createPersistedStatuslineEvent = createPersistedStatuslineEvent;
 exports.computeStatuslineEvent = computeStatuslineEvent;
@@ -146,16 +148,103 @@ function parseStatuslinePayload(input) {
     const parsed = JSON.parse(trimmed);
     return isRecord(parsed) ? parsed : {};
 }
+/** statusline 里 ctx 段标红用的 ANSI 颜色码，仅作用于展示，不进任何落盘/导出契约。 */
+const ANSI_RED = "\x1b[31m";
+const ANSI_RESET = "\x1b[0m";
+/** contextMax 超过该值就当作 1M 长上下文窗口，否则按 200K 档。 */
+const CTX_TIER_1M_MIN_MAX = 400_000;
+/**
+ * 各档位的内置默认阈值（百分比超阈值或已用 token 超阈值，任一满足即标红）。
+ *
+ * 200K 窗口余量小、到 80%（约 160K）才提醒；1M 窗口虽大但 50%（约 500K）已用很多，提前提醒。
+ */
+const CTX_TIER_DEFAULTS = {
+    "200k": { pct: 80, tokens: null },
+    "1m": { pct: 50, tokens: null },
+};
+/** 根据 contextMax 判断当前上下文窗口档位；拿不到 contextMax 时回退到 200K 档。 */
+function resolveCtxTier(contextMax) {
+    return contextMax !== null && contextMax > CTX_TIER_1M_MIN_MAX ? "1m" : "200k";
+}
+/** 把 `120000` / `120k` / `0.5m` 这类写法解析成整数 token；非法或空返回 null。 */
+function parseTokenThreshold(raw) {
+    const text = raw.trim().toLowerCase();
+    if (text === "") {
+        return null;
+    }
+    const multiplier = text.endsWith("m") ? 1_000_000 : text.endsWith("k") ? 1_000 : 1;
+    const numericPart = multiplier === 1 ? text : text.slice(0, -1);
+    const parsed = Number(numericPart) * multiplier;
+    return Number.isFinite(parsed) && parsed >= 0 ? parsed : null;
+}
+/** 按顺序读第一个能解析成非负数的百分比环境变量；全部缺失/非法时返回 fallback。 */
+function readPctEnv(env, keys, fallback) {
+    for (const key of keys) {
+        const raw = (env[key] ?? "").trim();
+        if (raw === "") {
+            continue;
+        }
+        const parsed = Number(raw);
+        if (Number.isFinite(parsed) && parsed >= 0) {
+            return parsed;
+        }
+    }
+    return fallback;
+}
+/** 按顺序读第一个非空的 token 阈值环境变量；全部缺失时返回 fallback。 */
+function readTokensEnv(env, keys, fallback) {
+    for (const key of keys) {
+        const raw = env[key] ?? "";
+        if (raw.trim() === "") {
+            continue;
+        }
+        return parseTokenThreshold(raw);
+    }
+    return fallback;
+}
+/**
+ * 解析某个上下文档位的 ctx 标红阈值，两个条件取「或」：百分比超阈值，或已用 token 超阈值。
+ *
+ * 优先级：档位专属环境变量 > 通用环境变量 > 档位内置默认。
+ *
+ * - 百分比：`CCUS_CTX_RED_PCT_200K` / `CCUS_CTX_RED_PCT_1M` → `CCUS_CTX_RED_PCT` → 档位默认（200K=80，1M=50）。
+ * - token：`CCUS_CTX_RED_TOKENS_200K` / `CCUS_CTX_RED_TOKENS_1M` → `CCUS_CTX_RED_TOKENS` → 档位默认（默认不启用）。
+ *   token 支持 `120000` / `120k` / `0.5m` 写法。
+ *
+ * 阈值只影响 statusline 颜色展示，不改变 stdin/stdout 文本契约，也不落盘。
+ */
+function resolveCtxRedThresholds(contextMax, env = process.env) {
+    const tier = resolveCtxTier(contextMax);
+    const defaults = CTX_TIER_DEFAULTS[tier];
+    const tierSuffix = tier === "1m" ? "1M" : "200K";
+    const pct = readPctEnv(env, [`CCUS_CTX_RED_PCT_${tierSuffix}`, "CCUS_CTX_RED_PCT"], defaults.pct);
+    const tokens = readTokensEnv(env, [`CCUS_CTX_RED_TOKENS_${tierSuffix}`, "CCUS_CTX_RED_TOKENS"], defaults.tokens);
+    return { tier, pct, tokens };
+}
+/** 判断 ctx 是否达到标红条件：百分比超阈值，或已用 token 超阈值，任一满足即标红。 */
+function isContextHot(contextWindowPct, contextUsed, thresholds) {
+    if (contextWindowPct !== null && contextWindowPct > thresholds.pct) {
+        return true;
+    }
+    if (thresholds.tokens !== null && contextUsed !== null && contextUsed >= thresholds.tokens) {
+        return true;
+    }
+    return false;
+}
 /**
  * 生成真正显示在 Claude Code statusline 上的短文本。
  *
  * 这里必须保持单行、紧凑，避免污染 statusline 展示区域。
  */
-function formatStatusLine(event, gitBranch = null) {
+function formatStatusLine(event, gitBranch = null, env = process.env) {
     const timeLabel = (0, time_1.formatClock)(new Date(event.timestamp));
     const usageLabel = event.usagePct === null ? "5h --" : `5h ${event.usagePct.toFixed(1)}%`;
     const sevenDayLabel = event.sevenDayUsagePct === null ? "7d --" : `7d ${event.sevenDayUsagePct.toFixed(1)}%`;
-    const contextLabel = event.contextWindowPct === null ? "ctx --" : `ctx ${event.contextWindowPct.toFixed(1)}%`;
+    const contextText = event.contextWindowPct === null ? "ctx --" : `ctx ${event.contextWindowPct.toFixed(1)}%`;
+    // ctx 占用超阈值时整段标红，提醒上下文快满；阈值按窗口大小（200K / 1M）分档，由 resolveCtxRedThresholds 决定。
+    const contextLabel = isContextHot(event.contextWindowPct, event.contextUsed ?? null, resolveCtxRedThresholds(event.contextMax ?? null, env))
+        ? `${ANSI_RED}${contextText}${ANSI_RESET}`
+        : contextText;
     const modelLabel = event.modelName ?? "model --";
     const workspaceLabel = event.workspaceName ?? "workspace --";
     const segments = [usageLabel, sevenDayLabel, contextLabel, modelLabel, workspaceLabel];

package/package.json

@@ -1,35 +1,39 @@
-{
-  "name": "ccus-cli",
-  "version": "0.1.5",
-  "description": "Claude Code statusline usage logger and dashboard CLI",
-  "type": "commonjs",
-  "bin": {
-    "ccus": "dist/cli.js"
-  },
-  "files": [
-    "dist/cli.js",
-    "dist/lib/**/*.js",
-    "dist/types.js"
-  ],
-  "scripts": {
-    "build": "tsc -p tsconfig.json",
-    "prepublishOnly": "npm run build",
-    "test": "node --test dist/test",
-    "test:src": "node --import tsx --test src/test/payload.test.ts src/test/dashboard.test.ts src/test/export.test.ts src/test/storage.test.ts src/test/claude.test.ts src/test/aggregate.test.ts src/test/aggregate-dashboard.test.ts src/test/install.test.ts src/test/debug.test.ts src/test/time.test.ts src/test/update-check.test.ts"
-  },
-  "keywords": [
-    "claude-code",
-    "statusline",
-    "cli",
-    "dashboard"
-  ],
-  "license": "MIT",
-  "engines": {
-    "node": ">=20"
-  },
-  "devDependencies": {
-    "@types/node": "^24.0.0",
-    "tsx": "^4.20.0",
-    "typescript": "^5.8.3"
-  }
-}
+{
+  "name": "ccus-cli",
+  "version": "0.1.8",
+  "description": "Claude Code statusline usage logger and dashboard CLI",
+  "type": "commonjs",
+  "bin": {
+    "ccus": "dist/cli.js"
+  },
+  "files": [
+    "dist/cli.js",
+    "dist/lib/**/*.js",
+    "dist/types.js"
+  ],
+  "scripts": {
+    "build": "tsc -p tsconfig.json",
+    "prepublishOnly": "npm run build",
+    "test": "node --test dist/test",
+    "test:src": "node --import tsx --test src/test/payload.test.ts src/test/dashboard.test.ts src/test/export.test.ts src/test/storage.test.ts src/test/claude.test.ts src/test/aggregate.test.ts src/test/aggregate-dashboard.test.ts src/test/install.test.ts src/test/debug.test.ts src/test/time.test.ts src/test/update-check.test.ts"
+  },
+  "keywords": [
+    "claude-code",
+    "statusline",
+    "cli",
+    "dashboard"
+  ],
+  "repository": {
+    "type": "git",
+    "url": "https://github.com/gaoshang212/ccus-cli.git"
+  },
+  "license": "MIT",
+  "engines": {
+    "node": ">=20"
+  },
+  "devDependencies": {
+    "@types/node": "^24.0.0",
+    "tsx": "^4.20.0",
+    "typescript": "^5.8.3"
+  }
+}