feat: added roast ai agent and wrapped prisma update/create to prisma. to avoid race condition

Author: mrepol742

src/commands/roast.ts

@@ -0,0 +1,55 @@
+import { Message } from "../types/message";
+import log from "../components/utils/log";
+import agentHandler from "../components/ai/agentHandler";
+import { greetings } from "../components/utils/data";
+
+export const info = {
+  command: "roast",
+  description: "Interact with the Roast AI agent.",
+  usage: "roast <query>",
+  example: "roast Who are you?",
+  role: "user",
+  cooldown: 5000,
+};
+
+export default async function (msg: Message): Promise<void> {
+  const query = msg.body.replace(/^roast\b\s*/i, "").trim();
+  if (query.length === 0 && !msg.hasQuotedMsg) {
+    await msg.reply(greetings[Math.floor(Math.random() * greetings.length)]);
+    return;
+  }
+
+  const mentioned = msg.mentionedIds.length > 0;
+
+  const prompt = `You are Roast, your job is to roast anyone,
+  regarding of there status, no hard feelings your only doing your job.
+  You can use nasty emojis or actions, Only give one response at a time!
+  the Date today is %_TODAY_%. Response briefly and concisely,
+  ${mentioned && "You can mentioned user using @ (ps thats there handle/name/username whatever you'd like to call it)and"}
+  Now roast: ${query}`;
+
+  let text = await agentHandler(prompt);
+
+  if (!text) {
+    log.error("roast", "No response generated.");
+    await msg.reply("Sorry, I couldn't generate a response. Please try again.");
+    return;
+  }
+
+  const mentions: string[] = [];
+
+  if (mentioned) {
+    const mentionedContacts = await msg.getMentions();
+
+    for (let i = 0; i < mentionedContacts.length; i++) {
+      const c = mentionedContacts[i];
+      mentions.push(c.id._serialized);
+      text = text.replaceAll(
+        msg.mentionedIds[i].split("@")[0],
+        c.id._serialized.split("@")[0],
+      );
+    }
+  }
+
+  await msg.reply(text, undefined, { mentions });
+}

src/components/services/group.ts

@@ -6,26 +6,32 @@ import * as Sentry from "@sentry/node";
 export async function findOrCreateGroup(chat: WAWebJS.Chat): Promise<void> {
   try {
     if (!chat.isGroup) return;
+
     const groupChat = chat as GroupChat;
 
-    const group = await prisma.group
-      .update({
+    await prisma.$transaction(async (tx) => {
+      const existingGroup = await tx.group.findUnique({
         where: { gid: groupChat.id.user },
+      });
+
+      if (existingGroup) {
+        await tx.group.update({
+          where: { gid: groupChat.id.user },
+          data: {
+            name: groupChat.name,
+            description: groupChat.description,
+          },
+        });
+        return;
+      }
+
+      await tx.group.create({
         data: {
+          gid: groupChat.id.user,
           name: groupChat.name,
           description: groupChat.description,
         },
-      })
-      .catch(() => null);
-
-    if (group) return;
-
-    await prisma.group.create({
-      data: {
-        gid: groupChat.id.user,
-        name: groupChat.name,
-        description: groupChat.description,
-      },
+      });
     });
   } catch (error) {
     Sentry.captureException(error);

src/components/services/user.ts

@@ -47,59 +47,63 @@ export async function addUserQuizPoints(
 }
 
 export async function findOrCreateUser(msg: Message): Promise<boolean> {
-  try {
-    const lid = (msg.author ?? msg.from).split("@")[0];
+  const lid = (msg.author ?? msg.from).split("@")[0];
 
-    const min = 0.1;
-    const max = 0.9;
-    const rand = Math.random() * (max - min) + min;
-    const points = parseFloat(rand.toFixed(1));
+  const min = 0.1;
+  const max = 0.9;
+  const rand = Math.random() * (max - min) + min;
+  const points = parseFloat(rand.toFixed(1));
 
-    const user = await prisma.user
-      .update({
-        where: { lid },
-        data: {
-          commandCount: { increment: 1 },
-          points: {
-            increment: points,
+  try {
+    const result = await prisma.$transaction(async (tx) => {
+      const user = await tx.user.findUnique({ where: { lid } });
+
+      if (user) {
+        await tx.user.update({
+          where: { lid },
+          data: {
+            commandCount: { increment: 1 },
+            points: { increment: points },
           },
-        },
-      })
-      .catch(() => null);
+        });
+
+        return false;
+      }
 
-    if (user) return false;
+      const contact = await msg.getContact();
+      const name = contact?.pushname || contact?.name || "null";
+      const number = contact?.number || "0";
+      const countryCode = contact
+        ? await contact.getCountryCode().catch(() => "null")
+        : "null";
+      const about = contact ? await contact.getAbout().catch(() => null) : null;
 
-    const contact = await msg.getContact();
+      await Promise.allSettled([
+        msg.react("✅"),
+        tx.user.create({
+          data: {
+            lid,
+            name,
+            number,
+            countryCode,
+            type: contact?.isBusiness ? "business" : "private",
+            mode: msg.author ? "group" : "private",
+            about: about ? filterContent(about) : null,
+            commandCount: 1,
+            points,
+          },
+        }),
+      ]);
 
-    const name = contact?.pushname || contact?.name || "null";
-    const number = contact?.number || "0";
-    const countryCode = contact
-      ? await contact.getCountryCode().catch(() => "null")
-      : "null";
-    const about = contact ? await contact.getAbout().catch(() => null) : null;
+      return true;
+    });
 
-    await Promise.allSettled([
-      msg.react("✅"),
-      prisma.user.create({
-        data: {
-          lid,
-          name,
-          number,
-          countryCode,
-          type: contact?.isBusiness ? "business" : "private",
-          mode: msg.author ? "group" : "private",
-          about: about ? filterContent(about) : null,
-          commandCount: 1,
-          points,
-        },
-      }),
-    ]);
-    return true;
+    return result;
   } catch (error) {
     Sentry.captureException(error);
     log.error("Database", `Failed to find or create user.`, error);
+    return false;
   }
-  return false;
 }
 
 export async function getUserbyLid(lid: string) {
@@ -286,9 +290,11 @@ export async function deductUserPoints(
   points: number,
 ): Promise<void> {
   try {
-    prisma.user.update({
-      where: { lid },
-      data: { points: { decrement: points } },
+    await prisma.$transaction(async (tx) => {
+      await tx.user.update({
+        where: { lid },
+        data: { points: { decrement: points } },
+      });
     });
   } catch (error) {
     Sentry.captureException(error);
@@ -301,9 +307,11 @@ export async function addUserPoints(
   points: number,
 ): Promise<void> {
   try {
-    prisma.user.update({
-      where: { lid },
-      data: { points: { increment: points } },
+    await prisma.$transaction(async (tx) => {
+      await tx.user.update({
+        where: { lid },
+        data: { points: { increment: points } },
+      });
     });
   } catch (error) {
     Sentry.captureException(error);