Mercurial > minori
annotate src/anilist.cpp @ 4:5af270662505
Set override functions as override
| author | Paper <mrpapersonic@gmail.com> |
|---|---|
| date | Sat, 12 Aug 2023 12:08:16 -0400 |
| parents | 190ded9438c0 |
| children | 1d82f6e04d7d |
| rev | line source |
|---|---|
| 2 | 1 #include "window.h" |
| 2 #include "json.h" | |
| 3 #include <curl/curl.h> | |
| 4 #include <chrono> | |
| 5 #include <exception> | |
| 6 #include <format> | |
| 7 #include "anilist.h" | |
| 8 #include "anime.h" | |
| 9 #include "config.h" | |
| 10 #include "string_utils.h" | |
| 11 #define CLIENT_ID "13706" | |
| 12 | |
| 13 size_t AniList::CurlWriteCallback(void *contents, size_t size, size_t nmemb, void *userdata) { | |
| 14 ((std::string*)userdata)->append((char*)contents, size * nmemb); | |
| 15 return size * nmemb; | |
| 16 } | |
| 17 | |
| 18 std::string AniList::SendRequest(std::string data) { | |
| 19 struct curl_slist *list = NULL; | |
| 20 std::string userdata; | |
| 21 curl = curl_easy_init(); | |
| 22 if (curl) { | |
| 23 list = curl_slist_append(list, "Accept: application/json"); | |
| 24 list = curl_slist_append(list, "Content-Type: application/json"); | |
| 25 std::string bearer = "Authorization: Bearer " + session.config.anilist.auth_token; | |
| 26 list = curl_slist_append(list, bearer.c_str()); | |
| 27 curl_easy_setopt(curl, CURLOPT_URL, "https://graphql.anilist.co"); | |
| 28 curl_easy_setopt(curl, CURLOPT_POSTFIELDS, data.c_str()); | |
| 29 curl_easy_setopt(curl, CURLOPT_HTTPHEADER, list); | |
| 30 curl_easy_setopt(curl, CURLOPT_WRITEDATA, &userdata); | |
| 31 curl_easy_setopt(curl, CURLOPT_WRITEFUNCTION, &CurlWriteCallback); | |
| 32 /* FIXME: This sucks. When using HTTPS, we should ALWAYS make sure that our peer | |
| 33 is actually valid. I assume the best way to go about this would be to bundle a | |
| 34 certificate file, and if it's not found we should *prompt the user* and ask them | |
| 35 if it's okay to contact AniList WITHOUT verification. If so, we're golden, and this | |
| 36 flag will be set. If not, we should abort mission. | |
| 37 | |
| 38 For this program, it's probably fine to just contact AniList without | |
| 39 HTTPS verification. However it should still be in the list of things to do... */ | |
|
4
5af270662505
Set override functions as override
Paper <mrpapersonic@gmail.com>
parents:
3
diff
changeset
|
40 curl_easy_setopt(curl, CURLOPT_SSL_VERIFYPEER, false); |
| 2 | 41 res = curl_easy_perform(curl); |
| 42 curl_slist_free_all(list); | |
| 43 if (res != CURLE_OK) { | |
| 44 QMessageBox box(QMessageBox::Icon::Critical, "", QString("curl_easy_perform(curl) failed!: ") + QString(curl_easy_strerror(res))); | |
| 45 box.exec(); | |
| 46 curl_easy_cleanup(curl); | |
| 47 return ""; | |
| 48 } | |
| 49 curl_easy_cleanup(curl); | |
| 50 return userdata; | |
| 51 } | |
| 52 return ""; | |
| 53 } | |
| 54 | |
| 55 int AniList::GetUserId(std::string name) { | |
| 56 #define QUERY "query ($name: String) {\n" \ | |
| 57 " User (name: $name) {\n" \ | |
| 58 " id\n" \ | |
| 59 " }\n" \ | |
| 60 "}\n" | |
| 61 nlohmann::json json = { | |
| 62 {"query", QUERY}, | |
| 63 {"variables", { | |
| 64 {"name", name} | |
| 65 }} | |
| 66 }; | |
| 67 auto ret = nlohmann::json::parse(SendRequest(json.dump())); | |
| 68 return ret["data"]["User"]["id"].get<int>(); | |
| 69 #undef QUERY | |
| 70 } | |
| 71 | |
| 72 /* Maps to convert string forms to our internal enums */ | |
| 73 | |
| 74 std::map<std::string, enum AnimeWatchingStatus> StringToAnimeWatchingMap = { | |
| 75 {"CURRENT", CURRENT}, | |
| 76 {"PLANNING", PLANNING}, | |
| 77 {"COMPLETED", COMPLETED}, | |
| 78 {"DROPPED", DROPPED}, | |
| 79 {"PAUSED", PAUSED}, | |
| 80 {"REPEATING", REPEATING} | |
| 81 }; | |
| 82 | |
| 83 std::map<std::string, enum AnimeAiringStatus> StringToAnimeAiringMap = { | |
| 84 {"FINISHED", FINISHED}, | |
| 85 {"RELEASING", RELEASING}, | |
| 86 {"NOT_YET_RELEASED", NOT_YET_RELEASED}, | |
| 87 {"CANCELLED", CANCELLED}, | |
| 88 {"HIATUS", HIATUS} | |
| 89 }; | |
| 90 | |
| 91 std::map<std::string, enum AnimeSeason> StringToAnimeSeasonMap = { | |
| 92 {"WINTER", WINTER}, | |
| 93 {"SPRING", SPRING}, | |
| 94 {"SUMMER", SUMMER}, | |
| 95 {"FALL", FALL} | |
| 96 }; | |
| 97 | |
| 98 std::map<std::string, enum AnimeFormat> StringToAnimeFormatMap = { | |
| 99 {"TV", TV}, | |
| 100 {"TV_SHORT", TV_SHORT}, | |
| 101 {"MOVIE", MOVIE}, | |
| 102 {"SPECIAL", SPECIAL}, | |
| 103 {"OVA", OVA}, | |
| 104 {"ONA", ONA}, | |
| 105 {"MUSIC", MUSIC}, | |
| 106 {"MANGA", MANGA}, | |
| 107 {"NOVEL", NOVEL}, | |
| 108 {"ONE_SHOT", ONE_SHOT} | |
| 109 }; | |
| 110 | |
| 111 int AniList::UpdateAnimeList(std::vector<AnimeList>* anime_lists, int id) { | |
| 112 /* NOTE: these should be in the qrc file */ | |
| 113 #define QUERY "query ($id: Int) {\n" \ | |
| 114 " MediaListCollection (userId: $id, type: ANIME) {\n" \ | |
| 115 " lists {\n" \ | |
| 116 " name\n" \ | |
| 117 " entries {\n" \ | |
| 118 " score\n" \ | |
| 119 " notes\n" \ | |
| 120 " progress\n" \ | |
| 121 " startedAt {\n" \ | |
| 122 " year\n" \ | |
| 123 " month\n" \ | |
| 124 " day\n" \ | |
| 125 " }\n" \ | |
| 126 " completedAt {\n" \ | |
| 127 " year\n" \ | |
| 128 " month\n" \ | |
| 129 " day\n" \ | |
| 130 " }\n" \ | |
| 131 " updatedAt\n" \ | |
| 132 " media {\n" \ | |
| 133 " id\n" \ | |
| 134 " title {\n" \ | |
| 135 " romaji\n" \ | |
| 136 " english\n" \ | |
| 137 " native\n" \ | |
| 138 " }\n" \ | |
| 139 " format\n" \ | |
| 140 " status\n" \ | |
| 141 " averageScore\n" \ | |
| 142 " season\n" \ | |
| 143 " startDate {\n" \ | |
| 144 " year\n" \ | |
| 145 " month\n" \ | |
| 146 " day\n" \ | |
| 147 " }\n" \ | |
| 148 " genres\n" \ | |
| 149 " episodes\n" \ | |
| 150 " duration\n" \ | |
| 151 " synonyms\n" \ | |
| 152 " description(asHtml: false)\n" \ | |
| 153 " }\n" \ | |
| 154 " }\n" \ | |
| 155 " }\n" \ | |
| 156 " }\n" \ | |
| 157 "}\n" | |
| 158 nlohmann::json json = { | |
| 159 {"query", QUERY}, | |
| 160 {"variables", { | |
| 161 {"id", id} | |
| 162 }} | |
| 163 }; | |
| 164 /* TODO: do a try catch here, catch any json errors and then call | |
| 165 Authorize() if needed */ | |
| 166 auto res = nlohmann::json::parse(SendRequest(json.dump())); | |
| 167 /* TODO: make sure that we actually need the wstring converter and see | |
| 168 if we can just get wide strings back from nlohmann::json */ | |
| 169 for (const auto& list : res["data"]["MediaListCollection"]["lists"].items()) { | |
| 170 /* why are the .key() values strings?? */ | |
| 171 AnimeList anime_list; | |
| 172 anime_list.name = StringUtils::Utf8ToWstr(JSON::GetString(list.value(), "name")); | |
| 173 for (const auto& entry : list.value()["entries"].items()) { | |
| 174 Anime anime; | |
| 175 anime.score = JSON::GetInt(entry.value(), "score"); | |
| 176 anime.progress = JSON::GetInt(entry.value(), "progress"); | |
| 177 anime.status = StringToAnimeWatchingMap[JSON::GetString(entry.value(), "status")]; | |
| 178 anime.notes = StringUtils::Utf8ToWstr(JSON::GetString(entry.value(), "notes")); | |
| 179 | |
| 180 anime.started.SetYear(JSON::GetInt(entry.value()["startedAt"], "year")); | |
| 181 anime.started.SetMonth(JSON::GetInt(entry.value()["startedAt"], "month")); | |
| 182 anime.started.SetDay(JSON::GetInt(entry.value()["startedAt"], "day")); | |
| 183 | |
| 184 anime.completed.SetYear(JSON::GetInt(entry.value()["completedAt"], "year")); | |
| 185 anime.completed.SetMonth(JSON::GetInt(entry.value()["completedAt"], "month")); | |
| 186 anime.completed.SetDay(JSON::GetInt(entry.value()["completedAt"], "day")); | |
| 187 | |
| 188 anime.updated = JSON::GetInt(entry.value(), "updatedAt"); | |
| 189 | |
| 190 anime.title.native = StringUtils::Utf8ToWstr(JSON::GetString(entry.value()["media"]["title"], "native")); | |
| 191 anime.title.english = StringUtils::Utf8ToWstr(JSON::GetString(entry.value()["media"]["title"], "english")); | |
| 192 anime.title.romaji = StringUtils::Utf8ToWstr(JSON::GetString(entry.value()["media"]["title"], "romaji")); | |
| 193 | |
| 194 anime.id = JSON::GetInt(entry.value()["media"], "id"); | |
| 195 anime.episodes = JSON::GetInt(entry.value()["media"], "episodes"); | |
| 196 anime.type = StringToAnimeFormatMap[JSON::GetString(entry.value()["media"], "format")]; | |
| 197 | |
| 198 anime.airing = StringToAnimeAiringMap[JSON::GetString(entry.value()["media"], "status")]; | |
| 199 | |
| 200 anime.air_date.SetYear(JSON::GetInt(entry.value()["media"]["startDate"], "year")); | |
| 201 anime.air_date.SetMonth(JSON::GetInt(entry.value()["media"]["startDate"], "month")); | |
| 202 anime.air_date.SetDay(JSON::GetInt(entry.value()["media"]["startDate"], "day")); | |
| 203 | |
| 204 anime.audience_score = JSON::GetInt(entry.value()["media"], "averageScore"); | |
| 205 anime.season = StringToAnimeSeasonMap[JSON::GetString(entry.value()["media"], "season")]; | |
| 206 anime.duration = JSON::GetInt(entry.value()["media"], "duration"); | |
| 207 anime.synopsis = StringUtils::TextifySynopsis(StringUtils::Utf8ToWstr(JSON::GetString(entry.value()["media"], "duration"))); | |
| 208 | |
| 209 if (entry.value()["media"]["genres"].is_array()) | |
| 210 anime.genres = entry.value()["media"]["genres"].get<std::vector<std::string>>(); | |
| 211 anime_list.Add(anime); | |
| 212 } | |
| 213 anime_lists->push_back(anime_list); | |
| 214 } | |
| 215 return 1; | |
| 216 #undef QUERY | |
| 217 } | |
| 218 | |
| 219 int AniList::Authorize() { | |
| 220 if (session.config.anilist.auth_token.empty()) { | |
| 221 /* Prompt for PIN */ | |
| 222 QDesktopServices::openUrl(QUrl("https://anilist.co/api/v2/oauth/authorize?client_id=" CLIENT_ID "&response_type=token")); | |
| 223 bool ok; | |
| 224 QString token = QInputDialog::getText(0, "Credentials needed!", "Please enter the code given to you after logging in to AniList:", QLineEdit::Normal, "", &ok); | |
| 225 if (ok && !token.isEmpty()) { | |
| 226 session.config.anilist.auth_token = token.toStdString(); | |
| 227 } else { // fail | |
| 228 return 0; | |
| 229 } | |
| 230 } | |
| 231 return 1; | |
| 232 } |