Adicionando rota de download de midia em base64 ou buffer

src/api/controllers/chat.controller.ts

@@ -2,6 +2,7 @@ import {
   ArchiveChatDto,
   BlockUserDto,
   DeleteMessage,
+  DownloadMediaMessageDto,
   getBase64FromMediaMessageDto,
   MarkChatUnreadDto,
   NumberDto,
@@ -82,6 +83,10 @@ export class ChatController {
     return await this.waMonitor.waInstances[instanceName].updatePrivacySettings(data);
   }
 
+  public async downloadMediaMessage({ instanceName }: InstanceDto, data: DownloadMediaMessageDto) {
+    return await this.waMonitor.waInstances[instanceName].downloadMediaMessage(data);
+  }
+
   public async fetchBusinessProfile({ instanceName }: InstanceDto, data: ProfilePictureDto) {
     return await this.waMonitor.waInstances[instanceName].fetchBusinessProfile(data.number);
   }

src/api/dto/chat.dto.ts

@@ -1,4 +1,6 @@
 import {
+  DownloadableMessage,
+  MediaType,
   proto,
   WAPresence,
   WAPrivacyGroupAddValue,
@@ -94,6 +96,13 @@ export class PrivacySettingDto {
   groupadd: WAPrivacyGroupAddValue;
 }
 
+export class DownloadMediaMessageDto {
+  downloadableMessage: DownloadableMessage;
+  type: MediaType;
+  timeout?: number;
+  returnType?: 'base64' | 'buffer' = 'buffer';
+}
+
 export class DeleteMessage {
   id: string;
   fromMe: boolean;

src/api/integrations/channel/whatsapp/whatsapp.baileys.service.ts

@@ -3,6 +3,7 @@ import {
   ArchiveChatDto,
   BlockUserDto,
   DeleteMessage,
+  DownloadMediaMessageDto,
   getBase64FromMediaMessageDto,
   LastMessage,
   MarkChatUnreadDto,
@@ -77,6 +78,7 @@ import { Instance } from '@prisma/client';
 import { createJid } from '@utils/createJid';
 import { makeProxyAgent } from '@utils/makeProxyAgent';
 import { getOnWhatsappCache, saveOnWhatsappCache } from '@utils/onWhatsappCache';
+import { readStreamWithTimeout } from '@utils/readStreamWithTimeout';
 import { status } from '@utils/renderStatus';
 import useMultiFileAuthStatePrisma from '@utils/use-multi-file-auth-state-prisma';
 import { AuthStateProvider } from '@utils/use-multi-file-auth-state-provider-files';
@@ -92,6 +94,7 @@ import makeWASocket, {
   Contact,
   delay,
   DisconnectReason,
+  downloadContentFromMessage,
   downloadMediaMessage,
   fetchLatestBaileysVersion,
   generateWAMessageFromContent,
@@ -3133,6 +3136,28 @@ export class BaileysStartupService extends ChannelStartupService {
     }
   }
 
+  public async downloadMediaMessage(downloadMedia: DownloadMediaMessageDto) {
+    try {
+      const media = await downloadContentFromMessage(
+        {
+          ...downloadMedia.downloadableMessage,
+        },
+        downloadMedia.type,
+      );
+
+      const buffer = await readStreamWithTimeout(media, downloadMedia.timeout);
+
+      if (downloadMedia.returnType === 'base64') {
+        const base64Data = buffer.toString('base64');
+        return base64Data;
+      }
+
+      return buffer;
+    } catch (error) {
+      throw new InternalServerErrorException('Error downloading media message', error.toString());
+    }
+  }
+
   public async fetchBusinessProfile(number: string): Promise<NumberBusiness> {
     try {
       const jid = number ? createJid(number) : this.instance.wuid;

src/api/routes/chat.router.ts

@@ -3,6 +3,7 @@ import {
   ArchiveChatDto,
   BlockUserDto,
   DeleteMessage,
+  DownloadMediaMessageDto,
   getBase64FromMediaMessageDto,
   MarkChatUnreadDto,
   NumberDto,
@@ -24,6 +25,7 @@ import {
   blockUserSchema,
   contactValidateSchema,
   deleteMessageSchema,
+  downloadMediaMessageSchema,
   markChatUnreadSchema,
   messageUpSchema,
   messageValidateSchema,
@@ -267,6 +269,16 @@ export class ChatRouter extends RouterBroker {
         });
 
         return res.status(HttpStatus.CREATED).json(response);
+      })
+      .post(this.routerPath('downloadMediaMessage'), ...guards, async (req, res) => {
+        const response = await this.dataValidate<DownloadMediaMessageDto>({
+          request: req,
+          schema: downloadMediaMessageSchema,
+          ClassRef: DownloadMediaMessageDto,
+          execute: (instance, data) => chatController.downloadMediaMessage(instance, data),
+        });
+
+        return res.status(HttpStatus.OK).json(response);
       });
   }
 

src/utils/readStreamWithTimeout.ts

@@ -0,0 +1,22 @@
+import { promisify } from 'util';
+
+export async function readStreamWithTimeout(stream: any, timeout: number = 30_000) {
+  const setTimeoutPromise = promisify(setTimeout);
+
+  let buffer = Buffer.from([]);
+  const chunks = [];
+  const timer: any = setTimeoutPromise(timeout).then(() => {
+    stream.destroy(new Error('Stream timeout'));
+  });
+
+  try {
+    for await (const chunk of stream) {
+      chunks.push(chunk);
+    }
+    buffer = Buffer.concat(chunks);
+  } finally {
+    clearTimeout(timer);
+  }
+
+  return buffer;
+}

src/validate/chat.schema.ts

@@ -287,6 +287,26 @@ export const privacySettingsSchema: JSONSchema7 = {
   ...isNotEmpty('readreceipts', 'profile', 'status', 'online', 'last', 'groupadd'),
 };
 
+export const downloadMediaMessageSchema: JSONSchema7 = {
+  $id: v4(),
+  type: 'object',
+  properties: {
+    downloadableMessage: {
+      type: 'object',
+      properties: {
+        url: { type: 'string' },
+        mediaKey: { type: 'string' },
+        directPath: { type: 'string' },
+      },
+      ...isNotEmpty('url', 'mediaKey', 'directPath'),
+    },
+    type: { type: 'string', enum: ['image', 'video', 'audio', 'document', 'sticker', 'ptt'] },
+    timeout: { type: 'number', default: 30_000 },
+    returnType: { type: 'string', enum: ['base64', 'buffer'], default: 'buffer' },
+  },
+  ...isNotEmpty('downloadableMessage', 'type'),
+};
+
 export const profileNameSchema: JSONSchema7 = {
   $id: v4(),
   type: 'object',