package main

// comfyFetchScript is added to the pages of ComfyUI that we serve. ComfyUI's own "Download" buttons make the browser fetch
// a model file, which lands on the computer of the person who clicked; this script catches those downloads (a Hugging Face
// model address) and sends them to the machine instead, which downloads the file into its models folder (the agent's
// /_wxch/fetch, reached through the tunnel). A small button next to the logo opens a box for pasting an address by hand,
// which also works when a ComfyUI version does its download in a way the script does not recognise.
//
// The text is plain ES5 so that it runs in any browser; there are no backticks in it (it is a Go raw string).
const comfyFetchScript = `<script>(function(){
if(window.__wxchFetch)return;window.__wxchFetch=1;
var th=/^th/i.test(navigator.language||'');
function T(a,b){return th?a:b}
var HF=/^https:\/\/(?:[a-z0-9-]+\.)*(?:huggingface\.co|hf\.co)\//i,EXT=/\.(safetensors|ckpt|pt|pth|bin|gguf|sft|onnx)$/i;
var DIRS=['checkpoints','diffusion_models','text_encoders','vae','loras','clip','clip_vision','unet','controlnet','upscale_models','latent_upscale_models','model_patches','audio_encoders','embeddings','vae_approx','style_models','gligen','hypernetworks','photomaker','diffusers'];
function dirOf(u){try{var p=new URL(u).pathname.split('/');for(var i=p.length-2;i>=0;i--){if(DIRS.indexOf(p[i])>=0)return p[i]}}catch(e){}return ''}
function nameOf(u){try{var p=new URL(u).pathname.split('/');return decodeURIComponent(p[p.length-1])}catch(e){return ''}}
function isModel(u){try{var x=new URL(u,location.href);return HF.test(x.href)&&EXT.test(x.pathname)}catch(e){return false}}
var box,list,urlIn,dirSel,msg,timer;
function el(t,s,txt){var e=document.createElement(t);if(s)e.style.cssText=s;if(txt!=null)e.textContent=txt;return e}
function build(){
 if(box)return;
 var bg='background:#1c1f24;color:#e8e8e8;border:1px solid #555;border-radius:8px;';
 box=el('div',bg+'position:fixed;left:8px;bottom:42px;z-index:2147483000;width:min(380px,92vw);padding:10px;font:13px system-ui,sans-serif;display:none;box-shadow:0 4px 18px #0008');
 box.appendChild(el('div','font-weight:600;margin-bottom:6px',T('ดาวน์โหลดโมเดลลงเครื่อง (ไม่ผ่านคอมพิวเตอร์ของคุณ)','Download a model onto the machine (not your computer)')));
 urlIn=el('input','width:100%;box-sizing:border-box;margin:3px 0;padding:6px;'+bg);urlIn.placeholder='https://huggingface.co/.../resolve/main/....safetensors';
 dirSel=el('select','width:100%;box-sizing:border-box;margin:3px 0;padding:6px;'+bg);
 dirSel.appendChild(new Option(T('— เลือกโฟลเดอร์ปลายทาง —','— pick the folder —'),''));
 DIRS.forEach(function(d){dirSel.appendChild(new Option(d,d))});
 var go=el('button','padding:6px 12px;margin:3px 0;cursor:pointer;'+bg,T('ดาวน์โหลด','Download'));
 go.onclick=function(){send(urlIn.value.trim(),dirSel.value,'')};
 msg=el('div','margin:4px 0;min-height:1.2em;color:#f0b');
 list=el('div','max-height:30vh;overflow:auto');
 [urlIn,dirSel,go,msg,list].forEach(function(n){box.appendChild(n)});
 document.body.appendChild(box);
 var b=el('button','position:fixed;left:40px;bottom:8px;z-index:2147483000;padding:3px 8px;font:12px system-ui;cursor:pointer;opacity:.85;'+bg,'⬇ '+T('โมเดล','models'));
 b.title=T('ดาวน์โหลดโมเดลลงเครื่อง','Download a model onto the machine');
 b.onclick=function(){toggle()};document.body.appendChild(b);
}
function toggle(open){build();box.style.display=(open===undefined?box.style.display==='none':open)?'block':'none';if(box.style.display==='block')poll()}
function gb(n){return (n/1e9).toFixed(n>1e10?0:1)+' GB'}
function render(jobs){
 if(!list)return;list.textContent='';
 jobs.forEach(function(j){
  var pct=j.total?Math.floor(100*j.done/j.total):0,t=j.dir+'/'+j.name+' — ';
  if(j.state==='done')t+=T('เสร็จแล้ว (กด r ใน ComfyUI เพื่อโหลดรายการใหม่)','done (press r in ComfyUI to refresh the lists)');
  else if(j.state==='failed')t+=T('ไม่สำเร็จ: ','failed: ')+j.err;
  else t+=pct+'% ('+gb(j.done)+' / '+gb(j.total)+')';
  list.appendChild(el('div','margin:3px 0;word-break:break-all;'+(j.state==='failed'?'color:#f77':j.state==='done'?'color:#8fe8b8':''),t));
 });
}
function poll(){
 clearTimeout(timer);
 fetch('/_wxch/fetch',{credentials:'same-origin'}).then(function(r){return r.json()}).then(function(d){
  render(d.jobs||[]);
  var busy=(d.jobs||[]).some(function(j){return j.state==='starting'||j.state==='downloading'});
  if(busy)timer=setTimeout(poll,2000);
 }).catch(function(){});
}
function send(u,dir,name){
 build();toggle(true);urlIn.value=u;if(dir)dirSel.value=dir;msg.textContent='';
 if(!dir){msg.textContent=T('เลือกโฟลเดอร์ปลายทางแล้วกดดาวน์โหลด','Pick the folder, then press Download');return}
 fetch('/_wxch/fetch',{method:'POST',credentials:'same-origin',headers:{'Content-Type':'application/json'},body:JSON.stringify({url:u,dir:dir,name:name})})
 .then(function(r){return r.json()}).then(function(d){if(d.error){msg.textContent=d.error}else{msg.textContent='';poll()}})
 .catch(function(e){msg.textContent=String(e)});
}
function grab(u,name){
 if(!isModel(u))return false;
 var abs=new URL(u,location.href).href;
 send(abs,dirOf(abs),name||nameOf(abs));return true;
}
var oc=HTMLAnchorElement.prototype.click;
HTMLAnchorElement.prototype.click=function(){if(this.href&&grab(this.href,this.getAttribute('download')||''))return;return oc.apply(this,arguments)};
document.addEventListener('click',function(e){var a=e.target&&e.target.closest?e.target.closest('a[href]'):null;if(a&&grab(a.href,a.getAttribute('download')||'')){e.preventDefault();e.stopPropagation()}},true);
var wo=window.open;window.open=function(u){if(u&&grab(String(u),''))return null;return wo.apply(this,arguments)};
if(document.readyState==='loading')document.addEventListener('DOMContentLoaded',build);else build();
})();</script>`