mirror of
https://github.com/MercuryWorkshop/scramjet.git
synced 2025-05-15 23:30:00 -04:00
set on the host arg
This commit is contained in:
Toshit Chawda
parent
c4e90a50ab
commit
51abf6577f
1 changed files with 140 additions and 130 deletions
|
|
@ -2,150 +2,161 @@ import { BareResponseFetch } from "@mercuryworkshop/bare-mux";
|
||||||
import IDBMap from "@webreflection/idb-map";
|
import IDBMap from "@webreflection/idb-map";
|
||||||
|
|
||||||
declare global {
|
declare global {
|
||||||
interface Window {
|
interface Window {
|
||||||
ScramjetServiceWorker;
|
ScramjetServiceWorker;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
self.ScramjetServiceWorker = class ScramjetServiceWorker {
|
self.ScramjetServiceWorker = class ScramjetServiceWorker {
|
||||||
client: typeof self.$scramjet.shared.util.BareClient.prototype;
|
client: typeof self.$scramjet.shared.util.BareClient.prototype;
|
||||||
config: typeof self.$scramjet.config;
|
config: typeof self.$scramjet.config;
|
||||||
|
|
||||||
constructor(config = self.$scramjet.config) {
|
constructor(config = self.$scramjet.config) {
|
||||||
this.client = new self.$scramjet.shared.util.BareClient();
|
this.client = new self.$scramjet.shared.util.BareClient();
|
||||||
if (!config.prefix) config.prefix = "/scramjet/";
|
if (!config.prefix) config.prefix = "/scramjet/";
|
||||||
this.config = config;
|
this.config = config;
|
||||||
}
|
}
|
||||||
|
|
||||||
route({ request }: FetchEvent) {
|
route({ request }: FetchEvent) {
|
||||||
if (request.url.startsWith(location.origin + this.config.prefix)) return true;
|
if (request.url.startsWith(location.origin + this.config.prefix)) return true;
|
||||||
else return false;
|
else return false;
|
||||||
}
|
}
|
||||||
|
|
||||||
async fetch({ request }: FetchEvent) {
|
async fetch({ request }: FetchEvent) {
|
||||||
const urlParam = new URLSearchParams(new URL(request.url).search);
|
const urlParam = new URLSearchParams(new URL(request.url).search);
|
||||||
const { encodeUrl, decodeUrl } = self.$scramjet.shared.url;
|
const { encodeUrl, decodeUrl, rewriteHeaders, rewriteHtml, rewriteJs, rewriteCss, rewriteWorkers } = self.$scramjet.shared;
|
||||||
const { rewriteHeaders, rewriteHtml, rewriteJs, rewriteCss, rewriteWorkers } = self.$scramjet.shared.rewrite;
|
|
||||||
|
|
||||||
if (urlParam.has("url")) {
|
if (urlParam.has("url")) {
|
||||||
return Response.redirect(encodeUrl(urlParam.get("url"), new URL(urlParam.get("url"))))
|
return Response.redirect(encodeUrl(urlParam.get("url"), new URL(urlParam.get("url"))))
|
||||||
}
|
}
|
||||||
|
|
||||||
try {
|
try {
|
||||||
const url = new URL(decodeUrl(request.url));
|
const url = new URL(decodeUrl(request.url));
|
||||||
|
|
||||||
const cookieStore = new IDBMap(url.origin, { durability: "relaxed", prefix: "Cookies" });
|
const cookieStore = new IDBMap(url.host, { durability: "relaxed", prefix: "Cookies" });
|
||||||
|
|
||||||
const response: BareResponseFetch = await this.client.fetch(url, {
|
const response: BareResponseFetch = await this.client.fetch(url, {
|
||||||
method: request.method,
|
method: request.method,
|
||||||
body: request.body,
|
body: request.body,
|
||||||
headers: request.headers,
|
headers: request.headers,
|
||||||
credentials: "omit",
|
credentials: "omit",
|
||||||
mode: request.mode === "cors" ? request.mode : "same-origin",
|
mode: request.mode === "cors" ? request.mode : "same-origin",
|
||||||
cache: request.cache,
|
cache: request.cache,
|
||||||
redirect: request.redirect,
|
redirect: request.redirect,
|
||||||
//@ts-ignore why the fuck is this not typed mircosoft
|
//@ts-ignore why the fuck is this not typed mircosoft
|
||||||
duplex: "half",
|
duplex: "half",
|
||||||
});
|
});
|
||||||
|
|
||||||
let responseBody;
|
let responseBody;
|
||||||
const responseHeaders = rewriteHeaders(response.rawHeaders, url);
|
const responseHeaders = rewriteHeaders(response.rawHeaders, url);
|
||||||
|
|
||||||
for (const cookie of (responseHeaders["set-cookie"] || []) as string[]) {
|
for (const cookie of (responseHeaders["set-cookie"] || []) as string[]) {
|
||||||
let cookieParsed = cookie.split(";").map(x=>x.trim().split("="));
|
let cookieParsed = cookie.split(";").map(x => x.trim().split("="));
|
||||||
|
|
||||||
let [key, value] = cookieParsed.shift();
|
let [key, value] = cookieParsed.shift();
|
||||||
value = value.replace("\"", "");
|
value = value.replace("\"", "");
|
||||||
cookieStore.set(key, { value: value, args: cookieParsed });
|
|
||||||
|
const hostArg = cookieParsed.find(x => x[0].toLowerCase() === "domain");
|
||||||
|
cookieParsed = cookieParsed.filter(x => x[0].toLowerCase() !== "domain");
|
||||||
|
let host = hostArg ? hostArg[1] : undefined;
|
||||||
|
|
||||||
|
if (host && host !== url.host) {
|
||||||
|
if (host.startsWith(".")) host = host.slice(1);
|
||||||
|
const cookieStore = new IDBMap(host, { durability: "relaxed", prefix: "Cookies" });
|
||||||
|
cookieStore.set(key, { value: value, args: cookieParsed });
|
||||||
|
} else {
|
||||||
|
cookieStore.set(key, { value: value, args: cookieParsed });
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
if (response.body) {
|
if (response.body) {
|
||||||
switch (request.destination) {
|
switch (request.destination) {
|
||||||
case "iframe":
|
case "iframe":
|
||||||
case "document":
|
case "document":
|
||||||
if (responseHeaders["content-type"].toString().startsWith("text/html")) {
|
if (responseHeaders["content-type"].toString().startsWith("text/html")) {
|
||||||
responseBody = rewriteHtml(await response.text(), url);
|
responseBody = rewriteHtml(await response.text(), url);
|
||||||
} else {
|
} else {
|
||||||
responseBody = response.body;
|
responseBody = response.body;
|
||||||
}
|
}
|
||||||
break;
|
break;
|
||||||
case "script":
|
case "script":
|
||||||
responseBody = rewriteJs(await response.text(), url);
|
responseBody = rewriteJs(await response.text(), url);
|
||||||
break;
|
break;
|
||||||
case "style":
|
case "style":
|
||||||
responseBody = rewriteCss(await response.text(), url);
|
responseBody = rewriteCss(await response.text(), url);
|
||||||
break;
|
break;
|
||||||
case "sharedworker":
|
case "sharedworker":
|
||||||
case "worker":
|
case "worker":
|
||||||
responseBody = rewriteWorkers(await response.text(), url);
|
responseBody = rewriteWorkers(await response.text(), url);
|
||||||
break;
|
break;
|
||||||
default:
|
default:
|
||||||
responseBody = response.body;
|
responseBody = response.body;
|
||||||
break;
|
break;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
// downloads
|
// downloads
|
||||||
if (["document", "iframe"].includes(request.destination)) {
|
if (["document", "iframe"].includes(request.destination)) {
|
||||||
const header = responseHeaders["content-disposition"];
|
const header = responseHeaders["content-disposition"];
|
||||||
|
|
||||||
// validate header and test for filename
|
// validate header and test for filename
|
||||||
if (!/\s*?((inline|attachment);\s*?)filename=/i.test(header)) {
|
if (!/\s*?((inline|attachment);\s*?)filename=/i.test(header)) {
|
||||||
// if filename= wasn"t specified then maybe the remote specified to download this as an attachment?
|
// if filename= wasn"t specified then maybe the remote specified to download this as an attachment?
|
||||||
// if it"s invalid then we can still possibly test for the attachment/inline type
|
// if it"s invalid then we can still possibly test for the attachment/inline type
|
||||||
const type = /^\s*?attachment/i.test(header)
|
const type = /^\s*?attachment/i.test(header)
|
||||||
? "attachment"
|
? "attachment"
|
||||||
: "inline";
|
: "inline";
|
||||||
|
|
||||||
// set the filename
|
// set the filename
|
||||||
const [filename] = new URL(response.finalURL).pathname
|
const [filename] = new URL(response.finalURL).pathname
|
||||||
.split("/")
|
.split("/")
|
||||||
.slice(-1);
|
.slice(-1);
|
||||||
|
|
||||||
responseHeaders[
|
responseHeaders[
|
||||||
"content-disposition"
|
"content-disposition"
|
||||||
] = `${type}; filename=${JSON.stringify(filename)}`;
|
] = `${type}; filename=${JSON.stringify(filename)}`;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
if (responseHeaders["accept"] === "text/event-stream") {
|
if (responseHeaders["accept"] === "text/event-stream") {
|
||||||
responseHeaders["content-type"] = "text/event-stream";
|
responseHeaders["content-type"] = "text/event-stream";
|
||||||
}
|
}
|
||||||
if (crossOriginIsolated) {
|
if (crossOriginIsolated) {
|
||||||
responseHeaders["Cross-Origin-Embedder-Policy"] = "require-corp";
|
responseHeaders["Cross-Origin-Embedder-Policy"] = "require-corp";
|
||||||
}
|
}
|
||||||
|
|
||||||
return new Response(responseBody, {
|
return new Response(responseBody, {
|
||||||
headers: responseHeaders as HeadersInit,
|
headers: responseHeaders as HeadersInit,
|
||||||
status: response.status,
|
status: response.status,
|
||||||
statusText: response.statusText
|
statusText: response.statusText
|
||||||
})
|
})
|
||||||
} catch (err) {
|
} catch (err) {
|
||||||
if (!["document", "iframe"].includes(request.destination))
|
if (!["document", "iframe"].includes(request.destination))
|
||||||
return new Response(undefined, { status: 500 });
|
return new Response(undefined, { status: 500 });
|
||||||
|
|
||||||
console.error(err);
|
console.error(err);
|
||||||
|
|
||||||
return renderError(err, decodeUrl(request.url));
|
return renderError(err, decodeUrl(request.url));
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
function errorTemplate(
|
function errorTemplate(
|
||||||
trace: string,
|
trace: string,
|
||||||
fetchedURL: string,
|
fetchedURL: string,
|
||||||
) {
|
) {
|
||||||
// turn script into a data URI so we don"t have to escape any HTML values
|
// turn script into a data URI so we don"t have to escape any HTML values
|
||||||
const script = `
|
const script = `
|
||||||
errorTrace.value = ${JSON.stringify(trace)};
|
errorTrace.value = ${JSON.stringify(trace)};
|
||||||
fetchedURL.textContent = ${JSON.stringify(fetchedURL)};
|
fetchedURL.textContent = ${JSON.stringify(fetchedURL)};
|
||||||
for (const node of document.querySelectorAll("#hostname")) node.textContent = ${JSON.stringify(
|
for (const node of document.querySelectorAll("#hostname")) node.textContent = ${JSON.stringify(
|
||||||
location.hostname
|
location.hostname
|
||||||
)};
|
)};
|
||||||
reload.addEventListener("click", () => location.reload());
|
reload.addEventListener("click", () => location.reload());
|
||||||
version.textContent = "0.0.1";
|
version.textContent = "0.0.1";
|
||||||
`
|
`
|
||||||
|
|
||||||
return (
|
return (
|
||||||
`<!DOCTYPE html>
|
`<!DOCTYPE html>
|
||||||
<html>
|
<html>
|
||||||
<head>
|
<head>
|
||||||
<meta charset="utf-8" />
|
<meta charset="utf-8" />
|
||||||
|
|
@ -177,13 +188,12 @@ function errorTemplate(
|
||||||
<button id="reload">Reload</button>
|
<button id="reload">Reload</button>
|
||||||
<hr />
|
<hr />
|
||||||
<p><i>Scramjet v<span id="version"></span></i></p>
|
<p><i>Scramjet v<span id="version"></span></i></p>
|
||||||
<script src="${
|
<script src="${"data:application/javascript," + encodeURIComponent(script)
|
||||||
"data:application/javascript," + encodeURIComponent(script)
|
}"></script>
|
||||||
}"></script>
|
|
||||||
</body>
|
</body>
|
||||||
</html>
|
</html>
|
||||||
`
|
`
|
||||||
);
|
);
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
|
|
@ -192,22 +202,22 @@ function errorTemplate(
|
||||||
* @param {string} fetchedURL
|
* @param {string} fetchedURL
|
||||||
*/
|
*/
|
||||||
function renderError(err, fetchedURL) {
|
function renderError(err, fetchedURL) {
|
||||||
const headers = {
|
const headers = {
|
||||||
"content-type": "text/html",
|
"content-type": "text/html",
|
||||||
};
|
};
|
||||||
if (crossOriginIsolated) {
|
if (crossOriginIsolated) {
|
||||||
headers["Cross-Origin-Embedd'er-Policy"] = "require-corp";
|
headers["Cross-Origin-Embedd'er-Policy"] = "require-corp";
|
||||||
}
|
}
|
||||||
|
|
||||||
return new Response(
|
return new Response(
|
||||||
errorTemplate(
|
errorTemplate(
|
||||||
String(err),
|
String(err),
|
||||||
fetchedURL
|
fetchedURL
|
||||||
),
|
),
|
||||||
{
|
{
|
||||||
status: 500,
|
status: 500,
|
||||||
headers: headers
|
headers: headers
|
||||||
}
|
}
|
||||||
);
|
);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue