#!/usr/bin/python3

##
## util/generate_test_case.py
##
## Copyright (c) 2021-2024 Bartek Kryza <bkryza@gmail.com>
##
## Licensed under the Apache License, Version 2.0 (the "License");
## you may not use this file except in compliance with the License.
## You may obtain a copy of the License at
##
##     http://www.apache.org/licenses/LICENSE-2.0
##
## Unless required by applicable law or agreed to in writing, software
## distributed under the License is distributed on an "AS IS" BASIS,
## WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
## See the License for the specific language governing permissions and
## limitations under the License.
##

import os
import sys
import jinja2

TEST_CASE_MULTIPLIER = 10000

CLASS_DIAGRAM_TEST_CASE_EXAMPLES = """
    CHECK_CLASS_DIAGRAM(*config, diagram, *model, [](const auto &src) {
            // REQUIRE(HasTitle(src, "Basic class diagram example"));

            // REQUIRE(!IsClass(src, "NOSUCHCLASS"));
            // REQUIRE(IsAbstractClass(src, "A"));
            // REQUIRE(IsClass(src, "B"));
            // REQUIRE(IsBaseClass(src, "A", "B"));

            // REQUIRE(IsMethod<Public, Abstract>(src, "A", "foo_a"));

            // REQUIRE(IsAssociation<Private>(src, "D", "A", "as"));

            // REQUIRE(HasNote(src, "A", "left", "This is class A"));
        });
"""

SEQUENCE_DIAGRAM_TEST_CASE_EXAMPLES = """
    CHECK_SEQUENCE_DIAGRAM(
       *config, diagram, *model,
        [](const auto &src) {
            // REQUIRE(HasTitle(src, "Basic sequence diagram example"));

            REQUIRE(MessageOrder(src,
                 {
                     //
                    // {"tmain()", "A", "A()"},    //
                    // {"B", "A", "log_result(int)", Static{}}     //
                }));

            // REQUIRE(!HasMessage(src, {"A", {"detail", "C"}, "add(int,int)"}));

            // REQUIRE(HasComment(src, "t20001 test diagram of type sequence"));

            // REQUIRE(HasMessageComment(src, "tmain()", "Just add 2 numbers"));
        });
"""

PACKAGE_DIAGRAM_TEST_CASE_EXAMPLES = """
    CHECK_PACKAGE_DIAGRAM(*config, diagram, *model, [](const auto &src) {
        // REQUIRE(IsNamespacePackage(src, "A"s, "AA"s));

        // REQUIRE(IsNamespacePackage(src, "B"s, "BB"s, "BBB"s));

        // REQUIRE(IsDependency(src, "BBB", "A1"));
    });
"""

INCLUDE_DIAGRAM_TEST_CASE_EXAMPLES = """
    CHECK_INCLUDE_DIAGRAM(config, diagram, *model, [](const auto &src) {
        // REQUIRE(IsFolder(src, "include/lib1"));
        // REQUIRE(IsFile(src, "include/lib1/lib1.h"));

        // REQUIRE(IsSystemHeader(src, "string"));

        // REQUIRE(IsHeaderDependency(
        //    src, "src/t40001.cc", "include/t40001_include1.h"));

        // REQUIRE(IsSystemHeaderDependency(
        //    src, "include/t40001_include1.h", "string"));
    });
"""


def test_case_already_exists(name):
    return os.path.isdir(os.path.join(os.path.dirname(__file__), '..', name))


def test_case_category(type):
    if type == 'class':
        return 0
    elif type == 'sequence':
        return 2 * TEST_CASE_MULTIPLIER
    elif type == 'package':
        return 3 * TEST_CASE_MULTIPLIER
    elif type == 'include':
        return 4 * TEST_CASE_MULTIPLIER
    else:
        print(f"Invalid test case type '{type}' - aborting...")
        sys.exit(1)


def examples_for_type(type):
    if type == 'class':
        return CLASS_DIAGRAM_TEST_CASE_EXAMPLES
    elif type == 'sequence':
        return SEQUENCE_DIAGRAM_TEST_CASE_EXAMPLES
    elif type == 'package':
        return PACKAGE_DIAGRAM_TEST_CASE_EXAMPLES
    elif type == 'include':
        return INCLUDE_DIAGRAM_TEST_CASE_EXAMPLES
    else:
        print(f"Invalid test case type '{type}' - aborting...")
        sys.exit(1)


def generate(environment, variables, src, dst, dst_path):
    template = environment.get_template(src)

    output = template.render(variables)

    with open(os.path.join(dst_path, dst), 'wb+') as fh:
        fh.write(output.encode("UTF-8"))


def main(args):
    if len(args) != 2:
        print("Usage: ./generate_test_cases.py <type> <number>")

    test_case_type = args[0]
    test_case_number = int(args[1])
    test_case_name = f't{test_case_category(test_case_type) + test_case_number:05d}'

    if test_case_already_exists(test_case_name):
        print(f"Test case 'test_case_name' already exists - aborting...")
        sys.exit(1)

    template_directory = os.path.join(os.path.dirname(__file__), 'templates', 'test_cases')

    loader = jinja2.FileSystemLoader(searchpath=template_directory)

    env_parameters = dict(
        loader=loader,
        undefined=jinja2.StrictUndefined
    )
    environment = jinja2.Environment(**env_parameters)

    test_case_directory = os.path.join(os.path.dirname(__file__), '..', 'tests', test_case_name)

    os.mkdir(test_case_directory)

    examples = examples_for_type(test_case_type)

    variables = dict(type = test_case_type,
                     TYPE = test_case_type.upper(),
                     name = test_case_name,
                     examples = examples)

    generate(environment, variables, '.clang-uml', '.clang-uml', test_case_directory)
    generate(environment, variables, 't00000.cc', f'{test_case_name}.cc', test_case_directory)
    generate(environment, variables, 'test_case.h', 'test_case.h', test_case_directory)


if __name__ == '__main__':
    main(sys.argv[1:])