franpcode/Baileys
1
0
Fork 0
mirror of https://github.com/FranP-code/Baileys.git synced 2025-10-13 00:32:22 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat: implement message retry handling

Browse Source 
so if a message fails to decrypt on the other user's end -- it can be retried
This commit is contained in:
Adhiraj Singh
2021-12-04 13:47:24 +05:30
parent 903871d180
commit 78fd72c8e5
6 changed files with 91 additions and 33 deletions
Download Patch File Download Diff File Expand all files Collapse all files

66
src/Socket/messages-recv.ts
View File

@@ -6,6 +6,7 @@ import { proto } from "../../WAProto"
import { KEY_BUNDLE_TYPE } from "../Defaults"
import { makeChatsSocket } from "./chats"
import { extractGroupMetadata } from "./groups"
import { Boom } from "@hapi/boom"
const getStatusFromReceiptType = (type: string | undefined) => {
if(type === 'read' || type === 'read-self') {
@@ -24,6 +25,7 @@ export const makeMessagesRecvSocket = (config: SocketConfig) => {
ev,
authState,
ws,
assertSession,
assertingPreKeys,
sendNode,
relayMessage,
@@ -464,30 +466,64 @@ export const makeMessagesRecvSocket = (config: SocketConfig) => {
}
})
const sendMessagesAgain = async(key: proto.IMessageKey, ids: string[]) => {
const participant = key.participant || key.remoteJid
await assertSession(participant, true)
logger.debug({ key, ids }, 'recv retry request, forced new session')
const msgs = await Promise.all(
ids.map(id => (
config.getMessage({ ...key, id })
))
)
const missingMsgIdx = msgs.findIndex(m => !m)
if(missingMsgIdx >= 0) {
throw new Boom(
`recv request to retry message, but message "${ids[missingMsgIdx]}" not available`,
{ statusCode: 404, data: { key } }
)
}
for(let i = 0; i < msgs.length;i++) {
await relayMessage(key.remoteJid, msgs[i], {
messageId: ids[i],
participant
})
}
}
const handleReceipt = async(node: BinaryNode) => {
const { attrs, content } = node
const status = getStatusFromReceiptType(attrs.type)
const remoteJid = attrs.recipient || attrs.from
const fromMe = attrs.recipient ? false : true
const ids = [attrs.id]
if(Array.isArray(content)) {
const items = getBinaryNodeChildren(content[0], 'item')
ids.push(...items.map(i => i.attrs.id))
}
const key: proto.IMessageKey = {
remoteJid,
id: '',
fromMe,
participant: attrs.participant
}
const status = getStatusFromReceiptType(attrs.type)
if(typeof status !== 'undefined' && !areJidsSameUser(attrs.from, authState.creds.me?.id)) {
const ids = [attrs.id]
if(Array.isArray(content)) {
const items = getBinaryNodeChildren(content[0], 'item')
ids.push(...items.map(i => i.attrs.id))
}
const remoteJid = attrs.recipient || attrs.from
const fromMe = attrs.recipient ? false : true
ev.emit('messages.update', ids.map(id => ({
key: {
remoteJid,
id: id,
fromMe,
participant: attrs.participant
},
key: { ...key, id },
update: { status }
})))
}
if(attrs.type === 'retry') {
await sendMessagesAgain(key, ids)
}
await sendMessageAck(node, { class: 'receipt', type: attrs.type })
}