Protocol Buffers - Google's data interchange format (grpc依赖)
https://developers.google.com/protocol-buffers/
135 lines
4.8 KiB
135 lines
4.8 KiB
# Copyright (c) 2023, 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 |
|
"""lua_proto_library(): a rule for building Lua protos.""" |
|
|
|
load("@bazel_skylib//lib:paths.bzl", "paths") |
|
load("//bazel/common:proto_info.bzl", "ProtoInfo") |
|
|
|
# Generic support code ######################################################### |
|
|
|
# begin:github_only |
|
_is_google3 = False |
|
# end:github_only |
|
|
|
# begin:google_only |
|
# _is_google3 = True |
|
# end:google_only |
|
|
|
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(ctx, file): |
|
real_short_path = _get_real_short_path(file) |
|
root = file.path[:-len(real_short_path) - 1] |
|
if not _is_google3 and ctx.rule.attr.strip_import_prefix: |
|
root = paths.join(root, ctx.rule.attr.strip_import_prefix[1:]) |
|
return root |
|
|
|
def _generate_output_file(ctx, src, extension): |
|
package = ctx.label.package |
|
if not _is_google3 and ctx.rule.attr.strip_import_prefix and ctx.rule.attr.strip_import_prefix != "/": |
|
package = package[len(ctx.rule.attr.strip_import_prefix):] |
|
real_short_path = _get_real_short_path(src) |
|
real_short_path = paths.relativize(real_short_path, package) |
|
output_filename = paths.replace_extension(real_short_path, extension) |
|
ret = ctx.actions.declare_file(output_filename) |
|
return ret |
|
|
|
# upb_proto_library / upb_proto_reflection_library shared code ################# |
|
|
|
_LuaFilesInfo = provider( |
|
"A set of lua files generated from .proto files", |
|
fields = ["files"], |
|
) |
|
|
|
def _compile_upb_protos(ctx, proto_info, proto_sources): |
|
files = [_generate_output_file(ctx, name, "_pb.lua") for name in proto_sources] |
|
transitive_sets = proto_info.transitive_descriptor_sets.to_list() |
|
ctx.actions.run( |
|
inputs = depset( |
|
direct = [proto_info.direct_descriptor_set], |
|
transitive = [proto_info.transitive_descriptor_sets], |
|
), |
|
tools = [ctx.executable._upbc], |
|
outputs = files, |
|
executable = ctx.executable._protoc, |
|
arguments = [ |
|
"--lua_out=" + _get_real_root(ctx, files[0]), |
|
"--plugin=protoc-gen-lua=" + ctx.executable._upbc.path, |
|
"--descriptor_set_in=" + ctx.configuration.host_path_separator.join([f.path for f in transitive_sets]), |
|
] + |
|
[_get_real_short_path(file) for file in proto_sources], |
|
progress_message = "Generating Lua protos for :" + ctx.label.name, |
|
) |
|
return files |
|
|
|
def _lua_proto_rule_impl(ctx): |
|
if len(ctx.attr.deps) != 1: |
|
fail("only one deps dependency allowed.") |
|
dep = ctx.attr.deps[0] |
|
if _LuaFilesInfo not in dep: |
|
fail("proto_library rule must generate _LuaFilesInfo (aspect should have handled this).") |
|
files = dep[_LuaFilesInfo].files |
|
return [ |
|
DefaultInfo( |
|
files = files, |
|
data_runfiles = ctx.runfiles(files = files.to_list()), |
|
), |
|
] |
|
|
|
def _lua_proto_library_aspect_impl(target, ctx): |
|
proto_info = target[ProtoInfo] |
|
files = _compile_upb_protos(ctx, proto_info, proto_info.direct_sources) |
|
deps = ctx.rule.attr.deps |
|
transitive = [dep[_LuaFilesInfo].files for dep in deps if _LuaFilesInfo in dep] |
|
return [_LuaFilesInfo(files = depset(direct = files, transitive = transitive))] |
|
|
|
# lua_proto_library() ########################################################## |
|
|
|
_lua_proto_library_aspect = aspect( |
|
attrs = { |
|
"_upbc": attr.label( |
|
executable = True, |
|
cfg = "exec", |
|
default = "//lua:protoc-gen-lua", |
|
), |
|
"_protoc": attr.label( |
|
executable = True, |
|
cfg = "exec", |
|
default = "//:protoc", |
|
), |
|
}, |
|
implementation = _lua_proto_library_aspect_impl, |
|
provides = [_LuaFilesInfo], |
|
attr_aspects = ["deps"], |
|
fragments = ["cpp"], |
|
) |
|
|
|
lua_proto_library = rule( |
|
implementation = _lua_proto_rule_impl, |
|
attrs = { |
|
"deps": attr.label_list( |
|
aspects = [_lua_proto_library_aspect], |
|
allow_rules = ["proto_library"], |
|
providers = [ProtoInfo], |
|
), |
|
}, |
|
)
|
|
|