import os
from hashlib import md5
from pathlib import Path

from supervision.assets.list import MEDIA_ASSETS, Assets
from supervision.utils.file import _download_to_file
from supervision.utils.logger import _get_logger

logger = _get_logger(__name__)


def is_md5_hash_matching(filename: str | Path, original_md5_hash: str) -> bool:
    """
    Check if the MD5 hash of a file matches the original hash.

    Note: MD5 is used here for file integrity checking (detecting corruption),
    not for cryptographic security purposes.

    Args:
        filename: The path to the file to be checked.
        original_md5_hash: The original MD5 hash to compare against.

    Returns:
        True if the hashes match, False otherwise.
    """
    if not os.path.exists(filename):
        return False

    with open(filename, "rb") as file:
        file_contents = file.read()
        computed_md5_hash = md5(file_contents, usedforsecurity=False)

    return computed_md5_hash.hexdigest() == original_md5_hash


def _download_asset(filename: str, destination: Path) -> None:
    """
    Download asset bytes to the target destination via a temporary file.
    """
    _download_to_file(MEDIA_ASSETS[filename][0], destination, timeout=30.0, stream=True)


def _download_verified_asset(
    filename: str,
    original_md5_hash: str,
    destination: Path,
    check_target: str | Path,
    retry_on_mismatch: bool = True,
) -> None:
    """
    Download an asset and reject payloads whose MD5 does not match the catalog.
    """
    _download_asset(filename, destination)

    if is_md5_hash_matching(check_target, original_md5_hash):
        return

    logger.warning("File corrupted. Re-downloading...")
    os.remove(check_target)

    if retry_on_mismatch:
        _download_verified_asset(
            filename=filename,
            original_md5_hash=original_md5_hash,
            destination=destination,
            check_target=check_target,
            retry_on_mismatch=False,
        )
        return

    raise ValueError(f"Downloaded asset {filename!r} failed MD5 verification.")


def download_assets(
    asset_name: Assets | str,
    directory: str | Path | None = None,
) -> str:
    """
    Download a specified asset if it doesn't already exist or is corrupted.

    Args:
        asset_name: The name or type of the asset to be downloaded.
        directory: Optional output directory. Defaults to the current working
            directory for backward compatibility.

    Returns:
        The downloaded asset path. When `directory` is omitted, this preserves
        the historical filename-only return value.

    Example:
        ```pycon
        >>> from supervision.assets import download_assets, ImageAssets, VideoAssets
        >>> download_assets(VideoAssets.VEHICLES)  # doctest: +SKIP
        'vehicles.mp4'

        >>> download_assets(ImageAssets.PEOPLE_WALKING)  # doctest: +SKIP
        'people-walking.jpg'

        ```
    """

    filename = asset_name.filename if isinstance(asset_name, Assets) else asset_name
    if directory is None:
        destination = Path.cwd() / filename
        check_target: str | Path = filename
        return_value = filename
    else:
        destination_directory = Path(directory).expanduser().resolve()
        destination = destination_directory / filename
        check_target = str(destination)
        return_value = str(destination)

    if filename in MEDIA_ASSETS:
        original_md5_hash = MEDIA_ASSETS[filename][1]
        if not Path(check_target).exists():
            logger.info("Downloading %s assets", filename)
            _download_verified_asset(
                filename=filename,
                original_md5_hash=original_md5_hash,
                destination=destination,
                check_target=check_target,
            )
        else:
            if not is_md5_hash_matching(check_target, original_md5_hash):
                logger.warning("File corrupted. Re-downloading...")
                os.remove(check_target)
                _download_verified_asset(
                    filename=filename,
                    original_md5_hash=original_md5_hash,
                    destination=destination,
                    check_target=check_target,
                )

            logger.info("%s asset download complete.", filename)
    else:
        valid_assets = ", ".join(filename for filename in MEDIA_ASSETS.keys())
        raise ValueError(
            f"Invalid asset. It should be one of the following: {valid_assets}."
        )

    return return_value