1// SuperTux
2// Copyright (C) 2007 Christoph Sommer <christoph.sommer@2007.expires.deltadevelopment.de>
3// 2014 Ingo Ruhnke <grumbel@gmail.com>
4//
5// This program is free software: you can redistribute it and/or modify
6// it under the terms of the GNU General Public License as published by
7// the Free Software Foundation, either version 3 of the License, or
8// (at your option) any later version.
9//
10// This program is distributed in the hope that it will be useful,
11// but WITHOUT ANY WARRANTY; without even the implied warranty of
12// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
13// GNU General Public License for more details.
14//
15// You should have received a copy of the GNU General Public License
16// along with this program. If not, see <http://www.gnu.org/licenses/>.
17
18#ifndef HEADER_SUPERTUX_ADDON_DOWNLOADER_HPP
19#define HEADER_SUPERTUX_ADDON_DOWNLOADER_HPP
20
21#include <curl/curl.h>
22#include <curl/easy.h>
23#include <functional>
24#include <memory>
25#include <string>
26#include <vector>
27
28typedef int TransferId;
29class Downloader;
30
31class TransferStatus final
32{
33public:
34 Downloader& m_downloader;
35 TransferId id;
36 std::vector<std::function<void (bool)> > callbacks;
37
38 int dltotal;
39 int dlnow;
40 int ultotal;
41 int ulnow;
42
43 std::string error_msg;
44
45 TransferStatus(Downloader& downloader, TransferId id_) :
46 m_downloader(downloader),
47 id(id_),
48 callbacks(),
49 dltotal(0),
50 dlnow(0),
51 ultotal(0),
52 ulnow(0),
53 error_msg()
54 {}
55
56 void abort();
57 void update();
58
59 void then(const std::function<void (bool)>& callback)
60 {
61 callbacks.push_back(callback);
62 }
63};
64
65using TransferStatusPtr = std::shared_ptr<TransferStatus>;
66
67class Transfer;
68
69class Downloader final
70{
71private:
72 CURLM* m_multi_handle;
73 std::vector<std::unique_ptr<Transfer> > m_transfers;
74 int m_next_transfer_id;
75
76public:
77 Downloader();
78 ~Downloader();
79
80 /** Download \a url and return the result as string */
81 std::string download(const std::string& url);
82
83 /** Download \a url and store the result in \a filename */
84 void download(const std::string& url, const std::string& filename);
85
86 void download(const std::string& url,
87 size_t (*write_func)(void* ptr, size_t size, size_t nmemb, void* userdata),
88 void* userdata);
89
90 void update();
91
92 TransferStatusPtr request_download(const std::string& url, const std::string& filename);
93 void abort(TransferId id);
94
95private:
96 Downloader(const Downloader&) = delete;
97 Downloader& operator=(const Downloader&) = delete;
98};
99
100#endif
101
102/* EOF */
103