@goldenqueen/bai 1.0.0

package/lib/Socket/setup.js

@@ -0,0 +1,433 @@
+const WAProto = require('../../WAProto').proto;
+const crypto = require('crypto');
+
+class NotForrAll {
+    constructor(utils, waUploadToServer, relayMessageFn) {
+        this.utils = utils;
+        this.relayMessage = relayMessageFn
+        this.waUploadToServer = waUploadToServer;
+    }
+
+    detectType(content) {
+        if (content.requestPaymentMessage) return 'PAYMENT';
+        if (content.productMessage) return 'PRODUCT';
+        if (content.interactiveMessage) return 'INTERACTIVE';
+        if (content.albumMessage) return 'ALBUM';
+        if (content.eventMessage) return 'EVENT';
+        if (content.pollResultMessage) return 'POLL_RESULT'
+        return null;
+    }
+
+    async handlePayment(content, quoted) {
+        const data = content.requestPaymentMessage;
+        let notes = {};
+
+        if (data.sticker?.stickerMessage) {
+            notes = {
+                stickerMessage: {
+                    ...data.sticker.stickerMessage,
+                    contextInfo: {
+                        stanzaId: quoted?.key?.id,
+                        participant: quoted?.key?.participant || content.sender,
+                        quotedMessage: quoted?.message
+                    }
+                }
+            };
+        } else if (data.note) {
+            notes = {
+                extendedTextMessage: {
+                    text: data.note,
+                    contextInfo: {
+                        stanzaId: quoted?.key?.id,
+                        participant: quoted?.key?.participant || content.sender,
+                        quotedMessage: quoted?.message
+                    }
+                }
+            };
+        }
+
+        return {
+            requestPaymentMessage: WAProto.Message.RequestPaymentMessage.fromObject({
+                expiryTimestamp: data.expiry || 0,
+                amount1000: data.amount || 0,
+                currencyCodeIso4217: data.currency || "IDR",
+                requestFrom: data.from || "0@s.whatsapp.net",
+                noteMessage: notes,
+                background: data.background ?? {
+                    id: "DEFAULT",
+                    placeholderArgb: 0xFFF0F0F0
+                }
+            })
+        };
+    }
+        
+        async handleProduct(content, jid, quoted) {
+            const {
+                title, 
+                description, 
+                thumbnail,
+                productId, 
+                retailerId, 
+                url, 
+                body = "", 
+                footer = "", 
+                buttons = [],
+            } = 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,
+                                        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) {
+                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 }
+                    );
+                } 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;
+                    }
+                }
+        
+                media = await this.utils.prepareWAMessageMedia(
+                    documentPayload,
+                    { 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 || "" }
+            };
+            
+            if (buttons && buttons.length > 0) {
+                interactiveMessage.nativeFlowMessage = {
+                    buttons: buttons
+                };
+       
+                if (nativeFlowMessage) {
+                    interactiveMessage.nativeFlowMessage = {
+                        ...interactiveMessage.nativeFlowMessage,
+                        ...nativeFlowMessage
+                    };
+                }
+            } else if (nativeFlowMessage) {
+                interactiveMessage.nativeFlowMessage = nativeFlowMessage;
+            }
+            
+            if (media) {
+                interactiveMessage.header = {
+                    title: "",
+                    hasMediaAttachment: true,
+                    ...media
+                };
+            } else {
+                interactiveMessage.header = {
+                    title: "",
+                    hasMediaAttachment: false
+                };
+            }
+
+            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,
+                });
+                
+                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: "NotForrAll",
+                    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;
+        }   
+ 
+        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: "Yaoii — Baileys",
+                                    newsletterJid: "120363374309117436@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
+                        }
+                    }
+                }
+            }, { 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;
+        }
+    }
+
+module.exports = NotForrAll;

package/lib/Socket/setup.ts

@@ -0,0 +1,219 @@
+// setup.d.ts
+import { proto } from '../../WAProto';
+
+declare namespace NotForrAll {
+    interface MediaUploadOptions {
+        fileEncSha256?: Buffer;
+        mediaType?: string;
+        newsletter?: boolean;
+    }
+
+    type WAMediaUploadFunction = (
+        stream: Buffer | NodeJS.ReadableStream, 
+        options?: MediaUploadOptions
+    ) => Promise<{ url: string; directPath: string }>;
+
+    interface WAMessageContentGenerationOptions {
+        upload?: WAMediaUploadFunction;
+        mediaCache?: any;
+        options?: any;
+        logger?: any;
+    }
+
+    interface StickerMessage {
+        url: string;
+        fileSha256: Buffer | string;
+        fileEncSha256: Buffer | string;
+        mediaKey: Buffer | string;
+        mimetype: string;
+        directPath: string;
+        fileLength: number | string;
+        mediaKeyTimestamp: number | string;
+        isAnimated?: boolean;
+        stickerSentTs?: number | string;
+        isAvatar?: boolean;
+        isAiSticker?: boolean;
+        isLottie?: boolean;
+    }
+
+    interface PaymentMessage {
+        amount: number;
+        currency?: string;
+        from?: string;
+        expiry?: number;
+        sticker?: { stickerMessage: StickerMessage };
+        note?: string;
+        background?: {
+            id?: string;
+            fileLength?: string;
+            width?: number;
+            height?: number;
+            mimetype?: string;
+            placeholderArgb?: number;
+            textArgb?: number;
+            subtextArgb?: number;
+        };
+    }
+
+    interface ProductMessage {
+        title: string;
+        description: string;
+        thumbnail: Buffer | { url: string };
+        productId: string;
+        retailerId: string;
+        url: string;
+        body?: string;
+        footer?: string;
+        buttons?: proto.Message.InteractiveMessage.INativeFlowButton[];
+        priceAmount1000?: number | null;
+        currencyCode?: string;
+    }
+
+    interface InteractiveMessage {
+        title: string;
+        footer?: string;
+        thumbnail?: string;
+        image?: string | Buffer | { url: string };
+        video?: string | Buffer | { url: string };
+        document?: Buffer;
+        mimetype?: string;
+        fileName?: string;
+        jpegThumbnail?: Buffer; // Hanya Buffer saja
+        contextInfo?: {
+            mentionedJid?: string[];
+            forwardingScore?: number;
+            isForwarded?: boolean;
+            externalAdReply?: {
+                title?: string;
+                body?: string;
+                mediaType?: number;
+                thumbnailUrl?: string;
+                mediaUrl?: string;
+                sourceUrl?: string;
+                showAdAttribution?: boolean;
+                renderLargerThumbnail?: boolean;
+                [key: string]: any;
+            };
+            [key: string]: any;
+        };
+        externalAdReply?: {
+            title?: string;
+            body?: string;
+            mediaType?: number;
+            thumbnailUrl?: string;
+            mediaUrl?: string;
+            sourceUrl?: string;
+            showAdAttribution?: boolean;
+            renderLargerThumbnail?: boolean;
+            [key: string]: any;
+        };
+        buttons?: proto.Message.InteractiveMessage.INativeFlowButton[];
+        nativeFlowMessage?: {
+            messageParamsJson?: string;
+            buttons?: proto.Message.InteractiveMessage.INativeFlowButton[];
+            [key: string]: any;
+        };
+    }
+
+    interface AlbumItem {
+        image?: { url: string; caption?: string };
+        video?: { url: string; caption?: string };
+    }
+
+    interface EventMessageLocation {
+        degreesLatitude: number;
+        degreesLongitude: number;
+        name: string;
+    }
+
+    interface EventMessage {
+        isCanceled?: boolean;
+        name: string;
+        description: string;
+        location?: EventMessageLocation;
+        joinLink?: string;
+        startTime?: string | number;
+        endTime?: string | number;
+        extraGuestsAllowed?: boolean;
+    }
+    
+    interface PollVote {
+        optionName: string;
+        optionVoteCount: string | number;
+    }
+    
+    interface PollResultMessage {
+        name: string;
+        pollVotes: PollVote[];
+    }
+ 
+    interface MessageContent {
+        requestPaymentMessage?: PaymentMessage;
+        productMessage?: ProductMessage;
+        interactiveMessage?: InteractiveMessage;
+        albumMessage?: AlbumItem[];
+        eventMessage?: EventMessage;
+        pollResultMessage?: PollResultMessage;
+        sender?: string;
+    }
+
+    interface MessageOptions {
+        quoted?: proto.IWebMessageInfo;
+        filter?: boolean;
+    }
+
+    interface Utils {
+        prepareWAMessageMedia: (media: any, options: WAMessageContentGenerationOptions) => Promise<any>;
+        generateWAMessageContent: (content: any, options: WAMessageContentGenerationOptions) => Promise<any>;
+        generateWAMessageFromContent: (jid: string, content: any, options?: any) => Promise<any>;
+        generateWAMessage: (jid: string, content: any, options?: any) => Promise<any>;
+        generateMessageID: () => string;
+    }
+}
+
+declare class NotForrAll {
+    constructor(
+        utils: NotForrAll.Utils,
+        waUploadToServer: NotForrAll.WAMediaUploadFunction,
+        relayMessageFn?: (jid: string, content: any, options?: any) => Promise<any>
+    );
+    
+    detectType(content: NotForrAll.MessageContent): 'PAYMENT' | 'PRODUCT' | 'INTERACTIVE' | 'ALBUM' | 'EVENT' | 'POLL_RESULT' | null;
+
+    handlePayment(
+        content: { requestPaymentMessage: NotForrAll.PaymentMessage },
+        quoted?: proto.IWebMessageInfo
+    ): Promise<{ requestPaymentMessage: proto.Message.RequestPaymentMessage }>;
+
+    handleProduct(
+        content: { productMessage: NotForrAll.ProductMessage },
+        jid: string,
+        quoted?: proto.IWebMessageInfo
+    ): Promise<{ viewOnceMessage: proto.Message.ViewOnceMessage }>;
+
+    handleInteractive(
+        content: { interactiveMessage: NotForrAll.InteractiveMessage },
+        jid: string,
+        quoted?: proto.IWebMessageInfo
+    ): Promise<{ interactiveMessage: proto.Message.InteractiveMessage }>;
+
+    handleAlbum(
+        content: { albumMessage: NotForrAll.AlbumItem[] },
+        jid: string,
+        quoted?: proto.IWebMessageInfo
+    ): Promise<any>;
+
+    handleEvent(
+        content: { eventMessage: NotForrAll.EventMessage },
+        jid: string,
+        quoted?: proto.IWebMessageInfo
+    ): Promise<any>;
+    
+    handlePollResult(
+        content: { pollResultMessage: NotForrAll.PollResultMessage },
+        jid: string,
+        quoted?: proto.IWebMessageInfo
+    ): Promise<any>;
+}
+
+export = NotForrAll;

package/lib/Socket/socket.d.ts

@@ -0,0 +1,43 @@
+import { Boom } from '@hapi/boom';
+import { SocketConfig } from '../Types';
+import { BinaryNode } from '../WABinary';
+import { WebSocketClient } from './Client';
+/**
+ * Connects to WA servers and performs:
+ * - simple queries (no retry mechanism, wait for connection establishment)
+ * - listen to messages and emit events
+ * - query phone connection
+ */
+export declare const makeSocket: (config: SocketConfig) => {
+    type: "md";
+    ws: WebSocketClient;
+    ev: import("../Types").BaileysEventEmitter & {
+        process(handler: (events: Partial<import("../Types").BaileysEventMap>) => void | Promise<void>): (() => void);
+        buffer(): void;
+        createBufferedFunction<A extends any[], T>(work: (...args: A) => Promise<T>): ((...args: A) => Promise<T>);
+        flush(force?: boolean): boolean;
+        isBuffering(): boolean;
+    };
+    authState: {
+        creds: import("../Types").AuthenticationCreds;
+        keys: import("../Types").SignalKeyStoreWithTransaction;
+    };
+    signalRepository: import("../Types").SignalRepository;
+    readonly user: import("../Types").Contact | undefined;
+    generateMessageTag: () => string;
+    query: (node: BinaryNode, timeoutMs?: number) => Promise<any>;
+    waitForMessage: <T>(msgId: string, timeoutMs?: number | undefined) => Promise<any>;
+    waitForSocketOpen: () => Promise<void>;
+    sendRawMessage: (data: Uint8Array | Buffer) => Promise<void>;
+    sendNode: (frame: BinaryNode) => Promise<void>;
+    logout: (msg?: string) => Promise<void>;
+    end: (error: Error | undefined) => void;
+    onUnexpectedError: (err: Error | Boom, msg: string) => void;
+    uploadPreKeys: (count?: number) => Promise<void>;
+    uploadPreKeysToServerIfRequired: () => Promise<void>;
+    requestPairingCode: (phoneNumber: any, pairKey?: string) => Promise<string>;
+    /** Waits for the connection to WA to reach a state */
+    waitForConnectionUpdate: (check: (u: Partial<import("../Types").ConnectionState>) => Promise<boolean | undefined>, timeoutMs?: number) => Promise<void>;
+    sendWAMBuffer: (wamBuffer: Buffer) => Promise<any>;
+};
+export type Socket = ReturnType<typeof makeSocket>;