wanhose/cookie-dialog-monster
wanhose/cookie-dialog-monster
2
5
Fork 0
Code Issues 304 Pull Requests Releases 18 Wiki Activity

fix(api): rate limit bug

Browse Source
This commit is contained in:
wanhose 2024-11-19 12:19:21 +01:00
parent d16f86cd32
commit 6434dd6c1c
2 changed files with 10 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
packages/api/src/index.ts
View File

@ -17,6 +17,7 @@ import v6IssuesRoutes from 'routes/v6/issues';
import v6ReportRoutes from 'routes/v6/report'; import v6ReportRoutes from 'routes/v6/report';
import v6VersionRoutes from 'routes/v6/version'; import v6VersionRoutes from 'routes/v6/version';
import environment from 'services/environment'; import environment from 'services/environment';
import { keyGenerator } from 'services/rateLimit';
const server = fastify({ logger: true }); const server = fastify({ logger: true });
@ -30,6 +31,7 @@ server.register(cors, {
server.register(rateLimit, { server.register(rateLimit, {
global: false, global: false,
keyGenerator,
}); });
server.register(v1EntriesRoutes, { prefix: '/rest/v1' }); server.register(v1EntriesRoutes, { prefix: '/rest/v1' });

8
packages/api/src/services/rateLimit.ts
View File

@ -1,3 +1,5 @@
import type { FastifyRequest } from 'fastify';
export const RATE_LIMIT_1_PER_HOUR = { export const RATE_LIMIT_1_PER_HOUR = {
max: 1, max: 1,
timeWindow: '1 hour', timeWindow: '1 hour',
@ -17,3 +19,9 @@ export const RATE_LIMIT_3_PER_MIN = {
max: 3, max: 3,
timeWindow: '1 minute', timeWindow: '1 minute',
}; };
export function keyGenerator(req: FastifyRequest): string {
const userIdentifier = req.headers['x-forwarded-for'] || req.headers['x-real-ip'] || req.ip;
return `${userIdentifier}:${req.routerPath}`;
}