Show the update banner only when the running build differs from the server and make Refresh UI land the new shell on flaky links
This commit is contained in:
152
frontend/app.js
152
frontend/app.js
@@ -925,6 +925,7 @@ function toast(msg, { duration = 2200 } = {}) {
|
||||
first.classList.add('toast-exit');
|
||||
setTimeout(() => first.remove(), 280);
|
||||
}
|
||||
return t;
|
||||
}
|
||||
function uid() { return Date.now().toString(36) + Math.random().toString(36).slice(2, 7); }
|
||||
|
||||
@@ -5081,24 +5082,41 @@ function importBackup(e) {
|
||||
// live and we surface the existing update banner so the user can reload.
|
||||
// ============================================================================
|
||||
|
||||
// The build this page is actually running: the server stamps it into
|
||||
// index.html (<meta name="ytp-build">), and the service worker caches that
|
||||
// index.html together with the rest of the shell, so it always describes the
|
||||
// code in this tab. Null when served without the server (static test serve).
|
||||
const RUNNING_BUILD = (() => {
|
||||
const v = document.querySelector('meta[name="ytp-build"]')?.content || '';
|
||||
return v && !v.startsWith('__') ? v : null;
|
||||
})();
|
||||
|
||||
// Fallback baseline for pages without the meta tag: the first tag seen.
|
||||
let _knownBuildTag = null;
|
||||
|
||||
async function checkBuildTag() {
|
||||
try {
|
||||
const res = await fetch('/api/version', { cache: 'no-store' });
|
||||
if (!res.ok) return;
|
||||
const data = await res.json();
|
||||
const tag = data.buildTag;
|
||||
if (!tag) return;
|
||||
if (!_knownBuildTag) {
|
||||
_knownBuildTag = tag; // baseline on first successful fetch
|
||||
return;
|
||||
}
|
||||
if (tag !== _knownBuildTag) {
|
||||
_knownBuildTag = tag; // update so we don't show the banner twice
|
||||
showUpdateBanner();
|
||||
}
|
||||
} catch { /* network error — try again next interval */ }
|
||||
async function fetchServerBuild() {
|
||||
const res = await fetch('/api/version', { cache: 'no-store' });
|
||||
if (!res.ok) return null;
|
||||
const data = await res.json();
|
||||
return data.buildTag || null;
|
||||
}
|
||||
|
||||
// The ONLY way the update banner opens: the server runs a different build
|
||||
// than this page. Every trigger (the poll, a waiting worker, the activate
|
||||
// broadcast) funnels through here, so none of them can re-open the banner
|
||||
// once the page is current — which is what kept happening before.
|
||||
async function maybeShowUpdateBanner() {
|
||||
if (_updateBannerShown || _updating) return;
|
||||
let server = null;
|
||||
try { server = await fetchServerBuild(); } catch { return; /* offline — next poll */ }
|
||||
if (!server) return;
|
||||
const running = RUNNING_BUILD || _knownBuildTag;
|
||||
if (!running) { _knownBuildTag = server; return; }
|
||||
if (server !== running) showUpdateBanner(server);
|
||||
}
|
||||
|
||||
function checkBuildTag() {
|
||||
return maybeShowUpdateBanner();
|
||||
}
|
||||
|
||||
function pollBuildTag() {
|
||||
@@ -5134,7 +5152,9 @@ async function hardReloadUI() {
|
||||
}
|
||||
|
||||
let _updateBannerShown = false;
|
||||
function showUpdateBanner() {
|
||||
let _updating = false;
|
||||
let _updateToast = null;
|
||||
function showUpdateBanner(target) {
|
||||
// Only show the dialog once per page load
|
||||
if (_updateBannerShown) return;
|
||||
_updateBannerShown = true;
|
||||
@@ -5146,7 +5166,7 @@ function showUpdateBanner() {
|
||||
{ label: 'Later', onClick: closeModal },
|
||||
{ label: 'Refresh UI', primary: true, onClick: async () => {
|
||||
closeModal();
|
||||
await applyUpdate();
|
||||
await applyUpdate(target);
|
||||
}},
|
||||
]);
|
||||
}
|
||||
@@ -5155,16 +5175,85 @@ function showUpdateBanner() {
|
||||
// waiting worker without re-querying getRegistration().
|
||||
let _swReg = null;
|
||||
|
||||
// Applies a pending SW update in place (see sw-update.js for the full
|
||||
// rationale — this used to call hardReloadUI(), which caused the "Update
|
||||
// ready" banner to reappear right after being applied).
|
||||
async function applyUpdate() {
|
||||
const reg = _swReg || (('serviceWorker' in navigator) ? await navigator.serviceWorker.getRegistration() : null);
|
||||
await window.SwUpdate.applyUpdate({
|
||||
reg,
|
||||
container: navigator.serviceWorker,
|
||||
reload: () => window.location.reload(),
|
||||
});
|
||||
// "Refresh UI": download the new shell into the SW caches (all-or-nothing),
|
||||
// activate a waiting worker if any, reload once — see sw-update.js. The
|
||||
// attempt is recorded in sessionStorage so the reloaded page can confirm it
|
||||
// is really running `target` (checkUpdateOutcome) instead of looping.
|
||||
const UPDATE_STATE_KEY = 'ytpUpdateAttempt';
|
||||
const UPDATE_MAX_ATTEMPTS = 3;
|
||||
const SHELL_CACHE_RE = /^ytplayer-(?!thumbs$|fonts$)/;
|
||||
|
||||
function readUpdateState() {
|
||||
try { return JSON.parse(sessionStorage.getItem(UPDATE_STATE_KEY) || 'null'); } catch { return null; }
|
||||
}
|
||||
function writeUpdateState(st) {
|
||||
try {
|
||||
if (st) sessionStorage.setItem(UPDATE_STATE_KEY, JSON.stringify(st));
|
||||
else sessionStorage.removeItem(UPDATE_STATE_KEY);
|
||||
} catch { /* storage blocked — the update still applies, just unverified */ }
|
||||
}
|
||||
|
||||
async function applyUpdate(target, { quiet = false } = {}) {
|
||||
if (_updating) return;
|
||||
_updating = true;
|
||||
const prev = readUpdateState();
|
||||
writeUpdateState({ target, attempt: (prev ? prev.attempt : 0) + 1 });
|
||||
if (!quiet) _updateToast = toast('Downloading the update…', { duration: 125000 });
|
||||
try {
|
||||
const reg = _swReg || (('serviceWorker' in navigator) ? await navigator.serviceWorker.getRegistration() : null);
|
||||
const SU = window.SwUpdate;
|
||||
const reload = () => window.location.reload();
|
||||
if (!SU) { reload(); return; } // helper failed to load (flaky link): a reload is the best we can do
|
||||
await SU.applyUpdate({
|
||||
reg,
|
||||
container: navigator.serviceWorker,
|
||||
reload,
|
||||
refreshShell: ('caches' in window) ? () => SU.refreshShellInPlace({
|
||||
cachesApi: caches,
|
||||
fetchFn: (url, init) => fetch(url, init),
|
||||
isShellCache: (name) => SHELL_CACHE_RE.test(name),
|
||||
timeoutMs: 120000,
|
||||
attempts: 3,
|
||||
attemptTimeoutMs: 25000,
|
||||
setTimeout: (fn, ms) => setTimeout(fn, ms),
|
||||
}) : null,
|
||||
});
|
||||
} catch (err) {
|
||||
_updating = false;
|
||||
writeUpdateState(null);
|
||||
if (_updateToast) { _updateToast.remove(); _updateToast = null; }
|
||||
toast('⚠ Update didn’t download (' + (err && err.message ? err.message : 'connection problem') + '). It will be offered again.', { duration: 6000 });
|
||||
// Let the next poll re-offer it.
|
||||
_updateBannerShown = false;
|
||||
}
|
||||
}
|
||||
|
||||
// After an update reload: confirm the page now runs the build it asked for.
|
||||
// If a stubborn cache still served the old one, retry (bounded) instead of
|
||||
// re-opening the banner in a loop.
|
||||
async function checkUpdateOutcome() {
|
||||
const st = readUpdateState();
|
||||
if (!st) return;
|
||||
// Suppress the banner while this is being decided.
|
||||
_updateBannerShown = true;
|
||||
let server = null;
|
||||
try { server = await fetchServerBuild(); } catch { /* offline */ }
|
||||
// Current if it runs what was asked for — or whatever is deployed now (a
|
||||
// newer deploy may have landed in between).
|
||||
if (!RUNNING_BUILD || RUNNING_BUILD === st.target || (server && RUNNING_BUILD === server)) {
|
||||
writeUpdateState(null);
|
||||
_updateBannerShown = false;
|
||||
if (RUNNING_BUILD) toast('Updated ✓');
|
||||
return;
|
||||
}
|
||||
if (st.attempt < UPDATE_MAX_ATTEMPTS) {
|
||||
_updateToast = toast('Finishing the update…', { duration: 125000 });
|
||||
setTimeout(() => applyUpdate(server || st.target, { quiet: true }), 1500);
|
||||
return;
|
||||
}
|
||||
_updateBannerShown = false;
|
||||
writeUpdateState(null);
|
||||
toast('⚠ Couldn’t finish updating — the connection looks unstable. It will be offered again later.', { duration: 6000 });
|
||||
}
|
||||
|
||||
async function registerServiceWorker() {
|
||||
@@ -5186,20 +5275,20 @@ async function registerServiceWorker() {
|
||||
|
||||
// If a new SW is already waiting (e.g. user refreshed after an update),
|
||||
// show the dialog right away.
|
||||
if (reg.waiting && hasController()) { showUpdateBanner(); return; }
|
||||
if (reg.waiting && hasController()) maybeShowUpdateBanner();
|
||||
|
||||
// Listen for a new SW installing after the page is open.
|
||||
reg.addEventListener('updatefound', () => {
|
||||
const sw = reg.installing;
|
||||
if (!sw) return;
|
||||
sw.addEventListener('statechange', () => {
|
||||
if (sw.state === 'installed' && reg.waiting && hasController()) showUpdateBanner();
|
||||
if (sw.state === 'installed' && reg.waiting && hasController()) maybeShowUpdateBanner();
|
||||
});
|
||||
});
|
||||
|
||||
// The SW can also broadcast SW_UPDATE_AVAILABLE on its own activate.
|
||||
navigator.serviceWorker.addEventListener('message', (e) => {
|
||||
if (e.data && e.data.type === 'SW_UPDATE_AVAILABLE') showUpdateBanner();
|
||||
if (e.data && e.data.type === 'SW_UPDATE_AVAILABLE') maybeShowUpdateBanner();
|
||||
});
|
||||
|
||||
// Check for updates in the background (useful for long-lived sessions)
|
||||
@@ -5269,6 +5358,7 @@ async function boot() {
|
||||
els.searchInput.focus();
|
||||
|
||||
// Register service worker + ping server with fingerprint (WEB mode only)
|
||||
if (WEB) checkUpdateOutcome();
|
||||
registerServiceWorker();
|
||||
if (WEB) pollBuildTag();
|
||||
if (WEB) {
|
||||
|
||||
@@ -4,6 +4,9 @@
|
||||
<meta charset="UTF-8" />
|
||||
<meta name="viewport" content="width=device-width, initial-scale=1.0, viewport-fit=cover" />
|
||||
<meta name="theme-color" content="#1a1311" />
|
||||
<!-- Build this page was served from — the server stamps it at request time;
|
||||
app.js compares it with /api/version before offering an update. -->
|
||||
<meta name="ytp-build" content="__BUILD_TAG__" />
|
||||
<meta name="mobile-web-app-capable" content="yes" />
|
||||
<meta name="apple-mobile-web-app-capable" content="yes" />
|
||||
<meta name="apple-mobile-web-app-status-bar-style" content="black-translucent" />
|
||||
|
||||
@@ -1,5 +1,5 @@
|
||||
/* ============================================================================
|
||||
* sw-update — applies a waiting service-worker update in place.
|
||||
* sw-update — makes "Refresh UI" actually land the new build.
|
||||
*
|
||||
* Framework-free and dependency-free on purpose (same pattern as
|
||||
* async-guard.js):
|
||||
@@ -7,53 +7,128 @@
|
||||
* global `window.SwUpdate`).
|
||||
* • `require`-able by `node --test` (CommonJS `module.exports`).
|
||||
*
|
||||
* Bug this fixes (Task #61): "Update ready" kept reappearing right after the
|
||||
* user clicked "Reload now" / "Refresh UI". The old flow called
|
||||
* hardReloadUI(), which unregisters the service worker and wipes every cache
|
||||
* before navigating — forcing a brand-new install on the next load. That
|
||||
* fresh install briefly has `reg.waiting` truthy again (a normal but
|
||||
* transient SW lifecycle state), which registerServiceWorker() misread as a
|
||||
* genuinely new update and re-showed the banner immediately.
|
||||
* History: the "Update available keeps showing" loop was patched three times
|
||||
* by chasing individual triggers (a transient `reg.waiting`, the activate
|
||||
* broadcast, the buildTag poll firing before a worker was waiting). It kept
|
||||
* coming back on the real deployment because the homelab link is slow and
|
||||
* drops requests: the new worker's install (`cache.addAll`, all-or-nothing)
|
||||
* failed, "Refresh UI" reloaded into the OLD cached shell, and the next
|
||||
* successful install re-opened the banner — sometimes on a page that was
|
||||
* already current (a failed install leaves a partial cache that the
|
||||
* activate handler then mistook for a previous deploy).
|
||||
*
|
||||
* Fix: activate the *already-installed* waiting worker in place —
|
||||
* postMessage SKIP_WAITING to it, wait for it to actually take control
|
||||
* (`controllerchange`), and only then reload. The reloaded page is served by
|
||||
* the new SW from its very first request, and no fresh install/registration
|
||||
* cycle happens, so the banner has nothing to spuriously re-trigger on.
|
||||
* The fix no longer depends on the service-worker install lifecycle:
|
||||
* 1. refreshShellInPlace() downloads a fresh copy of every file the shell
|
||||
* caches hold (cache-busted, so even an old worker's cache-first
|
||||
* handler can't answer with the stale copy) — ALL of them or nothing —
|
||||
* and writes them into every versioned shell cache. Whichever worker
|
||||
* serves the next load, it serves the new build.
|
||||
* 2. Only then is a waiting worker (if any) activated, and the page
|
||||
* reloaded once.
|
||||
* The banner itself only opens when the build the page is running differs
|
||||
* from the server's (see app.js maybeShowUpdateBanner), so no lifecycle
|
||||
* event can re-open it once the page is current.
|
||||
* ========================================================================== */
|
||||
(function (root) {
|
||||
'use strict';
|
||||
|
||||
const BUST_PARAM = '__ytpfresh';
|
||||
|
||||
function withTimeout(promise, ms, setTimeoutFn, label) {
|
||||
if (!ms || !setTimeoutFn) return promise;
|
||||
return Promise.race([
|
||||
promise,
|
||||
new Promise((_, reject) => setTimeoutFn(() => reject(new Error(label || 'timed out')), ms)),
|
||||
]);
|
||||
}
|
||||
|
||||
/**
|
||||
* Applies a pending SW update: messages the waiting worker to skipWaiting(),
|
||||
* waits for controllerchange, then reloads exactly once.
|
||||
* Re-downloads every file held by the versioned shell caches and replaces
|
||||
* the cached copies. All-or-nothing: if any download fails, nothing is
|
||||
* written and it throws, so a flaky connection can never leave a
|
||||
* half-old/half-new shell.
|
||||
*
|
||||
* @param {object} opts
|
||||
* @param {ServiceWorkerRegistration|null} opts.reg the current registration
|
||||
* @param {ServiceWorkerContainer} opts.container navigator.serviceWorker
|
||||
* @param {() => void} opts.reload called at most once
|
||||
* @param {(fn: () => void, ms: number) => any} [opts.setTimeout] injectable for tests
|
||||
* @returns {Promise<void>}
|
||||
* @param {CacheStorage} opts.cachesApi
|
||||
* @param {(url: string, init: object) => Promise<Response>} opts.fetchFn
|
||||
* @param {(name: string) => boolean} opts.isShellCache
|
||||
* @param {string} [opts.bust] cache-busting token
|
||||
* @param {number} [opts.timeoutMs] overall download budget
|
||||
* @param {number} [opts.attempts] tries per file (default 3)
|
||||
* @param {number} [opts.attemptTimeoutMs] bound on a single try
|
||||
* @param {Function} [opts.setTimeout]
|
||||
* @returns {Promise<{refreshed: number, caches: number}>}
|
||||
*/
|
||||
async function applyUpdate({ reg, container, reload, setTimeout: setTimeoutFn }) {
|
||||
const scheduleTimeout = setTimeoutFn || (typeof setTimeout !== 'undefined' ? setTimeout : null);
|
||||
let waiting = reg && reg.waiting;
|
||||
async function refreshShellInPlace({ cachesApi, fetchFn, isShellCache, bust, timeoutMs, attempts, attemptTimeoutMs, setTimeout: setTimeoutFn }) {
|
||||
const names = (await cachesApi.keys()).filter(isShellCache);
|
||||
if (!names.length) return { refreshed: 0, caches: 0 }; // uncontrolled page: a reload already hits the network
|
||||
|
||||
if (!waiting && reg && typeof reg.update === 'function') {
|
||||
// The banner can be triggered by the server buildTag poll before the
|
||||
// browser has fetched the new sw.js at all. With no waiting worker, a
|
||||
// bare reload would be served the OLD cache-first shell, the new SW
|
||||
// would then install in the background, and the banner would reappear
|
||||
// — the "update available keeps showing" loop. Fetch the update now
|
||||
// and wait (bounded) for it to reach `installed` so a single click
|
||||
// activates the new version.
|
||||
try { await reg.update(); } catch { /* offline / fetch failed — fall through */ }
|
||||
waiting = reg.waiting || (await waitForInstalled(reg, scheduleTimeout, 8000));
|
||||
const urls = new Set();
|
||||
for (const name of names) {
|
||||
const cache = await cachesApi.open(name);
|
||||
for (const req of await cache.keys()) {
|
||||
const u = new URL(req.url);
|
||||
if (u.searchParams.has(BUST_PARAM)) { await cache.delete(req); continue; } // leftovers of a failed try
|
||||
urls.add(u.href);
|
||||
}
|
||||
}
|
||||
|
||||
const token = bust || String(Date.now());
|
||||
// Each file gets a few attempts (the homelab link drops requests), each
|
||||
// bounded so one hung connection can't eat the whole budget.
|
||||
const fetchOne = async (url) => {
|
||||
let last = null;
|
||||
for (let attempt = 0; attempt < (attempts || 3); attempt++) {
|
||||
const u = new URL(url);
|
||||
u.searchParams.set(BUST_PARAM, token + '-' + attempt);
|
||||
const ctl = typeof AbortController !== 'undefined' ? new AbortController() : null;
|
||||
try {
|
||||
const res = await withTimeout(
|
||||
fetchFn(u.href, { cache: 'reload', credentials: 'same-origin', signal: ctl ? ctl.signal : undefined }),
|
||||
attemptTimeoutMs, setTimeoutFn, `${new URL(url).pathname} timed out`);
|
||||
if (res && res.ok) return [url, res];
|
||||
last = new Error(`${new URL(url).pathname} → ${res ? res.status : 'no response'}`);
|
||||
} catch (err) {
|
||||
if (ctl) { try { ctl.abort(); } catch { /* already done */ } }
|
||||
last = err;
|
||||
}
|
||||
}
|
||||
throw last;
|
||||
};
|
||||
const fetched = await withTimeout(Promise.all([...urls].map(fetchOne)),
|
||||
timeoutMs, setTimeoutFn, 'update download timed out');
|
||||
|
||||
for (const name of names) {
|
||||
const cache = await cachesApi.open(name);
|
||||
for (const [url, res] of fetched) await cache.put(url, res.clone());
|
||||
// An old worker's cache-first handler may have stored the busted URLs.
|
||||
for (const req of await cache.keys()) {
|
||||
if (new URL(req.url).searchParams.has(BUST_PARAM)) await cache.delete(req);
|
||||
}
|
||||
}
|
||||
return { refreshed: fetched.length, caches: names.length };
|
||||
}
|
||||
|
||||
/**
|
||||
* Applies an update: refresh the shell in place (throws on failure — the
|
||||
* caller reports it and nothing reloads), then activate a waiting worker if
|
||||
* there is one, then reload exactly once.
|
||||
*
|
||||
* @param {object} opts
|
||||
* @param {ServiceWorkerRegistration|null} opts.reg
|
||||
* @param {ServiceWorkerContainer} opts.container
|
||||
* @param {() => void} opts.reload called at most once
|
||||
* @param {() => Promise<any>} [opts.refreshShell] refreshShellInPlace bound to real APIs
|
||||
* @param {Function} [opts.setTimeout] injectable for tests
|
||||
* @returns {Promise<void>}
|
||||
*/
|
||||
async function applyUpdate({ reg, container, reload, refreshShell, setTimeout: setTimeoutFn }) {
|
||||
const scheduleTimeout = setTimeoutFn || (typeof setTimeout !== 'undefined' ? setTimeout : null);
|
||||
|
||||
if (refreshShell) await refreshShell();
|
||||
|
||||
const waiting = reg && reg.waiting;
|
||||
if (!waiting) {
|
||||
// Nothing to activate (e.g. banner was shown from a broadcast message
|
||||
// rather than an actual waiting worker) — just reload.
|
||||
reload();
|
||||
return;
|
||||
}
|
||||
@@ -64,34 +139,14 @@
|
||||
reloaded = true;
|
||||
reload();
|
||||
};
|
||||
|
||||
// The waiting worker's cache was refreshed too, so activating it is safe
|
||||
// either way; reload on controllerchange, or after a short safety net.
|
||||
container.addEventListener('controllerchange', reloadOnce, { once: true });
|
||||
// Safety net in case controllerchange never fires (e.g. no controller yet).
|
||||
if (scheduleTimeout) scheduleTimeout(reloadOnce, 3000);
|
||||
|
||||
if (scheduleTimeout) scheduleTimeout(reloadOnce, 4000);
|
||||
waiting.postMessage({ type: 'SKIP_WAITING' });
|
||||
}
|
||||
|
||||
/**
|
||||
* Waits for reg.installing to reach the `installed` state (at which point
|
||||
* it becomes reg.waiting), bounded by a timeout. Resolves with the waiting
|
||||
* worker or null.
|
||||
*/
|
||||
function waitForInstalled(reg, scheduleTimeout, ms) {
|
||||
return new Promise((resolve) => {
|
||||
const sw = reg.installing;
|
||||
if (!sw || typeof sw.addEventListener !== 'function') { resolve(null); return; }
|
||||
let settled = false;
|
||||
const settle = (v) => { if (!settled) { settled = true; resolve(v); } };
|
||||
sw.addEventListener('statechange', () => {
|
||||
if (sw.state === 'installed') settle(reg.waiting || sw);
|
||||
else if (sw.state === 'redundant') settle(null);
|
||||
});
|
||||
if (scheduleTimeout) scheduleTimeout(() => settle(reg.waiting || null), ms);
|
||||
});
|
||||
}
|
||||
|
||||
const SwUpdate = { applyUpdate };
|
||||
const SwUpdate = { applyUpdate, refreshShellInPlace, BUST_PARAM };
|
||||
|
||||
if (typeof module !== 'undefined' && module.exports) {
|
||||
module.exports = SwUpdate;
|
||||
|
||||
@@ -1,22 +1,24 @@
|
||||
'use strict';
|
||||
|
||||
/**
|
||||
* Unit tests for sw-update.js (Task #61).
|
||||
* Unit tests for sw-update.js — the "Update available keeps showing after
|
||||
* Refresh UI" loop.
|
||||
*
|
||||
* Bug: "Update ready / Reload now" reappeared immediately after the user
|
||||
* clicked "Reload now". Root cause — the old flow unregistered the service
|
||||
* worker and wiped every cache before navigating, forcing a brand-new
|
||||
* install on the next load; that fresh install's registration briefly has
|
||||
* `reg.waiting` truthy again (a normal but transient SW lifecycle state),
|
||||
* which was misread as a new pending update and re-showed the banner.
|
||||
* On the real deployment (slow homelab link that drops requests) the new
|
||||
* service worker's all-or-nothing install kept failing, so "Refresh UI"
|
||||
* reloaded into the old cached shell and the banner came back. The fix
|
||||
* refreshes the cached shell in place (all-or-nothing, cache-busted) before
|
||||
* activating any waiting worker and reloading once.
|
||||
*
|
||||
* These tests drive applyUpdate() directly against mocked registration /
|
||||
* container / reload objects — no real service worker or browser needed.
|
||||
* No real service worker or browser needed: fake CacheStorage / fetch /
|
||||
* registration / container objects drive the functions directly.
|
||||
*/
|
||||
|
||||
const { test } = require('node:test');
|
||||
const assert = require('node:assert');
|
||||
const { applyUpdate } = require('./sw-update');
|
||||
const { applyUpdate, refreshShellInPlace, BUST_PARAM } = require('./sw-update');
|
||||
|
||||
const ORIGIN = 'https://worship.example';
|
||||
|
||||
// A minimal fake ServiceWorkerContainer supporting addEventListener/once.
|
||||
function fakeContainer() {
|
||||
@@ -29,172 +31,183 @@ function fakeContainer() {
|
||||
const fns = (listeners.controllerchange || []).slice();
|
||||
for (const { fn, once } of fns) {
|
||||
fn();
|
||||
if (once) {
|
||||
listeners.controllerchange = listeners.controllerchange.filter((l) => l.fn !== fn);
|
||||
}
|
||||
if (once) listeners.controllerchange = listeners.controllerchange.filter((l) => l.fn !== fn);
|
||||
}
|
||||
},
|
||||
};
|
||||
}
|
||||
|
||||
test('messages the waiting worker to skipWaiting and reloads only after controllerchange', async () => {
|
||||
const messages = [];
|
||||
const waiting = { postMessage: (m) => messages.push(m) };
|
||||
const container = fakeContainer();
|
||||
let reloadCount = 0;
|
||||
|
||||
const done = applyUpdate({
|
||||
reg: { waiting },
|
||||
container,
|
||||
reload: () => { reloadCount++; },
|
||||
setTimeout: () => {}, // no-op — we drive controllerchange manually
|
||||
});
|
||||
|
||||
// SKIP_WAITING should be sent immediately, before any reload.
|
||||
await Promise.resolve();
|
||||
assert.deepStrictEqual(messages, [{ type: 'SKIP_WAITING' }]);
|
||||
assert.strictEqual(reloadCount, 0, 'must not reload before the new SW has taken control');
|
||||
|
||||
container.fireControllerChange();
|
||||
await done;
|
||||
assert.strictEqual(reloadCount, 1, 'reloads exactly once after controllerchange');
|
||||
});
|
||||
|
||||
test('reloads only once even if controllerchange fires more than once (no reload loop)', async () => {
|
||||
const waiting = { postMessage: () => {} };
|
||||
const container = fakeContainer();
|
||||
let reloadCount = 0;
|
||||
|
||||
const done = applyUpdate({
|
||||
reg: { waiting },
|
||||
container,
|
||||
reload: () => { reloadCount++; },
|
||||
setTimeout: () => {},
|
||||
});
|
||||
|
||||
container.fireControllerChange();
|
||||
container.fireControllerChange(); // simulate a spurious second event
|
||||
await done;
|
||||
|
||||
assert.strictEqual(reloadCount, 1, 'reload must be idempotent — no loop');
|
||||
});
|
||||
|
||||
test('falls back to a plain reload when there is no waiting worker', async () => {
|
||||
const container = fakeContainer();
|
||||
let reloadCount = 0;
|
||||
|
||||
await applyUpdate({
|
||||
reg: { waiting: null },
|
||||
container,
|
||||
reload: () => { reloadCount++; },
|
||||
setTimeout: () => { throw new Error('timeout should not be scheduled without a waiting worker'); },
|
||||
});
|
||||
|
||||
assert.strictEqual(reloadCount, 1);
|
||||
});
|
||||
|
||||
test('with no waiting worker, fetches the SW update and activates the newly installed worker (buildTag-poll path)', async () => {
|
||||
// Simulates: server redeployed (banner shown by the /api/version poll) but
|
||||
// the browser hasn't fetched the new sw.js yet — reg.waiting is null until
|
||||
// reg.update() is called and the new worker finishes installing.
|
||||
const messages = [];
|
||||
const container = fakeContainer();
|
||||
let reloadCount = 0;
|
||||
|
||||
const stateListeners = [];
|
||||
const installing = {
|
||||
state: 'installing',
|
||||
addEventListener: (type, fn) => { if (type === 'statechange') stateListeners.push(fn); },
|
||||
postMessage: (m) => messages.push(m),
|
||||
};
|
||||
const reg = {
|
||||
waiting: null,
|
||||
installing: null,
|
||||
update() {
|
||||
// Browser found a byte-different sw.js → a new worker starts installing.
|
||||
this.installing = installing;
|
||||
return Promise.resolve();
|
||||
// Fake CacheStorage: { name: { url: body } }.
|
||||
function fakeCaches(initial) {
|
||||
const store = new Map(Object.entries(initial).map(([k, v]) => [k, new Map(Object.entries(v))]));
|
||||
return {
|
||||
store,
|
||||
async keys() { return [...store.keys()]; },
|
||||
async open(name) {
|
||||
if (!store.has(name)) store.set(name, new Map());
|
||||
const m = store.get(name);
|
||||
return {
|
||||
async keys() { return [...m.keys()].map((url) => ({ url })); },
|
||||
async put(url, res) { m.set(url, res.body); },
|
||||
async delete(req) { return m.delete(req.url); },
|
||||
};
|
||||
},
|
||||
};
|
||||
}
|
||||
|
||||
const done = applyUpdate({
|
||||
reg,
|
||||
container,
|
||||
reload: () => { reloadCount++; },
|
||||
setTimeout: () => {}, // no-op — we drive state transitions manually
|
||||
const res = (body, status = 200) => ({ ok: status >= 200 && status < 300, status, body, clone() { return res(body, status); } });
|
||||
const isShellCache = (n) => /^ytplayer-(?!thumbs$|fonts$)/.test(n);
|
||||
|
||||
test('refreshShellInPlace replaces every cached shell file in every shell cache, cache-busted', async () => {
|
||||
const cachesApi = fakeCaches({
|
||||
'ytplayer-old': { [`${ORIGIN}/`]: 'old-index', [`${ORIGIN}/app.js`]: 'old-app' },
|
||||
'ytplayer-new': { [`${ORIGIN}/app.js`]: 'old-app', [`${ORIGIN}/video-edit.js`]: 'old-edit' },
|
||||
'ytplayer-thumbs': { [`${ORIGIN}/thumb.jpg`]: 'thumb' },
|
||||
});
|
||||
|
||||
// Let applyUpdate reach the waitForInstalled stage, then finish the install.
|
||||
await Promise.resolve(); await Promise.resolve();
|
||||
installing.state = 'installed';
|
||||
reg.waiting = installing;
|
||||
stateListeners.forEach((fn) => fn());
|
||||
await Promise.resolve(); await Promise.resolve();
|
||||
|
||||
assert.deepStrictEqual(messages, [{ type: 'SKIP_WAITING' }], 'skip-waiting sent to the freshly installed worker');
|
||||
assert.strictEqual(reloadCount, 0, 'must not reload before the new SW takes control');
|
||||
|
||||
container.fireControllerChange();
|
||||
await done;
|
||||
assert.strictEqual(reloadCount, 1);
|
||||
const fetched = [];
|
||||
const r = await refreshShellInPlace({
|
||||
cachesApi,
|
||||
isShellCache,
|
||||
bust: 'T1',
|
||||
fetchFn: async (url, init) => {
|
||||
fetched.push({ url, init });
|
||||
return res('NEW ' + new URL(url).pathname);
|
||||
},
|
||||
});
|
||||
assert.strictEqual(r.refreshed, 3);
|
||||
// Busted so an old worker's cache-first handler can't answer from its cache.
|
||||
for (const f of fetched) {
|
||||
assert.strictEqual(new URL(f.url).searchParams.get(BUST_PARAM), 'T1-0');
|
||||
assert.strictEqual(f.init.cache, 'reload');
|
||||
}
|
||||
const old = cachesApi.store.get('ytplayer-old');
|
||||
assert.strictEqual(old.get(`${ORIGIN}/`), 'NEW /');
|
||||
assert.strictEqual(old.get(`${ORIGIN}/app.js`), 'NEW /app.js');
|
||||
assert.strictEqual(old.get(`${ORIGIN}/video-edit.js`), 'NEW /video-edit.js');
|
||||
assert.strictEqual(cachesApi.store.get('ytplayer-new').get(`${ORIGIN}/`), 'NEW /');
|
||||
assert.strictEqual(cachesApi.store.get('ytplayer-thumbs').get(`${ORIGIN}/thumb.jpg`), 'thumb', 'utility caches untouched');
|
||||
});
|
||||
|
||||
test('with no waiting worker and no update found, reloads once after the bounded wait', async () => {
|
||||
const container = fakeContainer();
|
||||
let reloadCount = 0;
|
||||
const timeouts = [];
|
||||
test('refreshShellInPlace is all-or-nothing: one failed download writes nothing and throws', async () => {
|
||||
const cachesApi = fakeCaches({
|
||||
'ytplayer-old': { [`${ORIGIN}/`]: 'old-index', [`${ORIGIN}/app.js`]: 'old-app', [`${ORIGIN}/styles.css`]: 'old-css' },
|
||||
});
|
||||
await assert.rejects(refreshShellInPlace({
|
||||
cachesApi,
|
||||
isShellCache,
|
||||
fetchFn: async (url) => (url.includes('/styles.css') ? res('gateway timeout', 504) : res('NEW')),
|
||||
}), /styles\.css/, 'fails after its retries are used up');
|
||||
const old = cachesApi.store.get('ytplayer-old');
|
||||
assert.deepStrictEqual([...old.values()], ['old-index', 'old-app', 'old-css'], 'no partial shell');
|
||||
});
|
||||
|
||||
const reg = {
|
||||
waiting: null,
|
||||
installing: null,
|
||||
update: () => Promise.resolve(), // update check ran; nothing new
|
||||
test('refreshShellInPlace retries a dropped download with a fresh cache-bust token', async () => {
|
||||
const cachesApi = fakeCaches({ 'ytplayer-old': { [`${ORIGIN}/app.js`]: 'old-app', [`${ORIGIN}/`]: 'old-index' } });
|
||||
const tries = [];
|
||||
await refreshShellInPlace({
|
||||
cachesApi, isShellCache, bust: 'T3',
|
||||
fetchFn: async (url) => {
|
||||
tries.push(url);
|
||||
if (url.includes('/app.js') && tries.filter((t) => t.includes('/app.js')).length === 1) throw new TypeError('network error');
|
||||
return res('NEW');
|
||||
},
|
||||
});
|
||||
const appTries = tries.filter((t) => t.includes('/app.js')).map((t) => new URL(t).searchParams.get(BUST_PARAM));
|
||||
assert.deepStrictEqual(appTries, ['T3-0', 'T3-1']);
|
||||
assert.strictEqual(cachesApi.store.get('ytplayer-old').get(`${ORIGIN}/app.js`), 'NEW');
|
||||
});
|
||||
|
||||
test('refreshShellInPlace drops busted URLs an old cache-first worker stored on the way through', async () => {
|
||||
const cachesApi = fakeCaches({ 'ytplayer-old': { [`${ORIGIN}/app.js`]: 'old-app' } });
|
||||
const fetchFn = async (url) => {
|
||||
// Simulate the old worker's cacheFirst caching the busted request.
|
||||
(await cachesApi.open('ytplayer-old')).put(url, res('NEW'));
|
||||
return res('NEW');
|
||||
};
|
||||
|
||||
const done = applyUpdate({
|
||||
reg,
|
||||
container,
|
||||
reload: () => { reloadCount++; },
|
||||
setTimeout: (fn) => { timeouts.push(fn); },
|
||||
});
|
||||
|
||||
await Promise.resolve(); await Promise.resolve();
|
||||
// reg.installing is null → waitForInstalled resolves immediately with null.
|
||||
await done;
|
||||
assert.strictEqual(reloadCount, 1, 'plain reload when the update check finds nothing');
|
||||
await refreshShellInPlace({ cachesApi, isShellCache, fetchFn, bust: 'T2' });
|
||||
assert.deepStrictEqual([...cachesApi.store.get('ytplayer-old').keys()], [`${ORIGIN}/app.js`]);
|
||||
});
|
||||
|
||||
test('falls back to a plain reload when there is no registration at all', async () => {
|
||||
const container = fakeContainer();
|
||||
let reloadCount = 0;
|
||||
|
||||
await applyUpdate({
|
||||
reg: null,
|
||||
container,
|
||||
reload: () => { reloadCount++; },
|
||||
test('refreshShellInPlace times out instead of hanging on a stalled request', async () => {
|
||||
const cachesApi = fakeCaches({ 'ytplayer-old': { [`${ORIGIN}/app.js`]: 'old-app' } });
|
||||
let fire = null;
|
||||
const p = refreshShellInPlace({
|
||||
cachesApi, isShellCache,
|
||||
fetchFn: () => new Promise(() => {}), // never settles
|
||||
timeoutMs: 1000,
|
||||
setTimeout: (fn, ms) => { if (ms === 1000) fire = fn; },
|
||||
});
|
||||
await new Promise((r) => setImmediate(r));
|
||||
fire();
|
||||
await assert.rejects(p, /timed out/);
|
||||
assert.strictEqual(cachesApi.store.get('ytplayer-old').get(`${ORIGIN}/app.js`), 'old-app');
|
||||
});
|
||||
|
||||
assert.strictEqual(reloadCount, 1);
|
||||
test('refreshShellInPlace with no shell caches (uncontrolled page) is a no-op', async () => {
|
||||
const r = await refreshShellInPlace({ cachesApi: fakeCaches({}), isShellCache, fetchFn: async () => { throw new Error('no fetch expected'); } });
|
||||
assert.deepStrictEqual(r, { refreshed: 0, caches: 0 });
|
||||
});
|
||||
|
||||
test('applyUpdate refreshes first, then activates the waiting worker, then reloads once on controllerchange', async () => {
|
||||
const order = [];
|
||||
const waiting = { postMessage: (m) => order.push('msg:' + m.type) };
|
||||
const container = fakeContainer();
|
||||
let reloads = 0;
|
||||
await applyUpdate({
|
||||
reg: { waiting },
|
||||
container,
|
||||
reload: () => { reloads++; order.push('reload'); },
|
||||
refreshShell: async () => { order.push('refresh'); },
|
||||
setTimeout: () => {},
|
||||
});
|
||||
assert.deepStrictEqual(order, ['refresh', 'msg:SKIP_WAITING']);
|
||||
assert.strictEqual(reloads, 0, 'must not reload before the new worker has taken control');
|
||||
container.fireControllerChange();
|
||||
container.fireControllerChange();
|
||||
assert.strictEqual(reloads, 1, 'reloads exactly once');
|
||||
});
|
||||
|
||||
test('applyUpdate with no waiting worker reloads right after the refresh', async () => {
|
||||
const order = [];
|
||||
await applyUpdate({
|
||||
reg: { waiting: null },
|
||||
container: fakeContainer(),
|
||||
reload: () => order.push('reload'),
|
||||
refreshShell: async () => { order.push('refresh'); },
|
||||
});
|
||||
assert.deepStrictEqual(order, ['refresh', 'reload']);
|
||||
});
|
||||
|
||||
test('applyUpdate does NOT reload or activate anything when the refresh fails', async () => {
|
||||
const messages = [];
|
||||
let reloads = 0;
|
||||
await assert.rejects(applyUpdate({
|
||||
reg: { waiting: { postMessage: (m) => messages.push(m) } },
|
||||
container: fakeContainer(),
|
||||
reload: () => { reloads++; },
|
||||
refreshShell: async () => { throw new Error('/app.js → 504'); },
|
||||
}), /504/);
|
||||
assert.strictEqual(reloads, 0, 'reloading now would just serve the old shell again (the loop)');
|
||||
assert.deepStrictEqual(messages, []);
|
||||
});
|
||||
|
||||
test('the timeout safety net reloads once if controllerchange never fires', async () => {
|
||||
const waiting = { postMessage: () => {} };
|
||||
let fire = null;
|
||||
let reloads = 0;
|
||||
const container = fakeContainer();
|
||||
let reloadCount = 0;
|
||||
let scheduledFn = null;
|
||||
|
||||
await applyUpdate({
|
||||
reg: { waiting },
|
||||
reg: { waiting: { postMessage() {} } },
|
||||
container,
|
||||
reload: () => { reloadCount++; },
|
||||
setTimeout: (fn) => { scheduledFn = fn; }, // capture instead of real timer
|
||||
reload: () => { reloads++; },
|
||||
setTimeout: (fn) => { fire = fn; },
|
||||
});
|
||||
|
||||
assert.strictEqual(reloadCount, 0, 'not reloaded yet — timeout not fired');
|
||||
scheduledFn(); // simulate the timeout elapsing
|
||||
assert.strictEqual(reloadCount, 1);
|
||||
|
||||
// A late controllerchange after the timeout already reloaded must not
|
||||
// trigger a second reload.
|
||||
fire();
|
||||
container.fireControllerChange();
|
||||
assert.strictEqual(reloadCount, 1, 'no double reload once the timeout fallback has fired');
|
||||
assert.strictEqual(reloads, 1);
|
||||
});
|
||||
|
||||
test('applyUpdate with no registration at all still reloads', async () => {
|
||||
let reloads = 0;
|
||||
await applyUpdate({ reg: null, container: fakeContainer(), reload: () => { reloads++; } });
|
||||
assert.strictEqual(reloads, 1);
|
||||
});
|
||||
|
||||
@@ -11,12 +11,12 @@
|
||||
* Auto-update flow:
|
||||
* 1. New SW installs alongside the old one and waits (skipWaiting() is NOT
|
||||
* called automatically — see the install handler below).
|
||||
* 2. Client (app.js registerServiceWorker()) notices the waiting worker
|
||||
* and shows the "Update ready" banner.
|
||||
* 3. User clicks "Reload now" → client (applyUpdate() in sw-update.js)
|
||||
* posts SKIP_WAITING to the waiting worker and waits for it to take
|
||||
* control (controllerchange) before reloading — it does NOT unregister
|
||||
* or wipe caches, so the reload doesn't force a brand-new install.
|
||||
* 2. The client (app.js) shows the "Update available" banner only when the
|
||||
* build it is running differs from the server's /api/version buildTag —
|
||||
* a waiting worker or a broadcast is merely a prompt to re-check.
|
||||
* 3. "Refresh UI" (sw-update.js) re-downloads every shell file into every
|
||||
* versioned shell cache (all-or-nothing), activates a waiting worker if
|
||||
* there is one (SKIP_WAITING), and reloads once.
|
||||
* 4. activate: if an older *versioned shell cache* is found (i.e. this
|
||||
* activation is genuinely replacing a previous deploy, not just the
|
||||
* first-ever install of a freshly (re)registered worker), broadcast
|
||||
@@ -68,14 +68,40 @@ const SHELL = [
|
||||
];
|
||||
|
||||
// ---- Install: pre-cache the app shell ----
|
||||
// skipWaiting() is NOT called here — the page activates a waiting worker
|
||||
// from "Refresh UI" (see sw-update.js).
|
||||
self.addEventListener('install', (e) => {
|
||||
e.waitUntil(
|
||||
caches.open(CACHE).then((cache) => cache.addAll(SHELL))
|
||||
// skipWaiting() is NOT called here — we wait for the client to confirm
|
||||
// before activating, so the update banner can appear first.
|
||||
);
|
||||
e.waitUntil(precacheShell());
|
||||
});
|
||||
|
||||
// Fresh requests (`cache: 'reload'` skips the browser HTTP cache, which could
|
||||
// otherwise hand the new worker yesterday's app.js), a few attempts per file
|
||||
// because the homelab link drops requests, and a failed install deletes its
|
||||
// partial cache — a leftover ytplayer-<tag> cache is exactly what activate
|
||||
// uses to decide "this is a genuine update", and it used to re-open the
|
||||
// update banner on pages that were already current.
|
||||
const PRECACHE_ATTEMPTS = 3;
|
||||
async function precacheShell() {
|
||||
const cache = await caches.open(CACHE);
|
||||
try {
|
||||
await Promise.all(SHELL.map(async (url) => {
|
||||
let lastErr = null;
|
||||
for (let attempt = 0; attempt < PRECACHE_ATTEMPTS; attempt++) {
|
||||
try {
|
||||
const res = await fetch(new Request(url, { cache: 'reload' }));
|
||||
if (!res.ok) throw new Error(url + ' → ' + res.status);
|
||||
await cache.put(url, res);
|
||||
return;
|
||||
} catch (err) { lastErr = err; }
|
||||
}
|
||||
throw lastErr;
|
||||
}));
|
||||
} catch (err) {
|
||||
await caches.delete(CACHE);
|
||||
throw err;
|
||||
}
|
||||
}
|
||||
|
||||
// ---- Activate: evict old caches, claim clients, notify about update ----
|
||||
self.addEventListener('activate', (e) => {
|
||||
e.waitUntil((async () => {
|
||||
|
||||
@@ -27,8 +27,9 @@ const SW_SOURCE = fs.readFileSync(path.join(__dirname, 'sw.js'), 'utf8');
|
||||
|
||||
// Builds a fresh sandboxed SW environment with the given starting cache keys
|
||||
// and returns handles to drive/inspect it.
|
||||
function loadSw(initialCacheKeys) {
|
||||
function loadSw(initialCacheKeys, { fetchImpl } = {}) {
|
||||
const cacheStore = new Set(initialCacheKeys);
|
||||
const puts = [];
|
||||
const deleted = [];
|
||||
const clientMessages = [];
|
||||
|
||||
@@ -50,14 +51,15 @@ function loadSw(initialCacheKeys) {
|
||||
async delete(key) { deleted.push(key); return cacheStore.delete(key); },
|
||||
async open(key) {
|
||||
cacheStore.add(key);
|
||||
return { addAll: async () => {}, match: async () => undefined, put: async () => {} };
|
||||
return { addAll: async () => {}, match: async () => undefined, put: async (url) => { puts.push(url); } };
|
||||
},
|
||||
};
|
||||
|
||||
const sandbox = {
|
||||
self: self_,
|
||||
caches: caches_,
|
||||
fetch: async () => { throw new Error('fetch not mocked'); },
|
||||
fetch: fetchImpl || (async () => { throw new Error('fetch not mocked'); }),
|
||||
Request: class { constructor(url, init) { this.url = url; Object.assign(this, init); } },
|
||||
Response: class { constructor(body, init) { this.body = body; Object.assign(this, init); } },
|
||||
URL,
|
||||
console,
|
||||
@@ -72,7 +74,14 @@ function loadSw(initialCacheKeys) {
|
||||
await waitPromise;
|
||||
}
|
||||
|
||||
return { triggerActivate, deleted, clientMessages, cacheStoreRemaining: () => Array.from(cacheStore) };
|
||||
async function triggerInstall() {
|
||||
let waitPromise = Promise.resolve();
|
||||
const event = { waitUntil: (p) => { waitPromise = p; } };
|
||||
for (const fn of listeners.install || []) fn(event);
|
||||
await waitPromise;
|
||||
}
|
||||
|
||||
return { triggerActivate, triggerInstall, deleted, puts, clientMessages, cacheStoreRemaining: () => Array.from(cacheStore) };
|
||||
}
|
||||
|
||||
test('activate does NOT broadcast on a fresh install (only the current-version cache exists)', async () => {
|
||||
@@ -115,3 +124,29 @@ test('activate reports an update and still preserves utility caches together', a
|
||||
['ytplayer-fonts', 'ytplayer-thumbs', 'ytplayer-v1.0.4'].sort()
|
||||
);
|
||||
});
|
||||
|
||||
test('install fetches every shell file past the HTTP cache and retries dropped requests', async () => {
|
||||
const seen = {};
|
||||
const sw = loadSw([], {
|
||||
fetchImpl: async (req) => {
|
||||
seen[req.url] = (seen[req.url] || 0) + 1;
|
||||
assert.strictEqual(req.cache, 'reload', 'must bypass the browser HTTP cache');
|
||||
// The homelab link drops the first request for app.js.
|
||||
if (req.url === '/app.js' && seen[req.url] === 1) throw new TypeError('network error');
|
||||
return { ok: true, status: 200 };
|
||||
},
|
||||
});
|
||||
await sw.triggerInstall();
|
||||
assert.strictEqual(seen['/app.js'], 2, 'a dropped request is retried');
|
||||
assert.ok(sw.puts.includes('/app.js') && sw.puts.includes('/'), 'shell files cached');
|
||||
assert.deepStrictEqual(sw.deleted, []);
|
||||
});
|
||||
|
||||
test('a failed install deletes its partial cache so activate never mistakes it for an old deploy', async () => {
|
||||
const sw = loadSw([], {
|
||||
fetchImpl: async (req) => (req.url === '/styles.css' ? { ok: false, status: 504 } : { ok: true, status: 200 }),
|
||||
});
|
||||
await assert.rejects(sw.triggerInstall(), /styles\.css/);
|
||||
assert.deepStrictEqual(sw.deleted, ['ytplayer-v1.0.4'], 'partial cache removed');
|
||||
assert.deepStrictEqual(sw.cacheStoreRemaining(), []);
|
||||
});
|
||||
|
||||
Reference in New Issue
Block a user