HummingBirdAnimeClient/src/AnimeClient/Model/AnimeCollection.php

<?php declare(strict_types=1);
/**
 * Hummingbird Anime List Client
 *
 * An API client for Kitsu to manage anime and manga watch lists
 *
 * PHP version 8
 *
 * @copyright   2015 - 2022  Timothy J. Warren <tim@timshome.page>
 * @license     http://www.opensource.org/licenses/mit-license.html  MIT License
 * @version     5.2
 * @link        https://git.timshome.page/timw4mail/HummingBirdAnimeClient
 */

namespace Aviat\AnimeClient\Model;

use Aviat\Ion\Di\ContainerInterface;
use PDO;
use PDOException;

/**
 * Model for getting anime collection data
 */
final class AnimeCollection extends Collection
{
	/**
	 * Anime API Model
	 */
	protected Anime $animeModel;

	/**
	 * Create the collection model
	 */
	public function __construct(ContainerInterface $container)
	{
		parent::__construct($container);
		$this->animeModel = $container->get('anime-model');
	}

	/**
	 * Get collection from the database, and organize by media type
	 *
	 * @return mixed[]
	 */
	public function getCollection(): array
	{
		if ($this->db === NULL)
		{
			return [];
		}

		$rawCollection = $this->getCollectionFromDatabase();

		$collection = [];

		foreach ($rawCollection as $row)
		{
			if (array_key_exists($row['media'], $collection))
			{
				$collection[$row['media']][] = $row;
			}
			else
			{
				$collection[$row['media']] = [$row];
			}
		}

		return $collection;
	}

	/**
	 * Get the collection from the database
	 *
	 * @return mixed[]
	 */
	public function getFlatCollection(): array
	{
		if ($this->db === NULL)
		{
			return [];
		}

		$query = $this->db->select('a.hummingbird_id, slug, title, alternate_title, show_type,
			 age_rating, episode_count, episode_length, cover_image, notes')
			->from('anime_set a')
			->orderBy('title')
			->get();

		// Add genres associated with each item
		$rows = $query->fetchAll(PDO::FETCH_ASSOC);
		$genres = $this->getGenreList();
		$media = $this->getMediaList();

		if (empty($rows))
		{
			return [];
		}

		foreach ($rows as &$row)
		{
			$id = $row['hummingbird_id'];

			$row['genres'] = array_key_exists($id, $genres)
				? $genres[$id]
				: [];

			$row['media'] = array_key_exists($id, $media)
				? $media[$id]
				: [];

			sort($row['genres']);
		}

		return $rows;
	}

	/**
	 * Get list of media types
	 *
	 * @return array<string, mixed[]>
	 */
	public function getMediaTypeList(): array
	{
		if ($this->db === NULL)
		{
			return [];
		}

		$flatList = [];

		$query = $this->db->select('id, type')
			->from('media')
			->get();

		$rows = $query->fetchAll(PDO::FETCH_ASSOC);
		if (empty($rows))
		{
			return [];
		}

		foreach ($rows as $row)
		{
			$flatList[$row['id']] = $row['type'];
		}

		// Organize the media types into groups
		// @TODO: make this more database-driven, rather than hardcoded
		return [
			'Common' => [
				2 => $flatList[2], // Blu-ray
				3 => $flatList[3], // DVD
				7 => $flatList[7], // Digital
				4 => $flatList[4], // Bootleg
			],
			'Retro' => [
				5 => $flatList[5], // LaserDisc
				6 => $flatList[6], // VHS
				9 => $flatList[9], // Betamax
				8 => $flatList[8], // Video CD
			],
			'Other' => [
				10 => $flatList[10], // UMD
				11 => $flatList[11], // Other
			],
		];
	}

	/**
	 * Add an item to the anime collection
	 */
	public function add(mixed $data): void
	{
		if ($this->db === NULL)
		{
			return;
		}

		// Check that the anime doesn't already exist
		if ($this->has($data['id']))
		{
			return;
		}

		$id = $data['id'];
		$anime = (object) $this->animeModel->getAnimeById($id);

		$this->db->set([
			'hummingbird_id' => $id,
			'slug' => $anime->slug,
			'title' => array_shift($anime->titles),
			'alternate_title' => implode('<br />', $anime->titles),
			'show_type' => $anime->show_type,
			'age_rating' => $anime->age_rating,
			'cover_image' => $anime->cover_image,
			'episode_count' => $anime->episode_count,
			'episode_length' => $anime->episode_length,
			'notes' => $data['notes'],
		])->insert('anime_set');

		$this->updateMediaLink($id, $data['media_id']);
		$this->updateGenres($id);
	}

	/**
	 * Verify that an item was added
	 */
	public function wasAdded(array $data): bool
	{
		if ($this->db === NULL)
		{
			return FALSE;
		}

		$row = $this->get($data['id']);

		return ! empty($row);
	}

	/**
	 * Update a collection item
	 */
	public function update(array $data): void
	{
		if ($this->db === NULL)
		{
			return;
		}

		// If there's no id to update, don't update
		if ( ! array_key_exists('hummingbird_id', $data))
		{
			return;
		}

		$id = $data['hummingbird_id'];
		$media = $data['media_id'];
		unset($data['hummingbird_id'], $data['media_id']);

		// If updating from the 'add' page, there
		// might be no data to actually update in
		// the anime_set table
		if ( ! empty($data))
		{
			$this->db->set($data)
				->where('hummingbird_id', $id)
				->update('anime_set');
		}

		// Update media and genres
		$this->updateMediaLink($id, $media);
		$this->updateGenres($id);
	}

	/**
	 * Verify that the collection item was updated
	 */
	public function wasUpdated(array $data): bool
	{
		if ($this->db === NULL)
		{
			return FALSE;
		}

		$row = $this->get($data['hummingbird_id']);

		foreach ($data as $key => $value)
		{
			if (is_array($row[$key]))
			{
				continue;
			}

			if ((string) $row[$key] !== (string) $value)
			{
				return FALSE;
			}
		}

		return TRUE;
	}

	/**
	 * Remove a collection item
	 */
	public function delete(array $data): void
	{
		if ($this->db === NULL)
		{
			return;
		}

		// If there's no id to update, don't delete
		if ( ! array_key_exists('hummingbird_id', $data))
		{
			return;
		}

		$this->db->beginTransaction();

		$this->db->where('hummingbird_id', $data['hummingbird_id'])
			->delete('anime_set_genre_link');

		$this->db->where('hummingbird_id', $data['hummingbird_id'])
			->delete('anime_set_media_link');

		$this->db->where('hummingbird_id', $data['hummingbird_id'])
			->delete('anime_set');

		$this->db->commit();
	}

	public function wasDeleted(array $data): bool
	{
		if ($this->db === NULL)
		{
			return FALSE;
		}

		return $this->has($data['hummingbird_id']) === FALSE;
	}

	/**
	 * Get the details of a collection item
	 *
	 * @return mixed[]
	 */
	public function get(int|string $kitsuId): array
	{
		if ($this->db === NULL)
		{
			return [];
		}

		// Get the main row data
		$row = $this->db->from('anime_set')
			->where('hummingbird_id', $kitsuId)
			->get()
			->fetch(PDO::FETCH_ASSOC);

		if ($row === FALSE)
		{
			return [];
		}

		// Get the media ids
		$mediaRows = $this->db->select('media_id')
			->from('anime_set_media_link')
			->where('hummingbird_id', $kitsuId)
			->get()
			->fetchAll(PDO::FETCH_ASSOC);

		if (empty($mediaRows))
		{
			return [];
		}

		$row['media_id'] = array_column($mediaRows, 'media_id');

		return $row;
	}

	/**
	 * Does this anime already exist in the collection?
	 */
	public function has(int|string $kitsuId): bool
	{
		if ($this->db === NULL)
		{
			return FALSE;
		}

		$row = $this->db->select('hummingbird_id')
			->from('anime_set')
			->where('hummingbird_id', $kitsuId)
			->get()
			->fetch(PDO::FETCH_ASSOC);

		return ! empty($row);
	}

	/**
	 * Get genres for anime collection items
	 *
	 * @return mixed[]
	 */
	public function getGenreList(array $filter = []): array
	{
		if ($this->db === NULL)
		{
			return [];
		}

		$output = [];

		// Catch the missing table PDOException
		// so that the collection does not show an
		// error by default
		try
		{
			$this->db->select('hummingbird_id, genre')
				->from('anime_set_genre_link gl')
				->join('genres g', 'g.id=gl.genre_id', 'left');

			if ( ! empty($filter))
			{
				$this->db->whereIn('hummingbird_id', $filter);
			}

			$query = $this->db->orderBy('hummingbird_id')
				->orderBy('genre')
				->get();

			$rows = $query->fetchAll(PDO::FETCH_ASSOC);
			if (empty($rows))
			{
				return [];
			}

			foreach ($rows as $row)
			{
				$id = $row['hummingbird_id'];
				$genre = $row['genre'];

				// Empty genre names aren't useful
				if (empty($genre))
				{
					continue;
				}

				if (array_key_exists($id, $output))
				{
					$output[$id][] = $genre;
				}
				else
				{
					$output[$id] = [$genre];
				}
			}
		}
		catch (PDOException)
		{
		}

		$this->db->resetQuery();

		return $output;
	}

	/**
	 * Get media for anime collection items
	 *
	 * @return mixed[]
	 */
	public function getMediaList(array $filter = []): array
	{
		if ($this->db === NULL)
		{
			return [];
		}

		$output = [];

		// Catch the missing table PDOException
		// so that the collection does not show an
		// error by default
		try
		{
			$this->db->select('m.type as media, hummingbird_id')
				->from('anime_set_media_link ml')
				->join('media m', 'm.id=ml.media_id', 'left');

			if ( ! empty($filter))
			{
				$this->db->whereIn('hummingbird_id', $filter);
			}

			$query = $this->db->orderBy('hummingbird_id')
				->orderBy('media')
				->get();

			$rows = $query->fetchAll(PDO::FETCH_ASSOC);
			if (empty($rows))
			{
				return [];
			}

			foreach ($rows as $row)
			{
				$id = $row['hummingbird_id'];
				$media = $row['media'];

				// Empty genre names aren't useful
				if (empty($media))
				{
					continue;
				}

				if (array_key_exists($id, $output))
				{
					$output[$id][] = $media;
				}
				else
				{
					$output[$id] = [$media];
				}
			}
		}
		catch (PDOException)
		{
		}

		$this->db->resetQuery();

		return $output;
	}

	private function updateMediaLink(string $animeId, array $media): void
	{
		if ($this->db === NULL)
		{
			return;
		}

		$this->db->beginTransaction();

		// Delete the old entries
		$this->db->where('hummingbird_id', $animeId)
			->delete('anime_set_media_link');

		// Add the new entries
		$entries = [];

		foreach ($media as $id)
		{
			$entries[] = [
				'hummingbird_id' => $animeId,
				'media_id' => $id,
			];
		}

		$this->db->insertBatch('anime_set_media_link', $entries);

		$this->db->commit();
	}

	/**
	 * Update genre information for selected anime
	 *
	 * @param string $animeId The current anime
	 */
	private function updateGenres(string $animeId): void
	{
		if ($this->db === NULL)
		{
			return;
		}

		// Get api information
		$anime = $this->animeModel->getAnimeById($animeId);

		$this->addNewGenres($anime['genres']);

		$genreInfo = $this->getGenreData();
		$genres = $genreInfo['genres'];
		$links = $genreInfo['links'];

		$linksToInsert = [];

		foreach ($anime['genres'] as $animeGenre)
		{
			// Update link table
			// Get id of genre to put in link table
			$flippedGenres = array_flip($genres);
			$genreId = $flippedGenres[$animeGenre];

			$animeLinks = $links[$animeId] ?? [];

			if ( ! in_array($flippedGenres[$animeGenre], $animeLinks, TRUE))
			{
				$linksToInsert[] = [
					'hummingbird_id' => $animeId,
					'genre_id' => $genreId,
				];
			}
		}

		if ($this->db !== NULL && ! empty($linksToInsert))
		{
			try
			{
				$this->db->insertBatch('anime_set_genre_link', $linksToInsert);
			}
			catch (PDOException)
			{
			}
		}
	}

	/**
	 * Add genres to the database
	 */
	private function addNewGenres(array $genres): void
	{
		if ($this->db === NULL)
		{
			return;
		}

		$existingGenres = $this->getExistingGenres();
		$newGenres = array_diff($genres, $existingGenres);

		$insert = [];

		foreach ($newGenres as $genre)
		{
			$insert[] = [
				'genre' => $genre,
			];
		}

		try
		{
			$this->db->insertBatch('genres', $insert);
		}
		catch (PDOException)
		{
			// dump($e);
		}
	}

	/**
	 * Get list of existing genres
	 *
	 * @return array<string, mixed[]>
	 */
	private function getGenreData(): array
	{
		return [
			'genres' => $this->getExistingGenres(),
			'links' => $this->getExistingGenreLinkEntries(),
		];
	}

	/**
	 * @return mixed[]
	 */
	private function getExistingGenres(): array
	{
		if ($this->db === NULL)
		{
			return [];
		}

		$genres = [];

		// Get existing genres
		$query = $this->db->select('id, genre')
			->from('genres')
			->get();

		$rows = $query->fetchAll(PDO::FETCH_ASSOC);
		if (empty($rows))
		{
			return [];
		}

		foreach ($rows as $genre)
		{
			$genres[$genre['id']] = $genre['genre'];
		}

		$this->db->resetQuery();

		return $genres;
	}

	/**
	 * @return mixed[]
	 */
	private function getExistingGenreLinkEntries(): array
	{
		if ($this->db === NULL)
		{
			return [];
		}

		$links = [];

		$query = $this->db->select('hummingbird_id, genre_id')
			->from('anime_set_genre_link')
			->get();

		$rows = $query->fetchAll(PDO::FETCH_ASSOC);
		if (empty($rows))
		{
			return [];
		}

		foreach ($rows as $link)
		{
			if (array_key_exists($link['hummingbird_id'], $links))
			{
				$links[$link['hummingbird_id']][] = $link['genre_id'];
			}
			else
			{
				$links[$link['hummingbird_id']] = [$link['genre_id']];
			}
		}

		$this->db->resetQuery();

		return $links;
	}

	/**
	 * Get full collection from the database
	 *
	 * @return mixed[]
	 */
	private function getCollectionFromDatabase(): array
	{
		if ($this->db === NULL)
		{
			return [];
		}

		$query = $this->db->select('a.hummingbird_id, slug, title, alternate_title, show_type,
			 age_rating, episode_count, episode_length, cover_image, notes, media.type as media')
			->from('anime_set a')
			->join('anime_set_media_link ml', 'ml.hummingbird_id=a.hummingbird_id', 'inner')
			->join('media', 'media.id=ml.media_id', 'inner')
			->orderBy('media')
			->orderBy('title')
			->groupBy('a.hummingbird_id, media.type')
			->get();

		// Add genres associated with each item
		$rows = $query->fetchAll(PDO::FETCH_ASSOC);
		if (empty($rows))
		{
			return [];
		}

		$genres = $this->getGenreList();

		foreach ($rows as &$row)
		{
			$id = $row['hummingbird_id'];

			$row['genres'] = array_key_exists($id, $genres)
				? $genres[$id]
				: [];

			sort($row['genres']);
		}

		return $rows;
	}
}

// End of AnimeCollectionModel.php