From fb93c890caccddeef69e848d1c562c2bd25f5fdb Mon Sep 17 00:00:00 2001
From: Davidson Gomes <davidsongviolao@gmail.com>
Date: Thu, 15 Aug 2024 20:24:57 -0300
Subject: [PATCH] feat: dify send images

---
 .../dify/services/dify.service.ts             | 371 +++++++++++++++---
 1 file changed, 315 insertions(+), 56 deletions(-)

diff --git a/src/api/integrations/dify/services/dify.service.ts b/src/api/integrations/dify/services/dify.service.ts
index 597232b7..7a2cc6ba 100644
--- a/src/api/integrations/dify/services/dify.service.ts
+++ b/src/api/integrations/dify/services/dify.service.ts
@@ -1118,14 +1118,51 @@ export class DifyService {
 
       const message = response?.data?.answer;
 
-      await instance.textMessage(
-        {
-          number: remoteJid.split('@')[0],
-          delay: settings?.delayMessage || 1000,
-          text: message,
-        },
-        false,
-      );
+      const regex = /!?\[(.*?)\]\((.*?)\)/g;
+
+      const result = [];
+      let lastIndex = 0;
+
+      let match;
+      while ((match = regex.exec(message)) !== null) {
+        if (match.index > lastIndex) {
+          result.push({ text: message.slice(lastIndex, match.index).trim() });
+        }
+
+        result.push({ caption: match[1], url: match[2] });
+
+        lastIndex = regex.lastIndex;
+      }
+
+      if (lastIndex < message.length) {
+        result.push({ text: message.slice(lastIndex).trim() });
+      }
+
+      for (const item of result) {
+        if (item.text) {
+          await instance.textMessage(
+            {
+              number: remoteJid.split('@')[0],
+              delay: settings?.delayMessage || 1000,
+              text: item.text,
+            },
+            false,
+          );
+        }
+
+        if (item.url) {
+          await instance.mediaMessage(
+            {
+              number: remoteJid.split('@')[0],
+              delay: settings?.delayMessage || 1000,
+              mediatype: 'image',
+              media: item.url,
+              caption: item.caption,
+            },
+            false,
+          );
+        }
+      }
 
       await this.prismaRepository.integrationSession.update({
         where: {
@@ -1186,14 +1223,51 @@ export class DifyService {
 
       const message = response?.data?.answer;
 
-      await instance.textMessage(
-        {
-          number: remoteJid.split('@')[0],
-          delay: settings?.delayMessage || 1000,
-          text: message,
-        },
-        false,
-      );
+      const regex = /!?\[(.*?)\]\((.*?)\)/g;
+
+      const result = [];
+      let lastIndex = 0;
+
+      let match;
+      while ((match = regex.exec(message)) !== null) {
+        if (match.index > lastIndex) {
+          result.push({ text: message.slice(lastIndex, match.index).trim() });
+        }
+
+        result.push({ caption: match[1], url: match[2] });
+
+        lastIndex = regex.lastIndex;
+      }
+
+      if (lastIndex < message.length) {
+        result.push({ text: message.slice(lastIndex).trim() });
+      }
+
+      for (const item of result) {
+        if (item.text) {
+          await instance.textMessage(
+            {
+              number: remoteJid.split('@')[0],
+              delay: settings?.delayMessage || 1000,
+              text: item.text,
+            },
+            false,
+          );
+        }
+
+        if (item.url) {
+          await instance.mediaMessage(
+            {
+              number: remoteJid.split('@')[0],
+              delay: settings?.delayMessage || 1000,
+              mediatype: 'image',
+              media: item.url,
+              caption: item.caption,
+            },
+            false,
+          );
+        }
+      }
 
       await this.prismaRepository.integrationSession.update({
         where: {
@@ -1274,14 +1348,51 @@ export class DifyService {
 
         const message = response?.data?.answer;
 
-        await instance.textMessage(
-          {
-            number: remoteJid.split('@')[0],
-            delay: settings?.delayMessage || 1000,
-            text: message,
-          },
-          false,
-        );
+        const regex = /!?\[(.*?)\]\((.*?)\)/g;
+
+        const result = [];
+        let lastIndex = 0;
+
+        let match;
+        while ((match = regex.exec(message)) !== null) {
+          if (match.index > lastIndex) {
+            result.push({ text: message.slice(lastIndex, match.index).trim() });
+          }
+
+          result.push({ caption: match[1], url: match[2] });
+
+          lastIndex = regex.lastIndex;
+        }
+
+        if (lastIndex < message.length) {
+          result.push({ text: message.slice(lastIndex).trim() });
+        }
+
+        for (const item of result) {
+          if (item.text) {
+            await instance.textMessage(
+              {
+                number: remoteJid.split('@')[0],
+                delay: settings?.delayMessage || 1000,
+                text: item.text,
+              },
+              false,
+            );
+          }
+
+          if (item.url) {
+            await instance.mediaMessage(
+              {
+                number: remoteJid.split('@')[0],
+                delay: settings?.delayMessage || 1000,
+                mediatype: 'image',
+                media: item.url,
+                caption: item.caption,
+              },
+              false,
+            );
+          }
+        }
 
         await this.prismaRepository.integrationSession.update({
           where: {
@@ -1346,14 +1457,51 @@ export class DifyService {
 
       const message = response?.data?.data.outputs.text;
 
-      await instance.textMessage(
-        {
-          number: remoteJid.split('@')[0],
-          delay: settings?.delayMessage || 1000,
-          text: message,
-        },
-        false,
-      );
+      const regex = /!?\[(.*?)\]\((.*?)\)/g;
+
+      const result = [];
+      let lastIndex = 0;
+
+      let match;
+      while ((match = regex.exec(message)) !== null) {
+        if (match.index > lastIndex) {
+          result.push({ text: message.slice(lastIndex, match.index).trim() });
+        }
+
+        result.push({ caption: match[1], url: match[2] });
+
+        lastIndex = regex.lastIndex;
+      }
+
+      if (lastIndex < message.length) {
+        result.push({ text: message.slice(lastIndex).trim() });
+      }
+
+      for (const item of result) {
+        if (item.text) {
+          await instance.textMessage(
+            {
+              number: remoteJid.split('@')[0],
+              delay: settings?.delayMessage || 1000,
+              text: item.text,
+            },
+            false,
+          );
+        }
+
+        if (item.url) {
+          await instance.mediaMessage(
+            {
+              number: remoteJid.split('@')[0],
+              delay: settings?.delayMessage || 1000,
+              mediatype: 'image',
+              media: item.url,
+              caption: item.caption,
+            },
+            false,
+          );
+        }
+      }
 
       if (settings.keepOpen) {
         await this.prismaRepository.integrationSession.update({
@@ -1523,14 +1671,51 @@ export class DifyService {
 
       const message = response?.data?.answer;
 
-      await instance.textMessage(
-        {
-          number: remoteJid.split('@')[0],
-          delay: settings?.delayMessage || 1000,
-          text: message,
-        },
-        false,
-      );
+      const regex = /!?\[(.*?)\]\((.*?)\)/g;
+
+      const result = [];
+      let lastIndex = 0;
+
+      let match;
+      while ((match = regex.exec(message)) !== null) {
+        if (match.index > lastIndex) {
+          result.push({ text: message.slice(lastIndex, match.index).trim() });
+        }
+
+        result.push({ caption: match[1], url: match[2] });
+
+        lastIndex = regex.lastIndex;
+      }
+
+      if (lastIndex < message.length) {
+        result.push({ text: message.slice(lastIndex).trim() });
+      }
+
+      for (const item of result) {
+        if (item.text) {
+          await instance.textMessage(
+            {
+              number: remoteJid.split('@')[0],
+              delay: settings?.delayMessage || 1000,
+              text: item.text,
+            },
+            false,
+          );
+        }
+
+        if (item.url) {
+          await instance.mediaMessage(
+            {
+              number: remoteJid.split('@')[0],
+              delay: settings?.delayMessage || 1000,
+              mediatype: 'image',
+              media: item.url,
+              caption: item.caption,
+            },
+            false,
+          );
+        }
+      }
 
       await this.prismaRepository.integrationSession.update({
         where: {
@@ -1591,14 +1776,51 @@ export class DifyService {
 
       const message = response?.data?.answer;
 
-      await instance.textMessage(
-        {
-          number: remoteJid.split('@')[0],
-          delay: settings?.delayMessage || 1000,
-          text: message,
-        },
-        false,
-      );
+      const regex = /!?\[(.*?)\]\((.*?)\)/g;
+
+      const result = [];
+      let lastIndex = 0;
+
+      let match;
+      while ((match = regex.exec(message)) !== null) {
+        if (match.index > lastIndex) {
+          result.push({ text: message.slice(lastIndex, match.index).trim() });
+        }
+
+        result.push({ caption: match[1], url: match[2] });
+
+        lastIndex = regex.lastIndex;
+      }
+
+      if (lastIndex < message.length) {
+        result.push({ text: message.slice(lastIndex).trim() });
+      }
+
+      for (const item of result) {
+        if (item.text) {
+          await instance.textMessage(
+            {
+              number: remoteJid.split('@')[0],
+              delay: settings?.delayMessage || 1000,
+              text: item.text,
+            },
+            false,
+          );
+        }
+
+        if (item.url) {
+          await instance.mediaMessage(
+            {
+              number: remoteJid.split('@')[0],
+              delay: settings?.delayMessage || 1000,
+              mediatype: 'image',
+              media: item.url,
+              caption: item.caption,
+            },
+            false,
+          );
+        }
+      }
 
       await this.prismaRepository.integrationSession.update({
         where: {
@@ -1758,14 +1980,51 @@ export class DifyService {
 
       const message = response?.data?.data.outputs.text;
 
-      await instance.textMessage(
-        {
-          number: remoteJid.split('@')[0],
-          delay: settings?.delayMessage || 1000,
-          text: message,
-        },
-        false,
-      );
+      const regex = /!?\[(.*?)\]\((.*?)\)/g;
+
+      const result = [];
+      let lastIndex = 0;
+
+      let match;
+      while ((match = regex.exec(message)) !== null) {
+        if (match.index > lastIndex) {
+          result.push({ text: message.slice(lastIndex, match.index).trim() });
+        }
+
+        result.push({ caption: match[1], url: match[2] });
+
+        lastIndex = regex.lastIndex;
+      }
+
+      if (lastIndex < message.length) {
+        result.push({ text: message.slice(lastIndex).trim() });
+      }
+
+      for (const item of result) {
+        if (item.text) {
+          await instance.textMessage(
+            {
+              number: remoteJid.split('@')[0],
+              delay: settings?.delayMessage || 1000,
+              text: item.text,
+            },
+            false,
+          );
+        }
+
+        if (item.url) {
+          await instance.mediaMessage(
+            {
+              number: remoteJid.split('@')[0],
+              delay: settings?.delayMessage || 1000,
+              mediatype: 'image',
+              media: item.url,
+              caption: item.caption,
+            },
+            false,
+          );
+        }
+      }
 
       if (settings.keepOpen) {
         await this.prismaRepository.integrationSession.update({