load("@bazel_skylib//rules:common_settings.bzl", "bool_flag")
load("@python_version_repo//:py_version.bzl", "REQUIREMENTS")
load("@rules_python//python:defs.bzl", "py_binary", "py_library")
load("@rules_python//python:pip.bzl", "compile_pip_requirements")

compile_pip_requirements(
    name = "requirements",
    extra_args = [
        "--allow-unsafe",
        "--build-isolation",
    ],
    generate_hashes = True,
    requirements_in = "requirements.in",
    requirements_txt = REQUIREMENTS,
)

compile_pip_requirements(
    name = "requirements_nightly",
    data = ["test-requirements.txt"],
    extra_args = [
        "--allow-unsafe",
        "--build-isolation",
        "--extra-index-url=https://pypi.anaconda.org/scientific-python-nightly-wheels/simple",
        "--pre",
        "--upgrade",
    ],
    generate_hashes = False,
    requirements_in = "requirements.in",
    requirements_txt = REQUIREMENTS,
)

compile_pip_requirements(
    name = "requirements_dev",
    extra_args = [
        "--allow-unsafe",
        "--build-isolation",
        "--upgrade",
    ],
    generate_hashes = False,
    requirements_in = "requirements.in",
    requirements_txt = REQUIREMENTS,
)

# Flag indicating if the target requires pre-built wheel.
bool_flag(
    name = "wheel_dependency",
    build_setting_default = False,
)

config_setting(
    name = "enable_wheel_dependency",
    flag_values = {
        ":wheel_dependency": "True",
    },
)

filegroup(
    name = "manylinux_compliance_test",
    srcs = ["manylinux_compliance_test.py"],
    visibility = ["//visibility:public"],
)

py_binary(
    name = "unpack_wheel_and_unzip_archive_files",
    srcs = ["unpack_wheel_and_unzip_archive_files.py"],
    main = "unpack_wheel_and_unzip_archive_files.py",
    visibility = ["//visibility:public"],
)

py_library(
    name = "setup_py_nvidia_dependencies_util",
    srcs = ["setup_py_nvidia_dependencies_util.py"],
    visibility = ["//visibility:public"],
)