kage-baileys 3.0.6 → 3.2.3

package/lib/Utils/automation-tools.js

@@ -8,7 +8,7 @@ function unwrapNormalMedia(raw) {
   let msg = raw?.message;
   for (let i = 0; i < 12 && msg; i++) {
     if (msg.viewOnceMessage || msg.viewOnceMessageV2 || msg.viewOnceMessageV2Extension) throw new Error('VIEW_ONCE_PROTECTED: pide al remitente una copia normal con su consentimiento.');
-    const wrapper = ['ephemeralMessage', 'documentWithCaptionMessage'].find(k => msg[k]?.message);
+    const wrapper = ['deviceSentMessage', 'ephemeralMessage', 'documentWithCaptionMessage'].find(k => msg[k]?.message);
     if (wrapper) { msg = msg[wrapper].message; continue; }
     for (const type of ['imageMessage', 'videoMessage', 'audioMessage', 'documentMessage', 'stickerMessage']) {
       if (!msg[type]) continue;
@@ -33,16 +33,32 @@ async function downloadNormalMedia(conn, raw, options = {}) {
   const chunks = [];
   let size = 0;
   const job = (async () => {
-    stream = await download({ ...raw, message: media.message }, 'stream', { options: { timeout: timeoutMs } }, { logger, reuploadRequest: conn.updateMediaMessage?.bind(conn) });
-    if (stopped) { stream.destroy?.(); throw new Error('MEDIA_TIMEOUT'); }
-    for await (const chunk of stream) {
-      if (stopped) throw new Error('MEDIA_TIMEOUT');
-      const data = Buffer.from(chunk); size += data.length;
-      if (size > maxBytes) throw new Error('MEDIA_TOO_LARGE: se interrumpió al alcanzar el límite.');
-      chunks.push(data);
+    let target = { ...raw, message: media.message };
+    for (let attempt = 0; attempt < 2; attempt++) {
+      chunks.length = 0; size = 0;
+      try {
+        stream = await download(target, 'stream', { options: { timeout: timeoutMs } }, { logger, reuploadRequest: conn.updateMediaMessage?.bind(conn) });
+        if (stopped) { stream.destroy?.(); throw new Error('MEDIA_TIMEOUT'); }
+        for await (const chunk of stream) {
+          if (stopped) throw new Error('MEDIA_TIMEOUT');
+          const data = Buffer.from(chunk); size += data.length;
+          if (size > maxBytes) throw new Error('MEDIA_TOO_LARGE: se interrumpió al alcanzar el límite.');
+          chunks.push(data);
+        }
+        if (!size) throw new Error('EMPTY_MEDIA: WhatsApp no entregó datos.');
+        return { buffer: Buffer.concat(chunks, size), type: media.type, mimetype: media.value.mimetype || 'application/octet-stream', caption: media.value.caption || '', bytes: size };
+      } catch (error) {
+        stream?.destroy?.();
+        const code = error?.output?.statusCode || error?.response?.status || error?.statusCode;
+        const retry = [403,404,408,410,429,500,502,503,504].includes(Number(code)) || /EMPTY_MEDIA|ECONNRESET|ETIMEDOUT|ENOTFOUND/.test(error?.message || error?.code || '');
+        if (attempt || stopped || !retry) throw error;
+        if ([403,404,410].includes(Number(code)) && typeof conn.updateMediaMessage === 'function') {
+          const updated = await conn.updateMediaMessage(target);
+          if (updated?.message) target = updated;
+        }
+        if (stopped) throw new Error('MEDIA_TIMEOUT');
+      }
     }
-    if (!size) throw new Error('EMPTY_MEDIA: WhatsApp no entregó datos.');
-    return { buffer: Buffer.concat(chunks, size), type: media.type, mimetype: media.value.mimetype || 'application/octet-stream', caption: media.value.caption || '', bytes: size };
   })();
   try {
     return await Promise.race([job, new Promise((_, reject) => { timer = setTimeout(() => reject(new Error('MEDIA_TIMEOUT: tiempo de espera agotado.')), timeoutMs); })]);
@@ -96,7 +112,7 @@ function handleAutoMedia(raw, conn, settings = {}, options = {}) {
   while (state.seen.size > 500) state.seen.delete(state.seen.keys().next().value);
   state.pending++;
   const job = state.tail.then(() => saveNormalMedia(conn, raw, options));
-  state.tail = job.catch(() => {}).finally(() => { state.pending--; });
+  state.tail = job.catch(() => { state.seen.delete(id); }).finally(() => { state.pending--; });
   return job;
 }
 async function sendAIResponse(conn, jid, text, options = {}) {
@@ -113,6 +129,12 @@ function installKageTools(conn, options = {}) {
   Object.defineProperty(conn, '__kageToolsInstalled', { value: true });
   const status = require('./status-tools');
   const country = require('./country-info');
+  conn.inspectMessage = raw => require('./message-inspector').inspectMessage(raw);
+  conn.getMessageDiagnostics = () => require('./message-inspector').getMessageDiagnostics(conn);
+  conn.enableEventTracking = opts => require('./message-inspector').installMessageInspector(conn, opts);
+  conn.refreshReceivedStatuses = opts => require('./status-refresh').refreshReceivedStatuses(conn, opts);
+  if (options.eventTracking) conn.enableEventTracking(options.eventOptions || {});
+  if (options.banner) require('./terminal-banner').showKageBanner();
   conn.getReceivedStatuses = (includeUnavailable = false) => status.listStatuses(conn, includeUnavailable);
   conn.getCountryForJid = jid => country.getCountryForJid(conn, jid);
   conn.downloadNormalMedia = (raw, opts) => downloadNormalMedia(conn, raw, opts);
@@ -129,17 +151,25 @@ function installKageTools(conn, options = {}) {
   conn.enableMediaAutomation = (settings = {}) => {
     if (conn.__kageStopMedia) return conn.__kageStopMedia;
     let active = true;
+    const recent = new Map();
     const scoped = new Proxy(settings, { get(target, key) { return !active && ['autoDownloadChats','autoViewOnceChats','autoViewOnceConsent'].includes(key) ? undefined : target[key]; } });
     const upsert = event => {
-      if (event?.type !== 'notify') return;
+      if (event?.type && !['notify', 'append'].includes(event.type)) return;
       for (const raw of event.messages || []) {
+        const id = JSON.stringify([raw.key?.remoteJid, raw.key?.id]);
+        if (raw.messageTimestamp) recent.set(id, raw.messageTimestamp);
+        while(recent.size > 1000) recent.delete(recent.keys().next().value);
         handleAutoMedia(raw, conn, scoped).catch(error => conn.logger?.warn?.({ error: error.message }, 'media automation'));
         require('./view-once-compat').handleAutoViewOnce(raw, conn, scoped).catch(error => conn.logger?.warn?.({ error: error.message }, 'view once automation'));
       }
     };
-    const stop = () => { active = false; conn.ev.off('messages.upsert', upsert); conn.ev.off('connection.update', closed); delete conn.__kageStopMedia; };
+    const updated = items => { for (const item of items || []) if (item.update?.message) {
+      const timestamp = item.update.messageTimestamp || recent.get(JSON.stringify([item.key?.remoteJid, item.key?.id]));
+      if (timestamp) upsert({ type: 'append', messages: [{ ...item.update, key: item.key, messageTimestamp: timestamp }] });
+    } };
+    const stop = () => { active = false; recent.clear(); conn.ev.off('messages.update', updated); conn.ev.off('messages.upsert', upsert); conn.ev.off('connection.update', closed); delete conn.__kageStopMedia; };
     const closed = update => { if (update.connection === 'close') stop(); };
-    conn.ev.on('messages.upsert', upsert); conn.ev.on('connection.update', closed);
+    conn.ev.on('messages.upsert', upsert); conn.ev.on('messages.update', updated); conn.ev.on('connection.update', closed);
     conn.__kageStopMedia = stop;
     return stop;
   };
@@ -151,11 +181,16 @@ function installKageTools(conn, options = {}) {
     let active = true;
     const scopedSettings = new Proxy(settings, { get(target, key) { return key === 'statusSave' ? active && target[key] : target[key]; } });
     const upsert = event => { for (const raw of event?.messages || []) status.handleStatus(raw, conn, scopedSettings).catch(e => conn.logger?.warn?.({ error: e.message }, 'status tracking')); };
+    const history = event => upsert(event);
     const deleted = event => { for (const key of event?.keys || []) status.markStatusDeleted(conn, key); };
-    const updated = items => { for (const item of items || []) if (item.key?.remoteJid === 'status@broadcast' && item.update?.message === null) status.markStatusDeleted(conn, item.key); };
+    const updated = items => { for (const item of items || []) {
+      if (item.key?.remoteJid === 'status@broadcast' && item.update?.message === null) status.markStatusDeleted(conn, item.key);
+      else if (item.update?.message) upsert({ messages: [{ ...item.update, key: item.key }] });
+    } }; 
     const closed = update => { if (update.connection === 'close') stop?.(); };
     const timer = setInterval(() => status.flushStatuses(conn), 60000); timer.unref?.();
     conn.ev.on('messages.upsert', upsert);
+    conn.ev.on('messaging-history.set', history);
     conn.ev.on('messages.delete', deleted);
     conn.ev.on('messages.update', updated);
     conn.ev.on('connection.update', closed);
@@ -163,7 +198,7 @@ function installKageTools(conn, options = {}) {
       if (!active) return;
       active = false;
       clearInterval(timer);
-      conn.ev.off('messages.upsert', upsert); conn.ev.off('messages.delete', deleted);
+      conn.ev.off('messages.upsert', upsert); conn.ev.off('messaging-history.set', history); conn.ev.off('messages.delete', deleted);
       conn.ev.off('messages.update', updated); conn.ev.off('connection.update', closed);
       status.flushStatuses(conn); stop = undefined;
     };

package/lib/Utils/baileys-event-stream.js

@@ -0,0 +1,146 @@
+"use strict"
+
+Object.defineProperty(exports, "__esModule", { value: true })
+
+const Events = require("events")
+const {
+  promises, 
+  createReadStream 
+} = require("fs")
+const { writeFile } = require("fs/promises")
+const { createInterface } = require("readline")
+const { delay } = require("./generics")
+const { makeMutex } = require("./make-mutex")
+
+/**
+ * Captures events from a baileys event emitter & stores them in a file
+ * @param ev The event emitter to read events from
+ * @param filename File to save to
+ */
+const captureEventStream = (ev, filename) => {
+    const oldEmit = ev.emit
+    // write mutex so data is appended in order
+    const writeMutex = makeMutex()
+    // monkey patch eventemitter to capture all events
+    ev.emit = function (...args) {
+        const content = JSON.stringify({ timestamp: Date.now(), event: args[0], data: args[1] }) + '\n'
+        const result = oldEmit.apply(ev, args)
+        writeMutex.mutex(async () => {
+            await promises.writeFile(filename, content, { flag: 'a' })
+        })
+        return result
+    }
+}
+
+/**
+ * Read event file and emit events from there
+ * @param filename filename containing event data
+ * @param delayIntervalMs delay between each event emit
+ */
+const readAndEmitEventStream = (filename, delayIntervalMs = 0) => {
+    const ev = new Events()
+    const fireEvents = async () => {
+        // from: https://stackoverflow.com/questions/6156501/read-a-file-one-line-at-a-time-in-node-js
+        const fileStream = createReadStream(filename)
+        const rl = createInterface({
+            input: fileStream,
+            crlfDelay: Infinity
+        })
+        // Note: we use the crlfDelay option to recognize all instances of CR LF
+        // ('\r\n') in input.txt as a single line break.
+        for await (const line of rl) {
+            if (line) {
+                const { event, data } = JSON.parse(line)
+                ev.emit(event, data)
+                delayIntervalMs && await delay(delayIntervalMs)
+            }
+        }
+        fileStream.close()
+    }
+    return {
+        ev,
+        task: fireEvents()
+    }
+}
+
+module.exports = {
+  captureEventStream, 
+  readAndEmitEventStream
+}
+// Kage: compatibilidad de API.
+const __kageModule58 = (() => {
+const module = { exports: {} }; const exports = module.exports;
+"use strict"
+
+Object.defineProperty(exports, "__esModule", { value: true })
+
+const Events = require("events")
+const {
+  promises, 
+  createReadStream 
+} = require("fs")
+const { writeFile } = require("fs/promises")
+const { createInterface } = require("readline")
+const { delay } = require("./generics")
+const { makeMutex } = require("./make-mutex")
+
+/**
+ * Captures events from a baileys event emitter & stores them in a file
+ * @param ev The event emitter to read events from
+ * @param filename File to save to
+ */
+const captureEventStream = (ev, filename) => {
+    const oldEmit = ev.emit
+    // write mutex so data is appended in order
+    const writeMutex = makeMutex()
+    // monkey patch eventemitter to capture all events
+    ev.emit = function (...args) {
+        const content = JSON.stringify({ timestamp: Date.now(), event: args[0], data: args[1] }) + '\n'
+        const result = oldEmit.apply(ev, args)
+        writeMutex.mutex(async () => {
+            await promises.writeFile(filename, content, { flag: 'a' })
+        })
+        return result
+    }
+}
+
+/**
+ * Read event file and emit events from there
+ * @param filename filename containing event data
+ * @param delayIntervalMs delay between each event emit
+ */
+const readAndEmitEventStream = (filename, delayIntervalMs = 0) => {
+    const ev = new Events()
+    const fireEvents = async () => {
+        // from: https://stackoverflow.com/questions/6156501/read-a-file-one-line-at-a-time-in-node-js
+        const fileStream = createReadStream(filename)
+        const rl = createInterface({
+            input: fileStream,
+            crlfDelay: Infinity
+        })
+        // Note: we use the crlfDelay option to recognize all instances of CR LF
+        // ('\r\n') in input.txt as a single line break.
+        for await (const line of rl) {
+            if (line) {
+                const { event, data } = JSON.parse(line)
+                ev.emit(event, data)
+                delayIntervalMs && await delay(delayIntervalMs)
+            }
+        }
+        fileStream.close()
+    }
+    return {
+        ev,
+        task: fireEvents()
+    }
+}
+
+module.exports = {
+  captureEventStream, 
+  readAndEmitEventStream
+}
+return module.exports;
+})();
+for (const key of Object.keys(__kageModule58)) {
+ if (key !== '__esModule' && !Object.prototype.hasOwnProperty.call(exports,key)) Object.defineProperty(exports,key,{enumerable:true,configurable:true,get:()=>__kageModule58[key]});
+}

package/lib/Utils/browser-utils.js

@@ -39,3 +39,98 @@ Object.assign(exports.Browsers, {
     ubuntu: browser => ['Ubuntu', browser || 'Chrome', '22.04.0'],
     appropriate: browser => exports.Browsers(browser || 'Chrome')
 });
+
+// Kage: compatibilidad de API.
+const __kageModule59 = (() => {
+const module = { exports: {} }; const exports = module.exports;
+"use strict"
+
+Object.defineProperty(exports, "__esModule", { value: true })
+
+const { 
+  platform, 
+  release 
+} = require("os") 
+const { proto } = require("../../WAProto")
+
+const PLATFORM_MAP = {
+    aix: 'AIX',
+    darwin: 'Mac OS',
+    win32: 'Windows',
+    android: 'Android',
+    freebsd: 'FreeBSD',
+    openbsd: 'OpenBSD',
+    sunos: 'Solaris',
+    linux: undefined,
+    haiku: undefined,
+    cygwin: undefined,
+    netbsd: undefined
+}
+
+const Browsers = {
+    ubuntu: browser => ['Ubuntu', browser, '22.04.4'],
+    macOS: browser => ['Mac OS', browser, '14.4.1'],
+    baileys: browser => ['Baileys', browser, '6.5.0'],
+    windows: browser => ['Windows', browser, '10.0.22631'],
+    //android: browser => [browser, 'Android', ''],
+    /** The appropriate browser based on your OS & release */
+    appropriate: browser => [PLATFORM_MAP[platform()] || 'Ubuntu', browser, release()]
+}
+
+const getPlatformId = (browser) => {
+    const platformType = proto.DeviceProps.PlatformType[browser.toUpperCase()]
+    return platformType ? platformType.toString() : '1' //chrome
+}
+
+module.exports = {
+  Browsers, 
+  getPlatformId
+}
+return module.exports;
+})();
+for (const key of Object.keys(__kageModule59)) {
+ if (key !== '__esModule' && !Object.prototype.hasOwnProperty.call(exports,key)) Object.defineProperty(exports,key,{enumerable:true,configurable:true,get:()=>__kageModule59[key]});
+}
+
+// Kage: funciones avanzadas.
+const __kageModule162 = (() => {
+const module = { exports: {} }; const exports = module.exports;
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.getPlatformId = exports.Browsers = void 0;
+const node_os_1 = require("node:os");
+const index_js_1 = require("../../WAProto/index.js");
+const PLATFORM_MAP = {
+    aix: 'AIX',
+    darwin: 'Mac OS',
+    win32: 'Windows',
+    android: 'Android',
+    freebsd: 'FreeBSD',
+    openbsd: 'OpenBSD',
+    sunos: 'Solaris',
+    linux: undefined,
+    haiku: undefined,
+    cygwin: undefined,
+    netbsd: undefined
+};
+exports.Browsers = {
+    ubuntu: browser => ['Ubuntu', browser, '22.04.4'],
+    macOS: browser => ['Mac OS', browser, '14.4.1'],
+    baileys: browser => ['Baileys', browser, '6.5.0'],
+    windows: browser => ['Windows', browser, '10.0.22631'],
+    android: browser => [browser, 'Android', ''],
+    /** The appropriate browser based on your OS & release */
+    appropriate: browser => [PLATFORM_MAP[(0, node_os_1.platform)()] || 'Ubuntu', browser, (0, node_os_1.release)()]
+};
+const getPlatformId = (browser) => {
+    const platformType = index_js_1.proto.DeviceProps.PlatformType[browser.toUpperCase()];
+    return platformType ? platformType.toString() : '1'; //chrome
+};
+exports.getPlatformId = getPlatformId;
+
+return module.exports;
+})();
+for (const key of Object.keys(__kageModule162)) {
+ if (key !== '__esModule' && !Object.prototype.hasOwnProperty.call(exports,key)) Object.defineProperty(exports,key,{enumerable:true,configurable:true,get:()=>__kageModule162[key]});
+}
+Object.defineProperty(exports, Symbol.for('kage.advanced'), {value:__kageModule162, configurable:true});