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('/', 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();
 +    };
 +}