From c6b3a243b964a180f2ae08c0af4d0cff6c1283eb Mon Sep 17 00:00:00 2001 From: multipleof4 Date: Sat, 23 Aug 2025 08:39:08 -0700 Subject: [PATCH] Update sw.js --- public/sw.js | 122 ++++++++++++++++++++++++++++++++++++++++++++++----- 1 file changed, 111 insertions(+), 11 deletions(-) diff --git a/public/sw.js b/public/sw.js index 9478281..8dea5bd 100644 --- a/public/sw.js +++ b/public/sw.js @@ -1,11 +1,111 @@ -const CH='chat-stream-v1' -let bc,streams=new Map() -self.addEventListener('install',e=>self.skipWaiting()) -self.addEventListener('activate',e=>e.waitUntil(self.clients.claim())) -function send(m){try{bc||(bc=new BroadcastChannel(CH));bc.postMessage(m)}catch{}} -function hint(x){x=String(x||'');if(/401|unauthorized/i.test(x))return'Unauthorized (check API key).';if(/429|rate/i.test(x))return'Rate limited (slow down or upgrade).';if(/403|forbidden|access/i.test(x))return'Forbidden (model or key scope).';return'Request failed.'} -async function start(id,p){bc||(bc=new BroadcastChannel(CH));const body=typeof p.body==='string'?p.body:JSON.stringify(p.body||{}),ctrl=new AbortController();let done=false,buf='';streams.set(id,{ctrl,buf:()=>buf,done:()=>done});try{const r=await fetch(p.url,{method:'POST',headers:p.headers||{},body,signal:ctrl.signal});if(!r.ok)throw new Error((await r.text().catch(()=>''))||('HTTP '+r.status));const rd=r.body.getReader(),dc=new TextDecoder('utf-8');let acc='';const doneOnce=()=>{if(done)return;done=true;send({t:'done',id})};for(;;){const {value,done:d}=await rd.read();if(d)break;acc+=dc.decode(value,{stream:true});let i;while((i=acc.indexOf('\n\n'))!==-1){const raw=acc.slice(0,i).trim();acc=acc.slice(i+2);if(!raw||!raw.startsWith('data:'))continue;const data=raw.slice(5).trim();if(data==='[DONE]'){doneOnce();continue}try{const j=JSON.parse(data);const delta=j.choices?.[0]?.delta?.content??'';if(delta){buf+=delta;send({t:'delta',id,data:delta})}const fin=j.choices?.[0]?.finish_reason;if(fin)doneOnce()}catch{}}}doneOnce()}catch(e){const aborted=e?.name==='AbortError'||/abort/i.test(String(e?.message||''));if(aborted)send({t:'done',id,aborted:true});else send({t:'error',id,hint:hint(e?.message)})}finally{streams.delete(id)}} -function stop(id){const s=streams.get(id);if(!s)return;try{s.ctrl.abort()}catch{}finally{streams.delete(id)}} -function replay(id,offset=0){const s=streams.get(id);if(!s)return;const cur=s.buf(),remain=cur.slice(Math.max(0,offset));if(remain)send({t:'delta',id,data:remain});if(s.done())send({t:'done',id})} -bc=new BroadcastChannel(CH) -bc.onmessage=e=>{const m=e.data||{};if(m.t==='hello')send({t:'hello-ack'});else if(m.t==='start'&&m.id&&m.payload)start(m.id,m.payload);else if(m.t==='stop'&&m.id)stop(m.id);else if(m.t==='replay'&&m.id)replay(m.id,m.offset|0)} +self.importScripts('https://cdn.jsdelivr.net/npm/localforage@1.10.0/dist/localforage.min.js') + +localforage.config({name:'localforage',storeName:'keyvaluepairs'}) + +const TKEY='threads_v1' + +const now=()=>Date.now() +const gid=()=>Math.random().toString(36).slice(2,9) +const titleFrom=t=>(String(t||'').replace(/\s+/g,' ').trim().slice(0,60)||'Untitled') + +async function loadThreads(){const v=await localforage.getItem(TKEY);return Array.isArray(v)?v:[]} +async function saveThreads(v){await localforage.setItem(TKEY,v)} + +async function pickLatestThread(threads){if(!threads.length)return null;let best=threads[0],bu=+best.updatedAt||0;for(const t of threads){const u=+t.updatedAt||0;if(u>bu){best=t;bu=u}}return best} + +async function ensureThreadForWrite(reqMeta){ + let threads=await loadThreads() + let th=await pickLatestThread(threads) + if(!th){ + const id=gid() + const firstUser=(reqMeta.messages||[]).find(m=>m&&m.role==='user')?.content||'' + th={id,title:titleFrom(firstUser),pinned:false,updatedAt:now(),messages:[]} + threads.unshift(th) + await saveThreads(threads) + } + return th.id +} + +async function appendAssistantPlaceholder(threadId,meta){ + let threads=await loadThreads() + let th=threads.find(t=>t.id===threadId) + if(!th){th={id:threadId,title:'Untitled',pinned:false,updatedAt:now(),messages:[]};threads.unshift(th)} + const mid='sw_'+gid() + const msg={id:mid,role:'assistant',content:'',sune_name:meta?.sune_name||'',model:meta?.model||'',avatar:meta?.avatar||'',sw:true} + th.messages=[...(Array.isArray(th.messages)?th.messages:[]),msg] + th.updatedAt=now() + await saveThreads(threads) + return mid +} + +async function updateAssistantContent(threadId,mid,content){ + let threads=await loadThreads() + const th=threads.find(t=>t.id===threadId) + if(!th)return + const i=(th.messages||[]).findIndex(m=>m&&m.id===mid) + if(i<0)return + th.messages[i].content=content + th.updatedAt=now() + await saveThreads(threads) +} + +async function parseAndPersist(stream,threadId,mid){ + const reader=stream.getReader() + const dec=new TextDecoder('utf-8') + let buf='',full='',lastWrite=0 + const flush=async(force=false)=>{const t=Date.now();if(force||t-lastWrite>250){await updateAssistantContent(threadId,mid,full);lastWrite=t}} + while(true){ + const {value,done}=await reader.read() + if(done)break + buf+=dec.decode(value,{stream:true}) + let idx + while((idx=buf.indexOf('\n\n'))!==-1){ + const chunk=buf.slice(0,idx).trim() + buf=buf.slice(idx+2) + if(!chunk)continue + if(chunk.startsWith('data:')){ + const data=chunk.slice(5).trim() + if(data==='[DONE]'){await flush(true);return} + try{ + const json=JSON.parse(data) + const d=json&&json.choices&&json.choices[0]&&json.choices[0].delta&&json.choices[0].delta.content||'' + if(d){full+=d;await flush(false)} + }catch(_){} + } + } + } + await flush(true) +} + +async function handleOpenRouterEvent(event){ + const req=event.request + const url=new URL(req.url) + const isTarget=url.hostname==='openrouter.ai'&&/\/api\/v1\/chat\/completions$/.test(url.pathname) + if(!isTarget)return fetch(req) + + let reqMeta={} + try{const t=await req.clone().text();reqMeta=JSON.parse(t||'{}')}catch(_){} + if(!reqMeta||!reqMeta.stream){return fetch(req)} + + const res=await fetch(req) + if(!res.body)return res + + const [toClient,toTap]=res.body.tee() + + event.waitUntil((async()=>{ + try{ + const msgs=Array.isArray(reqMeta.messages)?reqMeta.messages:[] + const meta={sune_name:'',model:reqMeta.model||'',avatar:''} + const threadId=await ensureThreadForWrite({messages:msgs}) + const mid=await appendAssistantPlaceholder(threadId,meta) + await parseAndPersist(toTap,threadId,mid) + }catch(_){} + })()) + + const headers=new Headers(res.headers) + return new Response(toClient,{status:res.status,statusText:res.statusText,headers}) +} + +self.addEventListener('install',e=>{self.skipWaiting()}) +self.addEventListener('activate',e=>{e.waitUntil(self.clients.claim())}) +self.addEventListener('fetch',e=>{e.respondWith(handleOpenRouterEvent(e))})