#!/usr/bin/env python

import docker

from ansible.module_utils._text import to_native
from ansible.module_utils.basic import AnsibleModule


ANSIBLE_METADATA = {
    'metadata_version': '1.1',
    'status': ['preview'],
    'supported_by': 'community'
}


def get_docker_containers(module, container_list):
    from requests.exceptions import ConnectionError
    if len(container_list) == 0:
        pass
    client = docker.from_env()
    try:
        containers = client.containers.list()
        docker_list = [{'container_name': i.attrs['Name'].strip('/')} for i
                       in containers if i.attrs['Name'].strip('/') in
                       container_list]

        return docker_list
    except docker.errors.APIError as e:
        module.fail_json(
            msg='Error listing containers: {}'.format(to_native(e)))
    except ConnectionError as e:
        module.fail_json(
            msg='Error connecting to Docker: {}'.format(to_native(e))
        )


def get_systemd_services(module, service_unit_list):
    if len(service_unit_list) == 0:
        pass
    systemctl_path = \
        module.get_bin_path("systemctl",
                            opt_dirs=["/usr/bin", "/usr/local/bin"])
    if systemctl_path is None:
        return None
    systemd_list = []
    for i in service_unit_list:
        rc, stdout, stderr = \
            module.run_command("{} is-enabled {}".format(systemctl_path, i),
                               use_unsafe_shell=True)
        if stdout == "enabled\n":
            state_val = "enabled"
        else:
            state_val = "disabled"
        systemd_list.append({"name": i, "state": state_val})
    return systemd_list


def run_module():

    module_args = dict(operations_service_map=dict(type=dict, required=True),
                       operations_service_names=dict(type=list, required=True),
                       )

    module = AnsibleModule(
        argument_spec=module_args,
        supports_check_mode=True
    )

    service_map = module.params.get('service_map')
    service_names = module.params.get('services')

    services_to_restart = {i: service_map[i] for i in service_names}

    container_list = [j for i in service_names
                      for j in services_to_restart[i]['container_name']]
    service_unit_list = [j for i in service_names
                         for j in services_to_restart[i]['systemd_unit']]

    result = dict(
        ansible_facts=dict(
            docker_containers_to_restart=get_docker_containers(
                module, container_list),
            systemd_services_to_restart=get_systemd_services(
                module, service_unit_list),
        )
    )

    if module.check_mode:
        return result

    module.exit_json(**result)


def main():
    run_module()


if __name__ == "__main__":
    main()