Adhiraj Singh
126 lines
3.4 KiB
TypeScript
126 lines
3.4 KiB
TypeScript
import { promisify } from 'util'
|
|
import { inflate } from 'zlib'
|
|
import { proto } from '../../WAProto'
|
|
import { Chat, Contact, InitialReceivedChatsState } from '../Types'
|
|
import { isJidUser } from '../WABinary'
|
|
import { toNumber } from './generics'
|
|
import { normalizeMessageContent } from './messages'
|
|
import { downloadContentFromMessage } from './messages-media'
|
|
|
|
const inflatePromise = promisify(inflate)
|
|
|
|
export const downloadHistory = async(msg: proto.Message.IHistorySyncNotification) => {
|
|
const stream = await downloadContentFromMessage(msg, 'md-msg-hist')
|
|
const bufferArray: Buffer[] = []
|
|
for await (const chunk of stream) {
|
|
bufferArray.push(chunk)
|
|
}
|
|
|
|
let buffer = Buffer.concat(bufferArray)
|
|
|
|
// decompress buffer
|
|
buffer = await inflatePromise(buffer)
|
|
|
|
const syncData = proto.HistorySync.decode(buffer)
|
|
return syncData
|
|
}
|
|
|
|
export const processHistoryMessage = (
|
|
item: proto.IHistorySync,
|
|
historyCache: Set<string>,
|
|
recvChats: InitialReceivedChatsState
|
|
) => {
|
|
const messages: proto.IWebMessageInfo[] = []
|
|
const contacts: Contact[] = []
|
|
const chats: Chat[] = []
|
|
|
|
switch (item.syncType) {
|
|
case proto.HistorySync.HistorySyncType.INITIAL_BOOTSTRAP:
|
|
case proto.HistorySync.HistorySyncType.RECENT:
|
|
for(const chat of item.conversations!) {
|
|
const contactId = `c:${chat.id}`
|
|
if(chat.name && !historyCache.has(contactId)) {
|
|
contacts.push({ id: chat.id, name: chat.name })
|
|
historyCache.add(contactId)
|
|
}
|
|
|
|
const msgs = chat.messages || []
|
|
delete chat.messages
|
|
|
|
for(const item of msgs) {
|
|
const message = item.message!
|
|
const uqId = `${message.key.remoteJid}:${message.key.id}`
|
|
if(!historyCache.has(uqId)) {
|
|
messages.push(message)
|
|
|
|
let curItem = recvChats[message.key.remoteJid!]
|
|
const timestamp = toNumber(message.messageTimestamp)
|
|
if(!curItem || timestamp > curItem.lastMsgTimestamp) {
|
|
curItem = { lastMsgTimestamp: timestamp }
|
|
recvChats[chat.id] = curItem
|
|
// keep only the most recent message in the chat array
|
|
chat.messages = [{ message }]
|
|
}
|
|
|
|
if(
|
|
!message.key.fromMe
|
|
&& (!curItem?.lastMsgRecvTimestamp || timestamp > curItem.lastMsgRecvTimestamp)
|
|
) {
|
|
curItem.lastMsgRecvTimestamp = timestamp
|
|
}
|
|
|
|
historyCache.add(uqId)
|
|
}
|
|
}
|
|
|
|
if(!historyCache.has(chat.id)) {
|
|
if(isJidUser(chat.id) && chat.readOnly && chat.archived) {
|
|
chat.readOnly = false
|
|
}
|
|
|
|
chats.push(chat)
|
|
historyCache.add(chat.id)
|
|
}
|
|
}
|
|
|
|
break
|
|
case proto.HistorySync.HistorySyncType.PUSH_NAME:
|
|
for(const c of item.pushnames!) {
|
|
const contactId = `c:${c.id}`
|
|
if(!historyCache.has(contactId)) {
|
|
contacts.push({ notify: c.pushname!, id: c.id! })
|
|
historyCache.add(contactId)
|
|
}
|
|
}
|
|
|
|
break
|
|
case proto.HistorySync.HistorySyncType.INITIAL_STATUS_V3:
|
|
// TODO
|
|
break
|
|
}
|
|
|
|
const didProcess = !!(chats.length || messages.length || contacts.length)
|
|
|
|
return {
|
|
chats,
|
|
contacts,
|
|
messages,
|
|
didProcess,
|
|
}
|
|
}
|
|
|
|
export const downloadAndProcessHistorySyncNotification = async(
|
|
msg: proto.Message.IHistorySyncNotification,
|
|
historyCache: Set<string>,
|
|
recvChats: InitialReceivedChatsState
|
|
) => {
|
|
const historyMsg = await downloadHistory(msg)
|
|
return processHistoryMessage(historyMsg, historyCache, recvChats)
|
|
}
|
|
|
|
export const isHistoryMsg = (message: proto.IMessage) => {
|
|
const normalizedContent = !!message ? normalizeMessageContent(message) : undefined
|
|
const isAnyHistoryMsg = !!normalizedContent?.protocolMessage?.historySyncNotification
|
|
|
|
return isAnyHistoryMsg
|
|
} |