---
# Wait for upgrade
- name: Confirm node upgrade
  pause:
    echo: yes
    prompt: "Ready to upgrade node? (Press Enter to continue or Ctrl+C for other options)"
  when:
    - upgrade_node_confirm

- name: Wait before upgrade node
  pause:
    seconds: "{{ upgrade_node_pause_seconds }}"
  when:
    - not upgrade_node_confirm
    - upgrade_node_pause_seconds != 0

# Node Ready: type = ready, status = True
# Node NotReady: type = ready, status = Unknown
- name: See if node is in ready state
  command: >
    {{ kubectl }} get node {{ kube_override_hostname | default(inventory_hostname) }}
    -o jsonpath='{ range .status.conditions[?(@.type == "Ready")].status }{ @ }{ end }'
  register: kubectl_node_ready
  delegate_to: "{{ groups['kube_control_plane'][0] }}"
  failed_when: false
  changed_when: false

# SchedulingDisabled: unschedulable = true
# else unschedulable key doesn't exist
- name: See if node is schedulable
  command: >
    {{ kubectl }} get node {{ kube_override_hostname | default(inventory_hostname) }}
    -o jsonpath='{ .spec.unschedulable }'
  register: kubectl_node_schedulable
  delegate_to: "{{ groups['kube_control_plane'][0] }}"
  failed_when: false
  changed_when: false

- name: Set if node needs cordoning
  set_fact:
    # noqa: jinja[spacing]
    needs_cordoning: >-
      {% if (kubectl_node_ready.stdout == "True" and not kubectl_node_schedulable.stdout) or upgrade_node_always_cordon -%}
      true
      {%- else -%}
      false
      {%- endif %}

- name: Node draining
  delegate_to: "{{ groups['kube_control_plane'][0] }}"
  when:
    - needs_cordoning
  block:
    - name: Cordon node
      command: "{{ kubectl }} cordon {{ kube_override_hostname | default(inventory_hostname) }}"
      delegate_to: "{{ groups['kube_control_plane'][0] }}"
      changed_when: true

    - name: Drain node
      command: >-
        {{ kubectl }} drain
        --force
        --ignore-daemonsets
        --grace-period {{ hostvars['localhost']['drain_grace_period_after_failure'] | default(drain_grace_period) }}
        --timeout {{ hostvars['localhost']['drain_timeout_after_failure'] | default(drain_timeout) }}
        --delete-emptydir-data {{ kube_override_hostname | default(inventory_hostname) }}
        {% if drain_pod_selector %}--pod-selector '{{ drain_pod_selector }}'{% endif %}
      when: drain_nodes
      register: result
      failed_when:
        - result.rc != 0
        - not drain_fallback_enabled
      until: result.rc == 0
      retries: "{{ drain_retries }}"
      delay: "{{ drain_retry_delay_seconds }}"

    - name: Drain fallback
      when:
        - drain_nodes
        - drain_fallback_enabled
        - result.rc != 0
      block:
        - name: Set facts after regular drain has failed
          set_fact:
            drain_grace_period_after_failure: "{{ drain_fallback_grace_period }}"
            drain_timeout_after_failure: "{{ drain_fallback_timeout }}"
          delegate_to: localhost
          delegate_facts: yes
          run_once: yes

        - name: Drain node - fallback with disabled eviction
          command: >-
            {{ kubectl }} drain
            --force
            --ignore-daemonsets
            --grace-period {{ drain_fallback_grace_period }}
            --timeout {{ drain_fallback_timeout }}
            --delete-emptydir-data {{ kube_override_hostname | default(inventory_hostname) }}
            {% if drain_pod_selector %}--pod-selector '{{ drain_pod_selector }}'{% endif %}
            --disable-eviction
          register: drain_fallback_result
          until: drain_fallback_result.rc == 0
          retries: "{{ drain_fallback_retries }}"
          delay: "{{ drain_fallback_retry_delay_seconds }}"
          changed_when: drain_fallback_result.rc == 0

  rescue:
    - name: Set node back to schedulable
      command: "{{ kubectl }} uncordon {{ kube_override_hostname | default(inventory_hostname) }}"
      when: upgrade_node_uncordon_after_drain_failure
    - name: Fail after rescue
      fail:
        msg: "Failed to drain node {{ kube_override_hostname | default(inventory_hostname) }}"
      when: upgrade_node_fail_if_drain_fails