feat(arabot): start logging of sus notes

prisma/schema.prisma

@@ -38,23 +38,23 @@ model User {
   vegCurious     Boolean    @default(false)
   convinced      Boolean    @default(false)
   muted          Boolean    @default(false)
-  VerifyUser     Verify[]   @relation("user")
-  VerifyVerifier Verify[]   @relation("verifier")
-  SusUser        Sus[]      @relation("user")
-  SusMod         Sus[]      @relation("mod")
-  RestrictUser   Restrict[] @relation("user")
-  RestrictMod    Restrict[] @relation("mod")
-  BanUser        Ban[]      @relation("user")
-  BanMod         Ban[]      @relation("mod")
-  TempBanUser    TempBan[]  @relation("user")
-  TempBanMod     TempBan[]  @relation("mod")
+  VerifyUser     Verify[]   @relation("verUser")
+  VerifyVerifier Verify[]   @relation("verVerifier")
+  SusUser        Sus[]      @relation("susUser")
+  SusMod         Sus[]      @relation("susMod")
+  RestrictUser   Restrict[] @relation("restUser")
+  RestrictMod    Restrict[] @relation("restMod")
+  BanUser        Ban[]      @relation("banUser")
+  BanMod         Ban[]      @relation("banMod")
+  TempBanUser    TempBan[]  @relation("tbanUser")
+  TempBanMod     TempBan[]  @relation("tbanMod")
 }
 
 model Verify {
   id          Int      @id @default(autoincrement())
-  user        User     @relation("user", fields: [userId], references: [id])
+  user        User     @relation("verUser", fields: [userId], references: [id])
   userId      String
-  verifier    User?    @relation("verifier", fields: [verifierId], references: [id])
+  verifier    User?    @relation("verVerifier", fields: [verifierId], references: [id])
   verifierId  String?
   time        DateTime @default(now())
   timedOut    Boolean  @default(false) // If they got kicked out of verification because they timed out
@@ -66,17 +66,18 @@ model Verify {
 
 model Sus {
   id     Int    @id @default(autoincrement())
-  user   User   @relation("user", fields: [userId], references: [id])
+  user   User   @relation("susUser", fields: [userId], references: [id])
   userId String
-  mod    User   @relation("mod", fields: [modId], references: [id])
+  mod    User   @relation("susMod", fields: [modId], references: [id])
   modId  String
+  note   String
 }
 
 model Restrict {
   id        Int       @id @default(autoincrement())
-  user      User      @relation("user", fields: [userId], references: [id])
+  user      User      @relation("restUser", fields: [userId], references: [id])
   userId    String
-  mod       User      @relation("mod", fields: [modId], references: [id])
+  mod       User      @relation("restMod", fields: [modId], references: [id])
   modId     String
   startTime DateTime  @default(now())
   endedTime DateTime?
@@ -85,9 +86,9 @@ model Restrict {
 
 model Ban {
   id     Int      @id @default(autoincrement())
-  user   User     @relation("user", fields: [userId], references: [id])
+  user   User     @relation("banUser", fields: [userId], references: [id])
   userId String
-  mod    User     @relation("mod", fields: [modId], references: [id])
+  mod    User     @relation("banMod", fields: [modId], references: [id])
   modId  String
   time   DateTime @default(now())
   active Boolean  @default(true)
@@ -96,9 +97,9 @@ model Ban {
 
 model TempBan {
   id        Int      @id @default(autoincrement())
-  user      User     @relation("user", fields: [userId], references: [id])
+  user      User     @relation("tbanUser", fields: [userId], references: [id])
   userId    String
-  mod       User     @relation("mod", fields: [modId], references: [id])
+  mod       User     @relation("tbanMod", fields: [modId], references: [id])
   modId     String
   startTime DateTime @default(now())
   endedTime DateTime

src/commands/mod/sus.ts

@@ -0,0 +1,83 @@
+// 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 { Command } from '@sapphire/framework';
+import type { Message, GuildMember } from 'discord.js';
+import { addExistingUser, userExists } from '../../utils/dbExistingUser';
+import { PrismaClient } from '@prisma/client';
+
+export class SusCommand extends Command {
+  public constructor(context: Command.Context) {
+    super(context, {
+      name: 'sus',
+      description: 'Adds a sus note about a user.',
+    });
+  }
+
+  public async messageRun(message: Message) {
+    const msg = await message.channel.send('Ping?');
+
+    const content = `Pong from JavaScript! Bot Latency ${Math.round(this.container.client.ws.ping)}ms. API Latency ${msg.createdTimestamp - message.createdTimestamp}ms.`;
+
+    // Checks if the user exists
+    if (message.member === null) {
+      await message.channel.send('Member not found');
+      return;
+    }
+
+    // Checks if the user is on the database
+    if (!await userExists(message.member)) {
+      // If they are not on the database, add them to the database
+      await addExistingUser(message.member);
+    }
+
+    await addToDatabase(message.member, message.member, 'This is a note :D');
+
+    return msg.edit(content);
+  }
+}
+
+async function addToDatabase(user: GuildMember, mod: GuildMember, message: string) {
+  // Initialise the database connection
+  const prisma = new PrismaClient();
+
+  // Create variables for data input
+  const userId = user.id.toString();
+  const modId = mod.id.toString();
+
+  // Add the user to the database
+  await prisma.sus.create({
+    data: {
+      user: {
+        connect: {
+          id: userId,
+        },
+      },
+      mod: {
+        connect: {
+          id: modId,
+        },
+      },
+      note: message,
+    },
+  });
+
+  // Close the database connection
+  await prisma.$disconnect();
+}