# Copyright 2020 The Chromium Authors
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.
import argparse
import os.path
import subprocess
import sys
from typing import Any, Mapping
import wayland_protocol_c_arg_handling
import wayland_protocol_construction
import wayland_protocol_data_classes
import wayland_protocol_externals
import wayland_protocol_identifiers
def expand_template(template: str, context: Mapping[str, Any]) -> str:
"""Expands the template using context, and returns the result"""
# Loaded from third_party/jinja2 after a sys.path modification
import jinja2
env = jinja2.Environment(
loader=jinja2.FileSystemLoader(os.path.dirname(template)),
keep_trailing_newline=True, # newline-terminate generated files
lstrip_blocks=True,
trim_blocks=True) # so don't need {%- -%} everywhere
# Additional global functions
env.globals['get_c_arg_for_server_request_arg'] = (
wayland_protocol_c_arg_handling.get_c_arg_for_server_request_arg)
env.globals['get_c_arg_for_server_event_arg'] = (
wayland_protocol_c_arg_handling.get_c_arg_for_server_event_arg)
env.globals['get_c_return_type_for_client_request'] = (
wayland_protocol_c_arg_handling.get_c_return_type_for_client_request)
env.globals['get_c_arg_for_client_request_arg'] = (
wayland_protocol_c_arg_handling.get_c_arg_for_client_request_arg)
env.globals['get_c_arg_for_client_event_arg'] = (
wayland_protocol_c_arg_handling.get_c_arg_for_client_event_arg)
env.globals['get_construction_steps'] = (
wayland_protocol_construction.get_construction_steps)
env.globals['get_destructor'] = (
wayland_protocol_construction.get_destructor)
env.globals['get_external_interfaces_for_protocol'] = (
wayland_protocol_externals.get_external_interfaces_for_protocol)
env.globals['get_minimum_version_to_construct'] = (
wayland_protocol_construction.get_minimum_version_to_construct)
env.globals['get_versions_to_test_for_event_delivery'] = (
wayland_protocol_construction.get_versions_to_test_for_event_delivery)
env.globals['is_global_interface'] = (
wayland_protocol_construction.is_global_interface)
# Additional filters for transforming text
env.filters['kebab'] = wayland_protocol_identifiers.kebab_case
env.filters['pascal'] = wayland_protocol_identifiers.pascal_case
return env.get_template(os.path.basename(template)).render(context)
def clang_format_source_text(source_text: str, clang_format_path: str,
effective_filename: str) -> str:
"""Runs clang-format on source_text and returns the result."""
# clang-format the output, for better readability and for
# -Wmisleading-indentation.
return subprocess.run(
[clang_format_path, '--assume-filename', effective_filename],
input=source_text,
capture_output=True,
check=True,
text=True).stdout
def write_if_changed(source_text: str, output: str) -> str:
"""Writes source_text to output, but only if different."""
if os.path.exists(output):
with open(output, 'rt') as infile:
if infile.read() == source_text:
return
with open(output, 'wt') as outfile:
outfile.write(source_text)
def main():
parser = argparse.ArgumentParser()
parser.add_argument('third_party_path')
parser.add_argument('clang_format_path')
parser.add_argument('template')
parser.add_argument('output')
parser.add_argument('protocols', nargs='+')
args = parser.parse_args()
# Allows us to import jinga2
sys.path.append(args.third_party_path)
protocols = wayland_protocol_data_classes.Protocols.parse_xml_files(
args.protocols)
if len(protocols.protocols) == 1:
expanded = expand_template(args.template,
{'protocol': protocols.protocols[0]})
else:
expanded = expand_template(args.template,
{'protocols': protocols.protocols})
formatted = clang_format_source_text(expanded, args.clang_format_path,
args.output)
write_if_changed(formatted, args.output)
if __name__ == '__main__':
main()
Codebase Browser
chromium