// compare/probe.js — model probe/check system
import state from './state.js';
import { WAVE_FRAMES } from './icons.js';
import uiModule from '../ui.js';
import spinnerModule from '../spinner.js';

function _clearProbeWaves() {
  const rows = document.querySelectorAll('.compare-probe-row');
  rows.forEach(r => { if (r._waveInterval) { clearInterval(r._waveInterval); r._waveInterval = null; } });
}

async function _checkUnprobed() {
  const unprobed = state._selectedModels.filter(m => !state._probed.has(m.model));
  if (unprobed.length === 0) {
    if (uiModule) uiModule.showToast('All models verified');
    return;
  }

  // Whirlpool loader on the Probe button while the check runs.
  const _btn = document.getElementById('compare-check-btn');
  let _btnHTML = null, _wp = null;
  if (_btn) {
    _btnHTML = _btn.innerHTML;
    _btn.disabled = true;
    _btn.style.opacity = '0.7';
    try {
      _wp = spinnerModule.createWhirlpool(14);
      _btn.innerHTML = '';
      _btn.appendChild(_wp.element);
    } catch (_) { /* spinner best-effort */ }
  }

  // Quick inline probe — show toast with results
  const isBlind = state._blindMode;
  let ok = 0, fail = 0;
  try {
  for (const m of unprobed) {
    try {
      const _imageModelPrefixes = ['dall-e', 'gpt-image', 'chatgpt-image', 'stable-diffusion', 'sdxl', 'flux', 'midjourney'];
      if (_imageModelPrefixes.some(p => m.model.toLowerCase().includes(p))) {
        state._probed.add(m.model);
        ok++;
        continue;
      }
      const res = await fetch(`${state.API_BASE}/api/probe-selected`, {
        method: 'POST', credentials: 'same-origin',
        headers: { 'Content-Type': 'application/json' },
        body: JSON.stringify({ models: [{ endpoint_id: m.endpointId || '', model: m.model, endpoint: m.endpoint || '' }] }),
      });
      const data = await res.json();
      const result = (data.results || [])[0];
      if (result && result.status === 'ok') {
        state._probed.add(m.model);
        ok++;
      } else {
        fail++;
        const name = isBlind ? 'a model' : (m.name || m.model.split('/').pop());
        if (uiModule) uiModule.showToast(`${name} failed: ${result?.error || 'unknown'}`, 5000);
      }
    } catch (e) {
      fail++;
    }
  }
  if (fail === 0) {
    if (uiModule) uiModule.showToast(`${ok} model${ok > 1 ? 's' : ''} verified`);
  }
  } finally {
    // Restore the Probe button (its label/visibility is refreshed below).
    if (_btn) {
      _btn.disabled = false;
      _btn.style.opacity = '';
      if (_btnHTML !== null) _btn.innerHTML = _btnHTML;
    }
    if (window._updateCheckBtnState) window._updateCheckBtnState();
  }
}

export { _clearProbeWaves, _checkUnprobed };