Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(10)

Side by Side Diff: third_party/inspector_protocol/CodeGenerator.py

Issue 2493063005: [DevTools] Use inspector_protocol generator in content/browser/devtools. (Closed)
Patch Set: Created 4 years, 1 month ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch
OLDNEW
1 # Copyright 2016 The Chromium Authors. All rights reserved. 1 # Copyright 2016 The Chromium Authors. All rights reserved.
2 # Use of this source code is governed by a BSD-style license that can be 2 # Use of this source code is governed by a BSD-style license that can be
3 # found in the LICENSE file. 3 # found in the LICENSE file.
4 4
5 import os.path 5 import os.path
6 import sys 6 import sys
7 import optparse 7 import optparse
8 import collections 8 import collections
9 import functools 9 import functools
10 try: 10 try:
(...skipping 61 matching lines...) Expand 10 before | Expand all | Expand 10 after
72 exc = sys.exc_info()[1] 72 exc = sys.exc_info()[1]
73 sys.stderr.write("Failed to parse command-line arguments: %s\n\n" % exc) 73 sys.stderr.write("Failed to parse command-line arguments: %s\n\n" % exc)
74 exit(1) 74 exit(1)
75 75
76 try: 76 try:
77 config_json_file = open(config_file, "r") 77 config_json_file = open(config_file, "r")
78 config_json_string = config_json_file.read() 78 config_json_string = config_json_file.read()
79 config_partial = json_to_object(config_json_string, output_base, config_ base) 79 config_partial = json_to_object(config_json_string, output_base, config_ base)
80 config_json_file.close() 80 config_json_file.close()
81 defaults = { 81 defaults = {
82 ".use_snake_file_names": False,
83 ".use_title_case_methods": False,
82 ".imported": False, 84 ".imported": False,
83 ".imported.export_macro": "", 85 ".imported.export_macro": "",
84 ".imported.export_header": False, 86 ".imported.export_header": False,
85 ".imported.header": False, 87 ".imported.header": False,
86 ".imported.package": False, 88 ".imported.package": False,
87 ".protocol.export_macro": "", 89 ".protocol.export_macro": "",
88 ".protocol.export_header": False, 90 ".protocol.export_header": False,
89 ".protocol.options": False, 91 ".protocol.options": False,
90 ".exported": False, 92 ".exported": False,
91 ".exported.export_macro": "", 93 ".exported.export_macro": "",
(...skipping 19 matching lines...) Expand all
111 113
112 114
113 def dash_to_camelcase(word): 115 def dash_to_camelcase(word):
114 prefix = "" 116 prefix = ""
115 if word[0] == "-": 117 if word[0] == "-":
116 prefix = "Negative" 118 prefix = "Negative"
117 word = word[1:] 119 word = word[1:]
118 return prefix + "".join(to_title_case(x) or "-" for x in word.split("-")) 120 return prefix + "".join(to_title_case(x) or "-" for x in word.split("-"))
119 121
120 122
121 def initialize_jinja_env(jinja_dir, cache_dir): 123 def to_snake_case(name):
caseq 2016/11/12 02:17:04 return re.sub(r'(?<=[a-z])([A-Z])', r'_\1', name,
dgozman 2016/11/14 18:40:28 Done.
124 result = []
125 last = "A"
126 for char in name:
127 if char.isupper() and not last.isupper():
128 result.append("_")
129 result.append(char.lower())
130 return "".join(result)
131
132
133 def to_method_case(config, name):
134 if config.use_title_case_methods:
135 return to_title_case(name)
136 return name
137
138
139 def initialize_jinja_env(jinja_dir, cache_dir, config):
122 # pylint: disable=F0401 140 # pylint: disable=F0401
123 sys.path.insert(1, os.path.abspath(jinja_dir)) 141 sys.path.insert(1, os.path.abspath(jinja_dir))
124 import jinja2 142 import jinja2
125 143
126 jinja_env = jinja2.Environment( 144 jinja_env = jinja2.Environment(
127 loader=jinja2.FileSystemLoader(module_path), 145 loader=jinja2.FileSystemLoader(module_path),
128 # Bytecode cache is not concurrency-safe unless pre-cached: 146 # Bytecode cache is not concurrency-safe unless pre-cached:
129 # if pre-cached this is read-only, but writing creates a race condition. 147 # if pre-cached this is read-only, but writing creates a race condition.
130 bytecode_cache=jinja2.FileSystemBytecodeCache(cache_dir), 148 bytecode_cache=jinja2.FileSystemBytecodeCache(cache_dir),
131 keep_trailing_newline=True, # newline-terminate generated files 149 keep_trailing_newline=True, # newline-terminate generated files
132 lstrip_blocks=True, # so can indent control flow tags 150 lstrip_blocks=True, # so can indent control flow tags
133 trim_blocks=True) 151 trim_blocks=True)
134 jinja_env.filters.update({"to_title_case": to_title_case, "dash_to_camelcase ": dash_to_camelcase}) 152 jinja_env.filters.update({"to_title_case": to_title_case, "dash_to_camelcase ": dash_to_camelcase, "to_method_case": functools.partial(to_method_case, config )})
135 jinja_env.add_extension("jinja2.ext.loopcontrols") 153 jinja_env.add_extension("jinja2.ext.loopcontrols")
136 return jinja_env 154 return jinja_env
137 155
138 156
139 def patch_full_qualified_refs(protocol): 157 def patch_full_qualified_refs(protocol):
140 def patch_full_qualified_refs_in_domain(json, domain_name): 158 def patch_full_qualified_refs_in_domain(json, domain_name):
141 if isinstance(json, list): 159 if isinstance(json, list):
142 for item in json: 160 for item in json:
143 patch_full_qualified_refs_in_domain(item, domain_name) 161 patch_full_qualified_refs_in_domain(item, domain_name)
144 162
145 if not isinstance(json, dict): 163 if not isinstance(json, dict):
146 return 164 return
147 for key in json: 165 for key in json:
148 if key == "type" and json[key] == "string": 166 if key == "type" and json[key] == "string":
149 json[key] = domain_name + ".string" 167 json[key] = domain_name + ".string"
150 if key != "$ref": 168 if key != "$ref":
151 patch_full_qualified_refs_in_domain(json[key], domain_name) 169 patch_full_qualified_refs_in_domain(json[key], domain_name)
152 continue 170 continue
153 if json["$ref"].find(".") == -1: 171 if json["$ref"].find(".") == -1:
154 json["$ref"] = domain_name + "." + json["$ref"] 172 json["$ref"] = domain_name + "." + json["$ref"]
155 return 173 return
156 174
157 for domain in protocol.json_api["domains"]: 175 for domain in protocol.json_api["domains"]:
158 patch_full_qualified_refs_in_domain(domain, domain["domain"]) 176 patch_full_qualified_refs_in_domain(domain, domain["domain"])
159 177
160 178
161 def calculate_exports(protocol): 179 def calculate_imports_and_exports(config, protocol):
162 def calculate_exports_in_json(json_value): 180 def has_exports(json_value, clear):
163 has_exports = False 181 result = False
164 if isinstance(json_value, list): 182 if isinstance(json_value, list):
165 for item in json_value: 183 for item in json_value:
166 has_exports = calculate_exports_in_json(item) or has_exports 184 result = has_exports(item, clear) or result
167 if isinstance(json_value, dict): 185 if isinstance(json_value, dict):
168 has_exports = ("exported" in json_value and json_value["exported"]) or has_exports 186 if "exported" in json_value and json_value["exported"]:
187 result = True
188 if "exported" in json_value and clear:
189 del json_value["exported"]
169 for key in json_value: 190 for key in json_value:
170 has_exports = calculate_exports_in_json(json_value[key]) or has_ exports 191 result = has_exports(json_value[key], clear) or result
171 return has_exports 192 return result
172 193
173 protocol.json_api["has_exports"] = False 194 imported_domains = protocol.imported_domains
195 if config.protocol.options:
196 protocol.generate_domains = [rule.domain for rule in config.protocol.opt ions]
197 imported_domains = []
198 for domain in protocol.imported_domains:
caseq 2016/11/12 02:17:04 imported_domains = list(set(protocol.imported_doma
dgozman 2016/11/14 18:40:28 Done.
199 if domain not in protocol.generate_domains:
200 imported_domains.append(domain)
201 exported_domains = protocol.generate_domains
202
203 protocol.imported_domains = []
204 protocol.exported_domains = []
174 for domain_json in protocol.json_api["domains"]: 205 for domain_json in protocol.json_api["domains"]:
175 domain_json["has_exports"] = calculate_exports_in_json(domain_json) 206 domain = domain_json["domain"]
176 if domain_json["has_exports"] and domain_json["domain"] in protocol.gene rate_domains: 207 clear = domain not in exported_domains and domain not in imported_domain s
177 protocol.json_api["has_exports"] = True 208 if not has_exports(domain_json, clear):
209 continue
210 if domain in exported_domains:
211 protocol.exported_domains.append(domain)
212 if domain in imported_domains:
213 protocol.imported_domains.append(domain)
178 214
179 215
180 def create_imported_type_definition(domain_name, type, imported_namespace): 216 def create_imported_type_definition(domain_name, type, imported_namespace):
181 # pylint: disable=W0622 217 # pylint: disable=W0622
182 return { 218 return {
183 "return_type": "std::unique_ptr<%s::%s::API::%s>" % (imported_namespace, domain_name, type["id"]), 219 "return_type": "std::unique_ptr<%s::%s::API::%s>" % (imported_namespace, domain_name, type["id"]),
184 "pass_type": "std::unique_ptr<%s::%s::API::%s>" % (imported_namespace, d omain_name, type["id"]), 220 "pass_type": "std::unique_ptr<%s::%s::API::%s>" % (imported_namespace, d omain_name, type["id"]),
185 "to_raw_type": "%s.get()", 221 "to_raw_type": "%s.get()",
186 "to_pass_type": "std::move(%s)", 222 "to_pass_type": "std::move(%s)",
187 "to_rvalue": "std::move(%s)", 223 "to_rvalue": "std::move(%s)",
(...skipping 199 matching lines...) Expand 10 before | Expand all | Expand 10 after
387 423
388 def generate_disable(protocol, config, domain): 424 def generate_disable(protocol, config, domain):
389 if "commands" not in domain: 425 if "commands" not in domain:
390 return True 426 return True
391 for command in domain["commands"]: 427 for command in domain["commands"]:
392 if command["name"] == "disable" and generate_command(protocol, config, d omain["domain"], "disable"): 428 if command["name"] == "disable" and generate_command(protocol, config, d omain["domain"], "disable"):
393 return False 429 return False
394 return True 430 return True
395 431
396 432
397 def format_include(header): 433 def format_include(config, header, file_name=None):
398 return "\"" + header + "\"" if header[0] not in "<\"" else header 434 if file_name is not None:
435 header = header + "/" + file_name + ".h"
436 header = "\"" + header + "\"" if header[0] not in "<\"" else header
437 if config.use_snake_file_names:
438 header = to_snake_case(header)
439 return header
440
441
442 def to_file_name(config, file_name):
443 if config.use_snake_file_names:
444 return to_snake_case(file_name).replace(".cpp", ".cc")
445 return file_name
399 446
400 447
401 def read_protocol_file(file_name, json_api): 448 def read_protocol_file(file_name, json_api):
402 input_file = open(file_name, "r") 449 input_file = open(file_name, "r")
403 json_string = input_file.read() 450 json_string = input_file.read()
404 input_file.close() 451 input_file.close()
405 parsed_json = json.loads(json_string) 452 parsed_json = json.loads(json_string)
406 version = parsed_json["version"]["major"] + "." + parsed_json["version"]["mi nor"] 453 version = parsed_json["version"]["major"] + "." + parsed_json["version"]["mi nor"]
407 domains = [] 454 domains = []
408 for domain in parsed_json["domains"]: 455 for domain in parsed_json["domains"]:
409 domains.append(domain["domain"]) 456 domains.append(domain["domain"])
410 domain["version"] = version 457 domain["version"] = version
411 json_api["domains"] += parsed_json["domains"] 458 json_api["domains"] += parsed_json["domains"]
412 return domains 459 return domains
413 460
414 461
415 class Protocol(object): 462 class Protocol(object):
416 def __init__(self): 463 def __init__(self):
417 self.json_api = {} 464 self.json_api = {}
418 self.generate_domains = [] 465 self.generate_domains = []
419 self.imported_domains = [] 466 self.imported_domains = []
467 self.exported_domains = []
420 468
421 469
422 def main(): 470 def main():
423 jinja_dir, config_file, config = read_config() 471 jinja_dir, config_file, config = read_config()
424 472
425 protocol = Protocol() 473 protocol = Protocol()
426 protocol.json_api = {"domains": []} 474 protocol.json_api = {"domains": []}
427 protocol.generate_domains = read_protocol_file(config.protocol.path, protoco l.json_api) 475 protocol.generate_domains = read_protocol_file(config.protocol.path, protoco l.json_api)
428 protocol.imported_domains = read_protocol_file(config.imported.path, protoco l.json_api) if config.imported else [] 476 protocol.imported_domains = read_protocol_file(config.imported.path, protoco l.json_api) if config.imported else []
429 patch_full_qualified_refs(protocol) 477 patch_full_qualified_refs(protocol)
430 calculate_exports(protocol) 478 calculate_imports_and_exports(config, protocol)
431 create_type_definitions(protocol, "::".join(config.imported.namespace) if co nfig.imported else "") 479 create_type_definitions(protocol, "::".join(config.imported.namespace) if co nfig.imported else "")
432 480
433 if not config.exported: 481 if not config.exported and len(protocol.exported_domains):
434 for domain_json in protocol.json_api["domains"]: 482 sys.stderr.write("Domains [%s] are exported, but config is missing expor t entry\n\n" % ", ".join(protocol.exported_domains))
435 if domain_json["has_exports"] and domain_json["domain"] in protocol. generate_domains: 483 exit(1)
436 sys.stderr.write("Domain %s is exported, but config is missing e xport entry\n\n" % domain_json["domain"])
437 exit(1)
438 484
439 if not os.path.exists(config.protocol.output): 485 if not os.path.exists(config.protocol.output):
440 os.mkdir(config.protocol.output) 486 os.mkdir(config.protocol.output)
441 if protocol.json_api["has_exports"] and not os.path.exists(config.exported.o utput): 487 if len(protocol.exported_domains) and not os.path.exists(config.exported.out put):
442 os.mkdir(config.exported.output) 488 os.mkdir(config.exported.output)
443 jinja_env = initialize_jinja_env(jinja_dir, config.protocol.output) 489 jinja_env = initialize_jinja_env(jinja_dir, config.protocol.output, config)
444 490
445 inputs = [] 491 inputs = []
446 inputs.append(__file__) 492 inputs.append(__file__)
447 inputs.append(config_file) 493 inputs.append(config_file)
448 inputs.append(config.protocol.path) 494 inputs.append(config.protocol.path)
449 if config.imported: 495 if config.imported:
450 inputs.append(config.imported.path) 496 inputs.append(config.imported.path)
451 templates_dir = os.path.join(module_path, "templates") 497 templates_dir = os.path.join(module_path, "templates")
452 inputs.append(os.path.join(templates_dir, "TypeBuilder_h.template")) 498 inputs.append(os.path.join(templates_dir, "TypeBuilder_h.template"))
453 inputs.append(os.path.join(templates_dir, "TypeBuilder_cpp.template")) 499 inputs.append(os.path.join(templates_dir, "TypeBuilder_cpp.template"))
(...skipping 12 matching lines...) Expand all
466 template_context = { 512 template_context = {
467 "config": config, 513 "config": config,
468 "domain": domain, 514 "domain": domain,
469 "join_arrays": join_arrays, 515 "join_arrays": join_arrays,
470 "resolve_type": functools.partial(resolve_type, protocol), 516 "resolve_type": functools.partial(resolve_type, protocol),
471 "type_definition": functools.partial(type_definition, protocol), 517 "type_definition": functools.partial(type_definition, protocol),
472 "generate_command": functools.partial(generate_command, protocol, co nfig), 518 "generate_command": functools.partial(generate_command, protocol, co nfig),
473 "generate_event": functools.partial(generate_event, protocol, config ), 519 "generate_event": functools.partial(generate_event, protocol, config ),
474 "is_async_command": functools.partial(is_async_command, protocol, co nfig), 520 "is_async_command": functools.partial(is_async_command, protocol, co nfig),
475 "generate_disable": functools.partial(generate_disable, protocol, co nfig), 521 "generate_disable": functools.partial(generate_disable, protocol, co nfig),
476 "format_include": format_include 522 "format_include": functools.partial(format_include, config),
477 } 523 }
478 524
479 if domain["domain"] in protocol.generate_domains: 525 if domain["domain"] in protocol.generate_domains:
480 outputs[os.path.join(config.protocol.output, class_name + ".h")] = h _template.render(template_context) 526 outputs[os.path.join(config.protocol.output, to_file_name(config, cl ass_name + ".h"))] = h_template.render(template_context)
481 outputs[os.path.join(config.protocol.output, class_name + ".cpp")] = cpp_template.render(template_context) 527 outputs[os.path.join(config.protocol.output, to_file_name(config, cl ass_name + ".cpp"))] = cpp_template.render(template_context)
482 if domain["has_exports"]: 528 if domain["domain"] in protocol.exported_domains:
483 outputs[os.path.join(config.exported.output, class_name + ".h")] = exported_template.render(template_context) 529 outputs[os.path.join(config.exported.output, to_file_name(config , class_name + ".h"))] = exported_template.render(template_context)
484 if domain["domain"] in protocol.imported_domains and domain["has_exports "]: 530 if domain["domain"] in protocol.imported_domains:
485 outputs[os.path.join(config.protocol.output, class_name + ".h")] = i mported_template.render(template_context) 531 outputs[os.path.join(config.protocol.output, to_file_name(config, cl ass_name + ".h"))] = imported_template.render(template_context)
486 532
487 if config.lib: 533 if config.lib:
488 template_context = { 534 template_context = {
489 "config": config, 535 "config": config,
490 "format_include": format_include, 536 "format_include": functools.partial(format_include, config),
491 } 537 }
492 538
493 lib_templates_dir = os.path.join(module_path, "lib") 539 lib_templates_dir = os.path.join(module_path, "lib")
494 # Note these should be sorted in the right order. 540 # Note these should be sorted in the right order.
495 # TODO(dgozman): sort them programmatically based on commented includes. 541 # TODO(dgozman): sort them programmatically based on commented includes.
496 lib_h_templates = [ 542 lib_h_templates = [
497 "Collections_h.template", 543 "Collections_h.template",
498 "ErrorSupport_h.template", 544 "ErrorSupport_h.template",
499 "Values_h.template", 545 "Values_h.template",
500 "Object_h.template", 546 "Object_h.template",
(...skipping 20 matching lines...) Expand all
521 ] 567 ]
522 568
523 def generate_lib_file(file_name, template_files): 569 def generate_lib_file(file_name, template_files):
524 parts = [] 570 parts = []
525 for template_file in template_files: 571 for template_file in template_files:
526 inputs.append(os.path.join(lib_templates_dir, template_file)) 572 inputs.append(os.path.join(lib_templates_dir, template_file))
527 template = jinja_env.get_template("lib/" + template_file) 573 template = jinja_env.get_template("lib/" + template_file)
528 parts.append(template.render(template_context)) 574 parts.append(template.render(template_context))
529 outputs[file_name] = "\n\n".join(parts) 575 outputs[file_name] = "\n\n".join(parts)
530 576
531 generate_lib_file(os.path.join(config.lib.output, "Forward.h"), forward_ h_templates) 577 generate_lib_file(os.path.join(config.lib.output, to_file_name(config, " Forward.h")), forward_h_templates)
532 generate_lib_file(os.path.join(config.lib.output, "Protocol.h"), lib_h_t emplates) 578 generate_lib_file(os.path.join(config.lib.output, to_file_name(config, " Protocol.h")), lib_h_templates)
533 generate_lib_file(os.path.join(config.lib.output, "Protocol.cpp"), lib_c pp_templates) 579 generate_lib_file(os.path.join(config.lib.output, to_file_name(config, " Protocol.cpp")), lib_cpp_templates)
534 580
535 # Make gyp / make generatos happy, otherwise make rebuilds world. 581 # Make gyp / make generatos happy, otherwise make rebuilds world.
536 inputs_ts = max(map(os.path.getmtime, inputs)) 582 inputs_ts = max(map(os.path.getmtime, inputs))
537 up_to_date = True 583 up_to_date = True
538 for output_file in outputs.iterkeys(): 584 for output_file in outputs.iterkeys():
539 if not os.path.exists(output_file) or os.path.getmtime(output_file) < in puts_ts: 585 if not os.path.exists(output_file) or os.path.getmtime(output_file) < in puts_ts:
540 up_to_date = False 586 up_to_date = False
541 break 587 break
542 if up_to_date: 588 if up_to_date:
543 sys.exit() 589 sys.exit()
544 590
545 for file_name, content in outputs.iteritems(): 591 for file_name, content in outputs.iteritems():
546 out_file = open(file_name, "w") 592 out_file = open(file_name, "w")
547 out_file.write(content) 593 out_file.write(content)
548 out_file.close() 594 out_file.close()
549 595
550 596
551 main() 597 main()
OLDNEW

Powered by Google App Engine
This is Rietveld 408576698