Proxy server to inject shim (#178)

things to test:

- [x] allow real URL to open in new window
- [x] packaging in electron?
- [ ] does it work on windows?
- [x] make sure it works with older apps
- [x] what about cache / reuse? - maybe use a bigger range of ports??

src/ipc/handlers/app_handlers.ts

@@ -33,10 +33,14 @@ import log from "electron-log";
 import { getSupabaseProjectName } from "../../supabase_admin/supabase_management_client";
 import { createLoggedHandler } from "./safe_handle";
 import { getLanguageModelProviders } from "../shared/language_model_helpers";
+import { startProxy } from "../utils/start_proxy_server";
+import { Worker } from "worker_threads";
 
 const logger = log.scope("app_handlers");
 const handle = createLoggedHandler(logger);
 
+let proxyWorker: Worker | null = null;
+
 // Needed, otherwise electron in MacOS/Linux will not be able
 // to find node/pnpm.
 fixPath();
@@ -50,8 +54,13 @@ async function executeApp({
   appId: number;
   event: Electron.IpcMainInvokeEvent;
 }): Promise<void> {
+  if (proxyWorker) {
+    proxyWorker.terminate();
+    proxyWorker = null;
+  }
   await executeAppLocalNode({ appPath, appId, event });
 }
+
 async function executeAppLocalNode({
   appPath,
   appId,
@@ -90,14 +99,27 @@ async function executeAppLocalNode({
   runningApps.set(appId, { process, processId: currentProcessId });
 
   // Log output
-  process.stdout?.on("data", (data) => {
+  process.stdout?.on("data", async (data) => {
     const message = util.stripVTControlCharacters(data.toString());
     logger.debug(`App ${appId} (PID: ${process.pid}) stdout: ${message}`);
+
     event.sender.send("app:output", {
       type: "stdout",
       message,
       appId,
     });
+    const urlMatch = message.match(/(https?:\/\/localhost:\d+\/?)/);
+    if (urlMatch) {
+      proxyWorker = await startProxy(urlMatch[1], {
+        onStarted: (proxyUrl) => {
+          event.sender.send("app:output", {
+            type: "stdout",
+            message: `[dyad-proxy-server]started=[${proxyUrl}] original=[${urlMatch[1]}]`,
+            appId,
+          });
+        },
+      });
+    }
   });
 
   process.stderr?.on("data", (data) => {

src/ipc/utils/port_utils.ts

@@ -0,0 +1,48 @@
+import net from "net";
+
+export function findAvailablePort(
+  minPort: number,
+  maxPort: number,
+): Promise<number> {
+  return new Promise((resolve, reject) => {
+    let attempts = 0;
+    const maxAttempts = 3;
+
+    function tryPort() {
+      if (attempts >= maxAttempts) {
+        reject(
+          new Error(
+            `Failed to find an available port after ${maxAttempts} attempts.`,
+          ),
+        );
+        return;
+      }
+
+      attempts++;
+      const port =
+        Math.floor(Math.random() * (maxPort - minPort + 1)) + minPort;
+      const server = net.createServer();
+
+      server.once("error", (err: any) => {
+        if (err.code === "EADDRINUSE") {
+          // Port is in use, try another one
+          console.log(`Port ${port} is in use, trying another...`);
+          server.close(() => tryPort());
+        } else {
+          // Other error
+          server.close(() => reject(err));
+        }
+      });
+
+      server.once("listening", () => {
+        server.close(() => {
+          resolve(port);
+        });
+      });
+
+      server.listen(port, "localhost");
+    }
+
+    tryPort();
+  });
+}

src/ipc/utils/start_proxy_server.ts

@@ -0,0 +1,55 @@
+// startProxy.js – helper to launch proxy.js as a worker
+
+import { Worker } from "worker_threads";
+import path from "path";
+import { findAvailablePort } from "./port_utils";
+import log from "electron-log";
+
+const logger = log.scope("start_proxy_server");
+
+export async function startProxy(
+  targetOrigin: string,
+  opts: {
+    // host?: string;
+    // port?: number;
+    // env?: Record<string, string>;
+    onStarted?: (proxyUrl: string) => void;
+  } = {},
+) {
+  if (!/^https?:\/\//.test(targetOrigin))
+    throw new Error("startProxy: targetOrigin must be absolute http/https URL");
+  const port = await findAvailablePort(50_000, 60_000);
+  logger.info("Found available port", port);
+  const {
+    // host = "localhost",
+    // env = {}, // additional env vars to pass to the worker
+    onStarted,
+  } = opts;
+
+  const worker = new Worker(
+    path.resolve(__dirname, "..", "..", "worker", "proxy_server.js"),
+    {
+      env: {
+        ...process.env, // inherit parent env
+
+        TARGET_URL: targetOrigin,
+      },
+      workerData: {
+        targetOrigin,
+        port,
+      },
+    },
+  );
+
+  worker.on("message", (m) => {
+    logger.info("[proxy]", m);
+    if (typeof m === "string" && m.startsWith("proxy-server-start url=")) {
+      const url = m.substring("proxy-server-start url=".length);
+      onStarted?.(url);
+    }
+  });
+  worker.on("error", (e) => logger.error("[proxy] error:", e));
+  worker.on("exit", (c) => logger.info("[proxy] exit", c));
+
+  return worker; // let the caller keep a handle if desired
+}