// Checks the script that sends ComfyUI's model downloads to the machine (cmd/wxch-server/comfy_fetch.go):
//   node tests/modules/comfy_fetch_test.js
const fs = require('fs');
const go = fs.readFileSync(__dirname + '/../../cmd/wxch-server/comfy_fetch.go', 'utf8');
const m = go.match(/const comfyFetchScript = `<script>([\s\S]*?)<\/script>`/);
let failed = 0;
const ok = (n, v) => { console.log(v ? 'ok  ' : 'FAIL', n); if (!v) failed++; };
ok('the script is in the Go file and has no backtick inside', !!m && !m[1].includes('`'));
try { new Function(m[1]); ok('the script parses', true); } catch (e) { ok('the script parses: ' + e.message, false); }

// a document that accepts anything
const node = () => new Proxy({ style: {}, children: [], appendChild() {}, addEventListener() {}, closest() { return null; } }, { set(t, k, v) { t[k] = v; return true; } });
const calls = [], clicks = [];
const anchorProto = { click() { clicks.push(this.href); } };
const env = {
  navigator: { language: 'th-TH' },
  document: { readyState: 'complete', body: node(), createElement: node, addEventListener() {} },
  location: { href: 'https://comfy.example.org/' },
  HTMLAnchorElement: function () {}, window: {},
  Option: function (a, b) { this.text = a; this.value = b; },
  fetch: (u, o) => { calls.push({ u, o }); return Promise.resolve({ json: () => Promise.resolve({ jobs: [] }) }); },
  clearTimeout() {}, setTimeout() { return 1; },
};
env.HTMLAnchorElement.prototype = anchorProto;
env.window.open = () => 'opened';
new Function(...Object.keys(env), 'URL', m[1])(...Object.values(env), URL);
const hf = 'https://huggingface.co/Comfy-Org/z_image_turbo/resolve/main/split_files/vae/ae.safetensors';
calls.length = 0;
anchorProto.click.call({ href: hf, getAttribute: () => 'ae.safetensors' });
const post = calls.find(c => c.o && c.o.method === 'POST');
const body = post && JSON.parse(post.o.body);
ok('a Hugging Face model download is sent to the machine, with the folder taken from the address', !!post && post.u === '/_wxch/fetch' && body.dir === 'vae' && body.name === 'ae.safetensors' && body.url === hf);
ok('it did not start a download in the browser', clicks.length === 0);
anchorProto.click.call({ href: 'https://example.org/file.zip', getAttribute: () => '' });
ok('any other link behaves as before', clicks.length === 1 && clicks[0] === 'https://example.org/file.zip');
anchorProto.click.call({ href: 'https://evilhuggingface.co/a/resolve/main/vae/x.safetensors', getAttribute: () => '' });
ok('a look-alike host is not taken for Hugging Face', clicks.length === 2);
calls.length = 0;
anchorProto.click.call({ href: 'https://huggingface.co/a/b/resolve/main/model.safetensors', getAttribute: () => '' });
ok('without a folder in the address nothing is sent: the box asks for the folder', !calls.some(c => c.o && c.o.method === 'POST'));
ok('window.open of a model address is caught as well', env.window.open('https://huggingface.co/a/b/resolve/main/diffusion_models/x.safetensors') === null);
ok('window.open of anything else still opens', env.window.open('https://example.org/') === 'opened');
process.exit(failed ? 1 : 0);