joan/Haha-Yes
1
0
Fork 0
forked from Supositware/Haha-Yes
Code Pull requests Activity

make ratelimiter async (also actually fix message)

Browse source
This commit is contained in:
Supositware 2024-02-04 01:51:17 +01:00
parent 12ba7621b6
commit e6dca692ed
3 changed files with 19 additions and 17 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
events/client/interactionCreate.js
View file

@ -71,16 +71,16 @@ export default {
// Check if the limit of parallel execution has been reached
if (command.parallelLimit) {
const doParallelLimit = ratelimiter.checkParallel(interaction.user, commandName, command);
const doParallelLimit = await ratelimiter.checkParallel(interaction.user, commandName, command);
if (doParallelLimit.limited) {
return await interaction.reply({ content: doParallelLimit, ephemeral: true });
return await interaction.reply({ content: doParallelLimit.msg, ephemeral: true });
}
ratelimiter.addParallel(commandName);
}
// Check the ratelimit
const doRateLimit = ratelimiter.check(interaction.user, commandName, command);
const doRateLimit = await ratelimiter.check(interaction.user, commandName, command);
if (doRateLimit) {
return interaction.reply({ content: doRateLimit, ephemeral: true });
@ -107,8 +107,8 @@ export default {
}
await command.execute(interaction, args, client)
.then(() => {
const hasPrallelLimit = ratelimiter.checkParallel(interaction.user, commandName, command);
.then(async () => {
const hasPrallelLimit = await ratelimiter.checkParallel(interaction.user, commandName, command);
if (hasPrallelLimit) ratelimiter.removeParallel(commandName);
});
}

10
events/client/messageCreate.js
View file

@ -333,16 +333,16 @@ export default {
// Check if the limit of parallel execution has been reached
if (command.parallelLimit) {
const doParallelLimit = ratelimiter.checkParallel(message.author, commandName, command);
const doParallelLimit = await ratelimiter.checkParallel(message.author, commandName, command);
if (doParallelLimit.limited) {
return await message.reply({ content: doParallelLimit, ephemeral: true });
return await message.reply({ content: doParallelLimit.msg, ephemeral: true });
}
ratelimiter.addParallel(commandName);
}
// Check the ratelimit
const doRateLimit = ratelimiter.check(message.author, commandName, command);
const doRateLimit = await ratelimiter.check(message.author, commandName, command);
if (doRateLimit) {
return message.reply({ content: doRateLimit, ephemeral: true });
@ -432,8 +432,8 @@ export default {
}
await command.execute(message, args, client)
.then(() => {
const hasPrallelLimit = ratelimiter.checkParallel(message.author, commandName, command);
.then(async () => {
const hasPrallelLimit = await ratelimiter.checkParallel(message.author, commandName, command);
if (hasPrallelLimit) ratelimiter.removeParallel(commandName);
});
}

14
utils/ratelimiter.js
View file

@ -10,14 +10,16 @@ export default {
removeParallel,
checkParallel,
};
function check(user, commandName, commands) {
async 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) {
if (NODE_ENV !== 'development') {
if (user.id === ownerId) {
return false;
}
}
if (!ratelimit[userID]) {
ratelimit[userID] = {};
@ -38,7 +40,7 @@ function check(user, commandName, commands) {
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 } });
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}.`);
@ -56,7 +58,7 @@ function check(user, commandName, commands) {
return false;
}
function addParallel(commandName) {
async function addParallel(commandName) {
// console.log(`[ADD] Adding parallel to ${commandName}`);
if (!parallelLimit[commandName]) parallelLimit[commandName] = 0;
@ -67,7 +69,7 @@ function addParallel(commandName) {
parallelLimit[commandName] = prevNumber + 1;
}
function removeParallel(commandName) {
async function removeParallel(commandName) {
// console.log(`[REMOVE] Removing parallel to ${commandName}`);
// This shouldn't be possible
@ -81,7 +83,7 @@ function removeParallel(commandName) {
// console.log(`[REMOVE] current parallel limit: ${JSON.stringify(parallelLimit)}`);
}
function checkParallel(user, commandName, command) {
async function checkParallel(user, commandName, command) {
// Don't apply the rate limit to bot owner
if (NODE_ENV !== 'development') {
if (user.id === ownerId) {