src/addon/downloader.hpp

   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
  28 typedef int TransferId;
  29 class Downloader;
  30
  31 class TransferStatus
  32 {
  33 public:
  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
  65 using TransferStatusPtr = std::shared_ptr<TransferStatus>;
  66
  67 class Transfer;
  68
  69 class Downloader
  70 {
  71 private:
  72   CURLM* m_multi_handle;
  73   std::vector<std::unique_ptr<Transfer> > m_transfers;
  74   int m_next_transfer_id;
  75
  76 public:
  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
  95 private:
  96   Downloader(const Downloader&) = delete;
  97   Downloader& operator=(const Downloader&) = delete;
  98 };
  99
 100 #endif
 101
 102 /* EOF */