anipy_dmenu.py

from pathlib import Path
import anipy_cli
import dmenu
import typing

# Other functions

def error(msg: str) -> typing.NoReturn:
    # Tell the user the message
    dmenu.show([], prompt="Error: " + msg)
    exit(1)

def warning(msg: str) -> None:
    # Tell the user the message
    dmenu.show([], prompt="Warning: " + msg)

def get_list_of_str_nums(start: int, end: int) -> list[str]:
    """Returns a list of strings of numbers between the start and end.
    The start and end are inclusive"""

    return [str(i) for i in range(start, end + 1)]


# Helper functions for using dmenu


def show_select(links_and_names: tuple[list[str], list[str]] | typing.Literal[0]) -> str:
    """Lets user select the show from the list provided by anipy_cli.query.query.get_links
    Returns the link for the selected show (without the gogoanime part)"""

    if links_and_names == 0: error("Invaild input into show_select (links_and_names == 0)")

    names: list[str] = links_and_names[1]
    links: list[str] = links_and_names[0]

    # Show user selction for names
    selected_name = dmenu.show(names)
    if selected_name == None: exit(0)

    # Now get the link the corresponding to the name selected
    location = names.index(selected_name)

    return links[location]


def ep_select(entry: anipy_cli.entry) -> anipy_cli.entry:
    """Lets the user pick which episode number they want
    Returns an entry with .ep filled 
    Requires .catagory_url to be filled"""
    
    ep_class = anipy_cli.epHandler(entry)
    ep_class.get_latest()

    # Get options for avalible episodes
    if ep_class.entry.latest_ep < 1:
        error("No episodes avalible (entry.lastest_ep is a value less than 1)")
    
    ep: str | None = dmenu.show(get_list_of_str_nums(1, ep_class.entry.latest_ep))
    # Make sure it is valid
    if ep == None: exit(0)
    if not ep.isdigit(): error("Episode number chosen is not a number (ep.isdigit() return False)")
    if int(ep) > ep_class.entry.latest_ep: error("Episode number chosen is too big (ep > latest_ep)")
    if int(ep) < 1: error("Episode number chosen is too small (ep < 1)")

    ep_class.entry.ep = int(ep)
    return ep_class.gen_eplink()
    

# Functions for eaiser use in sub-commands

def save_entry(entry: anipy_cli.entry) -> None:
    """Save an entry to the cache file. Used for playing next."""
    save_path = Path.home() / ".cache" / "anipy_dmenu_ep"
    try:
        save_path.parent.mkdir(exist_ok=True)
    except PermissionError:
        warning(f"Unable to make the folder the save file should be in (PermissionError when running mkdir on: {str(save_path)})")
        exit(1)

    save_data: list[str] = [
            entry.category_url,
            str(entry.ep)
            ]
    try:
        with save_path.open('w') as save_file:
            for piece in save_data:
                save_file.write(piece + '\n')
    except PermissionError:
        warning(f"Unable to save episode (PermissionError when writing to: {str(save_path)})")
        return

def load_entry() -> anipy_cli.entry:
    """Loads an entry from the cache file"""
    save_path = Path.home() / ".cache" / "anipy_dmenu_ep"
    
    try:
        with save_path.open('r') as save_file:
            save_data = save_file.read()
    except PermissionError:
        warning(f"Unable to read episode (PermissionError when reading: {str(save_path)})")
        exit(1)

    if save_data == "": error("Failed to load entry (got nothing from reading save_file in load_entry")
    # TODO: Fix this later
    data = save_data.split('\n')
    return anipy_cli.entry(category_url=data[0], ep=int(data[1]))

def play_entry(entry: anipy_cli.entry) -> None:
    """Plays the given entry and does not return until mpv is quit

    Requires the show_name, ep, embed_url, and stream_url fields to be filled

    This function is a wrapper of anipy-cli's player.mpv function, but just works better for this project
    """

    subproc = anipy_cli.mpv(entry)

    # .communicate pretty much just waits until the subproc exits
    subproc.communicate()

    return

def play_from_cached_entry_data(entry: anipy_cli.entry) -> None:
    """Note: This function is subject to change, it will be deprecated after this project swiches to using anipy-cli's history functions

    This function plays an entry that only has the category_url and ep filled in. It was made to be used in the next, previous, and replay commands to reduce reused code.
    """

    entry = anipy_cli.epHandler(entry).gen_eplink()

    url_class = anipy_cli.videourl(entry, None)
    url_class.stream_url()
    entry = url_class.get_entry()

    # Play it
    play_entry(entry)

if __name__ == '__main__':
    error("This file is not intented to be run (__name__ == '__main__' in anipy_dmenu.py)")