205 lines
7.3 KiB
Python
205 lines
7.3 KiB
Python
"""Generated an open-api spec for a grpc api spec.
|
|
|
|
Reads the the api spec in protobuf format and generate an open-api spec.
|
|
Optionally applies settings from the grpc-service configuration.
|
|
"""
|
|
|
|
load("@rules_proto//proto:defs.bzl", "ProtoInfo")
|
|
|
|
# TODO(yannic): Replace with |proto_common.direct_source_infos| when
|
|
# https://github.com/bazelbuild/rules_proto/pull/22 lands.
|
|
def _direct_source_infos(proto_info, provided_sources = []):
|
|
"""Returns sequence of `ProtoFileInfo` for `proto_info`'s direct sources.
|
|
|
|
Files that are both in `proto_info`'s direct sources and in
|
|
`provided_sources` are skipped. This is useful, e.g., for well-known
|
|
protos that are already provided by the Protobuf runtime.
|
|
|
|
Args:
|
|
proto_info: An instance of `ProtoInfo`.
|
|
provided_sources: Optional. A sequence of files to ignore.
|
|
Usually, these files are already provided by the
|
|
Protocol Buffer runtime (e.g. Well-Known protos).
|
|
|
|
Returns: A sequence of `ProtoFileInfo` containing information about
|
|
`proto_info`'s direct sources.
|
|
"""
|
|
|
|
source_root = proto_info.proto_source_root
|
|
if "." == source_root:
|
|
return [struct(file = src, import_path = src.path) for src in proto_info.direct_sources]
|
|
|
|
offset = len(source_root) + 1 # + '/'.
|
|
|
|
infos = []
|
|
for src in proto_info.direct_sources:
|
|
# TODO(yannic): Remove this hack when we drop support for Bazel < 1.0.
|
|
local_offset = offset
|
|
if src.root.path and not source_root.startswith(src.root.path):
|
|
# Before Bazel 1.0, `proto_source_root` wasn't guaranteed to be a
|
|
# prefix of `src.path`. This could happend, e.g., if `file` was
|
|
# generated (https://github.com/bazelbuild/bazel/issues/9215).
|
|
local_offset += len(src.root.path) + 1 # + '/'.
|
|
infos.append(struct(file = src, import_path = src.path[local_offset:]))
|
|
|
|
return infos
|
|
|
|
def _run_proto_gen_swagger(
|
|
actions,
|
|
proto_info,
|
|
target_name,
|
|
transitive_proto_srcs,
|
|
protoc,
|
|
protoc_gen_swagger,
|
|
grpc_api_configuration,
|
|
single_output,
|
|
json_names_for_fields,
|
|
fqn_for_swagger_name):
|
|
args = actions.args()
|
|
|
|
args.add("--plugin", "protoc-gen-swagger=%s" % protoc_gen_swagger.path)
|
|
|
|
args.add("--swagger_opt", "logtostderr=true")
|
|
args.add("--swagger_opt", "allow_repeated_fields_in_body=true")
|
|
|
|
extra_inputs = []
|
|
if grpc_api_configuration:
|
|
extra_inputs.append(grpc_api_configuration)
|
|
args.add("--swagger_opt", "grpc_api_configuration=%s" % grpc_api_configuration.path)
|
|
|
|
if json_names_for_fields:
|
|
args.add("--swagger_opt", "json_names_for_fields=true")
|
|
|
|
if fqn_for_swagger_name:
|
|
args.add("--swagger_opt", "fqn_for_swagger_name=true")
|
|
|
|
proto_file_infos = _direct_source_infos(proto_info)
|
|
|
|
# TODO(yannic): Use |proto_info.transitive_descriptor_sets| when
|
|
# https://github.com/bazelbuild/bazel/issues/9337 is fixed.
|
|
args.add_all(proto_info.transitive_proto_path, format_each = "--proto_path=%s")
|
|
|
|
if single_output:
|
|
args.add("--swagger_opt", "allow_merge=true")
|
|
args.add("--swagger_opt", "merge_file_name=%s" % target_name)
|
|
|
|
swagger_file = actions.declare_file("%s.swagger.json" % target_name)
|
|
args.add("--swagger_out", swagger_file.dirname)
|
|
|
|
args.add_all([f.import_path for f in proto_file_infos])
|
|
|
|
actions.run(
|
|
executable = protoc,
|
|
tools = [protoc_gen_swagger],
|
|
inputs = depset(
|
|
direct = extra_inputs,
|
|
transitive = [transitive_proto_srcs],
|
|
),
|
|
outputs = [swagger_file],
|
|
arguments = [args],
|
|
)
|
|
|
|
return [swagger_file]
|
|
|
|
# TODO(yannic): We may be able to generate all files in a single action,
|
|
# but that will change at least the semantics of `use_go_template.proto`.
|
|
swagger_files = []
|
|
for proto_file_info in proto_file_infos:
|
|
# TODO(yannic): This probably doesn't work as expected: we only add this
|
|
# option after we have seen it, so `.proto` sources that happen to be
|
|
# in the list of `.proto` files before `use_go_template.proto` will be
|
|
# compiled without this option, and all sources that get compiled after
|
|
# `use_go_template.proto` will have this option on.
|
|
if proto_file_info.file.basename == "use_go_template.proto":
|
|
args.add("--swagger_opt", "use_go_templates=true")
|
|
|
|
file_name = "%s.swagger.json" % proto_file_info.import_path[:-len(".proto")]
|
|
swagger_file = actions.declare_file(
|
|
"_virtual_imports/%s/%s" % (target_name, file_name),
|
|
)
|
|
|
|
file_args = actions.args()
|
|
|
|
offset = len(file_name) + 1 # + '/'.
|
|
file_args.add("--swagger_out", swagger_file.path[:-offset])
|
|
|
|
file_args.add(proto_file_info.import_path)
|
|
|
|
actions.run(
|
|
executable = protoc,
|
|
tools = [protoc_gen_swagger],
|
|
inputs = depset(
|
|
direct = extra_inputs,
|
|
transitive = [transitive_proto_srcs],
|
|
),
|
|
outputs = [swagger_file],
|
|
arguments = [args, file_args],
|
|
)
|
|
swagger_files.append(swagger_file)
|
|
|
|
return swagger_files
|
|
|
|
def _proto_gen_swagger_impl(ctx):
|
|
proto = ctx.attr.proto[ProtoInfo]
|
|
return [
|
|
DefaultInfo(
|
|
files = depset(
|
|
_run_proto_gen_swagger(
|
|
actions = ctx.actions,
|
|
proto_info = proto,
|
|
target_name = ctx.attr.name,
|
|
transitive_proto_srcs = depset(
|
|
direct = ctx.files._well_known_protos,
|
|
transitive = [proto.transitive_sources],
|
|
),
|
|
protoc = ctx.executable._protoc,
|
|
protoc_gen_swagger = ctx.executable._protoc_gen_swagger,
|
|
grpc_api_configuration = ctx.file.grpc_api_configuration,
|
|
single_output = ctx.attr.single_output,
|
|
json_names_for_fields = ctx.attr.json_names_for_fields,
|
|
fqn_for_swagger_name = ctx.attr.fqn_for_swagger_name,
|
|
),
|
|
),
|
|
),
|
|
]
|
|
|
|
protoc_gen_swagger = rule(
|
|
attrs = {
|
|
"proto": attr.label(
|
|
mandatory = True,
|
|
providers = [ProtoInfo],
|
|
),
|
|
"grpc_api_configuration": attr.label(
|
|
allow_single_file = True,
|
|
mandatory = False,
|
|
),
|
|
"single_output": attr.bool(
|
|
default = False,
|
|
mandatory = False,
|
|
),
|
|
"json_names_for_fields": attr.bool(
|
|
default = False,
|
|
mandatory = False,
|
|
),
|
|
"fqn_for_swagger_name": attr.bool(
|
|
default = False,
|
|
mandatory = False,
|
|
),
|
|
"_protoc": attr.label(
|
|
default = "@com_google_protobuf//:protoc",
|
|
executable = True,
|
|
cfg = "host",
|
|
),
|
|
"_well_known_protos": attr.label(
|
|
default = "@com_google_protobuf//:well_known_protos",
|
|
allow_files = True,
|
|
),
|
|
"_protoc_gen_swagger": attr.label(
|
|
default = Label("//protoc-gen-swagger:protoc-gen-swagger"),
|
|
executable = True,
|
|
cfg = "host",
|
|
),
|
|
},
|
|
implementation = _proto_gen_swagger_impl,
|
|
)
|