Added AddonDialog and connected it into the whole non-blocking update stuff

[supertux.git] / src / addon / downloader.cpp

diff --git a/src/addon/downloader.cpp b/src/addon/downloader.cpp
index 7f795a5..f40a141 100644 (file)
--- a/src/addon/downloader.cpp
+++ b/src/addon/downloader.cpp
@@ -1,5 +1,6 @@
 //  SuperTux
-//  Copyright (C) 2014 Ingo Ruhnke <grumbel@gmail.com>
+//  Copyright (C) 2007 Christoph Sommer <christoph.sommer@2007.expires.deltadevelopment.de>
+//                2014 Ingo Ruhnke <grumbel@gmail.com>
 //
 //  This program is free software: you can redistribute it and/or modify
 //  it under the terms of the GNU General Public License as published by
@@ -16,10 +17,9 @@
 
 #include "addon/downloader.hpp"
 
-#include <curl/curl.h>
-#include <curl/easy.h>
-#include <physfs.h>
+#include <algorithm>
 #include <memory>
+#include <physfs.h>
 #include <stdexcept>
 
 #include "util/log.hpp"
@@ -46,13 +46,130 @@ size_t my_curl_physfs_write(void* ptr, size_t size, size_t nmemb, void* userdata
 
 } // namespace
 
-Downloader::Downloader()
+class Transfer
+{
+private:
+  Downloader& m_downloader;
+  TransferId m_id;
+
+  std::string m_url;
+  CURL* m_handle;
+  std::array<char, CURL_ERROR_SIZE> m_error_buffer;
+
+  TransferStatusPtr m_status;
+
+public:
+  Transfer(Downloader& downloader, TransferId id, const std::string& url) :
+    m_downloader(downloader),
+    m_id(id),
+    m_url(url),
+    m_handle(curl_easy_init()),
+    m_error_buffer(),
+    m_status(new TransferStatus(id))
+  {
+    if (!m_handle)
+    {
+      throw std::runtime_error("curl_easy_init() failed");
+    }
+    else
+    {
+      curl_easy_setopt(m_handle, CURLOPT_URL, url.c_str());
+      curl_easy_setopt(m_handle, CURLOPT_USERAGENT, "SuperTux/" PACKAGE_VERSION " libcURL");
+
+      curl_easy_setopt(m_handle, CURLOPT_WRITEDATA, this);
+      curl_easy_setopt(m_handle, CURLOPT_WRITEFUNCTION,
+                       [](void* ptr, size_t size, size_t nmemb, void* userdata) -> size_t
+                       {
+                         return static_cast<Transfer*>(userdata)
+                           ->on_data(ptr, size, nmemb);
+                       });
+
+      curl_easy_setopt(m_handle, CURLOPT_ERRORBUFFER, m_error_buffer.data());
+      curl_easy_setopt(m_handle, CURLOPT_NOSIGNAL, 1);
+      curl_easy_setopt(m_handle, CURLOPT_FAILONERROR, 1);
+      curl_easy_setopt(m_handle, CURLOPT_FOLLOWLOCATION, 1);
+
+      curl_easy_setopt(m_handle, CURLOPT_NOPROGRESS, 0);
+      curl_easy_setopt(m_handle, CURLOPT_XFERINFODATA, this);
+      curl_easy_setopt(m_handle, CURLOPT_XFERINFOFUNCTION,
+                       [](void* userdata,
+                          curl_off_t dltotal, curl_off_t dlnow,
+                          curl_off_t ultotal, curl_off_t ulnow)
+                       {
+                         return static_cast<Transfer*>(userdata)
+                           ->on_progress(dltotal, dlnow,
+                                         ultotal, ulnow);
+                       });
+    }
+  }
+
+  ~Transfer()
+  {
+    curl_easy_cleanup(m_handle);
+  }
+
+  TransferStatusPtr get_status() const
+  {
+    return m_status;
+  }
+
+  TransferId get_id() const
+  {
+    return m_id;
+  }
+
+  CURL* get_curl_handle() const
+  {
+    return m_handle;
+  }
+
+  std::string get_url() const
+  {
+    return m_url;
+  }
+
+  size_t on_data(void* ptr, size_t size, size_t nmemb)
+  {
+    return size * nmemb;;
+  }
+
+  void on_progress(curl_off_t dltotal, curl_off_t dlnow,
+                   curl_off_t ultotal, curl_off_t ulnow)
+  {
+    m_status->dltotal = dltotal;
+    m_status->dlnow = dlnow;
+
+    m_status->ultotal = ultotal;
+    m_status->ulnow = ulnow;
+  }
+
+private:
+  Transfer(const Transfer&) = delete;
+  Transfer& operator=(const Transfer&) = delete;
+};
+
+Downloader::Downloader() :
+  m_multi_handle(),
+  m_transfers(),
+  m_next_transfer_id(1)
 {
   curl_global_init(CURL_GLOBAL_ALL);
+  m_multi_handle = curl_multi_init();
+  if (!m_multi_handle)
+  {
+    throw std::runtime_error("curl_multi_init() failed");
+  }
 }
 
 Downloader::~Downloader()
 {
+  for(auto& transfer : m_transfers)
+  {
+    curl_multi_remove_handle(m_multi_handle, transfer->get_curl_handle());
+  }
+  m_transfers.clear();
+
+  curl_multi_cleanup(m_multi_handle);
   curl_global_cleanup();
 }
 
@@ -101,4 +218,62 @@ Downloader::download(const std::string& url, const std::string& filename)
   download(url, my_curl_physfs_write, fout.get());
 }
 
+void
+Downloader::abort(TransferId id)
+{
+  auto it = std::find_if(m_transfers.begin(), m_transfers.end(),
+                         [&id](const std::unique_ptr<Transfer>& rhs)
+                         {
+                           return id == rhs->get_id();
+                         });
+  if (it == m_transfers.end())
+  {
+    log_warning << "transfer not found: " << id << std::endl;
+  }
+  else
+  {
+    curl_multi_remove_handle(m_multi_handle, (*it)->get_curl_handle());
+    m_transfers.erase(it);
+  }
+}
+
+void
+Downloader::update()
+{
+  // read data from the network
+  CURLMcode ret;
+  int running_handles;
+  while((ret = curl_multi_perform(m_multi_handle, &running_handles)) == CURLM_CALL_MULTI_PERFORM)
+  {
+    log_debug << "updating" << std::endl;
+  }
+
+  // check if any downloads got finished
+  int msgs_in_queue;
+  CURLMsg* msg;
+  while ((msg = curl_multi_info_read(m_multi_handle, &msgs_in_queue)))
+  {
+    switch(msg->msg)
+    {
+      case CURLMSG_DONE:
+        curl_multi_remove_handle(m_multi_handle, msg->easy_handle);
+        //FIXME: finish_transfer(msg->easy_handle);
+        break;
+
+      default:
+        log_warning << "unhandled cURL message: " << msg->msg << std::endl;
+        break;
+    }
+  }
+}
+
+TransferStatusPtr
+Downloader::request_download(const std::string& url, const std::string& filename)
+{
+  std::unique_ptr<Transfer> transfer(new Transfer(*this, m_next_transfer_id++, url));
+  curl_multi_add_handle(m_multi_handle, transfer->get_curl_handle());
+  m_transfers.push_back(std::move(transfer));
+  return m_transfers.back()->get_status();
+}
+
 /* EOF */