From 15f759a52be249e62d482f74d6af7b81696c9189 Mon Sep 17 00:00:00 2001 From: Adhiraj Singh Date: Sun, 15 Aug 2021 13:16:03 +0530 Subject: [PATCH] include media fix from @saviosg --- src/Utils/messages-media.ts | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/src/Utils/messages-media.ts b/src/Utils/messages-media.ts index e8716c7..9404b34 100644 --- a/src/Utils/messages-media.ts +++ b/src/Utils/messages-media.ts @@ -16,6 +16,7 @@ import { generateMessageID, hkdf } from './generics' import { Boom } from '@hapi/boom' import { MediaType } from '../Types' import { DEFAULT_ORIGIN } from '../Defaults' +import { once } from 'events' export const hkdfInfoKey = (type: MediaType) => { if(type === 'sticker') type = 'image' @@ -173,6 +174,8 @@ export const encryptedStream = async(media: WAMediaUpload, mediaType: MediaType, for await(const data of stream) { fileLength += data.length sha256Plain = sha256Plain.update(data) + if (writeStream && !writeStream.write(data)) await once(writeStream, 'drain') + writeStream && writeStream.write(data) onChunk(aes.update(data)) } @@ -185,9 +188,9 @@ export const encryptedStream = async(media: WAMediaUpload, mediaType: MediaType, const fileEncSha256 = sha256Enc.digest() encWriteStream.write(mac) - encWriteStream.close() + encWriteStream.end() - writeStream && writeStream.close() + writeStream && writeStream.end() return { mediaKey,