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
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
|
# This Source Code Form is subject to the terms of the Mozilla Public
# License, v. 2.0. If a copy of the MPL was not distributed with this
# file, You can obtain one at https://mozilla.org/MPL/2.0/.
import hashlib
from os.path import dirname
from pathlib import Path
import jinja2
from buildconfig import config, topsrcdir # type: ignore
from mozbuild.util import memoize # type: ignore
from schema_parser import parse_and_validate
THIS_DIR = Path(dirname(__file__))
TEMPLATES = THIS_DIR / "templates"
@memoize
def get_deps():
# Any imported python module is added as a dependency automatically,
# so we only need the templates.
return {
*[str(p) for p in TEMPLATES.iterdir()],
str(THIS_DIR / "gecko-trace.schema.yaml"),
}
def generate_cpp_events(output_fd, *inputs):
"""Generate C++ events header"""
events = parse_and_validate(inputs if inputs else load_schema_index())
template = _jinja2_env().get_template("GeckoTraceEvents.h.jinja2")
output_fd.write(
template.render(
events=events,
enabled=config.substs.get("GECKO_TRACE_ENABLE", False),
# Generate a unique hash to prevent include guard conflicts when
# multiple event files are generated and included together (e.g., in gtests).
# This ensures each generated header has a distinct include guard.
input_hash=hashlib.sha256("".join(inputs).encode())
.hexdigest()
.upper()[:15],
)
)
return get_deps().union(load_schema_index() if not inputs else {})
def generate_glean_metrics(output_fd, *inputs):
events = parse_and_validate(inputs if inputs else load_schema_index())
template = _jinja2_env().get_template("generated-metrics.yaml.jinja2")
output_fd.write(
template.render(
events=events,
)
)
def generate_glean_adapter(output_fd, *inputs):
events = parse_and_validate(inputs if inputs else load_schema_index())
template = _jinja2_env().get_template("glean_adapter.rs.jinja2")
output_fd.write(
template.render(
events=events,
)
)
return get_deps().union(load_schema_index() if not inputs else {})
@memoize
def load_schema_index():
index = THIS_DIR.parent / "index.py"
with open(index) as f:
index_src = f.read()
global gecko_trace_files
exec(index_src, globals())
return [str(Path(topsrcdir) / x) for x in gecko_trace_files] # type: ignore
@memoize
def _jinja2_env():
from jinja2.exceptions import TemplateRuntimeError
from jinja2.ext import Extension
class RaiseExtension(Extension):
tags = set(["raise"])
def parse(self, parser):
lineno = next(parser.stream).lineno
message_node = parser.parse_expression()
return jinja2.nodes.CallBlock(
self.call_method("_raise", [message_node], lineno=lineno),
[],
[],
[],
lineno=lineno,
)
def _raise(self, msg, caller):
raise TemplateRuntimeError(msg)
def camelize(name):
"""Convert snake_case to PascalCase"""
return "".join(
word.capitalize()
for word in name.replace("-", "_").replace(".", "_").split("_")
)
def snake_case(name):
"""Convert name to snake_case"""
return name.replace(".", "_").replace("-", "_").lower()
def debug(value):
print(value)
return value
env = jinja2.Environment(
loader=jinja2.FileSystemLoader(TEMPLATES),
trim_blocks=True,
lstrip_blocks=True,
extensions=[RaiseExtension],
)
env.filters["camelize"] = camelize
env.filters["snake_case"] = snake_case
env.filters["debug"] = debug
return env
def main(*args): # mach requires this
pass
|
