from argparse import ArgumentParser, Namespace
from git_flow import COLOR_WHITE_BOLD,  COLOR_GREEN, COLOR_RED, COLOR_RESET, TRASH_BRANCH_PREFIX, WIP_BRANCH_PREFIX, GitFlowError
from git_flow.command.base import Command
from git_flow.git import Git


BRANCH_FORMAT = "%(refname:short)"


class BranchCommand(Command):
    def name(self) -> str:
        return "branch"

    def description(self) -> str:
        return """Lista ramas del repositorio, agrupandolas por entorno objetivo"""

    def setup_parser(self, parser: ArgumentParser) -> ArgumentParser:
        parser.add_argument(
            "environment",
            nargs="?",
            help="Entorno de las ramas a listar. Por defecto, es el entorno actual.",
        )

        parser.add_argument(
            "--trash",
            action="store_true",
            help=f"Listar ramas en la 'papelera de reciclaje' (ramas que empiezan con '{TRASH_BRANCH_PREFIX}')"
        )

        parser.add_argument(
            "--wip",
            action="store_true",
            help=f"Listar ramas 'en progreso' (ramas que empiezan con '{WIP_BRANCH_PREFIX}')"
        )

        parser.add_argument(
            "--all",
            action="store_true",
            help=f"Listar ramas de todos los entornos"
        )

        return parser

    def run(self, args: Namespace = Namespace()):
        self.ensure_initialized()

        self.current_branch = Git.get_current_branch()
        self.environments = self.flowconfig["flow.branches"].split(",")

        if self.current_branch in self.environments:
            current_environment = self.current_branch
        else:
            (current_environment, _) = self.get_branch_env_and_type(self.current_branch)

        if args.trash:
            self.show_branches(TRASH_BRANCH_PREFIX)
        elif args.wip:
            self.show_branches(WIP_BRANCH_PREFIX)
        elif args.all:
            self.show_envs()
            self.show_all_branches()
        elif args.environment is None:
            self.show_envs()
            self.show_env_branches(current_environment)
        elif args.environment in self.environments:
            self.show_envs()
            self.show_env_branches(args.environment)
        else:
            raise GitFlowError(f"'{args.environment}' no es un entorno válido.")

    def show_branches(self, prefix: str):
        branches = self.get_branches(prefix)

        if not branches:
            print("No hay ramas a mostrar.")
            return

        for branch in branches:
            prefix = (COLOR_RED + "*") if branch == self.current_branch else " "
            print(prefix + " " + branch + COLOR_RESET)

    def get_branches(self, prefix: str):
        return Git("for-each-ref", "refs/heads/" + prefix, format=BRANCH_FORMAT).lines()

    def show_all_branches(self):
        for environment in self.environments:
            self.show_env_branches(environment, len(self.environments) > 1)

        trash = len(self.get_branches(TRASH_BRANCH_PREFIX))

        if trash:
            print()
            print(COLOR_WHITE_BOLD + "Papelera: " + COLOR_RESET + str(trash) + " rama" + ("" if trash == 1 else "s"))

    def show_envs(self):
        print(COLOR_WHITE_BOLD + "Entornos" + COLOR_RESET)
        for environment in self.environments:
            prefix = (COLOR_GREEN + "*") if environment == self.current_branch else " "
            print(prefix + " " + environment + COLOR_RESET)
        print()

    def show_env_branches(self, environment: str, show_env: bool = False):
        is_first_env = environment == self.environments[0]
        pattern = "refs/heads/" if is_first_env else "refs/heads/release/" + environment + "/"
        pattern += "*/*"
        exclude = ["refs/heads/" + TRASH_BRANCH_PREFIX, "refs/heads/release/"] if is_first_env else []
        branches = Git("for-each-ref", pattern, exclude=exclude, format=BRANCH_FORMAT).lines()

        print(COLOR_WHITE_BOLD + (environment if show_env else "Ramas") + COLOR_RESET)

        if not branches:
            print("No hay ramas a mostrar.")
            return

        for branch in branches:
            prefix = (COLOR_GREEN + "*") if branch == self.current_branch else " "
            print(prefix + " " + branch + COLOR_RESET)