#!/usr/bin/env python3
import os
import subprocess
import argparse
from helpers import Colors, print_colored, SCRIPT_DESCRIPTION, SCRIPT_EPILOG

def create_new_branch(repo_path, branch_name):
    """
    Create a new branch from 'develop' branch in the repository
    """
    try:
        os.chdir(repo_path)
        print_colored(f"\n=== Creating new branch in repository: {repo_path} ===", Colors.HEADER, bold=True)

        if not os.path.exists('.git'):
            print_colored(f"The folder {repo_path} is not a git repository!", Colors.RED)
            return False

        current_branch = subprocess.check_output(['git', 'branch', '--show-current']).decode().strip()
        print_colored(f"Current branch: {current_branch}", Colors.BLUE)

        # Check if develop branch exists
        branches = subprocess.check_output(['git', 'branch', '-a']).decode()
        if 'develop' not in branches and 'origin/develop' not in branches:
            print_colored("The 'develop' branch does not exist!", Colors.RED)
            return False

        # First checkout develop branch
        subprocess.run(['git', 'checkout', 'develop'], check=True)
        print_colored("Switched to the 'develop' branch", Colors.GREEN)

        # Pull latest changes from develop
        print_colored("Pulling latest changes from develop...", Colors.BLUE)
        subprocess.run(['git', 'pull'], check=True)

        # Check if the new branch already exists
        if branch_name in branches:
            print_colored(f"Branch '{branch_name}' already exists!", Colors.YELLOW)
            return False

        # Create new branch from develop
        subprocess.run(['git', 'checkout', '-b', branch_name], check=True)
        print_colored(f"Successfully created and switched to branch: {branch_name}", Colors.GREEN)

        try:
            subprocess.run(['git', 'push', '-u', 'origin', branch_name], check=True)
            print_colored("Successfully pushed new branch to remote", Colors.GREEN)
        except subprocess.CalledProcessError:
            print_colored("Failed to push branch to remote. Branch created locally only.", Colors.YELLOW)

        return True

    except subprocess.CalledProcessError as e:
        print_colored(f"Error while working with git: {str(e)}", Colors.RED)
        return False
    except Exception as e:
        print_colored(f"An unexpected error occurred: {str(e)}", Colors.RED)
        return False

def git_checkout_and_pull(repo_path):
    """
    Switch to the 'develop' branch and pull the latest changes for the given repository.
    """
    try:
        os.chdir(repo_path)
        print_colored(f"\n=== Processing repository: {repo_path} ===", Colors.HEADER, bold=True)

        if not os.path.exists('.git'):
            print_colored(f"The folder {repo_path} is not a git repository!", Colors.RED)
            return False

        current_branch = subprocess.check_output(['git', 'branch', '--show-current']).decode().strip()
        print_colored(f"Current branch: {current_branch}", Colors.BLUE)

        branches = subprocess.check_output(['git', 'branch', '-a']).decode()
        if 'develop' not in branches and 'origin/develop' not in branches:
            print_colored("The 'develop' branch does not exist!", Colors.RED)
            return False

        subprocess.run(['git', 'checkout', 'develop'], check=True)
        print_colored("Switched to the 'develop' branch", Colors.GREEN)

        subprocess.run(['git', 'pull'], check=True)
        print_colored("Successfully pulled the latest changes", Colors.GREEN)

        return True

    except subprocess.CalledProcessError as e:
        print_colored(f"Error while working with git: {str(e)}", Colors.RED)
        return False
    except Exception as e:
        print_colored(f"An unexpected error occurred: {str(e)}", Colors.RED)
        return False

def load_repositories(config_file):
    """
    Load repository paths from a configuration file.
    """
    try:
        with open(config_file, 'r') as file:
            repositories = [line.strip() for line in file if line.strip()]
        return repositories
    except FileNotFoundError:
        print_colored(f"Configuration file '{config_file}' not found!", Colors.RED)
        return []
    except Exception as e:
        print_colored(f"An error occurred while reading the configuration file: {str(e)}", Colors.RED)
        return []

def main():
    # Setup argument parser with detailed help
    parser = argparse.ArgumentParser(
        prog='mob_repos.py',
        description=SCRIPT_DESCRIPTION,
        epilog=SCRIPT_EPILOG,
        formatter_class=argparse.RawDescriptionHelpFormatter
    )

    parser.add_argument(
        '-bc', '--branch-create',
        help='Create a new branch from develop in all repositories (e.g., feature/GND1111-testing-repo)',
        metavar='BRANCH_NAME'
    )

    parser.add_argument(
        '-c', '--config',
        help='Path to configuration file (default: repos.txt)',
        default='repos.txt',
        metavar='CONFIG_FILE'
    )

    args = parser.parse_args()

    # Load repository paths
    print_colored("Loading repositories from configuration file...", Colors.BLUE)
    repositories = load_repositories(args.config)

    if not repositories:
        print_colored("No repositories found in the configuration file. Exiting.", Colors.RED)
        return

    print_colored(f"Found {len(repositories)} repositories", Colors.GREEN)

    # Save the original working directory
    original_dir = os.getcwd()

    # Process each repository
    success_count = 0
    fail_count = 0

    for repo in repositories:
        if not os.path.exists(repo):
            print_colored(f"Repository {repo} does not exist!", Colors.RED)
            fail_count += 1
            continue

        if args.branch_create:
            success = create_new_branch(repo, args.branch_create)
        else:
            success = git_checkout_and_pull(repo)

        if success:
            success_count += 1
        else:
            fail_count += 1

        os.chdir(original_dir)

    # Print summary
    print_colored("\n=== Summary ===", Colors.HEADER, bold=True)
    print_colored(f"Successfully processed: {success_count} repositories", Colors.GREEN)
    if fail_count > 0:
        print_colored(f"Failed: {fail_count} repositories", Colors.RED)
    print_colored("Script execution completed!", Colors.GREEN)

if __name__ == "__main__":
    main()