config/common/cmake/make_gn_args.py

#!/usr/bin/env python3

#
#    Copyright (c) 2023 Project CHIP Authors
#    All rights reserved.
#
#    Licensed under the Apache License, Version 2.0 (the "License");
#    you may not use this file except in compliance with the License.
#    You may obtain a copy of the License at
#
#        http://www.apache.org/licenses/LICENSE-2.0
#
#    Unless required by applicable law or agreed to in writing, software
#    distributed under the License is distributed on an "AS IS" BASIS,
#    WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
#    See the License for the specific language governing permissions and
#    limitations under the License.
#

import argparse
import sys

GN_SPECIAL_SEPARATOR = "+|+"
GN_CFLAG_EXCLUDES = [
    '-fno-asynchronous-unwind-tables',
    '-fno-common',
    '-fno-defer-pop',
    '-fno-reorder-functions',
    '-ffunction-sections',
    '-fdata-sections',
    '-g*',
    '-O*',
    '-W*',
]


def write_gn_args(args):
    if args.module:
        sys.stdout.write(f'import("{args.module}")\n')

    for key, value in args.arg:
        sys.stdout.write(f'{key} = {value}\n')

    for key, value in args.arg_string:
        # Escaped quote and dollar sign characters
        filtered_value = value.replace('"', '\\"')
        filtered_value = filtered_value.replace('$', '\\$')
        sys.stdout.write(f'{key} = "{filtered_value}"\n')

    cflag_excludes = ', '.join([f'"{exclude}"'
                               for exclude in GN_CFLAG_EXCLUDES])

    for key, value in args.arg_cflags:
        filtered_value = value.split(" -")
        # Remove empty include paths and defines caused by Cmake generator expressions
        filtered_value = filter(lambda v: v != "D", filtered_value)
        filtered_value = filter(lambda v: v != "isystem", filtered_value)
        # Escaped quote and dollar sign characters
        filtered_value = (v.replace('"', '\\"') for v in filtered_value)
        filtered_value = (v.replace('$', '\\$') for v in filtered_value)
        # Remove white spaces around the argument and remove internal whitespace
        # for correct splitting in string_split() function
        filtered_value = (v.strip() for v in filtered_value)
        filtered_value = (v.replace(' ', '') for v in filtered_value)
        #  Remove duplicates
        filtered_value = list(dict.fromkeys(filtered_value))

        filtered_value_fmt = f"{GN_SPECIAL_SEPARATOR}-".join(filtered_value)
        sys.stdout.write(
            f'{key} = filter_exclude(string_split("{filtered_value_fmt}", "{GN_SPECIAL_SEPARATOR}"), [{cflag_excludes}])\n')


def main():
    parser = argparse.ArgumentParser(fromfile_prefix_chars='@')
    parser.add_argument('--module', action='store')
    parser.add_argument('--arg', action='append', nargs=2, default=[])
    parser.add_argument('--arg-string', action='append', nargs=2, default=[])
    parser.add_argument('--arg-cflags', action='append', nargs=2, default=[])
    args = parser.parse_args()
    write_gn_args(args)


if __name__ == "__main__":
    main()