From 9fe6d0fe509295d3ef93e1007fbed475737808c6 Mon Sep 17 00:00:00 2001 From: dec05eba Date: Tue, 17 Aug 2021 02:35:19 +0200 Subject: Add related and recommendations page to AniList --- src/plugins/AniList.cpp | 307 +++++++++++++++++++++++++++++++++++++++++++++--- 1 file changed, 293 insertions(+), 14 deletions(-) (limited to 'src/plugins/AniList.cpp') diff --git a/src/plugins/AniList.cpp b/src/plugins/AniList.cpp index 5bf069c..c4286e0 100644 --- a/src/plugins/AniList.cpp +++ b/src/plugins/AniList.cpp @@ -75,6 +75,83 @@ query ($id: Int) { } )END"; + static const std::string related_query_graphql = R"END( +query ($id: Int) { + Media (id: $id) { + relations { + edges { + relationType (version: 2) + node { + id + type + format + status (version: 2) + startDate { + year + month + day + } + endDate { + year + month + day + } + averageScore + episodes + chapters + genres + coverImage { + medium + } + title { + romaji + } + synonyms + } + } + } + } +} + )END"; + + static const std::string recommendations_query_graphql = R"END( +query ($id: Int, $page: Int, $perPage: Int) { + Media (id: $id) { + recommendations (page: $page, perPage: $perPage) { + nodes { + mediaRecommendation { + id + type + format + status (version: 2) + startDate { + year + month + day + } + endDate { + year + month + day + } + averageScore + episodes + chapters + genres + coverImage { + medium + } + title { + romaji + } + synonyms + } + } + } + } +} + )END"; + enum class AniListMediaType { ANIME, MANGA @@ -325,6 +402,36 @@ query ($id: Int) { return body_item; } + static const char* media_relation_to_readable(const char *media_relation) { + if(strcmp(media_relation, "ADAPTATION") == 0) + return "Adaption"; + if(strcmp(media_relation, "PREQUEL") == 0) + return "Prequel"; + if(strcmp(media_relation, "SEQUEL") == 0) + return "Sequel"; + if(strcmp(media_relation, "PARENT") == 0) + return "Parent"; + if(strcmp(media_relation, "SIDE_STORY") == 0) + return "Side story"; + if(strcmp(media_relation, "CHARACTER") == 0) + return "Character"; + if(strcmp(media_relation, "SUMMARY") == 0) + return "Summary"; + if(strcmp(media_relation, "ALTERNATIVE") == 0) + return "Alternative"; + if(strcmp(media_relation, "SPIN_OFF") == 0) + return "Spin-off"; + if(strcmp(media_relation, "OTHER") == 0) + return "Other"; + if(strcmp(media_relation, "SOURCE") == 0) + return "Source"; + if(strcmp(media_relation, "COMPILATION") == 0) + return "Compilation"; + if(strcmp(media_relation, "CONTAINS") == 0) + return "Contains"; + return media_relation; + } + SearchResult AniListSearchPage::search_page(const std::string &str, int page, BodyItems &result_items) { Json::Value variables_json(Json::objectValue); variables_json["page"] = page; @@ -391,18 +498,19 @@ query ($id: Int) { manga_items.push_back(std::move(body_item)); } - if(anime_items.empty() && manga_items.empty()) - return SearchResult::OK; - - auto anime_title_item = BodyItem::create(""); - anime_title_item->set_author("------------------------ Anime ------------------------"); - result_items.push_back(std::move(anime_title_item)); - result_items.insert(result_items.end(), std::move_iterator(anime_items.begin()), std::move_iterator(anime_items.end())); + if(!anime_items.empty()) { + auto anime_title_item = BodyItem::create(""); + anime_title_item->set_author("------------------------ Anime ------------------------"); + result_items.push_back(std::move(anime_title_item)); + result_items.insert(result_items.end(), std::move_iterator(anime_items.begin()), std::move_iterator(anime_items.end())); + } - auto manga_title_item = BodyItem::create(""); - manga_title_item->set_author("------------------------ Manga ------------------------"); - result_items.push_back(std::move(manga_title_item)); - result_items.insert(result_items.end(), std::move_iterator(manga_items.begin()), std::move_iterator(manga_items.end())); + if(!manga_items.empty()) { + auto manga_title_item = BodyItem::create(""); + manga_title_item->set_author("------------------------ Manga ------------------------"); + result_items.push_back(std::move(manga_title_item)); + result_items.insert(result_items.end(), std::move_iterator(manga_items.begin()), std::move_iterator(manga_items.end())); + } return SearchResult::OK; } @@ -415,8 +523,8 @@ query ($id: Int) { if(url.empty()) return PluginResult::OK; - result_tabs.push_back({ create_body(), std::make_unique(program, url), nullptr }); result_tabs.push_back({ create_body(), std::make_unique(program, url), nullptr }); + result_tabs.push_back({ create_body(), std::make_unique(program, url), nullptr }); result_tabs.push_back({ create_body(), std::make_unique(program, url), nullptr }); return PluginResult::OK; } @@ -429,13 +537,94 @@ query ($id: Int) { if(url.empty()) return PluginResult::OK; - result_tabs.push_back({ create_body(), std::make_unique(program, url), nullptr }); result_tabs.push_back({ create_body(), std::make_unique(program, url), nullptr }); + result_tabs.push_back({ create_body(), std::make_unique(program, url), nullptr }); result_tabs.push_back({ create_body(), std::make_unique(program, url), nullptr }); return PluginResult::OK; } PluginResult AniListRelatedPage::lazy_fetch(BodyItems &result_items) { + Json::Value variables_json(Json::objectValue); + variables_json["id"] = atoi(id.c_str()); + + Json::Value request_json(Json::objectValue); + request_json["query"] = related_query_graphql; + request_json["variables"] = std::move(variables_json); + + Json::StreamWriterBuilder json_builder; + json_builder["commentStyle"] = "None"; + json_builder["indentation"] = ""; + + std::vector additional_args = { + { "-X", "POST" }, + { "-H", "content-type: application/json" }, + { "--data-binary", Json::writeString(json_builder, request_json) } + }; + + Json::Value json_root; + std::string err_msg; + DownloadResult download_result = download_json(json_root, "https://graphql.anilist.co", std::move(additional_args), true, &err_msg); + if(download_result != DownloadResult::OK) return download_result_to_plugin_result(download_result); + + if(!json_root.isObject()) + return PluginResult::ERR; + + const Json::Value &errors_json = json_root["errors"]; + if(errors_json.isArray() && errors_json.size() > 0) { + const Json::Value &error_json = errors_json[0]; + if(error_json.isObject()) { + const Json::Value &error_message_json = error_json["message"]; + if(error_message_json.isString()) { + show_notification("QuickMedia", "AniList search failed, error: " + error_message_json.asString(), Urgency::CRITICAL); + return PluginResult::ERR; + } + } + } + + const Json::Value &data_json = json_root["data"]; + if(!data_json.isObject()) + return PluginResult::ERR; + + const Json::Value &media_json = data_json["Media"]; + if(!media_json.isObject()) + return PluginResult::ERR; + + const Json::Value &relations_json = media_json["relations"]; + if(!relations_json.isObject()) + return PluginResult::ERR; + + const Json::Value &edges_json = relations_json["edges"]; + if(!edges_json.isArray()) + return PluginResult::ERR; + + std::map body_items_by_relation_type; + for(const Json::Value &edge_json : edges_json) { + if(!edge_json.isObject()) + continue; + + const Json::Value &relation_type_json = edge_json["relationType"]; + if(!relation_type_json.isString()) + continue; + + AniListMediaType media_type; + auto body_item = media_json_to_body_item(edge_json["node"], ThumbnailSize::MEDIUM, media_type); + if(!body_item) + continue; + + const char *relation_type_readable = media_relation_to_readable(relation_type_json.asCString()); + body_items_by_relation_type[relation_type_readable].push_back(std::move(body_item)); + } + + for(auto &it : body_items_by_relation_type) { + if(it.second.empty()) + continue; + + auto anime_title_item = BodyItem::create(""); + anime_title_item->set_author("------------------------ " + it.first + " ------------------------"); + result_items.push_back(std::move(anime_title_item)); + result_items.insert(result_items.end(), std::move_iterator(it.second.begin()), std::move_iterator(it.second.end())); + } + return PluginResult::OK; } @@ -502,13 +691,103 @@ query ($id: Int) { if(url.empty()) return PluginResult::OK; - result_tabs.push_back({ create_body(), std::make_unique(program, url), nullptr }); result_tabs.push_back({ create_body(), std::make_unique(program, url), nullptr }); + result_tabs.push_back({ create_body(), std::make_unique(program, url), nullptr }); result_tabs.push_back({ create_body(), std::make_unique(program, url), nullptr }); return PluginResult::OK; } PluginResult AniListRecommendationsPage::lazy_fetch(BodyItems &result_items) { + return get_page("", 0, result_items); + } + + PluginResult AniListRecommendationsPage::get_page(const std::string&, int page, BodyItems &result_items) { + Json::Value variables_json(Json::objectValue); + variables_json["page"] = 1 + page; + variables_json["perPage"] = 20; + variables_json["id"] = atoi(id.c_str()); + + Json::Value request_json(Json::objectValue); + request_json["query"] = recommendations_query_graphql; + request_json["variables"] = std::move(variables_json); + + Json::StreamWriterBuilder json_builder; + json_builder["commentStyle"] = "None"; + json_builder["indentation"] = ""; + + std::vector additional_args = { + { "-X", "POST" }, + { "-H", "content-type: application/json" }, + { "--data-binary", Json::writeString(json_builder, request_json) } + }; + + Json::Value json_root; + std::string err_msg; + DownloadResult download_result = download_json(json_root, "https://graphql.anilist.co", std::move(additional_args), true, &err_msg); + if(download_result != DownloadResult::OK) return download_result_to_plugin_result(download_result); + + if(!json_root.isObject()) + return PluginResult::ERR; + + const Json::Value &errors_json = json_root["errors"]; + if(errors_json.isArray() && errors_json.size() > 0) { + const Json::Value &error_json = errors_json[0]; + if(error_json.isObject()) { + const Json::Value &error_message_json = error_json["message"]; + if(error_message_json.isString()) { + show_notification("QuickMedia", "AniList search failed, error: " + error_message_json.asString(), Urgency::CRITICAL); + return PluginResult::ERR; + } + } + } + + const Json::Value &data_json = json_root["data"]; + if(!data_json.isObject()) + return PluginResult::ERR; + + const Json::Value &media_json = data_json["Media"]; + if(!media_json.isObject()) + return PluginResult::ERR; + + const Json::Value &recommendations_json = media_json["recommendations"]; + if(!recommendations_json.isObject()) + return PluginResult::ERR; + + const Json::Value &nodes_json = recommendations_json["nodes"]; + if(!nodes_json.isArray()) + return PluginResult::ERR; + + BodyItems anime_items; + BodyItems manga_items; + for(const Json::Value &node_json : nodes_json) { + if(!node_json.isObject()) + continue; + + AniListMediaType media_type; + auto body_item = media_json_to_body_item(node_json["mediaRecommendation"], ThumbnailSize::MEDIUM, media_type); + if(!body_item) + continue; + + if(media_type == AniListMediaType::ANIME) + anime_items.push_back(std::move(body_item)); + else if(media_type == AniListMediaType::MANGA) + manga_items.push_back(std::move(body_item)); + } + + if(!anime_items.empty()) { + auto anime_title_item = BodyItem::create(""); + anime_title_item->set_author("------------------------ Anime ------------------------"); + result_items.push_back(std::move(anime_title_item)); + result_items.insert(result_items.end(), std::move_iterator(anime_items.begin()), std::move_iterator(anime_items.end())); + } + + if(!manga_items.empty()) { + auto manga_title_item = BodyItem::create(""); + manga_title_item->set_author("------------------------ Manga ------------------------"); + result_items.push_back(std::move(manga_title_item)); + result_items.insert(result_items.end(), std::move_iterator(manga_items.begin()), std::move_iterator(manga_items.end())); + } + return PluginResult::OK; } } \ No newline at end of file -- cgit v1.2.3