merge master

1 files changed, 123 insertions, 0 deletions

diff --git a/src/http.cpp b/src/http.cpp
new file mode 100644
index 0000000..790add7
--- /dev/null
+++ b/src/http.cpp
@@ -0,0 +1,123 @@
+#include "http.hpp"
+
+namespace http {
+request::request(EMethod method, const std::string &url)
+    : m_url(url) {
+    switch (method) {
+        case REQUEST_GET:
+            m_method = "GET";
+            break;
+        case REQUEST_POST:
+            m_method = "POST";
+            break;
+        case REQUEST_PATCH:
+            m_method = "PATCH";
+            break;
+        case REQUEST_PUT:
+            m_method = "PUT";
+            break;
+        case REQUEST_DELETE:
+            m_method = "DELETE";
+            break;
+        default:
+            m_method = "GET";
+            break;
+    }
+
+    prepare();
+}
+
+request::~request() {
+    if (m_curl != nullptr)
+        curl_easy_cleanup(m_curl);
+
+    if (m_header_list != nullptr)
+        curl_slist_free_all(m_header_list);
+}
+
+void request::set_verify_ssl(bool verify) {
+    curl_easy_setopt(m_curl, CURLOPT_SSL_VERIFYPEER, verify ? 1L : 0L);
+}
+
+void request::set_proxy(const std::string &proxy) {
+    curl_easy_setopt(m_curl, CURLOPT_PROXY, proxy.c_str());
+}
+
+void request::set_header(const std::string &name, const std::string &value) {
+    m_header_list = curl_slist_append(m_header_list, (name + ": " + value).c_str());
+}
+
+void request::set_body(const std::string &data) {
+    curl_easy_setopt(m_curl, CURLOPT_COPYPOSTFIELDS, data.c_str());
+}
+
+void request::set_user_agent(const std::string &data) {
+    curl_easy_setopt(m_curl, CURLOPT_USERAGENT, data.c_str());
+}
+
+response request::execute() {
+    if (m_curl == nullptr) {
+        auto response = detail::make_response(m_url, EStatusCode::ClientErrorCURLInit);
+        response.error_string = "curl pointer is null";
+    }
+
+    detail::check_init();
+
+    std::string str;
+    curl_easy_setopt(m_curl, CURLOPT_NOSIGNAL, 1L);
+    curl_easy_setopt(m_curl, CURLOPT_CUSTOMREQUEST, m_method);
+    curl_easy_setopt(m_curl, CURLOPT_URL, m_url.c_str());
+    curl_easy_setopt(m_curl, CURLOPT_FOLLOWLOCATION, 1);
+    curl_easy_setopt(m_curl, CURLOPT_WRITEFUNCTION, detail::curl_write_data_callback);
+    curl_easy_setopt(m_curl, CURLOPT_WRITEDATA, &str);
+    curl_easy_setopt(m_curl, CURLOPT_ERRORBUFFER, m_error_buf);
+    m_error_buf[0] = '\0';
+    if (m_header_list != nullptr)
+        curl_easy_setopt(m_curl, CURLOPT_HTTPHEADER, m_header_list);
+
+    CURLcode result = curl_easy_perform(m_curl);
+    if (result != CURLE_OK) {
+        auto response = detail::make_response(m_url, EStatusCode::ClientErrorCURLPerform);
+        response.error_string = curl_easy_strerror(result);
+        response.error_string += " " + std::string(m_error_buf);
+        return response;
+    }
+
+    long response_code = 0;
+    curl_easy_getinfo(m_curl, CURLINFO_RESPONSE_CODE, &response_code);
+
+    auto response = detail::make_response(m_url, response_code);
+    response.text = str;
+
+    return response;
+}
+
+void request::prepare() {
+    m_curl = curl_easy_init();
+}
+
+namespace detail {
+    size_t curl_write_data_callback(void *ptr, size_t size, size_t nmemb, void *userdata) {
+        const size_t n = size * nmemb;
+        static_cast<std::string*>(userdata)->append(static_cast<char*>(ptr), n);
+        return n;
+    }
+
+    response make_response(const std::string &url, int code) {
+        response r;
+        r.url = url;
+        r.status_code = static_cast<EStatusCode>(code);
+        if (code < http::EStatusCode::ClientErrorMax)
+            r.error = true;
+        return r;
+    }
+
+    void check_init() {
+        static bool initialized = false;
+        if (!initialized) {
+            curl_global_init(CURL_GLOBAL_ALL);
+            initialized = true;
+        }
+    }
+} // namespace detail
+} // namespace http