@areumtecnologia/wasockets 1.1.5 → 2.0.0-alpha.1

package/lib/Socket/community.js

@@ -119,6 +119,76 @@ const makeCommunitiesSocket = (config) => {
     	await cleanDirtyBits('groups') 
     }) 
     
+    const communityRequestParticipantsUpdate = async (jid, participants, action) => {
+        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 }
+        })
+    }
+
+    const communityFetchLinkedGroups = async (jid) => {
+        let communityJid = jid
+        let isCommunity = false
+        
+        // Try to determine if it is a subgroup or a community
+        const metadata = await groupMetadata(jid)
+        
+        if (metadata.linkedParent) {
+            // It is a subgroup, get the community jid
+            communityJid = metadata.linkedParent
+        } else {
+            // It is a community
+            isCommunity = true
+        }
+        
+        // Fetch all subgroups of the community
+        const result = await communityQuery(communityJid, 'get', [{ tag: 'sub_groups', attrs: {} }])
+        const linkedGroupsData = []
+        const subGroupsNode = getBinaryNodeChild(result, 'sub_groups')
+        
+        if (subGroupsNode) {
+            const groupNodes = getBinaryNodeChildren(subGroupsNode, 'group')
+            
+            for (const groupNode of groupNodes) {
+                const isDefault = groupNode.attrs.default_sub_group === 'true' || !!getBinaryNodeChild(groupNode, 'default_sub_group')
+                linkedGroupsData.push({
+                    id: groupNode.attrs.id ? jidEncode(groupNode.attrs.id, 'g.us') : undefined,
+                    subject: groupNode.attrs.subject || '',
+                    creation: groupNode.attrs.creation ? Number(groupNode.attrs.creation) : undefined,
+                    owner: groupNode.attrs.creator ? jidNormalizedUser(groupNode.attrs.creator) : undefined,
+                    size: groupNode.attrs.size ? Number(groupNode.attrs.size) : undefined,
+                    isDefaultSubGroup: isDefault
+                })
+            }
+        }
+        
+        return {
+            communityJid,
+            isCommunity,
+            linkedGroups: linkedGroupsData
+        }
+    }
+
     return {
         ...suki,
         communityQuery, 
@@ -225,46 +295,15 @@ const makeCommunitiesSocket = (config) => {
                 }
             ])
         },
-        communityFetchLinkedGroups: async (jid) => {
-            let communityJid = jid
-            let isCommunity = false
-            
-            // Try to determine if it is a subgroup or a community
-            const metadata = await groupMetadata(jid)
-            
-            if (metadata.linkedParent) {
-                // It is a subgroup, get the community jid
-                communityJid = metadata.linkedParent
-            }
-            
-            else {
-                // It is a community
-                isCommunity = true
-            }
-            
-            // Fetch all subgroups of the community
-            const result = await communityQuery(communityJid, 'get', [{ tag: 'sub_groups', attrs: {} }])
-            const linkedGroupsData = []
-            const subGroupsNode = getBinaryNodeChild(result, 'sub_groups')
-            
-            if (subGroupsNode) {
-                const groupNodes = getBinaryNodeChildren(subGroupsNode, 'group')
-                
-                for (const groupNode of groupNodes) {
-                    linkedGroupsData.push({
-                        id: groupNode.attrs.id ? jidEncode(groupNode.attrs.id, 'g.us') : undefined,
-                        subject: groupNode.attrs.subject || '',
-                        creation: groupNode.attrs.creation ? Number(groupNode.attrs.creation) : undefined,
-                        owner: groupNode.attrs.creator ? jidNormalizedUser(groupNode.attrs.creator) : undefined,
-                        size: groupNode.attrs.size ? Number(groupNode.attrs.size) : undefined
-                    })
-                }
-            }
-            
+        communityFetchLinkedGroups,
+        communityFetchSubGroups: async (jid) => {
+            const data = await communityFetchLinkedGroups(jid)
+            const defaultSubGroup = data.linkedGroups.find(g => g.isDefaultSubGroup) || null
+            const subGroups = data.linkedGroups.filter(g => !g.isDefaultSubGroup)
             return {
-                communityJid,
-                isCommunity,
-                linkedGroups: linkedGroupsData
+                ...data,
+                defaultSubGroup,
+                subGroups
             }
         },
         communityRequestParticipantsList: async (jid) => {
@@ -280,32 +319,7 @@ const makeCommunitiesSocket = (config) => {
             
             return participants.map(v => v.attrs)
         },
-        communityRequestParticipantsUpdate: async (jid, participants, action) => {
-            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 }
-            })
-        },
+        communityRequestParticipantsUpdate,
         communityParticipantsUpdate: async (jid, participants, action) => {
             const result = await communityQuery(jid, 'set', [
                 {
@@ -442,10 +456,53 @@ const makeCommunitiesSocket = (config) => {
             await communityQuery(jid, 'set', [{ tag: 'member_add_mode', attrs: {}, content: mode }])
         },
         communityJoinApprovalMode: async (jid, mode) => {
+            const state = (mode === true || mode === 'on') ? 'on' : (mode === false || mode === 'off' ? 'off' : mode)
             await communityQuery(jid, 'set', [
-                { tag: 'membership_approval_mode', attrs: {}, content: [{ tag: 'community_join', attrs: { state: mode } }] }
+                { tag: 'membership_approval_mode', attrs: {}, content: [{ tag: 'community_join', attrs: { state } }] }
             ])
         },
+        communityMembershipApprovalMode: async (jid, mode) => {
+            const state = (mode === true || mode === 'on') ? 'on' : 'off'
+            await communityQuery(jid, 'set', [
+                { tag: 'membership_approval_mode', attrs: {}, content: [{ tag: 'community_join', attrs: { state } }] }
+            ])
+        },
+        communityApprovePendingParticipants: async (jid, participants) => {
+            const list = Array.isArray(participants) ? participants : [participants]
+            return communityRequestParticipantsUpdate(jid, list, 'approve')
+        },
+        communityRejectPendingParticipants: async (jid, participants) => {
+            const list = Array.isArray(participants) ? participants : [participants]
+            return communityRequestParticipantsUpdate(jid, list, 'reject')
+        },
+        communityUpdatePermissions: async (jid, policies = {}) => {
+            const results = {}
+            if (typeof policies.announce === 'boolean') {
+                await communityQuery(jid, 'set', [{ tag: policies.announce ? 'announcement' : 'not_announcement', attrs: {} }])
+                results.announce = policies.announce
+            }
+            if (typeof policies.restrict === 'boolean') {
+                await communityQuery(jid, 'set', [{ tag: policies.restrict ? 'locked' : 'unlocked', attrs: {} }])
+                results.restrict = policies.restrict
+            }
+            if (policies.memberAddMode) {
+                await communityQuery(jid, 'set', [{ tag: 'member_add_mode', attrs: {}, content: policies.memberAddMode }])
+                results.memberAddMode = policies.memberAddMode
+            }
+            if (typeof policies.approvalMode !== 'undefined') {
+                const state = (policies.approvalMode === true || policies.approvalMode === 'on') ? 'on' : 'off'
+                await communityQuery(jid, 'set', [{ tag: 'membership_approval_mode', attrs: {}, content: [{ tag: 'community_join', attrs: { state } }] }])
+                results.approvalMode = state
+            }
+            if (typeof policies.ephemeral !== 'undefined') {
+                const content = policies.ephemeral ?
+                    { tag: 'ephemeral', attrs: { expiration: policies.ephemeral.toString() } } :
+                    { tag: 'not_ephemeral', attrs: {} }
+                await communityQuery(jid, 'set', [content])
+                results.ephemeral = policies.ephemeral
+            }
+            return results
+        },
         communityFetchAllParticipating
     }
 }
@@ -489,12 +546,12 @@ const extractCommunityMetadata = (result) => {
         memberAddMode,
         participants: getBinaryNodeChildren(community, 'participant').map(({ attrs }) => {
             return {
-                id: mode === WAMessageAddressingMode.LID ? community.phone_number : attrs.jid,
-                lid: mode === WAMessageAddressingMode.LID ? community.jid : attrs.lid, 
+                id: mode === WAMessageAddressingMode.LID ? attrs.phone_number : attrs.jid,
+                lid: mode === WAMessageAddressingMode.LID ? attrs.jid : attrs.lid, 
                 admin: (attrs.type || null),
             }
         }),
-        ephemeralDuration: eph ? Number(ph) : undefined, 
+        ephemeralDuration: eph ? Number(eph) : undefined, 
         addressingMode: mode
     }
     

package/lib/Socket/groups.js

@@ -40,8 +40,16 @@ const makeGroupsSocket = (config) => {
     }))
     
     const groupMetadata = async (jid) => {
+        if (config.cachedGroupMetadata) {
+            const cached = await config.cachedGroupMetadata(jid)
+            if (cached) return cached
+        }
         const result = await groupQuery(jid, 'get', [{ tag: 'query', attrs: { request: 'interactive' } }])
-        return extractGroupMetadata(result)
+        const metadata = extractGroupMetadata(result)
+        if (config.cachedGroupMetadata?.set && metadata) {
+            config.cachedGroupMetadata.set(jid, metadata)
+        }
+        return metadata
     }
     
     const groupFetchAllParticipating = async () => {
@@ -79,6 +87,14 @@ const makeGroupsSocket = (config) => {
             }
         }
         
+        if (config.cachedGroupMetadata?.set) {
+            for (const meta of Object.values(data)) {
+                if (meta?.id) {
+                    config.cachedGroupMetadata.set(meta.id, meta)
+                }
+            }
+        }
+
         suki.ev.emit('groups.update', Object.values(data))
         return data
     }
@@ -93,6 +109,31 @@ const makeGroupsSocket = (config) => {
         await cleanDirtyBits('groups')
     })
     
+    const groupRequestParticipantsUpdate = async (jid, participants, action) => {
+        const result = await groupQuery(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 }
+        })
+    }
+
     return {
         ...suki,
         groupQuery, 
@@ -117,6 +158,7 @@ const makeGroupsSocket = (config) => {
             return extractGroupMetadata(result)
         },
         groupLeave: async (id) => {
+            config.cachedGroupMetadata?.delete?.(id)
             await groupQuery('@g.us', 'set', [
                 {
                     tag: 'leave',
@@ -128,6 +170,7 @@ const makeGroupsSocket = (config) => {
             ])
         },
         groupUpdateSubject: async (jid, subject) => {
+            config.cachedGroupMetadata?.delete?.(jid)
             await groupQuery(jid, 'set', [
                 {
                     tag: 'subject',
@@ -149,30 +192,7 @@ const makeGroupsSocket = (config) => {
             
             return participants.map(v => v.attrs)
         },
-        groupRequestParticipantsUpdate: async (jid, participants, action) => {
-            const result = await groupQuery(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 }
-            })
-        },
+        groupRequestParticipantsUpdate,
         groupParticipantsUpdate: async (jid, participants, action) => {
             const result = await groupQuery(jid, 'set', [
                 {
@@ -187,11 +207,13 @@ const makeGroupsSocket = (config) => {
             const node = getBinaryNodeChild(result, action)
             const participantsAffected = getBinaryNodeChildren(node, 'participant')
             
+            config.cachedGroupMetadata?.delete?.(jid)
             return participantsAffected.map(p => {
                 return { status: p.attrs.error || '200', jid: p.attrs.jid, content: p }
             })
         },
         groupUpdateDescription: async (jid, description) => {
+            config.cachedGroupMetadata?.delete?.(jid)
             const metadata = await groupMetadata(jid)
             const prev = metadata.descId ? metadata.descId : null
             
@@ -300,13 +322,68 @@ const makeGroupsSocket = (config) => {
             await groupQuery(jid, 'set', [content])
         },
         groupSettingUpdate: async (jid, setting) => {
+            config.cachedGroupMetadata?.delete?.(jid)
             await groupQuery(jid, 'set', [{ tag: setting, attrs: {} }])
         },
         groupMemberAddMode: async (jid, mode) => {
+            config.cachedGroupMetadata?.delete?.(jid)
             await groupQuery(jid, 'set', [{ tag: 'member_add_mode', attrs: {}, content: mode }])
         },
         groupJoinApprovalMode: async (jid, mode) => {
-            await groupQuery(jid, 'set', [{ tag: 'membership_approval_mode', attrs: {}, content: [{ tag: 'group_join', attrs: { state: mode } }] }])
+            config.cachedGroupMetadata?.delete?.(jid)
+            const state = (mode === true || mode === 'on') ? 'on' : (mode === false || mode === 'off' ? 'off' : mode)
+            await groupQuery(jid, 'set', [{ tag: 'membership_approval_mode', attrs: {}, content: [{ tag: 'group_join', attrs: { state } }] }])
+        },
+        groupMembershipApprovalMode: async (jid, mode) => {
+            const state = (mode === true || mode === 'on') ? 'on' : 'off'
+            await groupQuery(jid, 'set', [{ tag: 'membership_approval_mode', attrs: {}, content: [{ tag: 'group_join', attrs: { state } }] }])
+        },
+        groupApprovePendingParticipants: async (jid, participants) => {
+            const list = Array.isArray(participants) ? participants : [participants]
+            return groupRequestParticipantsUpdate(jid, list, 'approve')
+        },
+        groupRejectPendingParticipants: async (jid, participants) => {
+            const list = Array.isArray(participants) ? participants : [participants]
+            return groupRequestParticipantsUpdate(jid, list, 'reject')
+        },
+        groupUpdatePermissions: async (jid, policies = {}) => {
+            config.cachedGroupMetadata?.delete?.(jid)
+            const results = {}
+            if (typeof policies.announce === 'boolean') {
+                await groupQuery(jid, 'set', [{ tag: policies.announce ? 'announcement' : 'not_announcement', attrs: {} }])
+                results.announce = policies.announce
+            }
+            if (typeof policies.restrict === 'boolean') {
+                await groupQuery(jid, 'set', [{ tag: policies.restrict ? 'locked' : 'unlocked', attrs: {} }])
+                results.restrict = policies.restrict
+            }
+            if (policies.memberAddMode) {
+                await groupQuery(jid, 'set', [{ tag: 'member_add_mode', attrs: {}, content: policies.memberAddMode }])
+                results.memberAddMode = policies.memberAddMode
+            }
+            if (typeof policies.approvalMode !== 'undefined') {
+                const state = (policies.approvalMode === true || policies.approvalMode === 'on') ? 'on' : 'off'
+                await groupQuery(jid, 'set', [{ tag: 'membership_approval_mode', attrs: {}, content: [{ tag: 'group_join', attrs: { state } }] }])
+                results.approvalMode = state
+            }
+            if (typeof policies.ephemeral !== 'undefined') {
+                const content = policies.ephemeral ?
+                    { tag: 'ephemeral', attrs: { expiration: policies.ephemeral.toString() } } :
+                    { tag: 'not_ephemeral', attrs: {} }
+                await groupQuery(jid, 'set', [content])
+                results.ephemeral = policies.ephemeral
+            }
+            return results
+        },
+        groupGetAdminJids: async (jid) => {
+            const metadata = await groupMetadata(jid)
+            return (metadata.participants || [])
+                .filter(p => !!p.admin)
+                .map(p => p.id)
+        },
+        groupGetParticipants: async (jid) => {
+            const metadata = await groupMetadata(jid)
+            return metadata.participants || []
         },
         groupFetchAllParticipating
     }

package/lib/Socket/index.js

@@ -2,11 +2,12 @@
 
 Object.defineProperty(exports, "__esModule", { value: true })
 
-const { DEFAULT_CONNECTION_CONFIG } = require("../Defaults/connection")
+const { DEFAULT_CONNECTION_CONFIG, createGroupMetadataCache } = require("../Defaults/connection")
 const { makeCommunitiesSocket } = require("./community")
+const { PluginManager } = require("../Plugins")
 
 // export the last socket layer
-const makeWASocket = (config) => {
+const makeWASocket = (config = {}) => {
 	const newConfig = {
     	...DEFAULT_CONNECTION_CONFIG,
    	 ...config
@@ -18,7 +19,30 @@ const makeWASocket = (config) => {
         newConfig.shouldSyncHistoryMessage = () => !!newConfig.syncFullHistory
     }
 
-    return makeCommunitiesSocket(newConfig)
+    if (!config.cachedGroupMetadata && newConfig.enableGroupCache) {
+        newConfig.cachedGroupMetadata = createGroupMetadataCache()
+    }
+
+    const sock = makeCommunitiesSocket(newConfig)
+    const pluginManager = new PluginManager(sock)
+
+    sock.plugins = pluginManager
+    sock.use = (plugin, options) => {
+        pluginManager.use(plugin, options)
+        return sock
+    }
+    sock.registerMiddleware = (hook, fn) => pluginManager.registerMiddleware(hook, fn)
+    sock.hasPlugin = (name) => pluginManager.hasPlugin(name)
+    sock.listPlugins = () => pluginManager.listPlugins()
+
+    // Auto-install plugins passed in configuration
+    if (Array.isArray(config.plugins)) {
+        for (const p of config.plugins) {
+            sock.use(p)
+        }
+    }
+
+    return sock
 }
 
 exports.default = makeWASocket

package/lib/Socket/messages-recv.js

@@ -302,6 +302,46 @@ const makeMessagesRecvSocket = (config) => {
         await query(stanza)
     }
     
+    const acceptCall = async (callId, callFrom) => {
+        const stanza = ({
+            tag: 'call',
+            attrs: {
+                from: authState.creds.me.id,
+                to: callFrom,
+            },
+            content: [{
+                    tag: 'accept',
+                    attrs: {
+                        'call-id': callId,
+                        'call-creator': callFrom,
+                    },
+                    content: undefined,
+                }],
+        })
+        
+        await query(stanza)
+    }
+
+    const terminateCall = async (callId, callFrom) => {
+        const stanza = ({
+            tag: 'call',
+            attrs: {
+                from: authState.creds.me.id,
+                to: callFrom,
+            },
+            content: [{
+                    tag: 'terminate',
+                    attrs: {
+                        'call-id': callId,
+                        'call-creator': callFrom,
+                    },
+                    content: undefined,
+                }],
+        })
+        
+        await query(stanza)
+    }
+    
     const sendRetryRequest = async (node, forceIncludeKeys = false) => {
         const { fullMessage } = decodeMessageNode(node, authState.creds.me.id, authState.creds.me.lid || '')
         const { key: msgKey } = fullMessage
@@ -1633,6 +1673,8 @@ const makeMessagesRecvSocket = (config) => {
         sendRetryRequest,
         offerCall, 
         rejectCall,
+        acceptCall,
+        terminateCall,
         fetchMessageHistory,
         requestPlaceholderResend,
         messageRetryManager