• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1"""Support for rust_proto_library_aspect unit-tests."""
2
3load(
4    "//rust:aspects.bzl",
5    "RustProtoInfo",
6    "rust_cc_proto_library_aspect",
7    "rust_upb_proto_library_aspect",
8)
9
10ActionsInfo = provider(
11    doc = ("A provider that exposes what actions were registered by rust_proto_library aspects " +
12           "on proto_libraries."),
13    fields = {"actions": "List[Action]: actions registered on proto_libraries."},
14)
15
16def _attach_upb_aspect_impl(ctx):
17    return [ctx.attr.dep[RustProtoInfo], ActionsInfo(actions = ctx.attr.dep.actions)]
18
19attach_upb_aspect = rule(
20    implementation = _attach_upb_aspect_impl,
21    attrs = {
22        "dep": attr.label(aspects = [rust_upb_proto_library_aspect]),
23    },
24)
25
26def _attach_cc_aspect_impl(ctx):
27    return [ctx.attr.dep[RustProtoInfo], ActionsInfo(actions = ctx.attr.dep.actions)]
28
29attach_cc_aspect = rule(
30    implementation = _attach_cc_aspect_impl,
31    attrs = {
32        "dep": attr.label(aspects = [rust_cc_proto_library_aspect]),
33    },
34)
35