#! /usr/bin/env python3

 # This file is under GNU General Public License 3.0

 # see LICENSE.txt

 import argparse

 from os import listdir, getcwd

 from os.path import isfile, join

 from re import sub

 import datetime

 srcpath = join(getcwd(), "src/engine_tests")

 ls = listdir(srcpath)

 suites = []

 indent = "    "

 parser = argparse.ArgumentParser()

 parser.add_argument("--exclude", "-x", help="Comma-separated list of tests to exclude from test suite build")

 args = parser.parse_args()

 exclude_arg = args.exclude

 exclude_list = []

 if exclude_arg:

     exclude_list = exclude_arg.split(",")

     for x in exclude_list:

         x.replace(" ","")

 for f in ls:

     if isfile(join(srcpath, f)):

         if (f.endswith(".cc")):

             suite = sub('\.cc$', '', f)

             if not suite in exclude_list:

                 suites.append(suite)

 license = (''

 '// This file is under GNU General Public License 3.0\n'

 '// see LICENSE.txt\n\n')

 header = license + "//\n// src/SuiteMaker.cc generated by gensuitemaker.py - changes may be overwritten. You've been warned!\n//\n\n"

 header += (''

 '#include <cpptest.h>\n'

 '#include <cpptest-suite.h>\n'

 '#include <memory>\n'

 '#include <vector>\n'

 '#include "SuiteMaker.h"\n')

 suiteincludes = ""

 for s in suites:

     suiteincludes += '#include "' + s + '.h"\n'

 output = header + "\n// Begin where we generate stuff\n" + suiteincludes + "\n"

 output += ("\n"

 "const char* SuiteMaker::all_suites[] = {\n")

 first = True

 for s in suites:

     output += indent + '"' + s + '",\n'

 output += "};\n\n"

 output += (

 '// This file is generated, so magic constants are ok.\n'

 'int SuiteMaker::num_suites = ')

 output += str(len(suites)) + ";\n\n"

 output += "void SuiteMaker::suitemaker_build(const char* test_class_name, const char* test_home, Test::Suite** test_suite) {\n"

 first = True

 for s in suites:

     output += indent

     if not first:

         output += "else "

     else:

         first = False

     output += ('if (strcmp(test_class_name, "' + s + '") == 0)\n' + 

                indent + indent + '*test_suite = new ' + s + '(test_class_name, test_home);\n')

 output += "}\n\n"

 output += (

 'void SuiteMaker::suitemaker_buildlist(const char** test_class_names, int num_to_run, const char* test_home, std::vector<Test::Suite*>& test_suites) {\n'

 '    for (int i = 0; i < num_to_run; i++) {\n'

 '        Test::Suite* suite = NULL;\n'

 '        SuiteMaker::suitemaker_build(test_class_names[i], test_home, &suite);\n'

 '        if (!suite)\n'

 '            throw std::runtime_error("Could not create a test suite instance."); // FIXME, better error, cleanup, obviously\n'

 '        test_suites.push_back(suite);\n'

 '    }\n'    

 '}'

 '\n'

 'void SuiteMaker::suitemaker_buildall(const char* test_home, std::vector<Test::Suite*>& test_suites) {\n'

 '    SuiteMaker::suitemaker_buildlist(SuiteMaker::all_suites, SuiteMaker::num_suites, test_home, test_suites);\n'

 '}\n\n')

 suitemaker = open("src/SuiteMaker.cc", 'w') 

 suitemaker.write(output)

 suitemaker.close()

 generated_at = ("src/SuiteMaker.cc" + " generated at " + datetime.datetime.now().strftime('%c'))

 print(generated_at)