[bazel][robotpy] Add mirror for robotpy's wpiuil and wpinet libraries (#8062)

Project import generated by Copybara.

GitOrigin-RevId: 92ea93d1b47a82667044bd0af05f7fdb34d2c2c2
This commit is contained in:
PJ Reiniger
2025-08-30 14:55:11 -04:00
committed by GitHub
parent 96004f9bb5
commit bd1dcc4358
96 changed files with 7271 additions and 64 deletions

View File

@@ -0,0 +1,110 @@
import argparse
import json
import tomli
from jinja2 import BaseLoader, Environment
from shared.bazel.rules.robotpy.generation_utils import (
fixup_python_dep_name,
fixup_root_package_name,
fixup_shared_lib_name,
)
def main():
parser = argparse.ArgumentParser()
parser.add_argument("--project_cfg")
parser.add_argument("--output_file")
parser.add_argument("--third_party_dirs", nargs="+")
args = parser.parse_args()
with open(args.project_cfg, "rb") as fp:
raw_config = tomli.load(fp)
def double_quotes(data):
if data:
return json.dumps(data)
return None
def get_pc_dep(library):
base_project = library.replace("robotpy-native-", "")
wpilib_project = fixup_root_package_name(base_project)
return f"//{wpilib_project}:native/{base_project}/{library}.pc"
def get_python_dep(library):
base_project = library.replace("robotpy-native-", "")
wpilib_project = fixup_root_package_name(base_project)
return f"//{fixup_root_package_name(wpilib_project)}:{fixup_python_dep_name(library)}"
env = Environment(loader=BaseLoader)
env.filters["double_quotes"] = double_quotes
env.filters["get_pc_dep"] = get_pc_dep
env.filters["get_python_dep"] = get_python_dep
template = env.from_string(BUILD_FILE_TEMPLATE)
nativelib_config = raw_config["tool"]["hatch"]["build"]["hooks"]["nativelib"]
project_name = nativelib_config["pcfile"][0]["name"]
root_package = fixup_root_package_name(project_name)
shared_library_name = fixup_shared_lib_name(project_name)
with open(args.output_file, "w") as f:
f.write(
template.render(
raw_project_config=raw_config["project"],
nativelib_config=nativelib_config,
root_package=root_package,
shared_library_name=shared_library_name,
third_party_dirs=args.third_party_dirs or [],
)
)
BUILD_FILE_TEMPLATE = """# THIS FILE IS AUTO GENERATED
load("@aspect_bazel_lib//lib:copy_to_directory.bzl", "copy_to_directory")
load("//shared/bazel/rules/robotpy:pybind_rules.bzl", "native_wrappery_library")
def define_native_wrapper(name, pyproject_toml = None):
copy_to_directory(
name = "{}.copy_headers".format(name),
srcs = native.glob(["src/main/native/include/**"]) + native.glob(["src/generated/main/native/include/**"], allow_empty = True){% if third_party_dirs %} + native.glob([
{%- for dir in third_party_dirs %}
"src/main/native/thirdparty/{{dir}}/include/**",
{%- endfor %}
]){%- endif %},
out = "native/{{nativelib_config.pcfile[0].name}}/include",
root_paths = ["src/main/native/include/"],
replace_prefixes = {
"{{root_package}}/src/generated/main/native/include": "",
"{{root_package}}/src/main/native/include": "",
{%- for dir in third_party_dirs %}
"{{root_package}}/src/main/native/thirdparty/{{dir}}/include": "",
{%- endfor %}
},
verbose = False,
visibility = ["//visibility:public"],
)
native_wrappery_library(
name = name,
pyproject_toml = pyproject_toml or "src/main/python/native-pyproject.toml",
libinit_file = "native/{{nativelib_config.pcfile[0].name}}/_init_{{raw_project_config.name.replace("-", "_")}}.py",
pc_file = "native/{{nativelib_config.pcfile[0].name}}/{{raw_project_config.name}}.pc",
pc_deps = [
{%- for dep in nativelib_config.pcfile[0].requires | sort %}
"{{dep | get_pc_dep}}",
{%- endfor %}
],
deps = [
{%- for dep in nativelib_config.pcfile[0].requires | sort %}
"{{dep | get_python_dep}}",
{%- endfor %}
],
headers = "{}.copy_headers".format(name),
native_shared_library = "shared/{{shared_library_name}}",
install_path = "native/{{nativelib_config.pcfile[0].name}}/",
)
"""
if __name__ == "__main__":
main()