refactor: Update rmbg.js and sticker.js to improve reply validation

Neeraj-x0 · May 15, 2024 · 939ebbe · 939ebbe
1 parent 70dd0fd
commit 939ebbe
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 8 deletions.
diff --git a/assets/plugins/rmbg.js b/assets/plugins/rmbg.js
@@ -15,6 +15,7 @@ command(
         "Set RemoveBg API Key in config.js \n Get it from https://www.remove.bg/api"
       );
     if (!message.reply_message && !message.reply_message.image)
+
       return await message.reply("Reply to an image");
     let buff = await m.quoted.download();
     let buffer = await removeBg(buff);

diff --git a/assets/plugins/sticker.js b/assets/plugins/sticker.js
@@ -9,14 +9,13 @@ command(
     type: "converter",
   },
   async (message, match, m) => {
-    if (
-      !(
-        message.reply_message.video ||
-        message.reply_message.image ||
-        message.reply_message.text
-      )
-    )
-      return await message.reply("_Reply to photo/video/text_");
+    const isValid =
+      message.reply_message.video ||
+      message.reply_message.image ||
+      message.reply_message.text ||
+      message.reply_message.mimetype.includes("image");
+
+    if (isValid) return await message.reply("_Reply to photo/video/text_");
     if (message.reply_message.text) {
       let buff = await textToImg(message.reply_message.text);
       return await message.sendMessage(

diff --git a/lib/Messages/ReplyMessage.js b/lib/Messages/ReplyMessage.js
@@ -64,6 +64,10 @@ class ReplyMessage extends Base {
         const audioMessage = quotedMessage.audioMessage;
         this.audio = audioMessage;
         this.mimetype = audioMessage.mimetype;
+      } else if (quotedMessage.documentMessage) {
+        const documentMessage = quotedMessage.documentMessage;
+        this.document = documentMessage;
+        this.mimetype = documentMessage.mimetype;
       }
     }