Mercurial > minori

comparison src/services/anilist.cc @ 258:862d0d8619f6

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
*: HUUUGE changes animia has been renamed to animone, so instead of thinking of a health condition, you think of a beautiful flower :) I've also edited some of the code for animone, but I have no idea if it even works or not because I don't have a mac or windows machine lying around. whoops! ... anyway, all of the changes divergent from Anisthesia are now licensed under BSD. it's possible that I could even rewrite most of the code to where I don't even have to keep the MIT license, but that's thinking too far into the future I've been slacking off on implementing the anime seasons page, mostly out of laziness. I think I'd have to create another db file specifically for the seasons anyway, this code is being pushed *primarily* because the hard drive it's on is failing! yay :)
author Paper <paper@paper.us.eu.org>
date Mon, 01 Apr 2024 02:43:44 -0400
parents c130f47f6f48
children dd211ff68b36
comparison
equal deleted inserted replaced
257:699a20c57dc8 258:862d0d8619f6
6 #include "core/json.h" 6 #include "core/json.h"
7 #include "core/session.h" 7 #include "core/session.h"
8 #include "core/strings.h" 8 #include "core/strings.h"
9 #include "gui/translate/anilist.h" 9 #include "gui/translate/anilist.h"
10 10
11 #include <QByteArray>
11 #include <QDate> 12 #include <QDate>
12 #include <QByteArray>
13 #include <QDesktopServices> 13 #include <QDesktopServices>
14 #include <QInputDialog> 14 #include <QInputDialog>
15 #include <QLineEdit> 15 #include <QLineEdit>
16 #include <QMessageBox> 16 #include <QMessageBox>
17 #include <QUrl> 17 #include <QUrl>
27 namespace AniList { 27 namespace AniList {
28 28
29 constexpr int CLIENT_ID = 13706; 29 constexpr int CLIENT_ID = 13706;
30 30
31 class Account { 31 class Account {
32 public: 32 public:
33 int UserId() const { return session.config.auth.anilist.user_id; } 33 int UserId() const { return session.config.auth.anilist.user_id; }
34 void SetUserId(const int id) { session.config.auth.anilist.user_id = id; } 34 void SetUserId(const int id) { session.config.auth.anilist.user_id = id; }
35 35
36 std::string AuthToken() const { return session.config.auth.anilist.auth_token; } 36 std::string AuthToken() const { return session.config.auth.anilist.auth_token; }
37 void SetAuthToken(std::string const& auth_token) { session.config.auth.anilist.auth_token = auth_token; } 37 void SetAuthToken(std::string const& auth_token) { session.config.auth.anilist.auth_token = auth_token; }
38 38
39 bool Authenticated() const { return !AuthToken().empty(); } 39 bool Authenticated() const { return !AuthToken().empty(); }
40 bool IsValid() const { return UserId() && Authenticated(); } 40 bool IsValid() const { return UserId() && Authenticated(); }
41 }; 41 };
42 42
43 static Account account; 43 static Account account;
44 44
45 std::string SendRequest(std::string data) { 45 std::string SendRequest(std::string data) {
46 std::vector<std::string> headers = {"Authorization: Bearer " + account.AuthToken(), "Accept: application/json", 46 std::vector<std::string> headers = {"Authorization: Bearer " + account.AuthToken(), "Accept: application/json",
47 "Content-Type: application/json"}; 47 "Content-Type: application/json"};
48 return Strings::ToUtf8String(HTTP::Post("https://graphql.anilist.co", data, headers)); 48 return Strings::ToUtf8String(HTTP::Post("https://graphql.anilist.co", data, headers));
49 } 49 }
50 50
51 nlohmann::json SendJSONRequest(nlohmann::json data) { 51 nlohmann::json SendJSONRequest(nlohmann::json data) {
52 std::string request = SendRequest(data.dump()); 52 std::string request = SendRequest(data.dump());
61 return {}; 61 return {};
62 } 62 }
63 63
64 if (ret.contains("/errors"_json_pointer) && ret.at("/errors"_json_pointer).is_array()) { 64 if (ret.contains("/errors"_json_pointer) && ret.at("/errors"_json_pointer).is_array()) {
65 for (const auto& error : ret.at("/errors"_json_pointer)) 65 for (const auto& error : ret.at("/errors"_json_pointer))
66 std::cerr << "[AniList] Received an error in response: " << JSON::GetString<std::string>(error, "/message"_json_pointer, "") << std::endl; 66 std::cerr << "[AniList] Received an error in response: "
67 << JSON::GetString<std::string>(error, "/message"_json_pointer, "") << std::endl;
67 68
68 return {}; 69 return {};
69 } 70 }
70 71
71 return ret; 72 return ret;
72 } 73 }
73 74
74 void ParseListStatus(std::string status, Anime::Anime& anime) { 75 void ParseListStatus(std::string status, Anime::Anime& anime) {
75 static const std::unordered_map<std::string, Anime::ListStatus> map = { 76 static const std::unordered_map<std::string, Anime::ListStatus> map = {
76 {"CURRENT", Anime::ListStatus::CURRENT }, 77 {"CURRENT", Anime::ListStatus::CURRENT },
77 {"PLANNING", Anime::ListStatus::PLANNING }, 78 {"PLANNING", Anime::ListStatus::PLANNING },
78 {"COMPLETED", Anime::ListStatus::COMPLETED}, 79 {"COMPLETED", Anime::ListStatus::COMPLETED},
79 {"DROPPED", Anime::ListStatus::DROPPED }, 80 {"DROPPED", Anime::ListStatus::DROPPED },
80 {"PAUSED", Anime::ListStatus::PAUSED } 81 {"PAUSED", Anime::ListStatus::PAUSED }
81 }; 82 };
82 83
83 if (status == "REPEATING") { 84 if (status == "REPEATING") {
84 anime.SetUserIsRewatching(true); 85 anime.SetUserIsRewatching(true);
85 anime.SetUserStatus(Anime::ListStatus::CURRENT); 86 anime.SetUserStatus(Anime::ListStatus::CURRENT);
86 return; 87 return;
125 ParseTitle(json.at("/title"_json_pointer), anime); 126 ParseTitle(json.at("/title"_json_pointer), anime);
126 127
127 anime.SetEpisodes(JSON::GetNumber(json, "/episodes"_json_pointer, 0)); 128 anime.SetEpisodes(JSON::GetNumber(json, "/episodes"_json_pointer, 0));
128 anime.SetFormat(Translate::AniList::ToSeriesFormat(JSON::GetString<std::string>(json, "/format"_json_pointer, ""))); 129 anime.SetFormat(Translate::AniList::ToSeriesFormat(JSON::GetString<std::string>(json, "/format"_json_pointer, "")));
129 130
130 anime.SetAiringStatus(Translate::AniList::ToSeriesStatus(JSON::GetString<std::string>(json, "/status"_json_pointer, ""))); 131 anime.SetAiringStatus(
132 Translate::AniList::ToSeriesStatus(JSON::GetString<std::string>(json, "/status"_json_pointer, "")));
131 133
132 anime.SetAirDate(Date(json["/startDate"_json_pointer])); 134 anime.SetAirDate(Date(json["/startDate"_json_pointer]));
133 135
134 anime.SetPosterUrl(JSON::GetString<std::string>(json, "/coverImage/large"_json_pointer, "")); 136 anime.SetPosterUrl(JSON::GetString<std::string>(json, "/coverImage/large"_json_pointer, ""));
135 137
177 return 0; 179 return 0;
178 } 180 }
179 181
180 /* NOTE: these really ought to be in the qrc file */ 182 /* NOTE: these really ought to be in the qrc file */
181 constexpr std::string_view query = "query ($id: Int) {\n" 183 constexpr std::string_view query = "query ($id: Int) {\n"
182 " MediaListCollection (userId: $id, type: ANIME) {\n" 184 " MediaListCollection (userId: $id, type: ANIME) {\n"
183 " lists {\n" 185 " lists {\n"
184 " name\n" 186 " name\n"
185 " entries {\n" 187 " entries {\n"
186 " score\n" 188 " score\n"
187 " notes\n" 189 " notes\n"
188 " status\n" 190 " status\n"
189 " progress\n" 191 " progress\n"
190 " startedAt {\n" 192 " startedAt {\n"
191 " year\n" 193 " year\n"
192 " month\n" 194 " month\n"
193 " day\n" 195 " day\n"
194 " }\n" 196 " }\n"
195 " completedAt {\n" 197 " completedAt {\n"
196 " year\n" 198 " year\n"
197 " month\n" 199 " month\n"
198 " day\n" 200 " day\n"
199 " }\n" 201 " }\n"
200 " updatedAt\n" 202 " updatedAt\n"
201 " media {\n" 203 " media {\n"
202 " coverImage {\n" 204 " coverImage {\n"
203 " large\n" 205 " large\n"
204 " }\n" 206 " }\n"
205 " id\n" 207 " id\n"
206 " title {\n" 208 " title {\n"
207 " romaji\n" 209 " romaji\n"
208 " english\n" 210 " english\n"
209 " native\n" 211 " native\n"
210 " }\n" 212 " }\n"
211 " format\n" 213 " format\n"
212 " status\n" 214 " status\n"
213 " averageScore\n" 215 " averageScore\n"
214 " season\n" 216 " season\n"
215 " startDate {\n" 217 " startDate {\n"
216 " year\n" 218 " year\n"
217 " month\n" 219 " month\n"
218 " day\n" 220 " day\n"
219 " }\n" 221 " }\n"
220 " genres\n" 222 " genres\n"
221 " episodes\n" 223 " episodes\n"
222 " duration\n" 224 " duration\n"
223 " synonyms\n" 225 " synonyms\n"
224 " description(asHtml: false)\n" 226 " description(asHtml: false)\n"
225 " }\n" 227 " }\n"
226 " }\n" 228 " }\n"
227 " }\n" 229 " }\n"
228 " }\n" 230 " }\n"
229 "}\n"; 231 "}\n";
230 // clang-format off 232 // clang-format off
231 nlohmann::json json = { 233 nlohmann::json json = {
232 {"query", query}, 234 {"query", query},
233 {"variables", { 235 {"variables", {
234 {"id", account.UserId()} 236 {"id", account.UserId()}
244 return 1; 246 return 1;
245 } 247 }
246 248
247 /* return is a vector of anime ids */ 249 /* return is a vector of anime ids */
248 std::vector<int> Search(const std::string& search) { 250 std::vector<int> Search(const std::string& search) {
249 constexpr std::string_view query = 251 constexpr std::string_view query = "query ($search: String) {\n"
250 "query ($search: String) {\n" 252 " Page (page: 1, perPage: 50) {\n"
251 " Page (page: 1, perPage: 50) {\n" 253 " media (search: $search, type: ANIME) {\n"
252 " media (search: $search, type: ANIME) {\n" 254 " coverImage {\n"
253 " coverImage {\n" 255 " large\n"
254 " large\n" 256 " }\n"
255 " }\n" 257 " id\n"
256 " id\n" 258 " title {\n"
257 " title {\n" 259 " romaji\n"
258 " romaji\n" 260 " english\n"
259 " english\n" 261 " native\n"
260 " native\n" 262 " }\n"
261 " }\n" 263 " format\n"
262 " format\n" 264 " status\n"
263 " status\n" 265 " averageScore\n"
264 " averageScore\n" 266 " season\n"
265 " season\n" 267 " startDate {\n"
266 " startDate {\n" 268 " year\n"
267 " year\n" 269 " month\n"
268 " month\n" 270 " day\n"
269 " day\n" 271 " }\n"
270 " }\n" 272 " genres\n"
271 " genres\n" 273 " episodes\n"
272 " episodes\n" 274 " duration\n"
273 " duration\n" 275 " synonyms\n"
274 " synonyms\n" 276 " description(asHtml: false)\n"
275 " description(asHtml: false)\n" 277 " }\n"
276 " }\n" 278 " }\n"
277 " }\n" 279 "}\n";
278 "}\n";
279 280
280 // clang-format off 281 // clang-format off
281 nlohmann::json json = { 282 nlohmann::json json = {
282 {"query", query}, 283 {"query", query},
283 {"variables", { 284 {"variables", {
306 * float score, 307 * float score,
307 * int scoreRaw, 308 * int scoreRaw,
308 * int progress, 309 * int progress,
309 * int progressVolumes, // manga-specific. 310 * int progressVolumes, // manga-specific.
310 * int repeat, // rewatch 311 * int repeat, // rewatch
311 * int priority, 312 * int priority,
312 * bool private, 313 * bool private,
313 * string notes, 314 * string notes,
314 * bool hiddenFromStatusLists, 315 * bool hiddenFromStatusLists,
315 * string[] customLists, 316 * string[] customLists,
316 * float[] advancedScores, 317 * float[] advancedScores,
317 * Date startedAt, 318 * Date startedAt,
318 * Date completedAt 319 * Date completedAt
319 **/ 320 **/
320 Anime::Anime& anime = Anime::db.items[id]; 321 Anime::Anime& anime = Anime::db.items[id];
321 if (!anime.IsInUserList()) 322 if (!anime.IsInUserList())
322 return 0; 323 return 0;
323 324
324 constexpr std::string_view query = 325 constexpr std::string_view query =
325 "mutation ($media_id: Int, $progress: Int, $status: MediaListStatus, $score: Int, $notes: String, $start: FuzzyDateInput, $comp: FuzzyDateInput, $repeat: Int) {\n" 326 "mutation ($media_id: Int, $progress: Int, $status: MediaListStatus, $score: Int, $notes: String, $start: "
326 " SaveMediaListEntry (mediaId: $media_id, progress: $progress, status: $status, scoreRaw: $score, notes: $notes, startedAt: $start, completedAt: $comp, repeat: $repeat) {\n" 327 "FuzzyDateInput, $comp: FuzzyDateInput, $repeat: Int) {\n"
327 " id\n" 328 " SaveMediaListEntry (mediaId: $media_id, progress: $progress, status: $status, scoreRaw: $score, notes: "
328 " }\n" 329 "$notes, startedAt: $start, completedAt: $comp, repeat: $repeat) {\n"
329 "}\n"; 330 " id\n"
331 " }\n"
332 "}\n";
330 // clang-format off 333 // clang-format off
331 nlohmann::json json = { 334 nlohmann::json json = {
332 {"query", query}, 335 {"query", query},
333 {"variables", { 336 {"variables", {
334 {"media_id", anime.GetId()}, 337 {"media_id", anime.GetId()},
353 return account.UserId(); 356 return account.UserId();
354 } 357 }
355 358
356 bool AuthorizeUser() { 359 bool AuthorizeUser() {
357 /* Prompt for PIN */ 360 /* Prompt for PIN */
358 QDesktopServices::openUrl( 361 QDesktopServices::openUrl(QUrl(Strings::ToQString("https://anilist.co/api/v2/oauth/authorize?client_id=" +
359 QUrl(Strings::ToQString("https://anilist.co/api/v2/oauth/authorize?client_id=" + Strings::ToUtf8String(CLIENT_ID) + "&response_type=token"))); 362 Strings::ToUtf8String(CLIENT_ID) + "&response_type=token")));
360 363
361 bool ok; 364 bool ok;
362 QString token = QInputDialog::getText( 365 QString token = QInputDialog::getText(
363 0, "Credentials needed!", "Please enter the code given to you after logging in to AniList:", QLineEdit::Normal, 366 0, "Credentials needed!", "Please enter the code given to you after logging in to AniList:", QLineEdit::Normal,
364 "", &ok); 367 "", &ok);
365 368
366 if (!ok || token.isEmpty()) 369 if (!ok || token.isEmpty())
367 return false; 370 return false;
368 371
369 account.SetAuthToken(Strings::ToUtf8String(token)); 372 account.SetAuthToken(Strings::ToUtf8String(token));
370 373
371 constexpr std::string_view query = "query {\n" 374 constexpr std::string_view query = "query {\n"
372 " Viewer {\n" 375 " Viewer {\n"
373 " id\n" 376 " id\n"
374 " name\n" 377 " name\n"
375 " mediaListOptions {\n" 378 " mediaListOptions {\n"
376 " scoreFormat\n" // this will be used... eventually 379 " scoreFormat\n" // this will be used... eventually
377 " }\n" 380 " }\n"
378 " }\n" 381 " }\n"
379 "}\n"; 382 "}\n";
380 nlohmann::json json = { 383 nlohmann::json json = {
381 {"query", query} 384 {"query", query}
382 }; 385 };
383 386
384 auto ret = SendJSONRequest(json); 387 auto ret = SendJSONRequest(json);
385 388
386 ParseUser(ret["data"]["Viewer"]); 389 ParseUser(ret["data"]["Viewer"]);
387 return true; 390 return true;