Review fixes

Author: ScriptSmith

src/wasm.rs

@@ -172,7 +172,7 @@ impl HadrianGateway {
 
         let response = tower::ServiceExt::oneshot(self.router.clone(), http_request)
             .await
-            .unwrap();
+            .map_err(|e| JsError::new(&format!("Router error: {e}")))?;
 
         convert_response(response).await
     }

ui/src/service-worker/register.ts

@@ -22,14 +22,19 @@ export async function registerWasmServiceWorker(): Promise<void> {
     // Wait for the SW to be active (handles installing, waiting, or already active)
     const sw = registration.active || registration.waiting || registration.installing;
     if (sw && sw.state !== "activated") {
-      await new Promise<void>((resolve) => {
-        sw.addEventListener("statechange", function handler() {
-          if (sw.state === "activated") {
-            sw.removeEventListener("statechange", handler);
-            resolve();
-          }
-        });
-      });
+      await Promise.race([
+        new Promise<void>((resolve) => {
+          sw.addEventListener("statechange", function handler() {
+            if (sw.state === "activated") {
+              sw.removeEventListener("statechange", handler);
+              resolve();
+            }
+          });
+        }),
+        new Promise<void>((_, reject) =>
+          setTimeout(() => reject(new Error("Service worker activation timed out")), 10_000)
+        ),
+      ]);
     }
 
     // Ensure this page is controlled by the SW — even after activation,

ui/src/service-worker/sqlite-bridge.ts

@@ -47,20 +47,36 @@ function saveToIndexedDB(data: Uint8Array): void {
   req.onupgradeneeded = () => req.result.createObjectStore(IDB_STORE);
   req.onsuccess = () => {
     const tx = req.result.transaction(IDB_STORE, "readwrite");
-    tx.objectStore(IDB_STORE).put(data, IDB_KEY);
+    const putReq = tx.objectStore(IDB_STORE).put(data, IDB_KEY);
+    putReq.onerror = () => console.error("[sqlite-bridge] IndexedDB put failed:", putReq.error);
+    tx.onerror = () => console.error("[sqlite-bridge] IndexedDB transaction failed:", tx.error);
   };
+  req.onerror = () => console.error("[sqlite-bridge] IndexedDB open failed:", req.error);
 }
 
 let saveTimer: ReturnType<typeof setTimeout> | null = null;
+
+function flushSave(): void {
+  if (!db || !saveTimer) return;
+  clearTimeout(saveTimer);
+  saveTimer = null;
+  const data = db.export();
+  saveToIndexedDB(new Uint8Array(data));
+}
+
 function debouncedSave(): void {
   if (!db) return;
   if (saveTimer) clearTimeout(saveTimer);
   saveTimer = setTimeout(() => {
+    saveTimer = null;
     const data = db!.export();
     saveToIndexedDB(new Uint8Array(data));
   }, 500);
 }
 
+// Flush pending saves before the service worker terminates
+globalThis.addEventListener("unload", flushSave);
+
 // ---------------------------------------------------------------------------
 // Parameter binding
 // ---------------------------------------------------------------------------
@@ -117,15 +133,18 @@ const bridge = {
     if (!db) throw new Error("Database not initialized — call init_database() first");
 
     const stmt = db.prepare(sql);
-    stmt.bind(bindParams(params));
-
-    const rows: Record<string, unknown>[] = [];
-    while (stmt.step()) {
-      const row = stmt.getAsObject();
-      rows.push(row as Record<string, unknown>);
+    try {
+      stmt.bind(bindParams(params));
+
+      const rows: Record<string, unknown>[] = [];
+      while (stmt.step()) {
+        const row = stmt.getAsObject();
+        rows.push(row as Record<string, unknown>);
+      }
+      return rows;
+    } finally {
+      stmt.free();
     }
-    stmt.free();
-    return rows;
   },
 
   async execute(