File size: 4,173 Bytes
bee6636 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 |
import { FakeServiceWorker } from "./fakesw";
import { handleFetch } from "./fetch";
import BareClient from "@mercuryworkshop/bare-mux";
import { ScramjetConfig } from "../types";
import { asyncSetWasm } from "../shared/rewriters/wasm";
import { CookieStore } from "../shared/cookie";
import { config, loadCodecs, setConfig } from "../shared";
export class ScramjetServiceWorker extends EventTarget {
client: BareClient;
config: ScramjetConfig;
syncPool: Record<number, (val?: any) => void> = {};
synctoken = 0;
cookieStore = new CookieStore();
serviceWorkers: FakeServiceWorker[] = [];
constructor() {
super();
this.client = new BareClient();
const db = indexedDB.open("$scramjet", 1);
db.onsuccess = () => {
const res = db.result;
const tx = res.transaction("cookies", "readonly");
const store = tx.objectStore("cookies");
const cookies = store.get("cookies");
cookies.onsuccess = () => {
if (cookies.result) {
this.cookieStore.load(cookies.result);
}
};
};
addEventListener("message", async ({ data }: { data: MessageC2W }) => {
if (!("scramjet$type" in data)) return;
if ("scramjet$token" in data) {
// (ack message)
const cb = this.syncPool[data.scramjet$token];
delete this.syncPool[data.scramjet$token];
cb(data);
return;
}
if (data.scramjet$type === "registerServiceWorker") {
this.serviceWorkers.push(new FakeServiceWorker(data.port, data.origin));
return;
}
if (data.scramjet$type === "cookie") {
this.cookieStore.setCookies([data.cookie], new URL(data.url));
const res = db.result;
const tx = res.transaction("cookies", "readwrite");
const store = tx.objectStore("cookies");
store.put(JSON.parse(this.cookieStore.dump()), "cookies");
}
if (data.scramjet$type === "loadConfig") {
this.config = data.config;
}
});
}
async dispatch(client: Client, data: MessageW2C): Promise<MessageC2W> {
const token = this.synctoken++;
let cb: (val: MessageC2W) => void;
const promise: Promise<MessageC2W> = new Promise((r) => (cb = r));
this.syncPool[token] = cb;
data.scramjet$token = token;
client.postMessage(data);
return await promise;
}
async loadConfig() {
if (this.config) return;
const request = indexedDB.open("$scramjet", 1);
return new Promise<void>((resolve, reject) => {
request.onsuccess = async () => {
const db = request.result;
const tx = db.transaction("config", "readonly");
const store = tx.objectStore("config");
const storedconfig = store.get("config");
storedconfig.onsuccess = async () => {
this.config = storedconfig.result;
setConfig(storedconfig.result);
await asyncSetWasm();
resolve();
};
storedconfig.onerror = () => reject(storedconfig.error);
};
request.onerror = () => reject(request.error);
});
}
route({ request }: FetchEvent) {
if (request.url.startsWith(location.origin + this.config.prefix))
return true;
else if (request.url.startsWith(location.origin + this.config.files.wasm))
return true;
else return false;
}
async fetch({ request, clientId }: FetchEvent) {
if (!this.config) await this.loadConfig();
const client = await self.clients.get(clientId);
return handleFetch.call(this, request, client);
}
}
// @ts-ignore
self.ScramjetServiceWorker = ScramjetServiceWorker;
type RegisterServiceWorkerMessage = {
scramjet$type: "registerServiceWorker";
port: MessagePort;
origin: string;
};
type CookieMessage = {
scramjet$type: "cookie";
cookie: string;
url: string;
};
type ConfigMessage = {
scramjet$type: "loadConfig";
config: ScramjetConfig;
};
type MessageCommon = {
scramjet$type: string;
scramjet$token?: number;
};
type MessageTypeC2W =
| RegisterServiceWorkerMessage
| CookieMessage
| ConfigMessage;
type MessageTypeW2C = CookieMessage;
// c2w: client to (service) worker
export type MessageC2W = MessageCommon & MessageTypeC2W;
export type MessageW2C = MessageCommon & MessageTypeW2C;
|