@peopl-health/nexus 1.0.3 → 1.1.1

package/lib/controllers/messageController.js

@@ -0,0 +1,218 @@
+const { ScheduledMessage } = require('../models/agendaMessageModel.js');
+
+const { getRecordByFilter } = require('../services/airtableService.js');
+const { sendScheduledMessage } = require('../services/whatsappService.js');
+
+const moment = require('moment-timezone');
+
+const sendMessageController = async (req, res) => {
+  const {
+    fileUrl,
+    message,
+    fileType,
+    code,
+    sendTime = new Date(),
+    timeZone = 'Etc/GMT',
+    hidePreview = false,
+    contentSid = null,
+    variables = null
+  } = req.body || {};
+  const author = process.env.USER_DB_MONGO;
+  const sendMoment = sendTime ? moment.tz(sendTime, timeZone) + 2500 : new Date();
+
+  try {
+    const scheduledMessage = new ScheduledMessage({
+      fileUrl,
+      message,
+      fileType,
+      timeZone: timeZone === '' ? null : timeZone,
+      sendTime: sendMoment,
+      contentSid: contentSid,
+      hidePreview,
+      code,
+      author,
+      extraDelay: 0,
+      variables
+    });
+    await scheduledMessage.save();
+    console.log(scheduledMessage);
+    const sentMessage = await sendScheduledMessage(scheduledMessage);
+    res.status(200).json({ 
+      status: 200, 
+      response: 'Message scheduled to be sent once!',
+      messageId: sentMessage ? sentMessage.sid : null
+    });
+  } catch (err) {
+    console.error('Error scheduling individual message:', err.message);
+    res.status(500).json({ status: false, error: err.message });
+  }
+};
+
+const sendBulkMessageController = async (req, res) => {
+  const {
+    fileUrl,
+    message,
+    fileType,
+    codes,
+    sendTime = new Date(),
+    timeZone = 'Etc/GMT',
+    hidePreview = false,
+    contentSid = null,
+    variables = null
+  } = req.body || {};
+  const author = process.env.USER_DB_MONGO;
+  const sendMoment = sendTime ? moment.tz(sendTime, timeZone) + 20*1000 : new Date();
+
+  try {
+    let numSend = 0;
+    let extraDelay = 0;
+    let curMessage = message;
+    const scheduledMessages = [];
+    for (const code of codes) {
+      const scheduledMessage = new ScheduledMessage({
+        fileUrl,
+        message: curMessage,
+        fileType,
+        timeZone: timeZone === '' ? null : timeZone,
+        sendTime: new Date(sendMoment + extraDelay),
+        contentSid: contentSid,
+        hidePreview,
+        code,
+        author,
+        extraDelay,
+        variables
+      });
+
+      // Add to scheduledMessages for both saving and scheduling
+      scheduledMessages.push(scheduledMessage);
+
+      // Increment delay and message counter
+      extraDelay += Math.floor(Math.random() * 5001) + 5000;
+      numSend += 1;
+    }
+
+    // Schedule all messages with Agenda in parallel
+    const sentMessages = await Promise.all(
+      scheduledMessages.map(async (message) => {
+        const savedMessage = await ScheduledMessage.create(message);
+        return sendScheduledMessage(savedMessage);
+      })
+    );
+
+    console.log(`Send bulk of ${numSend} messages`);
+
+    // Extract message IDs
+    const messageIds = sentMessages.map(msg => msg ? msg.sid : null).filter(id => id !== null);
+
+    res.status(200).json({ 
+      status: true, 
+      response: 'Bulk message sent',
+      messageIds: messageIds
+    });
+  } catch (err) {
+    console.log(err.message);
+    res.status(500).send(err.message);
+  }
+};
+
+const sendBulkMessageAirtableController = async (req, res) => {
+  const {
+    fileUrl,
+    message,
+    fileType,
+    baseId,
+    tableName,
+    columnPhone,
+    sendTime = new Date(),
+    timeZone = 'Etc/GMT',
+    hidePreview = false,
+    contentSid = null,
+    condition = '1',
+    variables = null
+  } = req.body || {};
+  const author = process.env.USER_DB_MONGO;
+  const sendMoment = sendTime ? moment.tz(sendTime, timeZone) + 20*1000 : new Date();
+
+  const regex = /\[(.*?)\]/g;
+  const envVariables = [];
+  let match;
+  while ((match = regex.exec(message)) !== null) {
+    envVariables.push(match[1]);
+  }
+
+  try {
+    const rows = await getRecordByFilter(baseId, tableName, condition);
+    let extraDelay = 0;
+    let curMessage = message;
+    const sentPhones = new Set();
+    const scheduledMessages = [];
+
+    for (const row of rows) {
+      let customMessage = curMessage;
+            
+      for (const envVar of envVariables) {
+        let value = row[envVar];
+        if (Array.isArray(value)) value = value[0];
+        if (value !== undefined) {
+          customMessage = customMessage.replace(`[${envVar}]`, value);
+        }
+      }
+
+      let code = row[columnPhone];
+      if (Array.isArray(code)) code = code[0];
+      if (!code) continue;
+      if (!code.includes('@g.us')) code = `${code}@s.whatsapp.net`;
+
+      if (sentPhones.has(code)) continue;
+      sentPhones.add(code);
+
+      // Prepare message object
+      const scheduledMessage = {
+        fileUrl,
+        message: customMessage,
+        fileType,
+        timeZone: timeZone === '' ? null : timeZone,
+        sendTime: new Date(sendMoment + extraDelay),
+        contentSid: contentSid,
+        hidePreview,
+        code,
+        author,
+        variables
+      };
+            
+      // Add to scheduledMessages for both saving and scheduling
+      scheduledMessages.push(scheduledMessage);
+
+      // Increment delay and message counter
+      extraDelay += Math.floor(Math.random() * 5001) + 5000;
+    }
+
+    // Schedule all messages with Agenda in parallel
+    const sentMessages = await Promise.all(
+      scheduledMessages.map(async (message) => {
+        const savedMessage = await ScheduledMessage.create(message);
+        return sendScheduledMessage(savedMessage);
+      })
+    );
+
+    // Extract message IDs
+    const messageIds = sentMessages.map(msg => msg ? msg.sid : null).filter(id => id !== null);
+
+    console.log(`Iterate over ${rows.length} rows`);
+
+    res.status(200).json({ 
+      status: true, 
+      response: 'Airtable message sent',
+      messageIds: messageIds
+    });
+  } catch (err) {
+    console.log(err.message);
+    res.status(500).send(err.message);
+  }
+};
+
+module.exports = {
+  sendMessageController,
+  sendBulkMessageController,
+  sendBulkMessageAirtableController
+};