#!/usr/bin/env python
# Copyright 2017 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 os
import re
import sys


def main():
    parser = argparse.ArgumentParser(
        sys.argv[0], description="Generate main file for Fuchsia dart test")
    parser.add_argument(
        "--out", help="Path to .dart file to generate", required=True)
    parser.add_argument(
        '--source',
        help='Test source file',
        required=True,
        default=[],
        action='append')
    args = parser.parse_args()

    out_dir = os.path.dirname(args.out)
    test_files = [os.path.relpath(t, out_dir) for t in args.source]

    outfile = open(args.out, 'w')
    outfile.write(
        '''// Generated by %s

    // ignore_for_file: directives_ordering
    // ignore_for_file: avoid_relative_lib_imports

    import 'dart:async';
    import 'package:fuchsia_test_helper/fuchsia_test_helper.dart';
    ''' % os.path.basename(__file__))

    for i, path in enumerate(test_files):
        outfile.write("import '%s' as test_%d;\n" % (path, i))

    outfile.write(
        '''
    Future<int> main(List<String> args) async {
      final int exitCode = await runFuchsiaTests(<MainFunction>[
    ''')

    for i in range(len(test_files)):
        outfile.write('test_%d.main,\n' % i)

    outfile.write(
        '''], args);

      exitFuchsiaTest(exitCode);
      return exitCode;
    }
    ''')

    outfile.close()


if __name__ == '__main__':
    main()