feat: add logout and test commands

Author: mrepol742

.env.example

@@ -18,10 +18,6 @@ PORT=3000
 # automatically reloads commands as they are edited
 AUTO_RELOAD=false
 
-# the phone number of the super admin, used for privileged commands
-# this should be a valid phone number in the format
-SUPER_ADMIN=09123456789
-
 # the database connection URL
 DATABASE_URL=mysql://root@localhost:3306/project_canis
 

src/commands/help.ts

@@ -4,9 +4,9 @@ import { commands } from "../index";
 
 export const info = {
   command: "help",
-  description: "List all available commands.",
-  usage: "help",
-  example: "help",
+  description: "List available commands and their usage.",
+  usage: "help [page] | [role]",
+  example: "help admin",
   role: "user",
   cooldown: 5000,
 };

src/commands/logout.ts

@@ -0,0 +1,24 @@
+import { Message } from "whatsapp-web.js";
+import { client } from "../components/client";
+import log from "../components/utils/log";
+
+export const info = {
+  command: "logout",
+  description: "Log out the client from WhatsApp.",
+  usage: "logout",
+  example: "logout",
+  role: "admin",
+  cooldown: 5000,
+};
+
+export default async function (msg: Message) {
+  if (!/^logout\b/i.test(msg.body)) return;
+
+  try {
+    await client.logout();
+    log.info("Logout", "Client logged out successfully.");
+  } catch (error) {
+    log.error("Logout", "Failed to log out the client.", error);
+    await msg.reply("An error occurred while trying to log out.");
+  }
+}

src/commands/qrcode.ts

@@ -35,7 +35,9 @@ export default async function (msg: Message) {
       await fs.writeFile(tempPath, response.data);
 
       const media = MessageMedia.fromFilePath(tempPath);
-      await msg.reply(media);
+      await msg.reply(media, msg.from, {
+        caption: query,
+      });
       await fs.unlink(tempPath);
     })
     .catch(async (error) => {

src/commands/reload.ts

@@ -10,7 +10,7 @@ export const info = {
   command: "reload",
   description: "Reload a specific command or all commands.",
   usage: "reload [command]",
-  example: "reload poli",
+  example: "reload ai",
   role: "admin",
   cooldown: 5000,
 };
@@ -36,39 +36,53 @@ export default async function (msg: Message) {
       }
     }
 
-    if (!found) await msg.reply(`Failed to reload command "${query}".`);
-    if (found) await msg.reply(`Reloaded command "${query}".`);
+    if (!found)
+      await msg.reply(
+        `
+    \`Failed to load\`
+    ${query}
+    `
+      );
+    if (found)
+      await msg.reply(
+        `
+      \`Successfully reloaded\`
+      ${query}
+      `
+      );
     return;
   }
 
   // Reload all commands
   let count = 0;
+  const newCommands: string[] = [];
+  const removeCommands: string[] = [];
   const commandsPath = path.join(__dirname, "..", "commands");
+
   fs.readdirSync(commandsPath).forEach((file) => {
     if (/\.js$|\.ts$/.test(file)) {
-      const filePath = path.join(commandsPath, file);
-      delete require.cache[require.resolve(filePath)];
-      const commandModule = require(filePath);
-
-      if (
-        typeof commandModule.default === "function" &&
-        commandModule.info &&
-        commandModule.info.command
-      ) {
-        commands[commandModule.info.command] = {
-          command: commandModule.info.command,
-          description: commandModule.info.description || "No description",
-          usage: commandModule.info.usage || "No usage",
-          example: commandModule.info.example || "No example",
-          role: commandModule.info.user || "user",
-          cooldown: commandModule.info.cooldown || 5000,
-          exec: commandModule.default,
-        };
-        count++;
-        log.info("Loader", `Reloaded command: ${commandModule.info.command}`);
+      const commandName = file.replace(/\.(js|ts)$/, "");
+      if (!commands[commandName]) {
+        newCommands.push(commandName);
+      } else {
+        removeCommands.push(commandName);
       }
+      Loader(file);
+      count++;
     }
   });
 
-  await msg.reply(`Reloaded ${count} commands.`);
+  let text = `
+  \`Reloaded\`
+  ${count} commands
+  `;
+
+  if (newCommands.length > 0) {
+    text += `
+  \`Found new command(s)\`
+  ${newCommands.join(", ")}
+  `;
+  }
+
+  await msg.reply(text);
 }

src/commands/stats.ts

@@ -3,6 +3,8 @@ import log from "../components/utils/log";
 import os from "os";
 import si from "systeminformation";
 import { getUserCount, getBlockUserCount } from "../components/services/user";
+import { client } from "../components/client";
+import { commands } from "../index";
 
 export const info = {
   command: "stats",
@@ -23,15 +25,25 @@ export default async function (msg: Message) {
     totalMemory: os.totalmem(),
   };
 
-  const [gpuInfo, osInfo, shell, networkInterfaces, userCount, blockUserCount] =
-    await Promise.all([
-      si.graphics(),
-      si.osInfo(),
-      si.shell(),
-      si.networkInterfaces(),
-      getUserCount(),
-      getBlockUserCount(),
-    ]);
+  const [
+    gpuInfo,
+    osInfo,
+    shell,
+    networkInterfaces,
+    userCount,
+    blockUserCount,
+    whatsAppVersion,
+    whatsAppState,
+  ] = await Promise.all([
+    si.graphics(),
+    si.osInfo(),
+    si.shell(),
+    si.networkInterfaces(),
+    getUserCount(),
+    getBlockUserCount(),
+    client.getWWebVersion(),
+    client.getState()
+  ]);
 
   const statsMessage = `
       \`System Monitor\`
@@ -54,6 +66,9 @@ export default async function (msg: Message) {
         .map((iface) => `${iface.iface} ${iface.speed} Mbps`)
         .join(", ")}
       Node.js: ${process.version}
+      WA: ${whatsAppVersion}
+      WA State: ${whatsAppState}
+      Commands: ${Object.keys(commands).length}
       Users: ${userCount}
       Blocked Users: ${blockUserCount}
     `;

src/commands/test.ts

@@ -0,0 +1,17 @@
+import { Message } from "whatsapp-web.js";
+import log from "../components/utils/log";
+
+export const info = {
+  command: "test",
+  description: "A simple test command.",
+  usage: "test",
+  example: "test",
+  role: "user",
+  cooldown: 5000,
+};
+
+export default async function (msg: Message) {
+    if (!/^test\b/i.test(msg.body)) return;
+    
+    await msg.reply("This is a test response.");
+}

src/commands/video.ts

@@ -66,7 +66,9 @@ export default async function (msg: Message) {
     `${title}.mp4`
   );
 
-  await msg.reply(media, msg.from);
+  await msg.reply(media, msg.from, {
+    caption: `${title}`,
+  });
   await fs.promises.unlink(tempPath);
   await fs.promises.unlink(tempPath + ".mp4");
 }

src/commands/wa.ts

@@ -0,0 +1,37 @@
+import { Message } from "whatsapp-web.js";
+import { client } from "../components/client";
+
+export const info = {
+  command: "wa",
+  description: "Set WhatsApp status or name.",
+  usage: "wa [status | name]",
+  example: "wa status",
+  role: "admin",
+  cooldown: 5000,
+};
+
+export default async function (msg: Message) {
+  const query = msg.body.replace(/^wa\b\s*/i, "").trim();
+  if (query.length !== 0) {
+    if (!/^(status|name)$/i.test(query)) {
+      await msg.reply(
+        "Invalid argument. Please use one of the following:\n\nstatus or name"
+      );
+      return;
+    }
+  }
+
+  if (msg.hasQuotedMsg) {
+    await msg.reply("Please reply to a message to set the status or name.");
+    return;
+  }
+
+  const quotedMsg = await msg.getQuotedMessage();
+  if (query === "status") {
+    client.setStatus(quotedMsg.body);
+    await msg.reply("Status updated successfully.");
+    return;
+  }
+  client.setDisplayName(quotedMsg.body);
+  await msg.reply("Name updated successfully.");
+}

src/components/events/message.ts

@@ -16,7 +16,6 @@ import Font from "../utils/font";
 const commandPrefix = process.env.COMMAND_PREFIX || "!";
 const commandPrefixLess = process.env.COMMAND_PREFIX_LESS === "true";
 const debug = process.env.DEBUG === "true";
-const superAdmin = process.env.SUPER_ADMIN || "";
 
 export default async function message(msg: Message) {
   // ignore message if it is older than 10 seconds
@@ -61,19 +60,19 @@ export default async function message(msg: Message) {
   /*
    * Rate limit commands to prevent abuse.
    */
-  if (senderId !== superAdmin) {
+  if (!msg.fromMe) {
     const rate = rateLimiter(msg.from);
     if (rate === null) return;
     if (!rate) {
-      msg.reply("You are sending commands too fast. Please wait a minute.");
+      msg.reply("Please wait a minute or so.");
       return;
     }
   }
 
   /*
    * Role base restrictions.
    */
-  if (handler.role === "admin" && !msg.fromMe && senderId !== superAdmin) {
+  if (handler.role === "admin" && !msg.fromMe) {
     return;
   }