zircon/tools/zither/scripts/create-file-with-includes.py

#!/usr/bin/env fuchsia-vendored-python
# Copyright 2022 The Fuchsia Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.

import argparse
import sys
import os

# Creates an empty C/C++/asm file with the given #includes.


def main():
    parser = argparse.ArgumentParser()
    parser.add_argument(
        "--output",
        help="Where to write the resulting main()-containing file",
        required=True,
    )
    parser.add_argument(
        "--comment",
        help="The file-level comment of the generated file",
        required=True,
    )
    parser.add_argument("includes", nargs="*", help="files to #include")
    args = parser.parse_args()

    contents = ["// %s" % args.comment]

    # Include headers before .inc files, as the latter will likely refer to
    # things defined in the former.
    for include in args.includes:
        if include.endswith(".h"):
            contents.append("#include <%s>" % include)
    for include in args.includes:
        if not include.endswith(".h"):
            contents.append("#include <%s>" % include)
    contents.append("")

    with open(args.output, "w") as output_file:
        output_file.write("\n".join(contents))

    return 0


if __name__ == "__main__":
    sys.exit(main())