mirror of https://github.com/grpc/grpc.git
commit
9383823811
8 changed files with 145 additions and 12 deletions
@ -0,0 +1,128 @@ |
||||
# Copyright 2015, Google Inc. |
||||
# All rights reserved. |
||||
# |
||||
# Redistribution and use in source and binary forms, with or without |
||||
# modification, are permitted provided that the following conditions are |
||||
# met: |
||||
# |
||||
# * Redistributions of source code must retain the above copyright |
||||
# notice, this list of conditions and the following disclaimer. |
||||
# * Redistributions in binary form must reproduce the above |
||||
# copyright notice, this list of conditions and the following disclaimer |
||||
# in the documentation and/or other materials provided with the |
||||
# distribution. |
||||
# * Neither the name of Google Inc. nor the names of its |
||||
# contributors may be used to endorse or promote products derived from |
||||
# this software without specific prior written permission. |
||||
# |
||||
# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS |
||||
# "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT |
||||
# LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR |
||||
# A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT |
||||
# OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, |
||||
# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT |
||||
# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, |
||||
# DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY |
||||
# THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT |
||||
# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE |
||||
# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
||||
|
||||
""" |
||||
Bazel macros to declare gRPC libraries automatically generated from proto files. |
||||
|
||||
This file declares two macros: |
||||
- objc_proto_library |
||||
- objc_grpc_library |
||||
""" |
||||
|
||||
def _lower_underscore_to_upper_camel(str): |
||||
humps = [] |
||||
for hump in str.split('_'): |
||||
humps += [hump[0].upper() + hump[1:]] |
||||
return "".join(humps) |
||||
|
||||
def _file_to_upper_camel(src): |
||||
elements = src.rpartition('/') |
||||
upper_camel = _lower_underscore_to_upper_camel(elements[-1]) |
||||
return "".join(elements[:-1] + [upper_camel]) |
||||
|
||||
def _file_with_extension(src, ext): |
||||
elements = src.rpartition('/') |
||||
basename = elements[-1].partition('.')[0] |
||||
return "".join(elements[:-1] + [basename, ext]) |
||||
|
||||
def _protoc_invocation(srcs, flags): |
||||
"""Returns a command line to invoke protoc from a genrule, on the given |
||||
sources, using the given flags. |
||||
""" |
||||
protoc_command = "$(location //external:protoc) -I . " |
||||
srcs_params = "" |
||||
for src in srcs: |
||||
srcs_params += " $(location %s)" % (src) |
||||
return protoc_command + flags + srcs_params |
||||
|
||||
def objc_proto_library(name, srcs, visibility=None): |
||||
"""Declares an objc_library for the code generated by protoc from the given |
||||
proto sources. This generated code doesn't include proto services. |
||||
""" |
||||
h_files = [] |
||||
m_files = [] |
||||
for src in srcs: |
||||
src = _file_to_upper_camel(src) |
||||
h_files += [_file_with_extension(src, ".pbobjc.h")] |
||||
m_files += [_file_with_extension(src, ".pbobjc.m")] |
||||
|
||||
protoc_flags = "--objc_out=$(GENDIR)" |
||||
|
||||
native.genrule( |
||||
name = name + "_codegen", |
||||
srcs = srcs + ["//external:protoc"], |
||||
outs = h_files + m_files, |
||||
cmd = _protoc_invocation(srcs, protoc_flags), |
||||
) |
||||
native.objc_library( |
||||
name = name, |
||||
hdrs = h_files, |
||||
includes = ["."], |
||||
non_arc_srcs = m_files, |
||||
deps = ["//external:protobuf_objc"], |
||||
visibility = visibility, |
||||
) |
||||
|
||||
def objc_grpc_library(name, services, other_messages, visibility=None): |
||||
"""Declares an objc_library for the code generated by gRPC and protoc from the |
||||
given proto sources (services and other_messages). The generated code doesn't |
||||
include proto services of the files passed as other_messages. |
||||
""" |
||||
objc_proto_library(name + "_messages", services + other_messages) |
||||
|
||||
h_files = [] |
||||
m_files = [] |
||||
for src in services: |
||||
src = _file_to_upper_camel(src) |
||||
h_files += [_file_with_extension(src, ".pbrpc.h")] |
||||
m_files += [_file_with_extension(src, ".pbrpc.m")] |
||||
|
||||
protoc_flags = ("--grpc_out=$(GENDIR) --plugin=" + |
||||
"protoc-gen-grpc=$(location //external:grpc_protoc_plugin_objc)") |
||||
|
||||
native.genrule( |
||||
name = name + "_codegen", |
||||
srcs = services + [ |
||||
"//external:grpc_protoc_plugin_objc", |
||||
"//external:protoc", |
||||
], |
||||
outs = h_files + m_files, |
||||
cmd = _protoc_invocation(services, protoc_flags), |
||||
) |
||||
native.objc_library( |
||||
name = name, |
||||
hdrs = h_files, |
||||
includes = ["."], |
||||
srcs = m_files, |
||||
deps = [ |
||||
":" + name + "_messages", |
||||
"//external:proto_objc_rpc", |
||||
], |
||||
visibility = visibility, |
||||
) |
Loading…
Reference in new issue