Merge pull request #3481 from SillyTavern/cache-buster Add cache buster middleware to clear browser cache on server restart

fc488bd4fe6b5ae862e3f84cff0ae48429d65249

Cohee <18619528+Cohee1207@users.noreply.github.com>

Signed

server.js+2 -1 Viewed

 import whitelistMiddleware from './src/middleware/whitelist.js';
 import multerMonkeyPatch from './src/middleware/multerMonkeyPatch.js';
 import initRequestProxy from './src/request-proxy.js';
+import getCacheBusterMiddleware from './src/middleware/cacheBuster.js';
 import {
     getVersion,
     getConfigValue,

 // Static files
 // Host index page
-app.get('/', (request, response) => {
+app.get('/', getCacheBusterMiddleware(), (request, response) => {
     if (shouldRedirectToLogin(request)) {
         const query = request.url.split('?')[1];
         const redirectUrl = query ? `/login?${query}` : '/login';

src/middleware/cacheBuster.js+28 -0 Viewed

+import crypto from 'node:crypto';
+import { DEFAULT_USER } from '../constants.js';
+/**
+ * Middleware to bust the browser cache for the current user.
+ * @returns {import('express').RequestHandler}
+ */
+export default function getCacheBusterMiddleware() {
+    /**
+     * @type {Set<string>} Handles/User-Agents that have already been busted.
+     */
+    const keys = new Set();
+    return (request, response, next) => {
+        const handle = request.user?.profile?.handle || DEFAULT_USER.handle;
+        const userAgent = request.headers['user-agent'] || '';
+        const hash = crypto.createHash('sha256').update(userAgent).digest('hex');
+        const key = `${handle}-${hash}`;
+        if (keys.has(key)) {
+            return next();
+        }
+        keys.add(key);
+        response.setHeader('Clear-Site-Data', '"cache"');
+        next();
+    };
+}