forked from Supositware/Haha-Yes
103 lines
No EOL
3.6 KiB
JavaScript
103 lines
No EOL
3.6 KiB
JavaScript
const ratelimit = {};
|
|
const parallelLimit = {};
|
|
const { ownerId, NODE_ENV } = process.env;
|
|
|
|
import db from '../models/index.js';
|
|
|
|
export default {
|
|
check,
|
|
addParallel,
|
|
removeParallel,
|
|
checkParallel,
|
|
};
|
|
async function check(user, commandName, commands) {
|
|
const userID = user.id;
|
|
const userTag = user.username;
|
|
|
|
// Don't apply the rate limit to bot owner
|
|
if (NODE_ENV !== 'development') {
|
|
if (user.id === ownerId) {
|
|
return false;
|
|
}
|
|
}
|
|
|
|
if (!ratelimit[userID]) {
|
|
ratelimit[userID] = {};
|
|
}
|
|
|
|
const date = new Date();
|
|
if (ratelimit[userID][commandName]) {
|
|
if (ratelimit[userID][commandName].cooldown && ratelimit[userID][commandName].limit === commands.ratelimit) {
|
|
if (date > ratelimit[userID][commandName].cooldown) {
|
|
ratelimit[userID][commandName].limit = 0;
|
|
ratelimit[userID][commandName].cooldown = undefined;
|
|
}
|
|
}
|
|
|
|
if (commands.ratelimit === ratelimit[userID][commandName].limit) {
|
|
const seconds = Math.floor((ratelimit[userID][commandName].cooldown - date) / 1000);
|
|
const minutes = Math.floor(seconds / 60);
|
|
const hours = Math.floor(minutes / 60);
|
|
const dateString = `${hours > 0 ? ` ${Math.floor(hours)} hours` : ''}${minutes > 0 ? ` ${Math.floor(minutes % 60)} minutes` : ''}${seconds > 0 ? ` ${Math.floor(seconds % 60)} seconds` : ''}`;
|
|
|
|
const isOptOut = await db.optout.findOne({ where: { userID: userID } });
|
|
|
|
const timestamp = new Date();
|
|
console.log(`[${timestamp.toISOString()}] \x1b[33m${ isOptOut ? 'A user' : `${userTag} (${userID})`}\x1b[0m is rate limited on \x1b[33m${commandName}\x1b[0m for${dateString}.`);
|
|
|
|
return `You are being rate limited. You can try again in${dateString}.`;
|
|
}
|
|
}
|
|
|
|
|
|
if (commands.ratelimit) {
|
|
date.setSeconds(date.getSeconds() + commands.cooldown);
|
|
ratelimit[userID][commandName] = { limit: ratelimit[userID][commandName] ? ratelimit[userID][commandName].limit + 1 : 1, cooldown: date };
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
async function addParallel(commandName) {
|
|
// console.log(`[ADD] Adding parallel to ${commandName}`);
|
|
if (!parallelLimit[commandName]) parallelLimit[commandName] = 0;
|
|
|
|
const prevNumber = parallelLimit[commandName];
|
|
|
|
// console.log(`[ADD] Previous parallel executions: ${prevNumber}`);
|
|
// console.log(`[ADD] Current parallel executions: ${JSON.stringify(parallelLimit)}`);
|
|
parallelLimit[commandName] = prevNumber + 1;
|
|
}
|
|
|
|
async function removeParallel(commandName) {
|
|
// console.log(`[REMOVE] Removing parallel to ${commandName}`);
|
|
|
|
// This shouldn't be possible
|
|
if (!parallelLimit[commandName]) parallelLimit[commandName] = 0;
|
|
|
|
const prevNumber = parallelLimit[commandName];
|
|
|
|
// console.log(`[REMOVE] previous number: ${prevNumber}`);
|
|
// console.log(`[REMOVE] previous parallel limit: ${JSON.stringify(parallelLimit)}`);
|
|
parallelLimit[commandName] = prevNumber - 1;
|
|
// console.log(`[REMOVE] current parallel limit: ${JSON.stringify(parallelLimit)}`);
|
|
}
|
|
|
|
async function checkParallel(user, commandName, command) {
|
|
// Don't apply the rate limit to bot owner
|
|
if (NODE_ENV !== 'development') {
|
|
if (user.id === ownerId) {
|
|
return false;
|
|
}
|
|
}
|
|
|
|
if (!parallelLimit[commandName]) parallelLimit[commandName] = 0;
|
|
|
|
// console.log(`[CHECK] command limit: ${command.parallelLimit}`);
|
|
// console.log(`[CHECK] current parallel executions: ${parallelLimit[commandName]}`);
|
|
if (parallelLimit[commandName] >= command.parallelLimit) {
|
|
return { limited: true, current: parallelLimit[commandName], max: command.parallelLimit, msg: `There are currently too many parallel execution of this command, please wait before retrying. (${parallelLimit[commandName]}/${command.parallelLimit})` };
|
|
}
|
|
|
|
return { limited: false, current: parallelLimit[commandName], max: command.parallelLimit };
|
|
} |