112 lines
3.4 KiB
JavaScript
112 lines
3.4 KiB
JavaScript
#!/usr/bin/env node
|
||
/**
|
||
* 跨 Hook 会话追踪 (Phase 3)
|
||
*
|
||
* 提供共享 traceId，使 3 个 hooks 的事件可以关联分析。
|
||
* 所有消费者必须 try/catch 包裹，fail-open。
|
||
*/
|
||
|
||
const fs = require('fs');
|
||
const { safeAppendJsonl } = require('../hooks/lib/safe-append.js');
|
||
const path = require('path');
|
||
|
||
// ─── 路径解析 ────────────────────────────────────────
|
||
let debugDir;
|
||
try {
|
||
const { PATHS } = require('./paths.config.js');
|
||
debugDir = PATHS.debugDir;
|
||
} catch {
|
||
debugDir = path.resolve(__dirname, '..', 'debug');
|
||
}
|
||
|
||
const SESSION_TRACE_FILE = path.join(debugDir, 'session-trace.json');
|
||
const SESSION_LOCK_FILE = path.join(debugDir, 'session-active.lock');
|
||
|
||
/**
|
||
* 获取当前会话的 traceId（同 session 复用）
|
||
* @returns {{ sessionId: string, traceId: string, startedAt: string, eventCount: number }}
|
||
*/
|
||
function getSessionTrace() {
|
||
if (!fs.existsSync(debugDir)) fs.mkdirSync(debugDir, { recursive: true });
|
||
|
||
// 读取当前 sessionId
|
||
let sessionId = 'unknown';
|
||
try {
|
||
if (fs.existsSync(SESSION_LOCK_FILE)) {
|
||
const lock = JSON.parse(fs.readFileSync(SESSION_LOCK_FILE, 'utf8'));
|
||
sessionId = lock.sessionId || lock.id || 'unknown';
|
||
}
|
||
} catch {}
|
||
|
||
// 读取或创建 trace 状态
|
||
let trace = null;
|
||
try {
|
||
if (fs.existsSync(SESSION_TRACE_FILE)) {
|
||
trace = JSON.parse(fs.readFileSync(SESSION_TRACE_FILE, 'utf8'));
|
||
}
|
||
} catch {}
|
||
|
||
// 同 session 复用 traceId，否则创建新的
|
||
if (trace && trace.sessionId === sessionId && sessionId !== 'unknown') {
|
||
return trace;
|
||
}
|
||
|
||
// 新 trace
|
||
const traceId = `t-${Date.now().toString(36)}-${Math.random().toString(36).slice(2, 8)}`;
|
||
const newTrace = {
|
||
sessionId,
|
||
traceId,
|
||
startedAt: new Date().toISOString(),
|
||
eventCount: 0,
|
||
};
|
||
|
||
try {
|
||
// W5: 原子写入 (temp+rename)
|
||
var tmp = SESSION_TRACE_FILE + '.tmp.' + process.pid;
|
||
fs.writeFileSync(tmp, JSON.stringify(newTrace, null, 2) + '\n');
|
||
fs.renameSync(tmp, SESSION_TRACE_FILE);
|
||
} catch {}
|
||
|
||
return newTrace;
|
||
}
|
||
|
||
/**
|
||
* 追加事件到日期分片的 trace 日志
|
||
* @param {string} hookName - 来源 hook 名称
|
||
* @param {string} eventType - 事件类型 (detection|block-warn|pass|outcome)
|
||
* @param {Object} data - 附加数据
|
||
*/
|
||
function appendTraceEvent(hookName, eventType, data) {
|
||
try {
|
||
if (!fs.existsSync(debugDir)) fs.mkdirSync(debugDir, { recursive: true });
|
||
|
||
const trace = getSessionTrace();
|
||
const dateStr = new Date().toISOString().slice(0, 10);
|
||
const logFile = path.join(debugDir, `trace-${dateStr}.jsonl`);
|
||
|
||
const event = {
|
||
ts: new Date().toISOString(),
|
||
traceId: trace.traceId,
|
||
sessionId: trace.sessionId,
|
||
hook: hookName,
|
||
event: eventType,
|
||
data: data || {},
|
||
};
|
||
|
||
safeAppendJsonl(logFile, event);
|
||
|
||
// 更新 eventCount (原子写入)
|
||
trace.eventCount = (trace.eventCount || 0) + 1;
|
||
try {
|
||
var tmp2 = SESSION_TRACE_FILE + '.tmp.' + process.pid;
|
||
fs.writeFileSync(tmp2, JSON.stringify(trace, null, 2) + '\n');
|
||
fs.renameSync(tmp2, SESSION_TRACE_FILE);
|
||
} catch {}
|
||
} catch {}
|
||
}
|
||
|
||
// ─── 导出 ─────────────────────────────────────────────
|
||
if (typeof module !== 'undefined') {
|
||
module.exports = { getSessionTrace, appendTraceEvent };
|
||
}
|