revanced-api/api/backends/entities.py

from typing import Optional
from dataclasses import dataclass


@dataclass
class Metadata(dict):
    """
    Represents the metadata of a release.

    Attributes:
        - tag_name (str): The name of the release tag.
        - name (str): The name of the release.
        - body (str): The body of the release.
        - draft (bool): Whether the release is a draft.
        - prerelease (bool): Whether the release is a prerelease.
        - created_at (str): The creation date of the release.
        - published_at (str): The publication date of the release.
    """

    def __init__(
        self,
        tag_name: str,
        name: str,
        draft: bool,
        prerelease: bool,
        created_at: str,
        published_at: str,
        body: str,
        repository: Optional[str] = None,
    ):
        dict.__init__(
            self,
            tag_name=tag_name,
            name=name,
            draft=draft,
            prerelease=prerelease,
            created_at=created_at,
            published_at=published_at,
            body=body,
            repository=repository,
        )


@dataclass
class Asset(dict):
    """
    Represents an asset in a release.

    Attributes:
        - name (str): The name of the asset.
        - content_type (str): The MIME type of the asset content.
        - download_count (int): The number of times the asset has been downloaded.
        - download_url (str): The URL to download the asset.
    """

    def __init__(
        self,
        name: str,
        content_type: str,
        download_count: int,
        browser_download_url: str,
    ):
        dict.__init__(
            self,
            name=name,
            content_type=content_type,
            download_count=download_count,
            browser_download_url=browser_download_url,
        )


@dataclass
class Release(dict):
    """
    Represents a release.

    Attributes:
        - metadata (Metadata): The metadata of the release.
        - assets (list[Asset]): The assets of the release.
    """

    def __init__(self, metadata: Metadata, assets: list[Asset]):
        dict.__init__(self, metadata=metadata, assets=assets)


@dataclass
class Contributor(dict):
    """
    Represents a contributor to a repository.

    Attributes:
        - login (str): The GitHub username of the contributor.
        - avatar_url (str): The URL to the contributor's avatar image.
        - html_url (str): The URL to the contributor's GitHub profile.
        - contributions (Optional[int]): The number of contributions the contributor has made to the repository.
    """

    def __init__(
        self,
        login: str,
        avatar_url: str,
        html_url: str,
        contributions: Optional[int] = None,
        bio: Optional[str] = None,
        keys: Optional[str] = None,
    ):
        match contributions, bio, keys:
            case None, None, None:
                dict.__init__(
                    self,
                    login=login,
                    avatar_url=avatar_url,
                    html_url=html_url,
                    bio=bio,
                    keys=keys,
                )
            case int(_), None, None:
                dict.__init__(
                    self,
                    login=login,
                    avatar_url=avatar_url,
                    html_url=html_url,
                    contributions=contributions,
                )
            case None, str(_), None:
                dict.__init__(
                    self,
                    login=login,
                    avatar_url=avatar_url,
                    html_url=html_url,
                    bio=bio,
                )
            case int(_), str(_), str(_):
                dict.__init__(
                    self,
                    login=login,
                    avatar_url=avatar_url,
                    html_url=html_url,
                    contributions=contributions,
                    bio=bio,
                    keys=keys,
                )
            case None, str(_), str(_):
                dict.__init__(
                    self,
                    login=login,
                    avatar_url=avatar_url,
                    html_url=html_url,
                    bio=bio,
                    keys=keys,
                )
            case _:
                raise ValueError("Invalid arguments")


@dataclass
class AppInfo(dict):
    """
    Represents the information of an app.

    Attributes:
        - name (str): The name of the app.
        - category (str): The app category.
        - logo (str): The base64 enconded app logo.
    """

    def __init__(self, name: str, category: str, logo: str):
        dict.__init__(
            self,
            name=name,
            category=category,
            logo=logo,
        )