From c1494ca03515cbfeea52ca89752930865984b9d5 Mon Sep 17 00:00:00 2001
From: Aditya Nandwana <adityanandwana19@gmail.com>
Date: Sat, 1 Feb 2025 16:04:35 +0530
Subject: [PATCH] Refactor logical message deletion in BaileysStartupService

---
 .../channel/whatsapp/whatsapp.baileys.service.ts | 16 +++++++++++-----
 1 file changed, 11 insertions(+), 5 deletions(-)

diff --git a/src/api/integrations/channel/whatsapp/whatsapp.baileys.service.ts b/src/api/integrations/channel/whatsapp/whatsapp.baileys.service.ts
index 2de39434..a5602ab6 100644
--- a/src/api/integrations/channel/whatsapp/whatsapp.baileys.service.ts
+++ b/src/api/integrations/channel/whatsapp/whatsapp.baileys.service.ts
@@ -3551,25 +3551,31 @@ export class BaileysStartupService extends ChannelStartupService {
         const messageId = response.message?.protocolMessage?.key?.id;
         if (messageId) {
           const isLogicalDeleted = configService.get<Database>('DATABASE').DELETE_DATA.LOGICAL_MESSAGE_DELETE;
-          let message = await this.prismaRepository.message.findUnique({
-            where: { id: messageId },
+          let message = await this.prismaRepository.message.findFirst({
+            where: {
+              key: {
+                path: ['id'],
+                equals: messageId,
+              },
+            },
           });
           if (isLogicalDeleted) {
             if (!message) return response;
             const existingKey = typeof message?.key === 'object' && message.key !== null ? message.key : {};
             message = await this.prismaRepository.message.update({
-              where: { id: messageId },
+              where: { id: message.id },
               data: {
                 key: {
                   ...existingKey,
                   deleted: true,
                 },
+                status: 'DELETED',
               },
             });
           } else {
             await this.prismaRepository.message.deleteMany({
               where: {
-                id: messageId,
+                id: message.id,
               },
             });
           }
@@ -3578,7 +3584,7 @@ export class BaileysStartupService extends ChannelStartupService {
             instanceId: message.instanceId,
             key: message.key,
             messageType: message.messageType,
-            status: message.status,
+            status: 'DELETED',
             source: message.source,
             messageTimestamp: message.messageTimestamp,
             pushName: message.pushName,