mercuryworkshop-mirror/scramjet
1
0
Fork 0
mirror of https://github.com/MercuryWorkshop/scramjet.git synced 2025-05-14 15:00:01 -04:00
Code Issues Projects Releases Packages Wiki Activity Actions

clean up worker code

Browse source
This commit is contained in:
Percs 2024-05-07 19:26:45 -05:00
parent 8c58108727
commit c189e54509
3 changed files with 1550 additions and 1212 deletions
Download patch file Download diff file Expand all files Collapse all files

2665
pnpm-lock.yaml generated
View file

File diff suppressed because it is too large Load diff

95
src/worker/index.ts
View file

@ -17,50 +17,65 @@ self.ScramjetServiceWorker = class ScramjetServiceWorker {
this.client = new BareClient(); this.client = new BareClient();
} }
route(event: Event | any) {
const { request } = event;
if (request.url.startsWith(location.origin + self.__scramjet$config.prefix)) return true;
else return false;
}
async fetch(event: FetchEvent) { async fetch(event: FetchEvent) {
const url = new URL(self.__scramjet$bundle.rewriters.url.decodeUrl(event.request.url)); try {
const url = new URL(self.__scramjet$bundle.rewriters.url.decodeUrl(event.request.url));
// implement header rewriting later // implement header rewriting later
const response: BareResponseFetch = await this.client.fetch(url, { const response: BareResponseFetch = await this.client.fetch(url, {
method: event.request.method, method: event.request.method,
body: event.request.body, body: event.request.body,
headers: event.request.headers headers: event.request.headers
}); });
let responseBody; let responseBody;
const responseHeaders = self.__scramjet$bundle.rewriters.rewriteHeaders(response.rawHeaders, origin); const responseHeaders = self.__scramjet$bundle.rewriters.rewriteHeaders(response.rawHeaders, origin);
if (response.body) {
switch (event.request.destination) {
case "iframe":
case "document":
responseBody = self.__scramjet$bundle.rewriters.rewriteHtml(await response.text(), url.origin);
break;
case "script":
responseBody = self.__scramjet$bundle.rewriters.rewriteJs(await response.text(), url.origin);
break;
case "style":
responseBody = self.__scramjet$bundle.rewriters.rewriteCss(await response.text(), url.origin);
break;
case "sharedworker":
break;
case "worker":
break;
default:
responseBody = response.body;
break;
}
}
if (responseHeaders.accept === 'text/event-stream') {
responseHeaders.headers['content-type'] = 'text/event-stream';
}
if (crossOriginIsolated) {
responseHeaders["Cross-Origin-Embedder-Policy"] = "require-corp";
}
switch (event.request.destination) { return new Response(responseBody, {
case "document": headers: responseHeaders as HeadersInit,
responseBody = self.__scramjet$bundle.rewriters.rewriteHtml(await response.text(), url.origin); status: response.status,
break; statusText: response.statusText
case "iframe": })
responseBody = self.__scramjet$bundle.rewriters.rewriteHtml(await response.text(), url.origin); } catch (err) {
break; if (!['document', 'iframe'].includes(event.request.destination))
case "script": return new Response(undefined, { status: 500 });
responseBody = self.__scramjet$bundle.rewriters.rewriteJs(await response.text(), url.origin);
break; console.error(err);
case "style":
responseBody = self.__scramjet$bundle.rewriters.rewriteCss(await response.text(), url.origin);
break;
case "sharedworker":
break;
case "worker":
break;
default:
responseBody = response.body;
break;
} }
if (crossOriginIsolated) {
responseHeaders["Cross-Origin-Embedder-Policy"] = "require-corp";
}
return new Response(responseBody, {
headers: responseHeaders as HeadersInit,
status: response.status,
statusText: response.statusText
})
} }
} }

2
static/sw.js
View file

@ -7,7 +7,7 @@ const scramjet = new ScramjetServiceWorker();
self.addEventListener("fetch", async (event) => { self.addEventListener("fetch", async (event) => {
event.respondWith((async() => { event.respondWith((async() => {
if (event.request.url.startsWith(location.origin + __scramjet$config.prefix)) { if (scramjet.route(event)) {
return await scramjet.fetch(event); return await scramjet.fetch(event);
} else { } else {
return await fetch(event.request); return await fetch(event.request);