@agimon-ai/po-file-edit 0.0.0-alpha.78 → 0.0.0-alpha.79

package/dist/extensions/server.cjs

@@ -28,6 +28,7 @@ let _agimon_ai_po_core_serverPlugin = require("@agimon-ai/po-core/serverPlugin")
 let _agimon_ai_po_core_extensionFile = require("@agimon-ai/po-core/extensionFile");
 let _agimon_ai_po_core_childSession = require("@agimon-ai/po-core/childSession");
 let _agimon_ai_po_core_hubChannel = require("@agimon-ai/po-core/hubChannel");
+let _agimon_ai_po_core_packageApi = require("@agimon-ai/po-core/packageApi");
 let node_fs_promises = require("node:fs/promises");
 node_fs_promises = __toESM(node_fs_promises, 1);
 let node_path = require("node:path");
@@ -64,12 +65,6 @@ const MINUTE_MS = 6e4;
 const SHORT_ID_LENGTH = 8;
 /** Pi's built-in grep prints match rows as `relative/path:line: text`. */
 const PI_GREP_ROW = /^(.+?):(\d+): /u;
-function forgetPresence(ledger, agentKey) {
-	for (const [file, entries] of ledger) {
-		entries.delete(agentKey);
-		if (entries.size === 0) ledger.delete(file);
-	}
-}
 function stringArg(args, key) {
 	if (typeof args !== "object" || args === null) return void 0;
 	const value = args[key];
@@ -105,26 +100,6 @@ function ago(elapsed) {
 	const minutes = Math.floor(elapsed / MINUTE_MS);
 	return minutes < 1 ? "under 1m ago" : `${minutes}m ago`;
 }
-function liveOthers(ledger, file, self) {
-	const entries = ledger.get(file);
-	if (entries === void 0) return [];
-	const live = [];
-	for (const [key, entry] of entries) {
-		if (key === self) continue;
-		if (entry.agent.isLive()) live.push(entry);
-		else entries.delete(key);
-	}
-	return live.sort((left, right) => right.at - left.at);
-}
-function record(ledger, file, agent, tag, at) {
-	const entries = ledger.get(file) ?? /* @__PURE__ */ new Map();
-	entries.set(agent.key, {
-		agent,
-		tag,
-		at
-	});
-	ledger.set(file, entries);
-}
 function renderNote(lines) {
 	const shown = lines.slice(0, MAX_NOTE_LINES);
 	const more = lines.length - shown.length;
@@ -140,14 +115,13 @@ function renderNote(lines) {
 * Returns the content with one note appended, or undefined when there is nothing to add, so callers
 * never rewrite a result needlessly.
 */
-async function observePresence(ledger, agent, call, now = Date.now()) {
+async function observePresence(presence, agent, call, now = Date.now()) {
 	const recording = EDIT_TOOLS.has(call.toolName) && !call.isError;
-	if (!recording && ledger.size === 0) return void 0;
 	const lines = [];
 	for (const file of await touchedFiles(call)) {
 		const real = await node_fs_promises.default.realpath(file).catch(() => void 0);
 		if (real === void 0) continue;
-		const others = liveOthers(ledger, real, agent.key);
+		const others = presence.touch(real, agent);
 		let bytes;
 		if (others.length > 0) {
 			bytes = await node_fs_promises.default.readFile(real).catch(() => void 0);
@@ -160,7 +134,11 @@ async function observePresence(ledger, agent, call, now = Date.now()) {
 		}
 		if (recording) {
 			bytes ??= await node_fs_promises.default.readFile(real).catch(() => void 0);
-			if (bytes !== void 0) record(ledger, real, agent, (0, _agimon_ai_po_hashline_files.computeFileTag)(bytes), now);
+			if (bytes !== void 0) presence.record(real, {
+				agent,
+				tag: (0, _agimon_ai_po_hashline_files.computeFileTag)(bytes),
+				at: now
+			});
 		}
 	}
 	if (lines.length === 0) return void 0;
@@ -171,7 +149,7 @@ async function observePresence(ledger, agent, call, now = Date.now()) {
 }
 /** Edit presence for one hub session and the in-process children it spawns. */
 function createSessionPresence(options) {
-	const { sessionId, ledger } = options;
+	const { sessionId, presence } = options;
 	const sessionAgent = {
 		key: `session:${sessionId}`,
 		label: () => `session ${sessionId}${options.workflowStep ? " (workflow step)" : ""}`,
@@ -193,7 +171,7 @@ function createSessionPresence(options) {
 				const toolName = event.toolName;
 				if (execution.sessionId !== sessionId || typeof toolName !== "string" || !Array.isArray(event.content)) return;
 				const content = event.content;
-				const next = await safely(() => observePresence(ledger, sessionAgent, {
+				const next = await safely(() => observePresence(presence, sessionAgent, {
 					toolName,
 					args: event.args,
 					content,
@@ -214,7 +192,7 @@ function createSessionPresence(options) {
 			return {
 				beforeTool: () => void 0,
 				async afterTool(event) {
-					const next = await safely(() => observePresence(ledger, child, {
+					const next = await safely(() => observePresence(presence, child, {
 						toolName: event.toolName,
 						args: event.args,
 						content: event.content,
@@ -223,10 +201,10 @@ function createSessionPresence(options) {
 					}));
 					return next === void 0 ? void 0 : { content: next };
 				},
-				dispose: async () => forgetPresence(ledger, child.key)
+				dispose: async () => presence.forget(child.key)
 			};
 		} },
-		dispose: () => forgetPresence(ledger, sessionAgent.key)
+		dispose: () => presence.forget(sessionAgent.key)
 	};
 }
 //#endregion
@@ -2139,11 +2117,11 @@ function createFileEditSession({ agent, host, context }) {
 //#endregion
 //#region src/extensions/workspaces/sessions/(backend)/_lib/serverRoot.ts
 function createPresence({ agent, host, context }) {
-	const ledger = host.context.editPresence;
-	if (!agent || !ledger) return void 0;
+	const presence = context.get(_agimon_ai_po_core_packageApi.PO_EDIT_PRESENCE_SERVICE);
+	if (!agent || !presence) return void 0;
 	const execution = agent.context;
 	return createSessionPresence({
-		ledger,
+		presence,
 		sessionId: execution.sessionId,
 		cwd: execution.cwd,
 		workflowStep: execution.sessionContext?.provenance === _agimon_ai_po_core_hubChannel.WORKFLOW_STEP_SESSION_PROVENANCE,