smyalygames/arabot
1
0
Fork 0
mirror of https://github.com/veganhacktivists/arabot.git synced 2025-05-19 12:34:14 +02:00
Code Issues Actions Packages Projects Releases Wiki Activity

feat(arabot): add selecting a restricted section to database

Browse Source
This commit is contained in:
smyalygames 2023-02-14 01:08:10 +00:00
parent 9e833b87fd
commit e544fa2eae
1 changed files with 29 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

47
src/commands/mod/restriction/restrict.ts
View File

@ -37,7 +37,13 @@ import type {
Snowflake, Snowflake,
} from 'discord.js'; } from 'discord.js';
import IDs from '#utils/ids'; import IDs from '#utils/ids';
import { addEmptyUser, updateUser, userExists } from '#utils/database/dbExistingUser'; import { randint } from '#utils/random';
import {
addEmptyUser,
updateUser,
userExists,
fetchRoles,
} from '#utils/database/dbExistingUser';
import { restrict, checkActive } from '#utils/database/restriction'; import { restrict, checkActive } from '#utils/database/restriction';
export async function restrictRun( export async function restrictRun(
@ -76,14 +82,19 @@ export async function restrictRun(
member = await guild.members.fetch(userId); member = await guild.members.fetch(userId);
} }
const restrictRoles = [
IDs.roles.restrictions.restricted1,
IDs.roles.restrictions.restricted2,
IDs.roles.restrictions.restricted3,
IDs.roles.restrictions.restricted4,
IDs.roles.restrictions.restricted1, // Vegan Restrict
];
let section = tolerance ? randint(3, 4) : randint(1, 2);
if (member !== undefined) { if (member !== undefined) {
// Checks if the user is not restricted // Checks if the user is not restricted
if (member.roles.cache.hasAny( if (member.roles.cache.hasAny(...restrictRoles)) {
IDs.roles.restrictions.restricted1,
IDs.roles.restrictions.restricted2,
IDs.roles.restrictions.restricted3,
IDs.roles.restrictions.restricted4,
)) {
info.message = `${member} is already restricted!`; info.message = `${member} is already restricted!`;
return info; return info;
} }
@ -92,8 +103,12 @@ export async function restrictRun(
await updateUser(member); await updateUser(member);
if (member.roles.cache.has(IDs.roles.vegan.vegan)) { if (member.roles.cache.has(IDs.roles.vegan.vegan)) {
await member.roles.add(IDs.roles.restrictions.restricted1); section = 5;
}
await member.roles.add(restrictRoles[section - 1]);
if (member.roles.cache.has(IDs.roles.vegan.vegan)) {
const voiceChannel = await guild.channels.create({ const voiceChannel = await guild.channels.create({
name: 'Restricted Voice Channel', name: 'Restricted Voice Channel',
type: ChannelType.GuildVoice, type: ChannelType.GuildVoice,
@ -188,14 +203,6 @@ export async function restrictRun(
); );
await restrictedChannel.send({ embeds: [embed] }); await restrictedChannel.send({ embeds: [embed] });
} else if (tolerance) {
await member.roles.add(Math.random() > 0.5
? IDs.roles.restrictions.restricted3
: IDs.roles.restrictions.restricted4);
} else {
await member.roles.add(Math.random() > 0.5
? IDs.roles.restrictions.restricted1
: IDs.roles.restrictions.restricted2);
} }
await member.roles.remove([ await member.roles.remove([
@ -209,11 +216,16 @@ export async function restrictRun(
]); ]);
} else if (!await userExists(userId)) { } else if (!await userExists(userId)) {
await addEmptyUser(userId); await addEmptyUser(userId);
const dbRoles = await fetchRoles(userId);
if (dbRoles.includes(IDs.roles.vegan.vegan)) {
section = 5;
}
} }
// Restrict the user on the database // Restrict the user on the database
await restrict(userId, modId, reason); await restrict(userId, modId, reason, section);
info.message = `Restricted ${member}`;
info.success = true; info.success = true;
// Log the ban // Log the ban
@ -243,7 +255,6 @@ export async function restrictRun(
await logChannel.send({ embeds: [message] }); await logChannel.send({ embeds: [message] });
info.message = `Restricted ${member}`;
return info; return info;
} }