FastAnime/fastanime/cli/interfaces.py

from __future__ import annotations

from InquirerPy import inquirer

from ..libs.anilist.anilist import AniList
from ..libs.anilist.anilist_data_schema import AnilistDataSchema
from ..libs.anime_provider.allanime.api import anime_provider
from .config import Config
from .utils.mpv import mpv
from .utils.utils import clear, fuzzy_inquirer, get_selected_anime, get_selected_server


def fetch_episode(config: Config, anime, translation_type, selected_anime):
    # fetch episode
    episode_number = fuzzy_inquirer(
        "Select Episode:",
        [*anime["show"]["availableEpisodesDetail"][translation_type], "back"],
    )
    if episode_number == "back":
        anime_provider_(
            config,
            selected_anime[0]["name"],
        )
        return
    print(config.translation_type)
    episode = anime_provider.get_anime_episode(
        selected_anime[0]["_id"], episode_number, config.translation_type
    )

    fetch_streams(config, episode, anime, translation_type, selected_anime)


def fetch_streams(config: Config, episode, *args):
    episode_streams = list(anime_provider.get_episode_streams(episode))

    server = fuzzy_inquirer(
        "Select Server:", [episode_stream[0] for episode_stream in episode_streams]
    )
    selected_server = get_selected_server(server, episode_streams)

    quality = config.quality
    links = selected_server[1]["links"]
    if quality > len(links) - 1:
        quality = len(links) - 1
    elif quality < 0:
        quality = 0
    stream_link = links[quality]["link"]
    print("Now playing:", args[-1][0]["name"])
    mpv(stream_link)
    clear()
    player_controls(config, episode, links, *args)


def player_controls(config: Config, episode, links: list, *args):
    def _back():
        fetch_streams(config, episode, *args)

    def _replay():
        pass

    def _next_episode():
        pass

    def _episodes():
        fetch_episode(config, *args)

    def _previous_episode():
        pass

    def _change_quality():
        options = [link["link"] for link in links]
        quality = fuzzy_inquirer("Select Quality:", options)
        config.quality = options.index(quality)  # set quality
        player_controls(config, episode, links, *args)

    def _change_translation_type():
        options = ["sub", "dub"]
        translation_type = fuzzy_inquirer("Select Translation Type:", options)
        config.translation_type = translation_type  # set trannslation type
        player_controls(config, episode, links, *args)

    options = {
        "Replay": _replay,
        "Next Episode": _next_episode,
        "Episodes": _episodes,
        "Previous Episode": _previous_episode,
        "Change Quality": _change_quality,
        "Change Translation Type": _change_translation_type,
        "Back": _back,
    }

    action = fuzzy_inquirer("Select Action:", options.keys())
    options[action]()


def anime_provider_(config: Config, anime_title, **kwargs):
    translation_type = config.translation_type
    search_results = anime_provider.search_for_anime(anime_title, translation_type)
    search_results_anime_titles = [
        anime["name"] for anime in search_results["shows"]["edges"]
    ]
    selected_anime_title = fuzzy_inquirer(
        "Select Search Result:",
        [*search_results_anime_titles, "back"],
        default=kwargs.get("default_anime_title", ""),
    )
    if selected_anime_title == "back":
        anilist(config)
        return
    fetch_anime_epiosode(
        config,
        selected_anime_title,
        search_results,
    )


def fetch_anime_epiosode(config, selected_anime_title, search_results):
    translation_type = config.translation_type
    selected_anime = get_selected_anime(selected_anime_title, search_results)
    anime = anime_provider.get_anime(selected_anime[0]["_id"])

    fetch_episode(config, anime, translation_type, selected_anime)


def _stream(config, anilist_data: AnilistDataSchema, preferred_lang="romaji"):
    anime_titles = [
        str(anime["title"][preferred_lang])
        for anime in anilist_data["data"]["Page"]["media"]
    ]
    selected_anime_title = fuzzy_inquirer("Select Anime:", anime_titles)
    anime_provider_(
        config, selected_anime_title, default_anime_title=selected_anime_title
    )


def anilist_options(config, anilist_data: AnilistDataSchema):
    def _watch_trailer():
        pass

    def _add_to_list():
        pass

    def _remove_from_list():
        pass

    def _view_info():
        pass

    options = {
        "stream": _stream,
        "watch trailer": _watch_trailer,
        "add to list": _add_to_list,
        "remove from list": _remove_from_list,
        "view info": _view_info,
        "back": anilist,
    }
    action = fuzzy_inquirer("Select Action:", options.keys())
    options[action](config, anilist_data)


def anilist(config, *args, **kwargs):
    def _anilist_search():
        search_term = inquirer.text(
            "Search:", instruction="Enter anime to search for"
        ).execute()

        return AniList.search(query=search_term)

    options = {
        "trending": AniList.get_trending,
        "search": _anilist_search,
        "most popular anime": AniList.get_most_popular,
        "most favourite anime": AniList.get_most_favourite,
        "most scored anime": AniList.get_most_scored,
        "upcoming anime": AniList.get_most_favourite,
        "recently updated anime": AniList.get_most_recently_updated,
    }
    action = fuzzy_inquirer("Select Action:", options.keys())
    anilist_data = options[action]()
    if anilist_data[0]:
        anilist_options(config, anilist_data[1])