BUILD.bazel

# Copyright 2022 Ant Group Co., Ltd.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#   http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

load("@python_versions//3.10:defs.bzl", compile_pip_requirements_3_10 = "compile_pip_requirements")
load("@python_versions//3.11:defs.bzl", compile_pip_requirements_3_11 = "compile_pip_requirements")

# load("@python_versions//3.9:defs.bzl", compile_pip_requirements_3_9 = "compile_pip_requirements")
load("@rules_python//python:packaging.bzl", "py_package", "py_wheel")
load("@rules_python//python:pip.bzl", "compile_pip_requirements")
load("//:version.bzl", "SPU_VERSION")

# compile_pip_requirements_3_9(
#     name = "requirements_3_9",
#     src = "requirements_3_9.txt",
#     requirements_txt = "requirements_lock_3_9.txt",
#     tags = ["manual"],
# )

compile_pip_requirements_3_10(
    name = "requirements_3_10",
    src = "requirements.txt",
    requirements_txt = "requirements_lock_3_10.txt",
    tags = ["manual"],
)

compile_pip_requirements_3_11(
    name = "requirements_3_11",
    src = "requirements.txt",
    requirements_txt = "requirements_lock_3_11.txt",
    tags = ["manual"],
)

compile_pip_requirements(
    name = "requirements-dev",
    src = "requirements-dev.txt",
    requirements_txt = "requirements_dev_lock.txt",
    tags = ["manual"],
)

exports_files([
    "README.md",
])

# # https://rules-python.readthedocs.io/en/latest/api/rules_python/python/packaging.html#py_wheel_rule
# py_wheel(
#     name = "spu_wheel",
#     abi = select(
#         {
#             "@rules_python//python/config_settings:is_python_3.10": "cp310",
#             "@rules_python//python/config_settings:is_python_3.11": "cp311",
#             # "@rules_python//python/config_settings:is_python_3.9": "cp39",
#             "//conditions:default": "none",
#         },
#     ),
#     author = "SecretFlow Team",
#     author_email = "secretflow-contact@service.alipay.com",
#     classifiers = [
#         "Programming Language :: Python :: 3.9",
#         "Programming Language :: Python :: 3.10",
#         "Programming Language :: Python :: 3.11",
#     ],
#     description_content_type = "text/markdown",
#     description_file = "README.md",
#     distribution = "spu",
#     extra_distinfo_files = {
#         "//:LICENSE": "LICENSE",
#     },
#     homepage = "https://github.com/secretflow/spu",
#     license = "Apache License 2.0",
#     # TODO: add other fields.
#     platform = select(
#         {
#             "@bazel_tools//src/conditions:linux_x86_64": "manylinux2014_x86_64",
#             "@bazel_tools//src/conditions:darwin_arm64": "macosx_13_0_arm64",
#             "@bazel_tools//src/conditions:linux_aarch64": "manylinux_2_28_aarch64",
#             "//conditions:default": "any",
#         },
#     ),
#     python_requires = ">=3.9",
#     python_tag = select(
#         {
#             "@rules_python//python/config_settings:is_python_3.10": "cp310",
#             "@rules_python//python/config_settings:is_python_3.11": "cp311",
#             # "@rules_python//python/config_settings:is_python_3.9": "cp39",
#             "//conditions:default": "py3",
#         },
#     ),
#     requires_file = "requirements.txt",
#     summary = "SPU aims to be a 'provable', 'measurable' secure computation device.",
#     twine = None,
#     version = SPU_VERSION,
#     deps = [
#         ":spu_pkg",
#     ],
# )

py_package(
    name = "spu_pkg",
    packages = [
        "libspu",
        "spu",
    ],
    visibility = ["//visibility:private"],
    deps = [
        "//spu:api",
        "//spu:init",
        "//spu:libpsi",
        "//spu:libspu",
        "//spu/ops/groupby",
        "//spu/utils:distributed",
    ],
)