@alannxd/baileys 6.0.6 → 6.0.9

package/lib/Socket/communities.js

@@ -1,13 +1,12 @@
-const WAProto_1 = require('../../WAProto');
-const Types_1 = require('../Types');
-const Utils_1 = require('../Utils');
-const WABinary_1 = require('../WABinary');
-const newsletter_1 = require('./newsletter');
-
-const makeCommunitiesSocket = (config) => {
-    const sock = (0, newsletter_1.makeNewsletterSocket)(config);
+import { proto } from '../../WAProto/index.js';
+import { WAMessageStubType } from '../Types/index.js';
+import { generateMessageID, generateMessageIDV2, unixTimestampSeconds } from '../Utils/index.js';
+import logger from '../Utils/logger.js';
+import { getBinaryNodeChild, getBinaryNodeChildren, getBinaryNodeChildString, jidEncode, jidNormalizedUser } from '../WABinary/index.js';
+import { makeBusinessSocket } from './business.js';
+export const makeCommunitiesSocket = (config) => {
+    const sock = makeBusinessSocket(config);
     const { authState, ev, query, upsertMessage } = sock;
-
     const communityQuery = async (jid, type, content) => query({
         tag: 'iq',
         attrs: {
@@ -17,12 +16,10 @@ const makeCommunitiesSocket = (config) => {
         },
         content
     });
-
     const communityMetadata = async (jid) => {
         const result = await communityQuery(jid, 'get', [{ tag: 'query', attrs: { request: 'interactive' } }]);
         return extractCommunityMetadata(result);
     };
-
     const communityFetchAllParticipating = async () => {
         const result = await query({
             tag: 'iq',
@@ -43,9 +40,9 @@ const makeCommunitiesSocket = (config) => {
             ]
         });
         const data = {};
-        const communitiesChild = WABinary_1.getBinaryNodeChild(result, 'communities');
+        const communitiesChild = getBinaryNodeChild(result, 'communities');
         if (communitiesChild) {
-            const communities = WABinary_1.getBinaryNodeChildren(communitiesChild, 'community');
+            const communities = getBinaryNodeChildren(communitiesChild, 'community');
             for (const communityNode of communities) {
                 const meta = extractCommunityMetadata({
                     tag: 'result',
@@ -58,15 +55,14 @@ const makeCommunitiesSocket = (config) => {
         sock.ev.emit('groups.update', Object.values(data));
         return data;
     };
-
     async function parseGroupResult(node) {
-        Utils_1.logger.info({ node }, 'parseGroupResult');
-        const groupNode = WABinary_1.getBinaryNodeChild(node, 'group');
+        logger.info({ node }, 'parseGroupResult');
+        const groupNode = getBinaryNodeChild(node, 'group');
         if (groupNode) {
             try {
-                Utils_1.logger.info({ groupNode }, 'groupNode');
+                logger.info({ groupNode }, 'groupNode');
                 const metadata = await sock.groupMetadata(`${groupNode.attrs.id}@g.us`);
-                return metadata? metadata : Optional.empty();
+                return metadata ? metadata : Optional.empty();
             }
             catch (error) {
                 console.error('Error parsing group metadata:', error);
@@ -75,26 +71,23 @@ const makeCommunitiesSocket = (config) => {
         }
         return Optional.empty();
     }
-
     const Optional = {
         empty: () => null,
-        of: (value) => (value!== null? { value } : null)
+        of: (value) => (value !== null ? { value } : null)
     };
-
     sock.ws.on('CB:ib,,dirty', async (node) => {
-        const { attrs } = WABinary_1.getBinaryNodeChild(node, 'dirty');
-        if (attrs.type!== 'communities') {
+        const { attrs } = getBinaryNodeChild(node, 'dirty');
+        if (attrs.type !== 'communities') {
             return;
         }
         await communityFetchAllParticipating();
         await sock.cleanDirtyBits('groups');
     });
-
     return {
-       ...sock,
+        ...sock,
         communityMetadata,
         communityCreate: async (subject, body) => {
-            const descriptionId = Utils_1.generateMessageID().substring(0, 12);
+            const descriptionId = generateMessageID().substring(0, 12);
             const result = await communityQuery('@g.us', 'set', [
                 {
                     tag: 'create',
@@ -129,7 +122,7 @@ const makeCommunitiesSocket = (config) => {
             return await parseGroupResult(result);
         },
         communityCreateGroup: async (subject, participants, parentCommunityJid) => {
-            const key = Utils_1.generateMessageIDV2();
+            const key = generateMessageIDV2();
             const result = await communityQuery('@g.us', 'set', [
                 {
                     tag: 'create',
@@ -138,7 +131,7 @@ const makeCommunitiesSocket = (config) => {
                         key
                     },
                     content: [
-                       ...participants.map(jid => ({
+                        ...participants.map(jid => ({
                             tag: 'participant',
                             attrs: { jid }
                         })),
@@ -193,25 +186,29 @@ 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 sock.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 = WABinary_1.getBinaryNodeChild(result, 'sub_groups');
+            const subGroupsNode = getBinaryNodeChild(result, 'sub_groups');
             if (subGroupsNode) {
-                const groupNodes = WABinary_1.getBinaryNodeChildren(subGroupsNode, 'group');
+                const groupNodes = getBinaryNodeChildren(subGroupsNode, 'group');
                 for (const groupNode of groupNodes) {
                     linkedGroupsData.push({
-                        id: groupNode.attrs.id? WABinary_1.jidEncode(groupNode.attrs.id, 'g.us') : undefined,
+                        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? WABinary_1.jidNormalizedUser(groupNode.attrs.creator) : undefined,
-                        size: groupNode.attrs.size? Number(groupNode.attrs.size) : undefined
+                        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
                     });
                 }
             }
@@ -228,8 +225,8 @@ const makeCommunitiesSocket = (config) => {
                     attrs: {}
                 }
             ]);
-            const node = WABinary_1.getBinaryNodeChild(result, 'membership_approval_requests');
-            const participants = WABinary_1.getBinaryNodeChildren(node, 'membership_approval_request');
+            const node = getBinaryNodeChild(result, 'membership_approval_requests');
+            const participants = getBinaryNodeChildren(node, 'membership_approval_request');
             return participants.map(v => v.attrs);
         },
         communityRequestParticipantsUpdate: async (jid, participants, action) => {
@@ -249,9 +246,9 @@ const makeCommunitiesSocket = (config) => {
                     ]
                 }
             ]);
-            const node = WABinary_1.getBinaryNodeChild(result, 'membership_requests_action');
-            const nodeAction = WABinary_1.getBinaryNodeChild(node, action);
-            const participantsAffected = WABinary_1.getBinaryNodeChildren(nodeAction, 'participant');
+            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 };
             });
@@ -260,56 +257,67 @@ const makeCommunitiesSocket = (config) => {
             const result = await communityQuery(jid, 'set', [
                 {
                     tag: action,
-                    attrs: action === 'remove'? { linked_groups: 'true' } : {},
+                    attrs: action === 'remove' ? { linked_groups: 'true' } : {},
                     content: participants.map(jid => ({
                         tag: 'participant',
                         attrs: { jid }
                     }))
                 }
             ]);
-            const node = WABinary_1.getBinaryNodeChild(result, action);
-            const participantsAffected = WABinary_1.getBinaryNodeChildren(node, 'participant');
+            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, description) => {
             const metadata = await communityMetadata(jid);
-            const prev = metadata.descId?? null;
+            const prev = metadata.descId ?? null;
             await communityQuery(jid, 'set', [
                 {
                     tag: 'description',
                     attrs: {
-                       ...(description? { id: Utils_1.generateMessageID() } : { delete: 'true' }),
-                       ...(prev? { prev } : {})
+                        ...(description ? { id: generateMessageID() } : { delete: 'true' }),
+                        ...(prev ? { prev } : {})
                     },
-                    content: description? [{ tag: 'body', attrs: {}, content: Buffer.from(description, 'utf-8') }] : undefined
+                    content: description ? [{ tag: 'body', attrs: {}, content: Buffer.from(description, 'utf-8') }] : undefined
                 }
             ]);
         },
         communityInviteCode: async (jid) => {
             const result = await communityQuery(jid, 'get', [{ tag: 'invite', attrs: {} }]);
-            const inviteNode = WABinary_1.getBinaryNodeChild(result, 'invite');
+            const inviteNode = getBinaryNodeChild(result, 'invite');
             return inviteNode?.attrs.code;
         },
         communityRevokeInvite: async (jid) => {
             const result = await communityQuery(jid, 'set', [{ tag: 'invite', attrs: {} }]);
-            const inviteNode = WABinary_1.getBinaryNodeChild(result, 'invite');
+            const inviteNode = getBinaryNodeChild(result, 'invite');
             return inviteNode?.attrs.code;
         },
         communityAcceptInvite: async (code) => {
             const results = await communityQuery('@g.us', 'set', [{ tag: 'invite', attrs: { code } }]);
-            const result = WABinary_1.getBinaryNodeChild(results, 'community');
+            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, invitedJid) => {
             const result = await communityQuery(communityJid, 'set', [
                 { tag: 'revoke', attrs: {}, content: [{ tag: 'participant', attrs: { jid: invitedJid } }] }
             ]);
-            return!!result;
+            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, inviteMessage) => {
-            key = typeof key === 'string'? { remoteJid: key } : key;
+            key = typeof key === 'string' ? { remoteJid: key } : key;
             const results = await communityQuery(inviteMessage.groupJid, 'set', [
                 {
                     tag: 'accept',
@@ -320,8 +328,11 @@ const makeCommunitiesSocket = (config) => {
                     }
                 }
             ]);
+            // if we have the full message key
+            // update the invite message to be expired
             if (key.id) {
-                inviteMessage = WAProto_1.proto.Message.GroupInviteMessage.fromObject(inviteMessage);
+                // create new invite message that is expired
+                inviteMessage = proto.Message.GroupInviteMessage.fromObject(inviteMessage);
                 inviteMessage.inviteExpiration = 0;
                 inviteMessage.inviteCode = '';
                 ev.emit('messages.update', [
@@ -335,17 +346,18 @@ const makeCommunitiesSocket = (config) => {
                     }
                 ]);
             }
+            // generate the community add message
             await upsertMessage({
                 key: {
                     remoteJid: inviteMessage.groupJid,
-                    id: Utils_1.generateMessageIDV2(sock.user?.id),
+                    id: generateMessageIDV2(sock.user?.id),
                     fromMe: false,
-                    participant: key.remoteJid
+                    participant: key.remoteJid // TODO: investigate if this makes any sense at all
                 },
-                messageStubType: Types_1.WAMessageStubType.GROUP_PARTICIPANT_ADD,
+                messageStubType: WAMessageStubType.GROUP_PARTICIPANT_ADD,
                 messageStubParameters: [JSON.stringify(authState.creds.me)],
                 participant: key.remoteJid,
-                messageTimestamp: Utils_1.unixTimestampSeconds()
+                messageTimestamp: unixTimestampSeconds()
             }, 'notify');
             return results.attrs.from;
         }),
@@ -355,7 +367,7 @@ const makeCommunitiesSocket = (config) => {
         },
         communityToggleEphemeral: async (jid, ephemeralExpiration) => {
             const content = ephemeralExpiration
-               ? { tag: 'ephemeral', attrs: { expiration: ephemeralExpiration.toString() } }
+                ? { tag: 'ephemeral', attrs: { expiration: ephemeralExpiration.toString() } }
                 : { tag: 'not_ephemeral', attrs: {} };
             await communityQuery(jid, 'set', [content]);
         },
@@ -373,49 +385,47 @@ const makeCommunitiesSocket = (config) => {
         communityFetchAllParticipating
     };
 };
-
-const extractCommunityMetadata = (result) => {
-    const community = WABinary_1.getBinaryNodeChild(result, 'community');
-    const descChild = WABinary_1.getBinaryNodeChild(community, 'description');
+export const extractCommunityMetadata = (result) => {
+    const community = getBinaryNodeChild(result, 'community');
+    const descChild = getBinaryNodeChild(community, 'description');
     let desc;
     let descId;
     if (descChild) {
-        desc = WABinary_1.getBinaryNodeChildString(descChild, 'body');
+        desc = getBinaryNodeChildString(descChild, 'body');
         descId = descChild.attrs.id;
     }
     const communityId = community.attrs.id?.includes('@')
-       ? community.attrs.id
-        : WABinary_1.jidEncode(community.attrs.id || '', 'g.us');
-    const eph = WABinary_1.getBinaryNodeChild(community, 'ephemeral')?.attrs.expiration;
-    const memberAddMode = WABinary_1.getBinaryNodeChildString(community, 'member_add_mode') === 'all_member_add';
+        ? 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 = {
         id: communityId,
         subject: community.attrs.subject || '',
         subjectOwner: community.attrs.s_o,
         subjectTime: Number(community.attrs.s_t || 0),
-        size: WABinary_1.getBinaryNodeChildren(community, 'participant').length,
+        size: getBinaryNodeChildren(community, 'participant').length,
         creation: Number(community.attrs.creation || 0),
-        owner: community.attrs.creator? WABinary_1.jidNormalizedUser(community.attrs.creator) : undefined,
+        owner: community.attrs.creator ? jidNormalizedUser(community.attrs.creator) : undefined,
         desc,
         descId,
-        linkedParent: WABinary_1.getBinaryNodeChild(community, 'linked_parent')?.attrs.jid || undefined,
-        restrict:!!WABinary_1.getBinaryNodeChild(community, 'locked'),
-        announce:!!WABinary_1.getBinaryNodeChild(community, 'announcement'),
-        isCommunity:!!WABinary_1.getBinaryNodeChild(community, 'parent'),
-        isCommunityAnnounce:!!WABinary_1.getBinaryNodeChild(community, 'default_sub_community'),
-        joinApprovalMode:!!WABinary_1.getBinaryNodeChild(community, 'membership_approval_mode'),
+        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: WABinary_1.getBinaryNodeChildren(community, 'participant').map(({ attrs }) => {
+        participants: getBinaryNodeChildren(community, 'participant').map(({ attrs }) => {
             return {
+                // TODO: IMPLEMENT THE PN/LID FIELDS HERE!!
                 id: attrs.jid,
                 admin: (attrs.type || null)
             };
         }),
-        ephemeralDuration: eph? +eph : undefined,
-        addressingMode: WABinary_1.getBinaryNodeChildString(community, 'addressing_mode')
+        ephemeralDuration: eph ? +eph : undefined,
+        addressingMode: getBinaryNodeChildString(community, 'addressing_mode')
     };
     return metadata;
 };
-
-exports.makeCommunitiesSocket = makeCommunitiesSocket;
-exports.extractCommunityMetadata = extractCommunityMetadata;
+//# sourceMappingURL=communities.js.map