# Copyright (c) 2018 The WebRTC project authors. All Rights Reserved.
#
# Use of this source code is governed by a BSD-style license
# that can be found in the LICENSE file in the root of the source
# tree. An additional intellectual property rights grant can be found
# in the file PATENTS.  All contributing project authors may
# be found in the AUTHORS file in the root of the source tree.

import argparse
import datetime
import os
import sys
import textwrap


def GenerateUmbrellaHeader():
    parser = argparse.ArgumentParser(description='Generate umbrella header')
    parser.add_argument("-o", "--out", type=str, help="Output file.")
    parser.add_argument("-s",
                        "--sources",
                        default=[],
                        type=str,
                        nargs='+',
                        help="Headers to include.")

    args = parser.parse_args()

    with open(args.out, "w") as outfile:
        outfile.write(
            textwrap.dedent("""\
    /*
     *  Copyright %d The WebRTC project authors. All Rights Reserved.
     *
     *  Use of this source code is governed by a BSD-style license
     *  that can be found in the LICENSE file in the root of the source
     *  tree. An additional intellectual property rights grant can be found
     *  in the file PATENTS.  All contributing project authors may
     *  be found in the AUTHORS file in the root of the source tree.
     */\n\n""" % datetime.datetime.now().year))

        for s in args.sources:
            outfile.write("#import <WebRTC/{}>\n".format(os.path.basename(s)))

    return 0


if __name__ == '__main__':
    sys.exit(GenerateUmbrellaHeader())