@agimon-ai/po-computer-use 0.0.0-alpha.2 → 0.0.0-alpha.21

package/dist/extensions/server.mjs

@@ -1,16 +1,16 @@
 import { a as COMMAND_NAME, i as COMMAND_DESCRIPTION, n as ComputerScriptExecutionError, r as ComputerScriptRunner, t as DefaultComputerUseExtensionService } from "../extensionService-DXvF7esO.mjs";
 import path from "node:path";
-import { composeServerProviderOverrides, defineServerPlugin } from "@agimon-ai/po-core/serverFacet";
-import { composeRootHooks, defineChannel, defineRoot, defineRoutedContribution } from "@agimon-ai/po-core/extensionFile";
+import { composeServerProviderOverrides, defineServerPlugin } from "@agimon-ai/po-core/serverPlugin";
+import { composeRootHooks, composeRootInjects, defineChannel, defineRoot, defineRoutedContribution, runCleanups } from "@agimon-ai/po-core/extensionFile";
 import { defineMinorMode, serverMinorModes } from "@agimon-ai/po-minor-mode";
 import { randomUUID } from "node:crypto";
-import { doomApiCallerFrom } from "@agimon-ai/po-core/packageApi";
+import { poApiCallerFrom } from "@agimon-ai/po-core/packageApi";
 import { apiResponse, defineApiRoutes } from "@agimon-ai/po-core/web";
 //#region src/types/computerUseApi.ts
 const API_BASE_PATH = "computer-use";
 const computerUseChannelType = "computer_use_state";
 const COMPUTER_USE_MODE_ID = "computer-use";
-const COMPUTER_USE_STATUS_KEY = "doom-computer-use";
+const COMPUTER_USE_STATUS_KEY = "po-computer-use";
 const COMPUTER_USE_CONFIRMATION_WINDOW_MS = 12e4;
 /**
 * Every path this API answers on, below its mount.
@@ -36,7 +36,7 @@ const COMPUTER_USE_ROUTES = {
 const COMPUTER_USE_WAKE_LIMIT = 1e6;
 //#endregion
 //#region src/services/mcpTools/index.ts
-const COMPUTER_USE_MCP_TOOLS_SERVICE = "doom/computer-use-mcp-tools";
+const COMPUTER_USE_MCP_TOOLS_SERVICE = "po/computer-use-mcp-tools";
 function mountMcpTools(tools) {
 	return (context) => {
 		context.plugin((providerContext) => {
@@ -411,7 +411,7 @@ var ComputerUseRequestBroker = class ComputerUseRequestBroker {
 	async requestActivation(request) {
 		if (!this.desktop?.available || !(this.desktop.authorizeActivation?.(request.headers) ?? this.desktop.authorize(request.headers))) return jsonError("Not found.", 404);
 		if (this.desktop.enabled === false) return jsonError("Enable computer use in global Desktop settings first.", 409);
-		const stamped = doomApiCallerFrom(request.headers);
+		const stamped = poApiCallerFrom(request.headers);
 		const caller = stamped?.locality === "remote" ? stamped : {
 			locality: "local",
 			stepUp: "not-required"
@@ -985,13 +985,18 @@ function createComputerUseServer(host, apiContext) {
 				if (client && state?.phase !== "inactive" && state?.phase !== "failed") await client.stop();
 			}
 		}],
-		onDispose() {
-			stopActivity?.();
-			stopActivity = void 0;
-			unsubscribeState?.();
-			broker.close();
-			restrictionListeners.clear();
-			host.context.client.setStatus(COMPUTER_USE_STATUS_KEY, void 0);
+		async onDispose() {
+			await runCleanups([
+				() => {
+					const stop = stopActivity;
+					stopActivity = void 0;
+					stop?.();
+				},
+				() => unsubscribeState?.(),
+				() => broker.close(),
+				() => restrictionListeners.clear(),
+				() => host.context.client.setStatus(COMPUTER_USE_STATUS_KEY, void 0)
+			], "Computer-use cleanup failed.");
 		}
 	};
 }
@@ -1336,17 +1341,18 @@ const via = (identity, value) => value === void 0 ? void 0 : {
 	...identity,
 	...value
 };
-const facet = defineServerPlugin({
+const serverPlugin = defineServerPlugin({
 	name: "@agimon-ai/po-computer-use",
 	providerOverrides: composeServerProviderOverrides(root_server_default),
+	injects: composeRootInjects({ session: [root_server_default] }),
 	global: () => ({ get channels() {
 		return [() => via({ frameType: "computer_use" }, computer_use_server_default$2())];
 	} }),
-	workspace: () => ({ get channels() {
-		return [() => via({ frameType: "computer_use" }, computer_use_server_default$2())];
+	workspace: (context) => ({ get channels() {
+		return defined([context.hosted("global") ? void 0 : () => via({ frameType: "computer_use" }, computer_use_server_default$2())]);
 	} }),
 	session: async (context) => {
-		const scopeSession = await root_server_default(context);
+		const scopeSession = await context.scopeRoot("session", root_server_default, context);
 		const contextSession = {
 			...context,
 			root: scopeSession.value
@@ -1357,7 +1363,7 @@ const facet = defineServerPlugin({
 				return defined([at(route_server_default, contextSession)]);
 			},
 			get channels() {
-				return [() => via({ frameType: "computer_use" }, computer_use_server_default$2())];
+				return defined([context.hosted("global") ? void 0 : () => via({ frameType: "computer_use" }, computer_use_server_default$2())]);
 			},
 			get commands() {
 				return defined([via({ name: "computer-use" }, at(computer_use_server_default$1, contextSession))]);
@@ -1383,6 +1389,6 @@ const facet = defineServerPlugin({
 	}
 });
 //#endregion
-export { facet as default, facet };
+export { serverPlugin as default, serverPlugin };
 
 //# sourceMappingURL=server.mjs.map