58 lines
2.4 KiB
TypeScript
58 lines
2.4 KiB
TypeScript
 
|
/* Build-time service worker injection. After Vite writes the bundle, the
|
||
|
|
hashed asset URLs referenced by dist/index.html and a content-derived
|
||
|
|
revision are written into dist/sw.js, so the offline shell is complete right
|
||
|
|
after install and every release ships a new worker with fresh caches.
|
||
|
|
Build tooling only — never imported by the app bundle. */
|
||
|
|
import { createHash } from "node:crypto";
|
||
|
|
import { readFileSync, writeFileSync } from "node:fs";
|
||
|
|
import { resolve } from "node:path";
|
||
|
|
import type { Plugin } from "vite";
|
||
|
|
|
||
|
|
export const REVISION_MARKER = "__NQ_BUILD_REVISION__";
|
||
|
|
const PRECACHE_PLACEHOLDER = "const PRECACHE_ASSETS = [];";
|
||
|
|
|
||
|
|
// Every same-origin URL an index.html loads via <script src> or <link href>,
|
||
|
|
// normalised to an absolute path, in document order and without duplicates.
|
||
|
|
export function extractShellAssets(indexHtml: string): string[] {
|
||
|
|
const tags = indexHtml.match(/<(?:script|link)\b[^>]*>/gi) ?? [];
|
||
|
|
const urls = tags.flatMap((tag) => {
|
||
|
|
const attr = /\b(?:src|href)\s*=\s*["']([^"']+)["']/i.exec(tag);
|
||
|
|
if (!attr) return [];
|
||
|
|
const url = new URL(attr[1], "https://same-origin.invalid/");
|
||
|
|
return url.origin === "https://same-origin.invalid" ? [url.pathname] : [];
|
||
|
|
});
|
||
|
|
return [...new Set(urls)];
|
||
|
|
}
|
||
|
|
|
||
|
|
// Returns swSource with the precache list and the build revision filled in.
|
||
|
|
export function injectServiceWorker(indexHtml: string, swSource: string): string {
|
||
|
|
if (!swSource.includes(PRECACHE_PLACEHOLDER) || !swSource.includes(REVISION_MARKER)) {
|
||
|
|
throw new Error("sw.js is missing the PRECACHE_ASSETS placeholder or revision marker");
|
||
|
|
}
|
||
|
|
const assets = extractShellAssets(indexHtml);
|
||
|
|
const revision = createHash("sha256")
|
||
|
|
.update(indexHtml)
|
||
|
|
.update(swSource)
|
||
|
|
.digest("hex")
|
||
|
|
.slice(0, 12);
|
||
|
|
return swSource
|
||
|
|
.replace(PRECACHE_PLACEHOLDER, `const PRECACHE_ASSETS = ${JSON.stringify(assets)};`)
|
||
|
|
.replaceAll(REVISION_MARKER, revision);
|
||
|
|
}
|
||
|
|
|
||
|
|
export function serviceWorkerInjectPlugin(): Plugin {
|
||
|
|
let outDir = "";
|
||
|
|
return {
|
||
|
|
name: "nestquest-sw-inject",
|
||
|
|
apply: "build",
|
||
|
|
configResolved(config) {
|
||
|
|
outDir = resolve(config.root, config.build.outDir);
|
||
|
|
},
|
||
|
|
writeBundle() {
|
||
|
|
const swPath = resolve(outDir, "sw.js");
|
||
|
|
const indexHtml = readFileSync(resolve(outDir, "index.html"), "utf8");
|
||
|
|
writeFileSync(swPath, injectServiceWorker(indexHtml, readFileSync(swPath, "utf8")));
|
||
|
|
},
|
||
|
|
};
|
||
|
|
}
|