mirror of
https://github.com/veganhacktivists/arabot.git
synced 2025-07-07 10:21:01 +02:00
89 lines
2.7 KiB
TypeScript
89 lines
2.7 KiB
TypeScript
// SPDX-License-Identifier: GPL-3.0-or-later
|
|
/*
|
|
Animal Rights Advocates Discord Bot
|
|
Copyright (C) 2022 Anthony Berg
|
|
|
|
This program is free software: you can redistribute it and/or modify
|
|
it under the terms of the GNU General Public License as published by
|
|
the Free Software Foundation, either version 3 of the License, or
|
|
(at your option) any later version.
|
|
|
|
This program is distributed in the hope that it will be useful,
|
|
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
GNU General Public License for more details.
|
|
|
|
You should have received a copy of the GNU General Public License
|
|
along with this program. If not, see <https://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
import type { GuildMember } from 'discord.js';
|
|
import { PrismaClient } from '@prisma/client';
|
|
import { IDs } from './ids';
|
|
|
|
// Checks if the user exists on the database
|
|
export async function userExists(user: GuildMember) {
|
|
// Initialises Prisma Client
|
|
const prisma = new PrismaClient();
|
|
|
|
// Counts if the user is on the database by their snowflake
|
|
const userExists = await prisma.user.count({
|
|
where: {
|
|
id: user.id,
|
|
},
|
|
});
|
|
|
|
// Close the database connection
|
|
await prisma.$disconnect();
|
|
|
|
// If the user is found on the database, then return true, otherwise, false.
|
|
if (userExists > 0) {
|
|
return true;
|
|
}
|
|
return false;
|
|
}
|
|
|
|
// Adds the user to the database if they were already on the server before the bot/database
|
|
export async function addExistingUser(user: GuildMember) {
|
|
// Initialises Prisma Client
|
|
const prisma = new PrismaClient();
|
|
|
|
// Counts if the user is on the database by their snowflake
|
|
const userExists = await prisma.user.count({
|
|
where: {
|
|
id: user.id,
|
|
},
|
|
});
|
|
|
|
// If the user is already in the database
|
|
if (userExists > 0) {
|
|
return;
|
|
}
|
|
|
|
// Checks what roles the user has
|
|
const hasVegan = user.roles.cache.has(IDs.roles.vegan.vegan);
|
|
const hasActivist = user.roles.cache.has(IDs.roles.vegan.activist);
|
|
const hasPlus = user.roles.cache.has(IDs.roles.vegan.plus);
|
|
const hasVegCurious = user.roles.cache.has(IDs.roles.nonvegan.vegCurious);
|
|
const hasConvinced = user.roles.cache.has(IDs.roles.nonvegan.convinced);
|
|
const hasTrusted = user.roles.cache.has(IDs.roles.trusted);
|
|
const hasMuted = user.roles.cache.has(IDs.roles.restrictions.muted);
|
|
|
|
// Create the user in the database
|
|
await prisma.user.create({
|
|
data: {
|
|
id: user.id,
|
|
vegan: hasVegan,
|
|
trusted: hasTrusted,
|
|
activist: hasActivist,
|
|
plus: hasPlus,
|
|
vegCurious: hasVegCurious,
|
|
convinced: hasConvinced,
|
|
muted: hasMuted,
|
|
},
|
|
});
|
|
|
|
// Close the database connection
|
|
await prisma.$disconnect();
|
|
}
|