mirror of https://github.com/tycrek/ass
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
46 lines
1.7 KiB
46 lines
1.7 KiB
import { EndpointRateLimitConfiguration } from 'ass';
|
|
import { NextFunction, Request, Response } from 'express';
|
|
import { rateLimit } from 'express-rate-limit';
|
|
|
|
/**
|
|
* map that contains rate limiter middleware for each group
|
|
*/
|
|
const rateLimiterGroups = new Map<string, (req: Request, res: Response, next: NextFunction) => void>();
|
|
|
|
export const setRateLimiter = (group: string, config: EndpointRateLimitConfiguration | undefined): (req: Request, res: Response, next: NextFunction) => void => {
|
|
if (config == null) { // config might be null if the user doesnt want a rate limit
|
|
rateLimiterGroups.set(group, (req, res, next) => {
|
|
next();
|
|
});
|
|
|
|
return rateLimiterGroups.get(group)!;
|
|
} else {
|
|
rateLimiterGroups.set(group, rateLimit({
|
|
limit: config.requests,
|
|
windowMs: config.duration * 1000,
|
|
skipFailedRequests: true,
|
|
legacyHeaders: false,
|
|
standardHeaders: 'draft-7',
|
|
keyGenerator: (req, res) => {
|
|
return req.ip || 'disconnected';
|
|
},
|
|
handler: (req, res) => {
|
|
res.status(429);
|
|
res.contentType('json');
|
|
res.send('{"success":false,"message":"Rate limit exceeded, try again later"}');
|
|
}
|
|
}));
|
|
|
|
return rateLimiterGroups.get(group)!;
|
|
}
|
|
}
|
|
/**
|
|
* creates middleware for rate limiting
|
|
*/
|
|
export const rateLimiterMiddleware = (group: string, config: EndpointRateLimitConfiguration | undefined): (req: Request, res: Response, next: NextFunction) => void => {
|
|
if (!rateLimiterGroups.has(group)) setRateLimiter(group, config);
|
|
|
|
return (req, res, next) => {
|
|
return rateLimiterGroups.get(group)!(req, res, next);
|
|
};
|
|
}; |