HummingBirdAnimeClient/src/AnimeClient/API/Kitsu/Model.php

828 lines
18 KiB
PHP
Raw Normal View History

2016-12-21 12:46:20 -05:00
<?php declare(strict_types=1);
/**
2017-02-15 16:13:32 -05:00
* Hummingbird Anime List Client
2016-12-21 12:46:20 -05:00
*
2018-08-22 13:48:27 -04:00
* An API client for Kitsu to manage anime and manga watch lists
2016-12-21 12:46:20 -05:00
*
2021-02-04 11:57:01 -05:00
* PHP version 8
2016-12-21 12:46:20 -05:00
*
2022-03-04 15:50:35 -05:00
* @copyright 2015 - 2022 Timothy J. Warren <tim@timshome.page>
2017-01-06 23:34:56 -05:00
* @license http://www.opensource.org/licenses/mit-license.html MIT License
2020-12-10 17:06:50 -05:00
* @version 5.2
2022-03-04 15:50:35 -05:00
* @link https://git.timshome.page/timw4mail/HummingBirdAnimeClient
*/
namespace Aviat\AnimeClient\API\Kitsu;
2016-12-21 12:46:20 -05:00
2020-10-09 16:18:45 -04:00
use Amp;
2017-01-03 21:06:49 -05:00
use Aviat\AnimeClient\API\Kitsu\Transformer\{
AnimeHistoryTransformer,
AnimeListTransformer,
2017-02-04 15:18:34 -05:00
AnimeTransformer,
2020-08-24 13:07:47 -04:00
LibraryEntryTransformer,
MangaHistoryTransformer,
MangaListTransformer,
MangaTransformer
2017-01-03 21:06:49 -05:00
};
use Aviat\AnimeClient\API\{
CacheTrait,
Enum\AnimeWatchingStatus\Kitsu as KitsuWatchingStatus,
Enum\MangaReadingStatus\Kitsu as KitsuReadingStatus,
Mapping\AnimeWatchingStatus,
Mapping\MangaReadingStatus
};
2020-12-10 15:59:37 -05:00
use Aviat\AnimeClient\Enum\MediaType;
use Aviat\AnimeClient\Kitsu as K;
use Aviat\AnimeClient\Types\{Anime, MangaPage};
2023-05-09 12:49:36 -04:00
use Aviat\AnimeClient\Types\{AnimeListItem, MangaListItem};
use Aviat\Ion\{
Di\ContainerAware,
Json
};
2021-02-12 13:09:57 -05:00
use Generator;
use function Amp\Promise\wait;
use function Aviat\AnimeClient\getApiClient;
use const Aviat\AnimeClient\SESSION_SEGMENT;
2019-12-09 14:34:23 -05:00
2016-12-21 12:46:20 -05:00
/**
* Kitsu API Model
*/
final class Model
{
use CacheTrait;
use ContainerAware;
2020-08-06 09:39:12 -04:00
use RequestBuilderTrait;
use MutationTrait;
2016-12-21 12:46:20 -05:00
2020-10-09 16:18:45 -04:00
protected const LIST_PAGE_SIZE = 100;
protected AnimeTransformer $animeTransformer;
protected MangaTransformer $mangaTransformer;
/**
2017-02-17 11:37:22 -05:00
* Constructor
*/
public function __construct(protected ListItem $listItem)
2016-12-21 12:46:20 -05:00
{
$this->animeTransformer = new AnimeTransformer();
2017-01-04 13:16:58 -05:00
$this->mangaTransformer = new MangaTransformer();
2016-12-21 12:46:20 -05:00
}
2017-03-28 14:34:33 -04:00
/**
* Get the access token from the Kitsu API
*/
2021-02-16 14:43:51 -05:00
public function authenticate(string $username, string $password): array|false
2017-03-28 14:34:33 -04:00
{
// K::AUTH_URL
$response = $this->requestBuilder->getResponse('POST', K::AUTH_URL, [
'headers' => [
'accept' => NULL,
'Content-type' => 'application/x-www-form-urlencoded',
'client_id' => NULL,
'client_secret' => NULL,
],
2017-03-28 14:34:33 -04:00
'form_params' => [
'grant_type' => 'password',
'username' => $username,
'password' => $password,
],
2017-03-28 14:34:33 -04:00
]);
2020-03-11 16:26:17 -04:00
$data = Json::decode(wait($response->getBody()->buffer()));
2018-01-16 14:58:07 -05:00
if (array_key_exists('error', $data))
{
2020-08-17 10:23:32 -04:00
dump([
'method' => self::class . '\\' . __METHOD__,
2020-08-17 10:23:32 -04:00
'error' => $data['error'],
'response' => $response,
]);
exit();
}
2018-10-05 14:32:05 -04:00
if (array_key_exists('access_token', $data))
{
return $data;
}
2017-03-28 14:34:33 -04:00
return FALSE;
}
/**
* Extend the current session with a refresh token
*/
2021-02-16 14:43:51 -05:00
public function reAuthenticate(string $token): array|false
{
$response = $this->requestBuilder->getResponse('POST', K::AUTH_URL, [
'headers' => [
'accept' => NULL,
'Content-type' => 'application/x-www-form-urlencoded',
'Accept-encoding' => '*',
],
'form_params' => [
'grant_type' => 'refresh_token',
'refresh_token' => $token,
],
]);
2020-03-11 16:26:17 -04:00
$data = Json::decode(wait($response->getBody()->buffer()));
if (array_key_exists('error', $data))
{
2020-08-17 10:23:32 -04:00
dump([
'method' => self::class . '\\' . __METHOD__,
2020-08-17 10:23:32 -04:00
'error' => $data['error'],
'response' => $response,
]);
exit();
}
if (array_key_exists('access_token', $data))
{
return $data;
}
return FALSE;
}
/**
* Get the userid for a username from Kitsu
*/
public function getUserIdByUsername(?string $username = NULL): string
{
2018-02-02 09:50:58 -05:00
if ($username === NULL)
2017-01-27 12:35:28 -05:00
{
$username = $this->getUsername();
}
2017-02-04 15:18:34 -05:00
return $this->getCached(K::AUTH_USER_ID_KEY, function (string $username) {
$data = $this->requestBuilder->runQuery('GetUserId', [
'slug' => $username,
]);
return $data['data']['findProfileBySlug']['id'] ?? NULL;
2020-05-08 19:15:21 -04:00
}, [$username]);
}
2017-03-08 12:55:49 -05:00
/**
* Get information about a character
*
* @return mixed[]
2017-03-08 12:55:49 -05:00
*/
public function getCharacter(string $slug): array
{
return $this->requestBuilder->runQuery('CharacterDetails', [
'slug' => $slug,
2017-03-08 12:55:49 -05:00
]);
}
/**
* Get information about a person
*
* @return mixed[]
*/
public function getPerson(string $slug): array
{
return $this->getCached("kitsu-person-{$slug}", fn () => $this->requestBuilder->runQuery('PersonDetails', [
'slug' => $slug,
2020-05-08 19:18:10 -04:00
]));
}
2017-03-08 13:46:50 -05:00
/**
* Get profile information for the configured user
*
* @return mixed[]
2017-03-08 13:46:50 -05:00
*/
2017-03-08 12:55:49 -05:00
public function getUserData(string $username): array
{
return $this->requestBuilder->runQuery('UserDetails', [
'slug' => $username,
2017-03-08 12:55:49 -05:00
]);
}
// -------------------------------------------------------------------------
// ! Anime-specific methods
// -------------------------------------------------------------------------
/**
* Get information about a particular anime
*/
public function getAnime(string $slug): Anime
{
$baseData = $this->requestBuilder->runQuery('AnimeDetails', [
'slug' => $slug,
]);
if (empty($baseData))
{
return Anime::from([]);
}
return $this->animeTransformer->transform($baseData);
}
public function getRandomAnime(): Anime
{
$baseData = $this->requestBuilder->runQuery('RandomMedia', [
'type' => 'ANIME',
]);
return $this->animeTransformer->transform($baseData);
}
2020-12-10 17:06:50 -05:00
public function getRandomLibraryAnime(string $status): Anime
{
// @TODO
return Anime::from([]);
}
/**
* Get information about a particular anime
*/
public function getAnimeById(string $animeId): Anime
{
$baseData = $this->requestBuilder->runQuery('AnimeDetailsById', [
'id' => $animeId,
]);
return $this->animeTransformer->transform($baseData);
}
/**
* Retrieve the data for the anime watch history page
*
* @return mixed[]
*/
public function getAnimeHistory(): array
{
$key = K::ANIME_HISTORY_LIST_CACHE_KEY;
$list = $this->cache->get($key, NULL);
if ($list === NULL)
{
$raw = $this->getHistoryList();
$list = (new AnimeHistoryTransformer())->transform($raw);
$this->cache->set($key, $list);
}
return $list;
}
/**
* Get the anime list for the configured user
*
* @param string $status - The watching status to filter the list with
* @return mixed[]
*/
public function getAnimeList(string $status): array
{
$key = "kitsu-anime-list-{$status}";
$list = $this->cache->get($key, NULL);
if ($list === NULL)
{
$data = $this->getList(MediaType::ANIME, $status);
// Bail out on no data
if (empty($data))
{
return [];
}
$transformer = new AnimeListTransformer();
$transformed = $transformer->transformCollection($data);
$keyed = [];
foreach ($transformed as $item)
{
$keyed[$item['id']] = $item;
}
$list = $keyed;
$this->cache->set($key, $list);
}
return $list;
}
/**
* Get the number of anime list items
*
* @param string $status - Optional status to filter by
*/
public function getAnimeListCount(string $status = ''): int
{
2020-12-10 15:59:37 -05:00
return $this->getListCount(MediaType::ANIME, $status);
}
/**
* Get all the anime entries, that are organized for output to html
*
* @return array<string, array>
*/
public function getFullOrganizedAnimeList(): array
{
$output = [];
$statuses = KitsuWatchingStatus::getConstList();
foreach ($statuses as $status)
{
$mappedStatus = AnimeWatchingStatus::KITSU_TO_TITLE[$status];
$output[$mappedStatus] = $this->getAnimeList($status);
}
return $output;
}
// -------------------------------------------------------------------------
// ! Manga-specific methods
// -------------------------------------------------------------------------
/**
* Get information about a particular manga
*/
public function getManga(string $slug): MangaPage
{
$baseData = $this->requestBuilder->runQuery('MangaDetails', [
'slug' => $slug,
]);
if (empty($baseData))
{
return MangaPage::from([]);
}
return $this->mangaTransformer->transform($baseData);
}
public function getRandomManga(): MangaPage
{
$baseData = $this->requestBuilder->runQuery('RandomMedia', [
'type' => 'MANGA',
]);
return $this->mangaTransformer->transform($baseData);
}
/**
* Get information about a particular manga
*/
public function getMangaById(string $mangaId): MangaPage
{
$baseData = $this->requestBuilder->runQuery('MangaDetailsById', [
'id' => $mangaId,
]);
return $this->mangaTransformer->transform($baseData);
}
/**
* Retrieve the data for the manga read history page
*
* @return mixed[]
*/
public function getMangaHistory(): array
{
$key = K::MANGA_HISTORY_LIST_CACHE_KEY;
$list = $this->cache->get($key, NULL);
if ($list === NULL)
{
$raw = $this->getHistoryList();
$list = (new MangaHistoryTransformer())->transform($raw);
$this->cache->set($key, $list);
}
return $list;
}
/**
* Get the manga list for the configured user
*
* @param string $status - The reading status by which to filter the list
* @return mixed[]
*/
public function getMangaList(string $status): array
{
$key = "kitsu-manga-list-{$status}";
$list = $this->cache->get($key, NULL);
if ($list === NULL)
{
$data = $this->getList(MediaType::MANGA, $status);
// Bail out on no data
if (empty($data))
{
return [];
}
$transformer = new MangaListTransformer();
$transformed = $transformer->transformCollection($data);
$keyed = [];
foreach ($transformed as $item)
{
$keyed[$item['id']] = $item;
}
$list = $keyed;
$this->cache->set($key, $list);
}
return $list;
}
/**
* Get the number of manga list items
*
* @param string $status - Optional status to filter by
*/
public function getMangaListCount(string $status = ''): int
{
2020-12-10 15:59:37 -05:00
return $this->getListCount(MediaType::MANGA, $status);
}
/**
* Get all Manga lists
*
* @return array<string, mixed[]>
*/
public function getFullOrganizedMangaList(): array
{
$statuses = KitsuReadingStatus::getConstList();
$output = [];
foreach ($statuses as $status)
{
$mappedStatus = MangaReadingStatus::KITSU_TO_TITLE[$status];
$output[$mappedStatus] = $this->getMangaList($status);
}
return $output;
}
// ------------------------------------------------------------------------
// Base methods
// ------------------------------------------------------------------------
2016-12-21 12:46:20 -05:00
/**
2017-03-28 14:34:33 -04:00
* Search for an anime or manga
2016-12-21 12:46:20 -05:00
*
2017-03-28 14:34:33 -04:00
* @param string $type - 'anime' or 'manga'
* @param string $query - name of the item to search for
* @return array<int, array<string, mixed>>
2016-12-21 12:46:20 -05:00
*/
2017-03-28 14:34:33 -04:00
public function search(string $type, string $query): array
2016-12-21 12:46:20 -05:00
{
2020-10-21 15:45:30 -04:00
$uType = ucfirst(strtolower($type));
$raw = $this->requestBuilder->runQuery("Search{$uType}", [
'query' => $query,
]);
2016-12-21 12:46:20 -05:00
2020-10-21 15:45:30 -04:00
$nodes = $raw['data']["search{$uType}ByTitle"]['nodes'];
$data = [];
2020-10-21 15:45:30 -04:00
foreach ($nodes as $item)
{
2020-10-21 15:45:30 -04:00
$searchItem = [
'id' => $item['id'],
'slug' => $item['slug'],
'coverImage' => K::getPosterImage($item),
2020-10-21 15:45:30 -04:00
'canonicalTitle' => $item['titles']['canonical'],
'titles' => array_values(K::getTitles($item['titles'])),
'libraryEntry' => $item['myLibraryEntry'],
2020-10-21 15:45:30 -04:00
];
2020-10-21 15:45:30 -04:00
// Search for MAL mapping
if (is_array($item['mappings']['nodes']))
{
foreach ($item['mappings']['nodes'] as $mapping)
{
if ($mapping['externalSite'] === 'MYANIMELIST_' . strtoupper($type))
2020-10-21 15:45:30 -04:00
{
$searchItem['mal_id'] = $mapping['externalId'];
break;
}
}
}
2020-10-21 15:45:30 -04:00
$data[] = $searchItem;
2016-12-21 12:46:20 -05:00
}
2020-10-21 15:45:30 -04:00
return $data;
2016-12-21 12:46:20 -05:00
}
/**
* Find a media item on Kitsu by its associated MAL id
*
* @param string $type "anime" or "manga"
*/
public function getKitsuIdFromMALId(string $malId, string $type = 'anime'): ?string
{
$raw = $this->requestBuilder->runQuery('GetIdByMapping', [
'id' => $malId,
'site' => strtoupper("MYANIMELIST_{$type}"),
]);
2017-04-10 15:31:35 -04:00
return $raw['data']['lookupMapping']['id'] ?? NULL;
}
/**
* Get the data for a specific list item, generally for editing
*
* @param string $listId - The unique identifier of that list item
*/
public function getListItem(string $listId): AnimeListItem|MangaListItem|array
{
$baseData = $this->listItem->get($listId);
2020-08-24 13:07:47 -04:00
if ( ! isset($baseData['data']['findLibraryEntryById']))
{
// We need to get the errors...
return $baseData;
}
2018-10-19 10:40:11 -04:00
2020-08-24 13:07:47 -04:00
return (new LibraryEntryTransformer())->transform($baseData['data']['findLibraryEntryById']);
}
/**
* @return mixed[]
*/
public function getThumbList(string $type): array
{
$statuses = [
'CURRENT',
'PLANNED',
'ON_HOLD',
'DROPPED',
'COMPLETED',
];
$pages = [];
// Although I can fetch the whole list without segregating by status,
// this way is much faster...
foreach ($statuses as $status)
{
2023-05-19 10:54:08 -04:00
foreach ($this->getPages($this->getThumbListPages(...), strtoupper($type), $status) as $page)
{
$pages[] = $page;
}
}
return array_merge(...$pages);
}
/**
2020-07-28 16:11:13 -04:00
* Get the data to sync Kitsu anime/manga list with another API
*
* @return mixed[]
*/
public function getSyncList(string $type): array
{
$statuses = [
'CURRENT',
'PLANNED',
'ON_HOLD',
'DROPPED',
'COMPLETED',
];
$pages = [];
// Although I can fetch the whole list without segregating by status,
// this way is much faster...
foreach ($statuses as $status)
{
2023-05-19 10:54:08 -04:00
foreach ($this->getPages($this->getSyncPages(...), strtoupper($type), $status) as $page)
{
$pages[] = $page;
}
}
return array_merge(...$pages);
}
2020-04-21 19:22:56 -04:00
/**
* Get the aggregated pages of anime or manga history
*
* @return mixed[]
2020-04-21 19:22:56 -04:00
*/
protected function getHistoryList(): array
2020-04-21 19:22:56 -04:00
{
return $this->requestBuilder->runQuery('GetUserHistory', [
'slug' => $this->getUsername(),
2020-04-21 19:22:56 -04:00
]);
}
2020-10-09 16:18:45 -04:00
/**
* Get the raw anime/manga list from GraphQL
*
* @return mixed[]
2020-10-09 16:18:45 -04:00
*/
protected function getList(string $type, string $status = ''): array
2020-10-09 16:18:45 -04:00
{
$pages = [];
2023-05-19 10:54:08 -04:00
foreach ($this->getPages($this->getListPages(...), strtoupper($type), strtoupper($status)) as $page)
2020-10-09 16:18:45 -04:00
{
$pages[] = $page;
}
return array_merge(...$pages);
}
private function getListPages(string $type, string $status = ''): Amp\Iterator
2020-10-09 16:18:45 -04:00
{
$cursor = '';
$username = $this->getUsername();
return new Amp\Producer(function (callable $emit) use ($type, $status, $cursor, $username): Generator {
2020-10-09 16:18:45 -04:00
while (TRUE)
{
$vars = [
'type' => $type,
'slug' => $username,
];
if ($status !== '')
{
$vars['status'] = $status;
}
2020-10-09 16:18:45 -04:00
if ($cursor !== '')
{
$vars['after'] = $cursor;
}
$request = $this->requestBuilder->queryRequest('GetLibrary', $vars);
$response = yield getApiClient()->request($request);
$json = yield $response->getBody()->buffer();
$rawData = Json::decode($json);
$data = $rawData['data']['findProfileBySlug']['library']['all'] ?? [];
$page = $data['pageInfo'] ?? [];
2020-10-09 16:18:45 -04:00
if (empty($data))
{
// Clear session, in case the error is an invalid token.
$segment = $this->container->get('session')
->getSegment(SESSION_SEGMENT);
$segment->clear();
2021-02-16 14:43:51 -05:00
// @TODO Proper Error logging
dump($rawData);
exit();
2020-10-09 16:18:45 -04:00
}
$cursor = $page['endCursor'];
yield $emit($data['nodes']);
if ($page['hasNextPage'] !== TRUE)
2020-10-09 16:18:45 -04:00
{
break;
}
}
});
}
private function getSyncPages(string $type, string $status): Amp\Iterator
{
$cursor = '';
$username = $this->getUsername();
return new Amp\Producer(function (callable $emit) use ($type, $status, $cursor, $username): Generator {
while (TRUE)
{
$vars = [
'type' => $type,
'slug' => $username,
'status' => $status,
];
if ($cursor !== '')
{
$vars['after'] = $cursor;
}
$request = $this->requestBuilder->queryRequest('GetSyncLibrary', $vars);
$response = yield getApiClient()->request($request);
$json = yield $response->getBody()->buffer();
$rawData = Json::decode($json);
$data = $rawData['data']['findProfileBySlug']['library']['all'] ?? [];
$page = $data['pageInfo'];
if (empty($data))
{
dump($rawData);
exit();
}
$cursor = $page['endCursor'];
yield $emit($data['nodes']);
if ($page['hasNextPage'] === FALSE)
{
break;
}
}
});
}
private function getThumbListPages(string $type, string $status): Amp\Iterator
{
$cursor = '';
$username = $this->getUsername();
return new Amp\Producer(function (callable $emit) use ($type, $status, $cursor, $username): Generator {
while (TRUE)
{
$vars = [
'type' => $type,
'slug' => $username,
'status' => $status,
];
if ($cursor !== '')
{
$vars['after'] = $cursor;
}
$request = $this->requestBuilder->queryRequest('GetLibraryThumbs', $vars);
$response = yield getApiClient()->request($request);
$json = yield $response->getBody()->buffer();
$rawData = Json::decode($json);
$data = $rawData['data']['findProfileBySlug']['library']['all'] ?? [];
$page = $data['pageInfo'];
if (empty($data))
{
dump($rawData);
exit();
}
$cursor = $page['endCursor'];
yield $emit($data['nodes']);
if ($page['hasNextPage'] === FALSE)
{
break;
}
}
});
}
2021-02-16 14:43:51 -05:00
private function getPages(callable $method, mixed ...$args): Generator
{
$items = $method(...$args);
while (wait($items->advance()))
{
yield $items->getCurrent();
}
}
protected function getUserId(): string
{
static $userId = NULL;
if ($userId === NULL)
{
$userId = $this->getUserIdByUsername($this->getUsername());
}
return $userId;
}
/**
* Get the kitsu username from config
*/
private function getUsername(): string
{
return $this->getContainer()
->get('config')
->get(['kitsu_username']);
}
2017-02-04 15:18:34 -05:00
private function getListCount(string $type, string $status = ''): int
{
2020-08-25 13:22:38 -04:00
$args = [
'type' => strtoupper($type),
'slug' => $this->getUsername(),
];
2020-08-25 13:22:38 -04:00
if ($status !== '')
{
2020-08-25 13:22:38 -04:00
$args['status'] = strtoupper($status);
}
2020-08-25 13:22:38 -04:00
$res = $this->requestBuilder->runQuery('GetLibraryCount', $args);
2020-08-25 13:22:38 -04:00
return $res['data']['findProfileBySlug']['library']['all']['totalCount'];
}
}