@agimon-ai/po-core 0.0.0-alpha.85 → 0.0.0-alpha.87

package/dist/headlessHub.cjs

@@ -161,6 +161,35 @@ function createHeadlessHub(options) {
 	};
 	const pendingHomeTopics = /* @__PURE__ */ new Set();
 	const homeProjections = /* @__PURE__ */ new Map();
+	const homeCandidates = /* @__PURE__ */ new Map();
+	const homeProjectionCandidates = /* @__PURE__ */ new Map();
+	const removeHomeProjection = (key) => {
+		homeProjections.delete(key);
+		for (const identity of homeProjectionCandidates.get(key) ?? []) {
+			const candidates = homeCandidates.get(identity);
+			candidates.delete(key);
+			if (candidates.size === 0) homeCandidates.delete(identity);
+		}
+		homeProjectionCandidates.delete(key);
+	};
+	const indexHomeProjection = (key, frameType, session, projection) => {
+		for (const identity of homeProjectionCandidates.get(key) ?? []) {
+			const candidates = homeCandidates.get(identity);
+			candidates.delete(key);
+			if (candidates.size === 0) homeCandidates.delete(identity);
+		}
+		const identities = new Set(projection.entities.map((entity) => JSON.stringify([
+			frameType,
+			session.workspaceId,
+			entity.id
+		])));
+		homeProjectionCandidates.set(key, identities);
+		for (const identity of identities) {
+			const candidates = homeCandidates.get(identity) ?? /* @__PURE__ */ new Set();
+			candidates.add(key);
+			homeCandidates.set(identity, candidates);
+		}
+	};
 	let homeBumpQueued = false;
 	const queueHomeBump = (session) => {
 		pendingHomeTopics.add("global:home");
@@ -393,10 +422,30 @@ function createHeadlessHub(options) {
 			const key = JSON.stringify([sessionId, frameType]);
 			let projection;
 			try {
-				projection = JSON.stringify(source.homeFor(scopeOf(session))) ?? "";
+				const home = require_home.parseHomeProjection(source.homeFor(scopeOf(session)));
+				if (!home) throw new Error("Invalid Home projection.");
+				indexHomeProjection(key, frameType, session, home);
+				projection = JSON.stringify({
+					coverage: home.coverage,
+					entities: home.entities.map((entity) => {
+						if (entity.kind !== "running" && entity.kind !== "queued" && entity.kind !== "completed") return entity;
+						const competing = (homeCandidates.get(JSON.stringify([
+							frameType,
+							session.workspaceId,
+							entity.id
+						]))?.size ?? 0) > 1;
+						return {
+							id: entity.id,
+							kind: entity.kind,
+							sessionId: entity.sessionId,
+							relatedSessionIds: entity.relatedSessionIds,
+							...competing ? { updatedAt: entity.updatedAt } : {}
+						};
+					})
+				});
 			} catch {
 				projection = "";
-				homeProjections.delete(key);
+				removeHomeProjection(key);
 			}
 			if (projection === "" || homeProjections.get(key) !== projection) {
 				if (projection !== "") homeProjections.set(key, projection);
@@ -557,6 +606,16 @@ function createHeadlessHub(options) {
 		} },
 		onNotice: (message) => options.onNotice?.(message)
 	});
+	const invalidateHomeSource = (session, frameType) => {
+		const key = JSON.stringify([session.id, frameType]);
+		removeHomeProjection(key);
+		const source = selectedChannel(session, frameType)?.source;
+		if (source?.homeFor) try {
+			const projection = require_home.parseHomeProjection(source.homeFor(scopeOf(session)));
+			if (projection) indexHomeProjection(key, frameType, session, projection);
+		} catch {}
+		queueHomeBump(session);
+	};
 	const startChannel = (channel, mount) => {
 		const key = `${mountKey(mount)}:${channel.frameType}`;
 		if (channels.has(key)) {
@@ -579,8 +638,13 @@ function createHeadlessHub(options) {
 			channels.set(key, started);
 			for (const session of sessions.values()) {
 				if (selectedChannel(session, channel.frameType) !== started) continue;
-				displaced.get(session.id)?.source.sessionRemoved?.(session.id);
-				source.sessionAdded?.(scopeOf(session));
+				invalidateHomeSource(session, channel.frameType);
+				try {
+					displaced.get(session.id)?.source.sessionRemoved?.(session.id);
+					source.sessionAdded?.(scopeOf(session));
+				} finally {
+					invalidateHomeSource(session, channel.frameType);
+				}
 			}
 			return started;
 		} catch (error) {
@@ -593,8 +657,13 @@ function createHeadlessHub(options) {
 		if (channels.get(key) !== started) return;
 		const served = [...sessions.values()].filter((session) => selectedChannel(session, channel.frameType) === started);
 		channels.delete(key);
+		for (const session of served) invalidateHomeSource(session, channel.frameType);
 		started.source.close();
-		for (const session of served) selectedChannel(session, channel.frameType)?.source.sessionAdded?.(scopeOf(session));
+		for (const session of served) try {
+			selectedChannel(session, channel.frameType)?.source.sessionAdded?.(scopeOf(session));
+		} finally {
+			invalidateHomeSource(session, channel.frameType);
+		}
 	};
 	const receiveMountedChannel = (mount, sessionId, frameType, payload, connectionId) => {
 		if (!connectionId) return false;
@@ -617,7 +686,7 @@ function createHeadlessHub(options) {
 		} finally {
 			sessions.delete(sessionId);
 			if (current.workspaceId !== void 0) workspacePresence.get(current.workspaceId)?.forget(`session:${sessionId}`);
-			for (const key of homeProjections.keys()) if (key.startsWith(`[${JSON.stringify(sessionId)},`)) homeProjections.delete(key);
+			for (const key of homeProjectionCandidates.keys()) if (key.startsWith(`[${JSON.stringify(sessionId)},`)) removeHomeProjection(key);
 			queueHomeBump(current);
 			pendingSetups.delete(sessionId);
 			activities.delete(sessionId);
@@ -862,7 +931,10 @@ function createHeadlessHub(options) {
 			stopSelection?.();
 			stopBackgroundWork?.();
 		});
-		for (const { source } of selectedChannels(current)) source.sessionAdded?.(scopeOf(session));
+		for (const { channel, source } of selectedChannels(current)) {
+			source.sessionAdded?.(scopeOf(session));
+			invalidateHomeSource(current, channel.frameType);
+		}
 		const active = () => sessions.get(session.id)?.host === session.host && !closed;
 		const cleanup = () => {
 			subscriptions.delete(cleanup);
@@ -1241,6 +1313,7 @@ function createHeadlessHub(options) {
 		},
 		registerWorkspace(workspace) {
 			workspaces.set(workspace.id, workspace);
+			queueHomeBump({ workspaceId: workspace.id });
 			emit({
 				kind: "workspace_upsert",
 				workspace
@@ -1265,6 +1338,7 @@ function createHeadlessHub(options) {
 				workspaces.delete(id);
 				compositionRevisions.delete(key);
 				for (const topic of revisions.keys()) if (topic.startsWith(`workspace:${id}:`)) revisions.delete(topic);
+				queueHomeBump({ workspaceId: id });
 				emit({
 					kind: "workspace_removed",
 					workspaceId: id