// 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