Begin move to cleo

rip/cli_cleo.py

@@ -0,0 +1,214 @@
+import logging
+import os
+
+from cleo.application import Application as BaseApplication
+from cleo.commands.command import Command
+
+from streamrip import __version__
+
+from .config import Config
+from .core import RipCore
+
+logging.basicConfig(level="WARNING")
+logger = logging.getLogger("streamrip")
+
+
+class DownloadCommand(Command):
+    """
+    Download items from a url
+
+    url
+        {--f|file=None : Path to a text file containing urls}
+        {urls?* : One or more Qobuz, Tidal, Deezer, or SoundCloud urls}
+    """
+
+    help = (
+        '\nDownload "Dreams" by Fleetwood Mac:\n'
+        "$ <fg=magenta>rip url https://www.deezer.com/en/track/63480987</>\n\n"
+        "Batch download urls from a text file named urls.txt:\n"
+        "$ <fg=magenta>rip --file urls.txt</>\n"
+    )
+
+    def handle(self):
+        config = Config()
+        core = RipCore(config)
+
+        urls = self.argument("urls")
+        path = self.option("file")
+        if path != "None":
+            if os.path.isfile(path):
+                core.handle_txt(path)
+            else:
+                self.line(
+                    f"<error>File <comment>{path}</comment> does not exist.</error>"
+                )
+                return 1
+
+        if urls:
+            core.handle_urls(";".join(urls))
+
+        if len(core) > 0:
+            core.download()
+        elif not urls and path == "None":
+            self.line("<error>Must pass arguments. See </><info>rip url -h</info>.")
+
+        return 0
+
+
+class SearchCommand(Command):
+    """
+    Search for and download items in interactive mode.
+
+    search
+        {query : The name to search for}
+        {--s|source=qobuz : Qobuz, Tidal, Soundcloud, Deezer, or Deezloader}
+        {--t|type=album : Album, Playlist, Track, or Artist}
+    """
+
+    def handle(self):
+        query = self.argument("query")
+        source, type = clean_options(self.option("source"), self.option("type"))
+
+        config = Config()
+        core = RipCore(config)
+
+        if core.interactive_search(query, source, type):
+            core.download()
+        else:
+            self.line("<error>No items chosen, exiting.</error>")
+
+
+class DiscoverCommand(Command):
+    """
+    Browse and download items in interactive mode.
+
+    discover
+        {--s|scrape : Download all of the items in the list}
+        {--m|max-items=50 : The number of items to fetch}
+        {list=ideal-discography : The list to fetch}
+    """
+
+    help = (
+        "\nAvailable options for <info>list</info>:\n\n"
+        "    • most-streamed\n"
+        "    • recent-releases\n"
+        "    • best-sellers\n"
+        "    • press-awards\n"
+        "    • ideal-discography\n"
+        "    • editor-picks\n"
+        "    • most-featured\n"
+        "    • qobuzissims\n"
+        "    • new-releases\n"
+        "    • new-releases-full\n"
+        "    • harmonia-mundi\n"
+        "    • universal-classic\n"
+        "    • universal-jazz\n"
+        "    • universal-jeunesse\n"
+        "    • universal-chanson\n"
+    )
+
+    def handle(self):
+        from streamrip.constants import QOBUZ_FEATURED_KEYS
+
+        chosen_list = self.argument("list")
+        scrape = self.option("scrape")
+        max_items = self.option("max-items")
+
+        if chosen_list not in QOBUZ_FEATURED_KEYS:
+            self.line(f'<error>Error: list "{chosen_list}" not available</error>')
+            self.line(self.help)
+            return 1
+
+        config = Config()
+        core = RipCore(config)
+
+        if scrape:
+            core.scrape(chosen_list, max_items)
+            core.download()
+            return 0
+
+        if core.interactive_search(
+            chosen_list, "qobuz", "featured", limit=int(max_items)
+        ):
+            core.download()
+        else:
+            self.line("<error>No items chosen, exiting.</error>")
+
+
+class LastfmCommand(Command):
+    """
+    Search for tracks from a list.fm playlist and download them.
+
+    lastfm
+        {--s|source=qobuz : The source to search for items on}
+        {urls* : Last.fm playlist urls}
+    """
+
+    def handle(self):
+        source = self.option("source")
+        urls = self.argument("urls")
+
+        config = Config()
+        core = RipCore(config)
+        config.session["lastfm"]["source"] = source
+        core.handle_lastfm_urls(";".join(urls))
+        core.download()
+
+
+class ConfigCommand(Command):
+    """
+    Manage the configuration file
+
+    {--o|open : Open the config file in the default application}
+    {--ov|open-vim : Open the config file in (neo)vim}
+    {--d|directory : Open the directory that the config file is located in}
+    {--p|path : Show the config file's path}
+    {--q|qobuz : Set the credentials for Qobuz}
+    {--t|tidal : Log into Tidal}
+    {--dz|deezer : Set the Deezer ARL}
+    {--reset : Reset the config file}
+    {--update : Reset the config file, keeping the credentials}
+    """
+
+
+class Application(BaseApplication):
+    def __init__(self):
+        super().__init__("rip", __version__)
+
+    def _run(self, io):
+        if io.is_debug():
+            logger.setLevel(logging.DEBUG)
+
+        super()._run(io)
+
+    # @property
+    # def _default_definition(self):
+    #     default_globals = super()._default_definition
+    #     default_globals.add_option(Option("convert", shortcut="c", flag=False))
+    #     return default_globals
+
+
+# class ConvertCommand(Command):
+#     pass
+
+
+# class RepairCommand(Command):
+#     pass
+
+
+def clean_options(*opts):
+    return tuple(o.replace("=", "").strip() for o in opts)
+
+
+def main():
+    application = Application()
+    application.add(DownloadCommand())
+    application.add(SearchCommand())
+    application.add(DiscoverCommand())
+    application.add(LastfmCommand())
+    # application.add(ConfigCommand())
+    application.run()
+
+
+if __name__ == "__main__":
+    main()