import os
import shlex

from ansible.errors import AnsibleActionFail
from ansible.module_utils.common.text.converters import to_text


def is_work_tree(action, repo_dir, task_vars):
    cmd_result = action._execute_module(
        module_name='ansible.builtin.command',
        module_args=dict(
            argv=['git', 'rev-parse', '--is-inside-work-tree'],
            chdir=repo_dir,
        ),
        task_vars=task_vars,
    )
    return not cmd_result.get('failed')


def get_status(action, repo_dir, task_vars):
    cmd_result = action._execute_module(
        module_name='ansible.builtin.command',
        module_args=dict(
            argv=['git', 'status', '--porcelain'],
            chdir=repo_dir,
        ),
        task_vars=task_vars,
    )
    if cmd_result.get('failed'):
        raise AnsibleActionFail(f"Couldn't get git status for {repo_dir}", result=cmd_result)
    return to_text(cmd_result['stdout'])


def get_dirty_file_list(action, repo_dir, task_vars):
    status = get_status(action, repo_dir, task_vars)
    status = status.splitlines()
    status = [line[3:] for line in status]
    return status


def get_remote_url(action, repo_dir, task_vars, remote='origin'):
    cmd_result = action._execute_module(
        module_name='ansible.builtin.command',
        module_args=dict(
            argv=['git', 'config', '--local', f'remote.{remote}.url'],
            chdir=repo_dir,
        ),
        task_vars=task_vars,
    )
    if cmd_result.get('failed'):
        raise AnsibleActionFail(f"Couldn't read the URL of remote '{remote}' in {repo_dir}", result=cmd_result)
    return cmd_result['stdout']


def update_remote(action, repo_dir, remote, task_vars, accept_hostkey=False):
    ssh = os.environ.get('GIT_SSH', 'ssh')
    ssh_opts = ''
    if accept_hostkey:
        ssh_opts += ' -o StrictHostKeyChecking=no'
    ssh_command = shlex.quote(ssh) + ssh_opts

    cmd_result = action._execute_module(
        module_name='ansible.builtin.command',
        module_args=dict(
            argv=['env', f'GIT_SSH_COMMAND={ssh_command}', 'git', 'remote', 'update', remote],
            chdir=repo_dir,
        ),
        task_vars=task_vars,
    )
    if cmd_result.get('failed'):
        raise AnsibleActionFail(f"Couldn't update remote '{remote}' for {repo_dir}", result=cmd_result)


def is_ancestor(action, repo_dir, remote, version, task_vars):
    looks_like_hash = all(((c >= '0' and c <= '9') or (c >= 'a' and c <= 'f') for c in version))

    remote_versions = [f'{remote}/{version}']
    if looks_like_hash:
        remote_versions += [version]

    for _version in remote_versions:
        cmd_result = action._execute_module(
            module_name='ansible.builtin.command',
            module_args=dict(
                argv=['git', 'merge-base', '--is-ancestor', 'HEAD', _version],
                chdir=repo_dir,
            ),
            task_vars=task_vars,
        )
        if cmd_result.get('failed'):
            if cmd_result.get('stderr'):
                continue
        else:
            return True

    raise AnsibleActionFail(f"Couldn't deduce how the HEAD of {repo_dir} relates to remote version {remote}/{version}")