From 3382b720e3b9c7f7ff23faeccfb0a056f6690d7d Mon Sep 17 00:00:00 2001
From: Benex254 <benextempest@gmail.com>
Date: Sun, 18 Aug 2024 12:33:51 +0300
Subject: [PATCH] feat(cli): add grab command

---
 fastanime/cli/__init__.py      |   1 +
 fastanime/cli/commands/grab.py | 120 +++++++++++++++++++++++++++++++++
 2 files changed, 121 insertions(+)
 create mode 100644 fastanime/cli/commands/grab.py

diff --git a/fastanime/cli/__init__.py b/fastanime/cli/__init__.py
index c7800bb..ae4cc62 100644
--- a/fastanime/cli/__init__.py
+++ b/fastanime/cli/__init__.py
@@ -16,6 +16,7 @@ commands = {
     "cache": "cache.cache",
     "completions": "completions.completions",
     "update": "update.update",
+    "grab": "grab.grab",
 }
 
 
diff --git a/fastanime/cli/commands/grab.py b/fastanime/cli/commands/grab.py
new file mode 100644
index 0000000..2d9e605
--- /dev/null
+++ b/fastanime/cli/commands/grab.py
@@ -0,0 +1,120 @@
+from typing import TYPE_CHECKING
+
+import click
+
+from ..completion_functions import anime_titles_shell_complete
+
+if TYPE_CHECKING:
+    from ..config import Config
+
+
+@click.command(
+    help="Helper command to get streams for anime to use externally in a non-python application",
+    short_help="Print anime streams to standard out",
+)
+@click.option(
+    "--anime-titles",
+    "--anime_title",
+    "-t",
+    required=True,
+    shell_complete=anime_titles_shell_complete,
+    multiple=True,
+    help="Specify which anime to download",
+)
+@click.option(
+    "--episode-range",
+    "-r",
+    help="A range of episodes to download (start-end)",
+)
+@click.pass_obj
+def grab(
+    config: "Config",
+    anime_titles: tuple,
+    episode_range,
+):
+    import json
+    from logging import getLogger
+    from sys import exit
+
+    from thefuzz import fuzz
+
+    from ...AnimeProvider import AnimeProvider
+
+    logger = getLogger(__name__)
+
+    anime_provider = AnimeProvider(config.provider)
+
+    grabbed_animes = []
+    for anime_title in anime_titles:
+        # ---- search for anime ----
+        search_results = anime_provider.search_for_anime(
+            anime_title, translation_type=config.translation_type
+        )
+        if not search_results:
+            exit(1)
+        search_results = search_results["results"]
+        search_results_ = {
+            search_result["title"]: search_result for search_result in search_results
+        }
+
+        search_result = max(
+            search_results_.keys(), key=lambda title: fuzz.ratio(title, anime_title)
+        )
+
+        # ---- fetch anime ----
+        anime = anime_provider.get_anime(search_results_[search_result]["id"])
+        if not anime:
+            exit(1)
+        episodes = sorted(
+            anime["availableEpisodesDetail"][config.translation_type], key=float
+        )
+        # where the magic happens
+        if episode_range:
+            if ":" in episode_range:
+                ep_range_tuple = episode_range.split(":")
+                if len(ep_range_tuple) == 2 and all(ep_range_tuple):
+                    episodes_start, episodes_end = ep_range_tuple
+                    episodes_range = episodes[int(episodes_start) : int(episodes_end)]
+                elif len(ep_range_tuple) == 3 and all(ep_range_tuple):
+                    episodes_start, episodes_end, step = ep_range_tuple
+                    episodes_range = episodes[
+                        int(episodes_start) : int(episodes_end) : int(step)
+                    ]
+                else:
+                    episodes_start, episodes_end = ep_range_tuple
+                    if episodes_start.strip():
+                        episodes_range = episodes[int(episodes_start) :]
+                    elif episodes_end.strip():
+                        episodes_range = episodes[: int(episodes_end)]
+                    else:
+                        episodes_range = episodes
+            else:
+                episodes_range = episodes[int(episode_range) :]
+
+        else:
+            episodes_range = sorted(episodes, key=float)
+
+        grabbed_anime = dict(anime)
+        grabbed_anime["requested_episodes"] = episodes_range
+        grabbed_anime["episodes_streams"] = {}
+        # lets download em
+        for episode in episodes_range:
+            try:
+                if episode not in episodes:
+                    continue
+                streams = anime_provider.get_episode_streams(
+                    anime, episode, config.translation_type
+                )
+                if not streams:
+                    continue
+                grabbed_anime["episodes_streams"][episode] = {
+                    server["server"]: server for server in streams
+                }
+
+            except Exception as e:
+                logger.error(e)
+        grabbed_animes.append(grabbed_anime)
+        if len(grabbed_animes) == 1:
+            print(json.dumps(grabbed_animes[0]))
+        else:
+            print(json.dumps(grabbed_animes))