Function startProxyServer

packages/cli/src/commands/ui-proxy.ts:193–233

(opts: StartProxyOptions)

];

export async function startProxyServer(opts: StartProxyOptions): Promise<ProxyServerHandle> {
  const timeoutMs = opts.upstreamTimeoutMs ?? DEFAULT_UPSTREAM_TIMEOUT_MS;
  const upgradeSockets = new Set<Duplex>();
  const httpServer: HttpServer = createHttpServer((req, res) => {
    if (rejectIfNotLoopbackApi(req, res)) return;
    forwardRequest(req, res, opts.upstreamHost, opts.upstreamPort, timeoutMs);
  });

  httpServer.on('upgrade', (req, clientSocket, head) => {
    if (rejectUpgradeIfNotLoopback(req, clientSocket)) return;
    proxyUpgrade(req, clientSocket, head, opts.upstreamHost, opts.upstreamPort, upgradeSockets);
  });

  await new Promise<void>((done, fail) => {
    const onError = (err: Error) => fail(err);
    httpServer.once('error', onError);
    httpServer.listen(opts.listenPort, opts.host, () => {
      httpServer.off('error', onError);
      done();
    });
  });

  const addr = httpServer.address();
  const port = typeof addr === 'object' && addr !== null ? addr.port : opts.listenPort;

  return {
    httpServer,
    port,
    close: () =>
      new Promise<void>((done) => {
        for (const sock of upgradeSockets) {
          try {
            sock.destroy();
          } catch {}
        }
        upgradeSockets.clear();
        httpServer.close(() => done());
        httpServer.closeIdleConnections();
      }),
  };
}

interface ProxyRequestOptions {
  upstreamHost: string;