revanced-api/api/backends/apkdl.py

from base64 import b64encode

from aiohttp import ClientResponse
from bs4 import BeautifulSoup
from sanic import SanicException
from toolz.functoolz import compose

from api.backends.backend import AppInfoProvider
from api.backends.entities import AppInfo
from api.utils.http_utils import http_get

name: str = "apkdl"
base_url: str = "https://apk-dl.com"


class ApkDl(AppInfoProvider):
    def __init__(self):
        super().__init__(name, base_url)

    async def get_app_info(self, package_name: str) -> AppInfo:
        """Fetches information about an Android app from the ApkDl website.

        Args:
            package_name (str): The package name of the app to fetch.

        Returns:
            AppInfo: An AppInfo object containing the name, category, and logo of the app.

        Raises:
            SanicException: If the HTTP request fails or the app data is incomplete or not found.
        """
        app_url: str = f"{base_url}/{package_name}"
        response: ClientResponse = await http_get(headers={}, url=app_url)
        if response.status != 200:
            raise SanicException(
                f"ApkDl: {response.status}", status_code=response.status
            )
        page = BeautifulSoup(await response.read(), "lxml")
        find_div_text = compose(
            lambda d: d.find_next_sibling("div"),
            lambda d: page.find("div", text=d),
        )
        fetch_logo_url = compose(
            lambda div: div.img["src"],
            lambda _: page.find("div", {"class": "logo"}),
        )
        logo_response: ClientResponse = await http_get(
            headers={}, url=fetch_logo_url(None)
        )
        logo: str = (
            f"data:image/png;base64,{b64encode(await logo_response.content.read()).decode('utf-8')}"
            if logo_response.status == 200
            else ""
        )
        app_data = dict(
            name=find_div_text("App Name").text,
            category=find_div_text("Category").text,
            logo=logo,
        )
        if not all(app_data.values()):
            raise SanicException(
                "ApkDl: App data incomplete or not found", status_code=500
            )
        return AppInfo(**app_data)