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

feat: migrate serviceworker registrations to a more robust system

Browse source
This commit is contained in:
Percs 2024-12-12 23:27:46 -06:00
parent 88d76852d6
commit b658a79812
1 changed files with 20 additions and 36 deletions
Download patch file Download diff file Expand all files Collapse all files

56
src/client/dom/serviceworker.ts
View file

@ -1,7 +1,6 @@
import { rewriteUrl } from "../../shared";
import { ScramjetClient } from "../client";
import { type MessageC2W } from "../../worker";
import { getOwnPropertyDescriptorHandler } from "../helpers";
import { flagEnabled } from "../../scramjet";
// we need a late order because we're mangling with addEventListener at a higher level
@ -14,9 +13,16 @@ export function disabled(_client: ScramjetClient, _self: Self) {
Reflect.deleteProperty(Navigator.prototype, "serviceWorker");
}
export default function (client: ScramjetClient, _self: Self) {
let registration;
type FakeRegistrationState = {
scope: string;
active: ServiceWorker;
};
export default function (client: ScramjetClient, _self: Self) {
const registrationmap: WeakMap<
ServiceWorkerRegistration,
FakeRegistrationState
> = new WeakMap();
client.Proxy("EventTarget.prototype.addEventListener", {
apply(ctx) {
if (registration === ctx.this) {
@ -61,6 +67,12 @@ export default function (client: ScramjetClient, _self: Self) {
client.Proxy("ServiceWorkerContainer.prototype.register", {
apply(ctx) {
const fakeRegistration = new EventTarget() as ServiceWorkerRegistration;
Object.setPrototypeOf(
fakeRegistration,
self.ServiceWorkerRegistration.prototype
);
fakeRegistration.constructor = ctx.fn;
let url = rewriteUrl(ctx.args[0], client.meta) + "?dest=serviceworker";
if (ctx.args[1] && ctx.args[1].type === "module") {
url += "&type=module";
@ -68,6 +80,10 @@ export default function (client: ScramjetClient, _self: Self) {
const worker = client.natives.construct("SharedWorker", url);
const handle = worker.port;
const state: FakeRegistrationState = {
scope: ctx.args[0],
active: handle as ServiceWorker,
};
const controller = client.descriptors.get(
"ServiceWorkerContainer.prototype.controller",
client.serviceWorker
@ -84,39 +100,7 @@ export default function (client: ScramjetClient, _self: Self) {
[handle]
);
const fakeRegistration = new Proxy(
{
__proto__: ServiceWorkerRegistration.prototype,
},
{
get(target, prop) {
if (prop === "installing") {
return null;
}
if (prop === "waiting") {
return null;
}
if (prop === "active") {
return handle;
}
if (prop === "scope") {
return ctx.args[0];
}
if (prop === "unregister") {
return () => {};
}
if (prop === "addEventListener") {
return () => {};
}
return Reflect.get(target, prop);
},
getOwnPropertyDescriptor: getOwnPropertyDescriptorHandler,
}
);
registration = fakeRegistration;
registrationmap.set(fakeRegistration, state);
ctx.return(new Promise((resolve) => resolve(fakeRegistration)));
},
});