1 files changed, 52 insertions, 0 deletions

@@ -0,0 +1,52 @@
+#include "imagestore.h"
+
+#include <curlcpp/curl_easy.h>
+
+#include <fstream>
+#include <sstream>
+
+Magick::Image imagestore::get(std::string key, std::string url) const {
+  std::string filename = path_ + "/" + key;
+
+  Magick::Image pic;
+
+  if (std::ifstream(filename)) {
+    pic.read(filename);
+  } else {
+    std::ostringstream imgbuf;
+    curl::curl_ios<std::ostringstream> imgios(imgbuf);
+    curl::curl_easy imghandle(imgios);
+
+    imghandle.add<CURLOPT_URL>(url.c_str());
+    imghandle.add<CURLOPT_CONNECTTIMEOUT>(30);
+    imghandle.add<CURLOPT_TIMEOUT>(300);
+
+    imghandle.perform();
+
+    if (imghandle.get_info<CURLINFO_RESPONSE_CODE>().get() != 200) {
+      throw std::runtime_error("Could not download image");
+    }
+
+    std::string content_type =
+        imghandle.get_info<CURLINFO_CONTENT_TYPE>().get();
+    if (content_type.substr(0, 6) != "image/") {
+      throw std::runtime_error("Could not download image");
+    }
+
+    std::string imgstr = imgbuf.str();
+    Magick::Blob img(imgstr.c_str(), imgstr.length());
+
+    try {
+      pic.read(img);
+    } catch (const Magick::ErrorOption& e) {
+      // Occurs when the the data downloaded from the server is malformed
+      std::cout << "Magick: " << e.what() << std::endl;
+
+      throw std::runtime_error("Could not download image");
+    }
+
+    pic.write(filename);
+  }
+
+  return pic;
+}