@alannxd/baileys 2.1.4 → 3.0.0

package/lib/Socket/dugong.js

@@ -4,12 +4,19 @@
 
 const WAProto = require('../../WAProto').proto;
 const crypto = require('crypto');
+const Utils_1 = require("../Utils");
 
 class kikyy {
     constructor(utils, waUploadToServer, relayMessageFn) {
         this.utils = utils;
         this.relayMessage = relayMessageFn
         this.waUploadToServer = waUploadToServer;
+        
+        this.bail = {
+            generateWAMessageContent: this.utils.generateWAMessageContent || Utils_1.generateWAMessageContent,
+            generateMessageID: Utils_1.generateMessageID,
+            getContentType: (msg) => Object.keys(msg.message || {})[0]
+        };
     }
 
     detectType(content) {
@@ -18,7 +25,8 @@ class kikyy {
         if (content.interactiveMessage) return 'INTERACTIVE';
         if (content.albumMessage) return 'ALBUM';
         if (content.eventMessage) return 'EVENT';
-        if (content.pollResultMessage) return 'POLL_RESULT'
+        if (content.pollResultMessage) return 'POLL_RESULT';
+        if (content.groupStatusMessage) return 'GROUP_STORY';
         return null;
     }
 
@@ -65,377 +73,412 @@ class kikyy {
         };
     }
         
-        async handleProduct(content, jid, quoted) {
-            const {
-                title, 
-                description, 
-                thumbnail,
-                productId, 
-                retailerId, 
-                url, 
-                body = "", 
-                footer = "", 
-                buttons = [],
-                priceAmount1000 = null,
-                currencyCode = "IDR"
-            } = content.productMessage;
-    
-            let productImage;
+    async handleProduct(content, jid, quoted) {
+        const {
+            title, 
+            description, 
+            thumbnail,
+            productId, 
+            retailerId, 
+            url, 
+            body = "", 
+            footer = "", 
+            buttons = [],
+            priceAmount1000 = null,
+            currencyCode = "IDR"
+        } = content.productMessage;
+
+        let productImage;
+
+        if (Buffer.isBuffer(thumbnail)) {
+            const { imageMessage } = await this.utils.generateWAMessageContent(
+                { image: thumbnail }, 
+                { upload: this.waUploadToServer }
+            );
+            productImage = imageMessage;
+        } else if (typeof thumbnail === 'object' && thumbnail.url) {
+            const { imageMessage } = await this.utils.generateWAMessageContent(
+                { image: { url: thumbnail.url }}, 
+                { upload: this.waUploadToServer }
+            );
+            productImage = imageMessage;
+        }
+
+        return {
+            viewOnceMessage: {
+                message: {
+                    interactiveMessage: {
+                        body: { text: body },
+                        footer: { text: footer },
+                        header: {
+                            title,
+                            hasMediaAttachment: true,
+                            productMessage: {
+                                product: {
+                                    productImage,
+                                    productId,
+                                    title,
+                                    description,
+                                    currencyCode,
+                                    priceAmount1000,
+                                    retailerId,
+                                    url,
+                                    productImageCount: 1
+                                },
+                                businessOwnerJid: "0@s.whatsapp.net"
+                            }
+                        },
+                        nativeFlowMessage: { buttons }
+                    }
+                }
+            }
+        };
+    }
     
-            if (Buffer.isBuffer(thumbnail)) {
-                const { imageMessage } = await this.utils.generateWAMessageContent(
-                    { image: thumbnail }, 
+    async handleInteractive(content, jid, quoted) {
+        const {
+            title,
+            footer,
+            thumbnail,
+            image,
+            video,
+            document,
+            mimetype,
+            fileName,
+            jpegThumbnail,
+            contextInfo,
+            externalAdReply,
+            buttons = [],
+            nativeFlowMessage,
+            header
+        } = content.interactiveMessage;
+
+        let media = null;
+        let mediaType = null;
+
+        if (thumbnail) {
+            media = await this.utils.prepareWAMessageMedia(
+                { image: { url: thumbnail } },
+                { upload: this.waUploadToServer }
+            );
+            mediaType = 'image';
+        } else if (image) {
+            if (typeof image === 'object' && image.url) {
+                media = await this.utils.prepareWAMessageMedia(
+                    { image: { url: image.url } },
                     { upload: this.waUploadToServer }
                 );
-                productImage = imageMessage;
-            } else if (typeof thumbnail === 'object' && thumbnail.url) {
-                const { imageMessage } = await this.utils.generateWAMessageContent(
-                    { image: { url: thumbnail.url }}, 
+            } else {
+                media = await this.utils.prepareWAMessageMedia(
+                    { image: image },
                     { upload: this.waUploadToServer }
                 );
-                productImage = imageMessage;
             }
-    
-            return {
-                viewOnceMessage: {
-                    message: {
-                        interactiveMessage: {
-                            body: { text: body },
-                            footer: { text: footer },
-                            header: {
-                                title,
-                                hasMediaAttachment: true,
-                                productMessage: {
-                                    product: {
-                                        productImage,
-                                        productId,
-                                        title,
-                                        description,
-                                        currencyCode,
-                                        priceAmount1000,
-                                        retailerId,
-                                        url,
-                                        productImageCount: 1
-                                    },
-                                    businessOwnerJid: "0@s.whatsapp.net"
-                                }
-                            },
-                            nativeFlowMessage: { buttons }
-                        }
-                    }
-                }
-            };
-        }
-        
-        async handleInteractive(content, jid, quoted) {
-            const {
-                title,
-                footer,
-                thumbnail,
-                image,
-                video,
-                document,
-                mimetype,
-                fileName,
-                jpegThumbnail,
-                contextInfo,
-                externalAdReply,
-                buttons = [],
-                nativeFlowMessage
-            } = content.interactiveMessage;
-            
-            let media = null;
-            let mediaType = null;
-            
-            if (thumbnail) {
+            mediaType = 'image';
+        } else if (video) {
+            if (typeof video === 'object' && video.url) {
                 media = await this.utils.prepareWAMessageMedia(
-                    { image: { url: thumbnail } },
+                    { video: { url: video.url } },
                     { upload: this.waUploadToServer }
                 );
-                mediaType = 'image';
-            } else if (image) {
-                if (typeof image === 'object' && image.url) {
-                    media = await this.utils.prepareWAMessageMedia(
-                        { image: { url: image.url } },
-                        { upload: this.waUploadToServer }
-                    );
-                } else {
-                    media = await this.utils.prepareWAMessageMedia(
-                        { image: image },
-                        { upload: this.waUploadToServer }
-                    );
-                }
-                mediaType = 'image';
-            } else if (video) {
-                if (typeof video === 'object' && video.url) {
-                    media = await this.utils.prepareWAMessageMedia(
-                        { video: { url: video.url } },
-                        { upload: this.waUploadToServer }
-                    );
-                } else {
-                    media = await this.utils.prepareWAMessageMedia(
-                        { video: video },
-                        { upload: this.waUploadToServer }
-                    );
-                }
-                mediaType = 'video';
-            } else if (document) {
-                let documentPayload = { document: document };
-                
-                if (jpegThumbnail) {
-                    if (typeof jpegThumbnail === 'object' && jpegThumbnail.url) {
-                        documentPayload.jpegThumbnail = { url: jpegThumbnail.url };
-                    } else {
-                        documentPayload.jpegThumbnail = jpegThumbnail;
-                    }
-                }
-        
+            } else {
                 media = await this.utils.prepareWAMessageMedia(
-                    documentPayload,
+                    { video: video },
                     { upload: this.waUploadToServer }
                 );
-        
-                if (fileName) {
-                    media.documentMessage.fileName = fileName;
-                }
-                if (mimetype) {
-                    media.documentMessage.mimetype = mimetype;
-                }
-                mediaType = 'document';
             }
-            
-            let interactiveMessage = {
-                body: { text: title || "" },
-                footer: { text: footer || "" }
+            mediaType = 'video';
+        } else if (document) {
+            let documentPayload = { 
+                document: document 
             };
-            
-            if (buttons && buttons.length > 0) {
-                interactiveMessage.nativeFlowMessage = {
-                    buttons: buttons
-                };
-       
-                if (nativeFlowMessage) {
-                    interactiveMessage.nativeFlowMessage = {
-                        ...interactiveMessage.nativeFlowMessage,
-                        ...nativeFlowMessage
-                    };
+            if (jpegThumbnail) {
+                if (typeof jpegThumbnail === 'object' && jpegThumbnail.url) {
+                    documentPayload.jpegThumbnail = { url: jpegThumbnail.url };
+                } else {
+                    documentPayload.jpegThumbnail = jpegThumbnail;
                 }
-            } else if (nativeFlowMessage) {
-                interactiveMessage.nativeFlowMessage = nativeFlowMessage;
             }
             
-            if (media) {
-                interactiveMessage.header = {
-                    title: "",
-                    hasMediaAttachment: true,
-                    ...media
-                };
-            } else {
-                interactiveMessage.header = {
-                    title: "",
-                    hasMediaAttachment: false
-                };
+            media = await this.utils.prepareWAMessageMedia(
+                documentPayload,
+                { upload: this.waUploadToServer }
+            );
+            if (fileName) {
+                media.documentMessage.fileName = fileName;
             }
-
-            let finalContextInfo = {};
-            
-            if (contextInfo) {
-                finalContextInfo = {
-                    mentionedJid: contextInfo.mentionedJid || [],
-                    forwardingScore: contextInfo.forwardingScore || 0,
-                    isForwarded: contextInfo.isForwarded || false,
-                    ...contextInfo
-                };
+            if (mimetype) {
+                media.documentMessage.mimetype = mimetype;
             }
-
-            if (externalAdReply) {
-                finalContextInfo.externalAdReply = {
-                    title: externalAdReply.title || "",
-                    body: externalAdReply.body || "",
-                    mediaType: externalAdReply.mediaType || 1,
-                    thumbnailUrl: externalAdReply.thumbnailUrl || "",
-                    mediaUrl: externalAdReply.mediaUrl || "",
-                    sourceUrl: externalAdReply.sourceUrl || "",
-                    showAdAttribution: externalAdReply.showAdAttribution || false,
-                    renderLargerThumbnail: externalAdReply.renderLargerThumbnail || false,
-                    ...externalAdReply
+            mediaType = 'document';
+        }
+        let interactiveMessage = {
+            body: { text: title || "" },
+            footer: { text: footer || "" }
+        };
+        if (buttons && buttons.length > 0) {
+            interactiveMessage.nativeFlowMessage = {
+                buttons: buttons
+            };
+            if (nativeFlowMessage) {
+                interactiveMessage.nativeFlowMessage = {
+                    ...interactiveMessage.nativeFlowMessage,
+                    ...nativeFlowMessage
                 };
             }
-            
-            if (Object.keys(finalContextInfo).length > 0) {
-                interactiveMessage.contextInfo = finalContextInfo;
-            }
-
-            return {
-                interactiveMessage: interactiveMessage
+        } else if (nativeFlowMessage) {
+            interactiveMessage.nativeFlowMessage = nativeFlowMessage;
+        }
+        
+        if (media) {
+            interactiveMessage.header = {
+                title: header || "",
+                hasMediaAttachment: true,
+                ...media
+            };
+        } else {
+            interactiveMessage.header = {
+                title: header || "",        
+                hasMediaAttachment: false
             };
         }
         
-        async handleAlbum(content, jid, quoted) {
-            const array = content.albumMessage;
-            const album = await this.utils.generateWAMessageFromContent(jid, {
-                messageContextInfo: {
-                    messageSecret: crypto.randomBytes(32),
-                },
-                albumMessage: {
-                    expectedImageCount: array.filter((a) => a.hasOwnProperty("image")).length,
-                    expectedVideoCount: array.filter((a) => a.hasOwnProperty("video")).length,
-                },
-            }, {
-                userJid: this.utils.generateMessageID().split('@')[0] + '@s.whatsapp.net',
-                quoted,
-                upload: this.waUploadToServer
+        let finalContextInfo = {};
+        if (contextInfo) {
+            finalContextInfo = {
+                mentionedJid: contextInfo.mentionedJid || [],
+                forwardingScore: contextInfo.forwardingScore || 0,
+                isForwarded: contextInfo.isForwarded || false,
+                ...contextInfo
+            };
+        }
+        
+        if (externalAdReply) {
+            finalContextInfo.externalAdReply = {
+                title: externalAdReply.title || "",
+                body: externalAdReply.body || "",
+                mediaType: externalAdReply.mediaType || 1,
+                thumbnailUrl: externalAdReply.thumbnailUrl || "",
+                mediaUrl: externalAdReply.mediaUrl || "",
+                sourceUrl: externalAdReply.sourceUrl || "",
+                showAdAttribution: externalAdReply.showAdAttribution || false,
+                renderLargerThumbnail: externalAdReply.renderLargerThumbnail || false,
+                ...externalAdReply
+            };
+        }
+        
+        if (Object.keys(finalContextInfo).length > 0) {
+            interactiveMessage.contextInfo = finalContextInfo;
+        }
+        return {
+            interactiveMessage: interactiveMessage
+        };
+    }
+    
+    async handleAlbum(content, jid, quoted) {
+        const array = content.albumMessage;
+        const album = await this.utils.generateWAMessageFromContent(jid, {
+            messageContextInfo: {
+                messageSecret: crypto.randomBytes(32),
+            },
+            albumMessage: {
+                expectedImageCount: array.filter((a) => a.hasOwnProperty("image")).length,
+                expectedVideoCount: array.filter((a) => a.hasOwnProperty("video")).length,
+            },
+        }, {
+            userJid: this.utils.generateMessageID().split('@')[0] + '@s.whatsapp.net',
+            quoted,
+            upload: this.waUploadToServer
+        });
+        
+        await this.relayMessage(jid, album.message, {
+            messageId: album.key.id,
+        });
+        
+        for (let content of array) {
+            const img = await this.utils.generateWAMessage(jid, content, {
+                upload: this.waUploadToServer,
             });
             
-            await this.relayMessage(jid, album.message, {
-                messageId: album.key.id,
-            });
+            img.message.messageContextInfo = {
+                messageSecret: crypto.randomBytes(32),
+                messageAssociation: {
+                    associationType: 1,
+                    parentMessageKey: album.key,
+                },    
+                participant: "0@s.whatsapp.net",
+                remoteJid: "status@broadcast",
+                forwardingScore: 99999,
+                isForwarded: true,
+                mentionedJid: [jid],
+                starred: true,
+                labels: ["Y", "Important"],
+                isHighlighted: true,
+                businessMessageForwardInfo: {
+                    businessOwnerJid: jid,
+                },
+                dataSharingContext: {
+                    showMmDisclosure: true,
+                },
+            };
+
+            img.message.forwardedNewsletterMessageInfo = {
+                newsletterJid: "0@newsletter",
+                serverMessageId: 1,
+                newsletterName: `WhatsApp`,
+                contentType: 1,
+                timestamp: new Date().toISOString(),
+                senderName: "kikyy dugonggg",
+                content: "Text Message",
+                priority: "high",
+                status: "sent",
+            };
             
-            for (let content of array) {
-                const img = await this.utils.generateWAMessage(jid, content, {
-                    upload: this.waUploadToServer,
-                });
-                
-                img.message.messageContextInfo = {
-                    messageSecret: crypto.randomBytes(32),
-                    messageAssociation: {
-                        associationType: 1,
-                        parentMessageKey: album.key,
-                    },    
-                    participant: "0@s.whatsapp.net",
-                    remoteJid: "status@broadcast",
-                    forwardingScore: 99999,
-                    isForwarded: true,
-                    mentionedJid: [jid],
-                    starred: true,
-                    labels: ["Y", "Important"],
-                    isHighlighted: true,
-                    businessMessageForwardInfo: {
-                        businessOwnerJid: jid,
-                    },
-                    dataSharingContext: {
-                        showMmDisclosure: true,
-                    },
-                };
+            img.message.disappearingMode = {
+                initiator: 3,
+                trigger: 4,
+                initiatorDeviceJid: jid,
+                initiatedByExternalService: true,
+                initiatedByUserDevice: true,
+                initiatedBySystem: true,      
+                initiatedByServer: true,
+                initiatedByAdmin: true,
+                initiatedByUser: true,
+                initiatedByApp: true,
+                initiatedByBot: true,
+                initiatedByMe: true,
+            };
 
-                img.message.forwardedNewsletterMessageInfo = {
-                    newsletterJid: "0@newsletter",
-                    serverMessageId: 1,
-                    newsletterName: `WhatsApp`,
-                    contentType: 1,
-                    timestamp: new Date().toISOString(),
-                    senderName: "kikyy dugonggg",
-                    content: "Text Message",
-                    priority: "high",
-                    status: "sent",
-                };
-                
-                img.message.disappearingMode = {
-                    initiator: 3,
-                    trigger: 4,
-                    initiatorDeviceJid: jid,
-                    initiatedByExternalService: true,
-                    initiatedByUserDevice: true,
-                    initiatedBySystem: true,      
-                    initiatedByServer: true,
-                    initiatedByAdmin: true,
-                    initiatedByUser: true,
-                    initiatedByApp: true,
-                    initiatedByBot: true,
-                    initiatedByMe: true,
-                };
+            await this.relayMessage(jid, img.message, {
+                messageId: img.key.id,
+                quoted: {
+                    key: {
+                        remoteJid: album.key.remoteJid,
+                        id: album.key.id,
+                        fromMe: true,
+                        participant: this.utils.generateMessageID().split('@')[0] + '@s.whatsapp.net',
+                    },
+                    message: album.message,
+                },
+            });
+        }
+        return album;
+    }   
 
-                await this.relayMessage(jid, img.message, {
-                    messageId: img.key.id,
-                    quoted: {
-                        key: {
-                            remoteJid: album.key.remoteJid,
-                            id: album.key.id,
-                            fromMe: true,
-                            participant: this.utils.generateMessageID().split('@')[0] + '@s.whatsapp.net',
-                        },
-                        message: album.message,
+    async handleEvent(content, jid, quoted) {
+        const eventData = content.eventMessage;
+        
+        const msg = await this.utils.generateWAMessageFromContent(jid, {
+            viewOnceMessage: {
+                message: {
+                    messageContextInfo: {
+                        deviceListMetadata: {},
+                        deviceListMetadataVersion: 2,
+                        messageSecret: crypto.randomBytes(32),
+                        supportPayload: JSON.stringify({
+                            version: 2,
+                            is_ai_message: true,
+                            should_show_system_message: true,
+                            ticket_id: crypto.randomBytes(16).toString('hex')
+                        })
                     },
-                });
-            }
-            return album;
-        }   
- 
-        async handleEvent(content, jid, quoted) {
-            const eventData = content.eventMessage;
-            
-            const msg = await this.utils.generateWAMessageFromContent(jid, {
-                viewOnceMessage: {
-                    message: {
-                        messageContextInfo: {
-                            deviceListMetadata: {},
-                            deviceListMetadataVersion: 2,
-                            messageSecret: crypto.randomBytes(32),
-                            supportPayload: JSON.stringify({
-                                version: 2,
-                                is_ai_message: true,
-                                should_show_system_message: true,
-                                ticket_id: crypto.randomBytes(16).toString('hex')
-                            })
+                    eventMessage: {
+                        contextInfo: {
+                            mentionedJid: [jid],
+                            participant: jid,
+                            remoteJid: "status@broadcast",
+                            forwardedNewsletterMessageInfo: {
+                                newsletterName: "shenvn.",
+                                newsletterJid: "120363297591152843@newsletter",
+                                serverMessageId: 1
+                            }
+                        },
+                        isCanceled: eventData.isCanceled || false,
+                        name: eventData.name,
+                        description: eventData.description,
+                        location: eventData.location || {
+                            degreesLatitude: 0,
+                            degreesLongitude: 0,
+                            name: "Location"
                         },
-                        eventMessage: {
-                            contextInfo: {
-                                mentionedJid: [jid],
-                                participant: jid,
-                                remoteJid: "status@broadcast",
-                                forwardedNewsletterMessageInfo: {
-                                    newsletterName: "shenvn.",
-                                    newsletterJid: "120363297591152843@newsletter",
-                                    serverMessageId: 1
-                                }
-                            },
-                            isCanceled: eventData.isCanceled || false,
-                            name: eventData.name,
-                            description: eventData.description,
-                            location: eventData.location || {
-                                degreesLatitude: 0,
-                                degreesLongitude: 0,
-                                name: "Location"
-                            },
-                            joinLink: eventData.joinLink || '',
-                            startTime: typeof eventData.startTime === 'string' ? parseInt(eventData.startTime) : eventData.startTime || Date.now(),
-                            endTime: typeof eventData.endTime === 'string' ? parseInt(eventData.endTime) : eventData.endTime || Date.now() + 3600000,
-                            extraGuestsAllowed: eventData.extraGuestsAllowed !== false
-                        }
+                        joinLink: eventData.joinLink || '',
+                        startTime: typeof eventData.startTime === 'string' ? parseInt(eventData.startTime) : eventData.startTime || Date.now(),
+                        endTime: typeof eventData.endTime === 'string' ? parseInt(eventData.endTime) : eventData.endTime || Date.now() + 3600000,
+                        extraGuestsAllowed: eventData.extraGuestsAllowed !== false
                     }
                 }
-            }, { quoted });
-            
-            await this.relayMessage(jid, msg.message, {
-                messageId: msg.key.id
-            });
-            return msg;
-        }
-        
-        async handlePollResult(content, jid, quoted) {
-            const pollData = content.pollResultMessage;
+            }
+        }, { quoted });
         
-            const msg = await this.utils.generateWAMessageFromContent(jid, {
-                pollResultSnapshotMessage: {
-                    name: pollData.name,
-                    pollVotes: pollData.pollVotes.map(vote => ({
-                        optionName: vote.optionName,
-                        optionVoteCount: typeof vote.optionVoteCount === 'number' 
-                        ? vote.optionVoteCount.toString() 
-                        : vote.optionVoteCount
-                    }))
-                }
-            }, {
-                userJid: this.utils.generateMessageID().split('@')[0] + '@s.whatsapp.net',
-                quoted
-            });
+        await this.relayMessage(jid, msg.message, {
+            messageId: msg.key.id
+        });
+        return msg;
+    }
+    
+    async handlePollResult(content, jid, quoted) {
+        const pollData = content.pollResultMessage;
+    
+        const msg = await this.utils.generateWAMessageFromContent(jid, {
+            pollResultSnapshotMessage: {
+                name: pollData.name,
+                pollVotes: pollData.pollVotes.map(vote => ({
+                    optionName: vote.optionName,
+                    optionVoteCount: typeof vote.optionVoteCount === 'number' 
+                    ? vote.optionVoteCount.toString() 
+                    : vote.optionVoteCount
+                }))
+            }
+        }, {
+            userJid: this.utils.generateMessageID().split('@')[0] + '@s.whatsapp.net',
+            quoted
+        });
+    
+        await this.relayMessage(jid, msg.message, {
+            messageId: msg.key.id
+        });
+
+        return msg;
+    }
+
+    async handleGroupStory(content, jid, quoted) {
+        const storyData = content.groupStatusMessage;
+        let waMsgContent;
         
-            await this.relayMessage(jid, msg.message, {
-                messageId: msg.key.id
-            });
-       
-            return msg;
+        if (storyData.message) {
+            waMsgContent = storyData;
+        } else {
+            if (typeof this.bail?.generateWAMessageContent === "function") {
+                waMsgContent = await this.bail.generateWAMessageContent(storyData, {
+                    upload: this.waUploadToServer
+                });
+            } else if (typeof this.utils?.generateWAMessageContent === "function") {
+                waMsgContent = await this.utils.generateWAMessageContent(storyData, {
+                    upload: this.waUploadToServer
+                });
+            } else if (typeof this.utils?.prepareMessageContent === "function") {
+                waMsgContent = await this.utils.prepareMessageContent(storyData, {
+                    upload: this.waUploadToServer
+                });
+            } else {
+                waMsgContent = await Utils_1.generateWAMessageContent(storyData, {
+                    upload: this.waUploadToServer
+                });
+            }
         }
+
+        let msg = {
+            message: {
+                groupStatusMessageV2: {
+                    message: waMsgContent.message || waMsgContent
+                }
+            }
+        };
+
+        return await this.relayMessage(jid, msg.message, {
+            messageId: this.bail.generateMessageID()
+        });
     }
+}
 
-module.exports = kikyy;
+module.exports = kikyy;