import json
import shutil
import subprocess
from pathlib import Path
from urllib.error import HTTPError
from urllib.request import urlopen, urlretrieve

from pyodide_lock import PyodideLockSpec

from . import build_env
from .common import exit_with_stdio
from .create_pypa_index import create_pypa_index
from .logger import logger


def _download_xbuildenv(
    version: str, xbuildenv_path: Path, *, url: str | None = None
) -> None:
    from shutil import rmtree, unpack_archive
    from tempfile import NamedTemporaryFile

    xbuildenv_url = (
        url
        or f"https://github.com/pyodide/pyodide/releases/download/{version}/xbuildenv-{version}.tar.bz2"
    )
    logger.info(f"Downloading xbuild environment from {xbuildenv_url}")
    rmtree(xbuildenv_path, ignore_errors=True)
    with NamedTemporaryFile(suffix=".tar") as f:
        urlretrieve(
            xbuildenv_url,
            f.name,
        )
        unpack_archive(f.name, xbuildenv_path)


def install_xbuildenv(version: str, xbuildenv_path: Path) -> Path:
    logger.info("Installing xbuild environment")

    xbuildenv_path = xbuildenv_path / "xbuildenv"
    xbuildenv_root = xbuildenv_path / "pyodide-root"

    if (xbuildenv_path / ".installed").exists():
        return xbuildenv_root

    # TODO: use a separate configuration file for variables that are used only in package building
    host_site_packages = Path(
        build_env._get_make_environment_vars(pyodide_root=xbuildenv_root)[
            "HOSTSITEPACKAGES"
        ]
    )
    host_site_packages.mkdir(exist_ok=True, parents=True)
    result = subprocess.run(
        [
            "pip",
            "install",
            "--no-user",
            "-t",
            host_site_packages,
            "-r",
            xbuildenv_path / "requirements.txt",
        ],
        capture_output=True,
        encoding="utf8",
    )
    if result.returncode != 0:
        exit_with_stdio(result)
    # Copy the site-packages-extras (coming from the cross-build-files meta.yaml
    # key) over the site-packages directory with the newly installed packages.
    shutil.copytree(
        xbuildenv_path / "site-packages-extras", host_site_packages, dirs_exist_ok=True
    )
    cdn_base = f"https://cdn.jsdelivr.net/pyodide/v{version}/full/"
    lockfile_path = xbuildenv_root / "dist" / "pyodide-lock.json"
    if lockfile_path.exists():
        lockfile = PyodideLockSpec.from_json(lockfile_path)
    else:
        try:
            with urlopen(cdn_base + "pyodide-lock.json") as response:
                lockfile_bytes = response.read()
        except HTTPError:
            # Try again with old url
            with urlopen(cdn_base + "repodata.json") as response:
                lockfile_bytes = response.read()
        lockfile = PyodideLockSpec(**json.loads(lockfile_bytes))
    create_pypa_index(lockfile.packages, xbuildenv_root, cdn_base)

    (xbuildenv_path / ".installed").touch()

    return xbuildenv_root


def install(path: Path, *, download: bool = True, url: str | None = None) -> Path:
    """
    Install cross-build environment.

    Parameters
    ----------
    path
        A path to the cross-build environment.
    download
        Whether to download the cross-build environment before installing it.
    url
        URL to download the cross-build environment from. This is only used
        if `download` is True. The URL should point to a tarball containing
        the cross-build environment. If not specified, the corresponding
        release on GitHub is used.

        Warning: if you are downloading from a version that is not the same
        as the current version of pyodide-build, make sure that the cross-build
        environment is compatible with the current version of Pyodide.

    Returns
    -------
    Path to the Pyodide root directory for the cross-build environment.
    """
    from . import __version__

    version = __version__

    if not download and not path.exists():
        logger.error("xbuild environment not exists")
        raise FileNotFoundError(path)

    if download and path.exists():
        logger.warning("xbuild environment already exists, skipping download")

    elif download:
        _download_xbuildenv(version, path, url=url)

    return install_xbuildenv(version, path)