fix: Improovment newsletter to send poll

wppconnect-team · Apr 11, 2024 · ca4382f · ca4382f
1 parent a041474
commit ca4382f
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 5 deletions.
diff --git a/src/chat/functions/sendRawMessage.ts b/src/chat/functions/sendRawMessage.ts
@@ -67,18 +67,24 @@ export async function sendRawMessage(
 
   debug(`sending message (${rawMessage.type}) with id ${rawMessage.id}`);
   let result = null as any;
-  if (chat?.isNewsletter) {
-    if (rawMessage.type !== ('chat' || 'image' || 'video'))
+  if (chat?.isNewsletter && rawMessage.type) {
+    const validTypes = ['chat', 'image', 'video', 'poll_creation'];
+    if (!validTypes.includes(rawMessage.type))
       throw new WPPError(
         'type_not_valid_for_newsletter',
-        'Please, send a valid type for send message to newsletter. Valid types: "chat", "image", "video"'
+        'Please, send a valid type for send message to newsletter. Valid types: "chat", "image", "video", "poll_creation"'
       );
     const msg = new MsgModel(rawMessage as any);
     await addNewsletterMsgsRecords([await msgDataFromMsgModel(msg)]);
     const resultNewsletter = await sendNewsletterMessageJob({
+      type:
+        rawMessage.type == 'chat' && !rawMessage.editMsgType
+          ? 'text'
+          : rawMessage.type == 'poll_creation'
+            ? 'pollCreation'
+            : 'media',
       msgData: rawMessage,
       msg: new MsgModel(rawMessage as any),
-      type: rawMessage.type == 'chat' ? 'text' : 'media',
       newsletterJid: chat.id.toString(),
     });
     chat.msgs.add(msg);

diff --git a/src/whatsapp/functions/sendNewsletterMessageJob.ts b/src/whatsapp/functions/sendNewsletterMessageJob.ts
@@ -15,14 +15,16 @@
  */
 
 import { exportModule } from '../exportModule';
+import { MsgModel } from '../models';
 
 /** @whatsapp 397995
  */
 export declare function sendNewsletterMessageJob(data: {
   msgData?: any;
-  msg?: any;
+  msg?: MsgModel;
   newsletterJid: string;
   type: string;
+  editType?: 'media' | 'text';
 }): Promise<{ ack: { t: number }; serverId: number; success: boolean }>;
 
 exportModule(