const ratelimit = {}; const parallelLimit = {}; const { ownerId } = process.env; import db from '../models/index.js'; export default { check, addParallel, removeParallel, checkParallel, }; function check(user, commandName, commands) { const userID = user.id; const userTag = user.username; // Don't apply the rate limit to bot owner if (userID === 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 = db.optout.findOne({ where: { userID: userID } }); if (isOptOut) { console.log(`A user is rate limited on \x1b[33m${commandName}\x1b[0m for${dateString}.`); } else { console.log(`\x1b[33m${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; } 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; } 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)}`); } function checkParallel(user, commandName, command) { // Don't apply the rate limit to bot owner if (user.id === ownerId) return false; // console.log(`[CHECK] command limit: ${command.parallelLimit}`); // console.log(`[CHECK] current parallel executions: ${parallelLimit[commandName]}`); if (parallelLimit[commandName] >= command.parallelLimit) { return 'There are currently too many parallel execution of this command, please wait before retrying.'; } return false; }