@agimon-ai/po-file-edit 0.0.0-alpha.44 → 0.0.0-alpha.46

package/dist/extensions/server.cjs

@@ -34,9 +34,8 @@ let node_path = require("node:path");
 node_path = __toESM(node_path, 1);
 let _agimon_ai_po_hashline = require("@agimon-ai/po-hashline");
 let _agimon_ai_po_hashline_files = require("@agimon-ai/po-hashline/files");
-let node_fs = require("node:fs");
-node_fs = __toESM(node_fs, 1);
 let _agimon_ai_po_core_childProcess = require("@agimon-ai/po-core/childProcess");
+let _agimon_ai_po_core_platform = require("@agimon-ai/po-core/platform");
 let _agimon_ai_po_core_telemetry = require("@agimon-ai/po-core/telemetry");
 let node_crypto = require("node:crypto");
 let _agimon_ai_po_core_piSettings = require("@agimon-ai/po-core/piSettings");
@@ -47,7 +46,6 @@ ignore = __toESM(ignore, 1);
 let node_os = require("node:os");
 node_os = __toESM(node_os, 1);
 let _agimon_ai_po_config = require("@agimon-ai/po-config");
-let _agimon_ai_po_core_platform = require("@agimon-ai/po-core/platform");
 let hono = require("hono");
 //#region src/constants/editPresence.ts
 const PRESENCE_NOTE_TITLE = "[Po: other active agents edited files in this result]";
@@ -717,17 +715,17 @@ var FileEditPaths = class {
 	* the hub and the session API can find it from another process knowing only
 	* those two things.
 	*/
-	timelinePath(cwd, sessionKey) {
+	async timelinePath(cwd, sessionKey) {
 		const directory = this.stateDirectory();
-		node_fs.default.mkdirSync(directory, { recursive: true });
-		return node_path.default.join(directory, `${hash(node_fs.default.realpathSync(cwd))}-${hash(sessionKey)}.jsonl`);
+		await node_fs_promises.default.mkdir(directory, { recursive: true });
+		return node_path.default.join(directory, `${hash(await node_fs_promises.default.realpath(cwd))}-${hash(sessionKey)}.jsonl`);
 	}
 	/**
 	* Where this session's content snapshots live: beside its timeline and named
 	* after it, so the two are found together and cleared together.
 	*/
-	snapshotsPath(cwd, sessionKey) {
-		return `${this.timelinePath(cwd, sessionKey).replace(/\.jsonl$/u, "")}.blobs`;
+	async snapshotsPath(cwd, sessionKey) {
+		return `${(await this.timelinePath(cwd, sessionKey)).replace(/\.jsonl$/u, "")}.blobs`;
 	}
 };
 //#endregion
@@ -1576,8 +1574,15 @@ function createFileEditsApi(options = {}) {
 	const app = new hono.Hono();
 	if (options.sessionId !== void 0 && cwd !== "") {
 		const sessionKey = paths.sessionKey(options.sessionId);
-		if (options.timeline === void 0) timeline.initialize(paths.timelinePath(cwd, sessionKey));
-		if (options.snapshots === void 0) snapshots.initialize(paths.snapshotsPath(cwd, sessionKey));
+		let initializing;
+		app.use(async (_context, next) => {
+			initializing ??= (async () => {
+				if (options.timeline === void 0) timeline.initialize(await paths.timelinePath(cwd, sessionKey));
+				if (options.snapshots === void 0) snapshots.initialize(await paths.snapshotsPath(cwd, sessionKey));
+			})();
+			await initializing;
+			await next();
+		});
 	}
 	/**
 	* The file a request names, rejected unless this session actually recorded a
@@ -1834,9 +1839,9 @@ async function sweepSessionState(options) {
 * startup.
 */
 /** Whether the file is still there to be opened. */
-function stillExists(filePath) {
+async function stillExists(filePath) {
 	try {
-		return node_fs.default.statSync(filePath).isFile();
+		return (await node_fs_promises.default.stat(filePath)).isFile();
 	} catch {
 		return false;
 	}
@@ -1851,15 +1856,17 @@ function stillExists(filePath) {
 * is gone" is a dead end; the timeline still holds the change, so a tab already
 * open on the file keeps working and says so.
 */
-function readSessionFiles(timelinePath, cwd) {
+async function readSessionFiles(timelinePath, cwd) {
 	let content;
 	try {
-		content = node_fs.default.readFileSync(timelinePath, "utf8");
+		content = await node_fs_promises.default.readFile(timelinePath, "utf8");
 	} catch {
 		return [];
 	}
 	const events = confirmedChanges(parseTimeline(content));
-	const items = foldEntries(events).filter((entry) => stillExists(entry.path)).map((entry) => ({
+	const entries = foldEntries(events);
+	const existing = await Promise.all(entries.map((entry) => stillExists(entry.path)));
+	const items = entries.filter((_entry, index) => existing[index]).map((entry) => ({
 		path: entry.path,
 		relPath: node_path.default.relative(cwd, entry.path) || entry.path,
 		tool: entry.tool,
@@ -1868,7 +1875,7 @@ function readSessionFiles(timelinePath, cwd) {
 		diffable: isDiffable(foldVersions(events, entry.path))
 	}));
 	try {
-		return filterPoIgnoredFiles(items, node_fs.default.readFileSync(node_path.default.join(cwd, ".poignore"), "utf8"));
+		return filterPoIgnoredFiles(items, await node_fs_promises.default.readFile(node_path.default.join(cwd, ".poignore"), "utf8"));
 	} catch {
 		return items;
 	}
@@ -1894,19 +1901,27 @@ function createFilesChannel() {
 				sessionAdded(scope) {
 					subscriptions.get(scope.sessionId)?.();
 					const paths = new FileEditPaths(scope.environment);
-					let timelinePath;
-					try {
-						timelinePath = paths.timelinePath(scope.cwd, paths.sessionKey(scope.sessionId));
-					} catch {}
-					const items = timelinePath === void 0 ? [] : readSessionFiles(timelinePath, scope.cwd);
-					latest.set(scope.sessionId, items);
-					host.publish(scope.sessionId, { items });
+					let current = true;
+					let receivedDirect = false;
 					const unsubscribe = host.directEvents.subscribe(filesChannelType, scope.sessionId, (payload) => {
 						if (!filesPayload(payload)) return;
+						receivedDirect = true;
 						latest.set(scope.sessionId, payload.items);
 						host.publish(scope.sessionId, payload);
 					}, { replayLatest: true });
-					subscriptions.set(scope.sessionId, unsubscribe);
+					subscriptions.set(scope.sessionId, () => {
+						current = false;
+						unsubscribe();
+					});
+					(async () => {
+						let items = [];
+						try {
+							items = await readSessionFiles(await paths.timelinePath(scope.cwd, paths.sessionKey(scope.sessionId)), scope.cwd);
+						} catch {}
+						if (!current || receivedDirect) return;
+						latest.set(scope.sessionId, items);
+						host.publish(scope.sessionId, { items });
+					})().catch((error) => host.onNotice(`File-edit channel seed failed: ${String(error)}`));
 				},
 				sessionRemoved(sessionId) {
 					subscriptions.get(sessionId)?.();
@@ -1925,9 +1940,11 @@ function createFilesChannel() {
 //#endregion
 //#region src/services/fileEditSession/index.ts
 const PO_IGNORE_FILE = ".poignore";
-function readPoIgnore(cwd) {
+async function readPoIgnore(cwd, context) {
 	try {
-		const matcher = createPoIgnoreMatcher(node_fs.default.readFileSync(node_path.default.join(cwd, PO_IGNORE_FILE), "utf8"));
+		const content = await (0, _agimon_ai_po_core_platform.readTextIfExists)((0, _agimon_ai_po_core_platform.requirePoFs)(context), node_path.default.join(cwd, PO_IGNORE_FILE), _agimon_ai_po_core_platform.BACKGROUND_CONTEXT);
+		if (content === void 0) return void 0;
+		const matcher = createPoIgnoreMatcher(content);
 		return matcher === void 0 ? void 0 : (filePath) => matcher(node_path.default.relative(cwd, filePath));
 	} catch {
 		return;
@@ -1937,7 +1954,7 @@ function stringValue(value, key) {
 	const candidate = value[key];
 	return typeof candidate === "string" && candidate.length > 0 ? candidate : void 0;
 }
-function createFileEditSession({ agent, host }) {
+function createFileEditSession({ agent, host, context }) {
 	if (!agent) return { api: [api] };
 	const environment = Object.freeze({ ...host.context.environment });
 	const paths = new FileEditPaths(environment);
@@ -1958,6 +1975,7 @@ function createFileEditSession({ agent, host }) {
 	let closed = false;
 	let disposal;
 	let stopping;
+	let starting;
 	let initialized = false;
 	const track = (work) => {
 		const task = work();
@@ -1975,8 +1993,9 @@ function createFileEditSession({ agent, host }) {
 	* string is deliberate: the group declares `hideWhenEmpty`, so a session that
 	* has edited nothing publishes a key with no content and stays hidden.
 	*/
-	const publish = (runtime) => {
-		const items = readSessionFiles(runtime.timelinePath, runtime.context.cwd);
+	const publish = async (runtime) => {
+		const items = await readSessionFiles(runtime.timelinePath, runtime.context.cwd);
+		if (active !== runtime || closed || stopping) return;
 		directEvents.publish(filesChannelType, runtime.context.sessionId, { items });
 		runtime.context.client.setStatus(filesStatusKey, items.length === 0 ? "" : `${items.length} ${items.length === 1 ? "file" : "files"}`);
 	};
@@ -1985,11 +2004,13 @@ function createFileEditSession({ agent, host }) {
 		const runtime = active;
 		active = void 0;
 		stopping = (async () => {
-			await Promise.allSettled(pending);
+			while (pending.size > 0) await Promise.allSettled(pending);
+			const startedRuntime = active;
+			active = void 0;
 			await editTracker.flush();
 			editTracker.reset();
 			toolStartedAt.clear();
-			runtime?.context.client.setStatus(filesStatusKey, void 0);
+			(runtime ?? startedRuntime)?.context.client.setStatus(filesStatusKey, void 0);
 		})();
 		return stopping;
 	};
@@ -2017,50 +2038,60 @@ function createFileEditSession({ agent, host }) {
 			name: "file-edits",
 			start: async (executionContext) => {
 				if (closed) throw new Error("File-edit session is closed.");
+				if (starting) return starting;
 				if (active) return stopActivity;
 				if (stopping) await stopping;
 				if (closed) throw new Error("File-edit session is closed.");
+				if (starting) return starting;
 				stopping = void 0;
-				const sessionKey = paths.sessionKey(executionContext.sessionId);
-				const timelinePath = paths.timelinePath(executionContext.cwd, sessionKey);
-				const snapshotsPath = paths.snapshotsPath(executionContext.cwd, sessionKey);
-				timeline.initialize(timelinePath);
-				snapshots.initialize(snapshotsPath);
-				const isIgnored = readPoIgnore(executionContext.cwd);
-				editTracker.reset({
-					exclude: [
+				starting = track(async () => {
+					const sessionKey = paths.sessionKey(executionContext.sessionId);
+					const timelinePath = await paths.timelinePath(executionContext.cwd, sessionKey);
+					const snapshotsPath = await paths.snapshotsPath(executionContext.cwd, sessionKey);
+					timeline.initialize(timelinePath);
+					snapshots.initialize(snapshotsPath);
+					const isIgnored = await readPoIgnore(executionContext.cwd, context);
+					editTracker.reset({
+						exclude: [
+							timelinePath,
+							`${timelinePath}.lock`,
+							snapshotsPath
+						],
+						...isIgnored === void 0 ? {} : { isIgnored }
+					});
+					const runtime = {
+						context: executionContext,
 						timelinePath,
-						`${timelinePath}.lock`,
 						snapshotsPath
-					],
-					...isIgnored === void 0 ? {} : { isIgnored }
-				});
-				const runtime = {
-					context: executionContext,
-					timelinePath,
-					snapshotsPath
-				};
-				active = runtime;
-				lastRuntime = runtime;
-				publish(runtime);
-				track(async () => {
-					try {
-						await sweepSessionState({
-							directory: paths.stateDirectory(),
-							keep: timelinePath
-						});
-					} catch {}
+					};
+					active = runtime;
+					lastRuntime = runtime;
+					await publish(runtime);
+					track(async () => {
+						try {
+							await sweepSessionState({
+								directory: paths.stateDirectory(),
+								keep: timelinePath
+							});
+						} catch {}
+					});
+					if (!initialized) {
+						initialized = true;
+						await track(() => telemetry.recordEvent("po_file_edit.timeline_started", { outcome: "initialized" }));
+					}
+					return () => active === runtime ? stopActivity() : Promise.resolve();
 				});
-				if (!initialized) {
-					initialized = true;
-					await track(() => telemetry.recordEvent("po_file_edit.timeline_started", { outcome: "initialized" }));
+				try {
+					return await starting;
+				} finally {
+					starting = void 0;
 				}
-				return () => active === runtime ? stopActivity() : Promise.resolve();
 			}
 		}],
 		hooks: [{
 			event: "tool_execution_start",
 			handle: (event, executionContext) => {
+				if (closed || stopping) return;
 				const runtime = active;
 				const toolCallId = stringValue(event, "toolCallId");
 				const toolName = stringValue(event, "toolName");
@@ -2079,6 +2110,7 @@ function createFileEditSession({ agent, host }) {
 		}, {
 			event: "tool_execution_end",
 			handle: (event, executionContext) => {
+				if (closed || stopping) return;
 				const runtime = active;
 				const toolCallId = stringValue(event, "toolCallId");
 				const toolName = stringValue(event, "toolName");
@@ -2090,7 +2122,7 @@ function createFileEditSession({ agent, host }) {
 					const isError = event.isError === true;
 					await editTracker.end(toolCallId, isError, executionContext.cwd);
 					if (active !== runtime) return;
-					publish(runtime);
+					await publish(runtime);
 					await telemetry.recordEvent("po_file_edit.edit_finished", {
 						"tool.name": toolName,
 						"tool.result.error": isError,