// node tests/modules/model_download_test.js
const assert = require('node:assert/strict');
const fs = require('node:fs');
const html = fs.readFileSync(__dirname + '/../../web/static/index.html', 'utf8');
const scripts = [...html.matchAll(/<script>([\s\S]*?)<\/script>/g)].map(m => m[1]);
for (const script of scripts) new Function(script);
const source = scripts.join('\n');
const state = source.slice(source.indexOf('const modelDownloads ='), source.indexOf('async function renderCatalog()'));
const jobCode = source.slice(source.indexOf('async function longJob('), source.indexOf('// the runtime to download with'));
const model = 'nvidia/test';
let row, onEvent, rejectJob, calls = 0;
const agent = { value: 'pgx' };
function newRow() {
  const parts = {
    'button[data-tag]': { dataset: { tag: model, runtime: 'vllm' }, disabled: false },
    '.clog': { textContent: '', classList: { toggle() {} } },
    progress: { style: {}, removeAttribute() {} },
  };
  return { querySelector: s => parts[s] };
}
row = newRow();
const $ = id => id === 'agent' ? agent : { querySelectorAll: () => [row] };
const rpc = (method, params, cb, machine) => {
  calls++;
  assert.equal(machine, 'pgx');
  onEvent = cb;
  return new Promise((resolve, reject) => { rejectJob = reject; });
};
const refresh = () => { row = newRow(); ui.paintCatalogDownloads(); };
const ui = new Function('$', 'rpc', 't', 'gb', 'esc2', 'refreshModels',
  state + jobCode + '\nreturn { longJob, paintCatalogDownloads };')($, rpc, x => x, String, String, refresh);
(async () => {
  const first = ui.longJob(row.querySelector('button[data-tag]'), row.querySelector('progress'), row.querySelector('.clog'),
    'models.pull', { runtime: 'vllm', name: model });
  onEvent({ status: 'downloading shard', total: 100, completed: 30 });
  refresh();
  assert.equal(row.querySelector('button[data-tag]').disabled, true, 'refresh must keep busy button disabled');
  assert.match(row.querySelector('.clog').textContent, /downloading shard/);
  assert.equal(row.querySelector('progress').value, 30);
  await ui.longJob(row.querySelector('button[data-tag]'), null, row.querySelector('.clog'),
    'models.pull', { runtime: 'vllm', name: model });
  assert.equal(calls, 1, 'duplicate click must not submit a second RPC');
  agent.value = 'other';
  refresh();
  assert.equal(row.querySelector('button[data-tag]').disabled, false, 'busy state belongs to its machine');
  agent.value = 'pgx';
  rejectJob(new Error('wrong size'));
  await first;
  assert.equal(row.querySelector('button[data-tag]').disabled, false);
  assert.equal(row.querySelector('.clog').textContent, 'wrong size', 'error must survive catalog refresh');
  console.log('ok model download: duplicate clicks, refresh, machine isolation, error persistence; inline JS syntax');
})().catch(err => { console.error(err); process.exitCode = 1; });