Index: third_party/protobuf/protobuf.bzl |
diff --git a/third_party/protobuf/protobuf.bzl b/third_party/protobuf/protobuf.bzl |
new file mode 100644 |
index 0000000000000000000000000000000000000000..bf748db664b971e933b8ae1cb3dbf0109d47caf8 |
--- /dev/null |
+++ b/third_party/protobuf/protobuf.bzl |
@@ -0,0 +1,291 @@ |
+# -*- mode: python; -*- PYTHON-PREPROCESSING-REQUIRED |
+ |
+def _GenDir(ctx): |
+ if not ctx.attr.includes: |
+ return "" |
+ if not ctx.attr.includes[0]: |
+ return ctx.label.package |
+ if not ctx.label.package: |
+ return ctx.attr.includes[0] |
+ return ctx.label.package + '/' + ctx.attr.includes[0] |
+ |
+def _CcOuts(srcs): |
+ return [s[:-len(".proto")] + ".pb.h" for s in srcs] + \ |
+ [s[:-len(".proto")] + ".pb.cc" for s in srcs] |
+ |
+def _PyOuts(srcs): |
+ return [s[:-len(".proto")] + "_pb2.py" for s in srcs] |
+ |
+def _RelativeOutputPath(path, include): |
+ if include == None: |
+ return path |
+ |
+ if not path.startswith(include): |
+ fail("Include path %s isn't part of the path %s." % (include, path)) |
+ |
+ if include and include[-1] != '/': |
+ include = include + '/' |
+ |
+ path = path[len(include):] |
+ |
+ if not path.startswith(PACKAGE_NAME): |
+ fail("The package %s is not within the path %s" % (PACKAGE_NAME, path)) |
+ |
+ if not PACKAGE_NAME: |
+ return path |
+ |
+ return path[len(PACKAGE_NAME)+1:] |
+ |
+ |
+ |
+def _proto_gen_impl(ctx): |
+ """General implementation for generating protos""" |
+ srcs = ctx.files.srcs |
+ deps = [] |
+ deps += ctx.files.srcs |
+ gen_dir = _GenDir(ctx) |
+ if gen_dir: |
+ import_flags = ["-I" + gen_dir] |
+ else: |
+ import_flags = ["-I."] |
+ |
+ for dep in ctx.attr.deps: |
+ import_flags += dep.proto.import_flags |
+ deps += dep.proto.deps |
+ |
+ args = [] |
+ if ctx.attr.gen_cc: |
+ args += ["--cpp_out=" + ctx.var["GENDIR"] + "/" + gen_dir] |
+ if ctx.attr.gen_py: |
+ args += ["--python_out=" + ctx.var["GENDIR"] + "/" + gen_dir] |
+ |
+ if args: |
+ ctx.action( |
+ inputs=srcs + deps, |
+ outputs=ctx.outputs.outs, |
+ arguments=args + import_flags + [s.path for s in srcs], |
+ executable=ctx.executable.protoc, |
+ ) |
+ |
+ return struct( |
+ proto=struct( |
+ srcs=srcs, |
+ import_flags=import_flags, |
+ deps=deps, |
+ ), |
+ ) |
+ |
+_proto_gen = rule( |
+ attrs = { |
+ "srcs": attr.label_list(allow_files = True), |
+ "deps": attr.label_list(providers = ["proto"]), |
+ "includes": attr.string_list(), |
+ "protoc": attr.label( |
+ cfg = HOST_CFG, |
+ executable = True, |
+ single_file = True, |
+ mandatory = True, |
+ ), |
+ "gen_cc": attr.bool(), |
+ "gen_py": attr.bool(), |
+ "outs": attr.output_list(), |
+ }, |
+ output_to_genfiles = True, |
+ implementation = _proto_gen_impl, |
+) |
+ |
+def cc_proto_library( |
+ name, |
+ srcs=[], |
+ deps=[], |
+ cc_libs=[], |
+ include=None, |
+ protoc="//google/protobuf:protoc", |
+ internal_bootstrap_hack=False, |
+ default_runtime="//google/protobuf:protobuf", |
+ **kargs): |
+ """Bazel rule to create a C++ protobuf library from proto source files |
+ |
+ NOTE: the rule is only an internal workaround to generate protos. The |
+ interface may change and the rule may be removed when bazel has introduced |
+ the native rule. |
+ |
+ Args: |
+ name: the name of the cc_proto_library. |
+ srcs: the .proto files of the cc_proto_library. |
+ deps: a list of dependency labels; must be cc_proto_library. |
+ cc_libs: a list of other cc_library targets depended by the generated |
+ cc_library. |
+ include: a string indicating the include path of the .proto files. |
+ protoc: the label of the protocol compiler to generate the sources. |
+ internal_bootstrap_hack: a flag indicate the cc_proto_library is used only |
+ for bootstraping. When it is set to True, no files will be generated. |
+ The rule will simply be a provider for .proto files, so that other |
+ cc_proto_library can depend on it. |
+ default_runtime: the implicitly default runtime which will be depended on by |
+ the generated cc_library target. |
+ **kargs: other keyword arguments that are passed to cc_library. |
+ |
+ """ |
+ |
+ includes = [] |
+ if include != None: |
+ includes = [include] |
+ |
+ if internal_bootstrap_hack: |
+ # For pre-checked-in generated files, we add the internal_bootstrap_hack |
+ # which will skip the codegen action. |
+ _proto_gen( |
+ name=name + "_genproto", |
+ srcs=srcs, |
+ deps=[s + "_genproto" for s in deps], |
+ includes=includes, |
+ protoc=protoc, |
+ visibility=["//visibility:public"], |
+ ) |
+ # An empty cc_library to make rule dependency consistent. |
+ native.cc_library( |
+ name=name, |
+ **kargs) |
+ return |
+ |
+ outs = _CcOuts(srcs) |
+ _proto_gen( |
+ name=name + "_genproto", |
+ srcs=srcs, |
+ deps=[s + "_genproto" for s in deps], |
+ includes=includes, |
+ protoc=protoc, |
+ gen_cc=1, |
+ outs=outs, |
+ visibility=["//visibility:public"], |
+ ) |
+ |
+ if default_runtime and not default_runtime in cc_libs: |
+ cc_libs += [default_runtime] |
+ |
+ native.cc_library( |
+ name=name, |
+ srcs=outs, |
+ deps=cc_libs + deps, |
+ includes=includes, |
+ **kargs) |
+ |
+ |
+def internal_copied_filegroup( |
+ name, |
+ srcs, |
+ include, |
+ **kargs): |
+ """Bazel rule to fix sources file to workaround with python path issues. |
+ |
+ Args: |
+ name: the name of the internal_copied_filegroup rule, which will be the |
+ name of the generated filegroup. |
+ srcs: the source files to be copied. |
+ include: the expected import root of the source. |
+ **kargs: extra arguments that will be passed into the filegroup. |
+ """ |
+ outs = [_RelativeOutputPath(s, include) for s in srcs] |
+ |
+ native.genrule( |
+ name=name+"_genrule", |
+ srcs=srcs, |
+ outs=outs, |
+ cmd=" && ".join(["cp $(location %s) $(location %s)" % |
+ (s, _RelativeOutputPath(s, include)) |
+ for s in srcs])) |
+ |
+ native.filegroup( |
+ name=name, |
+ srcs=outs, |
+ **kargs) |
+ |
+ |
+def py_proto_library( |
+ name, |
+ srcs=[], |
+ deps=[], |
+ py_libs=[], |
+ py_extra_srcs=[], |
+ include=None, |
+ default_runtime="//google/protobuf:protobuf_python", |
+ protoc="//google/protobuf:protoc", |
+ **kargs): |
+ """Bazel rule to create a Python protobuf library from proto source files |
+ |
+ NOTE: the rule is only an internal workaround to generate protos. The |
+ interface may change and the rule may be removed when bazel has introduced |
+ the native rule. |
+ |
+ Args: |
+ name: the name of the py_proto_library. |
+ srcs: the .proto files of the py_proto_library. |
+ deps: a list of dependency labels; must be py_proto_library. |
+ py_libs: a list of other py_library targets depended by the generated |
+ py_library. |
+ py_extra_srcs: extra source files that will be added to the output |
+ py_library. This attribute is used for internal bootstrapping. |
+ include: a string indicating the include path of the .proto files. |
+ default_runtime: the implicitly default runtime which will be depended on by |
+ the generated py_library target. |
+ protoc: the label of the protocol compiler to generate the sources. |
+ **kargs: other keyword arguments that are passed to cc_library. |
+ |
+ """ |
+ outs = _PyOuts(srcs) |
+ |
+ includes = [] |
+ if include != None: |
+ includes = [include] |
+ |
+ _proto_gen( |
+ name=name + "_genproto", |
+ srcs=srcs, |
+ deps=[s + "_genproto" for s in deps], |
+ includes=includes, |
+ protoc=protoc, |
+ gen_py=1, |
+ outs=outs, |
+ visibility=["//visibility:public"], |
+ ) |
+ |
+ if include != None: |
+ # Copy the output files to the desired location to make the import work. |
+ internal_copied_filegroup_name=name + "_internal_copied_filegroup" |
+ internal_copied_filegroup( |
+ name=internal_copied_filegroup_name, |
+ srcs=outs, |
+ include=include) |
+ outs=[internal_copied_filegroup_name] |
+ |
+ if default_runtime and not default_runtime in py_libs + deps: |
+ py_libs += [default_runtime] |
+ |
+ native.py_library( |
+ name=name, |
+ srcs=outs+py_extra_srcs, |
+ deps=py_libs+deps, |
+ **kargs) |
+ |
+def internal_protobuf_py_tests( |
+ name, |
+ modules=[], |
+ **kargs): |
+ """Bazel rules to create batch tests for protobuf internal. |
+ |
+ Args: |
+ name: the name of the rule. |
+ modules: a list of modules for tests. The macro will create a py_test for |
+ each of the parameter with the source "google/protobuf/%s.py" |
+ kargs: extra parameters that will be passed into the py_test. |
+ |
+ """ |
+ for m in modules: |
+ s = _RelativeOutputPath( |
+ "python/google/protobuf/internal/%s.py" % m, "python") |
+ native.py_test( |
+ name="py_%s" % m, |
+ srcs=[s], |
+ main=s, |
+ **kargs) |