streamrip/music_dl/cli.py

# For tests

import logging
import os
from getpass import getpass

import click

from .config import Config
from .constants import CACHE_DIR, CONFIG_DIR, CONFIG_PATH
from .core import MusicDL
from .utils import init_log

logger = logging.getLogger(__name__)
config = Config(CONFIG_PATH)

if not os.path.isdir(CONFIG_DIR):
    os.makedirs(CONFIG_DIR)
if not os.path.isdir(CACHE_DIR):
    os.makedirs(CONFIG_DIR)

config = Config(CONFIG_PATH)
core = MusicDL(config)


@click.group()
@click.option(
    "--flush-cache",
    metavar="PATH",
    help="Flush the cache before running (only for extreme cases)",
)
@click.pass_context
def cli(ctx, **kwargs):
    """cli.

    $ rip www.qobuz.com/album/id1089374 convert -c ALAC -sr 48000

    > download and convert to alac, downsample to 48kHz

    $ rip config --read

    > Config(...)

    $ rip www.qobuz.com/artist/id223049 filter --studio-albums --no-repeats

    > download discography with given filters
    """


@click.command(name="dl")
@click.option("-q", "--quality", metavar="INT", help="Quality integer ID (5, 6, 7, 27)")
@click.option("-f", "--folder", metavar="PATH", help="Custom download folder")
@click.option("-s", "--search", metavar="QUERY")
@click.option("-nd", "--no-db", is_flag=True)
@click.option("-c", "--convert", metavar="CODEC")
@click.option("-sr", "--sampling-rate", metavar="INT")
@click.option("-bd", "--bit-depth", metavar="INT")
@click.option("--debug", default=False, is_flag=True, help="Enable debug logging")
@click.argument("items", nargs=-1)
@click.pass_context
def download(ctx, **kwargs):
    """
    Download an URL, space separated URLs or a text file with URLs.
    Mixed arguments are also supported.

    Examples:

        * `qobuz-dl dl https://some.url/some_type/some_id`

        * `qobuz-dl dl file_with_urls.txt`

        * `qobuz-dl dl URL URL URL`

    Supported sources and their types:

        * Deezer (album, artist, track, playlist)

        * Qobuz (album, artist, label, track, playlist)

        * Tidal (album, artist, track, playlist)
    """
    if kwargs.get("debug"):
        init_log()

    config.update_from_cli(**ctx.params)
    for item in kwargs["items"]:
        try:
            if os.path.isfile(item):
                core.from_txt(item)
                click.secho(f"File input found: {item}", fg="yellow")
            else:
                core.handle_url(item)
        except Exception as error:
            logger.error(error, exc_info=True)
            click.secho(
                f"{type(error).__name__} raised processing {item}: {error}", fg="red"
            )

    if ctx.params["convert"] is not None:
        core.convert_all(
            ctx.params["convert"],
            sampling_rate=ctx.params["sampling_rate"],
            bit_depth=ctx.params["bit_depth"],
        )


@click.command(name="config")
@click.option("-o", "--open", is_flag=True)
@click.option("-q", "--qobuz", is_flag=True)
@click.option("-t", "--tidal", is_flag=True)
def edit_config(open, qobuz, tidal):
    if open:
        # open in text editor
        click.launch(CONFIG_PATH)
        return

    if qobuz:
        config["qobuz"]["email"] = input("Qobuz email: ")
        config["qobuz"]["password"] = getpass("Qobuz password: ")
        config.save()
        click.secho(f"Config saved at {CONFIG_PATH}", fg="green")

    if tidal:
        config["tidal"]["email"] = input("Tidal email: ")
        config["tidal"]["password"] = getpass("Tidal password: ")
        config.save()
        click.secho(f"Config saved at {CONFIG_PATH}", fg="green")


@click.command()
@click.option(
    "-t",
    "--type",
    default="album",
    help="Type to search for. Can be album, artist, playlist, track",
)
@click.argument("QUERY")
def search(media_type, query):
    print(f"searching for {media_type} with {query=}")


@click.command()
def interactive():
    pass


@click.command()
@click.option("--no-extras", is_flag=True, help="Ignore extras")
@click.option("--no-features", is_flag=True, help="Ignore features")
@click.option("--studio-albums", is_flag=True, help="Ignore non-studio albums")
@click.option("--remaster-only", is_flag=True, help="Ignore non-remastered albums")
@click.option("--albums-only", is_flag=True, help="Ignore non-album downloads")
def filter(*args):
    print(f"filter {args=}")


@click.command()
@click.option(
    "--default-comment", metavar="COMMENT", help="Custom comment tag for audio files"
)
@click.option("--no-cover", help="Do not embed cover into audio file.")
def tags(default_comment, no_cover):
    print(f"{default_comment=}, {no_cover=}")


def main():
    cli.add_command(download)
    cli.add_command(filter)
    cli.add_command(tags)
    cli.add_command(edit_config)
    cli()


if __name__ == "__main__":
    main()