mirror of
https://github.com/revng/revng
synced 2026-06-21 14:07:57 +00:00
Alessandro Di Federico
67 lines
2.1 KiB
Python
Executable File
67 lines
2.1 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
#
|
|
# This file is distributed under the MIT License. See LICENSE.md for details.
|
|
#
|
|
|
|
import argparse
|
|
import shutil
|
|
import subprocess
|
|
import yaml
|
|
from pathlib import Path
|
|
|
|
from tuple_tree_generator.generators import CppHeadersGenerator
|
|
|
|
argparser = argparse.ArgumentParser()
|
|
argparser.add_argument("schema", help="YAML schema")
|
|
argparser.add_argument("output_dir", help="Output to this directory")
|
|
argparser.add_argument("--namespace", required=True, help="Base namespace for generated types")
|
|
output_format_group = argparser.add_mutually_exclusive_group(required=True)
|
|
output_format_group.add_argument("--cpp-headers", action="store_true", help="Emit C++ headers")
|
|
argparser.add_argument("--include-path-prefix", help="Prefixed to include paths. Only allowed with --cpp-headers")
|
|
|
|
|
|
def is_clang_format_available():
|
|
return shutil.which("clang-format")
|
|
|
|
|
|
def reformat(source: str) -> str:
|
|
"""Applies clang-format to the given source"""
|
|
return subprocess.check_output(
|
|
[
|
|
"clang-format",
|
|
],
|
|
input=source,
|
|
encoding="utf-8",
|
|
)
|
|
|
|
|
|
def main(args):
|
|
with open(args.schema) as f:
|
|
schema = yaml.safe_load(f)
|
|
|
|
if args.cpp_headers:
|
|
if args.include_path_prefix is None:
|
|
raise argparse.ArgumentError("You must provide --include-path-prefix when using --cpp-headers")
|
|
|
|
generator = CppHeadersGenerator(schema, args.namespace, user_include_path=args.include_path_prefix)
|
|
sources = generator.emit()
|
|
|
|
if is_clang_format_available():
|
|
sources = {name: reformat(source) for name, source in sources.items()}
|
|
else:
|
|
print("Warning: clang-format is not available, cannot format source")
|
|
else:
|
|
argparser.error("Provide --cpp-headers")
|
|
|
|
output_dir = Path(args.output_dir)
|
|
for output_path, output_source in sources.items():
|
|
full_output_path = output_dir / output_path
|
|
# Ensure the containing directory is created
|
|
full_output_path.parent.mkdir(exist_ok=True)
|
|
full_output_path.write_text(output_source)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
parsed_args = argparser.parse_args()
|
|
main(parsed_args)
|