Enhance message editing validation in BaileysStartupService

Author: matheusmartinsInsper

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

@@ -3341,6 +3341,16 @@ export class BaileysStartupService extends ChannelStartupService {
     }
 
     try {
+      const oldMessage: any = await this.getMessage(data.key, true);
+       if (!oldMessage) throw new NotFoundException('Message not found');
+       if (oldMessage?.key?.remoteJid !== jid) {
+         throw new BadRequestException('RemoteJid does not match');
+       }
+       if (oldMessage?.messageTimestamp > Date.now() + 900000) {
+         // 15 minutes in milliseconds
+         throw new BadRequestException('Message is older than 15 minutes');
+       }
+ 
       const response = await this.client.sendMessage(jid, {
         ...(options as any),
         edit: data.key,
@@ -3365,14 +3375,17 @@ export class BaileysStartupService extends ChannelStartupService {
             new BadRequestException('You cannot edit deleted messages');
           }
 
-          const updateMessage = this.prepareMessage({ ...response });
+          if (oldMessage.messageType === 'conversation' || oldMessage.messageType === 'extendedTextMessage') {
+            oldMessage.message.conversation = data.text;
+          } else {
+            oldMessage.message[oldMessage.messageType].caption = data.text;
+          }
           message = await this.prismaRepository.message.update({
             where: { id: message.id },
             data: {
-              message: {
-                ...updateMessage?.message?.[updateMessage.messageType]?.editedMessage,
-              },
+              message: oldMessage.message,
               status: 'EDITED',
+              messageTimestamp: Math.floor(Date.now() / 1000),
             },
           });
           const messageUpdate: any = {