amiudmodz 5.0.0 → 5.0.1

package/engine-requirements.js

@@ -7,4 +7,4 @@ if (major < 20) {
     `   Please upgrade to Node.js 20+ to proceed.\n`
   );
   process.exit(1);
-}
+}

package/lib/Defaults/index.js

@@ -141,7 +141,7 @@ exports.MEDIA_HKDF_KEY_MAPPING = {
 exports.MEDIA_KEYS = Object.keys(exports.MEDIA_PATH_MAP);
 exports.MIN_PREKEY_COUNT = 5;
 exports.INITIAL_PREKEY_COUNT = 30;
-exports.UPLOAD_TIMEOUT = 600000;
+exports.UPLOAD_TIMEOUT = 7200000; // 2 hours in ms
 exports.MIN_UPLOAD_INTERVAL = 5000;
 
 exports.TimeMs = {

package/lib/Utils/messages-media.js

@@ -400,23 +400,41 @@ const prepareStream = async (media, mediaType, { logger, saveOriginalFileIfRequi
     logger === null || logger === void 0 ? void 0 : logger.debug('fetched media stream');
     let bodyPath;
     let didSaveToTmpPath = false;
+    let fileLength = 0;
+    const sha256 = Crypto.createHash('sha256');
+    const encWriteStream = new stream_1.PassThrough();
+    
+    // if it's already a file, we can use its path
+    if (type === 'file') {
+        bodyPath = media.url;
+    } else if (saveOriginalFileIfRequired) {
+        bodyPath = (0, path_1.join)(getTmpFilesDirectory(), mediaType + (0, generics_1.generateMessageID)());
+        didSaveToTmpPath = true;
+    }
+
+    const writeStream = bodyPath && !media.url ? (0, fs_1.createWriteStream)(bodyPath) : undefined;
+
     try {
-        const buffer = await (0, exports.toBuffer)(stream);
-        if (type === 'file') {
-            bodyPath = media.url;
-        }
-        else if (saveOriginalFileIfRequired) {
-            bodyPath = (0, path_1.join)(getTmpFilesDirectory(), mediaType + (0, generics_1.generateMessageID)());
-            (0, fs_1.writeFileSync)(bodyPath, buffer);
-            didSaveToTmpPath = true;
+        for await (const chunk of stream) {
+            fileLength += chunk.length;
+            sha256.update(chunk);
+            if (writeStream) {
+                if (!writeStream.write(chunk)) {
+                    await (0, events_1.once)(writeStream, 'drain');
+                }
+            }
+            encWriteStream.write(chunk);
         }
-        const fileLength = buffer.length;
-        const fileSha256 = Crypto.createHash('sha256').update(buffer).digest();
-        stream === null || stream === void 0 ? void 0 : stream.destroy();
+
+        encWriteStream.end();
+        writeStream?.end();
+
+        const fileSha256 = sha256.digest();
         logger === null || logger === void 0 ? void 0 : logger.debug('prepare stream data successfully');
+        
         return {
             mediaKey: undefined,
-            encWriteStream: buffer,
+            encWriteStream, // Now a stream, not a buffer
             fileLength,
             fileSha256,
             fileEncSha256: undefined,
@@ -425,8 +443,9 @@ const prepareStream = async (media, mediaType, { logger, saveOriginalFileIfRequi
         };
     }
     catch (error) {
-
         stream.destroy();
+        encWriteStream.destroy();
+        writeStream?.destroy();
         if (didSaveToTmpPath) {
             try {
                 await fs_1.promises.unlink(bodyPath);
@@ -442,6 +461,7 @@ exports.prepareStream = prepareStream;
 const encryptedStream = async (media, mediaType, { logger, saveOriginalFileIfRequired, opts, isPtt, forceOpus } = {}) => {
     const { stream, type } = await (0, exports.getStream)(media, opts);
 
+    let finalStream = stream;
     let opusConverted = false;
     if (mediaType === 'audio' && (isPtt === true || forceOpus === true)) {
         try {
@@ -664,7 +684,7 @@ function extensionForMediaMessage(message) {
 }
 exports.extensionForMediaMessage = extensionForMediaMessage;
 const getWAUploadToServer = ({ customUploadHosts, fetchAgent, logger, options }, refreshMediaConn) => {
-    return async (stream, { mediaType, fileEncSha256B64, newsletter, timeoutMs }) => {
+    return async (stream, { mediaType, fileEncSha256B64, newsletter, timeoutMs, fileLength }) => {
         var _a, _b;
         const { default: axios } = await import('axios');
 
@@ -683,9 +703,9 @@ const getWAUploadToServer = ({ customUploadHosts, fetchAgent, logger, options },
             const url = `https://${hostname}${media}/${fileEncSha256B64}?auth=${auth}&token=${fileEncSha256B64}`;
             let result;
             try {
-                const bodyLength = Buffer.isBuffer(reqBody) ? reqBody.length : 0;
+                const bodyLength = fileLength || (Buffer.isBuffer(reqBody) ? reqBody.length : 0);
                 if (maxContentLengthBytes && bodyLength > maxContentLengthBytes) {
-                    throw new boom_1.Boom(`Body too large for "${hostname}"`, { statusCode: 413 });
+                    throw new boom_1.Boom(`Body too large for "${hostname}" (${bodyLength} > ${maxContentLengthBytes})`, { statusCode: 413 });
                 }
                 const body = await axios.post(url, reqBody, {
                     ...options,

package/lib/Utils/messages.js

@@ -165,7 +165,7 @@ const prepareWAMessageMedia = async (message, options) => {
 
     const [{ mediaUrl, directPath, handle }] = await Promise.all([
         (async () => {
-            const result = await options.upload(encWriteStream, { fileEncSha256B64, mediaType, timeoutMs: options.mediaUploadTimeoutMs });
+            const result = await options.upload(encWriteStream, { fileEncSha256B64, mediaType, timeoutMs: options.mediaUploadTimeoutMs, fileLength });
             logger === null || logger === void 0 ? void 0 : logger.debug({ mediaType, cacheableKey }, 'uploaded media');
             return result;
         })(),

package/package.json

@@ -1,6 +1,6 @@
 {
     "name": "amiudmodz",
-    "version": "5.0.0",
+    "version": "5.0.1",
     "description": "WhatsApp Baileys mod Powered by UDMODZ",
     "keywords": [
         "whatsapp",