rixty/SillyTavern
1
0
Fork 0
mirror of https://github.com/SillyTavern/SillyTavern.git synced 2025-06-05 21:59:27 +02:00
Code Issues Packages Projects Releases Wiki Activity

Refactor whitelist middleware to return a promise and update server.js to handle async initialization

Browse Source
This commit is contained in:
Cohee
2025-02-28 00:25:59 +02:00
parent 58fe54954a
commit 978b2cdb21
2 changed files with 7 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
src/middleware/whitelist.js
View File

@ -22,8 +22,6 @@ if (fs.existsSync(whitelistPath)) {
}
}
await resolveHostnames();
/**
* Get the client IP address from the request headers.
* @param {import('express').Request} req Express request object
@ -110,9 +108,9 @@ async function resolveHostnames() {
/**
* Returns a middleware function that checks if the client IP is in the whitelist.
* @returns {import('express').RequestHandler} The middleware function
* @returns {Promise<import('express').RequestHandler>} Promise that resolves to the middleware function
*/
export default function whitelistMiddleware() {
export default async function getWhitelistMiddleware() {
const forbiddenWebpage = Handlebars.compile(
safeReadFileSync('./public/error/forbidden-by-whitelist.html') ?? '',
);
@ -121,6 +119,8 @@ export default function whitelistMiddleware() {
'/favicon.ico',
];
await resolveHostnames();
return function (req, res, next) {
const clientIp = getIpFromRequest(req);
const forwardedIp = getForwardedIp(req);