From bbfa78ea9e04aacf6467ce4adff9fb91b4aab169 Mon Sep 17 00:00:00 2001
From: Rohit Rajan <rohit.rajan031101@gmail.com>
Date: Sun, 30 Nov 2025 22:13:49 +0530
Subject: [PATCH] fix: wss launch config

---
 browser/server.ts  | 7 +++++--
 docker-compose.yml | 2 ++
 2 files changed, 7 insertions(+), 2 deletions(-)

diff --git a/browser/server.ts b/browser/server.ts
index 2a70beef..8ee6ca6a 100644
--- a/browser/server.ts
+++ b/browser/server.ts
@@ -11,6 +11,7 @@ let browserServer: BrowserServer | null = null;
 // Configurable ports with defaults
 const BROWSER_WS_PORT = parseInt(process.env.BROWSER_WS_PORT || '3001', 10);
 const BROWSER_HEALTH_PORT = parseInt(process.env.BROWSER_HEALTH_PORT || '3002', 10);
+const BROWSER_WS_HOST = process.env.BROWSER_WS_HOST || 'localhost';
 
 async function start(): Promise<void> {
     console.log('Starting Maxun Browser Service...');
@@ -44,17 +45,19 @@ async function start(): Promise<void> {
         // Health check HTTP server
         const healthServer = http.createServer((req, res) => {
             if (req.url === '/health') {
+                const wsEndpoint = browserServer?.wsEndpoint();
                 res.writeHead(200, { 'Content-Type': 'application/json' });
                 res.end(JSON.stringify({
                     status: 'healthy',
-                    wsEndpoint: browserServer?.wsEndpoint(),
+                    wsEndpoint,
                     wsPort: BROWSER_WS_PORT,
                     healthPort: BROWSER_HEALTH_PORT,
                     timestamp: new Date().toISOString()
                 }));
             } else if (req.url === '/') {
                 res.writeHead(200, { 'Content-Type': 'text/plain' });
-                res.end(`Maxun Browser Service\nWebSocket: ${browserServer?.wsEndpoint()}\nHealth: http://localhost:${BROWSER_HEALTH_PORT}/health`);
+                const wsEndpoint = browserServer?.wsEndpoint().replace('localhost', BROWSER_WS_HOST) || '';
+                res.end(`Maxun Browser Service\nWebSocket: ${wsEndpoint}\nHealth: http://localhost:${BROWSER_HEALTH_PORT}/health`);
             } else {
                 res.writeHead(404);
                 res.end('Not Found');
diff --git a/docker-compose.yml b/docker-compose.yml
index 381c6671..700303fd 100644
--- a/docker-compose.yml
+++ b/docker-compose.yml
@@ -89,6 +89,8 @@ services:
       - DEBUG=pw:browser*
       - BROWSER_WS_PORT=${BROWSER_WS_PORT:-3001}
       - BROWSER_HEALTH_PORT=${BROWSER_HEALTH_PORT:-3002}
+      - BROWSER_WS_HOST=${BROWSER_WS_HOST:-browser}
+      - PLAYWRIGHT_BROWSERS_PATH=/ms-playwright
     restart: unless-stopped
     healthcheck:
       test: ["CMD", "curl", "-f", "http://localhost:${BROWSER_HEALTH_PORT:-3002}/health"]