feat: implement screenshot command to capture webpage images and handle URL validation

Author: mrepol742

src/commands/play.ts

@@ -36,14 +36,27 @@ export default async function play(msg: Message) {
     return;
   }
 
-  await msg.reply(`Download in progress... "${audio.title}"`);
+  // Only allow audios shorter than 10 minutes (600 seconds)
+  if (audio.length && audio.length.seconds > 600) {
+    await msg.reply(
+      "Sorry, only videos shorter than 10 minutes can be downloaded."
+    );
+    return;
+  }
+
+  await msg.react("👍");
 
   const stream = await yt.download(audio.id, {
     type: "video+audio",
     quality: "best",
     format: "mp4",
   });
 
+  if (!stream) {
+    await msg.reply("Failed to download the audio stream.");
+    return;
+  }
+
   const tempDir = "./.temp";
   await fs.mkdirSync(tempDir, { recursive: true });
 

src/commands/screenshot.ts

@@ -0,0 +1,49 @@
+import { Message, MessageMedia } from "whatsapp-web.js";
+import puppeteer from "puppeteer";
+import fs from "fs/promises";
+
+export const info = {
+  command: "screenshot",
+  description: "Take a screenshot of a webpage.",
+  usage: "screenshot <url>",
+  example: "screenshot https://example.com",
+  role: "user",
+  cooldown: 5000,
+};
+
+export default async function (msg: Message) {
+  const query = msg.body.replace(/^screenshot\b\s*/i, "").trim();
+  if (query.length === 0) {
+    await msg.reply("Please provide a URL to take a screenshot of.");
+    return;
+  }
+
+  // Validate URL
+  if (!/^https?:\/\//i.test(query)) {
+    await msg.reply(
+      "Please provide a valid URL starting with http:// or https://"
+    );
+    return;
+  }
+
+  const browser = await puppeteer.launch();
+  const page = await browser.newPage();
+
+  await page.goto(query, {
+    waitUntil: "networkidle2",
+    timeout: 60000,
+  });
+
+  const tempDir = "./.temp";
+  await fs.mkdir(tempDir, { recursive: true });
+
+  const path = `${tempDir}/${Date.now()}.png`;
+
+  await page.screenshot({ path });
+
+  await browser.close();
+
+  const media = MessageMedia.fromFilePath(path);
+  await msg.reply(media);
+  await fs.unlink(path);
+}

src/commands/stats.ts

@@ -32,17 +32,13 @@ export default async function (msg: Message) {
     networkInterfaces,
     userCount,
     blockUserCount,
-    whatsAppVersion,
-    whatsAppState,
   ] = await Promise.all([
     si.graphics(),
     si.osInfo(),
     si.shell(),
     si.networkInterfaces(),
     getUserCount(),
-    getBlockUserCount(),
-    client.getWWebVersion(),
-    client.getState()
+    getBlockUserCount()
   ]);
 
   const statsMessage = `
@@ -56,18 +52,10 @@ export default async function (msg: Message) {
     1024 ** 3
   ).toFixed(2)} GB
       VRam: ${gpuInfo.controllers.map((c) => c.vram).join(", ")} MB
-      Load Avg: ${os
-        .loadavg()
-        .map((n) => n.toFixed(2))
-        .join(", ")}
-      Process: #${process.pid} ${process.title}
       Shell: ${shell}
       Network: ${networkInterfaces
         .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/uptime.ts

@@ -1,7 +1,7 @@
 import { Message } from "whatsapp-web.js";
-import log from "../components/utils/log";
 import os from "os";
-import si from "systeminformation";
+import timestamp from "../components/utils/timestamp";
+import { client } from "../components/client";
 
 export const info = {
   command: "uptime",
@@ -15,12 +15,20 @@ export const info = {
 export default async function (msg: Message) {
   if (!/^uptime\b/i.test(msg.body)) return;
 
-  const uptimeMinutes = Math.floor(process.uptime() / 60);
+  const waStatus = await client.getState();
+  const waVersion = await client.getWWebVersion();
+
   const statsMessage = `
-      \`${uptimeMinutes} minutes\`
+      \`${timestamp(process.uptime())}\`
       
       ID: #${process.pid}
-      Platform: ${os.platform()} ${os.arch()}
+      LA: ${os
+        .loadavg()
+        .map((n) => n.toFixed(2))
+        .join(", ")}
+      Status: ${waStatus}
+      Version: ${waVersion}
+      Node.js: ${process.version}
       `;
 
   await msg.reply(statsMessage);

src/commands/video.ts

@@ -37,15 +37,25 @@ export default async function (msg: Message) {
     return;
   }
 
-  const title = video.title.toString();
-  await msg.reply(`Download in progress... "${title}"`);
+  // Only allow videos shorter than 10 minutes (600 seconds)
+  if (video.length && video.length.seconds > 600) {
+    await msg.reply("Sorry, only videos shorter than 10 minutes can be downloaded.");
+    return;
+  }
+  
+  await msg.react("👍");
 
   const stream = await yt.download(video.video_id, {
     type: "video+audio",
     quality: "best",
     format: "mp4",
   });
 
+  if (!stream) {
+    await msg.reply("Failed to download the video stream.");
+    return;
+  }
+
   const tempDir = "./.temp";
   await fs.promises.mkdir(tempDir, { recursive: true });
   const tempPath = path.join(tempDir, `${video.video_id}.mp4`);
@@ -63,11 +73,11 @@ export default async function (msg: Message) {
   const media = new MessageMedia(
     "audio/mpeg",
     audioBuffer.toString("base64"),
-    `${title}.mp4`
+    `${video.title}.mp4`
   );
 
   await msg.reply(media, msg.from, {
-    caption: `${title}`,
+    caption: `${video.title}`,
   });
   await fs.promises.unlink(tempPath);
   await fs.promises.unlink(tempPath + ".mp4");

src/components/utils/loader.ts

@@ -3,9 +3,6 @@ import fs from "fs";
 import path from "path";
 import { commands } from "../../index";
 
-export const command = "load";
-export const role = "admin";
-
 const commandsPath = path.join(__dirname, "..", "..", "commands");
 
 export default function (file: string, customPath?: string) {

src/components/utils/rateLimiter.ts

@@ -3,7 +3,7 @@ import log from "./log";
 
 const LIMIT = 3;
 const BASE_WINDOW_MS = 60 * 1000;
-const PENALTY_INCREMENT_MS = 60 * 1000;
+const PENALTY_INCREMENT_MS = 10 * 1000;
 
 function getKey(number: string) {
   return `rate:${number}`;

src/index.ts

@@ -17,8 +17,8 @@ const botName = process.env.PROJECT_CANIS_ALIAS || "Canis";
 const autoReload = process.env.AUTO_RELOAD === "true";
 const commandsPath = path.join(__dirname, "commands");
 
-log.info("Bot", `Welcome to ${botName}!`);
-log.info("Bot", `Command prefix: ${commandPrefix}`);
+log.info("Bot", `Initiating ${botName}...`);
+log.info("Bot", `prefix: ${commandPrefix}`);
 
 const commands: Record<
   string,
@@ -33,11 +33,11 @@ const commands: Record<
   }
 > = {};
 
-fs.readdirSync(commandsPath).forEach((file) => Loader(file));
+fs.readdirSync(commandsPath).forEach((file: string) => Loader(file));
 
 // Watch for changes
 if (autoReload)
-  fs.watch(commandsPath, (eventType, filename) => {
+  fs.watch(commandsPath, (eventType: string, filename: string | null) => {
     if (filename && /\.js$|\.ts$/.test(filename)) {
       try {
         Loader(filename);