@baileys-md/baileys 10.0.7

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (102) hide show
  1. package/LICENSE +21 -0
  2. package/README.md +2 -0
  3. package/WAProto/index.js +155693 -0
  4. package/WASignalGroup/GroupProtocol.js +1697 -0
  5. package/WASignalGroup/ciphertext_message.js +16 -0
  6. package/WASignalGroup/group_cipher.js +120 -0
  7. package/WASignalGroup/group_session_builder.js +46 -0
  8. package/WASignalGroup/index.js +5 -0
  9. package/WASignalGroup/keyhelper.js +21 -0
  10. package/WASignalGroup/protobufs.js +3 -0
  11. package/WASignalGroup/queue_job.js +69 -0
  12. package/WASignalGroup/sender_chain_key.js +50 -0
  13. package/WASignalGroup/sender_key_distribution_message.js +78 -0
  14. package/WASignalGroup/sender_key_message.js +92 -0
  15. package/WASignalGroup/sender_key_name.js +70 -0
  16. package/WASignalGroup/sender_key_record.js +56 -0
  17. package/WASignalGroup/sender_key_state.js +129 -0
  18. package/WASignalGroup/sender_message_key.js +39 -0
  19. package/lib/Defaults/baileys-version.json +3 -0
  20. package/lib/Defaults/index.js +122 -0
  21. package/lib/Defaults/phonenumber-mcc.json +223 -0
  22. package/lib/Signal/libsignal.js +161 -0
  23. package/lib/Socket/Client/abstract-socket-client.js +13 -0
  24. package/lib/Socket/Client/index.js +19 -0
  25. package/lib/Socket/Client/mobile-socket-client.js +65 -0
  26. package/lib/Socket/Client/web-socket-client.js +62 -0
  27. package/lib/Socket/business.js +260 -0
  28. package/lib/Socket/chats.js +882 -0
  29. package/lib/Socket/groups.js +327 -0
  30. package/lib/Socket/index.js +10 -0
  31. package/lib/Socket/messages-recv.js +1043 -0
  32. package/lib/Socket/messages-send.js +836 -0
  33. package/lib/Socket/newsletter.js +248 -0
  34. package/lib/Socket/registration.js +166 -0
  35. package/lib/Socket/socket.js +640 -0
  36. package/lib/Socket/usync.js +70 -0
  37. package/lib/Store/index.js +8 -0
  38. package/lib/Store/make-cache-manager-store.js +83 -0
  39. package/lib/Store/make-in-memory-store.js +431 -0
  40. package/lib/Store/make-mongo-store.js +567 -0
  41. package/lib/Store/make-ordered-dictionary.js +81 -0
  42. package/lib/Store/object-repository.js +27 -0
  43. package/lib/Types/Auth.js +2 -0
  44. package/lib/Types/Call.js +2 -0
  45. package/lib/Types/Chat.js +4 -0
  46. package/lib/Types/Contact.js +2 -0
  47. package/lib/Types/Events.js +2 -0
  48. package/lib/Types/GroupMetadata.js +2 -0
  49. package/lib/Types/Label.js +27 -0
  50. package/lib/Types/LabelAssociation.js +9 -0
  51. package/lib/Types/Message.js +7 -0
  52. package/lib/Types/Newsletter.js +18 -0
  53. package/lib/Types/Product.js +2 -0
  54. package/lib/Types/Signal.js +2 -0
  55. package/lib/Types/Socket.js +2 -0
  56. package/lib/Types/State.js +2 -0
  57. package/lib/Types/USync.js +2 -0
  58. package/lib/Types/index.js +42 -0
  59. package/lib/Utils/auth-utils.js +206 -0
  60. package/lib/Utils/baileys-event-stream.js +63 -0
  61. package/lib/Utils/business.js +234 -0
  62. package/lib/Utils/chat-utils.js +730 -0
  63. package/lib/Utils/crypto.js +151 -0
  64. package/lib/Utils/decode-wa-message.js +198 -0
  65. package/lib/Utils/event-buffer.js +520 -0
  66. package/lib/Utils/generics.js +445 -0
  67. package/lib/Utils/history.js +94 -0
  68. package/lib/Utils/index.js +33 -0
  69. package/lib/Utils/link-preview.js +126 -0
  70. package/lib/Utils/logger.js +7 -0
  71. package/lib/Utils/lt-hash.js +51 -0
  72. package/lib/Utils/make-mutex.js +43 -0
  73. package/lib/Utils/messages-media.js +724 -0
  74. package/lib/Utils/messages.js +1018 -0
  75. package/lib/Utils/noise-handler.js +155 -0
  76. package/lib/Utils/process-message.js +372 -0
  77. package/lib/Utils/signal.js +143 -0
  78. package/lib/Utils/use-multi-file-auth-state.js +93 -0
  79. package/lib/Utils/validate-connection.js +207 -0
  80. package/lib/WABinary/constants.js +40 -0
  81. package/lib/WABinary/decode.js +265 -0
  82. package/lib/WABinary/encode.js +250 -0
  83. package/lib/WABinary/generic-utils.js +110 -0
  84. package/lib/WABinary/index.js +21 -0
  85. package/lib/WABinary/jid-utils.js +69 -0
  86. package/lib/WABinary/types.js +2 -0
  87. package/lib/WAM/BinaryInfo.js +13 -0
  88. package/lib/WAM/constants.js +15350 -0
  89. package/lib/WAM/encode.js +155 -0
  90. package/lib/WAM/index.js +19 -0
  91. package/lib/WAUSync/Protocols/USyncContactProtocol.js +32 -0
  92. package/lib/WAUSync/Protocols/USyncDeviceProtocol.js +57 -0
  93. package/lib/WAUSync/Protocols/USyncDisappearingModeProtocol.js +30 -0
  94. package/lib/WAUSync/Protocols/USyncStatusProtocol.js +42 -0
  95. package/lib/WAUSync/Protocols/UsyncBotProfileProtocol.js +53 -0
  96. package/lib/WAUSync/Protocols/UsyncLIDProtocol.js +24 -0
  97. package/lib/WAUSync/Protocols/index.js +20 -0
  98. package/lib/WAUSync/USyncQuery.js +89 -0
  99. package/lib/WAUSync/USyncUser.js +26 -0
  100. package/lib/WAUSync/index.js +19 -0
  101. package/lib/index.js +33 -0
  102. package/package.json +84 -0
@@ -0,0 +1,724 @@
1
+ "use strict";
2
+ var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
3
+ if (k2 === undefined) k2 = k;
4
+ var desc = Object.getOwnPropertyDescriptor(m, k);
5
+ if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
6
+ desc = { enumerable: true, get: function() { return m[k]; } };
7
+ }
8
+ Object.defineProperty(o, k2, desc);
9
+ }) : (function(o, m, k, k2) {
10
+ if (k2 === undefined) k2 = k;
11
+ o[k2] = m[k];
12
+ }));
13
+ var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
14
+ Object.defineProperty(o, "default", { enumerable: true, value: v });
15
+ }) : function(o, v) {
16
+ o["default"] = v;
17
+ });
18
+ var __importStar = (this && this.__importStar) || (function () {
19
+ var ownKeys = function(o) {
20
+ ownKeys = Object.getOwnPropertyNames || function (o) {
21
+ var ar = [];
22
+ for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
23
+ return ar;
24
+ };
25
+ return ownKeys(o);
26
+ };
27
+ return function (mod) {
28
+ if (mod && mod.__esModule) return mod;
29
+ var result = {};
30
+ if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
31
+ __setModuleDefault(result, mod);
32
+ return result;
33
+ };
34
+ })();
35
+ var __importDefault = (this && this.__importDefault) || function (mod) {
36
+ return (mod && mod.__esModule) ? mod : { "default": mod };
37
+ };
38
+ Object.defineProperty(exports, "__esModule", { value: true });
39
+ exports.getStatusCodeForMediaRetry = exports.decryptMediaRetryData = exports.decodeMediaRetryNode = exports.encryptMediaRetryRequest = exports.getWAUploadToServer = exports.downloadEncryptedContent = exports.downloadContentFromMessage = exports.getUrlFromDirectPath = exports.encryptedStream = exports.prepareStream = exports.getHttpStream = exports.getStream = exports.toBuffer = exports.toReadable = exports.mediaMessageSHA256B64 = exports.generateProfilePicture = exports.encodeBase64EncodedStringForUpload = exports.extractImageThumb = exports.hkdfInfoKey = void 0;
40
+ exports.getMediaKeys = getMediaKeys;
41
+ exports.getAudioDuration = getAudioDuration;
42
+ exports.getAudioWaveform = getAudioWaveform;
43
+ exports.generateThumbnail = generateThumbnail;
44
+ exports.extensionForMediaMessage = extensionForMediaMessage;
45
+ const boom_1 = require("@hapi/boom");
46
+ const axios_1 = __importDefault(require("axios"));
47
+ const child_process_1 = require("child_process");
48
+ const Crypto = __importStar(require("crypto"));
49
+ const events_1 = require("events");
50
+ const fs_1 = require("fs");
51
+ const os_1 = require("os");
52
+ const path_1 = require("path");
53
+ const jimp_1 = __importDefault(require("jimp"));
54
+ const stream_1 = require("stream");
55
+ const WAProto_1 = require("../../WAProto");
56
+ const Defaults_1 = require("../Defaults");
57
+ const WABinary_1 = require("../WABinary");
58
+ const crypto_1 = require("./crypto");
59
+ const generics_1 = require("./generics");
60
+ const getTmpFilesDirectory = () => (0, os_1.tmpdir)();
61
+ const getImageProcessingLibrary = async () => {
62
+ const [_jimp, sharp] = await Promise.all([
63
+ (async () => {
64
+ const jimp = await (Promise.resolve().then(() => __importStar(require('jimp'))).catch(() => { }));
65
+ return jimp;
66
+ })(),
67
+ (async () => {
68
+ const sharp = await (Promise.resolve().then(() => __importStar(require('sharp'))).catch(() => { }));
69
+ return sharp;
70
+ })()
71
+ ]);
72
+ if (sharp) {
73
+ return { sharp };
74
+ }
75
+ const jimp = (_jimp === null || _jimp === void 0 ? void 0 : _jimp.default) || _jimp;
76
+ if (jimp) {
77
+ return { jimp };
78
+ }
79
+ throw new boom_1.Boom('No image processing library available');
80
+ };
81
+ const hkdfInfoKey = (type) => {
82
+ const hkdfInfo = Defaults_1.MEDIA_HKDF_KEY_MAPPING[type];
83
+ return `WhatsApp ${hkdfInfo} Keys`;
84
+ };
85
+ exports.hkdfInfoKey = hkdfInfoKey;
86
+ /** generates all the keys required to encrypt/decrypt & sign a media message */
87
+ async function getMediaKeys(buffer, mediaType) {
88
+ if (!buffer) {
89
+ throw new boom_1.Boom('Cannot derive from empty media key');
90
+ }
91
+ if (typeof buffer === 'string') {
92
+ buffer = Buffer.from(buffer.replace('data:;base64,', ''), 'base64');
93
+ }
94
+ // expand using HKDF to 112 bytes, also pass in the relevant app info
95
+ const expandedMediaKey = await (0, crypto_1.hkdf)(buffer, 112, { info: (0, exports.hkdfInfoKey)(mediaType) });
96
+ return {
97
+ iv: expandedMediaKey.slice(0, 16),
98
+ cipherKey: expandedMediaKey.slice(16, 48),
99
+ macKey: expandedMediaKey.slice(48, 80),
100
+ };
101
+ }
102
+ /** Extracts video thumb using FFMPEG */
103
+ const extractVideoThumb = async (path, destPath, time, size) => new Promise((resolve, reject) => {
104
+ const cmd = `ffmpeg -ss ${time} -i ${path} -y -vf scale=${size.width}:-1 -vframes 1 -f image2 ${destPath}`;
105
+ (0, child_process_1.exec)(cmd, (err) => {
106
+ if (err) {
107
+ reject(err);
108
+ }
109
+ else {
110
+ resolve();
111
+ }
112
+ });
113
+ });
114
+ const extractImageThumb = async (bufferOrFilePath, width = 32) => {
115
+ var _a, _b;
116
+ if (bufferOrFilePath instanceof stream_1.Readable) {
117
+ bufferOrFilePath = await (0, exports.toBuffer)(bufferOrFilePath);
118
+ }
119
+ const lib = await getImageProcessingLibrary();
120
+ if ('sharp' in lib && typeof ((_a = lib.sharp) === null || _a === void 0 ? void 0 : _a.default) === 'function') {
121
+ const img = lib.sharp.default(bufferOrFilePath);
122
+ const dimensions = await img.metadata();
123
+ const buffer = await img
124
+ .resize(width)
125
+ .jpeg({ quality: 50 })
126
+ .toBuffer();
127
+ return {
128
+ buffer,
129
+ original: {
130
+ width: dimensions.width,
131
+ height: dimensions.height,
132
+ },
133
+ };
134
+ }
135
+ else if ('jimp' in lib && typeof ((_b = lib.jimp) === null || _b === void 0 ? void 0 : _b.read) === 'function') {
136
+ const { read, MIME_JPEG, RESIZE_BILINEAR, AUTO } = lib.jimp;
137
+ const jimp = await read(bufferOrFilePath);
138
+ const dimensions = {
139
+ width: jimp.getWidth(),
140
+ height: jimp.getHeight()
141
+ };
142
+ const buffer = await jimp
143
+ .quality(50)
144
+ .resize(width, AUTO, RESIZE_BILINEAR)
145
+ .getBufferAsync(MIME_JPEG);
146
+ return {
147
+ buffer,
148
+ original: dimensions
149
+ };
150
+ }
151
+ else {
152
+ throw new boom_1.Boom('No image processing library available');
153
+ }
154
+ };
155
+ exports.extractImageThumb = extractImageThumb;
156
+ const encodeBase64EncodedStringForUpload = (b64) => (encodeURIComponent(b64
157
+ .replace(/\+/g, '-')
158
+ .replace(/\//g, '_')
159
+ .replace(/\=+$/, '')));
160
+ exports.encodeBase64EncodedStringForUpload = encodeBase64EncodedStringForUpload;
161
+ const generateProfilePicture = async (mediaUpload) => {
162
+ let bufferOrFilePath;
163
+ let img;
164
+ if (Buffer.isBuffer(mediaUpload)) {
165
+ bufferOrFilePath = mediaUpload;
166
+ }
167
+ else if ('url' in mediaUpload) {
168
+ bufferOrFilePath = mediaUpload.url.toString();
169
+ }
170
+ else {
171
+ bufferOrFilePath = await (0, exports.toBuffer)(mediaUpload.stream);
172
+ }
173
+ const jimp = await jimp_1.default.read(bufferOrFilePath);
174
+ const cropped = jimp.getWidth() > jimp.getHeight() ? jimp.resize(550, -1) : jimp.resize(-1, 650);
175
+ img = cropped
176
+ .quality(100)
177
+ .getBufferAsync(jimp_1.default.MIME_JPEG);
178
+ return {
179
+ img: await img,
180
+ };
181
+ };
182
+ exports.generateProfilePicture = generateProfilePicture;
183
+ /** gets the SHA256 of the given media message */
184
+ const mediaMessageSHA256B64 = (message) => {
185
+ const media = Object.values(message)[0];
186
+ return (media === null || media === void 0 ? void 0 : media.fileSha256) && Buffer.from(media.fileSha256).toString('base64');
187
+ };
188
+ exports.mediaMessageSHA256B64 = mediaMessageSHA256B64;
189
+ async function getAudioDuration(buffer) {
190
+ const musicMetadata = await Promise.resolve().then(() => __importStar(require('music-metadata')));
191
+ let metadata;
192
+ if (Buffer.isBuffer(buffer)) {
193
+ metadata = await musicMetadata.parseBuffer(buffer, undefined, { duration: true });
194
+ }
195
+ else if (typeof buffer === 'string') {
196
+ const rStream = (0, fs_1.createReadStream)(buffer);
197
+ try {
198
+ metadata = await musicMetadata.parseStream(rStream, undefined, { duration: true });
199
+ }
200
+ finally {
201
+ rStream.destroy();
202
+ }
203
+ }
204
+ else {
205
+ metadata = await musicMetadata.parseStream(buffer, undefined, { duration: true });
206
+ }
207
+ return metadata.format.duration;
208
+ }
209
+ /**
210
+ referenced from and modifying https://github.com/wppconnect-team/wa-js/blob/main/src/chat/functions/prepareAudioWaveform.ts
211
+ */
212
+ async function getAudioWaveform(buffer, logger) {
213
+ try {
214
+ const { default: decoder } = await eval('import(\'audio-decode\')');
215
+ let audioData;
216
+ if (Buffer.isBuffer(buffer)) {
217
+ audioData = buffer;
218
+ }
219
+ else if (typeof buffer === 'string') {
220
+ const rStream = (0, fs_1.createReadStream)(buffer);
221
+ audioData = await (0, exports.toBuffer)(rStream);
222
+ }
223
+ else {
224
+ audioData = await (0, exports.toBuffer)(buffer);
225
+ }
226
+ const audioBuffer = await decoder(audioData);
227
+ const rawData = audioBuffer.getChannelData(0); // We only need to work with one channel of data
228
+ const samples = 64; // Number of samples we want to have in our final data set
229
+ const blockSize = Math.floor(rawData.length / samples); // the number of samples in each subdivision
230
+ const filteredData = [];
231
+ for (let i = 0; i < samples; i++) {
232
+ const blockStart = blockSize * i; // the location of the first sample in the block
233
+ let sum = 0;
234
+ for (let j = 0; j < blockSize; j++) {
235
+ sum = sum + Math.abs(rawData[blockStart + j]); // find the sum of all the samples in the block
236
+ }
237
+ filteredData.push(sum / blockSize); // divide the sum by the block size to get the average
238
+ }
239
+ // This guarantees that the largest data point will be set to 1, and the rest of the data will scale proportionally.
240
+ const multiplier = Math.pow(Math.max(...filteredData), -1);
241
+ const normalizedData = filteredData.map((n) => n * multiplier);
242
+ // Generate waveform like WhatsApp
243
+ const waveform = new Uint8Array(normalizedData.map((n) => Math.floor(100 * n)));
244
+ return waveform;
245
+ }
246
+ catch (e) {
247
+ logger === null || logger === void 0 ? void 0 : logger.debug('Failed to generate waveform: ' + e);
248
+ }
249
+ }
250
+ const toReadable = (buffer) => {
251
+ const readable = new stream_1.Readable({ read: () => { } });
252
+ readable.push(buffer);
253
+ readable.push(null);
254
+ return readable;
255
+ };
256
+ exports.toReadable = toReadable;
257
+ const toBuffer = async (stream) => {
258
+ const chunks = [];
259
+ for await (const chunk of stream) {
260
+ chunks.push(chunk);
261
+ }
262
+ stream.destroy();
263
+ return Buffer.concat(chunks);
264
+ };
265
+ exports.toBuffer = toBuffer;
266
+ const getStream = async (item, opts) => {
267
+ if (Buffer.isBuffer(item)) {
268
+ return { stream: (0, exports.toReadable)(item), type: 'buffer' };
269
+ }
270
+ if ('stream' in item) {
271
+ return { stream: item.stream, type: 'readable' };
272
+ }
273
+ if (item.url.toString().startsWith('http://') || item.url.toString().startsWith('https://')) {
274
+ return { stream: await (0, exports.getHttpStream)(item.url, opts), type: 'remote' };
275
+ }
276
+ return { stream: (0, fs_1.createReadStream)(item.url), type: 'file' };
277
+ };
278
+ exports.getStream = getStream;
279
+ /** generates a thumbnail for a given media, if required */
280
+ async function generateThumbnail(file, mediaType, options) {
281
+ var _a;
282
+ let thumbnail;
283
+ let originalImageDimensions;
284
+ if (mediaType === 'image') {
285
+ const { buffer, original } = await (0, exports.extractImageThumb)(file);
286
+ thumbnail = buffer.toString('base64');
287
+ if (original.width && original.height) {
288
+ originalImageDimensions = {
289
+ width: original.width,
290
+ height: original.height,
291
+ };
292
+ }
293
+ }
294
+ else if (mediaType === 'video') {
295
+ const imgFilename = (0, path_1.join)(getTmpFilesDirectory(), (0, generics_1.generateMessageIDV2)() + '.jpg');
296
+ try {
297
+ await extractVideoThumb(file, imgFilename, '00:00:00', { width: 32, height: 32 });
298
+ const buff = await fs_1.promises.readFile(imgFilename);
299
+ thumbnail = buff.toString('base64');
300
+ await fs_1.promises.unlink(imgFilename);
301
+ }
302
+ catch (err) {
303
+ (_a = options.logger) === null || _a === void 0 ? void 0 : _a.debug('could not generate video thumb: ' + err);
304
+ }
305
+ }
306
+ return {
307
+ thumbnail,
308
+ originalImageDimensions
309
+ };
310
+ }
311
+ const getHttpStream = async (url, options = {}) => {
312
+ const fetched = await axios_1.default.get(url.toString(), { ...options, responseType: 'stream' });
313
+ return fetched.data;
314
+ };
315
+ exports.getHttpStream = getHttpStream;
316
+ const prepareStream = async (media, mediaType, { logger, saveOriginalFileIfRequired, opts } = {}) => {
317
+ const { stream, type } = await (0, exports.getStream)(media, opts);
318
+ logger === null || logger === void 0 ? void 0 : logger.debug('fetched media stream');
319
+ let bodyPath;
320
+ let didSaveToTmpPath = false;
321
+ try {
322
+ const buffer = await (0, exports.toBuffer)(stream);
323
+ if (type === 'file') {
324
+ bodyPath = media.url;
325
+ }
326
+ else if (saveOriginalFileIfRequired) {
327
+ bodyPath = (0, path_1.join)(getTmpFilesDirectory(), mediaType + (0, generics_1.generateMessageIDV2)());
328
+ (0, fs_1.writeFileSync)(bodyPath, buffer);
329
+ didSaveToTmpPath = true;
330
+ }
331
+ const fileLength = buffer.length;
332
+ const fileSha256 = Crypto.createHash('sha256').update(buffer).digest();
333
+ stream === null || stream === void 0 ? void 0 : stream.destroy();
334
+ logger === null || logger === void 0 ? void 0 : logger.debug('prepare stream data successfully');
335
+ return {
336
+ mediaKey: undefined,
337
+ encWriteStream: buffer,
338
+ fileLength,
339
+ fileSha256,
340
+ fileEncSha256: undefined,
341
+ bodyPath,
342
+ didSaveToTmpPath
343
+ };
344
+ }
345
+ catch (error) {
346
+ // destroy all streams with error
347
+ stream.destroy();
348
+ if (didSaveToTmpPath) {
349
+ try {
350
+ await fs_1.promises.unlink(bodyPath);
351
+ }
352
+ catch (err) {
353
+ logger === null || logger === void 0 ? void 0 : logger.error({ err }, 'failed to save to tmp path');
354
+ }
355
+ }
356
+ throw error;
357
+ }
358
+ };
359
+ exports.prepareStream = prepareStream;
360
+ const encryptedStream = async (media, mediaType, { logger, saveOriginalFileIfRequired, opts } = {}) => {
361
+ const { stream, type } = await (0, exports.getStream)(media, opts);
362
+ logger === null || logger === void 0 ? void 0 : logger.debug('fetched media stream');
363
+ const mediaKey = Crypto.randomBytes(32);
364
+ const { cipherKey, iv, macKey } = await getMediaKeys(mediaKey, mediaType);
365
+ const encWriteStream = new stream_1.Readable({ read: () => { } });
366
+ let bodyPath;
367
+ let writeStream;
368
+ let didSaveToTmpPath = false;
369
+ if (type === 'file') {
370
+ bodyPath = media.url;
371
+ }
372
+ else if (saveOriginalFileIfRequired) {
373
+ bodyPath = (0, path_1.join)(getTmpFilesDirectory(), mediaType + (0, generics_1.generateMessageIDV2)());
374
+ writeStream = (0, fs_1.createWriteStream)(bodyPath);
375
+ didSaveToTmpPath = true;
376
+ }
377
+ let fileLength = 0;
378
+ const aes = Crypto.createCipheriv('aes-256-cbc', cipherKey, iv);
379
+ let hmac = Crypto.createHmac('sha256', macKey).update(iv);
380
+ let sha256Plain = Crypto.createHash('sha256');
381
+ let sha256Enc = Crypto.createHash('sha256');
382
+ try {
383
+ for await (const data of stream) {
384
+ fileLength += data.length;
385
+ if (type === 'remote'
386
+ && (opts === null || opts === void 0 ? void 0 : opts.maxContentLength)
387
+ && fileLength + data.length > opts.maxContentLength) {
388
+ throw new boom_1.Boom(`content length exceeded when encrypting "${type}"`, {
389
+ data: { media, type }
390
+ });
391
+ }
392
+ sha256Plain = sha256Plain.update(data);
393
+ if (writeStream) {
394
+ if (!writeStream.write(data)) {
395
+ await (0, events_1.once)(writeStream, 'drain');
396
+ }
397
+ }
398
+ onChunk(aes.update(data));
399
+ }
400
+ onChunk(aes.final());
401
+ const mac = hmac.digest().slice(0, 10);
402
+ sha256Enc = sha256Enc.update(mac);
403
+ const fileSha256 = sha256Plain.digest();
404
+ const fileEncSha256 = sha256Enc.digest();
405
+ encWriteStream.push(mac);
406
+ encWriteStream.push(null);
407
+ writeStream === null || writeStream === void 0 ? void 0 : writeStream.end();
408
+ stream.destroy();
409
+ logger === null || logger === void 0 ? void 0 : logger.debug('encrypted data successfully');
410
+ return {
411
+ mediaKey,
412
+ encWriteStream,
413
+ bodyPath,
414
+ mac,
415
+ fileEncSha256,
416
+ fileSha256,
417
+ fileLength,
418
+ didSaveToTmpPath
419
+ };
420
+ }
421
+ catch (error) {
422
+ // destroy all streams with error
423
+ encWriteStream.destroy();
424
+ writeStream === null || writeStream === void 0 ? void 0 : writeStream.destroy();
425
+ aes.destroy();
426
+ hmac.destroy();
427
+ sha256Plain.destroy();
428
+ sha256Enc.destroy();
429
+ stream.destroy();
430
+ if (didSaveToTmpPath) {
431
+ try {
432
+ await fs_1.promises.unlink(bodyPath);
433
+ }
434
+ catch (err) {
435
+ logger === null || logger === void 0 ? void 0 : logger.error({ err }, 'failed to save to tmp path');
436
+ }
437
+ }
438
+ throw error;
439
+ }
440
+ function onChunk(buff) {
441
+ sha256Enc = sha256Enc.update(buff);
442
+ hmac = hmac.update(buff);
443
+ encWriteStream.push(buff);
444
+ }
445
+ };
446
+ exports.encryptedStream = encryptedStream;
447
+ const DEF_HOST = 'mmg.whatsapp.net';
448
+ const AES_CHUNK_SIZE = 16;
449
+ const toSmallestChunkSize = (num) => {
450
+ return Math.floor(num / AES_CHUNK_SIZE) * AES_CHUNK_SIZE;
451
+ };
452
+ const getUrlFromDirectPath = (directPath) => `https://${DEF_HOST}${directPath}`;
453
+ exports.getUrlFromDirectPath = getUrlFromDirectPath;
454
+ const downloadContentFromMessage = async ({ mediaKey, directPath, url }, type, opts = {}) => {
455
+ const downloadUrl = url || (0, exports.getUrlFromDirectPath)(directPath);
456
+ const keys = await getMediaKeys(mediaKey, type);
457
+ return (0, exports.downloadEncryptedContent)(downloadUrl, keys, opts);
458
+ };
459
+ exports.downloadContentFromMessage = downloadContentFromMessage;
460
+ /**
461
+ * Decrypts and downloads an AES256-CBC encrypted file given the keys.
462
+ * Assumes the SHA256 of the plaintext is appended to the end of the ciphertext
463
+ * */
464
+ const downloadEncryptedContent = async (downloadUrl, { cipherKey, iv }, { startByte, endByte, options } = {}) => {
465
+ let bytesFetched = 0;
466
+ let startChunk = 0;
467
+ let firstBlockIsIV = false;
468
+ // if a start byte is specified -- then we need to fetch the previous chunk as that will form the IV
469
+ if (startByte) {
470
+ const chunk = toSmallestChunkSize(startByte || 0);
471
+ if (chunk) {
472
+ startChunk = chunk - AES_CHUNK_SIZE;
473
+ bytesFetched = chunk;
474
+ firstBlockIsIV = true;
475
+ }
476
+ }
477
+ const endChunk = endByte ? toSmallestChunkSize(endByte || 0) + AES_CHUNK_SIZE : undefined;
478
+ const headers = {
479
+ ...(options === null || options === void 0 ? void 0 : options.headers) || {},
480
+ Origin: Defaults_1.DEFAULT_ORIGIN,
481
+ };
482
+ if (startChunk || endChunk) {
483
+ headers.Range = `bytes=${startChunk}-`;
484
+ if (endChunk) {
485
+ headers.Range += endChunk;
486
+ }
487
+ }
488
+ // download the message
489
+ const fetched = await (0, exports.getHttpStream)(downloadUrl, {
490
+ ...options || {},
491
+ headers,
492
+ maxBodyLength: Infinity,
493
+ maxContentLength: Infinity,
494
+ });
495
+ let remainingBytes = Buffer.from([]);
496
+ let aes;
497
+ const pushBytes = (bytes, push) => {
498
+ if (startByte || endByte) {
499
+ const start = bytesFetched >= startByte ? undefined : Math.max(startByte - bytesFetched, 0);
500
+ const end = bytesFetched + bytes.length < endByte ? undefined : Math.max(endByte - bytesFetched, 0);
501
+ push(bytes.slice(start, end));
502
+ bytesFetched += bytes.length;
503
+ }
504
+ else {
505
+ push(bytes);
506
+ }
507
+ };
508
+ const output = new stream_1.Transform({
509
+ transform(chunk, _, callback) {
510
+ let data = Buffer.concat([remainingBytes, chunk]);
511
+ const decryptLength = toSmallestChunkSize(data.length);
512
+ remainingBytes = data.slice(decryptLength);
513
+ data = data.slice(0, decryptLength);
514
+ if (!aes) {
515
+ let ivValue = iv;
516
+ if (firstBlockIsIV) {
517
+ ivValue = data.slice(0, AES_CHUNK_SIZE);
518
+ data = data.slice(AES_CHUNK_SIZE);
519
+ }
520
+ aes = Crypto.createDecipheriv('aes-256-cbc', cipherKey, ivValue);
521
+ // if an end byte that is not EOF is specified
522
+ // stop auto padding (PKCS7) -- otherwise throws an error for decryption
523
+ if (endByte) {
524
+ aes.setAutoPadding(false);
525
+ }
526
+ }
527
+ try {
528
+ pushBytes(aes.update(data), b => this.push(b));
529
+ callback();
530
+ }
531
+ catch (error) {
532
+ callback(error);
533
+ }
534
+ },
535
+ final(callback) {
536
+ try {
537
+ pushBytes(aes.final(), b => this.push(b));
538
+ callback();
539
+ }
540
+ catch (error) {
541
+ callback(error);
542
+ }
543
+ },
544
+ });
545
+ return fetched.pipe(output, { end: true });
546
+ };
547
+ exports.downloadEncryptedContent = downloadEncryptedContent;
548
+ function extensionForMediaMessage(message) {
549
+ const getExtension = (mimetype) => mimetype.split(';')[0].split('/')[1];
550
+ const type = Object.keys(message)[0];
551
+ let extension;
552
+ if (type === 'locationMessage' ||
553
+ type === 'liveLocationMessage' ||
554
+ type === 'productMessage') {
555
+ extension = '.jpeg';
556
+ }
557
+ else {
558
+ const messageContent = message[type];
559
+ extension = getExtension(messageContent.mimetype);
560
+ }
561
+ return extension;
562
+ }
563
+ const getWAUploadToServer = ({ customUploadHosts, fetchAgent, logger, options }, refreshMediaConn) => {
564
+ return async (stream, { mediaType, fileEncSha256B64, newsletter, timeoutMs }) => {
565
+ var _a, _b;
566
+ // send a query JSON to obtain the url & auth token to upload our media
567
+ let uploadInfo = await refreshMediaConn(false);
568
+ let urls;
569
+ const hosts = [...customUploadHosts, ...uploadInfo.hosts];
570
+ const chunks = [];
571
+ if (!Buffer.isBuffer(stream)) {
572
+ for await (const chunk of stream) {
573
+ chunks.push(chunk);
574
+ }
575
+ }
576
+ const reqBody = Buffer.isBuffer(stream) ? stream : Buffer.concat(chunks);
577
+ fileEncSha256B64 = (0, exports.encodeBase64EncodedStringForUpload)(fileEncSha256B64);
578
+ let media = Defaults_1.MEDIA_PATH_MAP[mediaType];
579
+ if (newsletter) {
580
+ media = media === null || media === void 0 ? void 0 : media.replace('/mms/', '/newsletter/newsletter-');
581
+ }
582
+ for (const { hostname, maxContentLengthBytes } of hosts) {
583
+ logger.debug(`uploading to "${hostname}"`);
584
+ const auth = encodeURIComponent(uploadInfo.auth); // the auth token
585
+ const url = `https://${hostname}${media}/${fileEncSha256B64}?auth=${auth}&token=${fileEncSha256B64}`;
586
+ let result;
587
+ try {
588
+ if (maxContentLengthBytes && reqBody.length > maxContentLengthBytes) {
589
+ throw new boom_1.Boom(`Body too large for "${hostname}"`, { statusCode: 413 });
590
+ }
591
+ const body = await axios_1.default.post(url, reqBody, {
592
+ ...options,
593
+ headers: {
594
+ ...options.headers || {},
595
+ 'Content-Type': 'application/octet-stream',
596
+ 'Origin': Defaults_1.DEFAULT_ORIGIN
597
+ },
598
+ httpsAgent: fetchAgent,
599
+ timeout: timeoutMs,
600
+ responseType: 'json',
601
+ maxBodyLength: Infinity,
602
+ maxContentLength: Infinity,
603
+ });
604
+ result = body.data;
605
+ if ((result === null || result === void 0 ? void 0 : result.url) || (result === null || result === void 0 ? void 0 : result.directPath)) {
606
+ urls = {
607
+ mediaUrl: result.url,
608
+ directPath: result.direct_path,
609
+ handle: result.handle
610
+ };
611
+ break;
612
+ }
613
+ else {
614
+ uploadInfo = await refreshMediaConn(true);
615
+ throw new Error(`upload failed, reason: ${JSON.stringify(result)}`);
616
+ }
617
+ }
618
+ catch (error) {
619
+ if (axios_1.default.isAxiosError(error)) {
620
+ result = (_a = error.response) === null || _a === void 0 ? void 0 : _a.data;
621
+ }
622
+ const isLast = hostname === ((_b = hosts[uploadInfo.hosts.length - 1]) === null || _b === void 0 ? void 0 : _b.hostname);
623
+ logger.warn({ trace: error.stack, uploadResult: result }, `Error in uploading to ${hostname} ${isLast ? '' : ', retrying...'}`);
624
+ }
625
+ }
626
+ if (!urls) {
627
+ throw new boom_1.Boom('Media upload failed on all hosts', { statusCode: 500 });
628
+ }
629
+ return urls;
630
+ };
631
+ };
632
+ exports.getWAUploadToServer = getWAUploadToServer;
633
+ const getMediaRetryKey = (mediaKey) => {
634
+ return (0, crypto_1.hkdf)(mediaKey, 32, { info: 'WhatsApp Media Retry Notification' });
635
+ };
636
+ /**
637
+ * Generate a binary node that will request the phone to re-upload the media & return the newly uploaded URL
638
+ */
639
+ const encryptMediaRetryRequest = async (key, mediaKey, meId) => {
640
+ const recp = { stanzaId: key.id };
641
+ const recpBuffer = WAProto_1.proto.ServerErrorReceipt.encode(recp).finish();
642
+ const iv = Crypto.randomBytes(12);
643
+ const retryKey = await getMediaRetryKey(mediaKey);
644
+ const ciphertext = (0, crypto_1.aesEncryptGCM)(recpBuffer, retryKey, iv, Buffer.from(key.id));
645
+ const req = {
646
+ tag: 'receipt',
647
+ attrs: {
648
+ id: key.id,
649
+ to: (0, WABinary_1.jidNormalizedUser)(meId),
650
+ type: 'server-error'
651
+ },
652
+ content: [
653
+ // this encrypt node is actually pretty useless
654
+ // the media is returned even without this node
655
+ // keeping it here to maintain parity with WA Web
656
+ {
657
+ tag: 'encrypt',
658
+ attrs: {},
659
+ content: [
660
+ { tag: 'enc_p', attrs: {}, content: ciphertext },
661
+ { tag: 'enc_iv', attrs: {}, content: iv }
662
+ ]
663
+ },
664
+ {
665
+ tag: 'rmr',
666
+ attrs: {
667
+ jid: key.remoteJid,
668
+ 'from_me': (!!key.fromMe).toString(),
669
+ // @ts-ignore
670
+ participant: key.participant || undefined
671
+ }
672
+ }
673
+ ]
674
+ };
675
+ return req;
676
+ };
677
+ exports.encryptMediaRetryRequest = encryptMediaRetryRequest;
678
+ const decodeMediaRetryNode = (node) => {
679
+ const rmrNode = (0, WABinary_1.getBinaryNodeChild)(node, 'rmr');
680
+ const event = {
681
+ key: {
682
+ id: node.attrs.id,
683
+ remoteJid: rmrNode.attrs.jid,
684
+ fromMe: rmrNode.attrs.from_me === 'true',
685
+ participant: rmrNode.attrs.participant
686
+ }
687
+ };
688
+ const errorNode = (0, WABinary_1.getBinaryNodeChild)(node, 'error');
689
+ if (errorNode) {
690
+ const errorCode = +errorNode.attrs.code;
691
+ event.error = new boom_1.Boom(`Failed to re-upload media (${errorCode})`, { data: errorNode.attrs, statusCode: (0, exports.getStatusCodeForMediaRetry)(errorCode) });
692
+ }
693
+ else {
694
+ const encryptedInfoNode = (0, WABinary_1.getBinaryNodeChild)(node, 'encrypt');
695
+ const ciphertext = (0, WABinary_1.getBinaryNodeChildBuffer)(encryptedInfoNode, 'enc_p');
696
+ const iv = (0, WABinary_1.getBinaryNodeChildBuffer)(encryptedInfoNode, 'enc_iv');
697
+ if (ciphertext && iv) {
698
+ event.media = { ciphertext, iv };
699
+ }
700
+ else {
701
+ event.error = new boom_1.Boom('Failed to re-upload media (missing ciphertext)', { statusCode: 404 });
702
+ }
703
+ }
704
+ return event;
705
+ };
706
+ exports.decodeMediaRetryNode = decodeMediaRetryNode;
707
+ const decryptMediaRetryData = async ({ ciphertext, iv }, mediaKey, msgId) => {
708
+ const retryKey = await getMediaRetryKey(mediaKey);
709
+ const plaintext = (0, crypto_1.aesDecryptGCM)(ciphertext, retryKey, iv, Buffer.from(msgId));
710
+ return WAProto_1.proto.MediaRetryNotification.decode(plaintext);
711
+ };
712
+ exports.decryptMediaRetryData = decryptMediaRetryData;
713
+ const getStatusCodeForMediaRetry = (code) => MEDIA_RETRY_STATUS_MAP[code];
714
+ exports.getStatusCodeForMediaRetry = getStatusCodeForMediaRetry;
715
+ const MEDIA_RETRY_STATUS_MAP = {
716
+ [WAProto_1.proto.MediaRetryNotification.ResultType.SUCCESS]: 200,
717
+ [WAProto_1.proto.MediaRetryNotification.ResultType.DECRYPTION_ERROR]: 412,
718
+ [WAProto_1.proto.MediaRetryNotification.ResultType.NOT_FOUND]: 404,
719
+ [WAProto_1.proto.MediaRetryNotification.ResultType.GENERAL_ERROR]: 418,
720
+ };
721
+ // eslint-disable-next-line @typescript-eslint/no-unused-vars
722
+ function __importStar(arg0) {
723
+ throw new Error('Function not implemented.');
724
+ }