feat: reduce user points when they get banned

Author: mrepol742

src/commands/fbdl.ts

@@ -31,19 +31,21 @@ export default async function (msg: Message) {
   if (!result.url)
     return await msg.reply("No video found at the provided URL.");
 
-  const response = await axios.get(result.hd, { responseType: "arraybuffer" });
+  const response = await axios.get(result.hd || result.sd, {
+    responseType: "arraybuffer",
+  });
 
   const tempDir = "./.temp";
-  await fs.mkdirSync(tempDir, { recursive: true });
+  fs.mkdirSync(tempDir, { recursive: true });
 
   const tempPath = `${tempDir}/fbdl_${Date.now()}.mp4`;
-  await fs.writeFileSync(tempPath, response.data);
+  fs.writeFileSync(tempPath, response.data);
 
   const audioBuffer = fs.readFileSync(tempPath);
   const media = new MessageMedia(
     "audio/mpeg",
     audioBuffer.toString("base64"),
-    `${result.title}.mp4`
+    `${result.title}.mp4`,
   );
 
   await msg.reply(media, msg.from);

src/components/events/message.ts

@@ -79,7 +79,7 @@ export default async function (msg: Message) {
    * Rate limit commands to prevent abuse.
    */
   if (!msg.fromMe) {
-    const rate = await rateLimiter(msg.from);
+    const rate = await rateLimiter(msg);
     if (rate) return;
     if (rate === null) {
       msg.reply("Please wait a minute or so.");

src/components/utils/rateLimiter.ts

@@ -1,5 +1,7 @@
+import { Message } from "whatsapp-web.js";
 import redis from "../redis";
 import log from "./log";
+import { prisma } from "../prisma";
 
 const LIMIT = 5;
 const BASE_WINDOW_MS = 30 * 1000;
@@ -10,8 +12,9 @@ function getKey(number: string) {
 }
 
 export default async function rateLimiter(
-  number: string,
+  msg: Message,
 ): Promise<boolean | null> {
+  const number = msg.from;
   const now = Date.now();
   const key = getKey(number);
 
@@ -47,7 +50,19 @@ export default async function rateLimiter(
       `User ${number} blocked until ${new Date(entry.penaltyUntil).toLocaleTimeString()}`,
     );
 
-    await redis.set(key, JSON.stringify(entry));
+    await Promise.all([
+      redis.set(key, JSON.stringify(entry)),
+      prisma.user.update({
+        where: {
+          lid: msg.author ? msg.author.split("@")[0] : msg.from.split("@")[0],
+        },
+        data: {
+          quizAnswered: {
+            decrement: 10,
+          },
+        },
+      }),
+    ]);
     if (entry.penaltyCount === 1) return null;
     return true;
   }