1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
|
# Copyright (C) 2019 The Android Open Source Project
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
# This file defines the proto_gen() rule that is used for generating protos
# with custom plugins (ipc and protozero).
def _proto_gen_impl(ctx):
proto_src = [
f
for dep in ctx.attr.deps
for f in dep[ProtoInfo].direct_sources
]
includes = [
f
for dep in ctx.attr.deps
for f in dep[ProtoInfo].transitive_imports.to_list()
]
proto_path = "."
out_dir = str(ctx.genfiles_dir.path)
strip_base_path = ""
if ctx.attr.root != "//":
# This path is hit in Google internal builds, where root is typically
# //third_party/perfetto.
proto_path = "."
# The below will likely be //third_party/perfetto/ but may also be any
# subdir under //third_party/perfetto.
last_slash_idx = ctx.build_file_path.rfind("/")
strip_base_path = ctx.build_file_path[:last_slash_idx + 1]
elif ctx.label.workspace_root:
# This path is hit when proto targets are built as @perfetto//:xxx
# instead of //:xxx. This happens in embedder builds. In this case,
# workspace_root == "external/perfetto" and we need to rebase the paths
# passed to protoc.
proto_path = ctx.label.workspace_root
out_dir += "/" + ctx.label.workspace_root
strip_base_path = ctx.label.workspace_root + "/"
out_files = []
suffix = ctx.attr.suffix
for src in proto_src:
base_path = src.path[:-len(".proto")]
if base_path.startswith(strip_base_path):
base_path = base_path[len(strip_base_path):]
out_files += [ctx.actions.declare_file(base_path + ".%s.h" % suffix)]
out_files += [ctx.actions.declare_file(base_path + ".%s.cc" % suffix)]
arguments = [
"--proto_path=" + proto_path,
]
plugin_deps = []
if ctx.attr.plugin:
arguments += [
"--plugin=protoc-gen-plugin=" + ctx.executable.plugin.path,
"--plugin_out=wrapper_namespace=pbzero:" + out_dir,
]
plugin_deps += [ctx.executable.plugin]
else:
arguments += [
"--cpp_out=" + out_dir,
]
arguments += [src.path for src in proto_src]
ctx.actions.run(
inputs = proto_src + includes + plugin_deps,
tools = plugin_deps,
outputs = out_files,
executable = ctx.executable.protoc,
arguments = arguments,
)
return [
DefaultInfo(files = depset(out_files)),
OutputGroupInfo(
cc = depset([f for f in out_files if f.path.endswith(".cc")]),
h = depset([f for f in out_files if f.path.endswith(".h")]),
),
]
proto_gen = rule(
attrs = {
"deps": attr.label_list(
mandatory = True,
allow_empty = False,
providers = [ProtoInfo],
),
"plugin": attr.label(
executable = True,
mandatory = False,
cfg = "host",
),
"suffix": attr.string(
mandatory = True,
),
"protoc": attr.label(
executable = True,
cfg = "host",
),
"root": attr.string(
mandatory = False,
default = "//",
),
},
output_to_genfiles = True,
implementation = _proto_gen_impl,
)
|
