/**

 * 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?

			struct MHD_Response* mhdResponse = MHD_create_response_from_buffer(sizeof (body), (void*) body, MHD_RESPMEM_PERSISTENT);

			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;

}

}

}

}