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

refactor(arabot): remove unnecessary not nulls

Browse Source
This commit is contained in:
smyalygames 2022-10-28 05:27:51 +01:00
parent 990d0f0dfa
commit baaa52aa14
4 changed files with 65 additions and 60 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

109
src/commands/mod/sus.ts
View File

@ -119,8 +119,8 @@ class SusCommand extends Command {
// Subcommand to add sus note // Subcommand to add sus note
private async addNote(interaction: Command.ChatInputInteraction) { private async addNote(interaction: Command.ChatInputInteraction) {
// Get the arguments // Get the arguments
let user = interaction.options.getUser('user'); const user = interaction.options.getUser('user');
let note = interaction.options.getString('note'); const note = interaction.options.getString('note');
const { guild } = interaction; const { guild } = interaction;
// Checks if all the variables are of the right type // Checks if all the variables are of the right type
@ -133,16 +133,13 @@ class SusCommand extends Command {
return; return;
} }
// Remove possibility of null from variables const mod = interaction.member.user;
user = user!;
const mod = interaction.member!.user;
note = note!;
// Add the data to the database // Add the data to the database
// Check if the user exists on the database // Check if the user exists on the database
const userGuildMember = guild!.members.cache.get(user.id); const userGuildMember = guild.members.cache.get(user.id);
const modGuildMember = guild!.members.cache.get(mod.id); const modGuildMember = guild.members.cache.get(mod.id);
// TODO potentially add a method to add user by Snowflake // TODO potentially add a method to add user by Snowflake
if (userGuildMember === undefined || modGuildMember === undefined) { if (userGuildMember === undefined || modGuildMember === undefined) {
await interaction.reply({ await interaction.reply({
@ -154,17 +151,17 @@ class SusCommand extends Command {
} }
// Check if user and mod are on the database // Check if user and mod are on the database
if (!await userExists(userGuildMember!)) { if (!await userExists(userGuildMember.id)) {
await addExistingUser(userGuildMember!); await addExistingUser(userGuildMember);
} }
if (!await userExists(modGuildMember!)) { if (!await userExists(modGuildMember.id)) {
await addExistingUser(modGuildMember!); await addExistingUser(modGuildMember);
} }
await addToDatabase(user.id, mod.id, note); await addToDatabase(user.id, mod.id, note);
// Give the user the sus role they don't already have the sus note // Give the user the sus role they don't already have the sus note
if (!userGuildMember.roles.cache.has(IDs.roles.restrictions.sus)) { if (!userGuildMember.roles.cache.has(IDs.roles.restrictions.sus)) {
await userGuildMember!.roles.add(IDs.roles.restrictions.sus); await userGuildMember.roles.add(IDs.roles.restrictions.sus);
} }
await interaction.reply({ await interaction.reply({
@ -176,7 +173,7 @@ class SusCommand extends Command {
private async listNote(interaction: Command.ChatInputInteraction) { private async listNote(interaction: Command.ChatInputInteraction) {
// Get the arguments // Get the arguments
let user = interaction.options.getUser('user'); const user = interaction.options.getUser('user');
const { guild } = interaction; const { guild } = interaction;
// Checks if all the variables are of the right type // Checks if all the variables are of the right type
@ -189,9 +186,6 @@ class SusCommand extends Command {
return; return;
} }
// Remove possibility of null from variables
user = user!;
// Gets the sus notes from the database // Gets the sus notes from the database
const notes = await findNotes(user.id, true); const notes = await findNotes(user.id, true);
@ -214,7 +208,7 @@ class SusCommand extends Command {
// Add up to 10 of the latest sus notes to the embed // Add up to 10 of the latest sus notes to the embed
for (let i = notes.length > 10 ? notes.length - 10 : 0; i < notes.length; i += 1) { for (let i = notes.length > 10 ? notes.length - 10 : 0; i < notes.length; i += 1) {
// Get mod name // Get mod name
const modGuildMember = guild!.members.cache.get(notes[i].modId); const modGuildMember = guild.members.cache.get(notes[i].modId);
let mod = notes[i].modId; let mod = notes[i].modId;
if (modGuildMember !== undefined) { if (modGuildMember !== undefined) {
mod = modGuildMember!.displayName; mod = modGuildMember!.displayName;
@ -236,7 +230,7 @@ class SusCommand extends Command {
private async removeNote(interaction: Command.ChatInputInteraction) { private async removeNote(interaction: Command.ChatInputInteraction) {
// Get the arguments // Get the arguments
let noteId = interaction.options.getInteger('id'); const noteId = interaction.options.getInteger('id');
const { guild, channel } = interaction; const { guild, channel } = interaction;
// Checks if all the variables are of the right type // Checks if all the variables are of the right type
@ -249,9 +243,6 @@ class SusCommand extends Command {
return; return;
} }
// Remove possibility of null from variables
noteId = noteId!;
// Get the note to be deleted // Get the note to be deleted
const note = await getNote(noteId); const note = await getNote(noteId);
@ -266,29 +257,39 @@ class SusCommand extends Command {
} }
// Get user GuildMembers for user and mod and person who ran command // Get user GuildMembers for user and mod and person who ran command
const user = await guild!.members.cache.get(note!.userId); const user = await guild.members.cache.get(note.userId);
const mod = await guild!.members.cache.get(note!.modId); const mod = await guild.members.cache.get(note.modId);
// TODO fix if user left the server
if (user === undefined || mod === undefined) {
await interaction.reply({
content: 'Error fetching users from Discord!',
ephemeral: true,
fetchReply: true,
});
return;
}
// Get user's name // Get user's name
let userName = note!.userId; let userName = note.userId;
if (user !== undefined) { if (user !== undefined) {
userName = user!.displayName; userName = user.displayName;
} }
// Get mod name // Get mod name
let modName = note!.modId; let modName = note.modId;
if (mod !== undefined) { if (mod !== undefined) {
modName = mod!.displayName; modName = mod.displayName;
} }
// Create an embed for the note // Create an embed for the note
const noteEmbed = new MessageEmbed() const noteEmbed = new MessageEmbed()
.setColor('#ff0000') .setColor('#ff0000')
.setTitle(`Sus note for ${userName}`) .setTitle(`Sus note for ${userName}`)
.setThumbnail(user!.avatarURL()!) // TODO avatar does not show when run .setThumbnail(user.avatarURL()!) // TODO avatar does not show when run
.addField( .addField(
`ID: ${noteId} | Moderator: ${modName} | Date: <t:${Math.floor(note!.time.getTime() / 1000)}>`, `ID: ${noteId} | Moderator: ${modName} | Date: <t:${Math.floor(note.time.getTime() / 1000)}>`,
note!.note, note.note,
); );
// Create buttons to delete or cancel the deletion // Create buttons to delete or cancel the deletion
@ -319,7 +320,7 @@ class SusCommand extends Command {
} }
// Listen for the button presses // Listen for the button presses
const collector = channel!.createMessageComponentCollector({ const collector = channel.createMessageComponentCollector({
max: 1, // Maximum of 1 button press max: 1, // Maximum of 1 button press
time: 15000, // 15 seconds time: 15000, // 15 seconds
}); });
@ -327,19 +328,19 @@ class SusCommand extends Command {
// Button pressed // Button pressed
collector.on('collect', async (button: ButtonInteraction) => { collector.on('collect', async (button: ButtonInteraction) => {
if (button.customId === `delete${noteId}`) { if (button.customId === `delete${noteId}`) {
await deactivateNote(noteId!); await deactivateNote(noteId);
await interaction.editReply({ await interaction.editReply({
content: `${user!}'s sus note (ID: ${noteId}) has been successfully removed`, content: `${user}'s sus note (ID: ${noteId}) has been successfully removed`,
embeds: [], embeds: [],
}); });
// TODO create a new Prisma function to only count and not to get a whole list of sus notes // TODO create a new Prisma function to only count and not to get a whole list of sus notes
// Check how many notes the user has and if 0, then remove sus note // Check how many notes the user has and if 0, then remove sus note
const notes = await findNotes(user!.id, true); const notes = await findNotes(user.id, true);
// Checks if there are no notes on the user and if there's none, remove the sus role // Checks if there are no notes on the user and if there's none, remove the sus role
if (notes.length === 0) { if (notes.length === 0) {
await user!.roles.remove(IDs.roles.restrictions.sus); await user.roles.remove(IDs.roles.restrictions.sus);
} }
} }
}); });
@ -367,7 +368,7 @@ class SusCommand extends Command {
return; return;
} }
const userGuildMember = guild!.members.cache.get(user!.id); const userGuildMember = guild.members.cache.get(user.id);
// Checks if managed to find GuildMember for the user // Checks if managed to find GuildMember for the user
if (userGuildMember === undefined) { if (userGuildMember === undefined) {
@ -381,12 +382,12 @@ class SusCommand extends Command {
// Check if the user had sus notes before trying to remove them // Check if the user had sus notes before trying to remove them
// Gets the sus notes from the database // Gets the sus notes from the database
const notes = await findNotes(user!.id, true); const notes = await findNotes(user.id, true);
// Checks if there are no notes on the user // Checks if there are no notes on the user
if (notes.length === 0) { if (notes.length === 0) {
await interaction.reply({ await interaction.reply({
content: `${user!} had no notes!`, content: `${user} had no notes!`,
ephemeral: true, ephemeral: true,
fetchReply: true, fetchReply: true,
}); });
@ -396,16 +397,16 @@ class SusCommand extends Command {
// Creates the embed to display the sus note // Creates the embed to display the sus note
const noteEmbed = new MessageEmbed() const noteEmbed = new MessageEmbed()
.setColor('#ff0000') .setColor('#ff0000')
.setTitle(`Delete ${notes.length} sus notes for ${user!.username}?`) .setTitle(`Delete ${notes.length} sus notes for ${user.username}?`)
.setThumbnail(user!.avatarURL()!); .setThumbnail(user.avatarURL()!);
// Add up to 10 of the latest sus notes to the embed // Add up to 10 of the latest sus notes to the embed
for (let i = notes.length > 10 ? notes.length - 10 : 0; i < notes.length; i += 1) { for (let i = notes.length > 10 ? notes.length - 10 : 0; i < notes.length; i += 1) {
// Get mod name // Get mod name
const modGuildMember = guild!.members.cache.get(notes[i].modId); const modGuildMember = guild.members.cache.get(notes[i].modId);
let mod = notes[i].modId; let mod = notes[i].modId;
if (modGuildMember !== undefined) { if (modGuildMember !== undefined) {
mod = modGuildMember!.displayName; mod = modGuildMember.displayName;
} }
// Add sus note to embed // Add sus note to embed
noteEmbed.addField( noteEmbed.addField(
@ -418,11 +419,11 @@ class SusCommand extends Command {
const buttons = new MessageActionRow<MessageButton>() const buttons = new MessageActionRow<MessageButton>()
.addComponents( .addComponents(
new MessageButton() new MessageButton()
.setCustomId(`delete${user!.id}`) .setCustomId(`delete${user.id}`)
.setLabel('Delete') .setLabel('Delete')
.setStyle(Constants.MessageButtonStyles.DANGER), .setStyle(Constants.MessageButtonStyles.DANGER),
new MessageButton() new MessageButton()
.setCustomId(`cancel${user!.id}`) .setCustomId(`cancel${user.id}`)
.setLabel('Cancel') .setLabel('Cancel')
.setStyle(Constants.MessageButtonStyles.SECONDARY), .setStyle(Constants.MessageButtonStyles.SECONDARY),
); );
@ -442,18 +443,18 @@ class SusCommand extends Command {
} }
// Listen for the button presses // Listen for the button presses
const collector = channel!.createMessageComponentCollector({ const collector = channel.createMessageComponentCollector({
max: 1, // Maximum of 1 button press max: 1, // Maximum of 1 button press
time: 15000, // 15 seconds time: 15000, // 15 seconds
}); });
// Button pressed // Button pressed
collector.on('collect', async (button: ButtonInteraction) => { collector.on('collect', async (button: ButtonInteraction) => {
if (button.customId === `delete${user!.id}`) { if (button.customId === `delete${user.id}`) {
// Remove sus note from database // Remove sus note from database
await deactivateAllNotes(user!.id); await deactivateAllNotes(user.id);
await interaction.editReply({ await interaction.editReply({
content: `Removed all of ${userGuildMember!}'s sus notes successfully`, content: `Removed all of ${userGuildMember}'s sus notes successfully`,
embeds: [], embeds: [],
}); });
} }
@ -467,7 +468,7 @@ class SusCommand extends Command {
}); });
// Remove sus role from the user // Remove sus role from the user
await userGuildMember!.roles.remove(IDs.roles.restrictions.sus); await userGuildMember.roles.remove(IDs.roles.restrictions.sus);
} }
// Non Application Command method of adding a sus note // Non Application Command method of adding a sus note
@ -498,17 +499,17 @@ class SusCommand extends Command {
} }
// Check if user and mod are on the database // Check if user and mod are on the database
if (!await userExists(user)) { if (!await userExists(user.id)) {
await addExistingUser(user); await addExistingUser(user);
} }
if (!await userExists(mod!)) { if (!await userExists(mod.id)) {
await addExistingUser(mod!); await addExistingUser(mod);
} }
await addToDatabase(user.id, mod.id, note); await addToDatabase(user.id, mod.id, note);
// Give the user the sus role they don't already have the sus note // Give the user the sus role they don't already have the sus note
if (!user.roles.cache.has(IDs.roles.restrictions.sus)) { if (!user.roles.cache.has(IDs.roles.restrictions.sus)) {
await user!.roles.add(IDs.roles.restrictions.sus); await user.roles.add(IDs.roles.restrictions.sus);
} }
// Checks if the user is xlevra to send a very kind message // Checks if the user is xlevra to send a very kind message

8
src/commands/mod/unban.ts
View File

@ -18,8 +18,12 @@
*/ */
import { Args, Command, RegisterBehavior } from '@sapphire/framework'; import { Args, Command, RegisterBehavior } from '@sapphire/framework';
import type { User, Message, TextChannel } from 'discord.js'; import type {
import { GuildBan } from 'discord.js'; User,
Message,
TextChannel,
GuildBan,
} from 'discord.js';
import IDs from '../../utils/ids'; import IDs from '../../utils/ids';
import { removeBan, checkActive, addBan } from '../../utils/database/ban'; import { removeBan, checkActive, addBan } from '../../utils/database/ban';

2
src/listeners/verification/joinVC.ts
View File

@ -499,7 +499,7 @@ class VerificationJoinVCListener extends Listener {
return; return;
} }
// Add verifier to database if they're not on the database // Add verifier to database if they're not on the database
if (!(await userExists(verifierGuildMember))) { if (!(await userExists(verifierGuildMember.id))) {
await addExistingUser(verifierGuildMember); await addExistingUser(verifierGuildMember);
} }

6
src/utils/database/dbExistingUser.ts
View File

@ -22,14 +22,14 @@ import { PrismaClient } from '@prisma/client';
import IDs from '../ids'; import IDs from '../ids';
// Checks if the user exists on the database // Checks if the user exists on the database
export async function userExists(user: GuildMember) { export async function userExists(userId: string) {
// Initialises Prisma Client // Initialises Prisma Client
const prisma = new PrismaClient(); const prisma = new PrismaClient();
// Counts if the user is on the database by their snowflake // Counts if the user is on the database by their snowflake
const userQuery = await prisma.user.count({ const userQuery = await prisma.user.count({
where: { where: {
id: user.id, id: userId,
}, },
}); });
@ -97,7 +97,7 @@ export async function addExistingUser(user: GuildMember) {
export async function updateUser(user: GuildMember) { export async function updateUser(user: GuildMember) {
// Check if the user is already on the database // Check if the user is already on the database
if (!(await userExists(user))) { if (!(await userExists(user.id))) {
await addExistingUser(user); await addExistingUser(user);
return; return;
} }