#pragma once

#include <winsock2.h>

#include <atomic>
#include <filesystem>
#include <functional>
#include <mutex>
#include <string>
#include <thread>
#include <vector>

class ControlServer
{
public:
	struct Callbacks
	{
		std::function<std::string()> getStateJson;
		std::function<bool(const std::string&, std::string&)> selectShader;
		std::function<bool(const std::string&, const std::string&, const std::string&, std::string&)> updateParameter;
		std::function<bool(const std::string&, std::string&)> resetParameters;
		std::function<bool(bool, std::string&)> setBypass;
		std::function<bool(double, std::string&)> setMixAmount;
		std::function<bool(std::string&)> reloadShader;
	};

	ControlServer();
	~ControlServer();

	bool Start(const std::filesystem::path& uiRoot, unsigned short preferredPort, const Callbacks& callbacks, std::string& error);
	void Stop();
	void BroadcastState();

	unsigned short GetPort() const { return mPort; }

private:
	struct ClientConnection
	{
		SOCKET socket = INVALID_SOCKET;
		bool websocket = false;
	};

	void ServerLoop();
	bool HandleHttpClient(SOCKET clientSocket);
	bool TryAcceptClient();
	bool SendHttpResponse(SOCKET clientSocket, const std::string& status, const std::string& contentType, const std::string& body);
	bool HandleHttpRequest(SOCKET clientSocket, const std::string& request);
	bool HandleWebSocketUpgrade(SOCKET clientSocket, const std::string& request);
	bool SendWebSocketText(SOCKET clientSocket, const std::string& payload);
	void BroadcastStateLocked();
	std::string LoadUiAsset(const std::string& relativePath, std::string& contentType) const;
	std::string BuildJsonResponse(bool success, const std::string& error = std::string()) const;
	static std::string Base64Encode(const unsigned char* data, DWORD dataLength);
	static std::string ComputeWebSocketAcceptKey(const std::string& clientKey);
	static std::string GetHeaderValue(const std::string& request, const std::string& headerName);
	static std::string GetRequestPath(const std::string& request);
	static std::string GetRequestMethod(const std::string& request);
	static std::string GetRequestBody(const std::string& request);

private:
	std::filesystem::path mUiRoot;
	Callbacks mCallbacks;
	SOCKET mListenSocket;
	unsigned short mPort;
	std::thread mThread;
	std::atomic<bool> mRunning;
	mutable std::mutex mMutex;
	std::vector<ClientConnection> mClients;
};