abaddon/discord/httpclient.cpp

#include "httpclient.hpp"
#include "httpclient.hpp"

//#define USE_LOCAL_PROXY
HTTPClient::HTTPClient(std::string api_base)
    : m_api_base(api_base) {
    m_dispatcher.connect(sigc::mem_fun(*this, &HTTPClient::RunCallbacks));
}

void HTTPClient::SetUserAgent(std::string agent) {
    m_agent = agent;
}

void HTTPClient::SetAuth(std::string auth) {
    m_authorization = auth;
}

void HTTPClient::MakeDELETE(const std::string &path, std::function<void(http::response_type r)> cb) {
    printf("DELETE %s\n", path.c_str());
    m_futures.push_back(std::async(std::launch::async, [this, path, cb] {
        http::request req(http::REQUEST_DELETE, m_api_base + path);
        req.set_header("Authorization", m_authorization);
        req.set_user_agent(m_agent != "" ? m_agent : "Abaddon");
#ifdef USE_LOCAL_PROXY
        req.set_proxy("http://127.0.0.1:8888");
        req.set_verify_ssl(false);
#endif

        auto res = req.execute();

        OnResponse(res, cb);
    }));
}

void HTTPClient::MakePATCH(const std::string &path, const std::string &payload, std::function<void(http::response_type r)> cb) {
    printf("PATCH %s\n", path.c_str());
    m_futures.push_back(std::async(std::launch::async, [this, path, cb, payload] {
        http::request req(http::REQUEST_PATCH, m_api_base + path);
        req.set_header("Authorization", m_authorization);
        req.set_header("Content-Type", "application/json");
        req.set_user_agent(m_agent != "" ? m_agent : "Abaddon");
        req.set_body(payload);
#ifdef USE_LOCAL_PROXY
        req.set_proxy("http://127.0.0.1:8888");
        req.set_verify_ssl(false);
#endif

        auto res = req.execute();

        OnResponse(res, cb);
    }));
}

void HTTPClient::MakePOST(const std::string &path, const std::string &payload, std::function<void(http::response_type r)> cb) {
    printf("POST %s\n", path.c_str());
    m_futures.push_back(std::async(std::launch::async, [this, path, cb, payload] {
        http::request req(http::REQUEST_POST, m_api_base + path);
        req.set_header("Authorization", m_authorization);
        req.set_header("Content-Type", "application/json");
        req.set_user_agent(m_agent != "" ? m_agent : "Abaddon");
        req.set_body(payload);
#ifdef USE_LOCAL_PROXY
        req.set_proxy("http://127.0.0.1:8888");
        req.set_verify_ssl(false);
#endif

        auto res = req.execute();

        OnResponse(res, cb);
    }));
}

void HTTPClient::MakePUT(const std::string &path, const std::string &payload, std::function<void(http::response_type r)> cb) {
    printf("PUT %s\n", path.c_str());
    m_futures.push_back(std::async(std::launch::async, [this, path, cb, payload] {
        http::request req(http::REQUEST_PUT, m_api_base + path);
        req.set_header("Authorization", m_authorization);
        req.set_header("Content-Type", "application/json");
        req.set_user_agent(m_agent != "" ? m_agent : "Abaddon");
        req.set_body(payload);
#ifdef USE_LOCAL_PROXY
        req.set_proxy("http://127.0.0.1:8888");
        req.set_verify_ssl(false);
#endif

        auto res = req.execute();

        OnResponse(res, cb);
    }));
}

void HTTPClient::MakeGET(const std::string &path, std::function<void(http::response_type r)> cb) {
    printf("GET %s\n", path.c_str());
    m_futures.push_back(std::async(std::launch::async, [this, path, cb] {
        http::request req(http::REQUEST_GET, m_api_base + path);
        req.set_header("Authorization", m_authorization);
        req.set_header("Content-Type", "application/json");
        req.set_user_agent(m_agent != "" ? m_agent : "Abaddon");
#ifdef USE_LOCAL_PROXY
        req.set_proxy("http://127.0.0.1:8888");
        req.set_verify_ssl(false);
#endif

        auto res = req.execute();

        OnResponse(res, cb);
    }));
}

void HTTPClient::CleanupFutures() {
    for (auto it = m_futures.begin(); it != m_futures.end();) {
        if (it->wait_for(std::chrono::seconds(0)) == std::future_status::ready)
            it = m_futures.erase(it);
        else
            it++;
    }
}

void HTTPClient::RunCallbacks() {
    m_mutex.lock();
    m_queue.front()();
    m_queue.pop();
    m_mutex.unlock();
}

void HTTPClient::OnResponse(const http::response_type &r, std::function<void(http::response_type r)> cb) {
    CleanupFutures();
    try {
        m_mutex.lock();
        m_queue.push([this, r, cb] { cb(r); });
        m_dispatcher.emit();
        m_mutex.unlock();
    } catch (const std::exception &e) {
        fprintf(stderr, "error handling response (%s, code %d): %s\n", r.url.c_str(), r.status_code, e.what());
    }
}