#!/usr/bin/python

from ansible.errors import AnsibleActionFail
from ansible.plugins.action import ActionBase
from ansible_collections.tensin.infra.plugins.module_utils import file, git


class ActionModule(ActionBase):
    def run(self, tmp=None, task_vars=None):
        result = super().run(tmp, task_vars)

        repo = self._task.args["repo"]
        dest = self._task.args["dest"]
        accept_hostkey = self._task.args.get("accept_hostkey", False)
        remote = self._task.args.get("remote", "origin")
        version = self._task.args["version"]

        builtin_args = {
            "repo": repo,
            "dest": dest,
            "accept_hostkey": accept_hostkey,
            "remote": remote,
            "version": version,
        }

        if not file.does_dir_exist(self, dest, task_vars):
            # If the destination directory doesn't exist, delegate to the
            # builtin module.
            return self._execute_module(
                module_name="ansible.builtin.git",
                module_args=builtin_args,
                task_vars=task_vars,
            )

        # ansible.builtin.git would just remove everything, fuck that.
        if not git.is_work_tree(self, dest, task_vars):
            raise AnsibleActionFail(
                f"Destination {dest} exists but is not a repository"
            )

        # Again, ansible.builtin.git would just remove everything, fuck that.
        current_url = git.get_remote_url(self, dest, task_vars, remote=remote)
        if current_url != repo:
            raise AnsibleActionFail(
                f"Existing remote URL '{current_url}' doesn't match '{repo}' for {dest}"
            )

        # ansible.builtin.git would abort if the `force` parameter is set to
        # `false`, but I don't want to depend on default values.
        dirty_files = git.get_dirty_file_list(self, dest, task_vars)
        if dirty_files:
            raise AnsibleActionFail(f"There are local changes in {dest}")

        # Fetch the remote.
        git.update_remote(self, dest, remote, task_vars, accept_hostkey=accept_hostkey)

        # If the current HEAD is not an ancestor of the requested version, fail
        # (again, unlike the builtin module).
        is_ancestor = git.is_ancestor(self, dest, remote, version, task_vars)
        if not is_ancestor:
            raise AnsibleActionFail(
                f"HEAD in {dest} is not an ancestor of {remote}/{version}"
            )

        # Finally, everything looks good, delegate to the builtin module.
        return self._execute_module(
            module_name="ansible.builtin.git",
            module_args=builtin_args,
            task_vars=task_vars,
        )