LLM-Labs-Local/ansible/roles/preflight/tasks/main.yml

- name: Detect WSL
  set_fact:
    courseware_is_wsl: "{{ 'microsoft' in ansible_kernel | lower or 'wsl' in ansible_kernel | lower }}"

- name: Fail on unsupported operating systems
  fail:
    msg: "Supported platforms are Apple Silicon macOS and Debian-family Linux/WSL."
  when: ansible_system not in ["Darwin", "Linux"]

- name: Fail on unsupported macOS architecture
  fail:
    msg: "This installer supports Apple Silicon Macs only."
  when:
    - ansible_system == "Darwin"
    - ansible_architecture not in ["arm64", "aarch64"]

- name: Fail on undersized macOS systems
  fail:
    msg: "This courseware assumes a modern Apple Silicon Mac with at least 16 GB of unified memory."
  when:
    - ansible_system == "Darwin"
    - (ansible_memtotal_mb | int) < 16000

- name: Check for Xcode command line tools
  command: xcode-select -p
  register: courseware_xcode_select
  changed_when: false
  when: ansible_system == "Darwin"

- name: Check for Homebrew
  command: which brew
  register: courseware_brew_check
  changed_when: false
  failed_when: false
  when: ansible_system == "Darwin"

- name: Fail when Xcode command line tools are missing
  fail:
    msg: "Install Xcode Command Line Tools first with 'xcode-select --install'."
  when:
    - ansible_system == "Darwin"
    - courseware_xcode_select.rc != 0

- name: Fail when Homebrew is missing
  fail:
    msg: "Install Homebrew first from https://brew.sh/."
  when:
    - ansible_system == "Darwin"
    - courseware_brew_check.rc != 0

- name: Fail on unsupported Linux family
  fail:
    msg: "This installer currently supports Debian and Ubuntu only."
  when:
    - ansible_system == "Linux"
    - ansible_os_family != "Debian"

- name: Query NVIDIA GPU memory
  command: nvidia-smi --query-gpu=memory.total --format=csv,noheader,nounits
  register: courseware_gpu_memory
  changed_when: false
  failed_when: false
  when: ansible_system == "Linux"

- name: Query NVIDIA GPU names
  command: nvidia-smi --query-gpu=name --format=csv,noheader
  register: courseware_gpu_names
  changed_when: false
  failed_when: false
  when: ansible_system == "Linux"

- name: Fail when no supported NVIDIA GPU is visible
  fail:
    msg: "Linux/WSL requires an NVIDIA GPU visible to nvidia-smi."
  when:
    - ansible_system == "Linux"
    - courseware_gpu_memory.rc != 0

- name: Fail when GPU VRAM is below baseline
  fail:
    msg: "This build assumes at least 8 GB of VRAM on Linux/WSL."
  when:
    - ansible_system == "Linux"
    - (courseware_gpu_memory.stdout_lines | map('int') | max) < 8192

- name: Check for CUDA compiler on Linux
  command: which nvcc
  register: courseware_preflight_nvcc
  changed_when: false
  failed_when: false
  when: ansible_system == "Linux"

- name: Check for CUDA runtime header on Linux
  stat:
    path: "{{ item }}"
  loop:
    - /usr/local/cuda/include/cuda_runtime.h
    - /usr/include/cuda_runtime.h
  register: courseware_preflight_cuda_headers
  when: ansible_system == "Linux"

- name: Set CUDA toolkit readiness
  set_fact:
    courseware_cuda_toolkit_ready: >-
      {{
        courseware_preflight_nvcc.rc == 0
        or (courseware_preflight_cuda_headers.results | selectattr('stat.exists', 'equalto', true) | list | length > 0)
      }}
  when: ansible_system == "Linux"

- name: Query distro CUDA toolkit apt candidate
  command: apt-cache policy nvidia-cuda-toolkit
  register: courseware_preflight_cuda_toolkit_policy
  changed_when: false
  failed_when: false
  when:
    - ansible_system == "Linux"
    - ansible_os_family == "Debian"

- name: Set distro CUDA toolkit package availability
  set_fact:
    courseware_preflight_cuda_toolkit_package_available: >-
      {{
        courseware_preflight_cuda_toolkit_policy.rc == 0
        and 'Candidate: (none)' not in courseware_preflight_cuda_toolkit_policy.stdout
      }}
  when:
    - ansible_system == "Linux"
    - ansible_os_family == "Debian"

- name: Fail when automatic WSL CUDA bootstrap is unsupported
  fail:
    msg: "Automatic CUDA bootstrap currently supports Ubuntu x86_64 on WSL only. For other WSL distros, install the CUDA toolkit manually before rerunning."
  when:
    - ansible_system == "Linux"
    - courseware_is_wsl
    - not courseware_cuda_toolkit_ready
    - ansible_distribution != "Ubuntu" or ansible_architecture not in ["x86_64", "amd64"]

- name: Install distro CUDA toolkit on Ubuntu WSL when available
  become: true
  apt:
    name: nvidia-cuda-toolkit
    state: present
    update_cache: true
  when:
    - ansible_system == "Linux"
    - courseware_is_wsl
    - not courseware_cuda_toolkit_ready
    - ansible_distribution == "Ubuntu"
    - ansible_architecture in ["x86_64", "amd64"]
    - courseware_preflight_cuda_toolkit_package_available | default(false)

- name: Recheck CUDA compiler after distro toolkit install
  command: which nvcc
  register: courseware_preflight_nvcc_after_distro_install
  changed_when: false
  failed_when: false
  when:
    - ansible_system == "Linux"
    - courseware_is_wsl
    - not courseware_cuda_toolkit_ready
    - ansible_distribution == "Ubuntu"
    - ansible_architecture in ["x86_64", "amd64"]
    - courseware_preflight_cuda_toolkit_package_available | default(false)

- name: Recheck CUDA runtime header after distro toolkit install
  stat:
    path: "{{ item }}"
  loop:
    - /usr/local/cuda/include/cuda_runtime.h
    - /usr/include/cuda_runtime.h
  register: courseware_preflight_cuda_headers_after_distro_install
  when:
    - ansible_system == "Linux"
    - courseware_is_wsl
    - not courseware_cuda_toolkit_ready
    - ansible_distribution == "Ubuntu"
    - ansible_architecture in ["x86_64", "amd64"]
    - courseware_preflight_cuda_toolkit_package_available | default(false)

- name: Refresh CUDA toolkit readiness after distro toolkit install
  set_fact:
    courseware_cuda_toolkit_ready: >-
      {{
        courseware_preflight_nvcc_after_distro_install.rc == 0
        or (courseware_preflight_cuda_headers_after_distro_install.results | selectattr('stat.exists', 'equalto', true) | list | length > 0)
      }}
  when:
    - ansible_system == "Linux"
    - courseware_is_wsl
    - not courseware_cuda_toolkit_ready
    - ansible_distribution == "Ubuntu"
    - ansible_architecture in ["x86_64", "amd64"]
    - courseware_preflight_cuda_toolkit_package_available | default(false)

- name: Remove legacy NVIDIA CUDA apt key when preparing WSL toolkit install
  become: true
  command: apt-key del 7fa2af80
  register: courseware_wsl_cuda_apt_key_delete
  changed_when: courseware_wsl_cuda_apt_key_delete.rc == 0
  failed_when: false
  when:
    - ansible_system == "Linux"
    - courseware_is_wsl
    - not courseware_cuda_toolkit_ready
    - ansible_distribution == "Ubuntu"
    - ansible_architecture in ["x86_64", "amd64"]

- name: Download NVIDIA WSL CUDA apt pin
  become: true
  get_url:
    url: "{{ courseware_wsl_cuda_pin_url }}"
    dest: "{{ courseware_wsl_cuda_pin_dest }}"
    mode: "0644"
    force: true
  when:
    - ansible_system == "Linux"
    - courseware_is_wsl
    - not courseware_cuda_toolkit_ready
    - ansible_distribution == "Ubuntu"
    - ansible_architecture in ["x86_64", "amd64"]

- name: Download NVIDIA WSL CUDA local installer
  get_url:
    url: "{{ courseware_wsl_cuda_installer_url }}"
    dest: "{{ courseware_wsl_cuda_installer_local_path }}"
    mode: "0644"
    force: false
  when:
    - ansible_system == "Linux"
    - courseware_is_wsl
    - not courseware_cuda_toolkit_ready
    - ansible_distribution == "Ubuntu"
    - ansible_architecture in ["x86_64", "amd64"]

- name: Install NVIDIA WSL CUDA local repository package
  become: true
  apt:
    deb: "{{ courseware_wsl_cuda_installer_local_path }}"
    state: present
  when:
    - ansible_system == "Linux"
    - courseware_is_wsl
    - not courseware_cuda_toolkit_ready
    - ansible_distribution == "Ubuntu"
    - ansible_architecture in ["x86_64", "amd64"]

- name: Find NVIDIA WSL CUDA keyring
  become: true
  find:
    paths: "{{ courseware_wsl_cuda_repo_dir }}"
    patterns: "cuda-*-keyring.gpg"
    file_type: file
  register: courseware_wsl_cuda_keyring
  when:
    - ansible_system == "Linux"
    - courseware_is_wsl
    - not courseware_cuda_toolkit_ready
    - ansible_distribution == "Ubuntu"
    - ansible_architecture in ["x86_64", "amd64"]

- name: Fail when NVIDIA WSL CUDA keyring is missing
  fail:
    msg: "The NVIDIA WSL CUDA repository package was installed, but its keyring file was not found under {{ courseware_wsl_cuda_repo_dir }}."
  when:
    - ansible_system == "Linux"
    - courseware_is_wsl
    - not courseware_cuda_toolkit_ready
    - ansible_distribution == "Ubuntu"
    - ansible_architecture in ["x86_64", "amd64"]
    - (courseware_wsl_cuda_keyring.files | length) == 0

- name: Copy NVIDIA WSL CUDA keyring into trusted keyrings
  become: true
  copy:
    src: "{{ courseware_wsl_cuda_keyring.files[0].path }}"
    dest: "/usr/share/keyrings/{{ courseware_wsl_cuda_keyring.files[0].path | basename }}"
    remote_src: true
    mode: "0644"
  when:
    - ansible_system == "Linux"
    - courseware_is_wsl
    - not courseware_cuda_toolkit_ready
    - ansible_distribution == "Ubuntu"
    - ansible_architecture in ["x86_64", "amd64"]
    - (courseware_wsl_cuda_keyring.files | length) > 0

- name: Install NVIDIA WSL CUDA toolkit
  become: true
  apt:
    name: "{{ courseware_wsl_cuda_toolkit_package }}"
    state: present
    update_cache: true
  when:
    - ansible_system == "Linux"
    - courseware_is_wsl
    - not courseware_cuda_toolkit_ready
    - ansible_distribution == "Ubuntu"
    - ansible_architecture in ["x86_64", "amd64"]

- name: Recheck CUDA compiler after WSL toolkit install
  command: which nvcc
  register: courseware_preflight_nvcc_after_install
  changed_when: false
  failed_when: false
  when:
    - ansible_system == "Linux"
    - courseware_is_wsl
    - not courseware_cuda_toolkit_ready
    - ansible_distribution == "Ubuntu"
    - ansible_architecture in ["x86_64", "amd64"]

- name: Fail when CUDA toolkit is still missing after WSL install attempt
  fail:
    msg: "The NVIDIA WSL CUDA toolkit install completed, but `nvcc` is still missing. Verify the repository package and rerun the installer."
  when:
    - ansible_system == "Linux"
    - courseware_is_wsl
    - not courseware_cuda_toolkit_ready
    - ansible_distribution == "Ubuntu"
    - ansible_architecture in ["x86_64", "amd64"]
    - courseware_preflight_nvcc_after_install.rc != 0

- name: Set runtime binary defaults
  set_fact:
    courseware_python_bin: >-
      {{ '/opt/homebrew/opt/python@3.11/bin/python3.11' if ansible_system == 'Darwin' else '/usr/bin/python3' }}
    courseware_ollama_bin: "ollama"