aboutsummaryrefslogtreecommitdiff
path: root/src/curl.cpp
blob: e141e7864767d8da98d42771141cf0a50045143e (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
#include "../include/curl.hpp"
#include "../include/env.hpp"
#include <curl/curl.h>
#include <cstring>

using namespace std;

#ifdef DEBUG
#define CURL_DEBUG
#endif

#undef CURL_DEBUG

class CurlSession
{
public:
    CurlSession()
    {
        curl_global_init(CURL_GLOBAL_ALL);
    }
};

static CurlSession curlSession;

namespace sibs
{
    // TODO: Instead of writing to file, reading from file and extracting it;
    // we can extract to file directly by putting libarchive code here
    size_t writeToFile(void *ptr, size_t size, size_t nmemb, void *stream)
    {
        return fwrite(ptr, size, nmemb, (FILE*)stream);
    }

    Result<bool> curl::downloadFile(const char *url, const char *filepath)
    {
        CURL *curl_handle = curl_easy_init();
        curl_easy_setopt(curl_handle, CURLOPT_URL, url);
#ifdef CURL_DEBUG
        long verbose = 1L;
        long noProgressMeter = 0L;
#else
        long verbose = 0L;
        long noProgressMeter = 1L;
#endif
        curl_easy_setopt(curl_handle, CURLOPT_VERBOSE, verbose);
        curl_easy_setopt(curl_handle, CURLOPT_NOPROGRESS, noProgressMeter);
        curl_easy_setopt(curl_handle, CURLOPT_WRITEFUNCTION, writeToFile);
        curl_easy_setopt(curl_handle, CURLOPT_FOLLOWLOCATION, true);

        FILE *file = fopen(filepath, "wb");
        if(!file)
        {
            int error = errno;
            curl_easy_cleanup(curl_handle);

            string errMsg = "Failed to open file for writing: ";
            errMsg += filepath;
            if(error != 0)
            {
                errMsg += "; Reason: ";
                errMsg += strerror(error);
                return Result<bool>::Err(errMsg);
            }
            else
            {
                return Result<bool>::Err(errMsg);
            }
        }

        curl_easy_setopt(curl_handle, CURLOPT_WRITEDATA, file);
        printf("Downloading from url: %s\n", url);
        CURLcode curlResponse = curl_easy_perform(curl_handle);

        long httpCode = 0;
        curl_easy_getinfo(curl_handle, CURLINFO_RESPONSE_CODE, &httpCode);
        curl_easy_cleanup(curl_handle);
        fclose(file);

        if(httpCode == 200 && curlResponse == CURLE_OK)
            return Result<bool>::Ok(true);

        if(httpCode != 200)
        {
            string errMsg = "Failed to download file from url: ";
            errMsg += url;
            errMsg += "\nReason: Expected http response code 200 (OK), got: ";
            errMsg += to_string(httpCode);
            return Result<bool>::Err(errMsg);
        }
        else
        {
            string errMsg = "Failed to download file from url: ";
            errMsg += url;
            errMsg += "\nReason: ";
            errMsg += curl_easy_strerror(curlResponse);
            return Result<bool>::Err(errMsg);
        }
    }
}