opencode-metrics-plugin 0.1.5 → 0.2.1

package/dist/summary-store.js

@@ -1,560 +0,0 @@
-import * as fs from "fs";
-import * as path from "path";
-import { createRequire } from "node:module";
-import { resolveDirs, getSummaryFile } from "./dirs.js";
-import { log } from "./logger.js";
-const require_ = createRequire(import.meta.url);
-let sqliteUnavailable = false;
-function loadSqlite() {
-    if (sqliteUnavailable)
-        return null;
-    try {
-        return require_("node:sqlite");
-    }
-    catch {
-        sqliteUnavailable = true;
-        log.info("[summary] node:sqlite 不可用（需 Node >= 22.5），摘要/详情库停用");
-        return null;
-    }
-}
-// ─── DDL（镜像 tnnotix sqlite 方言，幂等）───────────────────────────────────
-const DDL = `
-CREATE TABLE IF NOT EXISTS session_summaries (
-  session_id TEXT PRIMARY KEY,
-  task_id TEXT,
-  agent TEXT,
-  model TEXT,
-  directory TEXT,
-  title TEXT,
-  first_user_message TEXT,
-  step_count INTEGER NOT NULL DEFAULT 0,
-  tool_calls INTEGER NOT NULL DEFAULT 0,
-  total_tokens_input BIGINT NOT NULL DEFAULT 0,
-  total_tokens_output BIGINT NOT NULL DEFAULT 0,
-  total_tokens_cache_read BIGINT NOT NULL DEFAULT 0,
-  total_tokens_cache_write BIGINT NOT NULL DEFAULT 0,
-  total_tokens_reasoning BIGINT NOT NULL DEFAULT 0,
-  total_tokens BIGINT NOT NULL DEFAULT 0,
-  duration BIGINT NOT NULL DEFAULT 0,
-  created_at TEXT,
-  source_file TEXT UNIQUE,
-  ingested_at TEXT DEFAULT CURRENT_TIMESTAMP
-);
-
-CREATE TABLE IF NOT EXISTS session_details (
-  session_id TEXT PRIMARY KEY,
-  full_data TEXT
-);
-
-CREATE INDEX IF NOT EXISTS idx_session_summaries_created_at ON session_summaries(created_at);
-CREATE INDEX IF NOT EXISTS idx_session_summaries_task_id ON session_summaries(task_id);
-CREATE INDEX IF NOT EXISTS idx_session_summaries_model ON session_summaries(model);
-`;
-// ─── 连接缓存 ────────────────────────────────────────────────────────────────
-const dbCache = new Map();
-/** 打开（或复用）摘要库连接：建目录 + WAL + busy_timeout + 幂等建表；不可用返回 null。 */
-export function openSummaryDb(summaryFile) {
-    const cached = dbCache.get(summaryFile);
-    if (cached)
-        return cached;
-    const sqlite = loadSqlite();
-    if (!sqlite)
-        return null;
-    try {
-        fs.mkdirSync(path.dirname(summaryFile), { recursive: true });
-        const db = new sqlite.DatabaseSync(summaryFile);
-        db.exec("PRAGMA journal_mode=WAL");
-        db.exec("PRAGMA busy_timeout=5000");
-        db.exec(DDL);
-        dbCache.set(summaryFile, db);
-        return db;
-    }
-    catch (err) {
-        log.info("[summary] 打开摘要库失败", { summaryFile, error: String(err) });
-        return null;
-    }
-}
-/** 关闭全部缓存的连接（宿主 dispose 时可选调用）。 */
-export function closeSummaryDbs() {
-    for (const [file, db] of dbCache) {
-        try {
-            db.close();
-        }
-        catch { }
-        dbCache.delete(file);
-    }
-}
-// ─── 快照 → 摘要列提取（与 tnnotix extractSummaryFields 同口径）──────────────
-/** 归一化 created_at（number 时间戳 / ISO 字符串 / Date）为 ISO 字符串。 */
-export function toIsoString(value) {
-    if (value == null)
-        return null;
-    const d = value instanceof Date ? value : new Date(value);
-    return isNaN(d.getTime()) ? null : d.toISOString();
-}
-/** 从 rounds 提取首条用户消息（兼容 string 与 string[] 形态）。 */
-export function firstUserMessageOf(rounds) {
-    const raw = rounds?.find((r) => r.userMessage)?.userMessage;
-    if (raw == null)
-        return undefined;
-    return Array.isArray(raw) ? raw[0] : raw;
-}
-/** 摘要表一行所需的全部字段（文件→表共用的提取逻辑）。 */
-export function extractSummaryFields(m, filePath) {
-    const tokens = m.tokens ?? { input: 0, output: 0, reasoning: 0, cacheRead: 0, cacheWrite: 0, total: 0 };
-    return {
-        sessionId: m.sessionId,
-        taskId: null,
-        agent: m.header?.agent ?? "",
-        model: m.header?.model ?? "",
-        directory: m.header?.workingDirectory ?? "",
-        title: m.header?.title ?? null,
-        firstUserMessage: firstUserMessageOf(m.rounds) ?? null,
-        stepCount: m.steps?.length ?? 0,
-        toolCalls: m.tools?.totalCalls ?? 0,
-        totalTokensInput: tokens.input ?? 0,
-        totalTokensOutput: tokens.output ?? 0,
-        totalTokensCacheRead: tokens.cacheRead ?? 0,
-        totalTokensCacheWrite: tokens.cacheWrite ?? 0,
-        totalTokensReasoning: tokens.reasoning ?? 0,
-        totalTokens: tokens.total ?? 0,
-        duration: m.duration ?? 0,
-        createdAt: toIsoString(m.header?.startTime),
-        sourceFile: filePath,
-    };
-}
-function summaryValues(f) {
-    return [
-        f.sessionId, f.taskId, f.agent, f.model, f.directory, f.title, f.firstUserMessage,
-        f.stepCount, f.toolCalls,
-        f.totalTokensInput, f.totalTokensOutput,
-        f.totalTokensCacheRead, f.totalTokensCacheWrite,
-        f.totalTokensReasoning, f.totalTokens,
-        f.duration, f.createdAt, f.sourceFile,
-    ];
-}
-// ─── upsert（force = 总是刷新；soft = 既有 total_tokens>0 不覆盖）────────────
-const SUMMARY_INSERT = `INSERT INTO session_summaries
-  (session_id, task_id, agent, model, directory, title, first_user_message,
-   step_count, tool_calls,
-   total_tokens_input, total_tokens_output,
-   total_tokens_cache_read, total_tokens_cache_write,
-   total_tokens_reasoning, total_tokens,
-   duration, created_at, source_file)
-  VALUES (?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?)`;
-/**
- * 冲突更新 SET 子句（按 mode 构建）：
- * - 计数列恒刷新；first_user_message/title 用 COALESCE 只治愈不清空（防 compact 会话丢已采值）；
- * - force 额外全量刷新展示列（model/directory/agent/created_at/source_file），
- *   口径升级（如模型短名化）后能治愈存量行，不再停留在历史插入值。
- */
-const summaryUpdateSet = (mode) => `
-ON CONFLICT (session_id) DO UPDATE SET
-  step_count = excluded.step_count,
-  tool_calls = excluded.tool_calls,
-  total_tokens_input = excluded.total_tokens_input,
-  total_tokens_output = excluded.total_tokens_output,
-  total_tokens_cache_read = excluded.total_tokens_cache_read,
-  total_tokens_cache_write = excluded.total_tokens_cache_write,
-  total_tokens_reasoning = excluded.total_tokens_reasoning,
-  total_tokens = excluded.total_tokens,
-  duration = excluded.duration,
-  first_user_message = COALESCE(excluded.first_user_message, session_summaries.first_user_message),
-  title = COALESCE(excluded.title, session_summaries.title)${mode === "force" ? `,
-  model = excluded.model,
-  directory = excluded.directory,
-  agent = excluded.agent,
-  created_at = excluded.created_at,
-  source_file = excluded.source_file` : ""},
-  ingested_at = CURRENT_TIMESTAMP`;
-const DETAIL_SOFT_GUARD = `
-  WHERE session_details.full_data IS NULL OR
-    CAST(json_extract(session_details.full_data, '$.tokens.total') AS INTEGER) IS NULL OR
-    CAST(json_extract(session_details.full_data, '$.tokens.total') AS INTEGER) = 0`;
-/** 快照双表入库（事务内 summary + detail；source_file 保留首见值）。返回是否成功。 */
-export function upsertSessionSnapshot(summaryFile, snapshot, filePath, mode = "force") {
-    const db = openSummaryDb(summaryFile);
-    if (!db)
-        return false;
-    const fields = extractSummaryFields(snapshot, filePath);
-    const payload = JSON.stringify(snapshot, null, 2);
-    const summarySql = SUMMARY_INSERT + summaryUpdateSet(mode) + (mode === "soft" ? "\n  WHERE session_summaries.total_tokens = 0" : "");
-    const detailSql = `INSERT INTO session_details (session_id, full_data)
-    VALUES (?, ?)
-    ON CONFLICT (session_id) DO UPDATE SET full_data = excluded.full_data` + (mode === "soft" ? DETAIL_SOFT_GUARD : "");
-    try {
-        db.exec("BEGIN");
-        try {
-            db.prepare(summarySql).run(...summaryValues(fields));
-            db.prepare(detailSql).run(fields.sessionId, payload);
-            db.exec("COMMIT");
-        }
-        catch (err) {
-            try {
-                db.exec("ROLLBACK");
-            }
-            catch { }
-            throw err;
-        }
-        return true;
-    }
-    catch (err) {
-        log.info("[summary] 快照入库失败", { sessionId: fields.sessionId, mode, error: String(err) });
-        return false;
-    }
-}
-/** 逐元素累加 token 用量。 */
-export function sumTokens(items) {
-    return items.reduce((acc, t) => ({
-        input: acc.input + (t.input ?? 0),
-        output: acc.output + (t.output ?? 0),
-        reasoning: acc.reasoning + (t.reasoning ?? 0),
-        cacheRead: acc.cacheRead + (t.cacheRead ?? 0),
-        cacheWrite: acc.cacheWrite + (t.cacheWrite ?? 0),
-        total: acc.total + (t.total ?? 0),
-    }), { input: 0, output: 0, reasoning: 0, cacheRead: 0, cacheWrite: 0, total: 0 });
-}
-/** a 减去 b（用于剥离子代理 token 得 main tokens）。 */
-export function subtractTokens(a, b) {
-    return {
-        input: (a.input ?? 0) - (b.input ?? 0),
-        output: (a.output ?? 0) - (b.output ?? 0),
-        reasoning: (a.reasoning ?? 0) - (b.reasoning ?? 0),
-        cacheRead: (a.cacheRead ?? 0) - (b.cacheRead ?? 0),
-        cacheWrite: (a.cacheWrite ?? 0) - (b.cacheWrite ?? 0),
-        total: (a.total ?? 0) - (b.total ?? 0),
-    };
-}
-/** 快照 → 列表摘要（含 subagent 拆分出的 main/subagent tokens）。 */
-export function snapshotToSummary(m, filePath) {
-    const subagents = m.subagents ?? [];
-    const subagentTokens = subagents.length > 0 ? sumTokens(subagents.map((s) => s.tokens)) : undefined;
-    const mainTokens = subagentTokens ? subtractTokens(m.tokens, subagentTokens) : undefined;
-    return {
-        id: m.sessionId,
-        directory: m.header?.workingDirectory ?? "",
-        model: m.header?.model ?? "",
-        agent: m.header?.agent ?? "",
-        title: m.header?.title,
-        stepCount: m.steps?.length ?? 0,
-        totalTokens: m.tokens,
-        mainTokens,
-        subagentTokens,
-        subagentCount: subagents.length > 0 ? subagents.length : undefined,
-        totalCost: 0,
-        duration: m.duration ?? 0,
-        createdAt: m.header?.startTime ?? "",
-        startTime: m.startTime,
-        endTime: m.endTime,
-        toolCalls: m.tools?.totalCalls ?? 0,
-        successRate: m.tools?.successRate ?? 0,
-        firstUserMessage: firstUserMessageOf(m.rounds),
-        filePath,
-    };
-}
-/** 快照 → 详情；默认子代理摘要化（steps 抽为 stepCount，完整 steps 走 getSubagentSteps 懒加载）。 */
-export function detailFromRaw(raw, opts = {}) {
-    const detail = {
-        ...snapshotToSummary(raw),
-        steps: raw.steps ?? [],
-        tools: raw.tools,
-        codeStats: raw.codeStats,
-        rounds: raw.rounds ?? [],
-        subagents: raw.subagents,
-        skills: raw.skills,
-        planning: raw.planning,
-        systemPrompts: raw.systemPrompts,
-    };
-    if (Array.isArray(detail.subagents) && !opts.keepSubagentSteps) {
-        detail.subagents = detail.subagents.map((s) => ({
-            sessionId: s.sessionId,
-            agent: s.agent,
-            title: s.title,
-            tokens: s.tokens,
-            tools: s.tools,
-            ...(s.userMessages ? { userMessages: s.userMessages } : {}),
-            stepCount: s.steps?.length ?? s.stepCount ?? 0,
-        }));
-    }
-    return detail;
-}
-/** 保留完整子代理 steps 的详情（供直接渲染子代理树的宿主使用）。 */
-export function detailFullFromRaw(raw) {
-    return detailFromRaw(raw, { keepSubagentSteps: true });
-}
-function buildWhere(f) {
-    const clauses = [];
-    const params = [];
-    if (f.agent) {
-        clauses.push("agent = ?");
-        params.push(f.agent);
-    }
-    if (f.agents && f.agents.length > 0) {
-        clauses.push(`agent IN (${f.agents.map(() => "?").join(",")})`);
-        params.push(...f.agents);
-    }
-    if (f.taskId) {
-        clauses.push("task_id = ?");
-        params.push(f.taskId);
-    }
-    if (f.model) {
-        clauses.push("model = ?");
-        params.push(f.model);
-    }
-    if (f.sessionIds && f.sessionIds.length > 0) {
-        clauses.push(`session_id IN (${f.sessionIds.map(() => "?").join(",")})`);
-        params.push(...f.sessionIds);
-    }
-    if (f.directoryPrefix) {
-        clauses.push("lower(replace(directory, '\\', '/')) LIKE ?");
-        params.push(f.directoryPrefix.replace(/\\/g, "/").toLowerCase() + "%");
-    }
-    const sinceIso = toIsoString(f.since);
-    if (sinceIso) {
-        clauses.push("created_at >= ?");
-        params.push(sinceIso);
-    }
-    const untilIso = toIsoString(f.until);
-    if (untilIso) {
-        clauses.push("created_at <= ?");
-        params.push(untilIso);
-    }
-    return { where: clauses.length > 0 ? ` WHERE ${clauses.join(" AND ")}` : "", params };
-}
-const SUMMARY_LIST_COLS = `session_id AS "id", task_id AS "taskId", agent, model, directory, title,
-  first_user_message AS "firstUserMessage", step_count AS "stepCount",
-  tool_calls AS "toolCalls",
-  total_tokens_input, total_tokens_output,
-  total_tokens_cache_read, total_tokens_cache_write,
-  total_tokens_reasoning, total_tokens,
-  duration, created_at AS "createdAt", source_file AS "filePath"`;
-function toInt(v) {
-    const n = typeof v === "number" ? v : parseInt(String(v), 10);
-    return Number.isFinite(n) ? n : 0;
-}
-function rowToSummary(r) {
-    return {
-        id: r.id,
-        directory: r.directory ?? "",
-        model: r.model ?? "",
-        agent: r.agent ?? "",
-        title: r.title ? r.title : undefined,
-        stepCount: toInt(r.stepCount),
-        toolCalls: toInt(r.toolCalls),
-        totalTokens: {
-            input: toInt(r.total_tokens_input),
-            output: toInt(r.total_tokens_output),
-            cacheRead: toInt(r.total_tokens_cache_read),
-            cacheWrite: toInt(r.total_tokens_cache_write),
-            reasoning: toInt(r.total_tokens_reasoning),
-            total: toInt(r.total_tokens),
-        },
-        duration: toInt(r.duration),
-        createdAt: r.createdAt ?? "",
-        firstUserMessage: r.firstUserMessage ? r.firstUserMessage : undefined,
-        filePath: r.filePath ? r.filePath : undefined,
-        totalCost: 0,
-        successRate: 0,
-    };
-}
-function resolveSummaryFile(dirs) {
-    return resolveDirs(dirs).summaryFile;
-}
-/** 摘要筛选查询（created_at 倒序；目录/筛选参数归一后走 SQL）。同步函数。 */
-export function querySummaries(filter = {}, dirs) {
-    const db = openSummaryDb(resolveSummaryFile(dirs));
-    if (!db)
-        return [];
-    const { where, params } = buildWhere(filter);
-    let limitClause = "";
-    if (filter.limit != null || filter.offset != null) {
-        limitClause = " LIMIT ? OFFSET ?";
-        params.push(filter.limit ?? -1, filter.offset ?? 0);
-    }
-    const sql = `SELECT ${SUMMARY_LIST_COLS} FROM session_summaries${where} ORDER BY created_at DESC${limitClause}`;
-    const rows = db.prepare(sql).all(...params);
-    return rows.map(rowToSummary);
-}
-/** 按同一筛选条件计数（分页用）。 */
-export function countSummaries(filter = {}, dirs) {
-    const db = openSummaryDb(resolveSummaryFile(dirs));
-    if (!db)
-        return 0;
-    const { where, params } = buildWhere(filter);
-    const row = db.prepare(`SELECT COUNT(*) AS n FROM session_summaries${where}`).get(...params);
-    return toInt(row?.n);
-}
-// ─── 详情读取（full_data 优先 → 快照文件回退）──────────────────────────────
-const SESSION_ID_RE = /^[A-Za-z0-9_-]+$/;
-function readSnapshotFile(metricsDir, sessionId) {
-    if (!SESSION_ID_RE.test(sessionId))
-        return null;
-    const filePath = path.join(metricsDir, `${sessionId}.json`);
-    try {
-        const raw = JSON.parse(fs.readFileSync(filePath, "utf-8"));
-        if (!raw?.sessionId || !raw.header)
-            return null;
-        return { raw, filePath };
-    }
-    catch {
-        return null;
-    }
-}
-function loadRaw(sessionId, dirs) {
-    const db = openSummaryDb(resolveSummaryFile(dirs));
-    if (db) {
-        try {
-            const row = db.prepare("SELECT full_data FROM session_details WHERE session_id = ?").get(sessionId);
-            if (row?.full_data) {
-                const raw = JSON.parse(row.full_data);
-                if (raw?.sessionId)
-                    return { raw, fromDb: true };
-            }
-        }
-        catch { /* 回退文件 */ }
-    }
-    const file = readSnapshotFile(resolveDirs(dirs).metricsDir, sessionId);
-    return file ? { raw: file.raw, filePath: file.filePath, fromDb: false } : null;
-}
-/** 单会话详情（DB full_data → 文件回退）；默认子代理摘要化，fullSubagents=true 保留完整 steps。 */
-export function getDetail(sessionId, dirs, opts = {}) {
-    const found = loadRaw(sessionId, dirs);
-    if (!found)
-        return null;
-    const detail = detailFromRaw(found.raw, { keepSubagentSteps: opts.fullSubagents === true });
-    if (found.filePath)
-        detail.filePath = found.filePath;
-    return detail;
-}
-/** 子代理完整 steps 懒加载（DB full_data → 文件回退）。 */
-export function getSubagentSteps(sessionId, index, dirs) {
-    const found = loadRaw(sessionId, dirs);
-    const sa = found?.raw.subagents?.[index];
-    if (!sa)
-        return null;
-    return sa.steps ?? [];
-}
-/** 原始快照 JSON 字符串（文件优先 → DB full_data 回退；等价"下载"能力）。 */
-export function getSessionRaw(sessionId, dirs) {
-    if (!SESSION_ID_RE.test(sessionId))
-        return null;
-    try {
-        return fs.readFileSync(path.join(resolveDirs(dirs).metricsDir, `${sessionId}.json`), "utf-8");
-    }
-    catch { /* 回退 DB */ }
-    const db = openSummaryDb(resolveSummaryFile(dirs));
-    if (db) {
-        try {
-            const row = db.prepare("SELECT full_data FROM session_details WHERE session_id = ?").get(sessionId);
-            if (row?.full_data)
-                return row.full_data;
-        }
-        catch { }
-    }
-    return null;
-}
-/** 给会话标记任务分组（宿主任务关联）。返回是否命中行。 */
-export function setSessionTaskId(sessionId, taskId, dirs) {
-    const db = openSummaryDb(resolveSummaryFile(dirs));
-    if (!db)
-        return false;
-    const r = db.prepare("UPDATE session_summaries SET task_id = ? WHERE session_id = ?").run(taskId, sessionId);
-    return Number(r.changes) > 0;
-}
-/**
- * 扫描 metricsDir 现有快照 JSON 重建摘要行（soft：live 已有数据不覆盖）。
- * 摘要库缺省为全局共享库（多宿主聚合）；显式 summaryFile 可指向隔离库。
- * prune=true 时清理 source_file 已不存在于磁盘的孤儿行（双表同删）——
- * 共享库语义下会清理**全库**的孤儿行（含其他宿主目录已消失的快照），属预期行为。
- */
-export function reindexDir(metricsDir, opts = {}) {
-    const summaryFile = opts.summaryFile ?? getSummaryFile();
-    const result = { scanned: 0, upserted: 0, pruned: 0 };
-    let files = [];
-    try {
-        files = fs.readdirSync(metricsDir).filter((f) => f.endsWith(".json"));
-    }
-    catch {
-        return result;
-    }
-    for (const f of files) {
-        const filePath = path.join(metricsDir, f);
-        result.scanned++;
-        try {
-            const raw = JSON.parse(fs.readFileSync(filePath, "utf-8"));
-            if (!raw?.sessionId || !raw.header)
-                continue;
-            if (upsertSessionSnapshot(summaryFile, raw, filePath, "soft"))
-                result.upserted++;
-        }
-        catch { /* 跳过坏文件 */ }
-    }
-    if (opts.prune) {
-        const db = openSummaryDb(summaryFile);
-        if (db) {
-            try {
-                const rows = db.prepare("SELECT session_id, source_file FROM session_summaries").all();
-                db.exec("BEGIN");
-                try {
-                    const delSummary = db.prepare("DELETE FROM session_summaries WHERE session_id = ?");
-                    const delDetail = db.prepare("DELETE FROM session_details WHERE session_id = ?");
-                    for (const r of rows) {
-                        if (!r.source_file || !fs.existsSync(r.source_file)) {
-                            delSummary.run(r.session_id);
-                            delDetail.run(r.session_id);
-                            result.pruned++;
-                        }
-                    }
-                    db.exec("COMMIT");
-                }
-                catch (err) {
-                    try {
-                        db.exec("ROLLBACK");
-                    }
-                    catch { }
-                    throw err;
-                }
-            }
-            catch (err) {
-                log.info("[summary] prune 失败", { summaryFile, error: String(err) });
-            }
-        }
-    }
-    return result;
-}
-/**
- * 删除双表中的会话行（宿主删除链路用）。
- * 返回是否删除及被删行的 source_file；库不存在/行不存在时 deleted=false。
- */
-export function removeSession(sessionId, dirs) {
-    const db = openSummaryDb(resolveSummaryFile(dirs));
-    if (!db)
-        return { deleted: false };
-    try {
-        const row = db.prepare("SELECT source_file FROM session_summaries WHERE session_id = ?")
-            .get(sessionId);
-        db.exec("BEGIN");
-        let changes = 0;
-        try {
-            changes = Number(db.prepare("DELETE FROM session_summaries WHERE session_id = ?").run(sessionId).changes);
-            db.prepare("DELETE FROM session_details WHERE session_id = ?").run(sessionId);
-            db.exec("COMMIT");
-        }
-        catch (err) {
-            try {
-                db.exec("ROLLBACK");
-            }
-            catch { }
-            throw err;
-        }
-        return { deleted: changes > 0, sourceFile: row?.source_file ?? undefined };
-    }
-    catch (err) {
-        log.info("[summary] 删除会话行失败", { sessionId, error: String(err) });
-        return { deleted: false };
-    }
-}
-/** 进程级默认摘要库路径（与 getMetricsDir/getEventsDir 对称的便捷取法）。 */
-export function getDefaultSummaryFile() {
-    return getSummaryFile();
-}