#!/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-dir',
                        help='Path to test sources',
                        required=True)
    args = parser.parse_args()

    out_dir = os.path.dirname(args.out)
    test_files = []
    for root, dirs, files in os.walk(args.source_dir):
        for f in files:
            if not f.endswith('_test.dart'):
                continue
            test_files.append(os.path.relpath(os.path.join(root, f), out_dir))

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