193 lines
5.0 KiB
C++
193 lines
5.0 KiB
C++
#include "datadragon.h"
|
|
|
|
#include <ostream>
|
|
#include <stdexcept>
|
|
|
|
#include <curl/easy.h>
|
|
#include <Log.h>
|
|
|
|
#include <QJsonArray>
|
|
#include <QJsonObject>
|
|
|
|
|
|
static const std::string BASEURL = "https://ddragon.leagueoflegends.com/";
|
|
|
|
static size_t arrayWriteCallback(char* contents, size_t size, size_t nmemb, void* userdata) {
|
|
if(userdata) {
|
|
QByteArray* arr = (QByteArray*) userdata;
|
|
arr->append(contents, size * nmemb);
|
|
return size * nmemb;
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
|
|
template<typename T>
|
|
T convert(const QJsonValue& val);
|
|
|
|
template<>
|
|
int convert(const QJsonValue& val) {
|
|
if(val.isString())
|
|
return val.toString().toInt();
|
|
return val.toInt();
|
|
}
|
|
|
|
template<>
|
|
std::string convert(const QJsonValue& val) {
|
|
return val.toString().toStdString();
|
|
}
|
|
|
|
template<typename T>
|
|
static T getValue(const QJsonObject& obj, const char* key, const T& def) {
|
|
auto it = obj.constFind(key);
|
|
if(it != obj.constEnd()) {
|
|
return convert<T>(it.value());
|
|
}
|
|
return def;
|
|
}
|
|
|
|
DataDragon::DataDragon() {
|
|
curl = curl_easy_init();
|
|
curl_easy_setopt(curl, CURLOPT_FOLLOWLOCATION, 1L);
|
|
curl_easy_setopt(curl, CURLOPT_WRITEFUNCTION, arrayWriteCallback);
|
|
}
|
|
|
|
DataDragon::ChampData::ChampData(const QJsonObject& source) {
|
|
name = getValue<std::string>(source, "name", "");
|
|
id = getValue<std::string>(source, "id", "");
|
|
key = getValue<int>(source, "key", -1);
|
|
partype = getValue<std::string>(source, "partype", "");
|
|
title = getValue<std::string>(source, "title", "");
|
|
}
|
|
|
|
const std::string& DataDragon::getVersion() {
|
|
if(!version.empty()) {
|
|
return version;
|
|
}
|
|
|
|
QJsonDocument jversions = request("api/versions.json");
|
|
if(jversions.isArray()) {
|
|
QJsonArray jverarr = jversions.array();
|
|
if(!jverarr.empty()) {
|
|
version = jverarr.at(0).toString().toStdString();
|
|
Log::info << "got League version: " << version;
|
|
return version;
|
|
}
|
|
}
|
|
|
|
Log::error << "error parsing version object";
|
|
// empty version str
|
|
return version;
|
|
}
|
|
|
|
const std::vector<DataDragon::ChampData>& DataDragon::getChamps() {
|
|
if(!champs.empty()) {
|
|
return champs;
|
|
}
|
|
if(getVersion().empty()) {
|
|
return champs;
|
|
}
|
|
|
|
QJsonDocument jchamps = request(getCDNString() + "data/en_US/champion.json");
|
|
if(jchamps.isObject()) {
|
|
QJsonObject obj = jchamps.object();
|
|
auto it = obj.constFind("data");
|
|
if(it != obj.constEnd() && it.value().isObject()) {
|
|
QJsonObject jchampsdata = it.value().toObject();
|
|
for(auto champit = jchampsdata.constBegin(); champit != jchampsdata.constEnd(); champit++) {
|
|
if(champit.value().isObject()) {
|
|
champs.emplace_back(champit.value().toObject());
|
|
}
|
|
}
|
|
}
|
|
}
|
|
Log::info << "loaded " << champs.size() << " champs";
|
|
|
|
return champs;
|
|
}
|
|
|
|
cv::Mat DataDragon::getImage(const std::string& champid, ImageType imgtype) {
|
|
std::string url;
|
|
if(imgtype == ImageType::SQUARE) {
|
|
if(getVersion().empty()) {
|
|
return {};
|
|
}
|
|
|
|
// with version
|
|
url = getCDNString() + "img/champion/" + champid + ".png";
|
|
} else if(imgtype == ImageType::LOADING) {
|
|
// no version
|
|
url = "cdn/img/champion/loading/" + champid + "_0.jpg";
|
|
} else if(imgtype == ImageType::SPLASH) {
|
|
// no version!
|
|
url = "cdn/img/champion/splash/" + champid + "_0.jpg";
|
|
}
|
|
|
|
QByteArray arr = requestRaw(url);
|
|
if(arr.isEmpty()) {
|
|
Log::error << "image could not be loaded";
|
|
return {};
|
|
}
|
|
|
|
cv::Mat rawData(1, arr.size(), CV_8UC1, (void*) arr.data());
|
|
|
|
// propably an error
|
|
if(arr.size() < 1000) {
|
|
Log::info << "small icon url: " << getCDNString() << "img/champion/" << champid << ".png";
|
|
Log::info << "content: " << std::string(arr.data(), arr.size());
|
|
return {};
|
|
}
|
|
|
|
cv::Mat decodedImage = cv::imdecode(rawData, cv::IMREAD_COLOR);
|
|
return decodedImage;
|
|
}
|
|
|
|
std::string DataDragon::getCDNString() const {
|
|
return "cdn/" + version + "/";
|
|
}
|
|
|
|
QByteArray DataDragon::requestRaw(const std::string& url) {
|
|
if(!curl) return {};
|
|
|
|
std::string requrl = BASEURL + url;
|
|
QByteArray ba; //buffer
|
|
// std::cout << "[DEBUG] requrl is: " << requrl << std::endl;
|
|
curl_easy_setopt(curl, CURLOPT_URL, requrl.c_str());
|
|
|
|
// curl_easy_setopt(curl, CURLOPT_NOSIGNAL, 1); //Prevent "longjmp causes uninitialized stack frame" bug
|
|
// set callback data
|
|
curl_easy_setopt(curl, CURLOPT_WRITEDATA, &ba);
|
|
|
|
// Check for errors
|
|
CURLcode res = curl_easy_perform(curl);
|
|
if (res != CURLE_OK) {
|
|
if(res == CURLE_HTTP_RETURNED_ERROR) {
|
|
long responsecode = -1;
|
|
curl_easy_getinfo(curl, CURLINFO_RESPONSE_CODE, &responsecode);
|
|
Log::warn << "DataDragon request failed: " << url << " -> " << responsecode ;
|
|
} else {
|
|
Log::warn << "DataDragon request failed: " << url << " " << curl_easy_strerror(res);
|
|
}
|
|
return {};
|
|
}
|
|
return ba;
|
|
}
|
|
|
|
QJsonDocument DataDragon::request(const std::string& url) {
|
|
QByteArray arr = requestRaw(url);
|
|
if(arr.isEmpty()) return {};
|
|
|
|
QJsonParseError err;
|
|
QJsonDocument parsed = QJsonDocument::fromJson(arr, &err);
|
|
if(parsed.isNull() || err.error != QJsonParseError::NoError) {
|
|
Log::error << "DataDragon Jsonparse error " << err.errorString().toStdString() << " offset: " << err.offset;
|
|
return {};
|
|
}
|
|
|
|
return parsed;
|
|
}
|
|
|
|
std::ostream& operator<<(std::ostream& str, const DataDragon::ChampData& cd) {
|
|
return str << "[n: " << cd.name << " " << " k: " << cd.key << " id: " << cd.id << "]";
|
|
}
|