# 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())