#include "HandlerFactory.h"
#include "PingHandler.h"
#include "ColumnInfoHandler.h"
#include <Poco/URI.h>
#include <Poco/Net/HTTPServerRequest.h>
#include <common/logger_useful.h>

namespace DB
{

std::unique_ptr<HTTPRequestHandler> ODBCBridgeHandlerFactory::createRequestHandler(const HTTPServerRequest & request)
{
    Poco::URI uri{request.getURI()};
    LOG_TRACE(log, "Request URI: {}", uri.toString());

    if (uri.getPath() == "/ping" && request.getMethod() == Poco::Net::HTTPRequest::HTTP_GET)
        return std::make_unique<PingHandler>(keep_alive_timeout);

    if (request.getMethod() == Poco::Net::HTTPRequest::HTTP_POST)
    {

        if (uri.getPath() == "/columns_info")
#if USE_ODBC
            return std::make_unique<ODBCColumnsInfoHandler>(keep_alive_timeout, context);
#else
            return nullptr;
#endif
        else if (uri.getPath() == "/identifier_quote")
#if USE_ODBC
            return std::make_unique<IdentifierQuoteHandler>(keep_alive_timeout, context);
#else
            return nullptr;
#endif
        else if (uri.getPath() == "/schema_allowed")
#if USE_ODBC
            return std::make_unique<SchemaAllowedHandler>(keep_alive_timeout, context);
#else
            return nullptr;
#endif
        else if (uri.getPath() == "/write")
            return std::make_unique<ODBCHandler>(pool_map, keep_alive_timeout, context, "write");
        else
            return std::make_unique<ODBCHandler>(pool_map, keep_alive_timeout, context, "read");
    }
    return nullptr;
}
}