#!./.venv/bin/python
import os
import sys
import argparse
import logging
from git import Repo
from git.exc import GitCommandError, NoSuchPathError
from colorama import Fore
from colorama import Style
from collections import defaultdict

new_path = os.path.normpath(os.path.join(os.path.dirname(__file__), ".."))
sys.path.append(new_path)

from script.git_tool import GitTool

_logger = logging.getLogger(__name__)


def get_config():
    """Parse command line arguments, extracting the config file name,
    returning the union of config file and command line arguments

    :return: dict of config file settings and command line arguments
    """
    # TODO update description
    parser = argparse.ArgumentParser(
        formatter_class=argparse.RawDescriptionHelpFormatter,
        description="""Compare actual code with a manifest.""",
        epilog="""\
""",
    )
    parser.add_argument(
        "-m",
        "--manifest",
        default="./default.xml",
        help="The manifest to compare with actual code.",
    )
    args = parser.parse_args()
    return args


def main():
    config = get_config()
    git_tool = GitTool()

    dct_remote, dct_project, default_remote = git_tool.get_manifest_xml_info(
        filename=config.manifest, add_root=True
    )
    default_branch_name = default_remote.get(
        "@revision", git_tool.default_branch
    )
    dct_result = defaultdict(int)
    i = 0
    total = len(dct_project)
    for name, project in dct_project.items():
        i += 1
        path = project.get("@path")
        print(f"{i}/{total} - {path}")
        branch_name = project.get("@revision", default_branch_name)
        organization = project.get("@remote", git_tool.default_project_name)

        try:
            git_repo = Repo(path)
        except NoSuchPathError:
            print(
                f"{Fore.YELLOW}Warning{Style.RESET_ALL} missing project"
                f" {path}."
            )
            dct_result["WARNING"] += 1
            continue

        value = git_repo.git.branch("--show-current")
        if not value:
            # TODO maybe need to check divergence with local branch and not remote branch
            commit_head = git_repo.git.rev_parse("HEAD")
            try:
                commit_branch = git_repo.git.rev_parse(
                    f"{organization}/{branch_name}"
                )
            except GitCommandError:
                # Cannot get information
                if branch_name == commit_head:
                    print(
                        f"{Fore.GREEN}PASS{Style.RESET_ALL} Not on specified"
                        " branch, no divergence"
                    )
                    dct_result["PASS"] += 1
                else:
                    print(
                        f"{Fore.RED}ERROR{Style.RESET_ALL} manifest revision"
                        f" is {branch_name} and commit {commit_head}."
                    )
                    dct_result["ERROR"] += 1
                continue
            if commit_branch != commit_head:
                print(
                    f"{Fore.YELLOW}WARNING{Style.RESET_ALL} Not on specified"
                    " branch, got a divergence."
                )
                dct_result["WARNING"] += 1
            else:
                print(
                    f"{Fore.GREEN}PASS{Style.RESET_ALL} Not on specified"
                    " branch, no divergence"
                )
                dct_result["PASS"] += 1
        elif branch_name != value:
            print(
                f"{Fore.RED}ERROR{Style.RESET_ALL} manifest revision is"
                f" {branch_name} and actual revision is {value}."
            )
            dct_result["ERROR"] += 1
        else:
            print(f"{Fore.GREEN}PASS{Style.RESET_ALL}")
            dct_result["PASS"] += 1

    str_result = ""
    if dct_result["PASS"]:
        str_result += (
            f"{Fore.GREEN}PASS: {dct_result['PASS']}{Style.RESET_ALL}"
        )
    if dct_result["WARNING"]:
        if str_result:
            str_result += " "
        str_result += (
            f"{Fore.YELLOW}WARNING: {dct_result['WARNING']}{Style.RESET_ALL}"
        )
    if dct_result["ERROR"]:
        if str_result:
            str_result += " "
        str_result += (
            f"{Fore.RED}ERROR: {dct_result['ERROR']}{Style.RESET_ALL}"
        )
    print(str_result)


if __name__ == "__main__":
    main()