ccus-cli 0.1.6 → 0.1.8

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.6",
-  "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"
+  }
+}