import argparse
import os

from git_aggregator import main as gitaggregate_cmd
from git_aggregator.utils import working_directory_keeper
from jinja2 import Template
from loguru import logger
from plumbum import local
from plumbum.cmd import mkdir
from virtualenv.run import cli_run as virtualenv_cmd


def ensure_folder_exists(folder_path, mode=False):
    """Create a local folder.
    - directory is created if it doesn't exist.
    - mode is applied if defined.
    - a log is done at INFO level.
    """
    if not folder_path.exists():
        cmd = ["--parents", folder_path]
        if mode:
            cmd = ["--mode", "777"] + cmd
        logger.info("Creating folder '%s' ..." % (folder_path))
        mkdir(cmd)


def ensure_file_exists_from_template(file_path, template_name, **args):

    template = Template(template_name)
    output = template.render(args)

    if file_path.exists():
        # Check if content is different
        with open(file_path, "r") as file:
            data = file.read()
            file.close()
            if data == output:
                return

        log_text = "Updating file '%s' from template ..." % (file_path)
    else:
        log_text = "Creating file '%s' from template ..." % (file_path)

    with open(file_path, "w") as f:
        logger.info(log_text)
        f.write(output)
        f.close()


def git_aggregate(folder_path, config_path):
    args = argparse.Namespace(
        command="aggregate",
        config=str(config_path),
        jobs=1,
        dirmatch=None,
        do_push=False,
        expand_env=False,
        env_file=None,
        force=False,
    )
    with working_directory_keeper:
        os.chdir(folder_path)
        logger.info(
            "Gitaggregate source code for %s. This can take a while ..."
            % config_path
        )
        gitaggregate_cmd.run(args)


def create_virtualenv(folder_path, python_version):
    """
    Create a virtual env named ``env`` in the ``folder_path`` folder
    with the given ``python_version``.
    """

    with local.cwd(folder_path):
        logger.info(
            "Create Virtual Env in %s with version %s"
            % (folder_path, python_version)
        )
        virtualenv_cmd(["env", "--python", python_version])