1 files changed, 52 insertions, 0 deletions
diff --git a/src/zenserver/hub/httpproxyhandler.h b/src/zenserver/hub/httpproxyhandler.h
new file mode 100644
index 000000000..8667c0ca1
--- /dev/null
+++ b/src/zenserver/hub/httpproxyhandler.h
@@ -0,0 +1,52 @@
+// Copyright Epic Games, Inc. All Rights Reserved.
+
+#pragma once
+
+#include <zencore/thread.h>
+#include <zenhttp/httpserver.h>
+#include <zenhttp/websocket.h>
+
+#include <functional>
+#include <memory>
+#include <unordered_map>
+
+namespace zen {
+
+class HttpClient;
+
+class HttpProxyHandler
+{
+public:
+	using PortValidator = std::function<bool(uint16_t)>;
+
+	HttpProxyHandler();
+	explicit HttpProxyHandler(PortValidator ValidatePort);
+	~HttpProxyHandler();
+
+	void SetPortValidator(PortValidator ValidatePort);
+
+	HttpProxyHandler(const HttpProxyHandler&) = delete;
+	HttpProxyHandler& operator=(const HttpProxyHandler&) = delete;
+
+	void HandleProxyRequest(HttpServerRequest& Request, std::string_view PortStr, std::string_view PathTail);
+	void PrunePort(uint16_t Port);
+	void Shutdown();
+
+	void OnWebSocketOpen(Ref<WebSocketConnection> Connection, std::string_view RelativeUri);
+	void OnWebSocketMessage(WebSocketConnection& Conn, const WebSocketMessage& Msg);
+	void OnWebSocketClose(WebSocketConnection& Conn, uint16_t Code, std::string_view Reason);
+
+private:
+	PortValidator m_ValidatePort;
+
+	HttpClient& GetOrCreateProxyClient(uint16_t Port);
+
+	RwLock													  m_ProxyClientsLock;
+	std::unordered_map<uint16_t, std::unique_ptr<HttpClient>> m_ProxyClients;
+
+	struct WsBridge;
+	RwLock													m_WsBridgesLock;
+	std::unordered_map<WebSocketConnection*, Ref<WsBridge>> m_WsBridges;
+};
+
+}  // namespace zen