import json
from pathlib import Path
from typing import Optional

import git

from PyTeX.config.constants import BUILD_INFO_FILENAME
from PyTeX.errors import *

from .utils import BuildInfo, pytex_msg, TexFileToFormat


def build(
        src_dir: Optional[Path] = None,
        build_dir: Optional[Path] = None,
        input_file: Optional[Path] = None,
        author: Optional[str] = None,
        latex_name: str = 'prepend-author',           # name handling
        recursive: bool = False,                      # input control
        include_timestamp: bool = False,              # header
        include_pytex_version: bool = False,          # header
        include_license: bool = False,                # header
        include_git_version: bool = False,            # header
        include_pytex_info_text: bool = False,        # header
        extra_header: Optional[Path] = None,
        allow_dirty: bool = False,                    # versioning
        overwrite_existing_files: bool = False,       # output control
        build_all: bool = False,                      # output control / versioning
        write_build_information: bool = True,         # meta
        clean_old_files: bool = False
        ):
    pytex_msg('Getting git repository information...')
    if extra_header:
        if extra_header.exists():
            with open(extra_header, 'r') as f:
                text = f.readlines()
            extra_header = [line.rstrip() for line in text]
        else:
            raise ExtraHeaderFileNotFoundError
    current_build_info = BuildInfo(
        include_timestamp=include_timestamp,
        include_pytex_version=include_pytex_version,
        include_license=include_license,
        include_git_version=include_git_version,
        include_pytex_info_text=include_pytex_info_text,
        extra_header=extra_header,
        author=author,
        pytex_repo=git.Repo(__file__, search_parent_directories=True),
        packages_repo=git.Repo(src_dir, search_parent_directories=True)
    )
    print(r'[PyTeX] This is PyTex version {pytex_version}'.format(
        pytex_version=current_build_info.pytex_version
    ))
    input_dir = src_dir if src_dir else input_file.parent
    output_dir = build_dir if build_dir else input_file.parent

    last_build_info_file = output_dir / BUILD_INFO_FILENAME
    if last_build_info_file.exists():
        with open(output_dir / 'build_info.json', 'r') as f:
            last_build_info = json.load(f)
    else:
        last_build_info = None

    files = []
    if input_file:
        files.append(input_file)
    if src_dir:
        if recursive:
            for file in src_dir.rglob('*.pysty'):
                files.append(file)
            for file in src_dir.rglob('*.pycls'):
                files.append(file)
            for file in src_dir.rglob('*.pydict'):
                files.append(file)
            for file in src_dir.rglob('*.pysty3'):
                files.append(file)
            for file in src_dir.rglob('*.pycls3'):
                files.append(file)
            for file in src_dir.rglob('*.dtx'):
                files.append(file)
        else:
            for file in src_dir.glob('*.pysty'):
                files.append(file)
            for file in src_dir.glob('*.pycls'):
                files.append(file)
            for file in src_dir.glob('*.pydict'):
                files.append(file)
            for file in src_dir.glob('*.pysty3'):
                files.append(file)
            for file in src_dir.glob('*.pycls3'):
                files.append(file)
            for file in src_dir.glob('*.dtx'):
                files.append(file)

    sources_to_build = []
    for file in files:
        if last_build_info:
            last_build_info_for_this_file =\
                list(filter(lambda i: i['source file'] == str(file.relative_to(src_dir)), last_build_info['tex_sources']))
        else:
            last_build_info_for_this_file = []
        sources_to_build.append(
            TexFileToFormat(
                src_path=file,
                build_root=output_dir,
                src_root=src_dir,
                latex_name=latex_name,
                current_build_info=current_build_info,
                last_build_info=last_build_info_for_this_file,
                allow_dirty=allow_dirty,
                overwrite_existing_files=overwrite_existing_files,
                build_all=build_all
            ))

    info_dict = {
        'build_time': '',
        'source files': {
            'version': current_build_info.packages_version,
            'commit': current_build_info.packages_hash,
            'dirty': current_build_info.package_repo.is_dirty(untracked_files=True)
        },
        'pytex': {
            'version': current_build_info.pytex_version,
            'commit': current_build_info.pytex_hash,
            'dirty': current_build_info.pytex_repo.is_dirty(untracked_files=True)
        },
        'tex_sources': [

        ]
    }

    for source in sources_to_build:
        infos = source.format()
        for info in infos:
            info_dict['tex_sources'].append(info)

    built_files = [info['name'] for info in info_dict['tex_sources']]
    if last_build_info:
        lastly_built_files = [info['name'] for info in last_build_info['tex_sources']]
    else:
        lastly_built_files = []
    if clean_old_files:
        for file in output_dir.rglob('*'):
            if str(file.relative_to(output_dir)) in lastly_built_files:
                if str(file.relative_to(output_dir)) not in built_files:
                    print(f'[PyTeX] Removing old built file {str(file.relative_to(output_dir))}')
                    file.unlink()
            elif not str(file.relative_to(output_dir)) in built_files:
                if not file.is_dir() and not str(file.relative_to(output_dir)) == 'build_info.json':
                    # PyTeX does not at all know something about this file
                    raise UnknownFileInBuildDirectoryError(file.relative_to(output_dir))

    if write_build_information:
        with open(output_dir / 'build_info.json', 'w') as f:
            json.dump(info_dict, f, indent=4)
    pytex_msg('Build done')