socketon 1.31.2-rc → 1.51.16

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 (213) hide show
  1. package/README.md +313 -159
  2. package/WAProto/WAProto.proto +5311 -0
  3. package/WAProto/index.js +65801 -141371
  4. package/lib/Defaults/index.js +117 -141
  5. package/lib/KeyDB/BinarySearch.js +20 -0
  6. package/lib/KeyDB/KeyedDB.js +167 -0
  7. package/lib/KeyDB/index.js +4 -0
  8. package/lib/Signal/Group/ciphertext-message.js +12 -14
  9. package/lib/Signal/Group/group-session-builder.js +10 -42
  10. package/lib/Signal/Group/group_cipher.js +75 -87
  11. package/lib/Signal/Group/index.js +13 -57
  12. package/lib/Signal/Group/keyhelper.js +17 -52
  13. package/lib/Signal/Group/sender-chain-key.js +27 -33
  14. package/lib/Signal/Group/sender-key-distribution-message.js +62 -63
  15. package/lib/Signal/Group/sender-key-message.js +65 -66
  16. package/lib/Signal/Group/sender-key-name.js +45 -44
  17. package/lib/Signal/Group/sender-key-record.js +39 -49
  18. package/lib/Signal/Group/sender-key-state.js +80 -93
  19. package/lib/Signal/Group/sender-message-key.js +27 -28
  20. package/lib/Signal/libsignal.js +313 -163
  21. package/lib/Signal/lid-mapping.js +155 -0
  22. package/lib/Socket/Client/index.js +4 -19
  23. package/lib/Socket/Client/types.js +13 -0
  24. package/lib/Socket/Client/websocket.js +52 -0
  25. package/lib/Socket/Client/websocket.js.bak +53 -0
  26. package/lib/Socket/business.js +359 -242
  27. package/lib/Socket/chats.js +846 -935
  28. package/lib/Socket/communities.js +413 -0
  29. package/lib/Socket/groups.js +304 -309
  30. package/lib/Socket/index.js +15 -10
  31. package/lib/Socket/messages-recv.js +1107 -1054
  32. package/lib/Socket/messages-send.js +639 -448
  33. package/lib/Socket/mex.js +45 -0
  34. package/lib/Socket/newsletter.js +240 -324
  35. package/lib/Socket/socket.js +794 -651
  36. package/lib/Socket/socketon.js +402 -0
  37. package/lib/Store/index.js +6 -10
  38. package/lib/Store/make-cache-manager-store.js +73 -81
  39. package/lib/Store/make-in-memory-store.js +286 -423
  40. package/lib/Store/make-ordered-dictionary.js +77 -79
  41. package/lib/Store/object-repository.js +24 -26
  42. package/lib/Types/Auth.js +3 -2
  43. package/lib/Types/Bussines.js +3 -0
  44. package/lib/Types/Call.js +3 -2
  45. package/lib/Types/Chat.js +9 -4
  46. package/lib/Types/Contact.js +3 -2
  47. package/lib/Types/Events.js +3 -2
  48. package/lib/Types/GroupMetadata.js +3 -2
  49. package/lib/Types/Label.js +24 -26
  50. package/lib/Types/LabelAssociation.js +6 -8
  51. package/lib/Types/Message.js +12 -9
  52. package/lib/Types/Newsletter.js +33 -38
  53. package/lib/Types/Newsletter.js.bak +33 -0
  54. package/lib/Types/Product.js +3 -2
  55. package/lib/Types/Signal.js +3 -2
  56. package/lib/Types/Socket.js +4 -2
  57. package/lib/Types/State.js +11 -2
  58. package/lib/Types/USync.js +3 -2
  59. package/lib/Types/index.js +27 -41
  60. package/lib/Utils/auth-utils.js +211 -198
  61. package/lib/Utils/baileys-event-stream.js +42 -61
  62. package/lib/Utils/browser-utils.js +25 -0
  63. package/lib/Utils/business.js +213 -214
  64. package/lib/Utils/chat-utils.js +710 -687
  65. package/lib/Utils/crypto.js +112 -133
  66. package/lib/Utils/decode-wa-message.js +252 -183
  67. package/lib/Utils/decode-wa-message.js.bak +267 -0
  68. package/lib/Utils/event-buffer.js +510 -496
  69. package/lib/Utils/generics.js +319 -392
  70. package/lib/Utils/history.js +83 -92
  71. package/lib/Utils/index.js +21 -33
  72. package/lib/Utils/link-preview.js +71 -83
  73. package/lib/Utils/logger.js +5 -7
  74. package/lib/Utils/lt-hash.js +40 -46
  75. package/lib/Utils/make-mutex.js +34 -41
  76. package/lib/Utils/message-retry-manager.js +113 -0
  77. package/lib/Utils/messages-media.js +550 -768
  78. package/lib/Utils/messages.js +354 -263
  79. package/lib/Utils/noise-handler.js +138 -149
  80. package/lib/Utils/pre-key-manager.js +85 -0
  81. package/lib/Utils/process-message.js +323 -303
  82. package/lib/Utils/signal.js +149 -141
  83. package/lib/Utils/use-multi-file-auth-state.js +95 -103
  84. package/lib/Utils/validate-connection.js +183 -214
  85. package/lib/WABinary/constants.js +1298 -35
  86. package/lib/WABinary/decode.js +237 -249
  87. package/lib/WABinary/encode.js +213 -260
  88. package/lib/WABinary/generic-utils.js +56 -65
  89. package/lib/WABinary/index.js +7 -21
  90. package/lib/WABinary/jid-utils.js +89 -58
  91. package/lib/WABinary/types.js +3 -2
  92. package/lib/WAM/BinaryInfo.js +10 -12
  93. package/lib/WAM/constants.js +22851 -15348
  94. package/lib/WAM/encode.js +135 -136
  95. package/lib/WAM/index.js +5 -19
  96. package/lib/WAUSync/Protocols/USyncContactProtocol.js +28 -30
  97. package/lib/WAUSync/Protocols/USyncDeviceProtocol.js +49 -53
  98. package/lib/WAUSync/Protocols/USyncDisappearingModeProtocol.js +27 -28
  99. package/lib/WAUSync/Protocols/USyncStatusProtocol.js +36 -39
  100. package/lib/WAUSync/Protocols/UsyncBotProfileProtocol.js +50 -50
  101. package/lib/WAUSync/Protocols/UsyncLIDProtocol.js +26 -20
  102. package/lib/WAUSync/Protocols/index.js +6 -20
  103. package/lib/WAUSync/USyncQuery.js +86 -85
  104. package/lib/WAUSync/USyncUser.js +23 -25
  105. package/lib/WAUSync/index.js +5 -19
  106. package/lib/index.js +27 -35
  107. package/package.json +85 -95
  108. package/engine-requirements.js +0 -10
  109. package/lib/Defaults/baileys-version.json +0 -3
  110. package/lib/Defaults/index.d.ts +0 -53
  111. package/lib/Defaults/phonenumber-mcc.json +0 -223
  112. package/lib/Signal/Group/ciphertext-message.d.ts +0 -9
  113. package/lib/Signal/Group/group-session-builder.d.ts +0 -14
  114. package/lib/Signal/Group/group_cipher.d.ts +0 -17
  115. package/lib/Signal/Group/index.d.ts +0 -11
  116. package/lib/Signal/Group/keyhelper.d.ts +0 -10
  117. package/lib/Signal/Group/queue-job.d.ts +0 -1
  118. package/lib/Signal/Group/queue-job.js +0 -57
  119. package/lib/Signal/Group/sender-chain-key.d.ts +0 -13
  120. package/lib/Signal/Group/sender-key-distribution-message.d.ts +0 -16
  121. package/lib/Signal/Group/sender-key-message.d.ts +0 -18
  122. package/lib/Signal/Group/sender-key-name.d.ts +0 -17
  123. package/lib/Signal/Group/sender-key-record.d.ts +0 -30
  124. package/lib/Signal/Group/sender-key-state.d.ts +0 -38
  125. package/lib/Signal/Group/sender-message-key.d.ts +0 -11
  126. package/lib/Signal/libsignal.d.ts +0 -3
  127. package/lib/Socket/Client/abstract-socket-client.d.ts +0 -17
  128. package/lib/Socket/Client/abstract-socket-client.js +0 -13
  129. package/lib/Socket/Client/index.d.ts +0 -3
  130. package/lib/Socket/Client/mobile-socket-client.d.ts +0 -13
  131. package/lib/Socket/Client/mobile-socket-client.js +0 -65
  132. package/lib/Socket/Client/web-socket-client.d.ts +0 -12
  133. package/lib/Socket/Client/web-socket-client.js +0 -62
  134. package/lib/Socket/business.d.ts +0 -171
  135. package/lib/Socket/chats.d.ts +0 -267
  136. package/lib/Socket/dugong.d.ts +0 -254
  137. package/lib/Socket/dugong.js +0 -484
  138. package/lib/Socket/groups.d.ts +0 -115
  139. package/lib/Socket/index.d.ts +0 -173
  140. package/lib/Socket/messages-recv.d.ts +0 -161
  141. package/lib/Socket/messages-send.d.ts +0 -149
  142. package/lib/Socket/newsletter.d.ts +0 -134
  143. package/lib/Socket/registration.d.ts +0 -267
  144. package/lib/Socket/registration.js +0 -166
  145. package/lib/Socket/socket.d.ts +0 -43
  146. package/lib/Socket/usync.d.ts +0 -36
  147. package/lib/Socket/usync.js +0 -70
  148. package/lib/Store/index.d.ts +0 -3
  149. package/lib/Store/make-cache-manager-store.d.ts +0 -13
  150. package/lib/Store/make-in-memory-store.d.ts +0 -118
  151. package/lib/Store/make-ordered-dictionary.d.ts +0 -13
  152. package/lib/Store/object-repository.d.ts +0 -10
  153. package/lib/Types/Auth.d.ts +0 -110
  154. package/lib/Types/Call.d.ts +0 -13
  155. package/lib/Types/Chat.d.ts +0 -102
  156. package/lib/Types/Contact.d.ts +0 -19
  157. package/lib/Types/Events.d.ts +0 -157
  158. package/lib/Types/GroupMetadata.d.ts +0 -55
  159. package/lib/Types/Label.d.ts +0 -35
  160. package/lib/Types/LabelAssociation.d.ts +0 -29
  161. package/lib/Types/Message.d.ts +0 -273
  162. package/lib/Types/Newsletter.d.ts +0 -103
  163. package/lib/Types/Product.d.ts +0 -78
  164. package/lib/Types/Signal.d.ts +0 -57
  165. package/lib/Types/Socket.d.ts +0 -111
  166. package/lib/Types/State.d.ts +0 -27
  167. package/lib/Types/USync.d.ts +0 -25
  168. package/lib/Types/index.d.ts +0 -57
  169. package/lib/Utils/auth-utils.d.ts +0 -18
  170. package/lib/Utils/baileys-event-stream.d.ts +0 -16
  171. package/lib/Utils/business.d.ts +0 -22
  172. package/lib/Utils/chat-utils.d.ts +0 -71
  173. package/lib/Utils/crypto.d.ts +0 -41
  174. package/lib/Utils/decode-wa-message.d.ts +0 -19
  175. package/lib/Utils/event-buffer.d.ts +0 -35
  176. package/lib/Utils/generics.d.ts +0 -92
  177. package/lib/Utils/generics.js.bak +0 -433
  178. package/lib/Utils/history.d.ts +0 -15
  179. package/lib/Utils/index.d.ts +0 -17
  180. package/lib/Utils/link-preview.d.ts +0 -21
  181. package/lib/Utils/logger.d.ts +0 -4
  182. package/lib/Utils/lt-hash.d.ts +0 -12
  183. package/lib/Utils/make-mutex.d.ts +0 -7
  184. package/lib/Utils/messages-media.d.ts +0 -116
  185. package/lib/Utils/messages.d.ts +0 -77
  186. package/lib/Utils/noise-handler.d.ts +0 -21
  187. package/lib/Utils/process-message.d.ts +0 -41
  188. package/lib/Utils/signal.d.ts +0 -32
  189. package/lib/Utils/use-multi-file-auth-state.d.ts +0 -13
  190. package/lib/Utils/validate-connection.d.ts +0 -11
  191. package/lib/Utils/validate-connection.js.bak +0 -237
  192. package/lib/WABinary/constants.d.ts +0 -30
  193. package/lib/WABinary/decode.d.ts +0 -7
  194. package/lib/WABinary/encode.d.ts +0 -3
  195. package/lib/WABinary/generic-utils.d.ts +0 -17
  196. package/lib/WABinary/index.d.ts +0 -5
  197. package/lib/WABinary/jid-utils.d.ts +0 -31
  198. package/lib/WABinary/types.d.ts +0 -18
  199. package/lib/WAM/BinaryInfo.d.ts +0 -17
  200. package/lib/WAM/constants.d.ts +0 -38
  201. package/lib/WAM/encode.d.ts +0 -3
  202. package/lib/WAM/index.d.ts +0 -3
  203. package/lib/WAUSync/Protocols/USyncContactProtocol.d.ts +0 -9
  204. package/lib/WAUSync/Protocols/USyncDeviceProtocol.d.ts +0 -22
  205. package/lib/WAUSync/Protocols/USyncDisappearingModeProtocol.d.ts +0 -12
  206. package/lib/WAUSync/Protocols/USyncStatusProtocol.d.ts +0 -12
  207. package/lib/WAUSync/Protocols/UsyncBotProfileProtocol.d.ts +0 -25
  208. package/lib/WAUSync/Protocols/UsyncLIDProtocol.d.ts +0 -8
  209. package/lib/WAUSync/Protocols/index.d.ts +0 -4
  210. package/lib/WAUSync/USyncQuery.d.ts +0 -28
  211. package/lib/WAUSync/USyncUser.d.ts +0 -12
  212. package/lib/WAUSync/index.d.ts +0 -3
  213. package/lib/index.d.ts +0 -12
@@ -1,819 +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
- 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
- }
228
- }
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
- }
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;
282
140
  }
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;
141
+ //=======================================================//
142
+ export async function getAudioWaveform(buffer, logger) {
143
+ try {
144
+ const { default: decoder } = await import("audio-decode");
145
+ let audioData;
146
+ if (Buffer.isBuffer(buffer)) {
147
+ audioData = buffer;
148
+ }
149
+ else if (typeof buffer === "string") {
150
+ const rStream = createReadStream(buffer);
151
+ audioData = await toBuffer(rStream);
320
152
  }
153
+ else {
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
+ }
321
177
  }
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;
178
+ //=======================================================//
179
+ export const toReadable = (buffer) => {
180
+ const readable = new Readable({ read: () => { } });
181
+ readable.push(buffer);
182
+ readable.push(null);
183
+ return readable;
328
184
  };
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);
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);
337
193
  };
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' };
350
- };
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;
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" };
389
211
  };
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;
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");
396
228
  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
- }
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
+ };
242
+ }
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);
433
254
  };
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;
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");
447
292
  }
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
-
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();
472
325
  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;
326
+ await fs.unlink(encFilePath);
327
+ if (originalFilePath) {
328
+ await fs.unlink(originalFilePath);
329
+ }
531
330
  }
532
-
533
- function onChunk(buff) {
534
- sha256Enc = sha256Enc.update(buff);
535
- hmac = hmac.update(buff);
536
- encWriteStream.push(buff);
331
+ catch (err) {
332
+ logger?.error({ err }, "failed deleting tmp files");
537
333
  }
334
+ throw error;
335
+ }
538
336
  };
539
- exports.encryptedStream = encryptedStream;
540
- const DEF_HOST = 'mmg.whatsapp.net';
337
+ //=======================================================//
338
+ const DEF_HOST = "mmg.whatsapp.net";
541
339
  const AES_CHUNK_SIZE = 16;
542
340
  const toSmallestChunkSize = (num) => {
543
- return Math.floor(num / AES_CHUNK_SIZE) * AES_CHUNK_SIZE;
341
+ return Math.floor(num / AES_CHUNK_SIZE) * AES_CHUNK_SIZE;
544
342
  };
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);
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);
551
353
  };
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
- }
569
- }
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
- }
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;
580
395
  }
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;
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);
596
411
  }
597
- else {
598
- push(bytes);
412
+ aes = Crypto.createDecipheriv("aes-256-cbc", cipherKey, ivValue);
413
+ if (endByte) {
414
+ aes.setAutoPadding(false);
599
415
  }
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 });
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 });
639
436
  };
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;
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;
655
450
  }
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
- }
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();
670
485
  }
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-');
486
+ catch {
487
+ parsed = undefined;
676
488
  }
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
- }
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;
720
499
  }
721
- if (!urls) {
722
- 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)}`);
723
503
  }
724
- return urls;
725
- };
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
+ };
726
515
  };
727
- exports.getWAUploadToServer = getWAUploadToServer;
516
+ //=======================================================//
728
517
  const getMediaRetryKey = (mediaKey) => {
729
- return (0, crypto_1.hkdf)(mediaKey, 32, { info: 'WhatsApp Media Retry Notification' });
518
+ return hkdf(mediaKey, 32, { info: "WhatsApp Media Retry Notification" });
730
519
  };
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
- },
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: {},
747
538
  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
- }
539
+ { tag: "enc_p", attrs: {}, content: ciphertext },
540
+ { tag: "enc_iv", attrs: {}, content: iv }
768
541
  ]
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
542
+ },
543
+ {
544
+ tag: "rmr",
545
+ attrs: {
546
+ jid: key.remoteJid,
547
+ from_me: (!!key.fromMe).toString(),
548
+ participant: key.participant || undefined
781
549
  }
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) });
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 };
787
580
  }
788
581
  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
- }
582
+ event.error = new Boom("Failed to re-upload media (missing ciphertext)", { statusCode: 404 });
798
583
  }
799
- return event;
584
+ }
585
+ return event;
800
586
  };
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);
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);
806
592
  };
807
- exports.decryptMediaRetryData = decryptMediaRetryData;
808
- const getStatusCodeForMediaRetry = (code) => MEDIA_RETRY_STATUS_MAP[code];
809
- exports.getStatusCodeForMediaRetry = getStatusCodeForMediaRetry;
593
+ //=======================================================//
594
+ export const getStatusCodeForMediaRetry = (code) => MEDIA_RETRY_STATUS_MAP[code];
810
595
  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,
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
815
600
  };
816
- // eslint-disable-next-line @typescript-eslint/no-unused-vars
817
- function __importStar(arg0) {
818
- throw new Error('Function not implemented.');
819
- }
601
+ //=======================================================//