---
- name: Check whether reboot is required
  ansible.builtin.stat:
    path: /var/run/reboot-required
  register: nvidia_runtime_reboot_required

- name: Warn when reboot is required but disabled
  ansible.builtin.debug:
    msg: >-
      NVIDIA packages were installed but reboot is required.
      Set nvidia_runtime_reboot_if_needed=true to allow automatic reboot.
  when:
    - nvidia_runtime_reboot_required.stat.exists
    - not nvidia_runtime_reboot_if_needed | bool

- name: Reboot host when required and enabled
  ansible.builtin.reboot:
    msg: "Reboot triggered by nvidia_runtime_setup role"
    reboot_timeout: "{{ nvidia_runtime_reboot_timeout }}"
  when:
    - nvidia_runtime_reboot_required.stat.exists
    - nvidia_runtime_reboot_if_needed | bool