feat: add quiz functionality with tracking and user interaction

Author: mrepol742

prisma/migrations/20250722091631_add_quiz/migration.sql

@@ -0,0 +1,11 @@
+-- CreateTable
+CREATE TABLE `Quiz` (
+    `id` INTEGER NOT NULL AUTO_INCREMENT,
+    `mid` VARCHAR(191) NOT NULL,
+    `lid` VARCHAR(191) NOT NULL,
+    `qid` VARCHAR(191) NOT NULL,
+    `createdAt` DATETIME(3) NOT NULL DEFAULT CURRENT_TIMESTAMP(3),
+    `updatedAt` DATETIME(3) NOT NULL,
+
+    PRIMARY KEY (`id`)
+) DEFAULT CHARACTER SET utf8mb4 COLLATE utf8mb4_unicode_ci;

prisma/migrations/20250722092649_add_quiz_answered_at/migration.sql

@@ -0,0 +1,2 @@
+-- AlterTable
+ALTER TABLE `Quiz` ADD COLUMN `answeredAt` DATETIME(3) NULL;

prisma/migrations/20250722092719_add_quiz_mid_unique/migration.sql

@@ -0,0 +1,8 @@
+/*
+  Warnings:
+
+  - A unique constraint covering the columns `[mid]` on the table `Quiz` will be added. If there are existing duplicate values, this will fail.
+
+*/
+-- CreateIndex
+CREATE UNIQUE INDEX `Quiz_mid_key` ON `Quiz`(`mid`);

prisma/migrations/20250722100651_add_quiz_answered_count_user/migration.sql

@@ -0,0 +1,2 @@
+-- AlterTable
+ALTER TABLE `User` ADD COLUMN `quizAnswered` INTEGER NOT NULL DEFAULT 0;

prisma/schema.prisma

@@ -24,6 +24,7 @@ model User {
   mode         String
   about        String?
   commandCount Int      @default(0)
+  quizAnswered Int      @default(0)
   createdAt    DateTime @default(now())
   updatedAt    DateTime @updatedAt
 }
@@ -70,4 +71,14 @@ model Log {
   output    String?
   createdAt DateTime @default(now())
   updatedAt DateTime @updatedAt
-}
+}
+
+model Quiz {
+  id         Int       @id @default(autoincrement())
+  mid        String    @unique
+  lid        String
+  qid        String
+  answeredAt DateTime?
+  createdAt  DateTime  @default(now())
+  updatedAt  DateTime  @updatedAt
+}

src/commands/quiz.ts

@@ -1,5 +1,7 @@
 import { Message } from "whatsapp-web.js";
 import { quiz } from "../components/utils/data";
+import { newQuizAttempt } from "../components/services/quiz";
+import log from "../components/utils/log";
 
 export const info = {
   command: "quiz",
@@ -13,9 +15,12 @@ export const info = {
 export default async function (msg: Message) {
   if (!/^quiz$/i.test(msg.body)) return;
 
-  const response = quiz[Math.floor(Math.random() * quiz.length)];
+  const id = Math.floor(Math.random() * quiz.length);
+  const response = quiz[id];
+
   if (response.length === 0)
     return await msg.reply("I don't have any quiz questions right now...");
+
   let text = `
   \`${response.question}\`
   `;
@@ -29,5 +34,9 @@ export default async function (msg: Message) {
     `;
   }
 
-  await msg.reply(text);
+  const messageReturn = await msg.reply(text);
+  await Promise.all([
+    newQuizAttempt(messageReturn, id.toString()),
+    log.info("quiz", `Quiz question sent: ${response.question}`),
+  ]);
 }

src/commands/top.ts

@@ -28,7 +28,7 @@ export default async function (msg: Message) {
       .map((u, index) => {
         const displayName =
           u.name.length > 12 ? u.name.slice(0, 12) + "..." : u.name;
-        return `${index + 1}. ${displayName} - ${u.commandCount} Points`;
+        return `${index + 1}. ${displayName} - ${u.commandCount + u.quizAnswered} Points`;
       })
       .join("\n    ")}
     `;

src/components/events/message.ts

@@ -12,6 +12,7 @@ import sleep from "../utils/sleep";
 import { findOrCreateUser, isBlocked } from "../services/user";
 import { client } from "../client";
 import Font from "../utils/font";
+import quiz from "./quiz";
 
 const commandPrefix = process.env.COMMAND_PREFIX || "!";
 const commandPrefixLess = process.env.COMMAND_PREFIX_LESS === "true";
@@ -30,6 +31,15 @@ export default async function message(msg: Message) {
   )
     return; // ignore them all
 
+  /*
+   *
+   * Quiz command validation
+   */
+  if (msg.hasQuotedMsg) {
+    const quoted = await msg.getQuotedMessage();
+    if (await quiz(msg, quoted)) return;
+  }
+
   // process normalization
   msg.body = msg.body
     .normalize("NFKC")
@@ -129,13 +139,11 @@ export default async function message(msg: Message) {
     await Promise.all([
       handler.exec(msg),
       (async () => {
-        if (!msg.fromMe) {
-          const user = await findOrCreateUser(msg);
+        const user = await findOrCreateUser(msg);
 
-          if (user) {
-            await sleep(2000); // Prevent rate limiting issues
-            await msg.react("✅");
-          }
+        if (user) {
+          await sleep(2000); // Prevent rate limiting issues
+          await msg.react("✅");
         }
 
         return Promise.resolve();

src/components/events/quiz.ts

@@ -0,0 +1,37 @@
+import { Message } from "whatsapp-web.js";
+import { getQuizAttempts, setQuizAttemptAnswered } from "../services/quiz";
+import { quiz } from "../utils/data";
+
+const done = [
+  "Correct! 🎉",
+  "Well done! 👍",
+  "Nice job! ✅",
+  "You got it! 🥳",
+  "That's right! 👏",
+  "Excellent! 🌟",
+  "Great answer! 💡",
+  "Spot on! 🎯",
+  "Perfect! 🏆",
+  "You nailed it! 🔥",
+];
+
+export default async function (
+  msg: Message,
+  quoted: Message
+): Promise<boolean> {
+  if (!quoted.body) return false;
+
+  const quizAttempts = await getQuizAttempts(quoted);
+  if (quizAttempts) {
+    const question = quiz[parseInt(quizAttempts.qid)];
+
+    if (msg.body.trim().toLowerCase() === question.answer.toLowerCase()) {
+      await Promise.all([
+        msg.reply(done[Math.floor(Math.random() * done.length)]),
+        setQuizAttemptAnswered(msg, quoted),
+      ]);
+      return true;
+    }
+  }
+  return false;
+}

src/components/services/quiz.ts

@@ -0,0 +1,61 @@
+import { prisma } from "../prisma";
+import log from "../utils/log";
+import { Message } from "whatsapp-web.js";
+
+export async function newQuizAttempt(msg: Message, qid: string): Promise<void> {
+  try {
+    const lid = msg.id.remote.split("@")[0];
+    await prisma.quiz.create({
+      data: {
+        mid: msg.id.id,
+        lid,
+        qid,
+      },
+    });
+  } catch (error) {
+    log.error("Database", "Failed to add quiz.", error);
+  }
+}
+
+export async function getQuizAttempts(msg: Message): Promise<any | null> {
+  try {
+    if (!msg.id || !msg.id.remote) return [];
+
+    const quiz = await prisma.quiz.findUnique({
+      where: { mid: msg.id.id, answeredAt: null },
+    });
+
+    return quiz;
+  } catch (error) {
+    log.error("Database", `Failed to get quiz attempts.`, error);
+  }
+  return null;
+}
+
+export async function setQuizAttemptAnswered(
+  msg: Message,
+  quoted: Message
+): Promise<void> {
+  try {
+    if (!quoted.id || !quoted.id.remote) return;
+
+    await Promise.all([
+      prisma.quiz.update({
+        where: { mid: quoted.id.id },
+        data: { answeredAt: new Date() },
+      }),
+      prisma.user.update({
+        where: {
+          lid: msg.author ? msg.author.split("@")[0] : msg.from.split("@")[0],
+        },
+        data: {
+          quizAnswered: {
+            increment: 1,
+          },
+        },
+      }),
+    ]);
+  } catch (error) {
+    log.error("Database", `Failed to set quiz attempt as answered.`, error);
+  }
+}