import base64

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


def _get_stat(action, path, task_vars):
    stat_result = action._execute_module(
        module_name='ansible.builtin.stat',
        module_args=dict(
            path=path,
            get_attributes=False,
            get_checksum=False,
            get_mime=False,
        ),
        task_vars=task_vars,
    )
    if stat_result.get('failed'):
        raise AnsibleActionFail(f"Couldn't stat file {path}", result=stat_result)
    return stat_result['stat']


def does_file_exist(action, path, task_vars):
    return _get_stat(action, path, task_vars)['exists']


def does_dir_exist(action, path, task_vars):
    stat = _get_stat(action, path, task_vars)
    return stat['exists'] and stat.get('isdir', False)


def read_file(action, path, task_vars, default=False):
    if default is not None:
        if not does_file_exist(action, path, task_vars):
            return default
    read_result = action._execute_module(
        module_name='ansible.builtin.slurp',
        module_args=dict(src=path),
        task_vars=task_vars,
    )
    if read_result.get('failed'):
        raise AnsibleActionFail(f"Couldn't read file {path}", result=read_result)
    return base64.b64decode(to_text(read_result['content'])).decode('utf-8')