• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1# Copyright (c) 2009-2021, Google LLC
2# All rights reserved.
3#
4# Redistribution and use in source and binary forms, with or without
5# modification, are permitted provided that the following conditions are met:
6#     * Redistributions of source code must retain the above copyright
7#       notice, this list of conditions and the following disclaimer.
8#     * Redistributions in binary form must reproduce the above copyright
9#       notice, this list of conditions and the following disclaimer in the
10#       documentation and/or other materials provided with the distribution.
11#     * Neither the name of Google LLC nor the
12#       names of its contributors may be used to endorse or promote products
13#       derived from this software without specific prior written permission.
14#
15# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
16# ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
17# WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
18# DISCLAIMED. IN NO EVENT SHALL Google LLC BE LIABLE FOR ANY
19# DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
20# (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
21# LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
22# ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
23# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
24# SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
25
26"""Internal rules for building upb."""
27
28load(":upb_proto_library.bzl", "GeneratedSrcsInfo")
29
30# upb_amalgamation() rule, with file_list aspect.
31
32SrcList = provider(
33    fields = {
34        "srcs": "list of srcs",
35    },
36)
37
38def _file_list_aspect_impl(target, ctx):
39    if GeneratedSrcsInfo in target:
40        srcs = target[GeneratedSrcsInfo]
41        return [SrcList(srcs = srcs.srcs + srcs.hdrs)]
42
43    srcs = []
44    for src in ctx.rule.attr.srcs:
45        srcs += src.files.to_list()
46    for hdr in ctx.rule.attr.hdrs:
47        srcs += hdr.files.to_list()
48    for hdr in ctx.rule.attr.textual_hdrs:
49        srcs += hdr.files.to_list()
50    return [SrcList(srcs = srcs)]
51
52_file_list_aspect = aspect(
53    implementation = _file_list_aspect_impl,
54)
55
56def _upb_amalgamation(ctx):
57    inputs = []
58    for lib in ctx.attr.libs:
59        inputs += lib[SrcList].srcs
60    srcs = [src for src in inputs if not src.path.endswith("hpp")]
61    ctx.actions.run(
62        inputs = inputs,
63        outputs = ctx.outputs.outs,
64        arguments = [f.path for f in ctx.outputs.outs] + [f.path for f in srcs],
65        progress_message = "Making amalgamation",
66        executable = ctx.executable._amalgamator,
67    )
68    return []
69
70upb_amalgamation = rule(
71    attrs = {
72        "_amalgamator": attr.label(
73            executable = True,
74            cfg = "exec",
75            default = "//bazel:amalgamate",
76        ),
77        "prefix": attr.string(
78            default = "",
79        ),
80        "libs": attr.label_list(aspects = [_file_list_aspect]),
81        "outs": attr.output_list(),
82        "strip_import_prefix": attr.string_list(),
83    },
84    implementation = _upb_amalgamation,
85)
86