2021-10-23 11:35:13 +03:00
|
|
|
const { RateLimiter } = require("limiter");
|
2022-04-13 18:33:37 +03:00
|
|
|
const { log } = require("../src/util");
|
2021-10-23 11:35:13 +03:00
|
|
|
|
|
|
|
class KumaRateLimiter {
|
|
|
|
constructor(config) {
|
|
|
|
this.errorMessage = config.errorMessage;
|
|
|
|
this.rateLimiter = new RateLimiter(config);
|
|
|
|
}
|
|
|
|
|
|
|
|
async pass(callback, num = 1) {
|
|
|
|
const remainingRequests = await this.removeTokens(num);
|
2022-04-13 18:33:37 +03:00
|
|
|
log.info("rate-limit", "remaining requests: " + remainingRequests);
|
2021-10-23 11:35:13 +03:00
|
|
|
if (remainingRequests < 0) {
|
|
|
|
if (callback) {
|
|
|
|
callback({
|
|
|
|
ok: false,
|
|
|
|
msg: this.errorMessage,
|
|
|
|
});
|
|
|
|
}
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
|
|
|
async removeTokens(num = 1) {
|
|
|
|
return await this.rateLimiter.removeTokens(num);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
const loginRateLimiter = new KumaRateLimiter({
|
|
|
|
tokensPerInterval: 20,
|
|
|
|
interval: "minute",
|
|
|
|
fireImmediately: true,
|
|
|
|
errorMessage: "Too frequently, try again later."
|
|
|
|
});
|
|
|
|
|
2022-03-29 12:38:48 +03:00
|
|
|
const twoFaRateLimiter = new KumaRateLimiter({
|
|
|
|
tokensPerInterval: 30,
|
|
|
|
interval: "minute",
|
|
|
|
fireImmediately: true,
|
|
|
|
errorMessage: "Too frequently, try again later."
|
|
|
|
});
|
|
|
|
|
2021-10-23 11:35:13 +03:00
|
|
|
module.exports = {
|
2022-03-29 12:38:48 +03:00
|
|
|
loginRateLimiter,
|
|
|
|
twoFaRateLimiter,
|
2021-10-23 11:35:13 +03:00
|
|
|
};
|