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

fix: worker proxying

Browse source
This commit is contained in:
Percs 2024-10-11 15:19:22 -05:00
parent 3a7a7e95a5
commit 2a943adf91
1 changed files with 15 additions and 49 deletions
Download patch file Download diff file Expand all files Collapse all files

64
src/client/shared/worker.ts
View file

@ -1,3 +1,4 @@
import { iswindow } from "..";
import { BareMuxConnection } from "../../shared";
import { encodeUrl } from "../../shared/rewriters/url";
import type { MessageC2W } from "../../worker";
@ -6,7 +7,7 @@ import { ScramjetClient } from "../client";
const workerpostmessage = Worker.prototype.postMessage;
export default function (client: ScramjetClient, self: typeof globalThis) {
client.Proxy("Worker", {
const handler = {
construct({ args, call }) {
if (args[0] instanceof URL) args[0] = args[0].href;
if (args[0].startsWith("blob:") || args[0].startsWith("data:")) {
@ -48,57 +49,22 @@ export default function (client: ScramjetClient, self: typeof globalThis) {
);
})();
},
});
};
client.Proxy("Worklet.prototype.addModule", {
apply(ctx) {
if (ctx.args[0]) ctx.args[0] = encodeUrl(ctx.args[0], client.meta);
},
});
if (self.Worker) {
client.Proxy("Worker", handler);
}
client.Proxy("SharedWorker", {
construct({ args, call }) {
if (args[0] instanceof URL) args[0] = args[0].href;
if (args[0].startsWith("blob:") || args[0].startsWith("data:")) {
const data = syncfetch(client, args[0]);
const id = Math.random().toString(8).slice(5);
if (iswindow) {
client.Proxy("Worklet.prototype.addModule", {
apply(ctx) {
if (ctx.args[0]) ctx.args[0] = encodeUrl(ctx.args[0], client.meta);
},
});
args[0] = "/scramjet/worker?id=" + id;
if (args[1] && args[1].type === "module") {
args[0] += "&type=module";
}
args[0] += "&origin=" + encodeURIComponent(client.url.origin);
client.serviceWorker.controller?.postMessage({
scramjet$type: "dataworker",
data,
id,
} as MessageC2W);
} else {
args[0] = encodeUrl(args[0], client.meta) + "?dest=worker";
if (args[1] && args[1].type === "module") {
args[0] += "&type=module";
}
}
const worker = call();
const conn = new BareMuxConnection();
(async () => {
const port = await conn.getInnerPort();
workerpostmessage.call(
worker,
{
$scramjet$type: "baremuxinit",
port,
},
[port]
);
})();
},
});
// sharedworkers can only be constructed from window
client.Proxy("SharedWorker", handler);
}
}
function syncfetch(client: ScramjetClient, url: string) {