from pytest import fixture  # , raises
from pathlib import Path
from ruamel.yaml import YAML
from rougail import RougailConfig
from rougail.output_doc import RougailOutputDoc

from .custom import CustomOption


dico_dirs = Path('../rougail/tests/dictionaries')
test_ok = set()

excludes = set([
    '60_5family_dynamic_unknown_suffix',
    '60_5family_dynamic_variable_outside_sub_suffix',
])

for test in dico_dirs.iterdir():
    if (test / 'tiramisu').is_dir() and test.name not in excludes:
        test_ok.add(test.name)

test_ok = list(test_ok)
test_ok.sort()

# test_ok = ['00_9default_calculation_multi_optional2']


@fixture(scope="module", params=test_ok)
def test_dir(request):
    return request.param


def _test_dictionaries(test, output, namespace, examples=False):
    test_dir = dico_dirs / test
    rougailconfig = RougailConfig.copy()
    rougailconfig['step.output'] = 'doc'
    rougailconfig['doc.output_format'] = output
    rougailconfig['functions_files'] = [str(dico_dirs.parent / 'eosfunc' / 'test.py')]
    if examples:
        rougailconfig['doc.with_example'] = True
    else:
        rougailconfig['doc.with_example'] = False
#    rougailconfig['tiramisu_cache'] = "cache.py"
    dirs = [str(test_dir / 'dictionaries' / 'rougail')]
    subfolder = test_dir / 'dictionaries' / 'rougail2'
    if subfolder.is_dir():
        dirs.append(str(subfolder))
    rougailconfig['main_dictionaries'] = dirs
    if namespace:
        rougailconfig['main_namespace'] = 'Rougail'
    else:
        rougailconfig['main_namespace'] = None
    extra_dictionaries = {}
    extras = list((test_dir / 'dictionaries').iterdir())
    extras.sort()
    for extra in extras:
        if extra.name in ['rougail', 'rougail2']:
            continue
        if extra.is_dir():
            extra_dictionaries[extra.name] = [str(extra)]
    if extra_dictionaries:
        rougailconfig['extra_dictionaries'] = extra_dictionaries
    rougailconfig['custom_types']['custom'] = CustomOption
    inventory = RougailOutputDoc(rougailconfig=rougailconfig)
    try:
        doc = inventory.formater.header()
    except:
        doc = ''
    yaml = YAML()
    len_subdir = len(str(dico_dirs)) + 1
    if extra_dictionaries:
        all_dirs = [[rougailconfig['main_dictionaries']], rougailconfig['extra_dictionaries'].values()]
    else:
        all_dirs = [[rougailconfig['main_dictionaries']]]
    for r in all_dirs:
        for dirs in r:
            for d in dirs:
                for f in Path(d).iterdir():
                    if f.name.endswith('.yml') or f.name.endswith('.yaml'):
                        try:
                            doc += inventory.formater.title(str(f)[len_subdir:].split('/', 1)[-1], 1)
                        except:
                            pass
                        with f.open(encoding="utf8") as file_fh:
                            objects = yaml.load(file_fh)
                        try:
                            doc += inventory.formater.yaml(objects)
                        except:
                            pass
    if examples:
        name = 'examples'
        doc += inventory.gen_doc_examples()
    else:
        if namespace:
            name = 'base'
        else:
            name = 'no_namespace'
        doc += inventory.gen_doc()
    doc_file = Path('tests') / 'docs' / name / (test_dir.name + {'github': '.md', 'asciidoc': '.adoc', 'json': '.json'}.get(output))
    if not doc_file.is_file():
        with doc_file.open('w') as docfh:
            docfh.write(doc)
    with doc_file.open() as docfh:
        result = docfh.read()
    assert doc == result


def test_dictionaries_json(test_dir):
    _test_dictionaries(test_dir, 'json', True)


def test_dictionaries_github(test_dir):
    _test_dictionaries(test_dir, 'github', True)


def test_dictionaries_asciidoc(test_dir):
    _test_dictionaries(test_dir, 'asciidoc', True)


def test_dictionaries_examples(test_dir):
    _test_dictionaries(test_dir, 'github', True, True)


def test_dictionaries_json_no_namespace(test_dir):
    if (dico_dirs / test_dir / 'force_namespace').is_file():
        return
    _test_dictionaries(test_dir, 'json', False)


def test_dictionaries_github_no_namespace(test_dir):
    if (dico_dirs / test_dir / 'force_namespace').is_file():
        return
    _test_dictionaries(test_dir, 'github', False)


def test_dictionaries_asciidoc_no_namespace(test_dir):
    if (dico_dirs / test_dir / 'force_namespace').is_file():
        return
    _test_dictionaries(test_dir, 'asciidoc', False)