#pragma once

#include <zencore/httpserver.h>

#define _WINSOCKAPI_
#include <zencore/windows.h>
#include "iothreadpool.h"

#include <atlbase.h>
#include <http.h>

namespace zen {

/**
 * @brief Windows implementation of HTTP server based on http.sys
 *
 * This requires elevation to function
 */
class HttpSysServer : public HttpServer
{
	friend class HttpSysTransaction;

public:
	explicit HttpSysServer(int ThreadCount);
	~HttpSysServer();

	// HttpServer interface implementation

	virtual void Initialize(int BasePort) override;
	virtual void Run(bool TestMode) override;
	virtual void RequestExit() override;
	virtual void AddEndpoint(HttpService& Service) override;

private:
	void Initialize(const wchar_t* UrlPath);

	void StartServer();
	void OnHandlingRequest();
	void IssueNewRequestMaybe();

	inline bool IsOk() const { return m_IsOk; }

	void AddEndpoint(const char* Endpoint, HttpService& Service);
	void RemoveEndpoint(const char* Endpoint, HttpService& Service);

private:
	bool			m_IsOk				= false;
	bool			m_IsHttpInitialized = false;
	WinIoThreadPool m_ThreadPool;

	std::wstring		   m_BaseUri;  // http://*:nnnn/
	HTTP_SERVER_SESSION_ID m_HttpSessionId		= 0;
	HTTP_URL_GROUP_ID	   m_HttpUrlGroupId		= 0;
	HANDLE				   m_RequestQueueHandle = 0;
	std::atomic_int32_t	   m_PendingRequests{0};
	int32_t				   m_MinPendingRequests = 16;
	int32_t				   m_MaxPendingRequests = 128;
	Event				   m_ShutdownEvent;
};

}  // namespace zen