#include "imagestore.h"

#include <curlcpp/curl_easy.h>

#include <chrono>
#include <fstream>
#include <sstream>
#include <thread>

Magick::Image imagestore::get(std::string key, std::string url) const {
  std::lock_guard lock(mutex_);

  std::string filename = path_ + "/" + key;

  Magick::Image pic;

  if (std::ifstream(filename)) {
    pic.read(filename);
  } else {
    // This is to avoid getting rate limited.
    std::this_thread::sleep_for(std::chrono::milliseconds(50));

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

void imagestore::cleanup() const {
  std::lock_guard lock(mutex_);

  for (const auto& entry : std::filesystem::directory_iterator(path_)) {
    std::filesystem::remove_all(entry);
  }
}