@sixcore/baileys 1.0.2 → 1.1.0

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