ccus-cli 0.1.0

package/dist/lib/aggregate.js

@@ -0,0 +1,168 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.loadWeeklyExportBundles = loadWeeklyExportBundles;
+exports.buildAggregatedDetailRows = buildAggregatedDetailRows;
+exports.buildAggregatedDailyRows = buildAggregatedDailyRows;
+exports.buildAggregatedWeeklyRows = buildAggregatedWeeklyRows;
+const promises_1 = __importDefault(require("node:fs/promises"));
+const node_path_1 = __importDefault(require("node:path"));
+const payload_1 = require("./payload");
+const time_1 = require("./time");
+function isRecord(value) {
+    return typeof value === "object" && value !== null && !Array.isArray(value);
+}
+function isPersistedStatuslineEvent(value) {
+    return isRecord(value) && typeof value.timestamp === "string" && isRecord(value.rawPayload);
+}
+function hasWeeklyStatuslineShape(value) {
+    return (isRecord(value) &&
+        typeof value.sampleCount === "number" &&
+        typeof value.uniqueSessions === "number" &&
+        typeof value.uniqueWorkspaces === "number" &&
+        (typeof value.fiveHourLatestUsagePct === "number" || value.fiveHourLatestUsagePct === null) &&
+        (typeof value.fiveHourPeakUsagePct === "number" || value.fiveHourPeakUsagePct === null) &&
+        (typeof value.sevenDayLatestUsagePct === "number" || value.sevenDayLatestUsagePct === null) &&
+        (typeof value.sevenDayPeakUsagePct === "number" || value.sevenDayPeakUsagePct === null));
+}
+function isWeeklyExportBundle(value) {
+    if (!isRecord(value)) {
+        return false;
+    }
+    if (value.schemaVersion !== 6) {
+        return false;
+    }
+    if (!Array.isArray(value.rawEvents) || !Array.isArray(value.dailySummaries) || !isRecord(value.weeklySummary) || !isRecord(value.identity) || !isRecord(value.range)) {
+        return false;
+    }
+    return hasWeeklyStatuslineShape(value.weeklySummary.statusline);
+}
+function localDateKey(date) {
+    const year = date.getFullYear();
+    const month = String(date.getMonth() + 1).padStart(2, "0");
+    const day = String(date.getDate()).padStart(2, "0");
+    return `${year}-${month}-${day}`;
+}
+function startOfLocalWeek(date) {
+    const day = date.getDay();
+    const diff = day === 0 ? -6 : 1 - day;
+    const start = new Date(date);
+    start.setDate(date.getDate() + diff);
+    start.setHours(0, 0, 0, 0);
+    return start;
+}
+function weekKey(date) {
+    return localDateKey(startOfLocalWeek(date));
+}
+function toPersonKey(gitUserEmail, gitUserName) {
+    return (0, time_1.extractGitEmailAccount)(gitUserEmail) ?? gitUserName ?? "unknown";
+}
+async function collectBundleJsonFiles(directoryPath) {
+    const entries = await promises_1.default.readdir(directoryPath, { withFileTypes: true });
+    const nested = await Promise.all(entries.map(async (entry) => {
+        const fullPath = node_path_1.default.join(directoryPath, entry.name);
+        if (entry.isDirectory()) {
+            return collectBundleJsonFiles(fullPath);
+        }
+        return entry.isFile() && entry.name.endsWith(".json") ? [fullPath] : [];
+    }));
+    return nested.flat();
+}
+/** 读取目录里的 export bundle json 文件。 */
+async function loadWeeklyExportBundles(inputDir) {
+    const files = await collectBundleJsonFiles(inputDir);
+    const bundles = [];
+    const invalidFiles = [];
+    for (const filePath of files) {
+        try {
+            const content = await promises_1.default.readFile(filePath, "utf8");
+            const parsed = JSON.parse(content);
+            if (isWeeklyExportBundle(parsed)) {
+                bundles.push({ filePath, bundle: parsed });
+                continue;
+            }
+            invalidFiles.push(filePath);
+        }
+        catch {
+            continue;
+        }
+    }
+    if (invalidFiles.length > 0) {
+        throw new Error(`Unsupported export bundle schema in files: ${invalidFiles.join(", ")}. Re-export with current ccus so aggregate receives schemaVersion 6 bundles.`);
+    }
+    return bundles;
+}
+/** 从 bundle.rawEvents 展开 detail.csv 明细。 */
+function buildAggregatedDetailRows(bundles) {
+    const rows = [];
+    for (const { bundle } of bundles) {
+        const personKey = toPersonKey(bundle.identity.gitUserEmail, bundle.identity.gitUserName);
+        const tokensByDate = new Map(bundle.dailySummaries.map((day) => [day.date, day]));
+        for (const record of bundle.rawEvents.filter(isPersistedStatuslineEvent)) {
+            const event = (0, payload_1.computeStatuslineEvent)(record);
+            const ts = new Date(event.timestamp);
+            const dateKey = localDateKey(ts);
+            const dayTokens = tokensByDate.get(dateKey);
+            rows.push({
+                ...event,
+                personKey,
+                weekKey: weekKey(ts),
+                dateKey,
+                inputTokens: dayTokens?.inputTokens ?? 0,
+                outputTokens: dayTokens?.outputTokens ?? 0,
+                cacheReadInputTokens: dayTokens?.cacheReadInputTokens ?? 0,
+            });
+        }
+    }
+    return rows.sort((left, right) => left.timestamp.localeCompare(right.timestamp));
+}
+/** 直接从 bundle.dailySummaries 展开 daily.csv。 */
+function buildAggregatedDailyRows(bundles) {
+    const rows = [];
+    for (const { bundle } of bundles) {
+        const personKey = toPersonKey(bundle.identity.gitUserEmail, bundle.identity.gitUserName);
+        for (const item of bundle.dailySummaries) {
+            rows.push({
+                personKey,
+                date: item.date,
+                userMessageCount: item.userMessageCount,
+                apiRequestCount: item.apiRequestCount,
+                inputTokens: item.inputTokens,
+                outputTokens: item.outputTokens,
+                cacheReadInputTokens: item.cacheReadInputTokens,
+                sampleCount: item.sampleCount,
+                fiveHourPeakUsagePct: item.fiveHourPeakUsagePct,
+                fiveHourLatestUsagePct: item.fiveHourLatestUsagePct,
+                sevenDayPeakUsagePct: item.sevenDayPeakUsagePct,
+                sevenDayLatestUsagePct: item.sevenDayLatestUsagePct,
+                uniqueSessions: item.uniqueSessions,
+                uniqueWorkspaces: item.uniqueWorkspaces,
+            });
+        }
+    }
+    return rows.sort((left, right) => `${left.personKey}|${left.date}`.localeCompare(`${right.personKey}|${right.date}`));
+}
+/** 直接从 bundle.weeklySummary 展开 weekly.csv。 */
+function buildAggregatedWeeklyRows(bundles) {
+    return bundles
+        .map(({ bundle }) => ({
+        personKey: toPersonKey(bundle.identity.gitUserEmail, bundle.identity.gitUserName),
+        week: weekKey(new Date(bundle.range.start)),
+        userMessageCount: bundle.weeklySummary.counts.userMessageCount,
+        apiRequestCount: bundle.weeklySummary.counts.apiRequestCount,
+        inputTokens: bundle.weeklySummary.tokens.inputTokens,
+        outputTokens: bundle.weeklySummary.tokens.outputTokens,
+        cacheReadInputTokens: bundle.weeklySummary.tokens.cacheReadInputTokens,
+        sampleCount: bundle.weeklySummary.statusline.sampleCount,
+        fiveHourPeakUsagePct: bundle.weeklySummary.statusline.fiveHourPeakUsagePct,
+        fiveHourLatestUsagePct: bundle.weeklySummary.statusline.fiveHourLatestUsagePct,
+        sevenDayPeakUsagePct: bundle.weeklySummary.statusline.sevenDayPeakUsagePct,
+        sevenDayLatestUsagePct: bundle.weeklySummary.statusline.sevenDayLatestUsagePct,
+        uniqueSessions: bundle.weeklySummary.statusline.uniqueSessions,
+        uniqueWorkspaces: bundle.weeklySummary.statusline.uniqueWorkspaces,
+    }))
+        .sort((left, right) => `${left.personKey}|${left.week}`.localeCompare(`${right.personKey}|${right.week}`));
+}
+//# sourceMappingURL=aggregate.js.map

package/dist/lib/claude.js

@@ -0,0 +1,199 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.summarizeClaudeProjectUsage = summarizeClaudeProjectUsage;
+exports.summarizeClaudeProjectUsageByDay = summarizeClaudeProjectUsageByDay;
+const promises_1 = __importDefault(require("node:fs/promises"));
+const node_path_1 = __importDefault(require("node:path"));
+const paths_1 = require("./paths");
+function isRecord(value) {
+    return typeof value === "object" && value !== null && !Array.isArray(value);
+}
+function getNumber(value) {
+    return typeof value === "number" && Number.isFinite(value) ? value : 0;
+}
+function getString(value) {
+    return typeof value === "string" && value.length > 0 ? value : null;
+}
+async function collectProjectJsonlFiles(directoryPath) {
+    try {
+        const entries = await promises_1.default.readdir(directoryPath, { withFileTypes: true });
+        const nested = await Promise.all(entries.map(async (entry) => {
+            const fullPath = node_path_1.default.join(directoryPath, entry.name);
+            if (entry.isDirectory()) {
+                return collectProjectJsonlFiles(fullPath);
+            }
+            return entry.isFile() && entry.name.endsWith(".jsonl") ? [fullPath] : [];
+        }));
+        return nested.flat();
+    }
+    catch (error) {
+        if (error.code === "ENOENT") {
+            return [];
+        }
+        throw error;
+    }
+}
+function timestampInRange(timestamp, start, end) {
+    if (!timestamp) {
+        return false;
+    }
+    const value = new Date(timestamp).getTime();
+    return Number.isFinite(value) && value >= start.getTime() && value <= end.getTime();
+}
+function localDateKey(date) {
+    const year = date.getFullYear();
+    const month = String(date.getMonth() + 1).padStart(2, "0");
+    const day = String(date.getDate()).padStart(2, "0");
+    return `${year}-${month}-${day}`;
+}
+/**
+ * 判断一条 `type:"user"` 事件是否算作一次用户请求。
+ *
+ * Claude transcript 里 `type:"user"` 还会被用作 tool_result 回填——这种伪 user 事件必须排除，
+ * 否则 userMessageCount 会被高估近 10×。sidechain（子 agent）会话里的用户提示保留计入，
+ * 因为它们仍然代表团队让 Claude 做的事，不算工具机械回填。
+ */
+function isHumanUserMessage(record) {
+    if (record.type !== "user") {
+        return false;
+    }
+    if (record.isMeta === true) {
+        return false;
+    }
+    if (record.toolUseResult !== undefined && record.toolUseResult !== null) {
+        return false;
+    }
+    if (!isRecord(record.message)) {
+        return false;
+    }
+    const content = record.message.content;
+    if (Array.isArray(content)) {
+        const hasNonToolResult = content.some((item) => isRecord(item) && item.type !== "tool_result");
+        if (!hasNonToolResult) {
+            return false;
+        }
+    }
+    return true;
+}
+function summarizeProjectTranscript(content, start, end) {
+    const lines = content.split(/\r?\n/).map((line) => line.trim()).filter((line) => line.length > 0);
+    const summary = {
+        userMessageCount: 0,
+        apiRequestCount: 0,
+        inputTokens: 0,
+        outputTokens: 0,
+        cacheReadInputTokens: 0,
+    };
+    for (const line of lines) {
+        try {
+            const record = JSON.parse(line);
+            if (!isRecord(record) || !timestampInRange(getString(record.timestamp), start, end)) {
+                continue;
+            }
+            if (isHumanUserMessage(record)) {
+                summary.userMessageCount += 1;
+            }
+            if (record.type !== "assistant" || !isRecord(record.message) || !isRecord(record.message.usage)) {
+                continue;
+            }
+            summary.apiRequestCount += 1;
+            summary.inputTokens += getNumber(record.message.usage.input_tokens);
+            summary.outputTokens += getNumber(record.message.usage.output_tokens);
+            summary.cacheReadInputTokens += getNumber(record.message.usage.cache_read_input_tokens);
+        }
+        catch {
+            continue;
+        }
+    }
+    return summary;
+}
+/**
+ * 从 Claude 本地 project transcript 中统计本周消息数、请求数和 token 用量。
+ */
+async function summarizeClaudeProjectUsage(start, end) {
+    const claudeDataDir = (0, paths_1.getClaudeDataDir)();
+    const projectDir = node_path_1.default.join(claudeDataDir, "projects");
+    const files = await collectProjectJsonlFiles(projectDir);
+    const totals = {
+        userMessageCount: 0,
+        apiRequestCount: 0,
+        inputTokens: 0,
+        outputTokens: 0,
+        cacheReadInputTokens: 0,
+        matchedFileCount: files.length,
+        claudeDataDir,
+    };
+    for (const filePath of files) {
+        try {
+            const content = await promises_1.default.readFile(filePath, "utf8");
+            const summary = summarizeProjectTranscript(content, start, end);
+            totals.userMessageCount += summary.userMessageCount;
+            totals.apiRequestCount += summary.apiRequestCount;
+            totals.inputTokens += summary.inputTokens;
+            totals.outputTokens += summary.outputTokens;
+            totals.cacheReadInputTokens += summary.cacheReadInputTokens;
+        }
+        catch {
+            continue;
+        }
+    }
+    return totals;
+}
+/**
+ * 按天汇总 Claude project transcript 中的消息数、请求数和 token 用量。
+ */
+async function summarizeClaudeProjectUsageByDay(start, end) {
+    const claudeDataDir = (0, paths_1.getClaudeDataDir)();
+    const projectDir = node_path_1.default.join(claudeDataDir, "projects");
+    const files = await collectProjectJsonlFiles(projectDir);
+    const daily = new Map();
+    for (const filePath of files) {
+        let content = "";
+        try {
+            content = await promises_1.default.readFile(filePath, "utf8");
+        }
+        catch {
+            continue;
+        }
+        const lines = content.split(/\r?\n/).map((line) => line.trim()).filter((line) => line.length > 0);
+        for (const line of lines) {
+            try {
+                const record = JSON.parse(line);
+                if (!isRecord(record)) {
+                    continue;
+                }
+                const timestamp = getString(record.timestamp);
+                if (!timestampInRange(timestamp, start, end)) {
+                    continue;
+                }
+                const date = localDateKey(new Date(timestamp));
+                const current = daily.get(date) ?? {
+                    date,
+                    userMessageCount: 0,
+                    apiRequestCount: 0,
+                    inputTokens: 0,
+                    outputTokens: 0,
+                    cacheReadInputTokens: 0,
+                };
+                if (isHumanUserMessage(record)) {
+                    current.userMessageCount += 1;
+                }
+                if (record.type === "assistant" && isRecord(record.message) && isRecord(record.message.usage)) {
+                    current.apiRequestCount += 1;
+                    current.inputTokens += getNumber(record.message.usage.input_tokens);
+                    current.outputTokens += getNumber(record.message.usage.output_tokens);
+                    current.cacheReadInputTokens += getNumber(record.message.usage.cache_read_input_tokens);
+                }
+                daily.set(date, current);
+            }
+            catch {
+                continue;
+            }
+        }
+    }
+    return daily;
+}
+//# sourceMappingURL=claude.js.map