Hong Shin
4 months ago
committed by
Copybara-Service
parent
4e6b93ca4e
commit
fda2669a52
3 changed files with 1 additions and 321 deletions
@ -1,27 +0,0 @@ |
|||||||
# Copyright (c) 2009-2021, Google LLC |
|
||||||
# All rights reserved. |
|
||||||
# |
|
||||||
# Use of this source code is governed by a BSD-style |
|
||||||
# license that can be found in the LICENSE file or at |
|
||||||
# https://developers.google.com/open-source/licenses/bsd |
|
||||||
|
|
||||||
load("@bazel_skylib//:bzl_library.bzl", "bzl_library") |
|
||||||
|
|
||||||
# begin:google_only |
|
||||||
# package(default_applicable_licenses = ["//upb:license"]) |
|
||||||
# end:google_only |
|
||||||
|
|
||||||
licenses(["notice"]) |
|
||||||
|
|
||||||
bzl_library( |
|
||||||
name = "upb_cc_proto_library_bzl", |
|
||||||
srcs = ["upb_cc_proto_library.bzl"], |
|
||||||
visibility = ["//visibility:public"], |
|
||||||
deps = [ |
|
||||||
"//bazel:upb_proto_library_bzl", |
|
||||||
"//bazel/common:proto_info_bzl", |
|
||||||
"@bazel_skylib//lib:paths", |
|
||||||
"@bazel_tools//tools/cpp:toolchain_utils", |
|
||||||
"@bazel_tools//tools/cpp:toolchain_utils.bzl", |
|
||||||
], |
|
||||||
) |
|
||||||
@ -1,293 +0,0 @@ |
|||||||
# Copyright (c) 2009-2021, Google LLC |
|
||||||
# All rights reserved. |
|
||||||
# |
|
||||||
# Use of this source code is governed by a BSD-style |
|
||||||
# license that can be found in the LICENSE file or at |
|
||||||
# https://developers.google.com/open-source/licenses/bsd |
|
||||||
"""Public rules for using upb protos: |
|
||||||
- upb_cc_proto_library() |
|
||||||
""" |
|
||||||
|
|
||||||
load("@bazel_skylib//lib:paths.bzl", "paths") |
|
||||||
load("@bazel_tools//tools/cpp:toolchain_utils.bzl", "find_cpp_toolchain", "use_cpp_toolchain") |
|
||||||
load("//bazel:upb_proto_library.bzl", "GeneratedSrcsInfo", "UpbWrappedCcInfo", "upb_proto_library_aspect") |
|
||||||
load("//bazel/common:proto_info.bzl", "ProtoInfo") |
|
||||||
|
|
||||||
# begin:google_only |
|
||||||
# |
|
||||||
# def upb_use_cpp_toolchain(): |
|
||||||
# # TODO: We shouldn't need to add this to the result of use_cpp_toolchain(). |
|
||||||
# return [ |
|
||||||
# config_common.toolchain_type( |
|
||||||
# "@bazel_tools//tools/cpp:cc_runtimes_toolchain_type", |
|
||||||
# mandatory = False, |
|
||||||
# ), |
|
||||||
# ] + use_cpp_toolchain() |
|
||||||
# |
|
||||||
# end:google_only |
|
||||||
|
|
||||||
# begin:github_only |
|
||||||
def upb_use_cpp_toolchain(): |
|
||||||
return use_cpp_toolchain() |
|
||||||
|
|
||||||
# end:github_only |
|
||||||
|
|
||||||
# Generic support code ######################################################### |
|
||||||
|
|
||||||
def _get_real_short_path(file): |
|
||||||
# For some reason, files from other archives have short paths that look like: |
|
||||||
# ../com_google_protobuf/google/protobuf/descriptor.proto |
|
||||||
short_path = file.short_path |
|
||||||
if short_path.startswith("../"): |
|
||||||
second_slash = short_path.index("/", 3) |
|
||||||
short_path = short_path[second_slash + 1:] |
|
||||||
|
|
||||||
# Sometimes it has another few prefixes like: |
|
||||||
# _virtual_imports/any_proto/google/protobuf/any.proto |
|
||||||
# benchmarks/_virtual_imports/100_msgs_proto/benchmarks/100_msgs.proto |
|
||||||
# We want just google/protobuf/any.proto. |
|
||||||
virtual_imports = "_virtual_imports/" |
|
||||||
if virtual_imports in short_path: |
|
||||||
short_path = short_path.split(virtual_imports)[1].split("/", 1)[1] |
|
||||||
return short_path |
|
||||||
|
|
||||||
def _get_real_root(file): |
|
||||||
real_short_path = _get_real_short_path(file) |
|
||||||
return file.path[:-len(real_short_path) - 1] |
|
||||||
|
|
||||||
def _generate_output_file(ctx, src, extension): |
|
||||||
real_short_path = _get_real_short_path(src) |
|
||||||
real_short_path = paths.relativize(real_short_path, ctx.label.package) |
|
||||||
output_filename = paths.replace_extension(real_short_path, extension) |
|
||||||
ret = ctx.actions.declare_file(output_filename) |
|
||||||
return ret |
|
||||||
|
|
||||||
def _filter_none(elems): |
|
||||||
return [e for e in elems if e] |
|
||||||
|
|
||||||
def _cc_library_func(ctx, name, hdrs, srcs, copts, dep_ccinfos): |
|
||||||
"""Like cc_library(), but callable from rules. |
|
||||||
|
|
||||||
Args: |
|
||||||
ctx: Rule context. |
|
||||||
name: Unique name used to generate output files. |
|
||||||
hdrs: Public headers that can be #included from other rules. |
|
||||||
srcs: C/C++ source files. |
|
||||||
copts: Additional options for cc compilation. |
|
||||||
dep_ccinfos: CcInfo providers of dependencies we should build/link against. |
|
||||||
|
|
||||||
Returns: |
|
||||||
CcInfo provider for this compilation. |
|
||||||
""" |
|
||||||
|
|
||||||
# begin:google_only |
|
||||||
# cc_runtimes_toolchain = ctx.toolchains["@bazel_tools//tools/cpp:cc_runtimes_toolchain_type"] |
|
||||||
# if cc_runtimes_toolchain: |
|
||||||
# dep_ccinfos += [ |
|
||||||
# target[CcInfo] |
|
||||||
# for target in cc_runtimes_toolchain.cc_runtimes_info.runtimes |
|
||||||
# ] |
|
||||||
# |
|
||||||
# end:google_only |
|
||||||
|
|
||||||
compilation_contexts = [info.compilation_context for info in dep_ccinfos] |
|
||||||
linking_contexts = [info.linking_context for info in dep_ccinfos] |
|
||||||
toolchain = find_cpp_toolchain(ctx) |
|
||||||
feature_configuration = cc_common.configure_features( |
|
||||||
ctx = ctx, |
|
||||||
cc_toolchain = toolchain, |
|
||||||
requested_features = ctx.features, |
|
||||||
unsupported_features = ctx.disabled_features, |
|
||||||
) |
|
||||||
|
|
||||||
(compilation_context, compilation_outputs) = cc_common.compile( |
|
||||||
actions = ctx.actions, |
|
||||||
feature_configuration = feature_configuration, |
|
||||||
cc_toolchain = toolchain, |
|
||||||
name = name, |
|
||||||
srcs = srcs, |
|
||||||
public_hdrs = hdrs, |
|
||||||
user_compile_flags = copts, |
|
||||||
compilation_contexts = compilation_contexts, |
|
||||||
) |
|
||||||
|
|
||||||
# buildifier: disable=unused-variable |
|
||||||
(linking_context, linking_outputs) = cc_common.create_linking_context_from_compilation_outputs( |
|
||||||
actions = ctx.actions, |
|
||||||
name = name, |
|
||||||
feature_configuration = feature_configuration, |
|
||||||
cc_toolchain = toolchain, |
|
||||||
compilation_outputs = compilation_outputs, |
|
||||||
linking_contexts = linking_contexts, |
|
||||||
) |
|
||||||
|
|
||||||
return CcInfo( |
|
||||||
compilation_context = compilation_context, |
|
||||||
linking_context = linking_context, |
|
||||||
) |
|
||||||
|
|
||||||
# Dummy rule to expose select() copts to aspects ############################## |
|
||||||
|
|
||||||
UpbCcProtoLibraryCoptsInfo = provider( |
|
||||||
"Provides copts for upb cc proto targets", |
|
||||||
fields = { |
|
||||||
"copts": "copts for upb_cc_proto_library()", |
|
||||||
}, |
|
||||||
) |
|
||||||
|
|
||||||
def upb_cc_proto_library_copts_impl(ctx): |
|
||||||
return UpbCcProtoLibraryCoptsInfo(copts = ctx.attr.copts) |
|
||||||
|
|
||||||
upb_cc_proto_library_copts = rule( |
|
||||||
implementation = upb_cc_proto_library_copts_impl, |
|
||||||
attrs = {"copts": attr.string_list(default = [])}, |
|
||||||
) |
|
||||||
|
|
||||||
_UpbCcWrappedCcInfo = provider("Provider for cc_info for protos", fields = ["cc_info"]) |
|
||||||
_WrappedCcGeneratedSrcsInfo = provider("Provider for generated sources", fields = ["srcs"]) |
|
||||||
|
|
||||||
def _compile_upb_cc_protos(ctx, generator, proto_info, proto_sources): |
|
||||||
if len(proto_sources) == 0: |
|
||||||
return GeneratedSrcsInfo(srcs = [], hdrs = []) |
|
||||||
|
|
||||||
tool = getattr(ctx.executable, "_gen_" + generator) |
|
||||||
srcs = [_generate_output_file(ctx, name, ".upb.proto.cc") for name in proto_sources] |
|
||||||
hdrs = [_generate_output_file(ctx, name, ".upb.proto.h") for name in proto_sources] |
|
||||||
hdrs += [_generate_output_file(ctx, name, ".upb.fwd.h") for name in proto_sources] |
|
||||||
transitive_sets = proto_info.transitive_descriptor_sets.to_list() |
|
||||||
|
|
||||||
args = ctx.actions.args() |
|
||||||
args.use_param_file(param_file_arg = "@%s") |
|
||||||
args.set_param_file_format("multiline") |
|
||||||
|
|
||||||
args.add("--" + generator + "_out=" + _get_real_root(srcs[0])) |
|
||||||
args.add("--plugin=protoc-gen-" + generator + "=" + tool.path) |
|
||||||
args.add("--descriptor_set_in=" + ctx.configuration.host_path_separator.join([f.path for f in transitive_sets])) |
|
||||||
args.add_all(proto_sources, map_each = _get_real_short_path) |
|
||||||
|
|
||||||
ctx.actions.run( |
|
||||||
inputs = depset( |
|
||||||
direct = [proto_info.direct_descriptor_set], |
|
||||||
transitive = [proto_info.transitive_descriptor_sets], |
|
||||||
), |
|
||||||
tools = [tool], |
|
||||||
outputs = srcs + hdrs, |
|
||||||
executable = ctx.executable._protoc, |
|
||||||
arguments = [args], |
|
||||||
progress_message = "Generating upb cc protos for :" + ctx.label.name, |
|
||||||
) |
|
||||||
return GeneratedSrcsInfo(srcs = srcs, hdrs = hdrs) |
|
||||||
|
|
||||||
def _upb_cc_proto_rule_impl(ctx): |
|
||||||
if len(ctx.attr.deps) != 1: |
|
||||||
fail("only one deps dependency allowed.") |
|
||||||
dep = ctx.attr.deps[0] |
|
||||||
|
|
||||||
if _WrappedCcGeneratedSrcsInfo in dep: |
|
||||||
srcs = dep[_WrappedCcGeneratedSrcsInfo].srcs |
|
||||||
else: |
|
||||||
fail("proto_library rule must generate _WrappedCcGeneratedSrcsInfo (aspect should have " + |
|
||||||
"handled this).") |
|
||||||
|
|
||||||
if _UpbCcWrappedCcInfo in dep: |
|
||||||
cc_info = dep[_UpbCcWrappedCcInfo].cc_info |
|
||||||
elif UpbWrappedCcInfo in dep: |
|
||||||
cc_info = dep[UpbWrappedCcInfo].cc_info |
|
||||||
else: |
|
||||||
fail("proto_library rule must generate UpbWrappedCcInfo or " + |
|
||||||
"_UpbCcWrappedCcInfo (aspect should have handled this).") |
|
||||||
|
|
||||||
lib = cc_info.linking_context.linker_inputs.to_list()[0].libraries[0] |
|
||||||
files = _filter_none([ |
|
||||||
lib.static_library, |
|
||||||
lib.pic_static_library, |
|
||||||
lib.dynamic_library, |
|
||||||
]) |
|
||||||
return [ |
|
||||||
DefaultInfo(files = depset(files + srcs.hdrs + srcs.srcs)), |
|
||||||
srcs, |
|
||||||
cc_info, |
|
||||||
] |
|
||||||
|
|
||||||
def _upb_cc_proto_aspect_impl(target, ctx, generator, cc_provider, file_provider): |
|
||||||
proto_info = target[ProtoInfo] |
|
||||||
files = _compile_upb_cc_protos(ctx, generator, proto_info, proto_info.direct_sources) |
|
||||||
deps = ctx.rule.attr.deps + getattr(ctx.attr, "_" + generator) |
|
||||||
dep_ccinfos = [dep[CcInfo] for dep in deps if CcInfo in dep] |
|
||||||
dep_ccinfos += [dep[UpbWrappedCcInfo].cc_info for dep in deps if UpbWrappedCcInfo in dep] |
|
||||||
dep_ccinfos += [dep[_UpbCcWrappedCcInfo].cc_info for dep in deps if _UpbCcWrappedCcInfo in dep] |
|
||||||
if UpbWrappedCcInfo not in target: |
|
||||||
fail("Target should have UpbWrappedCcInfo provider") |
|
||||||
dep_ccinfos.append(target[UpbWrappedCcInfo].cc_info) |
|
||||||
cc_info = _cc_library_func( |
|
||||||
ctx = ctx, |
|
||||||
name = ctx.rule.attr.name + "." + generator, |
|
||||||
hdrs = files.hdrs, |
|
||||||
srcs = files.srcs, |
|
||||||
copts = ctx.attr._ccopts[UpbCcProtoLibraryCoptsInfo].copts, |
|
||||||
dep_ccinfos = dep_ccinfos, |
|
||||||
) |
|
||||||
return [cc_provider(cc_info = cc_info), file_provider(srcs = files)] |
|
||||||
|
|
||||||
def _upb_cc_proto_library_aspect_impl(target, ctx): |
|
||||||
return _upb_cc_proto_aspect_impl(target, ctx, "upbprotos", _UpbCcWrappedCcInfo, _WrappedCcGeneratedSrcsInfo) |
|
||||||
|
|
||||||
_upb_cc_proto_library_aspect = aspect( |
|
||||||
attrs = { |
|
||||||
"_ccopts": attr.label( |
|
||||||
default = "//protos:upb_cc_proto_library_copts__for_generated_code_only_do_not_use", |
|
||||||
), |
|
||||||
"_gen_upbprotos": attr.label( |
|
||||||
executable = True, |
|
||||||
cfg = "exec", |
|
||||||
default = "//src/google/protobuf/compiler/hpb:protoc-gen-upb-protos", |
|
||||||
), |
|
||||||
"_protoc": attr.label( |
|
||||||
executable = True, |
|
||||||
cfg = "exec", |
|
||||||
default = "//:protoc", |
|
||||||
), |
|
||||||
"_cc_toolchain": attr.label( |
|
||||||
default = "@bazel_tools//tools/cpp:current_cc_toolchain", |
|
||||||
), |
|
||||||
"_upbprotos": attr.label_list( |
|
||||||
default = [ |
|
||||||
# TODO: Add dependencies for cc runtime (absl/string etc..) |
|
||||||
"//upb:generated_cpp_support__only_for_generated_code_do_not_use__i_give_permission_to_break_me", |
|
||||||
"//protos:generated_protos_support__only_for_generated_code_do_not_use__i_give_permission_to_break_me", |
|
||||||
"@com_google_absl//absl/strings", |
|
||||||
"@com_google_absl//absl/status:statusor", |
|
||||||
"//protos", |
|
||||||
"//protos:repeated_field", |
|
||||||
], |
|
||||||
), |
|
||||||
}, |
|
||||||
implementation = _upb_cc_proto_library_aspect_impl, |
|
||||||
provides = [ |
|
||||||
_UpbCcWrappedCcInfo, |
|
||||||
_WrappedCcGeneratedSrcsInfo, |
|
||||||
], |
|
||||||
required_aspect_providers = [ |
|
||||||
UpbWrappedCcInfo, |
|
||||||
], |
|
||||||
attr_aspects = ["deps"], |
|
||||||
fragments = ["cpp"], |
|
||||||
toolchains = upb_use_cpp_toolchain(), |
|
||||||
) |
|
||||||
|
|
||||||
upb_cc_proto_library = rule( |
|
||||||
implementation = _upb_cc_proto_rule_impl, |
|
||||||
attrs = { |
|
||||||
"deps": attr.label_list( |
|
||||||
aspects = [ |
|
||||||
upb_proto_library_aspect, |
|
||||||
_upb_cc_proto_library_aspect, |
|
||||||
], |
|
||||||
allow_rules = ["proto_library"], |
|
||||||
providers = [ProtoInfo], |
|
||||||
), |
|
||||||
"_ccopts": attr.label( |
|
||||||
default = "//protos:upb_cc_proto_library_copts__for_generated_code_only_do_not_use", |
|
||||||
), |
|
||||||
}, |
|
||||||
) |
|
||||||
Loading…
Reference in new issue