import {readJobStream} from '@/lib/stream';
import type {Download} from '@/lib/types';
import {encodeWav} from './audio';

/** Raised when the session has lapsed, so the studio can show sign-in without losing the project. */
export class SignedOut extends Error{}

type Call={signal:AbortSignal;status:(text:string)=>void};

/** Call a generation route and follow its progress stream to the result. */
export async function callEngine<T>(url:string,body:FormData|object,{signal,status}:Call):Promise<T>{
 const init:RequestInit=body instanceof FormData
  ?{method:'POST',body,signal}
  :{method:'POST',body:JSON.stringify(body),headers:{'Content-Type':'application/json'},signal};
 const response=await fetch(url,init);
 if(!response.ok){
  const data=await response.json().catch(()=>({}));
  if(response.status===401)throw new SignedOut(data.error||'Sign in again to continue.');
  throw new Error(data.error||'That could not start.');
 }
 if(!response.body)throw new Error('The connection closed. Please try again.');
 const result=await readJobStream(response.body,{status});
 if(!result)throw new Error('The engine returned nothing. Please try again.');
 return result as T;
}

/**
 * Fetch an engine result through the app, which relays it with the user's
 * token. Hugging Face's gateway drops some requests with a 5xx (sometimes an
 * HTML error page), so those are retried; the finished job is not lost.
 */
export async function fetchResult(file:Download,signal:AbortSignal){
 let last='';
 for(let attempt=0;attempt<5;attempt++){
  if(attempt)await new Promise(done=>setTimeout(done,2000*attempt));
  signal.throwIfAborted();
  const response=await fetch('/api/audio?url='+encodeURIComponent(file.url),{signal}).catch(error=>{
   if(signal.aborted)throw error;
   return null;
  });
  if(response?.ok)return response.arrayBuffer();
  const text=response?await response.text().catch(()=>''):'';
  // The relay's own messages are plain sentences; anything else is a gateway page.
  last=text&&!/<html|<!doctype/i.test(text)&&text.length<300?text:'';
  if(response&&response.status<500&&response.status!==429)break;
 }
 throw new Error(last||'Hugging Face’s servers had a hiccup while sending the result. Run it again; your project is safe.');
}

export function wavFile(buffer:AudioBuffer,name='selection.wav'){
 return new File([encodeWav(buffer)],name,{type:'audio/wav'});
}