33 lines
1.0 KiB
TypeScript
33 lines
1.0 KiB
TypeScript
|
import { Context, Next } from 'hono';
|
||
|
|
import { rateLimiter } from 'hono-rate-limiter';
|
||
|
|
import settingsCache from '../lib/settings';
|
||
|
|
import { getClientIp } from '../lib/utils';
|
||
|
|
|
||
|
|
let rateLimitInstance: ReturnType<typeof rateLimiter> | null = null;
|
||
|
|
|
||
|
|
const ratelimit = async (c: Context, next: Next) => {
|
||
|
|
const instanceSettings = settingsCache.get('instanceSettings');
|
||
|
|
|
||
|
|
if (instanceSettings?.enableRateLimiting) {
|
||
|
|
if (rateLimitInstance === null) {
|
||
|
|
rateLimitInstance = rateLimiter({
|
||
|
|
windowMs: instanceSettings.rateLimitWindow * 1000, // Convert seconds to milliseconds
|
||
|
|
limit: instanceSettings.rateLimitRequests,
|
||
|
|
standardHeaders: true,
|
||
|
|
keyGenerator: (c) => getClientIp(c) || 'anonymous',
|
||
|
|
});
|
||
|
|
}
|
||
|
|
|
||
|
|
return rateLimitInstance(c, next);
|
||
|
|
}
|
||
|
|
|
||
|
|
// If rate limiting is disabled, ensure the limiter is cleared
|
||
|
|
if (rateLimitInstance !== null) {
|
||
|
|
rateLimitInstance = null;
|
||
|
|
}
|
||
|
|
|
||
|
|
await next();
|
||
|
|
};
|
||
|
|
|
||
|
|
export default ratelimit;
|