HttpClient.hpp

Go to the documentation of this file.

#pragma once
#ifndef _KURLYK_HTTP_CLIENT_HPP_INCLUDED
#define _KURLYK_HTTP_CLIENT_HPP_INCLUDED

 
namespace kurlyk {

    class HttpClient {
    public:

        HttpClient() {
            ensure_initialized();
            m_request.request_id = HttpRequestManager::get_instance().generate_request_id();
        }

        HttpClient(const std::string& host) :
                m_host(host) {
            ensure_initialized();
            m_request.request_id = HttpRequestManager::get_instance().generate_request_id();
        }
 
        HttpClient(const HttpClient&) = delete;
        void operator=(const HttpClient&) = delete;

        virtual ~HttpClient() {
            cancel_requests();
            auto& instance = HttpRequestManager::get_instance();
            if (is_general_limit_owned) {
                instance.remove_limit(m_request.general_rate_limit_id);
            }
            if (is_specific_limit_owned) {
                instance.remove_limit(m_request.specific_rate_limit_id);
            }
        }

        void cancel_requests() {
            auto promise = std::make_shared<std::promise<void>>();
            auto future = promise->get_future();
            HttpRequestManager::get_instance().cancel_request_by_id(m_request.request_id, [promise](){
                try {
                    promise->set_value();
                } catch (const std::future_error& e) {
                    if (e.code() == std::make_error_condition(std::future_errc::promise_already_satisfied)) {
                        KURLYK_HANDLE_ERROR(e, "Promise already satisfied in HttpClient::request callback");
                    } else {
                        KURLYK_HANDLE_ERROR(e, "Future error in HttpClient::request callback");
                    }
                } catch (const std::exception& e) {
                    KURLYK_HANDLE_ERROR(e, "Unhandled exception in HttpClient::request callback");
                } catch (...) {
                    // Unknown fatal error in request callback
                }
            });
            core::NetworkWorker::get_instance().notify();
            try {
                future.get();
            } catch (const std::exception& e) {
                KURLYK_HANDLE_ERROR(e, "cancel_requests() future.get() failed");
            }
        }

        void set_host(const std::string& host) {
            m_host = host;
        }

        void set_headers(const kurlyk::Headers& headers) {
            m_request.headers = headers;
        }

        void assign_rate_limit_id(
                long limit_id,
                RateLimitType type = RateLimitType::RL_GENERAL) {
            auto& instance = HttpRequestManager::get_instance();
            switch (type) {
            case RateLimitType::RL_GENERAL:
                if (is_general_limit_owned) {
                    instance.remove_limit(m_request.general_rate_limit_id);
                }
                m_request.general_rate_limit_id = limit_id;
                is_general_limit_owned = false;
                break;
            case RateLimitType::RL_SPECIFIC:
                if (is_specific_limit_owned) {
                    instance.remove_limit(m_request.specific_rate_limit_id);
                }
                m_request.specific_rate_limit_id = limit_id;
                is_specific_limit_owned = false;
                break;
            }
        }

        void set_rate_limit_id(
                long limit_id,
                RateLimitType type = RateLimitType::RL_GENERAL) {
            assign_rate_limit_id(limit_id, type);
        }

        void set_rate_limit(
                long requests_per_period,
                long period_ms,
                RateLimitType type = RateLimitType::RL_GENERAL) {
            auto& instance = HttpRequestManager::get_instance();
            switch (type) {
            case RateLimitType::RL_GENERAL:
                if (is_general_limit_owned) {
                    instance.remove_limit(m_request.general_rate_limit_id);
                }
                m_request.general_rate_limit_id = instance.create_rate_limit(requests_per_period, period_ms);
                is_general_limit_owned = true;
                break;
            case RateLimitType::RL_SPECIFIC:
                if (is_specific_limit_owned) {
                    instance.remove_limit(m_request.specific_rate_limit_id);
                }
                m_request.specific_rate_limit_id = instance.create_rate_limit(requests_per_period, period_ms);
                is_specific_limit_owned = true;
                break;
            }
        }

        void set_rate_limit_rpm(
                long requests_per_minute,
                RateLimitType type = RateLimitType::RL_GENERAL) {
            long period_ms = 60000; // 1 minute in milliseconds
            return set_rate_limit(requests_per_minute, period_ms, type);
        }

        void set_rate_limit_rps(
                long requests_per_second,
                RateLimitType type = RateLimitType::RL_GENERAL) {
            long period_ms = 1000; // 1 second in milliseconds
            return set_rate_limit(requests_per_second, period_ms, type);
        }

        void set_accept_encoding(
                bool identity = false,
                bool deflate = false,
                bool gzip = false,
                bool brotli = false) {
            m_request.set_accept_encoding(identity, deflate, gzip, brotli);
        }

        void set_accept_encoding(const std::string& value) {
            m_request.accept_encoding = value;
        }

        void set_accept_language(const std::string& value) {
            m_request.headers.emplace("Accept-Language", value);
        }

        void set_content_type(const std::string& value) {
            m_request.headers.emplace("Content-Type", value);
        }

        void set_origin(const std::string& value) {
            m_request.headers.emplace("Origin", value);
        }

        void set_referer(const std::string& value) {
            m_request.headers.emplace("Referer", value);
        }

        void set_dnt(const bool value) {
            if (value) m_request.headers.emplace("dnt", "1");
        }

        void set_follow_location(bool value) {
            m_request.follow_location = value;
        }

        void set_auto_referer(bool value) {
            m_request.auto_referer = value;
        }

        void set_proxy_tunnel(bool value) {
            m_request.proxy_tunnel = value;
        }

        void set_head_only(bool value) {
            m_request.head_only = value;
        }

        void set_proxy(
                    const std::string& ip,
                    int port,
                    ProxyType type = ProxyType::PROXY_HTTP) {
            m_request.set_proxy(ip, port, type);
        }

        void set_proxy(
                const std::string& ip,
                const int port,
                const std::string& username,
                const std::string& password,
                ProxyType type = ProxyType::PROXY_HTTP) {
            m_request.set_proxy(ip, port, username, password, type);
        }

        void set_proxy_auth(
                const std::string& username,
                const std::string& password) {
            m_request.set_proxy_auth(username, password);
        }

        void set_proxy_server(const std::string& server) {
            m_request.set_proxy_server(server);
        }

        void set_proxy_auth(const std::string& auth) {
            m_request.set_proxy_auth(auth);
        }

        void set_proxy_type(ProxyType type) {
            m_request.set_proxy_type(type);
        }

        void set_retry_attempts(long retry_attempts, long retry_delay_ms) {
            m_request.set_retry_attempts(retry_attempts, retry_delay_ms);
        }

        void add_valid_status(long status) {
            m_request.add_valid_status(status);
        }

        void set_valid_statuses(const std::set<long>& statuses) {
            m_request.set_valid_statuses(statuses);
        }

        void clear_valid_statuses() {
            m_request.clear_valid_statuses();
        }

        void set_user_agent(const std::string& user_agent) {
            m_request.set_user_agent(user_agent);
        }

        void set_cookie(const std::string& cookie) {
            m_request.set_cookie(cookie);
        }

        void set_cert_file(const std::string& cert_file) {
            m_request.set_cert_file(cert_file);
        }

        void set_ca_file(const std::string& ca_file) {
            m_request.set_ca_file(ca_file);
        }

        void set_timeout(long timeout) {
            m_request.set_timeout(timeout);
        }

        void set_connect_timeout(long connect_timeout) {
            m_request.set_connect_timeout(connect_timeout);
        }

        void set_verbose(bool verbose) {
            m_request.verbose = verbose;
        }

        void set_debug_header(bool debug_header) {
            m_request.debug_header = debug_header;
        }

        void set_max_redirects(long max_redirects) {
            m_request.max_redirects = max_redirects;
        }

        bool request(
                const std::string &method,
                const std::string& path,
                const QueryParams &query,
                const Headers &headers,
                const std::string &content,
                HttpResponseCallback callback) {
#           if __cplusplus >= 201402L
            std::unique_ptr<HttpRequest> request_ptr = std::make_unique<HttpRequest>(m_request);
#           else
            std::unique_ptr<HttpRequest> request_ptr = std::unique_ptr<HttpRequest>(new HttpRequest(m_request));
#           endif
            request_ptr->method = method;
            request_ptr->set_url(m_host, path, query);
            request_ptr->headers.insert(headers.begin(), headers.end());
            request_ptr->content = content;
            return request(std::move(request_ptr), std::move(callback));
        }

        bool request(
                const std::string &method,
                const std::string& path,
                const QueryParams &query,
                const Headers &headers,
                const std::string &content,
                long specific_rate_limit_id,
                HttpResponseCallback callback) {
#           if __cplusplus >= 201402L
            std::unique_ptr<HttpRequest> request_ptr = std::make_unique<HttpRequest>(m_request);
#           else
            std::unique_ptr<HttpRequest> request_ptr = std::unique_ptr<HttpRequest>(new HttpRequest(m_request));
#           endif
            request_ptr->method = method;
            request_ptr->set_url(m_host, path, query);
            request_ptr->headers.insert(headers.begin(), headers.end());
            request_ptr->content = content;
 
            // Set the specific rate limit ID for this request
            if (is_specific_limit_owned) {
                HttpRequestManager::get_instance().remove_limit(request_ptr->specific_rate_limit_id);
            }
            request_ptr->specific_rate_limit_id = specific_rate_limit_id;
            is_specific_limit_owned = false;
 
            return request(std::move(request_ptr), std::move(callback));
        }

        bool get(
                const std::string& path,
                const QueryParams& query,
                const Headers& headers,
                HttpResponseCallback callback) {
            return request("GET", path, query, headers, std::string(), std::move(callback));
        }

        bool post(
                const std::string& path,
                const QueryParams& query,
                const Headers& headers,
                const std::string& content,
                HttpResponseCallback callback) {
            return request("POST", path, query, headers, content, std::move(callback));
        }

        bool get(
                const std::string& path,
                const QueryParams& query,
                const Headers& headers,
                long specific_rate_limit_id,
                HttpResponseCallback callback) {
            return request("GET", path, query, headers, std::string(), specific_rate_limit_id, std::move(callback));
        }

        bool post(
                const std::string& path,
                const QueryParams& query,
                const Headers& headers,
                const std::string& content,
                long specific_rate_limit_id,
                HttpResponseCallback callback) {
            return request("POST", path, query, headers, content, specific_rate_limit_id, std::move(callback));
        }

        std::future<HttpResponsePtr> request(
                const std::string& method,
                const std::string& path,
                const QueryParams& query,
                const Headers& headers,
                const std::string& content) {
#           if __cplusplus >= 201402L
            auto request_ptr = std::make_unique<HttpRequest>(m_request);
#           else
            auto request_ptr = std::unique_ptr<HttpRequest>(new HttpRequest(m_request));
#           endif
            request_ptr->method = method;
            request_ptr->set_url(m_host, path, query);
            request_ptr->headers.insert(headers.begin(), headers.end());
            request_ptr->content = content;
 
            auto promise = std::make_shared<std::promise<HttpResponsePtr>>();
            auto future = promise->get_future();
 
            HttpResponseCallback callback = [promise](HttpResponsePtr response) {
                safe_set_response(promise, std::move(response));
            };
 
            safe_submit_request(
                promise, std::move(request_ptr), std::move(callback));
 
            return future;
        }

        std::future<HttpResponsePtr> request(
                const std::string& method,
                const std::string& path,
                const QueryParams& query,
                const Headers& headers,
                const std::string& content,
                long specific_rate_limit_id) {
#           if __cplusplus >= 201402L
            auto request_ptr = std::make_unique<HttpRequest>(m_request);
#           else
            auto request_ptr = std::unique_ptr<HttpRequest>(new HttpRequest(m_request));
#           endif
            request_ptr->method = method;
            request_ptr->set_url(m_host, path, query);
            request_ptr->headers.insert(headers.begin(), headers.end());
            request_ptr->content = content;
 
            // Set the specific rate limit ID for this request
            if (is_specific_limit_owned) {
                HttpRequestManager::get_instance().remove_limit(request_ptr->specific_rate_limit_id);
            }
            request_ptr->specific_rate_limit_id = specific_rate_limit_id;
            is_specific_limit_owned = false;
 
            auto promise = std::make_shared<std::promise<HttpResponsePtr>>();
            auto future = promise->get_future();
 
            HttpResponseCallback callback = [promise](HttpResponsePtr response) {
                safe_set_response(promise, std::move(response));
            };
 
            safe_submit_request(
                promise, std::move(request_ptr), std::move(callback));
 
            return future;
        }

        std::future<HttpResponsePtr> get(
                const std::string& path,
                const QueryParams& query,
                const Headers& headers) {
            return request("GET", path, query, headers, std::string());
        }

        std::future<HttpResponsePtr> post(
                const std::string& path,
                const QueryParams& query,
                const Headers& headers,
                const std::string& content) {
            return request("POST", path, query, headers, content);
        }

        std::future<HttpResponsePtr> get(
                const std::string& path,
                const QueryParams& query,
                const Headers& headers,
                long specific_rate_limit_id) {
            return request("GET", path, query, headers, std::string(), specific_rate_limit_id);
        }

        std::future<HttpResponsePtr> post(
                const std::string& path,
                const QueryParams& query,
                const Headers& headers,
                const std::string& content,
                long specific_rate_limit_id) {
            return request("POST", path, query, headers, content, specific_rate_limit_id);
        }
 
    private:
        HttpRequest m_request;  
        std::string m_host;     
        bool is_general_limit_owned = false; 
        bool is_specific_limit_owned = false; 

        bool request(
                std::unique_ptr<HttpRequest> request_ptr,
                HttpResponseCallback callback) {
            const bool status = HttpRequestManager::get_instance().add_request(std::move(request_ptr), std::move(callback));
            core::NetworkWorker::get_instance().notify();
            return status;
        }

        static void safe_set_response(
                std::shared_ptr<std::promise<HttpResponsePtr>> promise,
                HttpResponsePtr response) {
            if (!response || !response->ready) return;
            try {
                promise->set_value(std::move(response));
            } catch (const std::future_error& e) {
                if (e.code() == std::make_error_condition(std::future_errc::promise_already_satisfied)) {
                    KURLYK_HANDLE_ERROR(e, "Promise already satisfied in HttpClient::request callback");
                } else {
                    KURLYK_HANDLE_ERROR(e, "Future error in HttpClient::request callback");
                }
            } catch (const std::exception& e) {
                KURLYK_HANDLE_ERROR(e, "Unhandled exception in HttpClient::request callback");
            } catch (...) {
                // Unknown fatal error in request callback
            }
        }

        void safe_submit_request(
                std::shared_ptr<std::promise<HttpResponsePtr>> promise,
                std::unique_ptr<HttpRequest> request_ptr,
                HttpResponseCallback callback) {
            bool exception_set = false;
            try {
                if (!request(std::move(request_ptr), std::move(callback))) {
                    promise->set_exception(std::make_exception_ptr(
                        std::runtime_error("Failed to add request to RequestManager")));
                    exception_set = true;
                }
            } catch (const std::exception& e) {
                KURLYK_HANDLE_ERROR(e, "Exception while submitting request");
                if (!exception_set) {
                    try {
                        promise->set_exception(std::current_exception());
                    } catch (...) {} // fallback
                }
            } catch (...) {
                // Unknown fatal error while submitting request
                if (!exception_set) {
                    try {
                        promise->set_exception(std::current_exception());
                    } catch (...) {} // fallback
                }
            }
        }

        static void ensure_initialized() {
            static bool is_initialized = false;
            if (!is_initialized) {
                is_initialized = true;
                HttpRequestManager::get_instance();
                core::NetworkWorker::get_instance().start(KURLYK_AUTO_INIT_USE_ASYNC);
            }
        }
 
    }; // HttpClient
 
}; // namespace kurlyk
 
#endif // _KURLYK_HTTP_CLIENT_HPP_INCLUDED