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

add parseDomain

Browse source
This commit is contained in:
Toshit Chawda 2024-07-14 17:00:35 -07:00
parent 6eb9b6f12c
commit 546d30ca4e
No known key found for this signature in database
GPG key ID: 91480ED99E2B3D9D
5 changed files with 214 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

41
src/worker/index.ts
View file

@ -1,5 +1,7 @@
import { BareResponseFetch } from "@mercuryworkshop/bare-mux";
import IDBMap from "@webreflection/idb-map";
import { ParseResultType } from "parse-domain";
import { parse } from "path";
declare global {
interface Window {
@ -33,6 +35,7 @@ self.ScramjetServiceWorker = class ScramjetServiceWorker {
rewriteCss,
rewriteWorkers,
} = self.$scramjet.shared.rewrite;
const { parseDomain } = self.$scramjet.shared.util;
if (urlParam.has("url")) {
return Response.redirect(
@ -69,23 +72,45 @@ self.ScramjetServiceWorker = class ScramjetServiceWorker {
let [key, value] = cookieParsed.shift();
value = value.replace('"', "");
const hostArg = cookieParsed.find(
(x) => x[0].toLowerCase() === "domain"
);
cookieParsed = cookieParsed.filter(
(x) => x[0].toLowerCase() !== "domain"
);
const hostArg = cookieParsed.find((x) => x[0] === "Domain");
cookieParsed = cookieParsed.filter((x) => x[0] !== "Domain");
let host = hostArg ? hostArg[1] : undefined;
if (url.protocol === "http" && cookieParsed.includes(["Secure"]))
continue;
if (
cookieParsed.includes(["SameSite", "None"]) &&
!cookieParsed.includes(["Secure"])
)
continue;
if (host && host !== url.host) {
if (host.startsWith(".")) host = host.slice(1);
const urlDomain = parseDomain(url.hostname);
if (urlDomain.type === ParseResultType.Listed) {
const { subDomains: _, domain, topLevelDomains } = urlDomain;
if (!host.endsWith([domain, ...topLevelDomains].join(".")))
continue;
} else {
continue;
}
const realCookieStore = new IDBMap(host, {
durability: "relaxed",
prefix: "Cookies",
});
realCookieStore.set(key, { value: value, args: cookieParsed });
realCookieStore.set(key, {
value: value,
args: cookieParsed,
subdomain: true,
});
} else {
cookieStore.set(key, { value: value, args: cookieParsed });
cookieStore.set(key, {
value: value,
args: cookieParsed,
subdomain: false,
});
}
}