feat: add commands for pinning, unpinning, and fetching top users; improve wa command error handling

Author: mrepol742

src/commands/block.ts

@@ -19,7 +19,7 @@ export default async function (msg: Message) {
     return;
   }
 
-  if (/^--rem$/.test(msg.body)) {
+  if (/^block\s--rem/.test(msg.body)) {
     for (const userId of msg.mentionedIds) {
       const lid = userId.split("@")[0];
 
@@ -28,7 +28,7 @@ export default async function (msg: Message) {
       });
     }
 
-    await msg.react("✅");
+    await msg.react("❌");
     return;
   }
 

src/commands/meme.ts

@@ -0,0 +1,45 @@
+import { Message, MessageMedia } from "whatsapp-web.js";
+import axios from "axios";
+import log from "../components/utils/log";
+import fs from "fs/promises";
+
+export const info = {
+  command: "meme",
+  description: "Fetch a random meme image.",
+  usage: "meme",
+  example: "meme",
+  role: "user",
+  cooldown: 5000,
+};
+
+export default async function (msg: Message) {
+  if (!/^meme$/i.test(msg.body)) return;
+
+  await axios
+    .get("https://api.popcat.xyz/meme")
+    .then(async (response) => {
+      await axios
+        .get(response.data.content.image, {
+          responseType: "arraybuffer",
+        })
+        .then(async (response) => {
+          const tempDir = "./.temp";
+          await fs.mkdir(tempDir, { recursive: true });
+
+          const tempPath = `${tempDir}/meme_${Date.now()}.png`;
+          await fs.writeFile(tempPath, response.data);
+
+          const media = MessageMedia.fromFilePath(tempPath);
+          await msg.reply(media);
+          await fs.unlink(tempPath);
+        })
+        .catch(async (error) => {
+          log.error("meme", `Error fetching image: ${error.message}`);
+          await msg.reply("Error fetching image. Please try again later.");
+        });
+    })
+    .catch(async (error) => {
+      log.error("meme", `Error fetching data: ${error.message}`);
+      await msg.reply("Error fetching data. Please try again later.");
+    });
+}

src/commands/pin.ts

@@ -0,0 +1,28 @@
+import { Message } from "whatsapp-web.js";
+import log from "../components/utils/log";
+
+export const info = {
+  command: "pin",
+  description: "Pin a message for a long duration.",
+  usage: "pin",
+  example: "pin",
+  role: "user",
+  cooldown: 5000,
+};
+
+export default async function (msg: Message) {
+    if (!/^pin$/i.test(msg.body)) return;
+    
+    if (!msg.hasQuotedMsg) {
+        await msg.reply("Please reply to the message you want to pin.");
+        return;
+    }
+
+    const quotedMsg = await msg.getQuotedMessage();
+    if (!quotedMsg.body) {
+      await msg.reply("Please reply to a message with the new status or name.");
+      return;
+    }
+
+    await quotedMsg.pin(24 * 60 * 60 * 1000 * 1000);
+}

src/commands/top.ts

@@ -0,0 +1,37 @@
+import { Message } from "whatsapp-web.js";
+import log from "../components/utils/log";
+import { getUsers } from "../components/services/user";
+
+export const info = {
+  command: "top",
+  description: "Get the top users of the bot.",
+  usage: "top",
+  example: "top",
+  role: "user",
+  cooldown: 5000,
+};
+
+export default async function (msg: Message) {
+  if (!/^top$/i.test(msg.body)) return;
+
+  const user = await getUsers();
+  if (!user || user.length === 0) {
+    await msg.reply("No users found.");
+    return;
+  }
+
+  const text = `
+\`Top Users:\`
+
+    ${user
+      .filter((u, index) => u.commandCount !== 0 && index < 20)
+      .map((u, index) => {
+        const displayName =
+          u.name.length > 12 ? u.name.slice(0, 12) + "..." : u.name;
+        return `${index + 1}. ${displayName} - ${u.commandCount} Points`;
+      })
+      .join("\n    ")}
+    `;
+
+  await msg.reply(text);
+}

src/commands/unpin.ts

@@ -0,0 +1,28 @@
+import { Message } from "whatsapp-web.js";
+import log from "../components/utils/log";
+
+export const info = {
+  command: "unpin",
+  description: "Unpin a previously pinned message.",
+  usage: "unpin",
+  example: "unpin",
+  role: "user",
+  cooldown: 5000,
+};
+
+export default async function (msg: Message) {
+  if (!/^unpin$/i.test(msg.body)) return;
+
+  if (!msg.hasQuotedMsg) {
+    await msg.reply("Please reply to the message you want to upin.");
+    return;
+  }
+
+  const quotedMsg = await msg.getQuotedMessage();
+  if (!quotedMsg.body) {
+    await msg.reply("Please reply to a pinnmed message.");
+    return;
+  }
+
+  await quotedMsg.unpin();
+}

src/commands/wa.ts

@@ -27,6 +27,10 @@ export default async function (msg: Message) {
   }
 
   const quotedMsg = await msg.getQuotedMessage();
+  if (!quotedMsg.body) {
+    await msg.reply("Please reply to a message with the new status or name.");
+    return;
+  }
   if (query === "status") {
     client.setStatus(quotedMsg.body);
     await msg.reply("Status updated successfully.");

src/components/events/message.ts

@@ -22,7 +22,7 @@ export default async function message(msg: Message) {
   if (msg.timestamp < Date.now() / 1000 - 10) return;
 
   if (
-    msg.hasQuotedMsg ||
+    // (msg.hasQuotedMsg && !msg.fromMe) ||
     msg.isForwarded ||
     msg.isGif ||
     msg.isStatus ||

src/components/services/user.ts

@@ -95,3 +95,29 @@ export async function getBlockUserCount(): Promise<number> {
     return 0;
   }
 }
+
+export async function getUsers(): Promise<any[]> {
+  try {
+    const users = await prisma.user.groupBy({
+      by: ["number", "name"],
+      _sum: {
+        commandCount: true,
+      },
+      orderBy: {
+        _sum: {
+          commandCount: "desc",
+        },
+      },
+    });
+
+    // format
+    return users.map((u) => ({
+      name: u.name,
+      number: u.number,
+      commandCount: u._sum.commandCount,
+    }));
+  } catch (error) {
+    log.error("Database", `Failed to get users.`, error);
+    return [];
+  }
+}