Trakt

"""Client for movie recommendations."""

from urllib.parse import quote

from client.auth import AuthClient
from config.api import DEFAULT_LIMIT
from config.endpoints import TRAKT_ENDPOINTS
from models.recommendations.recommendation import TraktRecommendedMovie
from utils.api.error_types import AuthenticationRequiredError
from utils.api.errors import handle_api_errors
from utils.api.id_helpers import build_trakt_id_object


class MovieRecommendationsClient(AuthClient):
    """Client for fetching movie recommendations from Trakt."""

    @handle_api_errors
    async def get_movie_recommendations(
        self,
        limit: int = DEFAULT_LIMIT,
        ignore_collected: bool = False,
        ignore_watchlisted: bool = False,
    ) -> list[TraktRecommendedMovie]:
        """Fetch personalized movie recommendations.

        Note: The Trakt recommendations API does not support pagination.
        Use the limit parameter (max 100) to control number of results.

        Args:
            limit: Number of results to return (max 100).
            ignore_collected: Filter out movies user has collected.
            ignore_watchlisted: Filter out movies user has watchlisted.

        Returns:
            List of movie recommendations.

        Raises:
            AuthenticationRequiredError: If the client is not authenticated.
        """
        if not self.is_authenticated():
            raise AuthenticationRequiredError("get movie recommendations")

        endpoint = TRAKT_ENDPOINTS["recommendations_movies"]
        params: dict[str, str | int | bool] = {"limit": limit}

        if ignore_collected:
            params["ignore_collected"] = "true"
        if ignore_watchlisted:
            params["ignore_watchlisted"] = "true"

        return await self._make_typed_list_request(
            endpoint=endpoint,
            params=params,
            response_type=TraktRecommendedMovie,
        )

    @handle_api_errors
    async def hide_movie_recommendation(self, movie_id: str) -> bool:
        """Hide a movie from future recommendations.

        Args:
            movie_id: Trakt ID, slug, or IMDB ID.

        Returns:
            True if successfully hidden.

        Raises:
            AuthenticationRequiredError: If the client is not authenticated.
        """
        if not self.is_authenticated():
            raise AuthenticationRequiredError("hide movie recommendation")

        endpoint = TRAKT_ENDPOINTS["hide_movie_recommendation"].replace(
            ":id", quote(movie_id, safe="")
        )
        await self._delete_request(endpoint)
        return True

    @handle_api_errors
    async def unhide_movie_recommendation(self, movie_id: str) -> bool:
        """Unhide a movie to restore it in future recommendations.

        Args:
            movie_id: Trakt ID, slug, or IMDB ID.

        Returns:
            True if successfully unhidden.

        Raises:
            AuthenticationRequiredError: If the client is not authenticated.
        """
        if not self.is_authenticated():
            raise AuthenticationRequiredError("unhide movie recommendation")

        endpoint = TRAKT_ENDPOINTS["unhide_recommendations"]
        data = build_trakt_id_object(movie_id, "movies")
        await self._post_request(endpoint, data)
        return True