---
- name: Set_fact trigger_restart
  ansible.builtin.set_fact:
    trigger_restart: true
  loop: "{{ groups[osd_group_name] }}"
  when: hostvars[item]['handler_osd_status'] | default(False) | bool
  run_once: true

- name: Osd handler
  when: trigger_restart | default(False) | bool
  block:
    - name: Set _osd_handler_called before restart
      ansible.builtin.set_fact:
        _osd_handler_called: true

    - name: Unset noup flag
      ceph_osd_flag:
        name: noup
        cluster: "{{ cluster }}"
        state: absent
      environment:
        CEPH_CONTAINER_IMAGE: "{{ ceph_docker_registry + '/' + ceph_docker_image + ':' + ceph_docker_image_tag if containerized_deployment | bool else None }}"
        CEPH_CONTAINER_BINARY: "{{ container_binary }}"
      delegate_to: "{{ groups[mon_group_name][0] }}"
      run_once: true

    # This does not just restart OSDs but everything else too. Unfortunately
    # at this time the ansible role does not have an OSD id list to use
    # for restarting them specifically.
    # This does not need to run during a rolling update as the playbook will
    # restart all OSDs using the tasks "start ceph osd" or
    # "restart containerized ceph osd"
    - name: Copy osd restart script
      ansible.builtin.template:
        src: restart_osd_daemon.sh.j2
        dest: "{{ tmpdirpath.path }}/restart_osd_daemon.sh"
        owner: root
        group: root
        mode: "0750"
      when: tmpdirpath.path is defined

    - name: Get pool list
      ansible.builtin.command: "{{ ceph_cmd }} --cluster {{ cluster }} osd pool ls detail -f json"
      register: pool_list
      delegate_to: "{{ groups.get(mon_group_name, [])[0] }}"
      run_once: true
      changed_when: false
      check_mode: false

    - name: Get balancer module status
      ansible.builtin.command: "{{ ceph_cmd }} --cluster {{ cluster }} balancer status -f json"
      register: balancer_status
      run_once: true
      delegate_to: "{{ groups[mon_group_name][0] }}"
      changed_when: false
      check_mode: false

    - name: Set_fact pools_pgautoscaler_mode
      ansible.builtin.set_fact:
        pools_pgautoscaler_mode: "{{ pools_pgautoscaler_mode | default([]) | union([{'name': item.pool_name, 'mode': item.pg_autoscale_mode}]) }}"
      run_once: true
      with_items: "{{ pool_list.stdout | default('{}') | from_json }}"

    - name: Disable balancer
      ansible.builtin.command: "{{ ceph_cmd }} --cluster {{ cluster }} balancer off"
      run_once: true
      delegate_to: "{{ groups[mon_group_name][0] }}"
      changed_when: false
      when: (balancer_status.stdout | from_json)['active'] | bool

    - name: Disable pg autoscale on pools
      ceph_pool:
        name: "{{ item.name }}"
        cluster: "{{ cluster }}"
        pg_autoscale_mode: false
      with_items: "{{ pools_pgautoscaler_mode }}"
      delegate_to: "{{ groups.get(mon_group_name, [])[0] }}"
      run_once: true
      when:
        - pools_pgautoscaler_mode is defined
        - item.mode == 'on'
      environment:
        CEPH_CONTAINER_IMAGE: "{{ ceph_docker_registry + '/' + ceph_docker_image + ':' + ceph_docker_image_tag if containerized_deployment | bool else None }}"
        CEPH_CONTAINER_BINARY: "{{ container_binary }}"

    - name: Restart ceph osds daemon(s)
      ansible.builtin.command: /usr/bin/env bash {{ hostvars[item]['tmpdirpath']['path'] }}/restart_osd_daemon.sh
      when:
        - hostvars[item]['handler_osd_status'] | default(False) | bool
        - handler_health_osd_check | bool
        - hostvars[item]['_osd_handler_called'] | default(False) | bool
        - hostvars[item].tmpdirpath.path is defined
      with_items: "{{ groups[osd_group_name] | intersect(ansible_play_batch) }}"
      delegate_to: "{{ item }}"
      changed_when: false
      run_once: true

    - name: Set _osd_handler_called after restart
      ansible.builtin.set_fact:
        _osd_handler_called: false

    - name: Re-enable pg autoscale on pools
      ceph_pool:
        name: "{{ item.name }}"
        cluster: "{{ cluster }}"
        pg_autoscale_mode: true
      with_items: "{{ pools_pgautoscaler_mode }}"
      run_once: true
      delegate_to: "{{ groups.get(mon_group_name, [])[0] }}"
      when:
        - pools_pgautoscaler_mode is defined
        - item.mode == 'on'
      environment:
        CEPH_CONTAINER_IMAGE: "{{ ceph_docker_registry + '/' + ceph_docker_image + ':' + ceph_docker_image_tag if containerized_deployment | bool else None }}"
        CEPH_CONTAINER_BINARY: "{{ container_binary }}"

    - name: Re-enable balancer
      ansible.builtin.command: "{{ ceph_cmd }} --cluster {{ cluster }} balancer on"
      run_once: true
      delegate_to: "{{ groups[mon_group_name][0] }}"
      changed_when: false
      when: (balancer_status.stdout | from_json)['active'] | bool