From c344032c0a17cf679f3a847d131f95275c36d349 Mon Sep 17 00:00:00 2001
From: Bruno Windels <bruno@windels.cloud>
Date: Thu, 9 Dec 2021 11:39:28 +0100
Subject: [PATCH] transpile service worker and cleanup build plugin

---
 scripts/build-plugins/service-worker.js | 50 ++++++++++++-------------
 vite.config.js                          |  2 +-
 2 files changed, 26 insertions(+), 26 deletions(-)

diff --git a/scripts/build-plugins/service-worker.js b/scripts/build-plugins/service-worker.js
index 987debbe..e7a77048 100644
--- a/scripts/build-plugins/service-worker.js
+++ b/scripts/build-plugins/service-worker.js
@@ -8,42 +8,42 @@ function contentHash(str) {
     return hasher.digest();
 }
 
-module.exports = function injectServiceWorker(swFile) {
+module.exports = function injectServiceWorker(swFile, otherUncachedFiles) {
+    const swName = path.basename(swFile);
     let root;
     let version;
-    let manifestHref;
+
     return {
         name: "hydrogen:injectServiceWorker",
         apply: "build",
         enforce: "post",
+        buildStart() {
+            this.emitFile({
+                type: "chunk",
+                fileName: swName,
+                id: swFile,
+            });
+        },
         configResolved: config => {
             root = config.root;
             version = JSON.parse(config.define.HYDROGEN_VERSION); // unquote
         },
-        generateBundle: async function(_, bundle) {
-            const absoluteSwFile = path.resolve(root, swFile);
-            let swSource = await fs.readFile(absoluteSwFile, {encoding: "utf8"});
-            const index = bundle["index.html"];
-            if (!index) {
-                console.log("index not found", index);
-            }
-            const uncachedFileContentMap = {
-                "index.html": index.source,
-                "sw.js": swSource
-            };
+        generateBundle: async function(options, bundle) {
+            const uncachedFileNames = [swName].concat(otherUncachedFiles);
+            const uncachedFileContentMap = uncachedFileNames.reduce((map, fileName) => {
+                const chunkOrAsset = bundle[fileName];
+                if (!chunkOrAsset) {
+                    throw new Error("could not get content for uncached asset or chunk " + fileName);
+                }
+                map[fileName] = chunkOrAsset.source || chunkOrAsset.code;
+                return map;
+            }, {});
             const assets = Object.values(bundle);
-            const cachedFileNames = assets.map(o => o.fileName).filter(fileName => fileName !== "index.html");
+            const cachedFileNames = assets.map(o => o.fileName).filter(fileName => !uncachedFileContentMap[fileName]);
             const globalHash = getBuildHash(cachedFileNames, uncachedFileContentMap);
-            swSource = await buildServiceWorker(swSource, version, globalHash, assets);
-            const outputName = path.basename(absoluteSwFile);
-            // TODO: do normal build transformations for service worker too,
-            // I think if we emit it as a chunk rather than an asset it would
-            // but we can't emit chunks anymore in generateBundle I think ...
-            this.emitFile({
-                type: "asset",
-                fileName: outputName,
-                source: swSource
-            });
+            const sw = bundle[swName];
+            sw.code = replaceConstsInServiceWorker(sw.code, version, globalHash, assets);
+            console.log(`\nBuilt ${version} (${globalHash})`);
         }
     };
 }
@@ -75,7 +75,7 @@ function isPreCached(asset) {
             fileName.endsWith(".js") && !NON_PRECACHED_JS.includes(path.basename(name));
 }
 
-async function buildServiceWorker(swSource, version, globalHash, assets) {
+function replaceConstsInServiceWorker(swSource, version, globalHash, assets) {
     const unhashedPreCachedAssets = [];
     const hashedPreCachedAssets = [];
     const hashedCachedOnRequestAssets = [];
diff --git a/vite.config.js b/vite.config.js
index 28f9f5a6..ef41babf 100644
--- a/vite.config.js
+++ b/vite.config.js
@@ -50,7 +50,7 @@ export default {
         // important this comes before service worker
         // otherwise the manifest and the icons it refers to won't be cached
         injectWebManifest("assets/manifest.json"),
-        injectServiceWorker("sw.js"),
+        injectServiceWorker("./src/platform/web/sw.js", ["index.html"]),
     ],
     define: {
         "HYDROGEN_VERSION": JSON.stringify(version)