Revert "Delete viu_media/libs/provider directory"

This reverts commit 7e93e4e.

Author: Benexl

viu_media/libs/provider/__init__.py

@@ -0,0 +1 @@
+

viu_media/libs/provider/anime/__init__.py

@@ -0,0 +1,39 @@
+import re
+
+from .....core.constants import GRAPHQL_DIR
+
+SERVERS_AVAILABLE = [
+    "sharepoint",
+    "dropbox",
+    "gogoanime",
+    "weTransfer",
+    "wixmp",
+    "Yt",
+    "mp4-upload",
+]
+API_BASE_URL = "allanime.day"
+API_GRAPHQL_REFERER = "https://allanime.to/"
+API_GRAPHQL_ENDPOINT = f"https://api.{API_BASE_URL}/api/"
+API_GRAPHQL_HEADERS= {
+    "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0.0.0 Safari/537.36",
+    "Content-Type": "application/json",
+    "Origin": f"{API_GRAPHQL_REFERER}",
+}
+
+# search constants
+DEFAULT_COUNTRY_OF_ORIGIN = "all"
+DEFAULT_NSFW = True
+DEFAULT_UNKNOWN = True
+DEFAULT_PER_PAGE = 40
+DEFAULT_PAGE = 1
+
+# regex stuff
+MP4_SERVER_JUICY_STREAM_REGEX = re.compile(
+    r"video/mp4\",src:\"(https?://.*/video\.mp4)\""
+)
+
+# graphql files
+_GQL_QUERIES = GRAPHQL_DIR / "allanime" / "queries"
+SEARCH_GQL = _GQL_QUERIES / "search.gql"
+ANIME_GQL = _GQL_QUERIES / "anime.gql"
+EPISODE_GQL = _GQL_QUERIES / "episodes.gql"

viu_media/libs/provider/anime/allanime/__init__.py

@@ -0,0 +1,3 @@
+from .extractor import extract_server
+
+__all__ = ["extract_server"]

viu_media/libs/provider/anime/allanime/constants.py

@@ -0,0 +1,31 @@
+from ...types import EpisodeStream, Server
+from ..constants import API_BASE_URL
+from ..types import AllAnimeEpisode, AllAnimeSource
+from .base import BaseExtractor
+
+
+class AkExtractor(BaseExtractor):
+    @classmethod
+    def extract(
+        cls,
+        url,
+        client,
+        episode_number: str,
+        episode: AllAnimeEpisode,
+        source: AllAnimeSource,
+    ) -> Server:
+        response = client.get(
+            f"https://{API_BASE_URL}{url.replace('clock', 'clock.json')}",
+            timeout=10,
+        )
+        response.raise_for_status()
+        streams = response.json()
+
+        return Server(
+            name="Ak",
+            links=[
+                EpisodeStream(link=link, quality="1080") for link in streams["links"]
+            ],
+            episode_title=episode["notes"],
+            headers={"Referer": f"https://{API_BASE_URL}/"},
+        )

viu_media/libs/provider/anime/allanime/extractors/__init__.py

@@ -0,0 +1,20 @@
+from abc import ABC, abstractmethod
+
+from httpx import Client
+
+from ...types import Server
+from ..types import AllAnimeEpisode, AllAnimeSource
+
+
+class BaseExtractor(ABC):
+    @classmethod
+    @abstractmethod
+    def extract(
+        cls,
+        url: str,
+        client: Client,
+        episode_number: str,
+        episode: AllAnimeEpisode,
+        source: AllAnimeSource,
+    ) -> Server | None:
+        pass

viu_media/libs/provider/anime/allanime/extractors/ak.py

@@ -0,0 +1,31 @@
+from ...types import EpisodeStream, Server
+from ..constants import API_BASE_URL
+from ..types import AllAnimeEpisode, AllAnimeSource
+from .base import BaseExtractor
+
+
+class SakExtractor(BaseExtractor):
+    @classmethod
+    def extract(
+        cls,
+        url,
+        client,
+        episode_number: str,
+        episode: AllAnimeEpisode,
+        source: AllAnimeSource,
+    ) -> Server:
+        response = client.get(
+            f"https://{API_BASE_URL}{url.replace('clock', 'clock.json')}",
+            timeout=10,
+        )
+        response.raise_for_status()
+        streams = response.json()
+
+        return Server(
+            name="dropbox",
+            links=[
+                EpisodeStream(link=link, quality="1080") for link in streams["links"]
+            ],
+            episode_title=episode["notes"],
+            headers={"Referer": f"https://{API_BASE_URL}/"},
+        )

viu_media/libs/provider/anime/allanime/extractors/base.py

@@ -0,0 +1,65 @@
+from httpx import Client
+
+from ...types import Server
+from ..types import AllAnimeEpisode, AllAnimeSource
+from ..utils import debug_extractor, logger, one_digit_symmetric_xor
+from .ak import AkExtractor
+from .dropbox import SakExtractor
+from .filemoon import FmHlsExtractor, OkExtractor
+from .gogoanime import Lufmp4Extractor
+from .mp4_upload import Mp4Extractor
+from .sharepoint import Smp4Extractor
+from .streamsb import SsHlsExtractor
+from .vid_mp4 import VidMp4Extractor
+from .we_transfer import KirExtractor
+from .wixmp import DefaultExtractor
+from .yt_mp4 import YtExtractor
+
+AVAILABLE_SOURCES = {
+    "Sak": SakExtractor,
+    "S-mp4": Smp4Extractor,
+    "Luf-Mp4": Lufmp4Extractor,
+    "Default": DefaultExtractor,
+    "Yt-mp4": YtExtractor,
+    "Kir": KirExtractor,
+    "Mp4": Mp4Extractor,
+}
+OTHER_SOURCES = {
+    "Ak": AkExtractor,
+    "Vid-mp4": VidMp4Extractor,
+    "Ok": OkExtractor,
+    "Ss-Hls": SsHlsExtractor,
+    "Fm-Hls": FmHlsExtractor,
+}
+
+
+@debug_extractor
+def extract_server(
+    client: Client,
+    episode_number: str,
+    episode: AllAnimeEpisode,
+    source: AllAnimeSource,
+) -> Server | None:
+    url = source.get("sourceUrl")
+    if not url:
+        logger.debug(f"Url not found in source: {source}")
+        return
+
+    if url.startswith("--"):
+        url = one_digit_symmetric_xor(56, url[2:])
+
+        logger.debug(f"Decrypting url for source: {source['sourceName']}")
+    if source["sourceName"] in OTHER_SOURCES:
+        logger.debug(f"Found  {source['sourceName']} but ignoring")
+        return
+
+    if source["sourceName"] not in AVAILABLE_SOURCES:
+        logger.debug(
+            f"Found  {source['sourceName']} but did not expect it, its time to scrape lol"
+        )
+        return
+    logger.debug(f"Found  {source['sourceName']}")
+
+    return AVAILABLE_SOURCES[source["sourceName"]].extract(
+        url, client, episode_number, episode, source
+    )

viu_media/libs/provider/anime/allanime/extractors/dropbox.py

@@ -0,0 +1,62 @@
+from ...types import EpisodeStream, Server
+from ..constants import API_BASE_URL, MP4_SERVER_JUICY_STREAM_REGEX
+from ..types import AllAnimeEpisode, AllAnimeSource
+from .base import BaseExtractor
+
+
+# TODO: requires decoding obsfucated js (filemoon)
+class FmHlsExtractor(BaseExtractor):
+    @classmethod
+    def extract(
+        cls,
+        url,
+        client,
+        episode_number: str,
+        episode: AllAnimeEpisode,
+        source: AllAnimeSource,
+    ) -> Server:
+        response = client.get(
+            f"https://{API_BASE_URL}{url.replace('clock', 'clock.json')}",
+            timeout=10,
+        )
+        response.raise_for_status()
+
+        embed_html = response.text.replace(" ", "").replace("\n", "")
+        vid = MP4_SERVER_JUICY_STREAM_REGEX.search(embed_html)
+        if not vid:
+            raise Exception("")
+        return Server(
+            name="dropbox",
+            links=[EpisodeStream(link=vid.group(1), quality="1080")],
+            episode_title=episode["notes"],
+            headers={"Referer": "https://www.mp4upload.com/"},
+        )
+
+
+# TODO: requires decoding obsfucated js (filemoon)
+class OkExtractor(BaseExtractor):
+    @classmethod
+    def extract(
+        cls,
+        url,
+        client,
+        episode_number: str,
+        episode: AllAnimeEpisode,
+        source: AllAnimeSource,
+    ) -> Server:
+        response = client.get(
+            f"https://{API_BASE_URL}{url.replace('clock', 'clock.json')}",
+            timeout=10,
+        )
+        response.raise_for_status()
+
+        embed_html = response.text.replace(" ", "").replace("\n", "")
+        vid = MP4_SERVER_JUICY_STREAM_REGEX.search(embed_html)
+        if not vid:
+            raise Exception("")
+        return Server(
+            name="dropbox",
+            links=[EpisodeStream(link=vid.group(1), quality="1080")],
+            episode_title=episode["notes"],
+            headers={"Referer": "https://www.mp4upload.com/"},
+        )