#!/usr/bin/env python

import yaml
import os

github_workflow_file = '.github/workflows/ansible-test-plugins.yml'


def read_github_workflow_file():
    with open(github_workflow_file, 'r') as gh_file:
        try:
            return yaml.safe_load(gh_file)
        except yaml.YAMLError as exc:
            print(exc)


def extract_value(target, dict_yaml):
    for key, value in dict_yaml.items():
        if key == target:
            return value


def extract_matrix(workflow_yaml):
    jobs = extract_value('jobs', workflow_yaml)
    integration = extract_value('integration', jobs)
    strategy = extract_value('strategy', integration)
    matrix = extract_value('matrix', strategy)
    return matrix


def is_exclude(exclude_list, test_suite):
    test_is_excluded = False
    for excl in exclude_list:
        match = 0

        if 'ansible' in excl:
            if excl.get('ansible') == test_suite.get('ansible'):
                match += 1

        if 'db_engine_name' in excl:
            if excl.get('db_engine_name') == test_suite.get('db_engine_name'):
                match += 1

        if 'db_engine_version' in excl:
            if excl.get('db_engine_version') == test_suite.get('db_engine_version'):
                match += 1

        if 'python' in excl:
            if excl.get('python') == test_suite.get('python'):
                match += 1

        if 'connector_name' in excl:
            if excl.get('connector_name') == test_suite.get('connector_name'):
                match += 1

        if 'connector_version' in excl:
            if excl.get('connector_version') == test_suite.get('connector_version'):
                match += 1

        if match > 1:
            test_is_excluded = True
            return test_is_excluded

    return test_is_excluded


def main():
    workflow_yaml = read_github_workflow_file()
    tests_matrix_yaml = extract_matrix(workflow_yaml)

    matrix = []
    exclude_list = tests_matrix_yaml.get('exclude')
    for ansible in tests_matrix_yaml.get('ansible'):
        for db_engine_name in tests_matrix_yaml.get('db_engine_name'):
            for db_engine_version in tests_matrix_yaml.get('db_engine_version'):
                for python in tests_matrix_yaml.get('python'):
                    for connector_name in tests_matrix_yaml.get('connector_name'):
                        for connector_version in tests_matrix_yaml.get('connector_version'):
                            test_suite = {
                                'ansible': ansible,
                                'db_engine_name': db_engine_name,
                                'db_engine_version': db_engine_version,
                                'python': python,
                                'connector_name': connector_name,
                                'connector_version': connector_version
                            }
                            if not is_exclude(exclude_list, test_suite):
                                matrix.append(test_suite)

    for tests in matrix:
        a = tests.get('ansible')
        dn = tests.get('db_engine_name')
        dv = tests.get('db_engine_version')
        p = tests.get('python')
        cn = tests.get('connector_name')
        cv = tests.get('connector_version')
        make_cmd = (
            f'make '
            f'ansible="{a}" '
            f'db_engine_name="{dn}" '
            f'db_engine_version="{dv}" '
            f'python="{p}" '
            f'connector_name="{cn}" '
            f'connector_version="{cv}" '
            f'test-integration'
        )
        print(f'Run tests for: Ansible: {a}, DB: {dn} {dv}, Python: {p}, Connector: {cn} {cv}')
        os.system(make_cmd)
        # TODO, allow for CTRL+C to break the loop more easily
        # TODO, store the failures from this iteration
    # TODO, display a summary of failures from every iterations


if __name__ == '__main__':
    main()