mirror of
https://github.com/genuinetools/reg.git
synced 2024-07-04 12:34:17 -04:00
113 lines
3.6 KiB
Python
113 lines
3.6 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.
|
|
"""
|
|
|
|
def _collect_includes(gen_dir, srcs):
|
|
"""Build an include path mapping.
|
|
|
|
It is important to not just collect unique dirnames, to also support
|
|
proto files of the same name from different packages.
|
|
|
|
The implementation below is similar to what bazel does in its
|
|
ProtoCompileActionBuilder.java
|
|
"""
|
|
includes = []
|
|
for src in srcs:
|
|
ref_path = src.path
|
|
|
|
if ref_path.startswith(gen_dir):
|
|
ref_path = ref_path[len(gen_dir):].lstrip("/")
|
|
|
|
if src.owner.workspace_root:
|
|
workspace_root = src.owner.workspace_root
|
|
ref_path = ref_path[len(workspace_root):].lstrip("/")
|
|
|
|
include = ref_path + "=" + src.path
|
|
if include not in includes:
|
|
includes.append(include)
|
|
|
|
return includes
|
|
|
|
def _run_proto_gen_swagger(ctx, direct_proto_srcs, transitive_proto_srcs, actions, protoc, protoc_gen_swagger, grpc_api_configuration):
|
|
swagger_files = []
|
|
for proto in direct_proto_srcs:
|
|
swagger_file = actions.declare_file(
|
|
"%s.swagger.json" % proto.basename[:-len(".proto")],
|
|
sibling = proto,
|
|
)
|
|
|
|
inputs = direct_proto_srcs + transitive_proto_srcs + [protoc_gen_swagger]
|
|
|
|
options=["logtostderr=true"]
|
|
if grpc_api_configuration:
|
|
options.append("grpc_api_configuration=%s" % grpc_api_configuration.path)
|
|
inputs.append(grpc_api_configuration)
|
|
|
|
includes = _collect_includes(ctx.genfiles_dir.path, direct_proto_srcs + transitive_proto_srcs)
|
|
|
|
args = actions.args()
|
|
args.add("--plugin=%s" % protoc_gen_swagger.path)
|
|
args.add("--swagger_out=%s:%s" % (",".join(options), ctx.bin_dir.path))
|
|
args.add(["-I%s" % include for include in includes])
|
|
args.add(proto.path)
|
|
|
|
actions.run(
|
|
executable = protoc,
|
|
inputs = inputs,
|
|
outputs = [swagger_file],
|
|
arguments = [args],
|
|
)
|
|
|
|
swagger_files.append(swagger_file)
|
|
|
|
return swagger_files
|
|
|
|
def _proto_gen_swagger_impl(ctx):
|
|
proto = ctx.attr.proto.proto
|
|
grpc_api_configuration = ctx.file.grpc_api_configuration
|
|
|
|
return struct(
|
|
files=depset(
|
|
_run_proto_gen_swagger(
|
|
ctx,
|
|
direct_proto_srcs = proto.direct_sources,
|
|
transitive_proto_srcs = ctx.files._well_known_protos + proto.transitive_sources.to_list(),
|
|
actions = ctx.actions,
|
|
protoc = ctx.executable._protoc,
|
|
protoc_gen_swagger = ctx.executable._protoc_gen_swagger,
|
|
grpc_api_configuration = grpc_api_configuration
|
|
)
|
|
)
|
|
)
|
|
|
|
protoc_gen_swagger = rule(
|
|
attrs = {
|
|
"proto": attr.label(
|
|
allow_rules = ["proto_library"],
|
|
mandatory = True,
|
|
providers = ['proto'],
|
|
),
|
|
"grpc_api_configuration": attr.label(
|
|
allow_single_file=True,
|
|
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,
|
|
)
|