/**
 * Relational pipes
 * Copyright © 2022 František Kučera (Frantovo.cz, GlobalCode.info)
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, version 3 of the License.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program. If not, see <http://www.gnu.org/licenses/>.
 */

#include <microhttpd.h>
#include <stdexcept>

#include "HTTPServer.h"

namespace relpipe {
namespace tr {
namespace httpd {

class HTTPServer::HTTPServerImpl {
public:
	MHD_Daemon* mhd = nullptr;
	std::shared_ptr<RequestHandler> requestHandler;
};

void HTTPServer::setRequestHandler(std::shared_ptr<RequestHandler> handler) {
	impl->requestHandler = handler;
}

HTTPServer* HTTPServer::create(HTTPServer::Options options) {
	HTTPServer::HTTPServerImpl* impl = new HTTPServer::HTTPServerImpl();

	void* acceptCallbackData = impl;
	MHD_AcceptPolicyCallback acceptCallback = [](void* rawImpl, const struct sockaddr* addr, socklen_t addrlen) {
		if (rawImpl) {
			HTTPServer::HTTPServerImpl* impl = static_cast<HTTPServer::HTTPServerImpl*> (rawImpl);
			// TODO: call impl if present and has such method
			return MHD_YES;
		} else {
			return MHD_YES;
		}
	};

	void* accessCallbackData = impl;
	MHD_AccessHandlerCallback accessCallback = [](void* rawImpl, struct MHD_Connection* connection, const char* url, const char* method, const char* version, const char* upload_data, size_t* upload_data_size, void** con_cls) {
		HTTPServer::HTTPServerImpl* impl = static_cast<HTTPServer::HTTPServerImpl*> (rawImpl);
		// TODO: return also HTTP headers
		if (impl->requestHandler) {
			HTTPServer::Request request;
			const HTTPServer::Response response = impl->requestHandler->handle(request);
			struct MHD_Response* mhdResponse = MHD_create_response_from_buffer(response.body.size(), (void*) response.body.c_str(), MHD_RESPMEM_MUST_COPY);
			MHD_queue_response(connection, response.code, mhdResponse);
			MHD_destroy_response(mhdResponse);
			return MHD_YES;
		} else {
			// there might be a point in time when the HTTP server is started and HTTP handler is not set
			// TODO: just return MHD_NO?
			static const char body[] = "<h1>HTTP 404: Not found</h1>";
			struct MHD_Response* mhdResponse = MHD_create_response_from_buffer(sizeof (body), (void*) body, MHD_RESPMEM_PERSISTENT);
			MHD_queue_response(connection, 404, mhdResponse);
			MHD_destroy_response(mhdResponse);
			return MHD_YES;
		}
	};


	impl->mhd = MHD_start_daemon(MHD_USE_INTERNAL_POLLING_THREAD,
			options.tcpPort,
			acceptCallback, acceptCallbackData,
			accessCallback, accessCallbackData,
			MHD_OPTION_THREAD_POOL_SIZE, 10,
			MHD_OPTION_CONNECTION_TIMEOUT, 60,
			MHD_OPTION_END);


	if (impl->mhd) return new HTTPServer(impl);
	else throw std::logic_error("Unable to start MHD.");
}

HTTPServer::~HTTPServer() {
	MHD_stop_daemon(impl->mhd);
	delete impl;
}


}
}
}