// Runs the AI modules page code (the part of web/static/index.html between the markers) against fake data:
//   node tests/modules/ui_modules_test.js
const fs = require('fs');
const html = fs.readFileSync(__dirname + '/../../web/static/index.html', 'utf8');
const src = [...html.matchAll(/<script>([\s\S]*?)<\/script>/g)].map(x => x[1]).join('\n');
const a = src.indexOf('// ---- AI modules page'), b = src.indexOf("setInterval(() => { if (view === 'modules')");
if (a < 0 || b < 0) throw new Error('markers not found');
const code = src.slice(a, b);
const cat = JSON.parse(fs.readFileSync(__dirname + '/../../catalog/modules.json', 'utf8'));

const els = {};
const $ = id => els[id] || (els[id] = { style: {}, innerHTML: '', value: 'pgx' });
const calls = [];
let machine = { free_bytes: 500e9, total_bytes: 1e12, dir: '/mnt/models/modules', capabilities: cat.capabilities.map(c => ({ id: c.id, installed: false })), engines: cat.engines.map(e => ({ engine: e.id, state: '', image_present: false })) };
const rpc = async (method, params) => { calls.push(method + ':' + JSON.stringify(params)); if (method === 'modules.list') return machine; return null; };
const esc2 = s => String(s == null ? '' : s);
const fetchFn = async () => ({ json: async () => cat });
const body = code + '\nreturn { modRefresh, renderModules, modSizes, modHave, setSel: ids => { modSel = new Set(ids); }, clickBtn: ds => $("modBox").onclick({ target: { closest: sel => sel === "button" ? { id: "", disabled: false, dataset: ds } : null } }), clickInstall: () => $("modBox").onclick({ target: { closest: s => s === "button" ? { id: "modInstall", disabled: false, dataset: {} } : null } }), queue: () => modQueue };';
const ui = new Function('$', 't', 'lang', 'rpc', 'esc2', 'fetch', 'alert', 'confirm', 'view', body)($, k => k, 'en', rpc, esc2, fetchFn, () => {}, () => true, 'modules');

let failed = 0;
const ok = (name, v) => { console.log(v ? 'ok  ' : 'FAIL', name); if (!v) failed++; };
(async () => {
  await ui.modRefresh();
  const all = ui.modSizes(cat.capabilities.map(c => c.id), { files: new Set(), images: new Set(), works: new Set() });
  ok('everything installed is about 110 GB (68.2 models + 42.0 engines)', Math.abs(all.total - 110.16e9) < 0.1e9 && Math.abs(all.m - 68.16e9) < 0.01e9);
  ok('the page shows that total', /110\.2 GB/.test(els.modBox.innerHTML));
  ok('one card per capability, none ticked, none installed', (els.modBox.innerHTML.match(/data-modsel=/g) || []).length === 5 && !/checked/.test(els.modBox.innerHTML) && !/cInstalled/.test(els.modBox.innerHTML));
  ok('memory is marked as an estimate', /modEstimate/.test(els.modBox.innerHTML));
  // choose one: the engine image counts
  ui.setSel(['image.generate']); ui.renderModules();
  const one = ui.modSizes(['image.generate'], ui.modHave().have);
  ok('one capability alone carries the 38.5 GB image', Math.abs(one.e - 40.0e9) < 0.01e9 && Math.abs(one.m - 17.236e9) < 0.01e9);
  ok('the install button is enabled with enough space', !/id="modInstall" disabled/.test(els.modBox.innerHTML));
  // not enough space
  machine = { ...machine, free_bytes: 40e9 }; await ui.modRefresh();
  ok('too little free space disables install and says so', /id="modInstall" disabled/.test(els.modBox.innerHTML) && /modNoSpace/.test(els.modBox.innerHTML));
  machine = { ...machine, free_bytes: 500e9 };
  // whisper installed, image present: ComfyUI capability no longer pays for the image
  machine = { ...machine, capabilities: machine.capabilities.map(c => c.id === 'audio.transcribe' ? { ...c, installed: true } : c), engines: machine.engines.map(e => ({ ...e, image_present: true })) };
  await ui.modRefresh();
  const two = ui.modSizes(['image.generate'], ui.modHave().have);
  ok('with the image already there only the models and the engine folder are added', Math.abs(two.e - 1.5e9) < 0.01e9);
  ok('an installed capability shows a remove button and cannot be ticked', /data-modrm="audio.transcribe"/.test(els.modBox.innerHTML) && /data-modsel="audio.transcribe" disabled/.test(els.modBox.innerHTML));
  ok('the engine of an installed capability can be started and stopped', /data-modeng="start" data-eng="whisper"/.test(els.modBox.innerHTML) && !/data-eng="comfyui"/.test(els.modBox.innerHTML));
  // a running install shows progress
  machine = { ...machine, capabilities: machine.capabilities.map(c => c.id === 'image.edit' ? { ...c, task: { op: 'install', status: 'downloading x', total: 100e9, done: 25e9 } } : c) };
  await ui.modRefresh();
  ok('a running install shows a progress bar with bytes', /<progress max="100000000000" value="25000000000">/.test(els.modBox.innerHTML) && /25\.0 GB \/ 100\.0 GB/.test(els.modBox.innerHTML));
  ok('install is disabled while a task runs', /id="modInstall" disabled/.test(els.modBox.innerHTML));
  // a failed task shows its error
  machine = { ...machine, capabilities: machine.capabilities.map(c => c.id === 'image.edit' ? { ...c, task: { op: 'install', status: 'failed', err: 'no space', ended: 1 } } : c) };
  await ui.modRefresh();
  ok('a failed install shows why', /warn">no space/.test(els.modBox.innerHTML));
  // files added by hand: listed with their size, deletable after a confirmation
  machine = { ...machine, extra: [{ engine: 'comfyui', path: 'checkpoints/ltx-2.3-22b-dev-fp8.safetensors', size: 29.15e9 }, { engine: 'comfyui', path: 'text_encoders/gemma.safetensors', size: 9.45e9 }] };
  await ui.modRefresh();
  ok('added files are listed with their sizes and a total', els.modBox.innerHTML.includes('modExtra') && els.modBox.innerHTML.includes('checkpoints/ltx-2.3-22b-dev-fp8.safetensors') && (els.modBox.innerHTML.includes('29.2 GB') || els.modBox.innerHTML.includes('29.1 GB')) && els.modBox.innerHTML.includes('38.6 GB'));
  ok('each has a delete button naming the engine and the path', els.modBox.innerHTML.includes('data-modfile="checkpoints/ltx-2.3-22b-dev-fp8.safetensors" data-eng="comfyui"'));
  const before = calls.length;
  await ui.clickBtn({ modfile: 'checkpoints/ltx-2.3-22b-dev-fp8.safetensors', eng: 'comfyui', size: '29150000000' });
  ok('confirming sends the delete to the machine', calls.slice(before).some(c => c === 'modules.file_delete:' + JSON.stringify({ engine: 'comfyui', path: 'checkpoints/ltx-2.3-22b-dev-fp8.safetensors' })));
  machine = { ...machine, extra: [] };
  await ui.modRefresh();
  ok('with no added files the section is not shown', !/modExtra/.test(els.modBox.innerHTML));
  // an engine started by hand shows that it stays until stopped
  machine = { ...machine, engines: machine.engines.map(e => e.engine === 'whisper' ? { ...e, state: 'running', pinned: true } : e) };
  await ui.modRefresh();
  ok('a pinned running engine is marked', els.modBox.innerHTML.includes('modPinned'));
  machine = { ...machine, engines: machine.engines.map(e => e.engine === 'whisper' ? { ...e, pinned: false } : e) };
  await ui.modRefresh();
  ok('an engine that is not pinned is not marked', !els.modBox.innerHTML.includes('modPinned'));
  if (failed) process.exit(1);
})();