implement config option to disable ratelimit

This commit is contained in:
derrickmehaffy 2022-12-16 10:42:28 -07:00
parent 68a4ec7729
commit b75e8a1d10

View File

@ -8,6 +8,15 @@ const { RateLimitError } = utils.errors;
module.exports = module.exports =
(config, { strapi }) => (config, { strapi }) =>
async (ctx, next) => { async (ctx, next) => {
let ratelimitConfig = strapi.config.get('admin.ratelimit');
if (!ratelimitConfig || !ratelimitConfig.enabled) {
ratelimitConfig = {
enabled: true,
};
}
if (ratelimitConfig.enabled === true) {
const ratelimit = require('koa2-ratelimit').RateLimit; const ratelimit = require('koa2-ratelimit').RateLimit;
const userEmail = toLower(ctx.request.body.email) || 'unknownEmail'; const userEmail = toLower(ctx.request.body.email) || 'unknownEmail';
@ -19,7 +28,10 @@ module.exports =
handler() { handler() {
throw new RateLimitError(); throw new RateLimitError();
}, },
...strapi.config.get('admin.ratelimit'), ...ratelimitConfig,
...config, ...config,
})(ctx, next); })(ctx, next);
}
return next();
}; };