dnd-music-console/croaker/cli.py

import io
import logging
import os
import sys
from pathlib import Path
from textwrap import dedent
from typing import List, Optional

import typer
from dotenv import load_dotenv
from typing_extensions import Annotated

import croaker.path
from croaker.exceptions import ConfigurationError
from croaker.playlist import Playlist
from croaker.server import server

SETUP_HELP = """
# Root directory for croaker configuration and logs. See also croaker --root.
CROAKER_ROOT=~/.dnd/croaker

# where to store playlist sources
#PLAYLIST_ROOT=$CROAKER_ROOT/playlists

# Where the record the daemon's PID
#PIDFILE=$CROAKER_ROOT/croaker.pid

# Command and Control TCP Server bind address
HOST=0.0.0.0
PORT=8003

# the kinds of files to add to playlists
MEDIA_GLOB=*.mp3,*.flac,*.m4a

# Icecast2 configuration for Liquidsoap
ICECAST_PASSWORD=
ICECAST_MOUNT=
ICECAST_HOST=
ICECAST_PORT=
ICECAST_URL=
"""

app = typer.Typer()
app_state = {}

logger = logging.getLogger('cli')


@app.callback()
def main(
    context: typer.Context,
    root: Optional[Path] = typer.Option(
        Path("~/.dnd/croaker"),
        help="Path to the Croaker environment",
    ),
    debug: Optional[bool] = typer.Option(None, help="Enable debugging output"),
):
    load_dotenv(root.expanduser() / Path("defaults"))
    load_dotenv(stream=io.StringIO(SETUP_HELP))
    if debug is not None:
        if debug:
            os.environ["DEBUG"] = "1"
        else:
            del os.environ["DEBUG"]

    logging.basicConfig(
        format="%(asctime)s - %(name)s - %(levelname)s - %(message)s",
        level=logging.DEBUG if debug else logging.INFO,
    )

    try:
        croaker.path.root()
        croaker.path.playlist_root()
    except ConfigurationError as e:
        sys.stderr.write(f"{e}\n\n{SETUP_HELP}")
        sys.exit(1)


@app.command()
def setup(context: typer.Context):
    """
    (Re)Initialize Croaker.
    """
    sys.stderr.write("Interactive setup is not yet available. Sorry!\n")
    print(dedent(SETUP_HELP))


@app.command()
def start(
    context: typer.Context,
    daemonize: bool = typer.Option(True, help="Daemonize the server."),
):
    """
    Start the Croaker command and control server.
    """
    server.start(daemonize=daemonize)


@app.command()
def stop():
    """
    Terminate the server.
    """
    server.stop()


@app.command()
def add(
    playlist: str = typer.Argument(
        ...,
        help="Playlist name",
    ),
    theme: Optional[bool] = typer.Option(False, help="Make the first track the theme song."),
    tracks: Annotated[Optional[List[Path]], typer.Argument()] = None,
):
    """
    Recursively add one or more paths to the specified playlist.

    Tracks can be any combination of individual audio files and directories
    containing audio files; anything not already on the playlist will be
    added to it.

    If --theme is specified, the first track will be designated the playlist
    "theme." Theme songs get played first whenever the playlist is loaded,
    after which the playlist order is randomized.
    """
    pl = Playlist(name=playlist)
    pl.add(tracks, make_theme=theme)
    print(pl)


if __name__ == "__main__":
    app.main()