json-validator/include/jvalidate/compat/curl.h

#pragma once
#include <sstream>
#include <string_view>

#include <curl/curl.h>

#include <jvalidate/adapter.h>
#include <jvalidate/forward.h>
#include <jvalidate/uri.h>

namespace jvalidate {
inline size_t transfer_to_buffer(char * data, size_t size, size_t nmemb, void * userdata) {
  std::stringstream & ss = *reinterpret_cast<std::stringstream *>(userdata);
  size_t actual_size = size * nmemb;
  ss << std::string_view(data, actual_size);
  return actual_size;
}

template <typename JSON> bool curl_get(jvalidate::URI const & uri, JSON & out) {
  using jvalidate::adapter::load_file;
  using jvalidate::adapter::load_stream;
  if (uri.scheme().starts_with("http")) {
    std::stringstream ss;
    if (CURL * curl = curl_easy_init(); curl) {
      curl_easy_setopt(curl, CURLOPT_URL, uri.c_str());
      curl_easy_setopt(curl, CURLOPT_FOLLOWLOCATION, 1L);
      curl_easy_setopt(curl, CURLOPT_WRITEDATA, &ss);
      curl_easy_setopt(curl, CURLOPT_WRITEFUNCTION, &transfer_to_buffer);

      CURLcode res = curl_easy_perform(curl);
      curl_easy_cleanup(curl);

      if (res == CURLE_OK) {
        return load_stream(ss, out);
      }
    }
    return false;
  } else if (uri.scheme() == "file") {
    return load_file(uri.resource(), out);
  } else {
    return false;
  }
}
}