feat(database): add new database checks and ids for roles

src/utils/dbExistingUser.ts

@@ -0,0 +1,85 @@
+// 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);
+  const hasActivist = user.roles.cache.has(IDs.roles.activist);
+  const hasPlus = user.roles.cache.has(IDs.roles.plus);
+  const hasVegCurious = user.roles.cache.has(IDs.roles.vegCurious);
+  const hasConvinced = user.roles.cache.has(IDs.roles.convinced);
+  const hasTrusted = user.roles.cache.has(IDs.roles.trusted);
+  const hasMuted = user.roles.cache.has(IDs.roles.trusted);
+
+  // 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,
+    },
+  });
+}

src/utils/ids.ts

@@ -0,0 +1,32 @@
+/*
+    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/>.
+*/
+
+const IDs = {
+  roles: {
+    nonvegan: '774763753308815400',
+    vegan: '788114978020392982',
+    activist: '730915638746546257',
+    plus: '798682625619132428',
+    vegCurious: '832656046572961803',
+    convinced: '797132019166871612',
+    trusted: '731563158011117590',
+    muted: '730924813681688596',
+  },
+};
+
+export { IDs };