1 files changed, 46 insertions, 0 deletions
diff --git a/src/components/BaseHead.astro b/src/components/BaseHead.astro
index 912375e..41926ad 100644
--- a/src/components/BaseHead.astro
+++ b/src/components/BaseHead.astro
@@ -87,3 +87,49 @@ const {
 <meta property="twitter:image" content={image} />
 
 <ClientRouter />
+
+<script>
+  import {
+    Workbox,
+    type WorkboxLifecycleWaitingEvent,
+  } from "workbox-window";
+
+  if ("serviceWorker" in navigator) {
+    const wb = new Workbox("/sw.js", { type: "module" });
+
+    wb.addEventListener("activated", (_event) => {
+      const urlsToCache = [
+        location.href,
+        ...performance.getEntriesByType("resource").map((r) => r.name),
+      ];
+      wb.messageSW({
+        type: "CACHE_URLS",
+        payload: { urlsToCache },
+      });
+    });
+
+    const showSkipWaitingPrompt = async (
+      _event: WorkboxLifecycleWaitingEvent,
+    ) => {
+      wb.addEventListener("controlling", () => {
+        window.location.reload();
+      });
+
+      const updateAccepted = await promptForUpdate();
+
+      if (updateAccepted) {
+        wb.messageSkipWaiting();
+      }
+    };
+
+    wb.addEventListener("waiting", (event) => {
+      showSkipWaitingPrompt(event);
+    });
+
+    wb.register();
+  }
+
+  async function promptForUpdate(): Promise<boolean> {
+    return await Promise.resolve(confirm("update?"));
+  }
+</script>