@morlay/session-rdb 0.0.21-alpha.6 → 0.0.21-alpha.8

package/dist/{log-qCNkDjh5.mjs → log-DTcvsMz7.mjs}

@@ -1,5 +1,156 @@
 import { SessionSeq } from "@deepseek-ai/dsh-session";
 import { sessionFormatCatalog } from "@deepseek-ai/dsh-session-format-catalog";
+//#region src/usage.ts
+const SESSION_USAGE_PATH = "/api/session.usage";
+const ROLLING_DAYS = {
+	"7d": 7,
+	"30d": 30,
+	"90d": 90
+};
+function localDayStart(now) {
+	const start = new Date(now);
+	start.setHours(0, 0, 0, 0);
+	return start;
+}
+/**
+* 范围起点（含），不限时为 undefined。**所有起点都对齐到 host 本地时区的零点**——`day` / `week` 是
+* 今天 / 本周一的零点，`7d` / `30d` / `90d` 是「今天零点往前 N-1 天」（含今天共 N 个自然日）。
+* 对齐的意义：事件级表按毫秒过滤、会话汇总表按本地日过滤，两者因此严格等价。
+* @param range - 语义键。
+* @param now - 当前时刻。
+* @returns 起点毫秒时间戳，或 undefined。
+*/
+function resolveUsageSince(range, now) {
+	switch (range) {
+		case "all": return;
+		case "day": return localDayStart(now).getTime();
+		case "week": {
+			const start = localDayStart(now);
+			start.setDate(start.getDate() - (start.getDay() + 6) % 7);
+			return start.getTime();
+		}
+		default: {
+			const start = localDayStart(now);
+			start.setDate(start.getDate() - (ROLLING_DAYS[range] - 1));
+			return start.getTime();
+		}
+	}
+}
+function parseUsageRange(value) {
+	return value === "day" || value === "week" || value === "7d" || value === "30d" || value === "90d" ? value : "all";
+}
+/** 计入活动计数的事件类型（轮次 / 步骤 / 用户输入 / 工具调用）。 */
+const COUNTED_EVENT_TYPES = [
+	"turn/start",
+	"step/start",
+	"user/message",
+	"tool/call"
+];
+/** 按事件类型把计数累加到活动指标上（表里按类型存，读的时候折成四项）。 */
+function addActivityCount(target, type, count) {
+	switch (type) {
+		case "turn/start":
+			target.turns += count;
+			return target;
+		case "step/start":
+			target.steps += count;
+			return target;
+		case "user/message":
+			target.userInputs += count;
+			return target;
+		case "tool/call":
+			target.toolCalls += count;
+			return target;
+		default: return target;
+	}
+}
+/** 毫秒时间戳 → host 本地日（`YYYY-MM-DD`，与 SQL 的 localtime 口径一致）。 */
+function localDayKey(ms) {
+	const date = new Date(ms);
+	const pad = (value) => String(value).padStart(2, "0");
+	return `${String(date.getFullYear())}-${pad(date.getMonth() + 1)}-${pad(date.getDate())}`;
+}
+function emptyTotals() {
+	return {
+		turns: 0,
+		steps: 0,
+		userInputs: 0,
+		toolCalls: 0,
+		inputTokens: 0,
+		outputTokens: 0,
+		cacheReadTokens: 0,
+		reasoningTokens: 0,
+		totalTokens: 0
+	};
+}
+function addTokenTotals(target, row) {
+	target.inputTokens += row.inputTokens;
+	target.outputTokens += row.outputTokens;
+	target.cacheReadTokens += row.cacheReadTokens;
+	target.reasoningTokens += row.reasoningTokens;
+	target.totalTokens += row.totalTokens;
+	return target;
+}
+function addActivityTotals(target, row) {
+	target.turns += row.turns;
+	target.steps += row.steps;
+	target.userInputs += row.userInputs;
+	target.toolCalls += row.toolCalls;
+	return target;
+}
+/** 把一行的整套指标累加进目标（后端合并「是否子代理」两组时用）。 */
+function addTotals(target, row) {
+	addActivityTotals(target, row);
+	addTokenTotals(target, row);
+	return target;
+}
+/**
+* 用量统计通道：一次请求回报总览（含 subagent 拆分）、按天 × 模型的桶与按会话的行。
+* 聚合只算被会话引用的事件行——fork 共享行因此只计一次，已删会话留下的孤儿行不计。
+*/
+function registerSessionUsage(ctx, persistence) {
+	ctx.inject(["webServer", "connection"], (webCtx) => {
+		const webServer = webCtx.webServer;
+		const connection = webCtx.get("connection");
+		return webCtx.effect(() => webServer.register({
+			kind: "exact",
+			path: SESSION_USAGE_PATH,
+			handler: async (req, res) => {
+				const rejection = connection.requestRejection(req);
+				if (rejection !== void 0) {
+					res.writeHead(rejection);
+					res.end(rejection === 401 ? "unauthorized" : "forbidden");
+					return;
+				}
+				if (req.method !== "POST") {
+					res.writeHead(405, { "content-type": "application/json" });
+					res.end(JSON.stringify({ error: "method not allowed" }));
+					return;
+				}
+				const chunks = [];
+				for await (const chunk of req) chunks.push(chunk);
+				const raw = Buffer.concat(chunks).toString("utf8");
+				let envelope = {};
+				if (raw !== "") try {
+					envelope = JSON.parse(raw);
+				} catch {
+					res.writeHead(400, { "content-type": "application/json" });
+					res.end(JSON.stringify({ error: "request body is not JSON" }));
+					return;
+				}
+				try {
+					const report = await persistence.usageReport(resolveUsageSince(parseUsageRange(envelope.range), Date.now()));
+					res.writeHead(200, { "content-type": "application/json" });
+					res.end(JSON.stringify(report));
+				} catch (error) {
+					res.writeHead(500, { "content-type": "application/json" });
+					res.end(JSON.stringify({ error: error instanceof Error ? error.message : "usage report failed" }));
+				}
+			}
+		}), `session-rdb: ${SESSION_USAGE_PATH} route`);
+	});
+}
+//#endregion
 //#region src/log.ts
 function normalizeSurfaceOp(value) {
 	if (typeof value !== "object" || value === null || Array.isArray(value)) return value;
@@ -354,15 +505,15 @@ function renameLegacyPtcEvents(events) {
 *   （v4 起 system 消息只认这个 kind）；
 * - `tool/result` 的消息：v3 是 user 角色、结果块包在 `content[0]`；v4 是 `role: 'tool'` + 顶层
 *   `toolCallId` + `content` 就是结果块；
-* - 本仓库自己写过、上游已删除的事件类型（见 `LEGACY_OWN_EVENT_TYPES`）标 `ignorable: true`——v4 的
-*   校验对未知类型只接受可忽略信封。
+* - **本仓库自造、上游不认识的事件类型**（见 `OWN_EVENT_TYPES`：历史遗留的与我们当前在写的）标
+*   `ignorable: true`——v4 的校验对未知类型只接受可忽略信封。
 *
 * 只认这张白名单：**其它未知类型保持 fail loud**（那是「数据来自更新的 harness」的信号，不能吞掉）。
 */
 function normalizeToCurrentShape(events) {
 	for (const event of events) {
 		const type = event.type;
-		if (LEGACY_OWN_EVENT_TYPES.has(type)) {
+		if (OWN_EVENT_TYPES.has(type)) {
 			event.ignorable = true;
 			continue;
 		}
@@ -405,16 +556,36 @@ function liftToolResultMessage(message) {
 	};
 }
 /**
-* 本仓库写过、后来连同机制一起删除的事件类型：读的时候标 `ignorable: true` 让校验跳过它们。
-* 新增历史类型时登记到这里——**不要**用「所有未知类型」代替（那会掩盖更新版本的未知事件）。
+* 落库前给**本仓库自造的事件类型**补上 `ignorable: true` 信封。
+*
+* 上游的持久化校验（`validateStoredEvents`）只放行「已知类型」或「带可忽略信封的未知类型」，我们的类型
+* 按构造就在已知表之外——不补的话连**写入**都会被拒（`appendBatch` 里那道校验）。上游 `Session.append`
+* 没有承载这个信封的口子（第三个参数只服务 surface 事件），所以这一笔由持久化层补：canonical log 里的
+* 形状不变，落库的那一份带上信封。
+* @param events - 即将落库的批次（`materializeAppendBatch` 的 JSON 快照副本，可写）。
+*/
+function sealOwnEvents(events) {
+	for (const event of events) if (OWN_EVENT_TYPES.has(event.type)) event.ignorable = true;
+}
+/**
+* 本仓库自造、上游不认识的事件类型：读的时候标 `ignorable: true`，v4 的持久化校验才肯跳过它们
+* （`validateStoredEvents` 只放行「已知类型」或「带可忽略信封的未知类型」）。
+*
+* 两类都在这张表里：
+* - **历史遗留**：写过、后来连同机制一起删除的（`session-branch/version`）；
+* - **当前在写**：本仓库的插件自己声明的事件类型（`session-mode/selected`）——上游的
+*   `KNOWN_SESSION_EVENT_TYPES` 由上游仓库的声明生成，外部插件的事件按构造就在它之外。
+*
+* 新增自造事件类型时**必须**登记到这里，否则读路径 fail loud（真回归：`session-mode/selected` 漏登记，
+* 会话打不开）。**不要**用「所有未知类型」代替（那会掩盖更新版本的未知事件——那种必须拒读）。
 */
-const LEGACY_OWN_EVENT_TYPES = /* @__PURE__ */ new Set(["session-branch/version"]);
-/** 行里是否还带着旧代消息形状（`session-branch/version` 这类自造事件也算）——写路径曾把回退视图的结果
-* 以当前版本号落库，所以**版本号不可信**，读之前要按内容判一次。 */
-function hasLegacyShape(events) {
+const OWN_EVENT_TYPES = /* @__PURE__ */ new Set(["session-branch/version", "session-mode/selected"]);
+/** 这批事件是否需要按当前形状归一（旧代消息形状，或尚未标 `ignorable` 的自造事件类型）——写路径曾把回退
+* 视图的结果以当前版本号落库，所以**版本号不可信**，读之前要按内容判一次。 */
+function needsShapeAdoption(events) {
 	for (const event of events) {
 		const type = event.type;
-		if (LEGACY_OWN_EVENT_TYPES.has(type)) return true;
+		if (OWN_EVENT_TYPES.has(type)) return true;
 		if (type === "system/message") {
 			if ((event.data["message"]?.["source"])?.["kind"] === "plugin") return true;
 			continue;
@@ -555,10 +726,13 @@ function textField(value) {
 /**
 * 一条事件行的用量（只有带 `usage` 的 `assistant/message` 才有）：写路径据此记录
 * `t_event_usage`，统计因此不必逐行解析 JSON。两种 `f_data` 结构（带信封 / 老格式）都认。
+* 读侧维度一起物化：本地日由事件时间算，`f_referenced` 直接 1（这一批事件行与桥接行同事务，
+* 提交即被引用），`f_subagent` 取写入它的会话是否子代理。
 * @param event - 待写入的事件行。
+* @param subagent - 写入它的会话是否 `origin === 'subagent'`。
 * @returns 用量行，或该事件没有用量时的 undefined。
 */
-function usageRowOf(event) {
+function usageRowOf(event, subagent) {
 	if (event.fType !== "assistant/message") return void 0;
 	let parsed;
 	try {
@@ -573,8 +747,11 @@ function usageRowOf(event) {
 	return {
 		fEventId: event.fEventId,
 		fCreatedAt: event.fCreatedAt,
+		fDay: localDayKey(event.fCreatedAt),
 		fProvider: textField(source?.["provider"]),
 		fModel: textField(source?.["model"]),
+		fReferenced: 1,
+		fSubagent: subagent ? 1 : 0,
 		fInputTokens: numericField(usage["inputTokens"]),
 		fOutputTokens: numericField(usage["outputTokens"]),
 		fCacheReadTokens: numericField(usage["cacheReadTokens"]),
@@ -583,4 +760,4 @@ function usageRowOf(event) {
 	};
 }
 //#endregion
-export { sessionInsertRow as _, orphanInboxSpliceSeqs as a, toJsonlArtifact as b, repairAssistantSettlement as c, repairRequestHeaders as d, repairSurfaceOps as f, sessionConflictRow as g, scanRows as h, normalizeToCurrentShape as i, repairOrphanInboxSplices as l, rowToMeta as m, findSurfaceRepairs as n, recomputeReplaceProvenance as o, rowToEvent as p, hasLegacyShape as r, renameLegacyPtcEvents as s, LEGACY_OWN_EVENT_TYPES as t, repairReadView as u, syncMeteringRanges as v, usageRowOf as x, titleOfEventData as y };
+export { registerSessionUsage as A, COUNTED_EVENT_TYPES as C, addTotals as D, addTokenTotals as E, emptyTotals as O, usageRowOf as S, addActivityTotals as T, sessionConflictRow as _, orphanInboxSpliceSeqs as a, titleOfEventData as b, repairAssistantSettlement as c, repairRequestHeaders as d, repairSurfaceOps as f, sealOwnEvents as g, scanRows as h, normalizeToCurrentShape as i, localDayKey as k, repairOrphanInboxSplices as l, rowToMeta as m, findSurfaceRepairs as n, recomputeReplaceProvenance as o, rowToEvent as p, needsShapeAdoption as r, renameLegacyPtcEvents as s, OWN_EVENT_TYPES as t, repairReadView as u, sessionInsertRow as v, addActivityCount as w, toJsonlArtifact as x, syncMeteringRanges as y };