static new ThreadLocal<SS> WebSocketHTTPD_headers;
static new ThreadLocal<WebSocketHTTPD> WebSocketHTTPD_current;

sclass WebSocketHTTPD extends NanoWebSocketServer {
  *(int port, O factory) {
    super(port, webSocketFactory(factory));
    nanohttpd_socketTimeoutFix();
  }

  // The following is all copied from #1011857
  
  O onServe, onEndServe;
  volatile long requests;
  bool printServes = true;
  
  sclass Serving {
    S uri;
    SS header, parms, files;
  }
  
  Set<Serving> currentlyServing = synchroSet();

  public Response serve(S uri, Method method,
    SS header, SS parms, SS files) ctex {
    ++requests;
    temp tempSetThreadLocal(WebSocketHTTPD_current, this);
    Serving serving = nu(Serving.class, +uri, +header, +parms, +files);
    currentlyServing.add(serving);
    pcallF(onServe, serving);
    try {
      S remoteAddr = getClientIPFromHeaders(header);
      if (printServes)
        print(formatDateAndTime() + " Serving URI: " + quote(header.get("host") + uri) + " to: " + remoteAddr);
      
      /*if (nempty(remoteAddr) &&
        isTrue(callOpt(mc(), "isBlockedIP", remoteAddr))) {
        print("BLOCKED IP.");
        ret serve404();
      }*/
      
      try {
        Response response = cast callOpt(getMainClass(), "serve", uri, method, header, parms, files);
        if (response != null) ret response;
        O html;
        WebSocketHTTPD_headers.set(header);
        try {
          html = callHtmlMethod2(getMainClass(), uri, parms);
        } finally {
          WebSocketHTTPD_headers.set(null);
        }
        
        if (html != null)
          ret html instanceof S ? serveHTML((S) html) : (NanoHTTPD.Response) html;
        ret serve404();
      } catch (Throwable e) {
        printStackTrace(e);
        ret serveHTML("ERROR.");
      }
    } finally {
      pcallF(onEndServe, serving);
      currentlyServing.remove(serving);
    }
  }
}