@vanzxy/baileys 2.0.0

package/lib/Utils/message-search.js

@@ -0,0 +1,189 @@
+// Vanz@Add --- ported from Bail-master addons/message-search.ts (type-only
+// annotations dropped; behavior unchanged).
+// Vanz@Fix (bug 58): peel off ephemeral / view-once wrappers so search still
+// indexes text from disappearing-message chats instead of always seeing '{}'.
+const unwrapMessage = (content) => {
+    let c = content;
+    while (c && (c.ephemeralMessage || c.viewOnceMessage || c.viewOnceMessageV2 || c.viewOnceMessageV2Extension || c.documentWithCaptionMessage)) {
+        c = c.ephemeralMessage?.message ||
+            c.viewOnceMessage?.message ||
+            c.viewOnceMessageV2?.message ||
+            c.viewOnceMessageV2Extension?.message ||
+            c.documentWithCaptionMessage?.message;
+    }
+    return c;
+};
+export const extractMessageText = (message) => {
+    const c = unwrapMessage(message.message);
+    if (!c)
+        return '';
+    if (c.conversation)
+        return c.conversation;
+    if (c.extendedTextMessage?.text)
+        return c.extendedTextMessage.text;
+    if (c.imageMessage?.caption)
+        return c.imageMessage.caption;
+    if (c.videoMessage?.caption)
+        return c.videoMessage.caption;
+    if (c.documentMessage?.caption)
+        return c.documentMessage.caption;
+    if (c.documentMessage?.fileName)
+        return c.documentMessage.fileName;
+    if (c.locationMessage?.name)
+        return c.locationMessage.name;
+    if (c.locationMessage?.address)
+        return c.locationMessage.address;
+    if (c.contactMessage?.displayName)
+        return c.contactMessage.displayName;
+    if (c.pollCreationMessage?.name)
+        return c.pollCreationMessage.name;
+    return '';
+};
+const getMessageType = (message) => {
+    const c = unwrapMessage(message.message);
+    if (!c)
+        return 'other';
+    if (c.conversation || c.extendedTextMessage)
+        return 'text';
+    if (c.imageMessage)
+        return 'image';
+    if (c.videoMessage)
+        return 'video';
+    if (c.documentMessage)
+        return 'document';
+    if (c.audioMessage)
+        return 'audio';
+    if (c.stickerMessage)
+        return 'sticker';
+    if (c.locationMessage || c.liveLocationMessage)
+        return 'location';
+    if (c.contactMessage || c.contactsArrayMessage)
+        return 'contact';
+    return 'other';
+};
+export const calculateRelevance = (query, text, position) => {
+    let score = 100;
+    if (text.toLowerCase() === query.toLowerCase())
+        score += 50;
+    score -= Math.min(position / 10, 20);
+    const lt = text.toLowerCase(), lq = query.toLowerCase();
+    if (position === 0 ||
+        lt[position - 1] === ' ' ||
+        lt[position + lq.length] === ' ' ||
+        position + lq.length === text.length)
+        score += 20;
+    return Math.max(score, 0);
+};
+/** Plain-text substring search with relevance-sorted results. */
+export const searchMessages = (messages, query, options = {}) => {
+    const results = [];
+    const sq = options.caseSensitive ? query : query.toLowerCase();
+    for (const message of messages) {
+        if (options.jid && message.key.remoteJid !== options.jid)
+            continue;
+        const ts = message.messageTimestamp;
+        const mt = ts ? new Date((typeof ts === 'number' ? ts : Number(ts)) * 1000) : null;
+        if (options.fromDate && mt && mt < options.fromDate)
+            continue;
+        if (options.toDate && mt && mt > options.toDate)
+            continue;
+        if (options.fromSender && message.key.participant !== options.fromSender)
+            continue;
+        if (options.fromMe !== undefined && message.key.fromMe !== options.fromMe)
+            continue;
+        if (options.messageTypes?.length) {
+            if (!options.messageTypes.includes(getMessageType(message)))
+                continue;
+        }
+        const text = extractMessageText(message);
+        if (!text)
+            continue;
+        const st = options.caseSensitive ? text : text.toLowerCase();
+        const pos = st.indexOf(sq);
+        if (pos !== -1) {
+            results.push({
+                message,
+                matchedText: text.substring(Math.max(0, pos - 20), Math.min(text.length, pos + query.length + 20)),
+                matchPosition: pos,
+                relevanceScore: calculateRelevance(query, text, pos)
+            });
+        }
+        // Vanz@Fix (bug 59): limit used to break the scan loop *before* sorting,
+        // so it kept the first N matches in document order and could drop
+        // higher-relevance matches found later. Sort first, then slice.
+    }
+    results.sort((a, b) => b.relevanceScore - a.relevanceScore);
+    return options.limit ? results.slice(0, options.limit) : results;
+};
+/** Regex-based search (e.g. for commands/patterns), unordered by relevance. */
+export const searchMessagesRegex = (messages, pattern, options = {}) => {
+    const results = [];
+    for (const message of messages) {
+        if (options.jid && message.key.remoteJid !== options.jid)
+            continue;
+        if (options.fromSender && message.key.participant !== options.fromSender)
+            continue;
+        if (options.fromMe !== undefined && message.key.fromMe !== options.fromMe)
+            continue;
+        if (options.messageTypes?.length) {
+            if (!options.messageTypes.includes(getMessageType(message)))
+                continue;
+        }
+        const text = extractMessageText(message);
+        if (!text)
+            continue;
+        const match = text.match(pattern);
+        if (match)
+            results.push({ message, matchedText: match[0], matchPosition: match.index ?? 0, relevanceScore: 100 });
+        if (options.limit && results.length >= options.limit)
+            break;
+    }
+    return results;
+};
+/** Standalone searchable index — feed it messages independently of the main Store. */
+export class MessageSearchManager {
+    messages = [];
+    messageIndex = new Map();
+    addMessages(messages) {
+        for (const msg of messages) {
+            const id = msg.key.id;
+            if (id && !this.messageIndex.has(id)) {
+                this.messages.push(msg);
+                this.messageIndex.set(id, msg);
+            }
+        }
+    }
+    removeMessages(messageIds) {
+        const idSet = new Set(messageIds);
+        this.messages = this.messages.filter((m) => !idSet.has(m.key.id || ''));
+        for (const id of messageIds)
+            this.messageIndex.delete(id);
+    }
+    clear() {
+        this.messages = [];
+        this.messageIndex.clear();
+    }
+    get count() {
+        return this.messages.length;
+    }
+    search(query, options) {
+        return searchMessages(this.messages, query, options);
+    }
+    searchRegex(pattern, options) {
+        return searchMessagesRegex(this.messages, pattern, options);
+    }
+    getByJid(jid) {
+        return this.messages.filter((m) => m.key.remoteJid === jid);
+    }
+    getBySender(sender) {
+        return this.messages.filter((m) => m.key.participant === sender || m.key.remoteJid === sender);
+    }
+    getByType(type) {
+        return this.messages.filter((m) => getMessageType(m) === type);
+    }
+    getById(id) {
+        return this.messageIndex.get(id);
+    }
+}
+export const createMessageSearch = () => new MessageSearchManager();
+//# sourceMappingURL=message-search.js.map

package/lib/Utils/messages-media.d.ts

@@ -0,0 +1,135 @@
+/** generates all the keys required to encrypt/decrypt & sign a media message */
+export function getMediaKeys(buffer: any, mediaType: any): Promise<{
+    iv: any;
+    cipherKey: any;
+    macKey: any;
+}>;
+export function getAudioDuration(buffer: any): Promise<any>;
+/**
+  referenced from and modifying https://github.com/wppconnect-team/wa-js/blob/main/src/chat/functions/prepareAudioWaveform.ts
+ */
+export function getAudioWaveform(buffer: any, logger: any): Promise<Uint8Array<ArrayBuffer> | undefined>;
+/** generates a thumbnail for a given media, if required */
+export function generateThumbnail(file: any, mediaType: any, options: any): Promise<{
+    thumbnail: any;
+    originalImageDimensions: {
+        width: any;
+        height: any;
+    } | undefined;
+}>;
+export function extensionForMediaMessage(message: any): any;
+export function getImageProcessingLibrary(): Promise<any>;
+export function hkdfInfoKey(type: any): string;
+export function getRawMediaUploadData(media: any, mediaType: any, logger: any): Promise<{
+    filePath: any;
+    fileSha256: any;
+    fileLength: number;
+}>;
+export function extractVideoThumb(path: any, time: any, size: any): Promise<any>;
+export function extractImageThumb(bufferOrFilePath: any, width?: number): Promise<{
+    buffer: any;
+    original: {
+        width: any;
+        height: any;
+    };
+}>;
+export function encodeBase64EncodedStringForUpload(b64: any): string;
+export function generateProfilePicture(mediaUpload: any, dimensions: any): Promise<{
+    img: any;
+}>;
+export function mediaMessageSHA256B64(message: any): any;
+export function toReadable(buffer: any): any;
+export function toBuffer(stream: any): Promise<any>;
+export function getStream(item: any, opts: any): Promise<{
+    stream: any;
+    type: string;
+}>;
+export function getHttpStream(url: any, options?: {}): Promise<any>;
+export function encryptedStream(media: any, mediaType: any, { logger, saveOriginalFileIfRequired, opts }?: {}): Promise<{
+    mediaKey: any;
+    originalFilePath: any;
+    encFilePath: any;
+    mac: any;
+    fileEncSha256: any;
+    fileSha256: any;
+    fileLength: number;
+}>;
+export const DEF_MEDIA_HOST: "mmg.whatsapp.net";
+export function getUrlFromDirectPath(directPath: any, host?: string): string;
+export function downloadContentFromMessage({ mediaKey, directPath, url }: {
+    mediaKey: any;
+    directPath: any;
+    url: any;
+}, type: any, opts?: {}): Promise<any>;
+export function downloadEncryptedContent(downloadUrl: any, { cipherKey, iv }: {
+    cipherKey: any;
+    iv: any;
+}, { startByte, endByte, options }?: {}): Promise<any>;
+export function uploadWithNodeHttp({ url, filePath, headers, timeoutMs, agent }: {
+    url: any;
+    filePath: any;
+    headers: any;
+    timeoutMs: any;
+    agent: any;
+}, redirectCount?: number): Promise<any>;
+export function getWAUploadToServer({ customUploadHosts, fetchAgent, logger, options }: {
+    customUploadHosts: any;
+    fetchAgent: any;
+    logger: any;
+    options: any;
+}, refreshMediaConn: any): (filePath: any, { mediaType, fileEncSha256B64, timeoutMs, newsletter }: {
+    mediaType: any;
+    fileEncSha256B64: any;
+    timeoutMs: any;
+    newsletter: any;
+}) => Promise<{
+    mediaUrl: any;
+    directPath: any;
+    meta_hmac: any;
+    fbid: any;
+    ts: any;
+    thumbnailDirectPath: any;
+    thumbnailSha256: any;
+}>;
+export function encryptMediaRetryRequest(key: any, mediaKey: any, meId: any): {
+    tag: string;
+    attrs: {
+        id: any;
+        to: string;
+        type: string;
+    };
+    content: ({
+        tag: string;
+        attrs: {
+            jid?: undefined;
+            from_me?: undefined;
+            participant?: undefined;
+        };
+        content: {
+            tag: string;
+            attrs: {};
+            content: any;
+        }[];
+    } | {
+        tag: string;
+        attrs: {
+            jid: any;
+            from_me: string;
+            participant: any;
+        };
+        content?: undefined;
+    })[];
+};
+export function decodeMediaRetryNode(node: any): {
+    key: {
+        id: any;
+        remoteJid: any;
+        fromMe: boolean;
+        participant: any;
+    };
+};
+export function decryptMediaRetryData({ ciphertext, iv }: {
+    ciphertext: any;
+    iv: any;
+}, mediaKey: any, msgId: any): any;
+export function getStatusCodeForMediaRetry(code: any): number;