feat: fetch groups if ib:dirty

This commit is contained in:
SamuelScheit
2023-05-14 17:44:56 +02:00
parent 68a6fe40c8
commit 07562c2204
2 changed files with 60 additions and 41 deletions

View File

@@ -340,8 +340,8 @@ export const makeChatsSocket = (config: SocketConfig) => {
}
}
const updateAccountSyncTimestamp = async(fromTimestamp: number | string) => {
logger.info({ fromTimestamp }, 'requesting account sync')
const cleanDirtyBits = async(type: 'account_sync' | 'groups', fromTimestamp?: number | string) => {
logger.info({ fromTimestamp }, 'clean dirty bits ' + type)
await sendNode({
tag: 'iq',
attrs: {
@@ -354,8 +354,8 @@ export const makeChatsSocket = (config: SocketConfig) => {
{
tag: 'clean',
attrs: {
type: 'account_sync',
timestamp: fromTimestamp.toString(),
type,
...(fromTimestamp ? { timestamp: fromTimestamp.toString() } : null),
}
}
]
@@ -879,13 +879,16 @@ export const makeChatsSocket = (config: SocketConfig) => {
if(attrs.timestamp) {
let { lastAccountSyncTimestamp } = authState.creds
if(lastAccountSyncTimestamp) {
await updateAccountSyncTimestamp(lastAccountSyncTimestamp)
await cleanDirtyBits('account_sync', lastAccountSyncTimestamp)
}
lastAccountSyncTimestamp = +attrs.timestamp
ev.emit('creds.update', { lastAccountSyncTimestamp })
}
break
case 'groups':
// handled in groups.ts
break
default:
logger.info({ node }, 'received unknown sync')
@@ -945,6 +948,7 @@ export const makeChatsSocket = (config: SocketConfig) => {
updateDefaultDisappearingMode,
getBusinessProfile,
resyncAppState,
chatModify
chatModify,
cleanDirtyBits
}
}

View File

@@ -29,6 +29,55 @@ export const makeGroupsSocket = (config: SocketConfig) => {
return extractGroupMetadata(result)
}
const groupFetchAllParticipating = async() => {
const result = await query({
tag: 'iq',
attrs: {
to: '@g.us',
xmlns: 'w:g2',
type: 'get',
},
content: [
{
tag: 'participating',
attrs: { },
content: [
{ tag: 'participants', attrs: { } },
{ tag: 'description', attrs: { } }
]
}
]
})
const data: { [_: string]: GroupMetadata } = { }
const groupsChild = getBinaryNodeChild(result, 'groups')
if(groupsChild) {
const groups = getBinaryNodeChildren(groupsChild, 'group')
for(const groupNode of groups) {
const meta = extractGroupMetadata({
tag: 'result',
attrs: { },
content: [groupNode]
})
data[meta.id] = meta
}
}
sock.ev.emit('groups.update', Object.values(data))
return data
}
sock.ws.on('CB:ib,,dirty', async(node: BinaryNode) => {
const { attrs } = getBinaryNodeChild(node, 'dirty')!
if(attrs.type !== 'groups') {
return
}
await groupFetchAllParticipating()
await sock.cleanDirtyBits('groups')
})
return {
...sock,
groupMetadata,
@@ -211,41 +260,7 @@ export const makeGroupsSocket = (config: SocketConfig) => {
groupSettingUpdate: async(jid: string, setting: 'announcement' | 'not_announcement' | 'locked' | 'unlocked') => {
await groupQuery(jid, 'set', [ { tag: setting, attrs: { } } ])
},
groupFetchAllParticipating: async() => {
const result = await query({
tag: 'iq',
attrs: {
to: '@g.us',
xmlns: 'w:g2',
type: 'get',
},
content: [
{
tag: 'participating',
attrs: { },
content: [
{ tag: 'participants', attrs: { } },
{ tag: 'description', attrs: { } }
]
}
]
})
const data: { [_: string]: GroupMetadata } = { }
const groupsChild = getBinaryNodeChild(result, 'groups')
if(groupsChild) {
const groups = getBinaryNodeChildren(groupsChild, 'group')
for(const groupNode of groups) {
const meta = extractGroupMetadata({
tag: 'result',
attrs: { },
content: [groupNode]
})
data[meta.id] = meta
}
}
return data
}
groupFetchAllParticipating
}
}