@sixcore/baileys 1.0.0 → 1.0.2

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 (228) hide show
  1. package/WAProto/index.js +14270 -302
  2. package/jessica.js +91 -0
  3. package/lib/Defaults/baileys-version.json +1 -1
  4. package/lib/Defaults/index.js +117 -79
  5. package/lib/Defaults/phonenumber-mcc.json +223 -0
  6. package/lib/Signal/Group/ciphertext-message.d.ts +9 -0
  7. package/lib/Signal/Group/ciphertext-message.js +15 -0
  8. package/lib/Signal/Group/group-session-builder.d.ts +14 -0
  9. package/lib/Signal/Group/group-session-builder.js +64 -0
  10. package/lib/Signal/Group/group_cipher.d.ts +17 -0
  11. package/lib/Signal/Group/group_cipher.js +96 -0
  12. package/lib/Signal/Group/index.d.ts +11 -0
  13. package/lib/Signal/Group/index.js +57 -0
  14. package/lib/Signal/Group/keyhelper.d.ts +10 -0
  15. package/lib/Signal/Group/keyhelper.js +55 -0
  16. package/lib/Signal/Group/queue-job.d.ts +1 -0
  17. package/lib/Signal/Group/queue-job.js +57 -0
  18. package/lib/Signal/Group/sender-chain-key.d.ts +13 -0
  19. package/lib/Signal/Group/sender-chain-key.js +34 -0
  20. package/lib/Signal/Group/sender-key-distribution-message.d.ts +16 -0
  21. package/lib/Signal/Group/sender-key-distribution-message.js +66 -0
  22. package/lib/Signal/Group/sender-key-message.d.ts +18 -0
  23. package/lib/Signal/Group/sender-key-message.js +69 -0
  24. package/lib/Signal/Group/sender-key-name.d.ts +17 -0
  25. package/lib/Signal/Group/sender-key-name.js +51 -0
  26. package/lib/Signal/Group/sender-key-record.d.ts +30 -0
  27. package/lib/Signal/Group/sender-key-record.js +53 -0
  28. package/lib/Signal/Group/sender-key-state.d.ts +38 -0
  29. package/lib/Signal/Group/sender-key-state.js +99 -0
  30. package/lib/Signal/Group/sender-message-key.d.ts +11 -0
  31. package/{WASignalGroup/sender_message_key.js → lib/Signal/Group/sender-message-key.js} +6 -16
  32. package/lib/Signal/libsignal.js +51 -29
  33. package/lib/Socket/business.d.ts +43 -42
  34. package/lib/Socket/chats.d.ts +222 -36
  35. package/lib/Socket/chats.js +173 -153
  36. package/lib/Socket/dugong.d.ts +254 -0
  37. package/lib/Socket/dugong.js +484 -0
  38. package/lib/Socket/groups.d.ts +7 -7
  39. package/lib/Socket/groups.js +37 -35
  40. package/lib/Socket/index.d.ts +52 -51
  41. package/lib/Socket/index.js +1 -0
  42. package/lib/Socket/messages-recv.d.ts +37 -34
  43. package/lib/Socket/messages-recv.js +175 -37
  44. package/lib/Socket/messages-send.d.ts +12 -18
  45. package/lib/Socket/messages-send.js +396 -574
  46. package/lib/Socket/newsletter.d.ts +28 -26
  47. package/lib/Socket/newsletter.js +132 -121
  48. package/lib/Socket/registration.d.ts +52 -49
  49. package/lib/Socket/registration.js +7 -7
  50. package/lib/Socket/socket.d.ts +0 -1
  51. package/lib/Socket/socket.js +49 -27
  52. package/lib/Socket/usync.d.ts +10 -11
  53. package/lib/Store/make-cache-manager-store.d.ts +1 -2
  54. package/lib/Store/make-in-memory-store.d.ts +2 -2
  55. package/lib/Store/make-in-memory-store.js +1 -5
  56. package/lib/Store/make-ordered-dictionary.js +2 -2
  57. package/lib/Types/Auth.d.ts +1 -0
  58. package/lib/Types/Call.d.ts +1 -1
  59. package/lib/Types/Chat.d.ts +7 -12
  60. package/lib/Types/Events.d.ts +2 -17
  61. package/lib/Types/GroupMetadata.d.ts +2 -3
  62. package/lib/Types/Label.d.ts +0 -11
  63. package/lib/Types/Label.js +1 -1
  64. package/lib/Types/LabelAssociation.js +1 -1
  65. package/lib/Types/Message.d.ts +10 -170
  66. package/lib/Types/Newsletter.d.ts +97 -86
  67. package/lib/Types/Newsletter.js +38 -32
  68. package/lib/Types/Socket.d.ts +2 -7
  69. package/lib/Types/index.d.ts +0 -9
  70. package/lib/Types/index.js +1 -1
  71. package/lib/Utils/auth-utils.js +14 -35
  72. package/lib/Utils/business.d.ts +1 -1
  73. package/lib/Utils/business.js +2 -2
  74. package/lib/Utils/chat-utils.d.ts +12 -11
  75. package/lib/Utils/chat-utils.js +36 -52
  76. package/lib/Utils/crypto.d.ts +16 -15
  77. package/lib/Utils/crypto.js +26 -74
  78. package/lib/Utils/decode-wa-message.d.ts +0 -17
  79. package/lib/Utils/decode-wa-message.js +17 -53
  80. package/lib/Utils/event-buffer.js +7 -10
  81. package/lib/Utils/generics.d.ts +17 -13
  82. package/lib/Utils/generics.js +79 -58
  83. package/lib/Utils/history.d.ts +2 -6
  84. package/lib/Utils/history.js +6 -4
  85. package/lib/Utils/logger.d.ts +3 -1
  86. package/lib/Utils/lt-hash.js +12 -12
  87. package/lib/Utils/make-mutex.d.ts +2 -2
  88. package/lib/Utils/messages-media.d.ts +28 -25
  89. package/lib/Utils/messages-media.js +733 -557
  90. package/lib/Utils/messages.js +68 -473
  91. package/lib/Utils/noise-handler.d.ts +5 -4
  92. package/lib/Utils/noise-handler.js +14 -19
  93. package/lib/Utils/process-message.d.ts +5 -5
  94. package/lib/Utils/process-message.js +23 -75
  95. package/lib/Utils/signal.d.ts +1 -2
  96. package/lib/Utils/signal.js +26 -32
  97. package/lib/Utils/use-multi-file-auth-state.d.ts +1 -0
  98. package/lib/Utils/use-multi-file-auth-state.js +66 -242
  99. package/lib/Utils/validate-connection.d.ts +1 -1
  100. package/lib/Utils/validate-connection.js +88 -64
  101. package/lib/WABinary/constants.d.ts +27 -24
  102. package/lib/WABinary/decode.d.ts +2 -1
  103. package/lib/WABinary/decode.js +11 -23
  104. package/lib/WABinary/encode.d.ts +2 -1
  105. package/lib/WABinary/encode.js +147 -134
  106. package/lib/WABinary/generic-utils.d.ts +5 -2
  107. package/lib/WABinary/generic-utils.js +125 -37
  108. package/lib/WABinary/jid-utils.d.ts +1 -1
  109. package/lib/WAM/BinaryInfo.d.ts +11 -2
  110. package/lib/WAM/encode.d.ts +2 -1
  111. package/lib/WAUSync/Protocols/USyncStatusProtocol.js +3 -3
  112. package/lib/WAUSync/USyncUser.d.ts +2 -0
  113. package/lib/index.d.ts +12 -0
  114. package/lib/index.js +64 -1
  115. package/package.json +113 -51
  116. package/WAProto/GenerateStatics.sh +0 -4
  117. package/WAProto/WAProto.proto +0 -4357
  118. package/WAProto/index.d.ts +0 -50383
  119. package/WASignalGroup/GroupProtocol.js +0 -1697
  120. package/WASignalGroup/ciphertext_message.js +0 -16
  121. package/WASignalGroup/generate-proto.sh +0 -1
  122. package/WASignalGroup/group.proto +0 -42
  123. package/WASignalGroup/group_cipher.js +0 -120
  124. package/WASignalGroup/group_session_builder.js +0 -46
  125. package/WASignalGroup/index.js +0 -5
  126. package/WASignalGroup/keyhelper.js +0 -21
  127. package/WASignalGroup/protobufs.js +0 -3
  128. package/WASignalGroup/queue_job.js +0 -69
  129. package/WASignalGroup/sender_chain_key.js +0 -50
  130. package/WASignalGroup/sender_key_distribution_message.js +0 -78
  131. package/WASignalGroup/sender_key_message.js +0 -92
  132. package/WASignalGroup/sender_key_name.js +0 -70
  133. package/WASignalGroup/sender_key_record.js +0 -56
  134. package/WASignalGroup/sender_key_state.js +0 -129
  135. package/lib/Utils/use-single-file-auth-state.d.ts +0 -12
  136. package/lib/Utils/use-single-file-auth-state.js +0 -75
  137. package/src/Defaults/baileys-version.json +0 -3
  138. package/src/Defaults/index.ts +0 -133
  139. package/src/Signal/Group/ciphertext-message.ts +0 -9
  140. package/src/Signal/Group/group-session-builder.ts +0 -56
  141. package/src/Signal/Group/group_cipher.ts +0 -117
  142. package/src/Signal/Group/index.ts +0 -11
  143. package/src/Signal/Group/keyhelper.ts +0 -28
  144. package/src/Signal/Group/sender-chain-key.ts +0 -34
  145. package/src/Signal/Group/sender-key-distribution-message.ts +0 -95
  146. package/src/Signal/Group/sender-key-message.ts +0 -96
  147. package/src/Signal/Group/sender-key-name.ts +0 -66
  148. package/src/Signal/Group/sender-key-record.ts +0 -69
  149. package/src/Signal/Group/sender-key-state.ts +0 -134
  150. package/src/Signal/Group/sender-message-key.ts +0 -36
  151. package/src/Signal/libsignal.ts +0 -447
  152. package/src/Signal/lid-mapping.ts +0 -209
  153. package/src/Socket/Client/index.ts +0 -2
  154. package/src/Socket/Client/types.ts +0 -22
  155. package/src/Socket/Client/websocket.ts +0 -56
  156. package/src/Socket/business.ts +0 -421
  157. package/src/Socket/chats.ts +0 -1223
  158. package/src/Socket/communities.ts +0 -477
  159. package/src/Socket/groups.ts +0 -361
  160. package/src/Socket/index.ts +0 -22
  161. package/src/Socket/messages-recv.ts +0 -1563
  162. package/src/Socket/messages-send.ts +0 -1210
  163. package/src/Socket/mex.ts +0 -58
  164. package/src/Socket/newsletter.ts +0 -229
  165. package/src/Socket/socket.ts +0 -1072
  166. package/src/Types/Auth.ts +0 -115
  167. package/src/Types/Bussines.ts +0 -20
  168. package/src/Types/Call.ts +0 -14
  169. package/src/Types/Chat.ts +0 -138
  170. package/src/Types/Contact.ts +0 -24
  171. package/src/Types/Events.ts +0 -132
  172. package/src/Types/GroupMetadata.ts +0 -70
  173. package/src/Types/Label.ts +0 -48
  174. package/src/Types/LabelAssociation.ts +0 -35
  175. package/src/Types/Message.ts +0 -424
  176. package/src/Types/Newsletter.ts +0 -98
  177. package/src/Types/Product.ts +0 -85
  178. package/src/Types/Signal.ts +0 -76
  179. package/src/Types/Socket.ts +0 -150
  180. package/src/Types/State.ts +0 -43
  181. package/src/Types/USync.ts +0 -27
  182. package/src/Types/globals.d.ts +0 -8
  183. package/src/Types/index.ts +0 -67
  184. package/src/Utils/auth-utils.ts +0 -331
  185. package/src/Utils/browser-utils.ts +0 -31
  186. package/src/Utils/business.ts +0 -286
  187. package/src/Utils/chat-utils.ts +0 -933
  188. package/src/Utils/crypto.ts +0 -184
  189. package/src/Utils/decode-wa-message.ts +0 -355
  190. package/src/Utils/event-buffer.ts +0 -662
  191. package/src/Utils/generics.ts +0 -470
  192. package/src/Utils/history.ts +0 -114
  193. package/src/Utils/index.ts +0 -18
  194. package/src/Utils/link-preview.ts +0 -111
  195. package/src/Utils/logger.ts +0 -13
  196. package/src/Utils/lt-hash.ts +0 -65
  197. package/src/Utils/make-mutex.ts +0 -45
  198. package/src/Utils/message-retry-manager.ts +0 -229
  199. package/src/Utils/messages-media.ts +0 -820
  200. package/src/Utils/messages.ts +0 -1137
  201. package/src/Utils/noise-handler.ts +0 -192
  202. package/src/Utils/pre-key-manager.ts +0 -126
  203. package/src/Utils/process-message.ts +0 -622
  204. package/src/Utils/signal.ts +0 -214
  205. package/src/Utils/use-multi-file-auth-state.ts +0 -136
  206. package/src/Utils/validate-connection.ts +0 -253
  207. package/src/WABinary/constants.ts +0 -1305
  208. package/src/WABinary/decode.ts +0 -281
  209. package/src/WABinary/encode.ts +0 -253
  210. package/src/WABinary/generic-utils.ts +0 -127
  211. package/src/WABinary/index.ts +0 -5
  212. package/src/WABinary/jid-utils.ts +0 -128
  213. package/src/WABinary/types.ts +0 -17
  214. package/src/WAM/BinaryInfo.ts +0 -12
  215. package/src/WAM/constants.ts +0 -22889
  216. package/src/WAM/encode.ts +0 -169
  217. package/src/WAM/index.ts +0 -3
  218. package/src/WAUSync/Protocols/USyncContactProtocol.ts +0 -32
  219. package/src/WAUSync/Protocols/USyncDeviceProtocol.ts +0 -78
  220. package/src/WAUSync/Protocols/USyncDisappearingModeProtocol.ts +0 -35
  221. package/src/WAUSync/Protocols/USyncStatusProtocol.ts +0 -44
  222. package/src/WAUSync/Protocols/UsyncBotProfileProtocol.ts +0 -76
  223. package/src/WAUSync/Protocols/UsyncLIDProtocol.ts +0 -33
  224. package/src/WAUSync/Protocols/index.ts +0 -4
  225. package/src/WAUSync/USyncQuery.ts +0 -133
  226. package/src/WAUSync/USyncUser.ts +0 -32
  227. package/src/WAUSync/index.ts +0 -3
  228. package/src/index.ts +0 -13
@@ -1,1137 +0,0 @@
1
- import { Boom } from '@hapi/boom'
2
- import { randomBytes } from 'crypto'
3
- import { promises as fs } from 'fs'
4
- import { type Transform } from 'stream'
5
- import { proto } from '../../WAProto/index.js'
6
- import {
7
- CALL_AUDIO_PREFIX,
8
- CALL_VIDEO_PREFIX,
9
- MEDIA_KEYS,
10
- type MediaType,
11
- URL_REGEX,
12
- WA_DEFAULT_EPHEMERAL
13
- } from '../Defaults'
14
- import type {
15
- AnyMediaMessageContent,
16
- AnyMessageContent,
17
- DownloadableMessage,
18
- MessageContentGenerationOptions,
19
- MessageGenerationOptions,
20
- MessageGenerationOptionsFromContent,
21
- MessageUserReceipt,
22
- MessageWithContextInfo,
23
- WAMediaUpload,
24
- WAMessage,
25
- WAMessageContent,
26
- WAMessageKey,
27
- WATextMessage
28
- } from '../Types'
29
- import { WAMessageStatus, WAProto } from '../Types'
30
- import { isJidGroup, isJidNewsletter, isJidStatusBroadcast, jidNormalizedUser } from '../WABinary'
31
- import { sha256 } from './crypto'
32
- import { generateMessageIDV2, getKeyAuthor, unixTimestampSeconds } from './generics'
33
- import type { ILogger } from './logger'
34
- import {
35
- downloadContentFromMessage,
36
- encryptedStream,
37
- generateThumbnail,
38
- getAudioDuration,
39
- getAudioWaveform,
40
- getRawMediaUploadData,
41
- type MediaDownloadOptions
42
- } from './messages-media'
43
-
44
- type MediaUploadData = {
45
- media: WAMediaUpload
46
- caption?: string
47
- ptt?: boolean
48
- ptv?: boolean
49
- seconds?: number
50
- gifPlayback?: boolean
51
- fileName?: string
52
- jpegThumbnail?: string
53
- mimetype?: string
54
- width?: number
55
- height?: number
56
- waveform?: Uint8Array
57
- backgroundArgb?: number
58
- }
59
-
60
- const MIMETYPE_MAP: { [T in MediaType]?: string } = {
61
- image: 'image/jpeg',
62
- video: 'video/mp4',
63
- document: 'application/pdf',
64
- audio: 'audio/ogg; codecs=opus',
65
- sticker: 'image/webp',
66
- 'product-catalog-image': 'image/jpeg'
67
- }
68
-
69
- const MessageTypeProto = {
70
- image: WAProto.Message.ImageMessage,
71
- video: WAProto.Message.VideoMessage,
72
- audio: WAProto.Message.AudioMessage,
73
- sticker: WAProto.Message.StickerMessage,
74
- document: WAProto.Message.DocumentMessage
75
- } as const
76
-
77
- /**
78
- * Uses a regex to test whether the string contains a URL, and returns the URL if it does.
79
- * @param text eg. hello https://google.com
80
- * @returns the URL, eg. https://google.com
81
- */
82
- export const extractUrlFromText = (text: string) => text.match(URL_REGEX)?.[0]
83
-
84
- export const generateLinkPreviewIfRequired = async (
85
- text: string,
86
- getUrlInfo: MessageGenerationOptions['getUrlInfo'],
87
- logger: MessageGenerationOptions['logger']
88
- ) => {
89
- const url = extractUrlFromText(text)
90
- if (!!getUrlInfo && url) {
91
- try {
92
- const urlInfo = await getUrlInfo(url)
93
- return urlInfo
94
- } catch (error: any) {
95
- // ignore if fails
96
- logger?.warn({ trace: error.stack }, 'url generation failed')
97
- }
98
- }
99
- }
100
-
101
- const assertColor = async (color: any) => {
102
- let assertedColor
103
- if (typeof color === 'number') {
104
- assertedColor = color > 0 ? color : 0xffffffff + Number(color) + 1
105
- } else {
106
- let hex = color.trim().replace('#', '')
107
- if (hex.length <= 6) {
108
- hex = 'FF' + hex.padStart(6, '0')
109
- }
110
-
111
- assertedColor = parseInt(hex, 16)
112
- return assertedColor
113
- }
114
- }
115
-
116
- export const prepareWAMessageMedia = async (
117
- message: AnyMediaMessageContent,
118
- options: MessageContentGenerationOptions
119
- ) => {
120
- const logger = options.logger
121
-
122
- let mediaType: (typeof MEDIA_KEYS)[number] | undefined
123
- for (const key of MEDIA_KEYS) {
124
- if (key in message) {
125
- mediaType = key
126
- }
127
- }
128
-
129
- if (!mediaType) {
130
- throw new Boom('Invalid media type', { statusCode: 400 })
131
- }
132
-
133
- const uploadData: MediaUploadData = {
134
- ...message,
135
- media: (message as any)[mediaType]
136
- }
137
- delete (uploadData as any)[mediaType]
138
- // check if cacheable + generate cache key
139
- const cacheableKey =
140
- typeof uploadData.media === 'object' &&
141
- 'url' in uploadData.media &&
142
- !!uploadData.media.url &&
143
- !!options.mediaCache &&
144
- mediaType + ':' + uploadData.media.url.toString()
145
-
146
- if (mediaType === 'document' && !uploadData.fileName) {
147
- uploadData.fileName = 'file'
148
- }
149
-
150
- if (!uploadData.mimetype) {
151
- uploadData.mimetype = MIMETYPE_MAP[mediaType]
152
- }
153
-
154
- if (cacheableKey) {
155
- const mediaBuff = await options.mediaCache!.get<Buffer>(cacheableKey)
156
- if (mediaBuff) {
157
- logger?.debug({ cacheableKey }, 'got media cache hit')
158
-
159
- const obj = proto.Message.decode(mediaBuff)
160
- const key = `${mediaType}Message`
161
-
162
- Object.assign(obj[key as keyof proto.Message]!, { ...uploadData, media: undefined })
163
-
164
- return obj
165
- }
166
- }
167
-
168
- const isNewsletter = !!options.jid && isJidNewsletter(options.jid)
169
- if (isNewsletter) {
170
- logger?.info({ key: cacheableKey }, 'Preparing raw media for newsletter')
171
- const { filePath, fileSha256, fileLength } = await getRawMediaUploadData(
172
- uploadData.media,
173
- options.mediaTypeOverride || mediaType,
174
- logger
175
- )
176
-
177
- const fileSha256B64 = fileSha256.toString('base64')
178
- const { mediaUrl, directPath } = await options.upload(filePath, {
179
- fileEncSha256B64: fileSha256B64,
180
- mediaType: mediaType,
181
- timeoutMs: options.mediaUploadTimeoutMs
182
- })
183
-
184
- await fs.unlink(filePath)
185
-
186
- const obj = WAProto.Message.fromObject({
187
- // todo: add more support here
188
- [`${mediaType}Message`]: (MessageTypeProto as any)[mediaType].fromObject({
189
- url: mediaUrl,
190
- directPath,
191
- fileSha256,
192
- fileLength,
193
- ...uploadData,
194
- media: undefined
195
- })
196
- })
197
-
198
- if (uploadData.ptv) {
199
- obj.ptvMessage = obj.videoMessage
200
- delete obj.videoMessage
201
- }
202
-
203
- if (obj.stickerMessage) {
204
- obj.stickerMessage.stickerSentTs = Date.now()
205
- }
206
-
207
- if (cacheableKey) {
208
- logger?.debug({ cacheableKey }, 'set cache')
209
- await options.mediaCache!.set(cacheableKey, WAProto.Message.encode(obj).finish())
210
- }
211
-
212
- return obj
213
- }
214
-
215
- const requiresDurationComputation = mediaType === 'audio' && typeof uploadData.seconds === 'undefined'
216
- const requiresThumbnailComputation =
217
- (mediaType === 'image' || mediaType === 'video') && typeof uploadData['jpegThumbnail'] === 'undefined'
218
- const requiresWaveformProcessing = mediaType === 'audio' && uploadData.ptt === true
219
- const requiresAudioBackground = options.backgroundColor && mediaType === 'audio' && uploadData.ptt === true
220
- const requiresOriginalForSomeProcessing = requiresDurationComputation || requiresThumbnailComputation
221
- const { mediaKey, encFilePath, originalFilePath, fileEncSha256, fileSha256, fileLength } = await encryptedStream(
222
- uploadData.media,
223
- options.mediaTypeOverride || mediaType,
224
- {
225
- logger,
226
- saveOriginalFileIfRequired: requiresOriginalForSomeProcessing,
227
- opts: options.options
228
- }
229
- )
230
-
231
- const fileEncSha256B64 = fileEncSha256.toString('base64')
232
- const [{ mediaUrl, directPath }] = await Promise.all([
233
- (async () => {
234
- const result = await options.upload(encFilePath, {
235
- fileEncSha256B64,
236
- mediaType,
237
- timeoutMs: options.mediaUploadTimeoutMs
238
- })
239
- logger?.debug({ mediaType, cacheableKey }, 'uploaded media')
240
- return result
241
- })(),
242
- (async () => {
243
- try {
244
- if (requiresThumbnailComputation) {
245
- const { thumbnail, originalImageDimensions } = await generateThumbnail(
246
- originalFilePath!,
247
- mediaType as 'image' | 'video',
248
- options
249
- )
250
- uploadData.jpegThumbnail = thumbnail
251
- if (!uploadData.width && originalImageDimensions) {
252
- uploadData.width = originalImageDimensions.width
253
- uploadData.height = originalImageDimensions.height
254
- logger?.debug('set dimensions')
255
- }
256
-
257
- logger?.debug('generated thumbnail')
258
- }
259
-
260
- if (requiresDurationComputation) {
261
- uploadData.seconds = await getAudioDuration(originalFilePath!)
262
- logger?.debug('computed audio duration')
263
- }
264
-
265
- if (requiresWaveformProcessing) {
266
- uploadData.waveform = await getAudioWaveform(originalFilePath!, logger)
267
- logger?.debug('processed waveform')
268
- }
269
-
270
- if (requiresAudioBackground) {
271
- uploadData.backgroundArgb = await assertColor(options.backgroundColor)
272
- logger?.debug('computed backgroundColor audio status')
273
- }
274
- } catch (error) {
275
- logger?.warn({ trace: (error as any).stack }, 'failed to obtain extra info')
276
- }
277
- })()
278
- ]).finally(async () => {
279
- try {
280
- await fs.unlink(encFilePath)
281
- if (originalFilePath) {
282
- await fs.unlink(originalFilePath)
283
- }
284
-
285
- logger?.debug('removed tmp files')
286
- } catch (error) {
287
- logger?.warn('failed to remove tmp file')
288
- }
289
- })
290
-
291
- const obj = WAProto.Message.fromObject({
292
- [`${mediaType}Message`]: MessageTypeProto[mediaType as keyof typeof MessageTypeProto].fromObject({
293
- url: mediaUrl,
294
- directPath,
295
- mediaKey,
296
- fileEncSha256,
297
- fileSha256,
298
- fileLength,
299
- mediaKeyTimestamp: unixTimestampSeconds(),
300
- ...uploadData,
301
- media: undefined
302
- } as any)
303
- })
304
-
305
- if (uploadData.ptv) {
306
- obj.ptvMessage = obj.videoMessage
307
- delete obj.videoMessage
308
- }
309
-
310
- if (cacheableKey) {
311
- logger?.debug({ cacheableKey }, 'set cache')
312
- await options.mediaCache!.set(cacheableKey, WAProto.Message.encode(obj).finish())
313
- }
314
-
315
- return obj
316
- }
317
-
318
- export const prepareDisappearingMessageSettingContent = (ephemeralExpiration?: number) => {
319
- ephemeralExpiration = ephemeralExpiration || 0
320
- const content: WAMessageContent = {
321
- ephemeralMessage: {
322
- message: {
323
- protocolMessage: {
324
- type: WAProto.Message.ProtocolMessage.Type.EPHEMERAL_SETTING,
325
- ephemeralExpiration
326
- }
327
- }
328
- }
329
- }
330
- return WAProto.Message.fromObject(content)
331
- }
332
-
333
- /**
334
- * Generate forwarded message content like WA does
335
- * @param message the message to forward
336
- * @param options.forceForward will show the message as forwarded even if it is from you
337
- */
338
- export const generateForwardMessageContent = (message: WAMessage, forceForward?: boolean) => {
339
- let content = message.message
340
- if (!content) {
341
- throw new Boom('no content in message', { statusCode: 400 })
342
- }
343
-
344
- // hacky copy
345
- content = normalizeMessageContent(content)
346
- content = proto.Message.decode(proto.Message.encode(content!).finish())
347
-
348
- let key = Object.keys(content)[0] as keyof proto.IMessage
349
-
350
- let score = (content?.[key] as { contextInfo: proto.IContextInfo })?.contextInfo?.forwardingScore || 0
351
- score += message.key.fromMe && !forceForward ? 0 : 1
352
- if (key === 'conversation') {
353
- content.extendedTextMessage = { text: content[key] }
354
- delete content.conversation
355
-
356
- key = 'extendedTextMessage'
357
- }
358
-
359
- const key_ = content?.[key] as { contextInfo: proto.IContextInfo }
360
- if (score > 0) {
361
- key_.contextInfo = { forwardingScore: score, isForwarded: true }
362
- } else {
363
- key_.contextInfo = {}
364
- }
365
-
366
- return content
367
- }
368
-
369
- export const generateWAMessageContent = async (
370
- message: AnyMessageContent,
371
- options: MessageContentGenerationOptions
372
- ) => {
373
- let m: WAMessageContent = {}
374
- if ('text' in message) {
375
- const extContent = { text: message.text } as WATextMessage
376
-
377
- let urlInfo = message.linkPreview
378
- if (typeof urlInfo === 'undefined') {
379
- urlInfo = await generateLinkPreviewIfRequired(message.text, options.getUrlInfo, options.logger)
380
- }
381
-
382
- if (urlInfo) {
383
- extContent.matchedText = urlInfo['matched-text']
384
- extContent.jpegThumbnail = urlInfo.jpegThumbnail
385
- extContent.description = urlInfo.description
386
- extContent.title = urlInfo.title
387
- extContent.previewType = 0
388
-
389
- const img = urlInfo.highQualityThumbnail
390
- if (img) {
391
- extContent.thumbnailDirectPath = img.directPath
392
- extContent.mediaKey = img.mediaKey
393
- extContent.mediaKeyTimestamp = img.mediaKeyTimestamp
394
- extContent.thumbnailWidth = img.width
395
- extContent.thumbnailHeight = img.height
396
- extContent.thumbnailSha256 = img.fileSha256
397
- extContent.thumbnailEncSha256 = img.fileEncSha256
398
- }
399
- }
400
-
401
- if (options.backgroundColor) {
402
- extContent.backgroundArgb = await assertColor(options.backgroundColor)
403
- }
404
-
405
- if (options.font) {
406
- extContent.font = options.font
407
- }
408
-
409
- m.extendedTextMessage = extContent
410
- } else if ('contacts' in message) {
411
- const contactLen = message.contacts.contacts.length
412
- if (!contactLen) {
413
- throw new Boom('require atleast 1 contact', { statusCode: 400 })
414
- }
415
-
416
- if (contactLen === 1) {
417
- m.contactMessage = WAProto.Message.ContactMessage.create(message.contacts.contacts[0])
418
- } else {
419
- m.contactsArrayMessage = WAProto.Message.ContactsArrayMessage.create(message.contacts)
420
- }
421
- } else if ('location' in message) {
422
- m.locationMessage = WAProto.Message.LocationMessage.create(message.location)
423
- } else if ('react' in message) {
424
- if (!message.react.senderTimestampMs) {
425
- message.react.senderTimestampMs = Date.now()
426
- }
427
-
428
- m.reactionMessage = WAProto.Message.ReactionMessage.create(message.react)
429
- } else if ('delete' in message) {
430
- m.protocolMessage = {
431
- key: message.delete,
432
- type: WAProto.Message.ProtocolMessage.Type.REVOKE
433
- }
434
- } else if ('forward' in message) {
435
- m = generateForwardMessageContent(message.forward, message.force)
436
- } else if ('disappearingMessagesInChat' in message) {
437
- const exp =
438
- typeof message.disappearingMessagesInChat === 'boolean'
439
- ? message.disappearingMessagesInChat
440
- ? WA_DEFAULT_EPHEMERAL
441
- : 0
442
- : message.disappearingMessagesInChat
443
- m = prepareDisappearingMessageSettingContent(exp)
444
- } else if ('groupInvite' in message) {
445
- m.groupInviteMessage = {}
446
- m.groupInviteMessage.inviteCode = message.groupInvite.inviteCode
447
- m.groupInviteMessage.inviteExpiration = message.groupInvite.inviteExpiration
448
- m.groupInviteMessage.caption = message.groupInvite.text
449
-
450
- m.groupInviteMessage.groupJid = message.groupInvite.jid
451
- m.groupInviteMessage.groupName = message.groupInvite.subject
452
- //TODO: use built-in interface and get disappearing mode info etc.
453
- //TODO: cache / use store!?
454
- if (options.getProfilePicUrl) {
455
- const pfpUrl = await options.getProfilePicUrl(message.groupInvite.jid, 'preview')
456
- if (pfpUrl) {
457
- const resp = await fetch(pfpUrl, { method: 'GET', dispatcher: options?.options?.dispatcher })
458
- if (resp.ok) {
459
- const buf = Buffer.from(await resp.arrayBuffer())
460
- m.groupInviteMessage.jpegThumbnail = buf
461
- }
462
- }
463
- }
464
- } else if ('pin' in message) {
465
- m.pinInChatMessage = {}
466
- m.messageContextInfo = {}
467
-
468
- m.pinInChatMessage.key = message.pin
469
- m.pinInChatMessage.type = message.type
470
- m.pinInChatMessage.senderTimestampMs = Date.now()
471
-
472
- m.messageContextInfo.messageAddOnDurationInSecs = message.type === 1 ? message.time || 86400 : 0
473
- } else if ('buttonReply' in message) {
474
- switch (message.type) {
475
- case 'template':
476
- m.templateButtonReplyMessage = {
477
- selectedDisplayText: message.buttonReply.displayText,
478
- selectedId: message.buttonReply.id,
479
- selectedIndex: message.buttonReply.index
480
- }
481
- break
482
- case 'plain':
483
- m.buttonsResponseMessage = {
484
- selectedButtonId: message.buttonReply.id,
485
- selectedDisplayText: message.buttonReply.displayText,
486
- type: proto.Message.ButtonsResponseMessage.Type.DISPLAY_TEXT
487
- }
488
- break
489
- }
490
- } else if ('ptv' in message && message.ptv) {
491
- const { videoMessage } = await prepareWAMessageMedia({ video: message.video }, options)
492
- m.ptvMessage = videoMessage
493
- } else if ('product' in message) {
494
- const { imageMessage } = await prepareWAMessageMedia({ image: message.product.productImage }, options)
495
- m.productMessage = WAProto.Message.ProductMessage.create({
496
- ...message,
497
- product: {
498
- ...message.product,
499
- productImage: imageMessage
500
- }
501
- })
502
- } else if ('listReply' in message) {
503
- m.listResponseMessage = { ...message.listReply }
504
- } else if ('event' in message) {
505
- m.eventMessage = {}
506
- const startTime = Math.floor(message.event.startDate.getTime() / 1000)
507
-
508
- if (message.event.call && options.getCallLink) {
509
- const token = await options.getCallLink(message.event.call, { startTime })
510
- m.eventMessage.joinLink = (message.event.call === 'audio' ? CALL_AUDIO_PREFIX : CALL_VIDEO_PREFIX) + token
511
- }
512
-
513
- m.messageContextInfo = {
514
- // encKey
515
- messageSecret: message.event.messageSecret || randomBytes(32)
516
- }
517
-
518
- m.eventMessage.name = message.event.name
519
- m.eventMessage.description = message.event.description
520
- m.eventMessage.startTime = startTime
521
- m.eventMessage.endTime = message.event.endDate ? message.event.endDate.getTime() / 1000 : undefined
522
- m.eventMessage.isCanceled = message.event.isCancelled ?? false
523
- m.eventMessage.extraGuestsAllowed = message.event.extraGuestsAllowed
524
- m.eventMessage.isScheduleCall = message.event.isScheduleCall ?? false
525
- m.eventMessage.location = message.event.location
526
- } else if ('poll' in message) {
527
- message.poll.selectableCount ||= 0
528
- message.poll.toAnnouncementGroup ||= false
529
-
530
- if (!Array.isArray(message.poll.values)) {
531
- throw new Boom('Invalid poll values', { statusCode: 400 })
532
- }
533
-
534
- if (message.poll.selectableCount < 0 || message.poll.selectableCount > message.poll.values.length) {
535
- throw new Boom(`poll.selectableCount in poll should be >= 0 and <= ${message.poll.values.length}`, {
536
- statusCode: 400
537
- })
538
- }
539
-
540
- m.messageContextInfo = {
541
- // encKey
542
- messageSecret: message.poll.messageSecret || randomBytes(32)
543
- }
544
-
545
- const pollCreationMessage = {
546
- name: message.poll.name,
547
- selectableOptionsCount: message.poll.selectableCount,
548
- options: message.poll.values.map(optionName => ({ optionName }))
549
- }
550
-
551
- if (message.poll.toAnnouncementGroup) {
552
- // poll v2 is for community announcement groups (single select and multiple)
553
- m.pollCreationMessageV2 = pollCreationMessage
554
- } else {
555
- if (message.poll.selectableCount === 1) {
556
- //poll v3 is for single select polls
557
- m.pollCreationMessageV3 = pollCreationMessage
558
- } else {
559
- // poll for multiple choice polls
560
- m.pollCreationMessage = pollCreationMessage
561
- }
562
- }
563
- } else if ('sharePhoneNumber' in message) {
564
- m.protocolMessage = {
565
- type: proto.Message.ProtocolMessage.Type.SHARE_PHONE_NUMBER
566
- }
567
- } else if ('requestPhoneNumber' in message) {
568
- m.requestPhoneNumberMessage = {}
569
- } else if ('limitSharing' in message) {
570
- m.protocolMessage = {
571
- type: proto.Message.ProtocolMessage.Type.LIMIT_SHARING,
572
- limitSharing: {
573
- sharingLimited: message.limitSharing === true,
574
- trigger: 1,
575
- limitSharingSettingTimestamp: Date.now(),
576
- initiatedByMe: true
577
- }
578
- }
579
-
580
- // ─── PATCH: Botões ─────────────────────────────────────────────────────────
581
-
582
- } else if ('buttons' in message) {
583
- // Botões clássicos (legacy buttonsMessage)
584
- const { text, footer = '', title = '', buttons } = (message as any).buttons
585
- m.buttonsMessage = {
586
- contentText: text,
587
- footerText: footer,
588
- headerType: title
589
- ? proto.Message.ButtonsMessage.HeaderType.TEXT
590
- : proto.Message.ButtonsMessage.HeaderType.EMPTY,
591
- text: title || undefined,
592
- buttons: buttons.map((b: any, i: number) => ({
593
- buttonId: b.id || `btn_${i}`,
594
- buttonText: { displayText: b.text },
595
- type: proto.Message.ButtonsMessage.Button.Type.RESPONSE,
596
- })),
597
- }
598
-
599
- } else if ('templateButtons' in message) {
600
- // Template Buttons (URL, call, quickReply)
601
- const { text, footer = '', title = '', buttons } = (message as any).templateButtons
602
- m.templateMessage = {
603
- hydratedTemplate: {
604
- hydratedContentText: text,
605
- hydratedFooterText: footer,
606
- ...(title ? { hydratedTitleText: title } : {}),
607
- hydratedButtons: buttons.map((b: any) => {
608
- if (b.type === 'url') {
609
- return { urlButton: { displayText: b.text, url: b.url } }
610
- }
611
- if (b.type === 'call') {
612
- return { callButton: { displayText: b.text, phoneNumber: b.phone } }
613
- }
614
- return { quickReplyButton: { displayText: b.text, id: b.id || b.text } }
615
- }),
616
- },
617
- }
618
-
619
- } else if ('list' in message) {
620
- // List Message com seções
621
- const { title = '', text, footer = '', buttonText, sections } = (message as any).list
622
- m.listMessage = {
623
- title,
624
- description: text,
625
- footerText: footer,
626
- buttonText,
627
- listType: proto.Message.ListMessage.ListType.SINGLE_SELECT,
628
- sections: sections.map((s: any) => ({
629
- title: s.title || '',
630
- rows: s.rows.map((r: any) => ({
631
- rowId: r.id || r.title,
632
- title: r.title,
633
- description: r.description || '',
634
- })),
635
- })),
636
- }
637
-
638
- } else if ('interactive' in message) {
639
- // Interactive Message — novo protocolo MD 2024+
640
- const { body, footer = '', header = '', nativeFlowButtons } = (message as any).interactive
641
- m.interactiveMessage = {
642
- body: { text: body },
643
- footer: { text: footer },
644
- header: {
645
- title: header || '',
646
- hasMediaAttachment: false,
647
- },
648
- nativeFlowMessage: {
649
- buttons: nativeFlowButtons,
650
- messageParamsJson: '',
651
- },
652
- }
653
-
654
- } else {
655
- m = await prepareWAMessageMedia(message, options)
656
- }
657
-
658
- if ('viewOnce' in message && !!message.viewOnce) {
659
- m = { viewOnceMessage: { message: m } }
660
- }
661
-
662
- if ('mentions' in message && message.mentions?.length) {
663
- const messageType = Object.keys(m)[0]! as Extract<keyof proto.IMessage, MessageWithContextInfo>
664
- const key = m[messageType]
665
- if ('contextInfo' in key! && !!key.contextInfo) {
666
- key.contextInfo.mentionedJid = message.mentions
667
- } else if (key!) {
668
- key.contextInfo = {
669
- mentionedJid: message.mentions
670
- }
671
- }
672
- }
673
-
674
- if ('edit' in message) {
675
- m = {
676
- protocolMessage: {
677
- key: message.edit,
678
- editedMessage: m,
679
- timestampMs: Date.now(),
680
- type: WAProto.Message.ProtocolMessage.Type.MESSAGE_EDIT
681
- }
682
- }
683
- }
684
-
685
- if ('contextInfo' in message && !!message.contextInfo) {
686
- const messageType = Object.keys(m)[0]! as Extract<keyof proto.IMessage, MessageWithContextInfo>
687
- const key = m[messageType]
688
- if ('contextInfo' in key! && !!key.contextInfo) {
689
- key.contextInfo = { ...key.contextInfo, ...message.contextInfo }
690
- } else if (key!) {
691
- key.contextInfo = message.contextInfo
692
- }
693
- }
694
-
695
- return WAProto.Message.create(m)
696
- }
697
-
698
- export const generateWAMessageFromContent = (
699
- jid: string,
700
- message: WAMessageContent,
701
- options: MessageGenerationOptionsFromContent
702
- ) => {
703
- // set timestamp to now
704
- // if not specified
705
- if (!options.timestamp) {
706
- options.timestamp = new Date()
707
- }
708
-
709
- const innerMessage = normalizeMessageContent(message)!
710
- const key = getContentType(innerMessage)! as Exclude<keyof proto.IMessage, 'conversation'>
711
- const timestamp = unixTimestampSeconds(options.timestamp)
712
- const { quoted, userJid } = options
713
-
714
- if (quoted && !isJidNewsletter(jid)) {
715
- const participant = quoted.key.fromMe
716
- ? userJid // TODO: Add support for LIDs
717
- : quoted.participant || quoted.key.participant || quoted.key.remoteJid
718
-
719
- let quotedMsg = normalizeMessageContent(quoted.message)!
720
- const msgType = getContentType(quotedMsg)!
721
- // strip any redundant properties
722
- quotedMsg = proto.Message.create({ [msgType]: quotedMsg[msgType] })
723
-
724
- const quotedContent = quotedMsg[msgType]
725
- if (typeof quotedContent === 'object' && quotedContent && 'contextInfo' in quotedContent) {
726
- delete quotedContent.contextInfo
727
- }
728
-
729
- const contextInfo: proto.IContextInfo =
730
- ('contextInfo' in innerMessage[key]! && innerMessage[key]?.contextInfo) || {}
731
- contextInfo.participant = jidNormalizedUser(participant!)
732
- contextInfo.stanzaId = quoted.key.id
733
- contextInfo.quotedMessage = quotedMsg
734
-
735
- // if a participant is quoted, then it must be a group
736
- // hence, remoteJid of group must also be entered
737
- if (jid !== quoted.key.remoteJid) {
738
- contextInfo.remoteJid = quoted.key.remoteJid
739
- }
740
-
741
- if (contextInfo && innerMessage[key]) {
742
- /* @ts-ignore */
743
- innerMessage[key].contextInfo = contextInfo
744
- }
745
- }
746
-
747
- if (
748
- // if we want to send a disappearing message
749
- !!options?.ephemeralExpiration &&
750
- // and it's not a protocol message -- delete, toggle disappear message
751
- key !== 'protocolMessage' &&
752
- // already not converted to disappearing message
753
- key !== 'ephemeralMessage' &&
754
- // newsletters don't support ephemeral messages
755
- !isJidNewsletter(jid)
756
- ) {
757
- /* @ts-ignore */
758
- innerMessage[key].contextInfo = {
759
- ...((innerMessage[key] as any).contextInfo || {}),
760
- expiration: options.ephemeralExpiration || WA_DEFAULT_EPHEMERAL
761
- //ephemeralSettingTimestamp: options.ephemeralOptions.eph_setting_ts?.toString()
762
- }
763
- }
764
-
765
- message = WAProto.Message.create(message)
766
-
767
- const messageJSON = {
768
- key: {
769
- remoteJid: jid,
770
- fromMe: true,
771
- id: options?.messageId || generateMessageIDV2()
772
- },
773
- message: message,
774
- messageTimestamp: timestamp,
775
- messageStubParameters: [],
776
- participant: isJidGroup(jid) || isJidStatusBroadcast(jid) ? userJid : undefined, // TODO: Add support for LIDs
777
- status: WAMessageStatus.PENDING
778
- }
779
- return WAProto.WebMessageInfo.fromObject(messageJSON) as WAMessage
780
- }
781
-
782
- export const generateWAMessage = async (jid: string, content: AnyMessageContent, options: MessageGenerationOptions) => {
783
- // ensure msg ID is with every log
784
- options.logger = options?.logger?.child({ msgId: options.messageId })
785
- // Pass jid in the options to generateWAMessageContent
786
- return generateWAMessageFromContent(jid, await generateWAMessageContent(content, { ...options, jid }), options)
787
- }
788
-
789
- /** Get the key to access the true type of content */
790
- export const getContentType = (content: proto.IMessage | undefined) => {
791
- if (content) {
792
- const keys = Object.keys(content)
793
- const key = keys.find(k => (k === 'conversation' || k.includes('Message')) && k !== 'senderKeyDistributionMessage')
794
- return key as keyof typeof content
795
- }
796
- }
797
-
798
- /**
799
- * Normalizes ephemeral, view once messages to regular message content
800
- * Eg. image messages in ephemeral messages, in view once messages etc.
801
- * @param content
802
- * @returns
803
- */
804
- export const normalizeMessageContent = (content: WAMessageContent | null | undefined): WAMessageContent | undefined => {
805
- if (!content) {
806
- return undefined
807
- }
808
-
809
- // set max iterations to prevent an infinite loop
810
- for (let i = 0; i < 5; i++) {
811
- const inner = getFutureProofMessage(content)
812
- if (!inner) {
813
- break
814
- }
815
-
816
- content = inner.message
817
- }
818
-
819
- return content!
820
-
821
- function getFutureProofMessage(message: typeof content) {
822
- return (
823
- message?.ephemeralMessage ||
824
- message?.viewOnceMessage ||
825
- message?.documentWithCaptionMessage ||
826
- message?.viewOnceMessageV2 ||
827
- message?.viewOnceMessageV2Extension ||
828
- message?.editedMessage
829
- )
830
- }
831
- }
832
-
833
- /**
834
- * Extract the true message content from a message
835
- * Eg. extracts the inner message from a disappearing message/view once message
836
- */
837
- export const extractMessageContent = (content: WAMessageContent | undefined | null): WAMessageContent | undefined => {
838
- const extractFromTemplateMessage = (
839
- msg: proto.Message.TemplateMessage.IHydratedFourRowTemplate | proto.Message.IButtonsMessage
840
- ) => {
841
- if (msg.imageMessage) {
842
- return { imageMessage: msg.imageMessage }
843
- } else if (msg.documentMessage) {
844
- return { documentMessage: msg.documentMessage }
845
- } else if (msg.videoMessage) {
846
- return { videoMessage: msg.videoMessage }
847
- } else if (msg.locationMessage) {
848
- return { locationMessage: msg.locationMessage }
849
- } else {
850
- return {
851
- conversation:
852
- 'contentText' in msg ? msg.contentText : 'hydratedContentText' in msg ? msg.hydratedContentText : ''
853
- }
854
- }
855
- }
856
-
857
- content = normalizeMessageContent(content)
858
-
859
- if (content?.buttonsMessage) {
860
- return extractFromTemplateMessage(content.buttonsMessage)
861
- }
862
-
863
- if (content?.templateMessage?.hydratedFourRowTemplate) {
864
- return extractFromTemplateMessage(content?.templateMessage?.hydratedFourRowTemplate)
865
- }
866
-
867
- if (content?.templateMessage?.hydratedTemplate) {
868
- return extractFromTemplateMessage(content?.templateMessage?.hydratedTemplate)
869
- }
870
-
871
- if (content?.templateMessage?.fourRowTemplate) {
872
- return extractFromTemplateMessage(content?.templateMessage?.fourRowTemplate)
873
- }
874
-
875
- return content
876
- }
877
-
878
- /**
879
- * Returns the device predicted by message ID
880
- */
881
- export const getDevice = (id: string) =>
882
- /^3A.{18}$/.test(id)
883
- ? 'ios'
884
- : /^3E.{20}$/.test(id)
885
- ? 'web'
886
- : /^(.{21}|.{32})$/.test(id)
887
- ? 'android'
888
- : /^(3F|.{18}$)/.test(id)
889
- ? 'desktop'
890
- : 'unknown'
891
-
892
- /** Upserts a receipt in the message */
893
- export const updateMessageWithReceipt = (msg: Pick<WAMessage, 'userReceipt'>, receipt: MessageUserReceipt) => {
894
- msg.userReceipt = msg.userReceipt || []
895
- const recp = msg.userReceipt.find(m => m.userJid === receipt.userJid)
896
- if (recp) {
897
- Object.assign(recp, receipt)
898
- } else {
899
- msg.userReceipt.push(receipt)
900
- }
901
- }
902
-
903
- /** Update the message with a new reaction */
904
- export const updateMessageWithReaction = (msg: Pick<WAMessage, 'reactions'>, reaction: proto.IReaction) => {
905
- const authorID = getKeyAuthor(reaction.key)
906
-
907
- const reactions = (msg.reactions || []).filter(r => getKeyAuthor(r.key) !== authorID)
908
- reaction.text = reaction.text || ''
909
- reactions.push(reaction)
910
- msg.reactions = reactions
911
- }
912
-
913
- /** Update the message with a new poll update */
914
- export const updateMessageWithPollUpdate = (msg: Pick<WAMessage, 'pollUpdates'>, update: proto.IPollUpdate) => {
915
- const authorID = getKeyAuthor(update.pollUpdateMessageKey)
916
-
917
- const reactions = (msg.pollUpdates || []).filter(r => getKeyAuthor(r.pollUpdateMessageKey) !== authorID)
918
- if (update.vote?.selectedOptions?.length) {
919
- reactions.push(update)
920
- }
921
-
922
- msg.pollUpdates = reactions
923
- }
924
-
925
- /** Update the message with a new event response */
926
- export const updateMessageWithEventResponse = (
927
- msg: Pick<WAMessage, 'eventResponses'>,
928
- update: proto.IEventResponse
929
- ) => {
930
- const authorID = getKeyAuthor(update.eventResponseMessageKey)
931
-
932
- const responses = (msg.eventResponses || []).filter(r => getKeyAuthor(r.eventResponseMessageKey) !== authorID)
933
- responses.push(update)
934
-
935
- msg.eventResponses = responses
936
- }
937
-
938
- type VoteAggregation = {
939
- name: string
940
- voters: string[]
941
- }
942
-
943
- /**
944
- * Aggregates all poll updates in a poll.
945
- * @param msg the poll creation message
946
- * @param meId your jid
947
- * @returns A list of options & their voters
948
- */
949
- export function getAggregateVotesInPollMessage(
950
- { message, pollUpdates }: Pick<WAMessage, 'pollUpdates' | 'message'>,
951
- meId?: string
952
- ) {
953
- const opts =
954
- message?.pollCreationMessage?.options ||
955
- message?.pollCreationMessageV2?.options ||
956
- message?.pollCreationMessageV3?.options ||
957
- []
958
- const voteHashMap = opts.reduce(
959
- (acc, opt) => {
960
- const hash = sha256(Buffer.from(opt.optionName || '')).toString()
961
- acc[hash] = {
962
- name: opt.optionName || '',
963
- voters: []
964
- }
965
- return acc
966
- },
967
- {} as { [_: string]: VoteAggregation }
968
- )
969
-
970
- for (const update of pollUpdates || []) {
971
- const { vote } = update
972
- if (!vote) {
973
- continue
974
- }
975
-
976
- for (const option of vote.selectedOptions || []) {
977
- const hash = option.toString()
978
- let data = voteHashMap[hash]
979
- if (!data) {
980
- voteHashMap[hash] = {
981
- name: 'Unknown',
982
- voters: []
983
- }
984
- data = voteHashMap[hash]
985
- }
986
-
987
- voteHashMap[hash]!.voters.push(getKeyAuthor(update.pollUpdateMessageKey, meId))
988
- }
989
- }
990
-
991
- return Object.values(voteHashMap)
992
- }
993
-
994
- type ResponseAggregation = {
995
- response: string
996
- responders: string[]
997
- }
998
-
999
- /**
1000
- * Aggregates all event responses in an event message.
1001
- * @param msg the event creation message
1002
- * @param meId your jid
1003
- * @returns A list of response types & their responders
1004
- */
1005
- export function getAggregateResponsesInEventMessage(
1006
- { eventResponses }: Pick<WAMessage, 'eventResponses'>,
1007
- meId?: string
1008
- ) {
1009
- const responseTypes = ['GOING', 'NOT_GOING', 'MAYBE']
1010
- const responseMap: { [_: string]: ResponseAggregation } = {}
1011
-
1012
- for (const type of responseTypes) {
1013
- responseMap[type] = {
1014
- response: type,
1015
- responders: []
1016
- }
1017
- }
1018
-
1019
- for (const update of eventResponses || []) {
1020
- const responseType = (update as any).eventResponse || 'UNKNOWN'
1021
- if (responseType !== 'UNKNOWN' && responseMap[responseType]) {
1022
- responseMap[responseType].responders.push(getKeyAuthor(update.eventResponseMessageKey, meId))
1023
- }
1024
- }
1025
-
1026
- return Object.values(responseMap)
1027
- }
1028
-
1029
- /** Given a list of message keys, aggregates them by chat & sender. Useful for sending read receipts in bulk */
1030
- export const aggregateMessageKeysNotFromMe = (keys: WAMessageKey[]) => {
1031
- const keyMap: { [id: string]: { jid: string; participant: string | undefined; messageIds: string[] } } = {}
1032
- for (const { remoteJid, id, participant, fromMe } of keys) {
1033
- if (!fromMe) {
1034
- const uqKey = `${remoteJid}:${participant || ''}`
1035
- if (!keyMap[uqKey]) {
1036
- keyMap[uqKey] = {
1037
- jid: remoteJid!,
1038
- participant: participant!,
1039
- messageIds: []
1040
- }
1041
- }
1042
-
1043
- keyMap[uqKey].messageIds.push(id!)
1044
- }
1045
- }
1046
-
1047
- return Object.values(keyMap)
1048
- }
1049
-
1050
- type DownloadMediaMessageContext = {
1051
- reuploadRequest: (msg: WAMessage) => Promise<WAMessage>
1052
- logger: ILogger
1053
- }
1054
-
1055
- const REUPLOAD_REQUIRED_STATUS = [410, 404]
1056
-
1057
- /**
1058
- * Downloads the given message. Throws an error if it's not a media message
1059
- */
1060
- export const downloadMediaMessage = async <Type extends 'buffer' | 'stream'>(
1061
- message: WAMessage,
1062
- type: Type,
1063
- options: MediaDownloadOptions,
1064
- ctx?: DownloadMediaMessageContext
1065
- ) => {
1066
- const result = await downloadMsg().catch(async error => {
1067
- if (
1068
- ctx &&
1069
- typeof error?.status === 'number' && // treat errors with status as HTTP failures requiring reupload
1070
- REUPLOAD_REQUIRED_STATUS.includes(error.status as number)
1071
- ) {
1072
- ctx.logger.info({ key: message.key }, 'sending reupload media request...')
1073
- // request reupload
1074
- message = await ctx.reuploadRequest(message)
1075
- const result = await downloadMsg()
1076
- return result
1077
- }
1078
-
1079
- throw error
1080
- })
1081
-
1082
- return result as Type extends 'buffer' ? Buffer : Transform
1083
-
1084
- async function downloadMsg() {
1085
- const mContent = extractMessageContent(message.message)
1086
- if (!mContent) {
1087
- throw new Boom('No message present', { statusCode: 400, data: message })
1088
- }
1089
-
1090
- const contentType = getContentType(mContent)
1091
- let mediaType = contentType?.replace('Message', '') as MediaType
1092
- const media = mContent[contentType!]
1093
-
1094
- if (!media || typeof media !== 'object' || (!('url' in media) && !('thumbnailDirectPath' in media))) {
1095
- throw new Boom(`"${contentType}" message is not a media message`)
1096
- }
1097
-
1098
- let download: DownloadableMessage
1099
- if ('thumbnailDirectPath' in media && !('url' in media)) {
1100
- download = {
1101
- directPath: media.thumbnailDirectPath,
1102
- mediaKey: media.mediaKey
1103
- }
1104
- mediaType = 'thumbnail-link'
1105
- } else {
1106
- download = media
1107
- }
1108
-
1109
- const stream = await downloadContentFromMessage(download, mediaType, options)
1110
- if (type === 'buffer') {
1111
- const bufferArray: Buffer[] = []
1112
- for await (const chunk of stream) {
1113
- bufferArray.push(chunk)
1114
- }
1115
-
1116
- return Buffer.concat(bufferArray)
1117
- }
1118
-
1119
- return stream
1120
- }
1121
- }
1122
-
1123
- /** Checks whether the given message is a media message; if it is returns the inner content */
1124
- export const assertMediaContent = (content: proto.IMessage | null | undefined) => {
1125
- content = extractMessageContent(content)
1126
- const mediaContent =
1127
- content?.documentMessage ||
1128
- content?.imageMessage ||
1129
- content?.videoMessage ||
1130
- content?.audioMessage ||
1131
- content?.stickerMessage
1132
- if (!mediaContent) {
1133
- throw new Boom('given message is not a media message', { statusCode: 400, data: content })
1134
- }
1135
-
1136
- return mediaContent
1137
- }