Method wrapWebSocketHandler

src/router.ts:241–283

(
    route: WebSocketRoute | BrowserWebsocketRoute,
    handler: WebSocketRoute['handler'] | BrowserWebsocketRoute['handler'],
  )

  }

  protected wrapWebSocketHandler(
    route: WebSocketRoute | BrowserWebsocketRoute,
    handler: WebSocketRoute['handler'] | BrowserWebsocketRoute['handler'],
  ) {
    return async (req: Request, socket: stream.Duplex, head: Buffer) => {
      if (!isConnected(socket)) {
        this.log.warn(`WebSocket Request has closed prior to running`);
        return ROUTE_DID_NOT_RUN;
      }
      const logger = new this.logger(route.name, req);
      if (
        Object.getPrototypeOf(route) instanceof BrowserWebsocketRoute &&
        'browser' in route &&
        route.browser
      ) {
        const browser = await this.browserManager.getBrowserForRequest(
          req,
          route,
          logger,
        );

        if (!isConnected(socket)) {
          this.log.warn(`WebSocket Request has closed prior to running`);
          this.browserManager.complete(browser);
          return ROUTE_DID_NOT_RUN;
        }

        if (!browser) {
          return writeResponse(socket, 500, `Error loading the browser.`);
        }

        try {
          this.log.trace(`Running found WebSocket handler.`);
          await handler(req, socket, head, logger, browser);
        } finally {
          this.log.trace(`WebSocket Request handler has finished.`);
          this.browserManager.complete(browser);
        }
        return;
      }
      return (handler as WebSocketRoute['handler'])(req, socket, head, logger);
    };
  }

  public registerHTTPRoute(
    route: HTTPRoute | BrowserHTTPRoute,