---
- name: "📁 Create the shared model store before the daemons start"
  ansible.builtin.file:
    path: "{{ LMSTUDIO_SHARED_MODELS_PATH }}/{{ model.repo }}"
    state: directory
    mode: "0755"
  loop: "{{ LMSTUDIO_PRELOAD_MODELS }}"
  loop_control:
    loop_var: model
    label: "{{ model.file }}"

- name: "📁 Create the staging directory beside the model store"
  ansible.builtin.file:
    path: "{{ LMSTUDIO_SHARED_STAGING_PATH }}"
    state: directory
    mode: "0755"

- name: "🔍 Read what the shared model store already holds"
  ansible.builtin.stat:
    path: "{{ LMSTUDIO_SHARED_MODELS_PATH }}/{{ model.repo }}/{{ model.file }}"
    checksum_algorithm: sha256
  register: lmstudio_present
  changed_when: false
  loop: "{{ LMSTUDIO_PRELOAD_MODELS }}"
  loop_control:
    loop_var: model
    label: "{{ model.file }}"

- name: "🗄️ Measure the free space the shared model store offers"
  ansible.builtin.command: "df -P -B1 {{ LMSTUDIO_SHARED_MODELS_PATH | quote }}"
  register: lmstudio_export_space
  changed_when: false

- name: "🛑 Refuse to fetch models the shared store cannot hold"
  vars:
    _free: "{{ lmstudio_export_space.stdout_lines[1].split()[3] | int }}"
    _absent: >-
      {{ lmstudio_present.results
         | rejectattr('stat.exists')
         | map(attribute='model.bytes') | map('int') | sum }}
    _required: "{{ (_absent | int) + (LMSTUDIO_PRELOAD_LARGEST_BYTES | int) }}"
  ansible.builtin.fail:
    msg: >-
      The shared model store on {{ LMSTUDIO_NFS_SERVER }} has
      {{ (_free | int / 1024 / 1024 / 1024) | round(1) }} GiB free but needs
      {{ (_required | int / 1024 / 1024 / 1024) | round(1) }} GiB: it is missing
      {{ (_absent | int / 1024 / 1024 / 1024) | round(1) }} GiB of declared models,
      plus {{ (LMSTUDIO_PRELOAD_LARGEST_BYTES | int / 1024 / 1024 / 1024) | round(1) }}
      GiB for the largest one while it stages. Shrink AI_LOCAL_MODELS in
      group_vars/all/16_ai.yml or grow the export.
  when: _free | int < _required | int

- name: "⬇️ Fetch the models into the shared store"
  get_url_retry:
    url: "{{ model.url }}"
    dest: "{{ LMSTUDIO_SHARED_MODELS_PATH }}/{{ model.repo }}/{{ model.file }}"
    tmp_dest: "{{ LMSTUDIO_SHARED_STAGING_PATH }}"
    checksum: "sha256:{{ model.sha256 }}"
    mode: "0644"
  loop: "{{ LMSTUDIO_PRELOAD_MODELS }}"
  loop_control:
    loop_var: model
    label: "{{ model.file }}"
  register: lmstudio_fetch
  changed_when: false
  failed_when: false

- name: "🪞 Fetch from a mirror when the source did not deliver"
  get_url_retry:
    url: "{{ item.1 }}"
    dest: "{{ LMSTUDIO_SHARED_MODELS_PATH }}/{{ item.0.repo }}/{{ item.0.file }}"
    tmp_dest: "{{ LMSTUDIO_SHARED_STAGING_PATH }}"
    checksum: "sha256:{{ item.0.sha256 }}"
    mode: "0644"
  loop: "{{ LMSTUDIO_PRELOAD_MODELS | subelements('mirrors', skip_missing=True) }}"
  loop_control:
    label: "{{ item.0.file }} from {{ item.1 }}"
  register: lmstudio_fetch_mirror
  changed_when: false
  failed_when: false

- name: "🔍 Read back what the shared model store now holds"
  ansible.builtin.stat:
    path: "{{ LMSTUDIO_SHARED_MODELS_PATH }}/{{ model.repo }}/{{ model.file }}"
    checksum_algorithm: sha256
  register: lmstudio_seeded
  changed_when: false
  loop: "{{ LMSTUDIO_PRELOAD_MODELS }}"
  loop_control:
    loop_var: model
    label: "{{ model.file }}"

- name: "🛑 Fail when no source delivered a declared model"
  vars:
    _missing: >-
      {{ lmstudio_seeded.results
         | rejectattr('stat.exists')
         | map(attribute='model.file') | list }}
    _said: >-
      {{ (lmstudio_fetch.results | map(attribute='msg') | select | list)
         + (lmstudio_fetch_mirror.results | default([]) | map(attribute='msg') | select | list) }}
  ansible.builtin.fail:
    msg: >-
      The shared model store on {{ LMSTUDIO_NFS_SERVER }} is missing
      {{ _missing | join(', ') }}. What the sources answered:
      {{ _said | join(' | ') }}
  when: _missing | length > 0

- name: "🛑 Fail when a seeded model does not match its checksum"
  ansible.builtin.fail:
    msg: >-
      {{ item.model.file }} in the shared model store on {{ LMSTUDIO_NFS_SERVER }}
      hashes to {{ item.stat.checksum }}, but AI_LOCAL_MODELS declares
      {{ item.model.sha256 }}.
  loop: "{{ lmstudio_seeded.results | selectattr('stat.exists') | list }}"
  loop_control:
    label: "{{ item.model.file }}"
  when: item.stat.checksum != item.model.sha256