zapo-js 0.1.2 → 0.3.0

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (630) hide show
  1. package/README.md +16 -4
  2. package/dist/appstate/WaAppStateCrypto.js +50 -42
  3. package/dist/appstate/WaAppStateSyncClient.js +215 -133
  4. package/dist/appstate/{store/sqlite.js → encoding.js} +13 -8
  5. package/dist/appstate/index.js +9 -7
  6. package/dist/appstate/utils.js +0 -5
  7. package/dist/auth/WaAuthClient.js +55 -57
  8. package/dist/auth/credentials-flow.js +195 -0
  9. package/dist/auth/index.js +1 -6
  10. package/dist/auth/pairing/WaPairingFlow.js +39 -32
  11. package/dist/auth/pairing/{WaPairingCodeCrypto.js → pairing-code-crypto.js} +35 -17
  12. package/dist/client/WaClient.js +338 -174
  13. package/dist/client/WaClientFactory.js +399 -66
  14. package/dist/client/connection/WaConnectionManager.js +23 -11
  15. package/dist/client/coordinators/WaAbPropsCoordinator.js +141 -0
  16. package/dist/client/coordinators/WaBusinessCoordinator.js +232 -0
  17. package/dist/client/coordinators/WaEmailCoordinator.js +63 -0
  18. package/dist/client/coordinators/WaGroupCoordinator.js +11 -7
  19. package/dist/client/coordinators/WaIncomingNodeCoordinator.js +34 -8
  20. package/dist/client/coordinators/WaMessageDispatchCoordinator.js +341 -118
  21. package/dist/client/coordinators/WaOfflineResumeCoordinator.js +114 -0
  22. package/dist/client/coordinators/WaPassiveTasksCoordinator.js +97 -36
  23. package/dist/client/coordinators/WaPrivacyCoordinator.js +134 -0
  24. package/dist/client/coordinators/WaProfileCoordinator.js +214 -0
  25. package/dist/client/coordinators/WaRetryCoordinator.js +184 -30
  26. package/dist/client/coordinators/WaStreamControlCoordinator.js +18 -11
  27. package/dist/client/coordinators/WaTrustedContactTokenCoordinator.js +184 -0
  28. package/dist/client/dirty.js +41 -21
  29. package/dist/client/events/abprops.js +43 -0
  30. package/dist/client/events/devices.js +72 -0
  31. package/dist/client/events/group.js +3 -11
  32. package/dist/client/events/identity.js +22 -0
  33. package/dist/client/events/privacy-token.js +38 -0
  34. package/dist/client/events/registration.js +42 -0
  35. package/dist/client/history-sync.js +50 -9
  36. package/dist/client/incoming.js +74 -7
  37. package/dist/client/mailbox.js +40 -23
  38. package/dist/client/media.js +243 -0
  39. package/dist/client/messages.js +245 -92
  40. package/dist/client/messaging/fanout.js +21 -11
  41. package/dist/client/messaging/participants.js +6 -4
  42. package/dist/client/persistence/WriteBehindPersistence.js +129 -0
  43. package/dist/client/tokens/cs-token.js +50 -0
  44. package/dist/client/tokens/tc-token.js +25 -0
  45. package/dist/crypto/core/index.js +5 -2
  46. package/dist/crypto/core/keys.js +4 -4
  47. package/dist/crypto/core/nonce.js +2 -0
  48. package/dist/crypto/core/primitives.js +0 -8
  49. package/dist/crypto/core/random.js +24 -8
  50. package/dist/crypto/core/xeddsa.js +57 -0
  51. package/dist/crypto/curves/X25519.js +43 -6
  52. package/dist/crypto/curves/constants.js +2 -1
  53. package/dist/crypto/index.js +3 -0
  54. package/dist/crypto/math/constants.js +13 -36
  55. package/dist/crypto/math/edwards.js +171 -44
  56. package/dist/crypto/math/fe.js +706 -0
  57. package/dist/crypto/math/mod.js +10 -3
  58. package/dist/esm/appstate/WaAppStateCrypto.js +40 -32
  59. package/dist/esm/appstate/WaAppStateSyncClient.js +206 -124
  60. package/dist/esm/appstate/{store/sqlite.js → encoding.js} +13 -8
  61. package/dist/esm/appstate/index.js +2 -2
  62. package/dist/esm/appstate/{WaAppStateSyncResponseParser.js → response-parser.js} +1 -1
  63. package/dist/esm/appstate/utils.js +2 -5
  64. package/dist/esm/auth/WaAuthClient.js +52 -54
  65. package/dist/esm/auth/credentials-flow.js +190 -0
  66. package/dist/esm/auth/index.js +0 -2
  67. package/dist/esm/auth/pairing/WaPairingFlow.js +39 -32
  68. package/dist/esm/auth/pairing/{WaPairingCodeCrypto.js → pairing-code-crypto.js} +26 -10
  69. package/dist/esm/client/WaClient.js +339 -175
  70. package/dist/esm/client/WaClientFactory.js +401 -68
  71. package/dist/esm/client/connection/WaConnectionManager.js +23 -11
  72. package/dist/esm/client/coordinators/WaAbPropsCoordinator.js +137 -0
  73. package/dist/esm/client/coordinators/WaBusinessCoordinator.js +229 -0
  74. package/dist/esm/client/coordinators/WaEmailCoordinator.js +60 -0
  75. package/dist/esm/client/coordinators/WaGroupCoordinator.js +11 -7
  76. package/dist/esm/client/coordinators/WaIncomingNodeCoordinator.js +36 -10
  77. package/dist/esm/client/coordinators/WaMessageDispatchCoordinator.js +337 -114
  78. package/dist/esm/client/coordinators/WaOfflineResumeCoordinator.js +110 -0
  79. package/dist/esm/client/coordinators/WaPassiveTasksCoordinator.js +97 -36
  80. package/dist/esm/client/coordinators/WaPrivacyCoordinator.js +131 -0
  81. package/dist/esm/client/coordinators/WaProfileCoordinator.js +211 -0
  82. package/dist/esm/client/coordinators/WaRetryCoordinator.js +186 -32
  83. package/dist/esm/client/coordinators/WaStreamControlCoordinator.js +19 -12
  84. package/dist/esm/client/coordinators/WaTrustedContactTokenCoordinator.js +180 -0
  85. package/dist/esm/client/dirty.js +41 -21
  86. package/dist/esm/client/events/abprops.js +40 -0
  87. package/dist/esm/client/events/devices.js +68 -0
  88. package/dist/esm/client/events/group.js +3 -11
  89. package/dist/esm/client/events/identity.js +19 -0
  90. package/dist/esm/client/events/privacy-token.js +35 -0
  91. package/dist/esm/client/events/registration.js +39 -0
  92. package/dist/esm/client/history-sync.js +50 -9
  93. package/dist/esm/client/incoming.js +74 -8
  94. package/dist/esm/client/mailbox.js +40 -23
  95. package/dist/esm/client/media.js +234 -0
  96. package/dist/esm/client/messages.js +244 -91
  97. package/dist/esm/client/messaging/fanout.js +22 -12
  98. package/dist/esm/client/messaging/participants.js +6 -4
  99. package/dist/esm/client/persistence/WriteBehindPersistence.js +125 -0
  100. package/dist/esm/client/tokens/cs-token.js +46 -0
  101. package/dist/esm/client/tokens/tc-token.js +18 -0
  102. package/dist/esm/crypto/core/index.js +3 -2
  103. package/dist/esm/crypto/core/keys.js +1 -1
  104. package/dist/esm/crypto/core/nonce.js +2 -0
  105. package/dist/esm/crypto/core/primitives.js +0 -7
  106. package/dist/esm/crypto/core/random.js +23 -7
  107. package/dist/esm/crypto/core/xeddsa.js +53 -0
  108. package/dist/esm/crypto/curves/X25519.js +45 -8
  109. package/dist/esm/crypto/curves/constants.js +1 -0
  110. package/dist/esm/crypto/index.js +1 -0
  111. package/dist/esm/crypto/math/constants.js +12 -35
  112. package/dist/esm/crypto/math/edwards.js +174 -47
  113. package/dist/esm/crypto/math/fe.js +691 -0
  114. package/dist/esm/crypto/math/mod.js +10 -1
  115. package/dist/esm/index.js +1 -1
  116. package/dist/esm/infra/perf/BackgroundQueue.js +478 -0
  117. package/dist/esm/infra/perf/BoundedTaskQueue.js +3 -1
  118. package/dist/esm/infra/perf/PromiseDedup.js +20 -0
  119. package/dist/esm/infra/perf/SharedExclusiveGate.js +109 -0
  120. package/dist/esm/infra/perf/StoreLock.js +80 -0
  121. package/dist/esm/media/WaMediaCrypto.js +332 -55
  122. package/dist/esm/media/WaMediaTransferClient.js +69 -220
  123. package/dist/esm/media/constants.js +4 -1
  124. package/dist/esm/media/processor.js +1 -0
  125. package/dist/esm/message/WaMessageClient.js +26 -19
  126. package/dist/esm/message/addon-crypto.js +130 -3
  127. package/dist/esm/message/content.js +206 -14
  128. package/dist/esm/message/icdc.js +76 -0
  129. package/dist/esm/message/incoming.js +38 -24
  130. package/dist/esm/message/phash.js +35 -13
  131. package/dist/esm/message/reporting-token.js +17 -30
  132. package/dist/esm/message/use-case-secret.js +1 -1
  133. package/dist/esm/protocol/abprops.js +159 -0
  134. package/dist/esm/protocol/appstate.js +9 -40
  135. package/dist/esm/protocol/browser.js +24 -18
  136. package/dist/esm/protocol/constants.js +8 -4
  137. package/dist/esm/protocol/defaults.js +6 -0
  138. package/dist/esm/protocol/email.js +30 -0
  139. package/dist/esm/protocol/index.js +1 -2
  140. package/dist/esm/protocol/jid.js +142 -39
  141. package/dist/esm/protocol/message.js +61 -1
  142. package/dist/esm/protocol/nodes.js +8 -2
  143. package/dist/esm/protocol/notification.js +9 -1
  144. package/dist/esm/protocol/privacy-token.js +17 -0
  145. package/dist/esm/protocol/privacy.js +55 -0
  146. package/dist/esm/protocol/stream.js +26 -1
  147. package/dist/esm/retry/codec.js +216 -0
  148. package/dist/esm/retry/constants.js +1 -1
  149. package/dist/esm/retry/index.js +2 -2
  150. package/dist/esm/retry/parse.js +50 -30
  151. package/dist/esm/retry/reason.js +1 -1
  152. package/dist/esm/retry/replay.js +11 -7
  153. package/dist/esm/retry/tracker.js +50 -12
  154. package/dist/esm/signal/api/SignalDeviceSyncApi.js +52 -32
  155. package/dist/esm/signal/api/SignalDigestSyncApi.js +21 -15
  156. package/dist/esm/signal/api/SignalIdentitySyncApi.js +30 -15
  157. package/dist/esm/signal/api/SignalMissingPreKeysSyncApi.js +19 -8
  158. package/dist/esm/signal/api/SignalRotateKeyApi.js +4 -2
  159. package/dist/esm/signal/api/SignalSessionSyncApi.js +17 -8
  160. package/dist/esm/signal/api/result-map.js +10 -0
  161. package/dist/esm/signal/constants.js +0 -4
  162. package/dist/esm/signal/crypto/WaAdvSignature.js +5 -45
  163. package/dist/esm/signal/crypto/constants.js +0 -4
  164. package/dist/esm/signal/{store/sqlite.js → encoding.js} +40 -29
  165. package/dist/esm/signal/group/SenderKeyChain.js +3 -3
  166. package/dist/esm/signal/group/SenderKeyCodec.js +8 -8
  167. package/dist/esm/signal/group/SenderKeyManager.js +131 -109
  168. package/dist/esm/signal/index.js +1 -0
  169. package/dist/esm/signal/registration/keygen.js +8 -5
  170. package/dist/esm/signal/registration/utils.js +3 -2
  171. package/dist/esm/signal/session/SignalProtocol.js +158 -81
  172. package/dist/esm/signal/session/SignalRatchet.js +21 -10
  173. package/dist/esm/signal/session/SignalSerializer.js +5 -6
  174. package/dist/esm/signal/session/SignalSession.js +11 -9
  175. package/dist/esm/signal/session/resolver.js +140 -105
  176. package/dist/esm/store/contracts/identity.store.js +1 -0
  177. package/dist/esm/store/contracts/message-secret.store.js +1 -0
  178. package/dist/esm/store/contracts/pre-key.store.js +1 -0
  179. package/dist/esm/store/contracts/privacy-token.store.js +1 -0
  180. package/dist/esm/store/contracts/session.store.js +1 -0
  181. package/dist/esm/store/createStore.js +143 -193
  182. package/dist/esm/store/index.js +5 -10
  183. package/dist/esm/store/locks/appstate.lock.js +26 -0
  184. package/dist/esm/store/locks/auth.lock.js +15 -0
  185. package/dist/esm/store/locks/contact.lock.js +20 -0
  186. package/dist/esm/store/locks/device-list.lock.js +20 -0
  187. package/dist/esm/store/locks/identity.lock.js +16 -0
  188. package/dist/esm/store/locks/message-secret.lock.js +17 -0
  189. package/dist/esm/store/locks/message.lock.js +21 -0
  190. package/dist/esm/store/locks/participants.lock.js +20 -0
  191. package/dist/esm/store/locks/pre-key.lock.js +27 -0
  192. package/dist/esm/store/locks/privacy-token.lock.js +18 -0
  193. package/dist/esm/store/locks/retry.lock.js +29 -0
  194. package/dist/esm/store/locks/sender-key.lock.js +52 -0
  195. package/dist/esm/store/locks/session.lock.js +19 -0
  196. package/dist/esm/store/locks/signal.lock.js +39 -0
  197. package/dist/esm/store/locks/thread.lock.js +21 -0
  198. package/dist/esm/store/noop.store.js +21 -1
  199. package/dist/esm/store/providers/memory/appstate.store.js +22 -24
  200. package/dist/esm/store/providers/memory/device-list.store.js +13 -5
  201. package/dist/esm/store/providers/memory/identity.store.js +31 -0
  202. package/dist/esm/store/providers/memory/message-secret.store.js +81 -0
  203. package/dist/esm/store/providers/memory/participants.store.js +3 -0
  204. package/dist/esm/store/providers/memory/pre-key.store.js +97 -0
  205. package/dist/esm/store/providers/memory/privacy-token.store.js +43 -0
  206. package/dist/esm/store/providers/memory/retry.store.js +99 -10
  207. package/dist/esm/store/providers/memory/sender-key.store.js +6 -1
  208. package/dist/esm/store/providers/memory/session.store.js +45 -0
  209. package/dist/esm/store/providers/memory/signal.store.js +1 -147
  210. package/dist/esm/transport/WaComms.js +7 -4
  211. package/dist/esm/transport/WaWebSocket.js +9 -7
  212. package/dist/esm/transport/binary/constants.js +0 -30
  213. package/dist/esm/transport/binary/decoder.js +4 -4
  214. package/dist/esm/transport/binary/encoder.js +8 -15
  215. package/dist/esm/transport/binary/index.js +0 -1
  216. package/dist/esm/transport/index.js +6 -0
  217. package/dist/esm/transport/keepalive/WaKeepAlive.js +17 -8
  218. package/dist/esm/transport/node/WaMobileTcpSocket.js +114 -0
  219. package/dist/esm/transport/node/WaNodeOrchestrator.js +37 -22
  220. package/dist/esm/transport/node/builders/abprops.js +20 -0
  221. package/dist/esm/transport/node/builders/business.js +129 -0
  222. package/dist/esm/transport/node/builders/device.js +11 -0
  223. package/dist/esm/transport/node/builders/email.js +65 -0
  224. package/dist/esm/transport/node/builders/global.js +370 -0
  225. package/dist/esm/transport/node/builders/message.js +63 -239
  226. package/dist/esm/transport/node/builders/offline.js +14 -0
  227. package/dist/esm/transport/node/builders/pairing.js +0 -24
  228. package/dist/esm/transport/node/builders/prekeys.js +37 -40
  229. package/dist/esm/transport/node/builders/presence.js +13 -0
  230. package/dist/esm/transport/node/builders/privacy-token.js +37 -0
  231. package/dist/esm/transport/node/builders/privacy.js +48 -0
  232. package/dist/esm/transport/node/builders/profile.js +70 -0
  233. package/dist/esm/transport/node/builders/retry.js +11 -23
  234. package/dist/esm/transport/node/builders/usync.js +6 -2
  235. package/dist/esm/transport/node/helpers.js +43 -1
  236. package/dist/esm/transport/node/mex/argo-decoder.js +152 -0
  237. package/dist/esm/transport/node/mex/client.js +83 -0
  238. package/dist/esm/transport/node/mex/persist-ids.js +10 -0
  239. package/dist/esm/transport/node/usync.js +3 -33
  240. package/dist/esm/transport/node/xml.js +35 -14
  241. package/dist/esm/transport/noise/WaClientPayload.js +24 -19
  242. package/dist/esm/transport/noise/WaFrameCodec.js +2 -2
  243. package/dist/esm/transport/noise/WaMobileClientPayload.js +53 -0
  244. package/dist/esm/transport/noise/WaNoiseCert.js +9 -27
  245. package/dist/esm/transport/noise/WaNoiseSession.js +76 -34
  246. package/dist/esm/transport/noise/WaNoiseSocket.js +8 -4
  247. package/dist/esm/transport/stream/parse.js +8 -4
  248. package/dist/esm/util/bytes.js +22 -18
  249. package/dist/esm/util/index.js +5 -0
  250. package/dist/esm/util/primitives.js +3 -2
  251. package/dist/index.js +7 -1
  252. package/dist/infra/perf/BackgroundQueue.js +482 -0
  253. package/dist/infra/perf/BoundedTaskQueue.js +3 -1
  254. package/dist/infra/perf/PromiseDedup.js +24 -0
  255. package/dist/infra/perf/SharedExclusiveGate.js +113 -0
  256. package/dist/infra/perf/StoreLock.js +84 -0
  257. package/dist/media/WaMediaCrypto.js +328 -51
  258. package/dist/media/WaMediaTransferClient.js +72 -253
  259. package/dist/media/constants.js +5 -2
  260. package/dist/media/processor.js +2 -0
  261. package/dist/message/WaMessageClient.js +26 -19
  262. package/dist/message/addon-crypto.js +131 -0
  263. package/dist/message/content.js +211 -14
  264. package/dist/message/icdc.js +81 -0
  265. package/dist/message/incoming.js +38 -24
  266. package/dist/message/phash.js +35 -13
  267. package/dist/message/reporting-token.js +16 -30
  268. package/dist/message/use-case-secret.js +1 -1
  269. package/dist/protocol/abprops.js +163 -0
  270. package/dist/protocol/appstate.js +10 -41
  271. package/dist/protocol/browser.js +25 -18
  272. package/dist/protocol/constants.js +33 -2
  273. package/dist/protocol/defaults.js +6 -0
  274. package/dist/protocol/email.js +33 -0
  275. package/dist/protocol/index.js +8 -5
  276. package/dist/protocol/jid.js +149 -39
  277. package/dist/protocol/message.js +62 -2
  278. package/dist/protocol/nodes.js +8 -2
  279. package/dist/protocol/notification.js +10 -2
  280. package/dist/protocol/privacy-token.js +20 -0
  281. package/dist/protocol/privacy.js +58 -0
  282. package/dist/protocol/stream.js +27 -2
  283. package/dist/retry/codec.js +220 -0
  284. package/dist/retry/constants.js +1 -1
  285. package/dist/retry/index.js +5 -5
  286. package/dist/retry/parse.js +51 -30
  287. package/dist/retry/reason.js +1 -1
  288. package/dist/retry/replay.js +10 -6
  289. package/dist/retry/tracker.js +50 -12
  290. package/dist/signal/api/SignalDeviceSyncApi.js +51 -31
  291. package/dist/signal/api/SignalDigestSyncApi.js +21 -15
  292. package/dist/signal/api/SignalIdentitySyncApi.js +29 -14
  293. package/dist/signal/api/SignalMissingPreKeysSyncApi.js +17 -6
  294. package/dist/signal/api/SignalRotateKeyApi.js +4 -2
  295. package/dist/signal/api/SignalSessionSyncApi.js +16 -7
  296. package/dist/signal/api/result-map.js +13 -0
  297. package/dist/signal/constants.js +1 -5
  298. package/dist/signal/crypto/WaAdvSignature.js +2 -44
  299. package/dist/signal/crypto/constants.js +1 -5
  300. package/dist/signal/{store/sqlite.js → encoding.js} +41 -25
  301. package/dist/signal/group/SenderKeyChain.js +2 -2
  302. package/dist/signal/group/SenderKeyCodec.js +8 -8
  303. package/dist/signal/group/SenderKeyManager.js +130 -108
  304. package/dist/signal/index.js +13 -1
  305. package/dist/signal/registration/keygen.js +7 -4
  306. package/dist/signal/registration/utils.js +3 -2
  307. package/dist/signal/session/SignalProtocol.js +158 -81
  308. package/dist/signal/session/SignalRatchet.js +19 -8
  309. package/dist/signal/session/SignalSerializer.js +5 -6
  310. package/dist/signal/session/SignalSession.js +11 -9
  311. package/dist/signal/session/resolver.js +138 -103
  312. package/dist/store/contracts/identity.store.js +2 -0
  313. package/dist/store/contracts/message-secret.store.js +2 -0
  314. package/dist/store/contracts/pre-key.store.js +2 -0
  315. package/dist/store/contracts/privacy-token.store.js +2 -0
  316. package/dist/store/contracts/session.store.js +2 -0
  317. package/dist/store/createStore.js +142 -192
  318. package/dist/store/index.js +23 -33
  319. package/dist/store/locks/appstate.lock.js +29 -0
  320. package/dist/store/locks/auth.lock.js +18 -0
  321. package/dist/store/locks/contact.lock.js +23 -0
  322. package/dist/store/locks/device-list.lock.js +23 -0
  323. package/dist/store/locks/identity.lock.js +19 -0
  324. package/dist/store/locks/message-secret.lock.js +20 -0
  325. package/dist/store/locks/message.lock.js +24 -0
  326. package/dist/store/locks/participants.lock.js +23 -0
  327. package/dist/store/locks/pre-key.lock.js +30 -0
  328. package/dist/store/locks/privacy-token.lock.js +21 -0
  329. package/dist/store/locks/retry.lock.js +32 -0
  330. package/dist/store/locks/sender-key.lock.js +55 -0
  331. package/dist/store/locks/session.lock.js +22 -0
  332. package/dist/store/locks/signal.lock.js +42 -0
  333. package/dist/store/locks/thread.lock.js +24 -0
  334. package/dist/store/noop.store.js +22 -2
  335. package/dist/store/providers/memory/appstate.store.js +22 -24
  336. package/dist/store/providers/memory/device-list.store.js +13 -5
  337. package/dist/store/providers/memory/identity.store.js +35 -0
  338. package/dist/store/providers/memory/message-secret.store.js +85 -0
  339. package/dist/store/providers/memory/participants.store.js +3 -0
  340. package/dist/store/providers/memory/pre-key.store.js +101 -0
  341. package/dist/store/providers/memory/privacy-token.store.js +47 -0
  342. package/dist/store/providers/memory/retry.store.js +98 -9
  343. package/dist/store/providers/memory/sender-key.store.js +6 -1
  344. package/dist/store/providers/memory/session.store.js +49 -0
  345. package/dist/store/providers/memory/signal.store.js +1 -147
  346. package/dist/transport/WaComms.js +7 -4
  347. package/dist/transport/WaWebSocket.js +9 -7
  348. package/dist/transport/binary/constants.js +1 -31
  349. package/dist/transport/binary/decoder.js +4 -4
  350. package/dist/transport/binary/encoder.js +8 -15
  351. package/dist/transport/binary/index.js +0 -4
  352. package/dist/transport/index.js +17 -1
  353. package/dist/transport/keepalive/WaKeepAlive.js +17 -8
  354. package/dist/transport/node/WaMobileTcpSocket.js +118 -0
  355. package/dist/transport/node/WaNodeOrchestrator.js +36 -21
  356. package/dist/transport/node/builders/abprops.js +23 -0
  357. package/dist/transport/node/builders/business.js +137 -0
  358. package/dist/transport/node/builders/device.js +14 -0
  359. package/dist/transport/node/builders/email.js +72 -0
  360. package/dist/transport/node/builders/global.js +375 -0
  361. package/dist/transport/node/builders/message.js +64 -245
  362. package/dist/transport/node/builders/offline.js +17 -0
  363. package/dist/transport/node/builders/pairing.js +0 -26
  364. package/dist/transport/node/builders/prekeys.js +36 -39
  365. package/dist/transport/node/builders/presence.js +16 -0
  366. package/dist/transport/node/builders/privacy-token.js +42 -0
  367. package/dist/transport/node/builders/privacy.js +55 -0
  368. package/dist/transport/node/builders/profile.js +78 -0
  369. package/dist/transport/node/builders/retry.js +10 -22
  370. package/dist/transport/node/builders/usync.js +6 -2
  371. package/dist/transport/node/helpers.js +46 -1
  372. package/dist/transport/node/mex/argo-decoder.js +189 -0
  373. package/dist/transport/node/mex/client.js +86 -0
  374. package/dist/transport/node/mex/persist-ids.js +13 -0
  375. package/dist/transport/node/usync.js +2 -32
  376. package/dist/transport/node/xml.js +35 -14
  377. package/dist/transport/noise/WaClientPayload.js +26 -21
  378. package/dist/transport/noise/WaFrameCodec.js +1 -1
  379. package/dist/transport/noise/WaMobileClientPayload.js +56 -0
  380. package/dist/transport/noise/WaNoiseCert.js +8 -26
  381. package/dist/transport/noise/WaNoiseSession.js +75 -33
  382. package/dist/transport/noise/WaNoiseSocket.js +8 -4
  383. package/dist/transport/stream/parse.js +7 -3
  384. package/dist/types/appstate/WaAppStateCrypto.d.ts +11 -8
  385. package/dist/types/appstate/WaAppStateSyncClient.d.ts +6 -2
  386. package/dist/types/appstate/encoding.d.ts +7 -0
  387. package/dist/types/appstate/index.d.ts +3 -3
  388. package/dist/types/appstate/{WaAppStateSyncResponseParser.d.ts → response-parser.d.ts} +1 -1
  389. package/dist/types/appstate/types.d.ts +1 -1
  390. package/dist/types/appstate/utils.d.ts +0 -2
  391. package/dist/types/auth/WaAuthClient.d.ts +9 -3
  392. package/dist/types/auth/credentials-flow.d.ts +20 -0
  393. package/dist/types/auth/index.d.ts +0 -2
  394. package/dist/types/auth/pairing/WaPairingFlow.d.ts +3 -2
  395. package/dist/types/auth/pairing/{WaPairingCodeCrypto.d.ts → pairing-code-crypto.d.ts} +6 -1
  396. package/dist/types/auth/types.d.ts +41 -0
  397. package/dist/types/client/WaClient.d.ts +44 -18
  398. package/dist/types/client/WaClientFactory.d.ts +22 -8
  399. package/dist/types/client/connection/WaConnectionManager.d.ts +2 -0
  400. package/dist/types/client/coordinators/WaAbPropsCoordinator.d.ts +26 -0
  401. package/dist/types/client/coordinators/WaBusinessCoordinator.d.ts +57 -0
  402. package/dist/types/client/coordinators/WaEmailCoordinator.d.ts +24 -0
  403. package/dist/types/client/coordinators/WaIncomingNodeCoordinator.d.ts +9 -2
  404. package/dist/types/client/coordinators/WaMessageDispatchCoordinator.d.ts +29 -2
  405. package/dist/types/client/coordinators/WaOfflineResumeCoordinator.d.ts +31 -0
  406. package/dist/types/client/coordinators/WaPassiveTasksCoordinator.d.ts +16 -1
  407. package/dist/types/client/coordinators/WaPrivacyCoordinator.d.ts +26 -0
  408. package/dist/types/client/coordinators/WaProfileCoordinator.d.ts +38 -0
  409. package/dist/types/client/coordinators/WaRetryCoordinator.d.ts +12 -0
  410. package/dist/types/client/coordinators/WaStreamControlCoordinator.d.ts +3 -2
  411. package/dist/types/client/coordinators/WaTrustedContactTokenCoordinator.d.ts +55 -0
  412. package/dist/types/client/dirty.d.ts +3 -1
  413. package/dist/types/client/events/abprops.d.ts +14 -0
  414. package/dist/types/client/events/devices.d.ts +20 -0
  415. package/dist/types/client/events/identity.d.ts +9 -0
  416. package/dist/types/client/events/privacy-token.d.ts +7 -0
  417. package/dist/types/client/events/registration.d.ts +17 -0
  418. package/dist/types/client/history-sync.d.ts +9 -6
  419. package/dist/types/client/incoming.d.ts +9 -2
  420. package/dist/types/client/index.d.ts +1 -1
  421. package/dist/types/client/mailbox.d.ts +5 -5
  422. package/dist/types/client/media.d.ts +31 -0
  423. package/dist/types/client/messages.d.ts +3 -2
  424. package/dist/types/client/persistence/WriteBehindPersistence.d.ts +34 -0
  425. package/dist/types/client/tokens/cs-token.d.ts +10 -0
  426. package/dist/types/client/tokens/tc-token.d.ts +5 -0
  427. package/dist/types/client/types.d.ts +151 -4
  428. package/dist/types/crypto/core/index.d.ts +3 -2
  429. package/dist/types/crypto/core/nonce.d.ts +2 -0
  430. package/dist/types/crypto/core/primitives.d.ts +1 -2
  431. package/dist/types/crypto/core/random.d.ts +2 -1
  432. package/dist/types/crypto/core/xeddsa.d.ts +2 -0
  433. package/dist/types/crypto/curves/constants.d.ts +1 -0
  434. package/dist/types/crypto/index.d.ts +2 -0
  435. package/dist/types/crypto/math/constants.d.ts +4 -2
  436. package/dist/types/crypto/math/fe.d.ts +30 -0
  437. package/dist/types/crypto/math/mod.d.ts +0 -2
  438. package/dist/types/crypto/math/types.d.ts +11 -4
  439. package/dist/types/index.d.ts +6 -3
  440. package/dist/types/infra/log/ConsoleLogger.d.ts +1 -1
  441. package/dist/types/infra/log/PinoLogger.d.ts +1 -1
  442. package/dist/types/infra/perf/BackgroundQueue.d.ts +58 -0
  443. package/dist/types/infra/perf/PromiseDedup.d.ts +4 -0
  444. package/dist/types/infra/perf/SharedExclusiveGate.d.ts +17 -0
  445. package/dist/types/infra/perf/StoreLock.d.ts +11 -0
  446. package/dist/types/media/WaMediaCrypto.d.ts +16 -6
  447. package/dist/types/media/WaMediaTransferClient.d.ts +6 -23
  448. package/dist/types/media/constants.d.ts +3 -1
  449. package/dist/types/media/index.d.ts +2 -1
  450. package/dist/types/media/processor.d.ts +28 -0
  451. package/dist/types/media/types.d.ts +19 -5
  452. package/dist/types/message/addon-crypto.d.ts +34 -3
  453. package/dist/types/message/content.d.ts +11 -1
  454. package/dist/types/message/icdc.d.ts +13 -0
  455. package/dist/types/message/reporting-token.d.ts +0 -1
  456. package/dist/types/message/types.d.ts +42 -11
  457. package/dist/types/protocol/abprops.d.ts +142 -0
  458. package/dist/types/protocol/appstate.d.ts +0 -11
  459. package/dist/types/protocol/browser.d.ts +1 -0
  460. package/dist/types/protocol/constants.d.ts +12 -4
  461. package/dist/types/protocol/defaults.d.ts +6 -0
  462. package/dist/types/protocol/email.d.ts +32 -0
  463. package/dist/types/protocol/index.d.ts +1 -2
  464. package/dist/types/protocol/jid.d.ts +20 -2
  465. package/dist/types/protocol/message.d.ts +60 -0
  466. package/dist/types/protocol/nodes.d.ts +6 -0
  467. package/dist/types/protocol/notification.d.ts +8 -0
  468. package/dist/types/protocol/privacy-token.d.ts +17 -0
  469. package/dist/types/protocol/privacy.d.ts +75 -0
  470. package/dist/types/protocol/stream.d.ts +31 -0
  471. package/dist/types/retry/codec.d.ts +3 -0
  472. package/dist/types/retry/index.d.ts +3 -3
  473. package/dist/types/retry/parse.d.ts +5 -2
  474. package/dist/types/retry/reason.d.ts +1 -1
  475. package/dist/types/retry/tracker.d.ts +1 -0
  476. package/dist/types/retry/types.d.ts +6 -1
  477. package/dist/types/signal/api/SignalDeviceSyncApi.d.ts +2 -1
  478. package/dist/types/signal/api/SignalDigestSyncApi.d.ts +9 -0
  479. package/dist/types/signal/api/SignalIdentitySyncApi.d.ts +5 -3
  480. package/dist/types/signal/api/SignalRotateKeyApi.d.ts +4 -5
  481. package/dist/types/signal/api/SignalSessionSyncApi.d.ts +8 -6
  482. package/dist/types/signal/api/result-map.d.ts +1 -0
  483. package/dist/types/signal/constants.d.ts +0 -3
  484. package/dist/types/signal/crypto/WaAdvSignature.d.ts +0 -2
  485. package/dist/types/signal/crypto/constants.d.ts +0 -1
  486. package/dist/types/signal/{store/sqlite.d.ts → encoding.d.ts} +9 -3
  487. package/dist/types/signal/group/SenderKeyChain.d.ts +1 -1
  488. package/dist/types/signal/group/SenderKeyManager.d.ts +17 -7
  489. package/dist/types/signal/index.d.ts +2 -0
  490. package/dist/types/signal/registration/utils.d.ts +2 -1
  491. package/dist/types/signal/session/SignalProtocol.d.ts +21 -6
  492. package/dist/types/signal/session/SignalSerializer.d.ts +2 -1
  493. package/dist/types/signal/session/resolver.d.ts +11 -4
  494. package/dist/types/signal/types.d.ts +16 -4
  495. package/dist/types/store/contracts/appstate.store.d.ts +1 -1
  496. package/dist/types/store/contracts/identity.store.d.ts +11 -0
  497. package/dist/types/store/contracts/message-secret.store.d.ts +16 -0
  498. package/dist/types/store/contracts/pre-key.store.d.ts +13 -0
  499. package/dist/types/store/contracts/privacy-token.store.d.ts +16 -0
  500. package/dist/types/store/contracts/retry.store.d.ts +7 -0
  501. package/dist/types/store/contracts/session.store.d.ts +14 -0
  502. package/dist/types/store/contracts/signal.store.d.ts +1 -27
  503. package/dist/types/store/createStore.d.ts +1 -1
  504. package/dist/types/store/index.d.ts +12 -12
  505. package/dist/types/store/locks/appstate.lock.d.ts +3 -0
  506. package/dist/types/store/locks/auth.lock.d.ts +3 -0
  507. package/dist/types/store/locks/contact.lock.d.ts +3 -0
  508. package/dist/types/store/locks/device-list.lock.d.ts +2 -0
  509. package/dist/types/store/locks/identity.lock.d.ts +3 -0
  510. package/dist/types/store/locks/message-secret.lock.d.ts +3 -0
  511. package/dist/types/store/locks/message.lock.d.ts +3 -0
  512. package/dist/types/store/locks/participants.lock.d.ts +2 -0
  513. package/dist/types/store/locks/pre-key.lock.d.ts +3 -0
  514. package/dist/types/store/locks/privacy-token.lock.d.ts +2 -0
  515. package/dist/types/store/locks/retry.lock.d.ts +2 -0
  516. package/dist/types/store/locks/sender-key.lock.d.ts +3 -0
  517. package/dist/types/store/locks/session.lock.d.ts +3 -0
  518. package/dist/types/store/locks/signal.lock.d.ts +3 -0
  519. package/dist/types/store/locks/thread.lock.d.ts +3 -0
  520. package/dist/types/store/noop.store.d.ts +4 -0
  521. package/dist/types/store/providers/memory/appstate.store.d.ts +1 -1
  522. package/dist/types/store/providers/memory/identity.store.d.ts +18 -0
  523. package/dist/types/store/providers/memory/message-secret.store.d.ts +21 -0
  524. package/dist/types/store/providers/memory/pre-key.store.d.ts +23 -0
  525. package/dist/types/store/providers/memory/privacy-token.store.d.ts +13 -0
  526. package/dist/types/store/providers/memory/retry.store.d.ts +15 -1
  527. package/dist/types/store/providers/memory/session.store.d.ts +21 -0
  528. package/dist/types/store/providers/memory/signal.store.d.ts +2 -43
  529. package/dist/types/store/providers/memory/thread.store.d.ts +1 -1
  530. package/dist/types/store/types.d.ts +69 -61
  531. package/dist/types/transport/WaWebSocket.d.ts +1 -1
  532. package/dist/types/transport/binary/constants.d.ts +0 -30
  533. package/dist/types/transport/binary/index.d.ts +0 -1
  534. package/dist/types/transport/index.d.ts +8 -1
  535. package/dist/types/transport/keepalive/WaKeepAlive.d.ts +4 -1
  536. package/dist/types/transport/node/WaMobileTcpSocket.d.ts +18 -0
  537. package/dist/types/transport/node/WaNodeOrchestrator.d.ts +9 -6
  538. package/dist/types/transport/node/builders/abprops.d.ts +5 -0
  539. package/dist/types/transport/node/builders/business.d.ts +29 -0
  540. package/dist/types/transport/node/builders/device.d.ts +2 -0
  541. package/dist/types/transport/node/builders/email.d.ts +11 -0
  542. package/dist/types/transport/node/builders/global.d.ts +102 -0
  543. package/dist/types/transport/node/builders/message.d.ts +8 -7
  544. package/dist/types/transport/node/builders/offline.d.ts +2 -0
  545. package/dist/types/transport/node/builders/pairing.d.ts +0 -2
  546. package/dist/types/transport/node/builders/prekeys.d.ts +4 -3
  547. package/dist/types/transport/node/builders/presence.d.ts +6 -0
  548. package/dist/types/transport/node/builders/privacy-token.d.ts +9 -0
  549. package/dist/types/transport/node/builders/privacy.d.ts +7 -0
  550. package/dist/types/transport/node/builders/profile.d.ts +8 -0
  551. package/dist/types/transport/node/builders/retry.d.ts +0 -1
  552. package/dist/types/transport/node/helpers.d.ts +8 -0
  553. package/dist/types/transport/node/mex/argo-decoder.d.ts +11 -0
  554. package/dist/types/transport/node/mex/client.d.ts +18 -0
  555. package/dist/types/transport/node/mex/persist-ids.d.ts +14 -0
  556. package/dist/types/transport/noise/WaMobileClientPayload.d.ts +29 -0
  557. package/dist/types/transport/noise/WaNoiseCert.d.ts +7 -1
  558. package/dist/types/transport/noise/WaNoiseSession.d.ts +4 -2
  559. package/dist/types/transport/noise/WaNoiseSocket.d.ts +4 -2
  560. package/dist/types/transport/types.d.ts +8 -0
  561. package/dist/types/util/bytes.d.ts +1 -1
  562. package/dist/types/util/index.d.ts +5 -0
  563. package/dist/types/util/primitives.d.ts +0 -1
  564. package/dist/util/bytes.js +22 -18
  565. package/dist/util/index.js +23 -0
  566. package/dist/util/primitives.js +2 -2
  567. package/package.json +34 -10
  568. package/proto/index.js +1 -1
  569. package/dist/auth/flow/WaAuthCredentialsFlow.js +0 -130
  570. package/dist/auth/pairing/constants.js +0 -5
  571. package/dist/client/connection/WaKeyShareCoordinator.js +0 -63
  572. package/dist/crypto/core/constants.js +0 -4
  573. package/dist/esm/auth/flow/WaAuthCredentialsFlow.js +0 -125
  574. package/dist/esm/auth/pairing/constants.js +0 -2
  575. package/dist/esm/client/connection/WaKeyShareCoordinator.js +0 -59
  576. package/dist/esm/crypto/core/constants.js +0 -1
  577. package/dist/esm/retry/outbound.js +0 -82
  578. package/dist/esm/store/providers/sqlite/BaseSqliteStore.js +0 -37
  579. package/dist/esm/store/providers/sqlite/appstate.store.js +0 -250
  580. package/dist/esm/store/providers/sqlite/auth.store.js +0 -176
  581. package/dist/esm/store/providers/sqlite/connection.js +0 -245
  582. package/dist/esm/store/providers/sqlite/contact.store.js +0 -74
  583. package/dist/esm/store/providers/sqlite/device-list.store.js +0 -127
  584. package/dist/esm/store/providers/sqlite/message.store.js +0 -132
  585. package/dist/esm/store/providers/sqlite/migrations.js +0 -347
  586. package/dist/esm/store/providers/sqlite/participants.store.js +0 -77
  587. package/dist/esm/store/providers/sqlite/retry.store.js +0 -141
  588. package/dist/esm/store/providers/sqlite/sender-key.store.js +0 -198
  589. package/dist/esm/store/providers/sqlite/signal.store.js +0 -435
  590. package/dist/esm/store/providers/sqlite/table-names.js +0 -107
  591. package/dist/esm/store/providers/sqlite/thread.store.js +0 -85
  592. package/dist/esm/transport/node/builders/index.js +0 -8
  593. package/dist/retry/outbound.js +0 -87
  594. package/dist/store/providers/sqlite/BaseSqliteStore.js +0 -41
  595. package/dist/store/providers/sqlite/appstate.store.js +0 -254
  596. package/dist/store/providers/sqlite/auth.store.js +0 -180
  597. package/dist/store/providers/sqlite/connection.js +0 -281
  598. package/dist/store/providers/sqlite/contact.store.js +0 -78
  599. package/dist/store/providers/sqlite/device-list.store.js +0 -131
  600. package/dist/store/providers/sqlite/message.store.js +0 -136
  601. package/dist/store/providers/sqlite/migrations.js +0 -350
  602. package/dist/store/providers/sqlite/participants.store.js +0 -81
  603. package/dist/store/providers/sqlite/retry.store.js +0 -145
  604. package/dist/store/providers/sqlite/sender-key.store.js +0 -202
  605. package/dist/store/providers/sqlite/signal.store.js +0 -439
  606. package/dist/store/providers/sqlite/table-names.js +0 -113
  607. package/dist/store/providers/sqlite/thread.store.js +0 -89
  608. package/dist/transport/node/builders/index.js +0 -42
  609. package/dist/types/appstate/store/sqlite.d.ts +0 -7
  610. package/dist/types/auth/flow/WaAuthCredentialsFlow.d.ts +0 -14
  611. package/dist/types/auth/pairing/constants.d.ts +0 -2
  612. package/dist/types/client/connection/WaKeyShareCoordinator.d.ts +0 -14
  613. package/dist/types/crypto/core/constants.d.ts +0 -1
  614. package/dist/types/retry/outbound.d.ts +0 -4
  615. package/dist/types/store/providers/sqlite/BaseSqliteStore.d.ts +0 -12
  616. package/dist/types/store/providers/sqlite/appstate.store.d.ts +0 -17
  617. package/dist/types/store/providers/sqlite/auth.store.d.ts +0 -10
  618. package/dist/types/store/providers/sqlite/connection.d.ts +0 -10
  619. package/dist/types/store/providers/sqlite/contact.store.d.ts +0 -12
  620. package/dist/types/store/providers/sqlite/device-list.store.d.ts +0 -15
  621. package/dist/types/store/providers/sqlite/message.store.d.ts +0 -13
  622. package/dist/types/store/providers/sqlite/migrations.d.ts +0 -3
  623. package/dist/types/store/providers/sqlite/participants.store.d.ts +0 -12
  624. package/dist/types/store/providers/sqlite/retry.store.d.ts +0 -15
  625. package/dist/types/store/providers/sqlite/sender-key.store.d.ts +0 -24
  626. package/dist/types/store/providers/sqlite/signal.store.d.ts +0 -53
  627. package/dist/types/store/providers/sqlite/table-names.d.ts +0 -5
  628. package/dist/types/store/providers/sqlite/thread.store.d.ts +0 -13
  629. package/dist/types/transport/node/builders/index.d.ts +0 -8
  630. /package/dist/appstate/{WaAppStateSyncResponseParser.js → response-parser.js} +0 -0
@@ -0,0 +1,27 @@
1
+ import { SharedExclusiveGate } from '../../infra/perf/SharedExclusiveGate.js';
2
+ import { StoreLock } from '../../infra/perf/StoreLock.js';
3
+ const WA_PREKEY_KEY = 'prekey:prekeys';
4
+ const WA_PREKEY_SERVER_KEY = 'prekey:serverHasPreKeys';
5
+ const WA_PREKEY_CLEAR_KEY = 'prekey:clear';
6
+ export function withPreKeyLock(store) {
7
+ const lock = new StoreLock();
8
+ const gate = new SharedExclusiveGate();
9
+ const destroyStore = store;
10
+ return {
11
+ putPreKey: (record) => gate.runShared(() => lock.run(WA_PREKEY_KEY, () => store.putPreKey(record))),
12
+ getOrGenPreKeys: (count, generator) => gate.runShared(() => lock.run(WA_PREKEY_KEY, () => store.getOrGenPreKeys(count, generator))),
13
+ getPreKeyById: (keyId) => gate.runShared(() => store.getPreKeyById(keyId)),
14
+ getPreKeysById: (keyIds) => gate.runShared(() => store.getPreKeysById(keyIds)),
15
+ consumePreKeyById: (keyId) => gate.runShared(() => lock.run(WA_PREKEY_KEY, () => store.consumePreKeyById(keyId))),
16
+ getOrGenSinglePreKey: (generator) => gate.runShared(() => lock.run(WA_PREKEY_KEY, () => store.getOrGenSinglePreKey(generator))),
17
+ markKeyAsUploaded: (keyId) => gate.runShared(() => lock.run(WA_PREKEY_KEY, () => store.markKeyAsUploaded(keyId))),
18
+ setServerHasPreKeys: (value) => gate.runShared(() => lock.run(WA_PREKEY_SERVER_KEY, () => store.setServerHasPreKeys(value))),
19
+ getServerHasPreKeys: () => gate.runShared(() => store.getServerHasPreKeys()),
20
+ clear: () => gate.runExclusive(() => lock.run(WA_PREKEY_CLEAR_KEY, () => store.clear())),
21
+ destroy: async () => {
22
+ await gate.close();
23
+ await lock.shutdown();
24
+ await destroyStore.destroy?.();
25
+ }
26
+ };
27
+ }
@@ -0,0 +1,18 @@
1
+ import { SharedExclusiveGate } from '../../infra/perf/SharedExclusiveGate.js';
2
+ import { StoreLock } from '../../infra/perf/StoreLock.js';
3
+ export function withPrivacyTokenLock(store) {
4
+ const lock = new StoreLock();
5
+ const gate = new SharedExclusiveGate();
6
+ return {
7
+ destroy: async () => {
8
+ await gate.close();
9
+ await lock.shutdown();
10
+ await store.destroy?.();
11
+ },
12
+ upsert: (record) => gate.runShared(() => lock.run(`pt:${record.jid}`, () => store.upsert(record))),
13
+ upsertBatch: (records) => gate.runExclusive(() => store.upsertBatch(records)),
14
+ getByJid: (jid) => gate.runShared(() => store.getByJid(jid)),
15
+ deleteByJid: (jid) => gate.runShared(() => lock.run(`pt:${jid}`, () => store.deleteByJid(jid))),
16
+ clear: () => gate.runExclusive(() => store.clear())
17
+ };
18
+ }
@@ -0,0 +1,29 @@
1
+ import { SharedExclusiveGate } from '../../infra/perf/SharedExclusiveGate.js';
2
+ import { StoreLock } from '../../infra/perf/StoreLock.js';
3
+ const WA_RETRY_CLEANUP_KEY = 'retry:cleanup';
4
+ const WA_RETRY_CLEAR_KEY = 'retry:clear';
5
+ export function withRetryLock(store) {
6
+ const lock = new StoreLock();
7
+ const gate = new SharedExclusiveGate();
8
+ return {
9
+ getTtlMs: store.getTtlMs?.bind(store),
10
+ supportsRawReplayPayload: store.supportsRawReplayPayload?.bind(store),
11
+ destroy: async () => {
12
+ await gate.close();
13
+ await lock.shutdown();
14
+ await store.destroy?.();
15
+ },
16
+ getOutboundRequesterStatus: (messageId, requesterDeviceJid) => gate.runShared(() => store.getOutboundRequesterStatus(messageId, requesterDeviceJid)),
17
+ upsertOutboundMessage: (record) => gate.runShared(() => lock.run(`retry:outbound:${record.messageId}`, () => store.upsertOutboundMessage(record))),
18
+ deleteOutboundMessage: (messageId) => gate.runShared(() => lock.run(`retry:outbound:${messageId}`, () => store.deleteOutboundMessage(messageId))),
19
+ getOutboundMessage: (messageId) => gate.runShared(() => store.getOutboundMessage(messageId)),
20
+ updateOutboundMessageState: (messageId, state, updatedAtMs, expiresAtMs) => gate.runShared(() => lock.run(`retry:outbound:${messageId}`, () => store.updateOutboundMessageState(messageId, state, updatedAtMs, expiresAtMs))),
21
+ markOutboundRequesterDelivered: (messageId, requesterDeviceJid, updatedAtMs, expiresAtMs) => gate.runShared(() => lock.runMany([
22
+ `retry:outbound:${messageId}`,
23
+ `retry:outbound:${messageId}:${requesterDeviceJid}`
24
+ ], () => store.markOutboundRequesterDelivered(messageId, requesterDeviceJid, updatedAtMs, expiresAtMs))),
25
+ incrementInboundCounter: (messageId, requesterJid, updatedAtMs, expiresAtMs) => gate.runShared(() => lock.run(`retry:inbound:${messageId}:${requesterJid}`, () => store.incrementInboundCounter(messageId, requesterJid, updatedAtMs, expiresAtMs))),
26
+ cleanupExpired: (nowMs) => gate.runExclusive(() => lock.run(WA_RETRY_CLEANUP_KEY, () => store.cleanupExpired(nowMs))),
27
+ clear: () => gate.runExclusive(() => lock.run(WA_RETRY_CLEAR_KEY, () => store.clear()))
28
+ };
29
+ }
@@ -0,0 +1,52 @@
1
+ import { SharedExclusiveGate } from '../../infra/perf/SharedExclusiveGate.js';
2
+ import { StoreLock } from '../../infra/perf/StoreLock.js';
3
+ import { signalAddressKey } from '../../protocol/jid.js';
4
+ const WA_SENDER_KEY_CLEAR_KEY = 'senderKey:clear';
5
+ function senderKeyLockKey(groupId, sender) {
6
+ return `senderKey:${groupId}:${signalAddressKey(sender)}`;
7
+ }
8
+ function senderDistributionLockKey(groupId, sender) {
9
+ return `senderKeyDistribution:${groupId}:${signalAddressKey(sender)}`;
10
+ }
11
+ function senderAnyLockKey(sender) {
12
+ return `senderKey:any:${signalAddressKey(sender)}`;
13
+ }
14
+ export function withSenderKeyLock(store) {
15
+ const lock = new StoreLock();
16
+ const gate = new SharedExclusiveGate();
17
+ const destroyStore = store;
18
+ return {
19
+ upsertSenderKey: (record) => gate.runShared(() => lock.runMany([
20
+ senderAnyLockKey(record.sender),
21
+ senderKeyLockKey(record.groupId, record.sender)
22
+ ], () => store.upsertSenderKey(record))),
23
+ upsertSenderKeyDistribution: (record) => gate.runShared(() => lock.runMany([
24
+ senderAnyLockKey(record.sender),
25
+ senderDistributionLockKey(record.groupId, record.sender)
26
+ ], () => store.upsertSenderKeyDistribution(record))),
27
+ upsertSenderKeyDistributions: (records) => gate.runShared(() => lock.runMany(records.flatMap((record) => [
28
+ senderAnyLockKey(record.sender),
29
+ senderDistributionLockKey(record.groupId, record.sender)
30
+ ]), () => store.upsertSenderKeyDistributions(records))),
31
+ getGroupSenderKeyList: (groupId) => gate.runShared(() => store.getGroupSenderKeyList(groupId)),
32
+ getDeviceSenderKey: (groupId, sender) => gate.runShared(() => store.getDeviceSenderKey(groupId, sender)),
33
+ getDeviceSenderKeyDistributions: (groupId, senders) => gate.runShared(() => store.getDeviceSenderKeyDistributions(groupId, senders)),
34
+ deleteDeviceSenderKey: (target, groupId) => gate.runShared(() => lock.runMany(groupId
35
+ ? [
36
+ senderAnyLockKey(target),
37
+ senderKeyLockKey(groupId, target),
38
+ senderDistributionLockKey(groupId, target)
39
+ ]
40
+ : [senderAnyLockKey(target)], () => store.deleteDeviceSenderKey(target, groupId))),
41
+ markForgetSenderKey: (groupId, participants) => gate.runShared(() => lock.runMany(participants.flatMap((participant) => [
42
+ senderAnyLockKey(participant),
43
+ senderKeyLockKey(groupId, participant)
44
+ ]), () => store.markForgetSenderKey(groupId, participants))),
45
+ clear: () => gate.runExclusive(() => lock.run(WA_SENDER_KEY_CLEAR_KEY, () => store.clear())),
46
+ destroy: async () => {
47
+ await gate.close();
48
+ await lock.shutdown();
49
+ await destroyStore.destroy?.();
50
+ }
51
+ };
52
+ }
@@ -0,0 +1,19 @@
1
+ import { SharedExclusiveGate } from '../../infra/perf/SharedExclusiveGate.js';
2
+ export function withSessionLock(store) {
3
+ const gate = new SharedExclusiveGate();
4
+ const destroyStore = store;
5
+ return {
6
+ hasSession: (address) => gate.runShared(() => store.hasSession(address)),
7
+ hasSessions: (addresses) => gate.runShared(() => store.hasSessions(addresses)),
8
+ getSession: (address) => gate.runShared(() => store.getSession(address)),
9
+ getSessionsBatch: (addresses) => gate.runShared(() => store.getSessionsBatch(addresses)),
10
+ setSession: (address, session) => gate.runShared(() => store.setSession(address, session)),
11
+ setSessionsBatch: (entries) => gate.runShared(() => store.setSessionsBatch(entries)),
12
+ deleteSession: (address) => gate.runShared(() => store.deleteSession(address)),
13
+ clear: () => gate.runExclusive(() => store.clear()),
14
+ destroy: async () => {
15
+ await gate.close();
16
+ await destroyStore.destroy?.();
17
+ }
18
+ };
19
+ }
@@ -0,0 +1,39 @@
1
+ import { SharedExclusiveGate } from '../../infra/perf/SharedExclusiveGate.js';
2
+ import { StoreLock } from '../../infra/perf/StoreLock.js';
3
+ const WA_SIGNAL_REGISTRATION_KEY = 'signal:registration';
4
+ const WA_SIGNAL_SIGNED_PREKEY_KEY = 'signal:signedPreKey';
5
+ const WA_SIGNAL_CLEAR_KEY = 'signal:clear';
6
+ export function withSignalLock(store) {
7
+ const lock = new StoreLock();
8
+ const gate = new SharedExclusiveGate();
9
+ const destroyStore = store;
10
+ let cachedRegInfo;
11
+ return {
12
+ getRegistrationInfo: () => gate.runShared(async () => {
13
+ if (cachedRegInfo !== undefined) {
14
+ return cachedRegInfo;
15
+ }
16
+ const info = await store.getRegistrationInfo();
17
+ cachedRegInfo = info;
18
+ return info;
19
+ }),
20
+ setRegistrationInfo: (info) => gate.runShared(() => lock.run(WA_SIGNAL_REGISTRATION_KEY, async () => {
21
+ await store.setRegistrationInfo(info);
22
+ cachedRegInfo = info;
23
+ })),
24
+ getSignedPreKey: () => gate.runShared(() => store.getSignedPreKey()),
25
+ setSignedPreKey: (record) => gate.runShared(() => lock.run(WA_SIGNAL_SIGNED_PREKEY_KEY, () => store.setSignedPreKey(record))),
26
+ getSignedPreKeyById: (keyId) => gate.runShared(() => store.getSignedPreKeyById(keyId)),
27
+ setSignedPreKeyRotationTs: (value) => gate.runShared(() => lock.run(WA_SIGNAL_SIGNED_PREKEY_KEY, () => store.setSignedPreKeyRotationTs(value))),
28
+ getSignedPreKeyRotationTs: () => gate.runShared(() => store.getSignedPreKeyRotationTs()),
29
+ clear: () => gate.runExclusive(() => lock.run(WA_SIGNAL_CLEAR_KEY, async () => {
30
+ await store.clear();
31
+ cachedRegInfo = undefined;
32
+ })),
33
+ destroy: async () => {
34
+ await gate.close();
35
+ await lock.shutdown();
36
+ await destroyStore.destroy?.();
37
+ }
38
+ };
39
+ }
@@ -0,0 +1,21 @@
1
+ import { SharedExclusiveGate } from '../../infra/perf/SharedExclusiveGate.js';
2
+ import { StoreLock } from '../../infra/perf/StoreLock.js';
3
+ const WA_THREAD_CLEAR_KEY = 'thread:clear';
4
+ export function withThreadLock(store) {
5
+ const lock = new StoreLock();
6
+ const gate = new SharedExclusiveGate();
7
+ const destroyStore = store;
8
+ return {
9
+ upsert: (record) => gate.runShared(() => lock.run(`thread:${record.jid}`, () => store.upsert(record))),
10
+ upsertBatch: (records) => gate.runShared(() => lock.runMany(records.map((record) => `thread:${record.jid}`), () => store.upsertBatch(records))),
11
+ getByJid: (jid) => gate.runShared(() => store.getByJid(jid)),
12
+ list: (limit) => gate.runShared(() => store.list(limit)),
13
+ deleteByJid: (jid) => gate.runShared(() => lock.run(`thread:${jid}`, () => store.deleteByJid(jid))),
14
+ clear: () => gate.runExclusive(() => lock.run(WA_THREAD_CLEAR_KEY, () => store.clear())),
15
+ destroy: async () => {
16
+ await gate.close();
17
+ await lock.shutdown();
18
+ await destroyStore.destroy?.();
19
+ }
20
+ };
21
+ }
@@ -1,4 +1,12 @@
1
1
  const EMPTY_STORE_LIST = Object.freeze([]);
2
+ export const NOOP_MESSAGE_SECRET_STORE = Object.freeze({
3
+ get: async (_messageId) => null,
4
+ getBatch: async (messageIds) => new Array(messageIds.length).fill(null),
5
+ set: async (_messageId, _entry) => { },
6
+ setBatch: async (_entries) => { },
7
+ cleanupExpired: async (_nowMs) => 0,
8
+ clear: async () => { }
9
+ });
2
10
  export const NOOP_MESSAGE_STORE = Object.freeze({
3
11
  upsert: async (_record) => { },
4
12
  upsertBatch: async (_records) => { },
@@ -22,6 +30,18 @@ export const NOOP_CONTACT_STORE = Object.freeze({
22
30
  deleteByJid: async (_jid) => 0,
23
31
  clear: async () => { }
24
32
  });
33
+ export const NOOP_RETRY_STORE = Object.freeze({
34
+ getOutboundRequesterStatus: async () => null,
35
+ upsertOutboundMessage: async (_record) => { },
36
+ deleteOutboundMessage: async (_messageId) => 0,
37
+ getOutboundMessage: async (_messageId) => null,
38
+ updateOutboundMessageState: async () => { },
39
+ markOutboundRequesterDelivered: async () => { },
40
+ incrementInboundCounter: async () => 0,
41
+ cleanupExpired: async (_nowMs) => 0,
42
+ clear: async () => { },
43
+ destroy: async () => { }
44
+ });
25
45
  export const NOOP_PARTICIPANTS_STORE = Object.freeze({
26
46
  upsertGroupParticipants: async (_snapshot) => { },
27
47
  getGroupParticipants: async (_groupJid, _nowMs) => null,
@@ -32,7 +52,7 @@ export const NOOP_PARTICIPANTS_STORE = Object.freeze({
32
52
  });
33
53
  export const NOOP_DEVICE_LIST_STORE = Object.freeze({
34
54
  upsertUserDevicesBatch: async (_snapshots) => { },
35
- getUserDevicesBatch: async (userJids, _nowMs) => userJids.map(() => null),
55
+ getUserDevicesBatch: async (userJids, _nowMs) => new Array(userJids.length).fill(null),
36
56
  deleteUserDevices: async (_userJid) => 0,
37
57
  cleanupExpired: async (_nowMs) => 0,
38
58
  clear: async () => { },
@@ -7,13 +7,6 @@ const DEFAULT_APPSTATE_MEMORY_STORE_LIMITS = Object.freeze({
7
7
  syncKeys: 4096,
8
8
  collectionEntries: 50000
9
9
  });
10
- function toBoundedMap(entries, maxEntries) {
11
- const map = new Map();
12
- for (const [key, value] of entries) {
13
- setBoundedMapEntry(map, key, value, maxEntries);
14
- }
15
- return map;
16
- }
17
10
  export class WaAppStateMemoryStore {
18
11
  constructor(initial, options = {}) {
19
12
  this.keys = new Map();
@@ -28,11 +21,15 @@ export class WaAppStateMemoryStore {
28
21
  if (!collection) {
29
22
  continue;
30
23
  }
24
+ const indexValueMap = new Map();
25
+ for (const [key, value] of Object.entries(collection.indexValueMap)) {
26
+ setBoundedMapEntry(indexValueMap, key, value, this.maxCollectionEntries);
27
+ }
31
28
  this.collections.set(collectionName, {
32
29
  initialized: true,
33
30
  version: collection.version,
34
31
  hash: collection.hash,
35
- indexValueMap: toBoundedMap(Object.entries(collection.indexValueMap), this.maxCollectionEntries)
32
+ indexValueMap
36
33
  });
37
34
  }
38
35
  }
@@ -65,14 +62,22 @@ export class WaAppStateMemoryStore {
65
62
  }
66
63
  return inserted;
67
64
  }
68
- async getSyncKey(keyId) {
69
- return this.keys.get(bytesToHex(keyId)) ?? null;
65
+ async getSyncKeysBatch(keyIds) {
66
+ const keys = new Array(keyIds.length);
67
+ for (let index = 0; index < keyIds.length; index += 1) {
68
+ keys[index] = this.keys.get(bytesToHex(keyIds[index])) ?? null;
69
+ }
70
+ return keys;
70
71
  }
71
72
  async getSyncKeyData(keyId) {
72
73
  return this.keys.get(bytesToHex(keyId))?.keyData ?? null;
73
74
  }
74
75
  async getSyncKeyDataBatch(keyIds) {
75
- return keyIds.map((keyId) => this.keys.get(bytesToHex(keyId))?.keyData ?? null);
76
+ const keys = new Array(keyIds.length);
77
+ for (let index = 0; index < keyIds.length; index += 1) {
78
+ keys[index] = this.keys.get(bytesToHex(keyIds[index]))?.keyData ?? null;
79
+ }
80
+ return keys;
76
81
  }
77
82
  async getActiveSyncKey() {
78
83
  return pickActiveSyncKey(this.keys.values());
@@ -93,28 +98,21 @@ export class WaAppStateMemoryStore {
93
98
  async getCollectionStates(collections) {
94
99
  const result = new Array(collections.length);
95
100
  for (let index = 0; index < collections.length; index += 1) {
96
- const collection = collections[index];
97
- let state = this.collections.get(collection);
98
- if (!state) {
99
- state = {
100
- initialized: false,
101
- version: 0,
102
- hash: APP_STATE_EMPTY_LT_HASH,
103
- indexValueMap: new Map()
104
- };
105
- this.collections.set(collection, state);
106
- }
107
- result[index] = state;
101
+ result[index] = await this.getCollectionState(collections[index]);
108
102
  }
109
103
  return result;
110
104
  }
111
105
  async setCollectionStates(updates) {
112
106
  for (const update of updates) {
107
+ const indexValueMap = new Map();
108
+ for (const [key, value] of update.indexValueMap.entries()) {
109
+ setBoundedMapEntry(indexValueMap, key, value, this.maxCollectionEntries);
110
+ }
113
111
  this.collections.set(update.collection, {
114
112
  initialized: true,
115
113
  version: update.version,
116
114
  hash: update.hash,
117
- indexValueMap: toBoundedMap(update.indexValueMap.entries(), this.maxCollectionEntries)
115
+ indexValueMap
118
116
  });
119
117
  }
120
118
  }
@@ -6,6 +6,9 @@ const DEFAULTS = Object.freeze({
6
6
  });
7
7
  export class WaDeviceListMemoryStore {
8
8
  constructor(ttlMs = DEFAULTS.ttlMs, options = {}) {
9
+ if (!Number.isFinite(ttlMs) || ttlMs <= 0) {
10
+ throw new Error('device-list ttlMs must be a positive finite number');
11
+ }
9
12
  this.records = new Map();
10
13
  this.ttlMs = ttlMs;
11
14
  this.maxUsers = resolvePositive(options.maxUsers, DEFAULTS.maxUsers, 'WaDeviceListMemoryStoreOptions.maxUsers');
@@ -24,17 +27,22 @@ export class WaDeviceListMemoryStore {
24
27
  }
25
28
  }
26
29
  async getUserDevicesBatch(userJids, nowMs = Date.now()) {
27
- return userJids.map((userJid) => {
30
+ const snapshots = new Array(userJids.length);
31
+ for (let index = 0; index < userJids.length; index += 1) {
32
+ const userJid = userJids[index];
28
33
  const record = this.records.get(userJid);
29
34
  if (!record) {
30
- return null;
35
+ snapshots[index] = null;
36
+ continue;
31
37
  }
32
38
  if (record.expiresAtMs <= nowMs) {
33
39
  this.records.delete(userJid);
34
- return null;
40
+ snapshots[index] = null;
41
+ continue;
35
42
  }
36
- return record;
37
- });
43
+ snapshots[index] = record;
44
+ }
45
+ return snapshots;
38
46
  }
39
47
  async deleteUserDevices(userJid) {
40
48
  return this.records.delete(userJid) ? 1 : 0;
@@ -0,0 +1,31 @@
1
+ import { signalAddressKey } from '../../../protocol/jid.js';
2
+ import { resolvePositive } from '../../../util/coercion.js';
3
+ import { setBoundedMapEntry } from '../../../util/collections.js';
4
+ const DEFAULT_MAX_REMOTE_IDENTITIES = 8192;
5
+ export class WaIdentityMemoryStore {
6
+ constructor(options = {}) {
7
+ this.remoteIdentities = new Map();
8
+ this.maxRemoteIdentities = resolvePositive(options.maxRemoteIdentities, DEFAULT_MAX_REMOTE_IDENTITIES, 'WaIdentityMemoryStoreOptions.maxRemoteIdentities');
9
+ }
10
+ async getRemoteIdentity(address) {
11
+ return this.remoteIdentities.get(signalAddressKey(address)) ?? null;
12
+ }
13
+ async getRemoteIdentities(addresses) {
14
+ const result = new Array(addresses.length);
15
+ for (let i = 0; i < addresses.length; i += 1) {
16
+ result[i] = this.remoteIdentities.get(signalAddressKey(addresses[i])) ?? null;
17
+ }
18
+ return result;
19
+ }
20
+ async setRemoteIdentity(address, identityKey) {
21
+ setBoundedMapEntry(this.remoteIdentities, signalAddressKey(address), identityKey, this.maxRemoteIdentities);
22
+ }
23
+ async setRemoteIdentities(entries) {
24
+ for (const entry of entries) {
25
+ setBoundedMapEntry(this.remoteIdentities, signalAddressKey(entry.address), entry.identityKey, this.maxRemoteIdentities);
26
+ }
27
+ }
28
+ async clear() {
29
+ this.remoteIdentities.clear();
30
+ }
31
+ }
@@ -0,0 +1,81 @@
1
+ import { resolvePositive } from '../../../util/coercion.js';
2
+ import { resolveCleanupIntervalMs, setBoundedMapEntry } from '../../../util/collections.js';
3
+ const DEFAULTS = Object.freeze({
4
+ ttlMs: 30 * 60 * 1000,
5
+ maxSecrets: 10000
6
+ });
7
+ export class WaMessageSecretMemoryStore {
8
+ constructor(ttlMs = DEFAULTS.ttlMs, options = {}) {
9
+ if (!Number.isFinite(ttlMs) || ttlMs <= 0) {
10
+ throw new Error('message-secret ttlMs must be a positive finite number');
11
+ }
12
+ this.secrets = new Map();
13
+ this.ttlMs = ttlMs;
14
+ this.maxSecrets = resolvePositive(options.maxSecrets, DEFAULTS.maxSecrets, 'WaMessageSecretMemoryStoreOptions.maxSecrets');
15
+ this.cleanupTimer = setInterval(() => {
16
+ void this.cleanupExpired(Date.now());
17
+ }, resolveCleanupIntervalMs(ttlMs));
18
+ this.cleanupTimer.unref();
19
+ }
20
+ async get(messageId, nowMs = Date.now()) {
21
+ const cached = this.secrets.get(messageId);
22
+ if (!cached)
23
+ return null;
24
+ if (cached.expiresAtMs <= nowMs) {
25
+ this.secrets.delete(messageId);
26
+ return null;
27
+ }
28
+ return { secret: cached.secret, senderJid: cached.senderJid };
29
+ }
30
+ async getBatch(messageIds, nowMs = Date.now()) {
31
+ const result = new Array(messageIds.length);
32
+ for (let i = 0; i < messageIds.length; i += 1) {
33
+ const cached = this.secrets.get(messageIds[i]);
34
+ if (!cached) {
35
+ result[i] = null;
36
+ continue;
37
+ }
38
+ if (cached.expiresAtMs <= nowMs) {
39
+ this.secrets.delete(messageIds[i]);
40
+ result[i] = null;
41
+ continue;
42
+ }
43
+ result[i] = { secret: cached.secret, senderJid: cached.senderJid };
44
+ }
45
+ return result;
46
+ }
47
+ async set(messageId, entry) {
48
+ setBoundedMapEntry(this.secrets, messageId, {
49
+ secret: entry.secret,
50
+ senderJid: entry.senderJid,
51
+ expiresAtMs: Date.now() + this.ttlMs
52
+ }, this.maxSecrets);
53
+ }
54
+ async setBatch(entries) {
55
+ const nowMs = Date.now();
56
+ for (let i = 0; i < entries.length; i += 1) {
57
+ setBoundedMapEntry(this.secrets, entries[i].messageId, {
58
+ secret: entries[i].entry.secret,
59
+ senderJid: entries[i].entry.senderJid,
60
+ expiresAtMs: nowMs + this.ttlMs
61
+ }, this.maxSecrets);
62
+ }
63
+ }
64
+ async cleanupExpired(nowMs) {
65
+ let removed = 0;
66
+ for (const [messageId, entry] of this.secrets) {
67
+ if (entry.expiresAtMs > nowMs)
68
+ continue;
69
+ this.secrets.delete(messageId);
70
+ removed += 1;
71
+ }
72
+ return removed;
73
+ }
74
+ async clear() {
75
+ this.secrets.clear();
76
+ }
77
+ async destroy() {
78
+ clearInterval(this.cleanupTimer);
79
+ this.secrets.clear();
80
+ }
81
+ }
@@ -6,6 +6,9 @@ const DEFAULTS = Object.freeze({
6
6
  });
7
7
  export class WaParticipantsMemoryStore {
8
8
  constructor(ttlMs = DEFAULTS.ttlMs, options = {}) {
9
+ if (!Number.isFinite(ttlMs) || ttlMs <= 0) {
10
+ throw new Error('participants ttlMs must be a positive finite number');
11
+ }
9
12
  this.records = new Map();
10
13
  this.ttlMs = ttlMs;
11
14
  this.maxGroups = resolvePositive(options.maxGroups, DEFAULTS.maxGroups, 'WaParticipantsMemoryStoreOptions.maxGroups');
@@ -0,0 +1,97 @@
1
+ import { resolvePositive } from '../../../util/coercion.js';
2
+ import { setBoundedMapEntry } from '../../../util/collections.js';
3
+ const DEFAULT_MAX_PRE_KEYS = 4096;
4
+ export class WaPreKeyMemoryStore {
5
+ constructor(options = {}) {
6
+ this.preKeys = new Map();
7
+ this.uploadedPreKeys = new Set();
8
+ this.serverHasPreKeys = false;
9
+ this.nextPreKeyId = 1;
10
+ this.maxPreKeys = resolvePositive(options.maxPreKeys, DEFAULT_MAX_PRE_KEYS, 'WaPreKeyMemoryStoreOptions.maxPreKeys');
11
+ }
12
+ async putPreKey(record) {
13
+ setBoundedMapEntry(this.preKeys, record.keyId, record, this.maxPreKeys, (keyId) => {
14
+ this.uploadedPreKeys.delete(keyId);
15
+ });
16
+ if (record.keyId >= this.nextPreKeyId) {
17
+ this.nextPreKeyId = record.keyId + 1;
18
+ }
19
+ }
20
+ async getOrGenPreKeys(count, generator) {
21
+ if (!Number.isSafeInteger(count) || count <= 0) {
22
+ throw new Error(`invalid prekey count: ${count}`);
23
+ }
24
+ const available = [];
25
+ const availableKeyIds = [];
26
+ for (const keyId of this.preKeys.keys()) {
27
+ if (!this.uploadedPreKeys.has(keyId)) {
28
+ availableKeyIds.push(keyId);
29
+ }
30
+ }
31
+ availableKeyIds.sort((left, right) => left - right);
32
+ for (let index = 0; index < availableKeyIds.length; index += 1) {
33
+ const keyId = availableKeyIds[index];
34
+ const record = this.preKeys.get(keyId);
35
+ if (!record) {
36
+ continue;
37
+ }
38
+ available.push(record);
39
+ if (available.length >= count) {
40
+ return available;
41
+ }
42
+ }
43
+ while (available.length < count) {
44
+ const record = await generator(this.nextPreKeyId++);
45
+ setBoundedMapEntry(this.preKeys, record.keyId, record, this.maxPreKeys, (keyId) => {
46
+ this.uploadedPreKeys.delete(keyId);
47
+ });
48
+ available.push(record);
49
+ }
50
+ return available;
51
+ }
52
+ async getPreKeyById(keyId) {
53
+ return this.preKeys.get(keyId) ?? null;
54
+ }
55
+ async getPreKeysById(keyIds) {
56
+ const result = new Array(keyIds.length);
57
+ for (let i = 0; i < keyIds.length; i += 1) {
58
+ result[i] = this.preKeys.get(keyIds[i]) ?? null;
59
+ }
60
+ return result;
61
+ }
62
+ async consumePreKeyById(keyId) {
63
+ const record = this.preKeys.get(keyId) ?? null;
64
+ if (!record) {
65
+ return null;
66
+ }
67
+ this.preKeys.delete(keyId);
68
+ this.uploadedPreKeys.delete(keyId);
69
+ return record;
70
+ }
71
+ async getOrGenSinglePreKey(generator) {
72
+ const preKeys = await this.getOrGenPreKeys(1, generator);
73
+ return preKeys[0];
74
+ }
75
+ async markKeyAsUploaded(keyId) {
76
+ if (keyId < 0 || keyId >= this.nextPreKeyId) {
77
+ throw new Error(`prekey ${keyId} is out of boundary`);
78
+ }
79
+ for (const candidate of this.preKeys.keys()) {
80
+ if (candidate <= keyId) {
81
+ this.uploadedPreKeys.add(candidate);
82
+ }
83
+ }
84
+ }
85
+ async setServerHasPreKeys(value) {
86
+ this.serverHasPreKeys = value;
87
+ }
88
+ async getServerHasPreKeys() {
89
+ return this.serverHasPreKeys;
90
+ }
91
+ async clear() {
92
+ this.preKeys.clear();
93
+ this.uploadedPreKeys.clear();
94
+ this.serverHasPreKeys = false;
95
+ this.nextPreKeyId = 1;
96
+ }
97
+ }
@@ -0,0 +1,43 @@
1
+ import { setBoundedMapEntry } from '../../../util/collections.js';
2
+ const DEFAULT_MAX_ENTRIES = 10000;
3
+ export class WaPrivacyTokenMemoryStore {
4
+ constructor(maxEntries = DEFAULT_MAX_ENTRIES) {
5
+ this.records = new Map();
6
+ this.maxEntries = maxEntries;
7
+ }
8
+ async upsert(record) {
9
+ const existing = this.records.get(record.jid);
10
+ const merged = existing ? this.mergeRecord(existing, record) : record;
11
+ setBoundedMapEntry(this.records, record.jid, merged, this.maxEntries);
12
+ }
13
+ async upsertBatch(records) {
14
+ for (let i = 0; i < records.length; i += 1) {
15
+ const record = records[i];
16
+ const existing = this.records.get(record.jid);
17
+ const merged = existing ? this.mergeRecord(existing, record) : record;
18
+ setBoundedMapEntry(this.records, record.jid, merged, this.maxEntries);
19
+ }
20
+ }
21
+ async getByJid(jid) {
22
+ return this.records.get(jid) ?? null;
23
+ }
24
+ async deleteByJid(jid) {
25
+ return this.records.delete(jid) ? 1 : 0;
26
+ }
27
+ async clear() {
28
+ this.records.clear();
29
+ }
30
+ async destroy() {
31
+ this.records.clear();
32
+ }
33
+ mergeRecord(existing, incoming) {
34
+ return {
35
+ jid: incoming.jid,
36
+ tcToken: incoming.tcToken ?? existing.tcToken,
37
+ tcTokenTimestamp: incoming.tcTokenTimestamp ?? existing.tcTokenTimestamp,
38
+ tcTokenSenderTimestamp: incoming.tcTokenSenderTimestamp ?? existing.tcTokenSenderTimestamp,
39
+ nctSalt: incoming.nctSalt ?? existing.nctSalt,
40
+ updatedAtMs: incoming.updatedAtMs
41
+ };
42
+ }
43
+ }