@xopcai/xopc 0.0.367 → 0.0.369

package/dist/src/browser/providers/extension.js

@@ -1,378 +0,0 @@
-import { createLogger } from "../../utils/logger/index.js";
-import { init_logger } from "../../utils/logger.js";
-import { deleteBrowserTabBindingsByPrincipal, getBrowserTabBindingById, init_browser_tab_binding_repository } from "../../storage/sqlite/browser-tab-binding-repository.js";
-import { getDevice, init_device_access_repository } from "../../storage/sqlite/device-access-repository.js";
-import { browserWireAuthenticationPayload } from "../../../packages/browser-control-contract/src/index.js";
-import { isChromeExtensionOrigin } from "../../gateway/security/origin-check.js";
-import crypto from "node:crypto";
-//#region src/browser/providers/extension.ts
-/**
-* Extension browser provider — connects to the xopc Chrome Extension via WebSocket.
-*
-* The xopc process starts a WebSocket server; the Chrome Extension connects as a client.
-* Commands are sent over the WS connection and results are returned asynchronously.
-*/
-init_logger();
-init_device_access_repository();
-init_browser_tab_binding_repository();
-const log = createLogger("ExtensionProvider");
-const DEFAULT_PORT = 19820;
-const DEFAULT_HOST = "127.0.0.1";
-const DEFAULT_CONNECTION_TIMEOUT = 3e4;
-const DEFAULT_COMMAND_TIMEOUT = 3e4;
-/**
-* Manages a WebSocket server that the Chrome Extension connects to.
-* Sends commands and receives results over the connection.
-*/
-var ExtensionBrowserProvider = class {
-	name = "extension";
-	server = null;
-	wss = null;
-	clientWs = null;
-	pending = /* @__PURE__ */ new Map();
-	commandCounter = 0;
-	socketConnected = false;
-	handshakeReceived = false;
-	connected = false;
-	extensionProtocolVersion = null;
-	extensionVersion = null;
-	connectionId = null;
-	principalId = null;
-	config;
-	connectionWaiters = [];
-	constructor(config) {
-		this.config = {
-			port: config?.port ?? DEFAULT_PORT,
-			host: config?.host ?? DEFAULT_HOST,
-			connectionTimeout: config?.connectionTimeout ?? DEFAULT_CONNECTION_TIMEOUT,
-			commandTimeout: config?.commandTimeout ?? DEFAULT_COMMAND_TIMEOUT
-		};
-	}
-	/** Start the WebSocket server and wait for the extension to connect. */
-	async start() {
-		if (this.server) return;
-		const wsModule = await import("ws");
-		let WssClass = wsModule.WebSocketServer;
-		if (typeof WssClass !== "function") {
-			const def = wsModule.default;
-			WssClass = def?.WebSocketServer ?? def;
-		}
-		if (typeof WssClass !== "function") throw new Error("Failed to resolve WebSocketServer from \"ws\" package");
-		const WebSocketServer = WssClass;
-		const http = await import("node:http");
-		this.server = http.createServer((_req, res) => {
-			res.writeHead(200, { "Content-Type": "application/json" });
-			res.end(JSON.stringify({
-				ok: true,
-				...this.getConnectionStatus()
-			}));
-		});
-		this.wss = new WebSocketServer({
-			server: this.server,
-			path: "/browser-ext"
-		});
-		this.wss.on("connection", (ws, request) => {
-			const origin = request.headers.origin?.toLowerCase();
-			if (!isChromeExtensionOrigin(origin)) {
-				ws.close(4403, "Chrome extension origin required");
-				return;
-			}
-			const challenge = {
-				type: "auth_challenge",
-				protocolVersion: 5,
-				connectionId: crypto.randomUUID(),
-				challenge: crypto.randomBytes(32).toString("base64url"),
-				issuedAt: Date.now()
-			};
-			let authenticated = false;
-			const authenticationTimer = setTimeout(() => {
-				if (!authenticated) ws.close(4401, "Browser authentication timed out");
-			}, 5e3);
-			ws.send(JSON.stringify(challenge));
-			ws.on("message", (data) => {
-				if (authenticated) {
-					this._handleMessage(data.toString(), ws);
-					return;
-				}
-				if (!this._authenticateCandidate(data.toString(), origin, challenge)) {
-					clearTimeout(authenticationTimer);
-					ws.close(4401, "Browser authentication failed");
-					return;
-				}
-				authenticated = true;
-				clearTimeout(authenticationTimer);
-				const message = JSON.parse(data.toString());
-				const previousClient = this.clientWs;
-				this.clientWs = ws;
-				this.socketConnected = true;
-				this.handshakeReceived = true;
-				this.connected = true;
-				this.extensionProtocolVersion = message.protocolVersion;
-				this.extensionVersion = message.extensionVersion;
-				this.connectionId = message.connectionId;
-				this.principalId = message.principalId;
-				if (previousClient && previousClient !== ws) try {
-					previousClient.close(4001, "Connection replaced");
-				} catch {}
-				for (const waiter of this.connectionWaiters) waiter.resolve();
-				this.connectionWaiters = [];
-				log.info({
-					principalId: message.principalId,
-					extensionId: message.extensionId,
-					connectionId: message.connectionId
-				}, "Chrome extension authenticated");
-			});
-			ws.on("close", (code, reason) => {
-				if (this.clientWs !== ws) return;
-				log.warn({
-					code,
-					reason: reason.toString(),
-					pendingCount: this.pending.size
-				}, "Chrome Extension disconnected");
-				const disconnectedPrincipalId = this.principalId;
-				this.clientWs = null;
-				this.socketConnected = false;
-				this.handshakeReceived = false;
-				this.connected = false;
-				this.extensionProtocolVersion = null;
-				this.extensionVersion = null;
-				this.connectionId = null;
-				this.principalId = null;
-				if (disconnectedPrincipalId) deleteBrowserTabBindingsByPrincipal(disconnectedPrincipalId);
-				for (const [id, req] of this.pending) {
-					clearTimeout(req.timer);
-					req.reject(/* @__PURE__ */ new Error("Extension disconnected"));
-					this.pending.delete(id);
-				}
-			});
-		});
-		const wssEmitter = this.wss;
-		await new Promise((resolve, reject) => {
-			let settled = false;
-			const disposeFailedStart = async () => {
-				if (this.wss) {
-					try {
-						this.wss.close();
-					} catch {}
-					this.wss = null;
-				}
-				if (this.server) {
-					await new Promise((r) => {
-						this.server.close(() => r());
-					});
-					this.server = null;
-				}
-			};
-			const onStartError = (err) => {
-				if (settled) return;
-				settled = true;
-				this.server.removeListener("error", onStartError);
-				wssEmitter.removeListener("error", onStartError);
-				disposeFailedStart().then(() => reject(err));
-			};
-			const onListening = () => {
-				if (settled) return;
-				settled = true;
-				this.server.removeListener("error", onStartError);
-				wssEmitter.removeListener("error", onStartError);
-				log.info({
-					port: this.config.port,
-					host: this.config.host
-				}, "Extension WS server started");
-				const onRuntimeError = (err) => {
-					log.error({ err }, "Extension WS bridge runtime error");
-				};
-				this.server.on("error", onRuntimeError);
-				wssEmitter.on("error", onRuntimeError);
-				resolve();
-			};
-			this.server.on("error", onStartError);
-			wssEmitter.on("error", onStartError);
-			this.server.listen(this.config.port, this.config.host, onListening);
-		});
-	}
-	/** Wait for the Chrome Extension to connect. */
-	async waitForConnection(timeoutMs) {
-		if (this.connected) return;
-		if (this.socketConnected && this.handshakeReceived && this.extensionProtocolVersion !== 5) throw new Error(this.protocolMismatchMessage());
-		const timeout = timeoutMs ?? this.config.connectionTimeout;
-		return new Promise((resolve, reject) => {
-			const timer = setTimeout(() => {
-				const idx = this.connectionWaiters.findIndex((w) => w.resolve === resolve);
-				if (idx >= 0) this.connectionWaiters.splice(idx, 1);
-				reject(/* @__PURE__ */ new Error(`Extension connection timeout after ${timeout}ms. Is the Chrome Extension installed and enabled?`));
-			}, timeout);
-			this.connectionWaiters.push({
-				resolve: () => {
-					clearTimeout(timer);
-					resolve();
-				},
-				reject: (e) => {
-					clearTimeout(timer);
-					reject(e);
-				}
-			});
-		});
-	}
-	/** Send one Browser Control v2 action to the extension. */
-	async send(input, timeoutMs, visualFallback = true) {
-		if (!this.connected || !this.clientWs) {
-			const detail = this.socketConnected ? this.protocolMismatchMessage() : "Extension not connected. Ensure the Chrome Extension is installed and connected.";
-			throw new Error(detail);
-		}
-		if (input.target?.kind === "attached_tab") {
-			const binding = getBrowserTabBindingById(input.target.bindingId);
-			if (!binding || binding.principalId !== this.principalId) throw new Error("Attached tab binding does not belong to the connected browser");
-		}
-		const id = `cmd_${++this.commandCounter}_${Date.now()}`;
-		const cmd = {
-			id,
-			protocolVersion: 5,
-			connectionId: this.connectionId,
-			input,
-			timeoutMs: timeoutMs ?? this.config.commandTimeout,
-			visualFallback
-		};
-		return new Promise((resolve, reject) => {
-			const timer = setTimeout(() => {
-				this.pending.delete(id);
-				reject(/* @__PURE__ */ new Error(`Browser action timed out: ${input.action} (${cmd.timeoutMs}ms)`));
-			}, cmd.timeoutMs);
-			this.pending.set(id, {
-				resolve,
-				reject,
-				timer
-			});
-			try {
-				this.clientWs.send(JSON.stringify(cmd));
-			} catch (e) {
-				clearTimeout(timer);
-				this.pending.delete(id);
-				reject(e instanceof Error ? e : new Error(String(e)));
-			}
-		});
-	}
-	/** Whether the extension is currently connected. */
-	isConnected() {
-		return this.connected;
-	}
-	getConnectionStatus() {
-		return {
-			socketConnected: this.socketConnected,
-			connected: this.connected,
-			protocolVersion: this.extensionProtocolVersion,
-			expectedProtocolVersion: 5,
-			extensionVersion: this.extensionVersion,
-			principalId: this.principalId
-		};
-	}
-	protocolMismatchMessage() {
-		return `Browser extension protocol mismatch (expected 5, received ${this.extensionProtocolVersion ?? "unknown"}). Reload the extension in Chrome.`;
-	}
-	/** Shutdown the WebSocket server. */
-	async shutdown() {
-		for (const waiter of this.connectionWaiters) waiter.reject(/* @__PURE__ */ new Error("Provider shutting down"));
-		this.connectionWaiters = [];
-		for (const [id, req] of this.pending) {
-			clearTimeout(req.timer);
-			req.reject(/* @__PURE__ */ new Error("Provider shutting down"));
-			this.pending.delete(id);
-		}
-		if (this.clientWs) {
-			try {
-				this.clientWs.close();
-			} catch {}
-			this.clientWs = null;
-		}
-		if (this.wss) {
-			try {
-				this.wss.close();
-			} catch {}
-			this.wss = null;
-		}
-		if (this.server) {
-			await new Promise((resolve) => {
-				this.server.close(() => resolve());
-			});
-			this.server = null;
-		}
-		this.connected = false;
-		this.socketConnected = false;
-		this.handshakeReceived = false;
-		this.extensionProtocolVersion = null;
-		this.extensionVersion = null;
-		this.connectionId = null;
-		this.principalId = null;
-		log.info("Extension provider shut down");
-	}
-	_handleMessage(raw, sourceWs) {
-		if (sourceWs !== this.clientWs) return;
-		try {
-			const msg = JSON.parse(raw);
-			if (!this.connectionId || msg.connectionId !== this.connectionId) return;
-			if (msg.type === "status") {
-				const status = msg;
-				if (status.principalId !== this.principalId) return;
-				log.debug({ sessionCount: status.sessionCount }, "Browser extension status updated");
-				return;
-			}
-			if (isBrowserWireKeepAlive(msg)) return;
-			if (msg.type === "network_event") {
-				log.debug({
-					listenerId: msg.listenerId,
-					eventType: msg.eventType
-				}, "Network event");
-				return;
-			}
-			const id = msg.id;
-			const pending = this.pending.get(id);
-			if (pending) {
-				clearTimeout(pending.timer);
-				this.pending.delete(id);
-				if (!isBrowserWireResult(msg)) {
-					pending.reject(/* @__PURE__ */ new Error("Browser extension returned an incompatible response. Reload the extension in Chrome."));
-					return;
-				}
-				pending.resolve(msg);
-			}
-		} catch (e) {
-			log.error({ err: e }, "Failed to parse extension message");
-		}
-	}
-	_authenticateCandidate(raw, origin, challenge) {
-		try {
-			const message = JSON.parse(raw);
-			if (message.type !== "authenticate" || message.protocolVersion !== 5 || message.connectionId !== challenge.connectionId || `chrome-extension://${message.extensionId}` !== origin || Math.abs(Date.now() - challenge.issuedAt) > 5e3) return false;
-			const device = getDevice(message.principalId);
-			if (!device || device.revokedAt !== void 0 || device.platform !== "chrome" || device.extensionId !== message.extensionId) return false;
-			const payload = browserWireAuthenticationPayload({
-				protocolVersion: message.protocolVersion,
-				connectionId: message.connectionId,
-				challenge: challenge.challenge,
-				issuedAt: challenge.issuedAt,
-				principalId: message.principalId,
-				extensionId: message.extensionId,
-				extensionVersion: message.extensionVersion
-			});
-			return crypto.verify("sha256", Buffer.from(payload), {
-				key: crypto.createPublicKey({
-					key: device.publicKeyJwk,
-					format: "jwk"
-				}),
-				dsaEncoding: "ieee-p1363"
-			}, Buffer.from(message.signature, "base64url"));
-		} catch {
-			return false;
-		}
-	}
-};
-function isBrowserWireResult(value) {
-	if (!value || typeof value !== "object") return false;
-	const result = value.result;
-	return typeof value.id === "string" && typeof value.connectionId === "string" && Boolean(result) && typeof result === "object" && typeof result.ok === "boolean";
-}
-function isBrowserWireKeepAlive(value) {
-	return Boolean(value) && typeof value === "object" && value.type === "keepalive" && typeof value.connectionId === "string" && typeof value.timestamp === "number";
-}
-//#endregion
-export { ExtensionBrowserProvider, isBrowserWireResult };