tv_rename/network.cpp

#include "network.hpp"
#include <iostream>

#ifdef _WIN32

// shamelessly stolen from http://www.cplusplus.com/forum/windows/109799/

Curl::Curl() {
    connect = InternetOpen( L"WinInet/1.0", INTERNET_OPEN_TYPE_PRECONFIG,
                            nullptr, nullptr, 0 );
    if ( !connect )
        std::wcerr << "Something went wrong while connecting" << std::endl;
}

Curl::~Curl() {
    if ( connect )
        InternetCloseHandle( connect );
}

std::string Curl::execute( const std::wstring &url ) {
    std::string source{};
    source.reserve( 10000 );

    HINTERNET openAddress = InternetOpenUrl(
        connect, url.c_str(), nullptr, 0,
        INTERNET_FLAG_PRAGMA_NOCACHE | INTERNET_FLAG_KEEP_CONNECTION, 0 );

    if ( !openAddress ) {
        unsigned long errorNum = GetLastError();
        std::wcout << "Failed to open URL" << std::endl
                   << "Error No: " << errorNum << std::endl;
        return "";
    }

    char dataReceived[4096];
    unsigned long numberOfBytesRead = 0;
    while ( InternetReadFile( openAddress, dataReceived, 4096,
                              &numberOfBytesRead ) &&
            numberOfBytesRead ) {
        source.append( dataReceived, numberOfBytesRead );
    }

    InternetCloseHandle( openAddress );
    return source;
}

#else

size_t writeCallback( void *contents, size_t size, size_t nmemb,
                      void *target ) {
    *static_cast< std::string * >( target ) +=
        std::string( static_cast< char * >( contents ), size * nmemb );
    return size * nmemb;
}

Curl::Curl() {
    curl_global_init( CURL_GLOBAL_ALL );
    curl_handle = curl_easy_init();
    curl_easy_setopt( curl_handle, CURLOPT_WRITEFUNCTION, writeCallback );
    curl_easy_setopt( curl_handle, CURLOPT_USERAGENT, "libcurl-agent/1.0" );
}

Curl::~Curl() {
    curl_easy_cleanup( curl_handle );
    curl_slist_free_all( chunk );
    curl_global_cleanup();
}

void cleanUp( CURL *curl_handle ) {
    curl_easy_setopt( curl_handle, CURLOPT_POST, 0 );
    curl_easy_setopt( curl_handle, CURLOPT_POSTFIELDS, "" );
}

std::string Curl::get( const std::string &url ) {
    // get rid of garbage
    cleanUp( curl_handle );
    std::string source;
    source.reserve( 100000 );
    curl_easy_setopt( curl_handle, CURLOPT_WRITEDATA,
                      static_cast< void * >( &source ) );
    curl_easy_setopt( curl_handle, CURLOPT_URL, url.c_str() );
    curl_easy_setopt( curl_handle, CURLOPT_HTTPGET, 1 );
    auto res = curl_easy_perform( curl_handle );
    if ( res != CURLE_OK ) {
        std::cerr << "curl_easy_perform() failed: " << curl_easy_strerror( res )
                  << std::endl;
        return "";
    }
    return source;
}

void Curl::addHeader( const std::string &header ) {
    chunk = curl_slist_append( chunk, header.c_str() );
    curl_easy_setopt( curl_handle, CURLOPT_HTTPHEADER, chunk );
}

void Curl::clearHeader() {
    curl_slist_free_all( chunk );
    chunk = NULL;
    curl_easy_setopt( curl_handle, CURLOPT_HTTPHEADER, chunk );
}

std::string Curl::post( const std::string &url, const std::string &data ) {
    std::string source;
    source.reserve( 100000 );
    curl_easy_setopt( curl_handle, CURLOPT_URL, url.c_str() );
    curl_easy_setopt( curl_handle, CURLOPT_POST, 1 );
    curl_easy_setopt( curl_handle, CURLOPT_POSTFIELDS, data.c_str() );
    curl_easy_setopt( curl_handle, CURLOPT_WRITEDATA,
                      static_cast< void * >( &source ) );
    auto res = curl_easy_perform( curl_handle );
    if ( res != CURLE_OK ) {
        std::cerr << "curl_easy_perform() failed: " << curl_easy_strerror( res )
                  << std::endl;
        return "";
    }
    return source;
}

#endif