@agimon-ai/po-author 0.0.0-alpha.52 → 0.0.0-alpha.53

package/dist/extensions/native.mjs

@@ -177,6 +177,7 @@ const documentProfiles = (profiles) => profiles.map((profile) => ({
 }));
 var AuthorBrowserBridge = class {
 	#host;
+	#reconcileTabs;
 	#canvases = /* @__PURE__ */ new Map();
 	#parked = /* @__PURE__ */ new Map();
 	#opening = /* @__PURE__ */ new Map();
@@ -184,8 +185,9 @@ var AuthorBrowserBridge = class {
 	#generation = 0;
 	#disposed = false;
 	#releaseConnected;
-	constructor(host) {
+	constructor(host, reconcileTabs) {
 		this.#host = host;
+		this.#reconcileTabs = reconcileTabs;
 		this.#releaseConnected = host.onHubConnected(() => {
 			for (const canvas of this.#canvases.values()) {
 				canvas.ownerToken = void 0;
@@ -304,6 +306,21 @@ var AuthorBrowserBridge = class {
 			this.#dropLive(canvas);
 		}
 	}
+	/** Follows the server's open canvases: lets go of any it no longer has, such as one evicted for a newer one. */
+	canvases(sessionId, canvases) {
+		this.#reconcileTabs?.(this.#host, sessionId, canvases);
+		const open = new Set(canvases.map((canvas) => canvas.alias));
+		const gone = /* @__PURE__ */ new Set();
+		for (const entry of [
+			...this.#opening.values(),
+			...this.#parked.values(),
+			...this.#canvases.values()
+		]) {
+			const parked = "parked" in entry ? entry.parked : entry;
+			if (parked.sessionId === sessionId && !open.has(parked.alias)) gone.add(parked.alias);
+		}
+		for (const alias of gone) this.drop(sessionId, alias);
+	}
 	apply(sessionId, message) {
 		if (!this.#agentEnabled(sessionId) || message.alias === void 0) return;
 		const canvas = this.#canvases.get(canvasKey(sessionId, message.alias));
@@ -510,13 +527,18 @@ var AuthorBrowserBridge = class {
 		canvas.generation = ++this.#generation;
 	}
 };
+/** Key of the bridge that serves every session when the plugin is not mounted per session. */
+const GLOBAL_BRIDGE = "global";
 const authorBridgeRuntimes = new Store(/* @__PURE__ */ new Map());
-const bridgeFor = (sessionId) => authorBridgeRuntimes.state.get(sessionId) ?? authorBridgeRuntimes.state.get("global");
-function startAuthorBrowserBridge(runtime) {
-	const key = runtime.mount?.scope === "session" ? runtime.mount.sessionId : "global";
+/** The server's last open-canvas snapshot per session, for a bridge that starts after it arrived. */
+const latestCanvases = new Store(/* @__PURE__ */ new Map());
+const bridgeFor = (sessionId) => authorBridgeRuntimes.state.get(sessionId) ?? authorBridgeRuntimes.state.get(GLOBAL_BRIDGE);
+function startAuthorBrowserBridge(runtime, reconcileTabs) {
+	const key = runtime.mount?.scope === "session" ? runtime.mount.sessionId : GLOBAL_BRIDGE;
 	authorBridgeRuntimes.state.get(key)?.dispose();
-	const next = new AuthorBrowserBridge(runtime);
+	const next = new AuthorBrowserBridge(runtime, reconcileTabs);
 	authorBridgeRuntimes.setState((state) => new Map(state).set(key, next));
+	for (const [sessionId, canvases] of latestCanvases.state) if (key === GLOBAL_BRIDGE || key === sessionId) next.canvases(sessionId, canvases);
 	return () => {
 		if (authorBridgeRuntimes.state.get(key) !== next) return;
 		next.dispose();
@@ -531,9 +553,17 @@ async function openAuthorCanvas(sessionId, alias, profiles) {
 	await bridgeFor(sessionId)?.open(sessionId, alias, profiles);
 }
 function applyAuthorHubMessage(sessionId, message) {
-	bridgeFor(sessionId)?.apply(sessionId, message);
+	if (message.kind === "canvases") {
+		latestCanvases.setState((state) => new Map(state).set(sessionId, message.canvases));
+		bridgeFor(sessionId)?.canvases(sessionId, message.canvases);
+	} else bridgeFor(sessionId)?.apply(sessionId, message);
 }
 function dropAuthorViewportSession(sessionId, alias) {
+	if (alias === void 0 && latestCanvases.state.has(sessionId)) latestCanvases.setState((state) => {
+		const remaining = new Map(state);
+		remaining.delete(sessionId);
+		return remaining;
+	});
 	bridgeFor(sessionId)?.drop(sessionId, alias);
 }
 /** Why the agent cannot use this canvas here, such as another device holding it; undefined when it can. */
@@ -561,6 +591,7 @@ function parseAuthorHubMessage(input) {
 	if ((input.activation === "active" || input.activation === "inactive") && Number.isSafeInteger(input.capabilityCount) && input.capabilityCount >= 0) return input;
 	if (typeof input.kind !== "string") return null;
 	if (input.kind === "rejected") return typeof input.reason === "string" ? input : null;
+	if (input.kind === "canvases") return Array.isArray(input.canvases) && input.canvases.every((entry) => isRecord(entry) && typeof entry.alias === "string" && typeof entry.path === "string") ? input : null;
 	if (!Number.isSafeInteger(input.generation) || typeof input.ownerToken !== "string") return null;
 	if (input.kind === "accepted") return typeof input.leaseMs === "number" && (input.catalogToken === void 0 || typeof input.catalogToken === "string") ? input : null;
 	if ((input.kind === "request" || input.kind === "cancel") && typeof input.catalogToken === "string" && typeof input.requestId === "string") {