feat: community support (#1157)
* feat: communities * feat: update makeCommunitiesSocket * chore(release): v6.7.10 * fix: ajust create communitie * feat: update communities * revert: package.json * fix: Revert CHANGELOG.md * fix: Revert CHANGELOG.md * fix: Revert CHANGELOG.md * fix: revert package-lock.json * fix: revert package-lock.json * fix: revert yarn.lock * fix: remove comment * restore yarn.lock * fix: improve type imports and formatting in communities.ts * fix: remove trailing semicolon in baileys-version.ts * fix: Delete src/Defaults/baileys-version.ts --------- Co-authored-by: João Lucas <jlucaso@hotmail.com>
This commit is contained in:
João Victor Souza
committed by
GitHub
GitHub
parent
8d2cdbcf98
commit
0c20166a45
@@ -0,0 +1,394 @@
|
||||
import { proto } from '../../WAProto'
|
||||
import {
|
||||
type GroupMetadata,
|
||||
type GroupParticipant,
|
||||
type ParticipantAction,
|
||||
type SocketConfig,
|
||||
type WAMessageKey,
|
||||
WAMessageStubType
|
||||
} from '../Types'
|
||||
import { generateMessageID, generateMessageIDV2, unixTimestampSeconds } from '../Utils'
|
||||
import logger from '../Utils/logger'
|
||||
import {
|
||||
type BinaryNode,
|
||||
getBinaryNodeChild,
|
||||
getBinaryNodeChildren,
|
||||
getBinaryNodeChildString,
|
||||
jidEncode,
|
||||
jidNormalizedUser
|
||||
} from '../WABinary'
|
||||
import { makeBusinessSocket } from './business'
|
||||
|
||||
export const makeCommunitiesSocket = (config: SocketConfig) => {
|
||||
const sock = makeBusinessSocket(config)
|
||||
const { authState, ev, query, upsertMessage } = sock
|
||||
|
||||
const communityQuery = async (jid: string, type: 'get' | 'set', content: BinaryNode[]) =>
|
||||
query({
|
||||
tag: 'iq',
|
||||
attrs: {
|
||||
type,
|
||||
xmlns: 'w:g2',
|
||||
to: jid
|
||||
},
|
||||
content
|
||||
})
|
||||
|
||||
const communityMetadata = async (jid: string) => {
|
||||
const result = await communityQuery(jid, 'get', [{ tag: 'query', attrs: { request: 'interactive' } }])
|
||||
return extractCommunityMetadata(result)
|
||||
}
|
||||
|
||||
const communityFetchAllParticipating = 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 communitiesChild = getBinaryNodeChild(result, 'communities')
|
||||
if (communitiesChild) {
|
||||
const communities = getBinaryNodeChildren(communitiesChild, 'community')
|
||||
for (const communityNode of communities) {
|
||||
const meta = extractCommunityMetadata({
|
||||
tag: 'result',
|
||||
attrs: {},
|
||||
content: [communityNode]
|
||||
})
|
||||
data[meta.id] = meta
|
||||
}
|
||||
}
|
||||
|
||||
sock.ev.emit('groups.update', Object.values(data))
|
||||
|
||||
return data
|
||||
}
|
||||
|
||||
async function parseGroupResult(node: BinaryNode) {
|
||||
logger.info({ node }, 'parseGroupResult')
|
||||
const groupNode = getBinaryNodeChild(node, 'group')
|
||||
if (groupNode) {
|
||||
try {
|
||||
logger.info({ groupNode }, 'groupNode')
|
||||
const metadata = await sock.groupMetadata(`${groupNode.attrs.id}@g.us`)
|
||||
return metadata ? metadata : Optional.empty()
|
||||
} catch (error) {
|
||||
console.error('Error parsing group metadata:', error)
|
||||
return Optional.empty()
|
||||
}
|
||||
}
|
||||
|
||||
return Optional.empty()
|
||||
}
|
||||
|
||||
const Optional = {
|
||||
empty: () => null,
|
||||
of: (value: null) => (value !== null ? { value } : null)
|
||||
}
|
||||
|
||||
sock.ws.on('CB:ib,,dirty', async (node: BinaryNode) => {
|
||||
const { attrs } = getBinaryNodeChild(node, 'dirty')!
|
||||
if (attrs.type !== 'communities') {
|
||||
return
|
||||
}
|
||||
|
||||
await communityFetchAllParticipating()
|
||||
await sock.cleanDirtyBits('groups')
|
||||
})
|
||||
|
||||
return {
|
||||
...sock,
|
||||
communityMetadata,
|
||||
communityCreate: async (subject: string, body: string) => {
|
||||
const descriptionId = generateMessageID().substring(0, 12)
|
||||
|
||||
const result = await communityQuery('@g.us', 'set', [
|
||||
{
|
||||
tag: 'create',
|
||||
attrs: { subject },
|
||||
content: [
|
||||
{
|
||||
tag: 'description',
|
||||
attrs: { id: descriptionId },
|
||||
content: [
|
||||
{
|
||||
tag: 'body',
|
||||
attrs: {},
|
||||
content: Buffer.from(body || '', 'utf-8')
|
||||
}
|
||||
]
|
||||
},
|
||||
{
|
||||
tag: 'parent',
|
||||
attrs: { default_membership_approval_mode: 'request_required' }
|
||||
},
|
||||
{
|
||||
tag: 'allow_non_admin_sub_group_creation',
|
||||
attrs: {}
|
||||
},
|
||||
{
|
||||
tag: 'create_general_chat',
|
||||
attrs: {}
|
||||
}
|
||||
]
|
||||
}
|
||||
])
|
||||
|
||||
return await parseGroupResult(result)
|
||||
},
|
||||
communityLeave: async (id: string) => {
|
||||
await communityQuery('@g.us', 'set', [
|
||||
{
|
||||
tag: 'leave',
|
||||
attrs: {},
|
||||
content: [{ tag: 'community', attrs: { id } }]
|
||||
}
|
||||
])
|
||||
},
|
||||
communityUpdateSubject: async (jid: string, subject: string) => {
|
||||
await communityQuery(jid, 'set', [
|
||||
{
|
||||
tag: 'subject',
|
||||
attrs: {},
|
||||
content: Buffer.from(subject, 'utf-8')
|
||||
}
|
||||
])
|
||||
},
|
||||
communityRequestParticipantsList: async (jid: string) => {
|
||||
const result = await communityQuery(jid, 'get', [
|
||||
{
|
||||
tag: 'membership_approval_requests',
|
||||
attrs: {}
|
||||
}
|
||||
])
|
||||
const node = getBinaryNodeChild(result, 'membership_approval_requests')
|
||||
const participants = getBinaryNodeChildren(node, 'membership_approval_request')
|
||||
return participants.map(v => v.attrs)
|
||||
},
|
||||
communityRequestParticipantsUpdate: async (jid: string, participants: string[], action: 'approve' | 'reject') => {
|
||||
const result = await communityQuery(jid, 'set', [
|
||||
{
|
||||
tag: 'membership_requests_action',
|
||||
attrs: {},
|
||||
content: [
|
||||
{
|
||||
tag: action,
|
||||
attrs: {},
|
||||
content: participants.map(jid => ({
|
||||
tag: 'participant',
|
||||
attrs: { jid }
|
||||
}))
|
||||
}
|
||||
]
|
||||
}
|
||||
])
|
||||
const node = getBinaryNodeChild(result, 'membership_requests_action')
|
||||
const nodeAction = getBinaryNodeChild(node, action)
|
||||
const participantsAffected = getBinaryNodeChildren(nodeAction, 'participant')
|
||||
return participantsAffected.map(p => {
|
||||
return { status: p.attrs.error || '200', jid: p.attrs.jid }
|
||||
})
|
||||
},
|
||||
communityParticipantsUpdate: async (jid: string, participants: string[], action: ParticipantAction) => {
|
||||
const result = await communityQuery(jid, 'set', [
|
||||
{
|
||||
tag: action,
|
||||
attrs: {},
|
||||
content: participants.map(jid => ({
|
||||
tag: 'participant',
|
||||
attrs: { jid }
|
||||
}))
|
||||
}
|
||||
])
|
||||
const node = getBinaryNodeChild(result, action)
|
||||
const participantsAffected = getBinaryNodeChildren(node, 'participant')
|
||||
return participantsAffected.map(p => {
|
||||
return { status: p.attrs.error || '200', jid: p.attrs.jid, content: p }
|
||||
})
|
||||
},
|
||||
communityUpdateDescription: async (jid: string, description?: string) => {
|
||||
const metadata = await communityMetadata(jid)
|
||||
const prev = metadata.descId ?? null
|
||||
|
||||
await communityQuery(jid, 'set', [
|
||||
{
|
||||
tag: 'description',
|
||||
attrs: {
|
||||
...(description ? { id: generateMessageID() } : { delete: 'true' }),
|
||||
...(prev ? { prev } : {})
|
||||
},
|
||||
content: description ? [{ tag: 'body', attrs: {}, content: Buffer.from(description, 'utf-8') }] : undefined
|
||||
}
|
||||
])
|
||||
},
|
||||
communityInviteCode: async (jid: string) => {
|
||||
const result = await communityQuery(jid, 'get', [{ tag: 'invite', attrs: {} }])
|
||||
const inviteNode = getBinaryNodeChild(result, 'invite')
|
||||
return inviteNode?.attrs.code
|
||||
},
|
||||
communityRevokeInvite: async (jid: string) => {
|
||||
const result = await communityQuery(jid, 'set', [{ tag: 'invite', attrs: {} }])
|
||||
const inviteNode = getBinaryNodeChild(result, 'invite')
|
||||
return inviteNode?.attrs.code
|
||||
},
|
||||
communityAcceptInvite: async (code: string) => {
|
||||
const results = await communityQuery('@g.us', 'set', [{ tag: 'invite', attrs: { code } }])
|
||||
const result = getBinaryNodeChild(results, 'community')
|
||||
return result?.attrs.jid
|
||||
},
|
||||
|
||||
/**
|
||||
* revoke a v4 invite for someone
|
||||
* @param communityJid community jid
|
||||
* @param invitedJid jid of person you invited
|
||||
* @returns true if successful
|
||||
*/
|
||||
communityRevokeInviteV4: async (communityJid: string, invitedJid: string) => {
|
||||
const result = await communityQuery(communityJid, 'set', [
|
||||
{ tag: 'revoke', attrs: {}, content: [{ tag: 'participant', attrs: { jid: invitedJid } }] }
|
||||
])
|
||||
return !!result
|
||||
},
|
||||
|
||||
/**
|
||||
* accept a CommunityInviteMessage
|
||||
* @param key the key of the invite message, or optionally only provide the jid of the person who sent the invite
|
||||
* @param inviteMessage the message to accept
|
||||
*/
|
||||
communityAcceptInviteV4: ev.createBufferedFunction(
|
||||
async (key: string | WAMessageKey, inviteMessage: proto.Message.IGroupInviteMessage) => {
|
||||
key = typeof key === 'string' ? { remoteJid: key } : key
|
||||
const results = await communityQuery(inviteMessage.groupJid!, 'set', [
|
||||
{
|
||||
tag: 'accept',
|
||||
attrs: {
|
||||
code: inviteMessage.inviteCode!,
|
||||
expiration: inviteMessage.inviteExpiration!.toString(),
|
||||
admin: key.remoteJid!
|
||||
}
|
||||
}
|
||||
])
|
||||
|
||||
// if we have the full message key
|
||||
// update the invite message to be expired
|
||||
if (key.id) {
|
||||
// create new invite message that is expired
|
||||
inviteMessage = proto.Message.GroupInviteMessage.fromObject(inviteMessage)
|
||||
inviteMessage.inviteExpiration = 0
|
||||
inviteMessage.inviteCode = ''
|
||||
ev.emit('messages.update', [
|
||||
{
|
||||
key,
|
||||
update: {
|
||||
message: {
|
||||
groupInviteMessage: inviteMessage
|
||||
}
|
||||
}
|
||||
}
|
||||
])
|
||||
}
|
||||
|
||||
// generate the community add message
|
||||
await upsertMessage(
|
||||
{
|
||||
key: {
|
||||
remoteJid: inviteMessage.groupJid,
|
||||
id: generateMessageIDV2(sock.user?.id),
|
||||
fromMe: false,
|
||||
participant: key.remoteJid
|
||||
},
|
||||
messageStubType: WAMessageStubType.GROUP_PARTICIPANT_ADD,
|
||||
messageStubParameters: [authState.creds.me!.id],
|
||||
participant: key.remoteJid,
|
||||
messageTimestamp: unixTimestampSeconds()
|
||||
},
|
||||
'notify'
|
||||
)
|
||||
|
||||
return results.attrs.from
|
||||
}
|
||||
),
|
||||
communityGetInviteInfo: async (code: string) => {
|
||||
const results = await communityQuery('@g.us', 'get', [{ tag: 'invite', attrs: { code } }])
|
||||
return extractCommunityMetadata(results)
|
||||
},
|
||||
communityToggleEphemeral: async (jid: string, ephemeralExpiration: number) => {
|
||||
const content: BinaryNode = ephemeralExpiration
|
||||
? { tag: 'ephemeral', attrs: { expiration: ephemeralExpiration.toString() } }
|
||||
: { tag: 'not_ephemeral', attrs: {} }
|
||||
await communityQuery(jid, 'set', [content])
|
||||
},
|
||||
communitySettingUpdate: async (
|
||||
jid: string,
|
||||
setting: 'announcement' | 'not_announcement' | 'locked' | 'unlocked'
|
||||
) => {
|
||||
await communityQuery(jid, 'set', [{ tag: setting, attrs: {} }])
|
||||
},
|
||||
communityMemberAddMode: async (jid: string, mode: 'admin_add' | 'all_member_add') => {
|
||||
await communityQuery(jid, 'set', [{ tag: 'member_add_mode', attrs: {}, content: mode }])
|
||||
},
|
||||
communityJoinApprovalMode: async (jid: string, mode: 'on' | 'off') => {
|
||||
await communityQuery(jid, 'set', [
|
||||
{ tag: 'membership_approval_mode', attrs: {}, content: [{ tag: 'community_join', attrs: { state: mode } }] }
|
||||
])
|
||||
},
|
||||
communityFetchAllParticipating
|
||||
}
|
||||
}
|
||||
|
||||
export const extractCommunityMetadata = (result: BinaryNode) => {
|
||||
const community = getBinaryNodeChild(result, 'community')!
|
||||
const descChild = getBinaryNodeChild(community, 'description')
|
||||
let desc: string | undefined
|
||||
let descId: string | undefined
|
||||
if (descChild) {
|
||||
desc = getBinaryNodeChildString(descChild, 'body')
|
||||
descId = descChild.attrs.id
|
||||
}
|
||||
|
||||
const communityId = community.attrs.id?.includes('@')
|
||||
? community.attrs.id
|
||||
: jidEncode(community.attrs.id || '', 'g.us')
|
||||
const eph = getBinaryNodeChild(community, 'ephemeral')?.attrs.expiration
|
||||
const memberAddMode = getBinaryNodeChildString(community, 'member_add_mode') === 'all_member_add'
|
||||
const metadata: GroupMetadata = {
|
||||
id: communityId,
|
||||
subject: community.attrs.subject || '',
|
||||
subjectOwner: community.attrs.s_o,
|
||||
subjectTime: Number(community.attrs.s_t || 0),
|
||||
size: getBinaryNodeChildren(community, 'participant').length,
|
||||
creation: Number(community.attrs.creation || 0),
|
||||
owner: community.attrs.creator ? jidNormalizedUser(community.attrs.creator) : undefined,
|
||||
desc,
|
||||
descId,
|
||||
linkedParent: getBinaryNodeChild(community, 'linked_parent')?.attrs.jid || undefined,
|
||||
restrict: !!getBinaryNodeChild(community, 'locked'),
|
||||
announce: !!getBinaryNodeChild(community, 'announcement'),
|
||||
isCommunity: !!getBinaryNodeChild(community, 'parent'),
|
||||
isCommunityAnnounce: !!getBinaryNodeChild(community, 'default_sub_community'),
|
||||
joinApprovalMode: !!getBinaryNodeChild(community, 'membership_approval_mode'),
|
||||
memberAddMode,
|
||||
participants: getBinaryNodeChildren(community, 'participant').map(({ attrs }) => {
|
||||
return {
|
||||
id: attrs.jid!,
|
||||
admin: (attrs.type || null) as GroupParticipant['admin']
|
||||
}
|
||||
}),
|
||||
ephemeralDuration: eph ? +eph : undefined,
|
||||
addressingMode: getBinaryNodeChildString(community, 'addressing_mode')! as GroupMetadata['addressingMode']
|
||||
}
|
||||
return metadata
|
||||
}
|
||||
+2
-2
@@ -1,10 +1,10 @@
|
||||
import { DEFAULT_CONNECTION_CONFIG } from '../Defaults'
|
||||
import type { UserFacingSocketConfig } from '../Types'
|
||||
import { makeBusinessSocket } from './business'
|
||||
import { makeCommunitiesSocket } from './communities'
|
||||
|
||||
// export the last socket layer
|
||||
const makeWASocket = (config: UserFacingSocketConfig) =>
|
||||
makeBusinessSocket({
|
||||
makeCommunitiesSocket({
|
||||
...DEFAULT_CONNECTION_CONFIG,
|
||||
...config
|
||||
})
|
||||
|
||||
Reference in New Issue
Block a user