imagestore.cpp


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

#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;
}