opendevbrowser 0.0.10 → 0.0.11

package/dist/opendevbrowser.js

@@ -2276,21 +2276,30 @@ function buildContinuityNudgeMessage(filePath) {
 import { createServer } from "http";
 import { timingSafeEqual } from "crypto";
 import { WebSocket, WebSocketServer } from "ws";
+var DEFAULT_DISCOVERY_PORT = 8787;
+var CONFIG_PATH = "/config";
+var PAIR_PATH = "/pair";
 var RelayServer = class _RelayServer {
   running = false;
   baseUrl = null;
   port = null;
   server = null;
+  discoveryServer = null;
   extensionWss = null;
   cdpWss = null;
   extensionSocket = null;
   cdpSocket = null;
   extensionInfo = null;
   pairingToken = null;
+  configuredDiscoveryPort;
+  discoveryPort = null;
   handshakeAttempts = /* @__PURE__ */ new Map();
   cdpAllowlist = null;
   static MAX_HANDSHAKE_ATTEMPTS = 5;
   static RATE_LIMIT_WINDOW_MS = 6e4;
+  constructor(options = {}) {
+    this.configuredDiscoveryPort = options.discoveryPort ?? DEFAULT_DISCOVERY_PORT;
+  }
   async start(port = 8787) {
     if (this.running && this.baseUrl && this.port !== null) {
       return { url: this.baseUrl, port: this.port };
@@ -2335,7 +2344,15 @@ var RelayServer = class _RelayServer {
     this.server.on("request", (request, response) => {
       const pathname = new URL(request.url ?? "", "http://127.0.0.1").pathname;
       const origin = request.headers.origin;
-      if (pathname === "/pair" && request.method === "OPTIONS") {
+      if (pathname === CONFIG_PATH && request.method === "OPTIONS") {
+        this.handleConfigPreflight(origin, response);
+        return;
+      }
+      if (pathname === CONFIG_PATH && request.method === "GET") {
+        this.handleConfigRequest(origin, response);
+        return;
+      }
+      if (pathname === PAIR_PATH && request.method === "OPTIONS") {
         if (origin && origin.startsWith("chrome-extension://")) {
           response.setHeader("Access-Control-Allow-Origin", origin);
           response.setHeader("Access-Control-Allow-Methods", "GET, OPTIONS");
@@ -2345,7 +2362,7 @@ var RelayServer = class _RelayServer {
         response.end();
         return;
       }
-      if (pathname === "/pair" && request.method === "GET") {
+      if (pathname === PAIR_PATH && request.method === "GET") {
         const isLocalhost = !origin || origin.startsWith("chrome-extension://");
         if (!isLocalhost) {
           response.writeHead(403, { "Content-Type": "application/json" });
@@ -2405,6 +2422,13 @@ var RelayServer = class _RelayServer {
     this.port = address.port;
     this.baseUrl = `ws://127.0.0.1:${address.port}`;
     this.running = true;
+    try {
+      await this.startDiscoveryServer();
+    } catch (error) {
+      const message = error instanceof Error ? error.message : String(error);
+      console.warn(`[opendevbrowser] Discovery server failed to start: ${message}`);
+      this.stopDiscoveryServer();
+    }
     return { url: this.baseUrl, port: address.port };
   }
   stop() {
@@ -2412,6 +2436,7 @@ var RelayServer = class _RelayServer {
     this.baseUrl = null;
     this.port = null;
     this.extensionInfo = null;
+    this.stopDiscoveryServer();
     if (this.extensionSocket) {
       this.extensionSocket.close(1e3, "Relay stopped");
       this.extensionSocket = null;
@@ -2440,6 +2465,12 @@ var RelayServer = class _RelayServer {
   getCdpUrl() {
     return this.baseUrl ? `${this.baseUrl}/cdp` : null;
   }
+  getDiscoveryPort() {
+    if (this.port !== null && this.port === this.configuredDiscoveryPort) {
+      return this.port;
+    }
+    return this.discoveryPort;
+  }
   setToken(token) {
     const trimmed = typeof token === "string" ? token.trim() : "";
     this.pairingToken = trimmed.length ? trimmed : null;
@@ -2460,6 +2491,81 @@ var RelayServer = class _RelayServer {
     }
     return false;
   }
+  isExtensionOrigin(origin) {
+    return Boolean(origin && origin.startsWith("chrome-extension://"));
+  }
+  handleConfigPreflight(origin, response) {
+    if (this.isExtensionOrigin(origin)) {
+      response.setHeader("Access-Control-Allow-Origin", origin);
+      response.setHeader("Access-Control-Allow-Methods", "GET, OPTIONS");
+      response.setHeader("Access-Control-Allow-Headers", "Content-Type");
+    }
+    response.writeHead(204);
+    response.end();
+  }
+  handleConfigRequest(origin, response) {
+    if (!this.isExtensionOrigin(origin)) {
+      response.writeHead(403, { "Content-Type": "application/json" });
+      response.end(JSON.stringify({ error: "Forbidden: extension origin required" }));
+      return;
+    }
+    if (origin) {
+      response.setHeader("Access-Control-Allow-Origin", origin);
+    }
+    if (this.port === null) {
+      response.writeHead(503, { "Content-Type": "application/json" });
+      response.end(JSON.stringify({ error: "Relay not running" }));
+      return;
+    }
+    response.writeHead(200, {
+      "Content-Type": "application/json",
+      "Cache-Control": "no-store"
+    });
+    response.end(JSON.stringify({
+      relayPort: this.port,
+      pairingRequired: Boolean(this.pairingToken)
+    }));
+  }
+  async startDiscoveryServer() {
+    if (this.port === null || this.discoveryServer) {
+      return;
+    }
+    if (this.configuredDiscoveryPort > 0 && this.configuredDiscoveryPort === this.port) {
+      return;
+    }
+    this.discoveryServer = createServer((request, response) => {
+      const pathname = new URL(request.url ?? "", "http://127.0.0.1").pathname;
+      const origin = request.headers.origin;
+      if (pathname === CONFIG_PATH && request.method === "OPTIONS") {
+        this.handleConfigPreflight(origin, response);
+        return;
+      }
+      if (pathname === CONFIG_PATH && request.method === "GET") {
+        this.handleConfigRequest(origin, response);
+        return;
+      }
+      response.writeHead(404);
+      response.end();
+    });
+    await new Promise((resolve, reject) => {
+      this.discoveryServer?.once("error", reject);
+      this.discoveryServer?.listen(this.configuredDiscoveryPort, "127.0.0.1", () => {
+        resolve();
+      });
+    });
+    const address = this.discoveryServer.address();
+    if (!address) {
+      throw new Error("Discovery server did not expose a port");
+    }
+    this.discoveryPort = address.port;
+  }
+  stopDiscoveryServer() {
+    if (this.discoveryServer) {
+      this.discoveryServer.close();
+      this.discoveryServer = null;
+    }
+    this.discoveryPort = null;
+  }
   isRateLimited(ip) {
     const now = Date.now();
     const record = this.handshakeAttempts.get(ip);