@fazzcode/baileys 0.1.7 → 2.4.4

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