@prototypevip/baileys 0.0.1

package/lib/Socket/groups.js

@@ -0,0 +1,465 @@
+import { proto } from "../../WAProto/index.js";
+import { WAMessageAddressingMode, WAMessageStubType } from "../Types/index.js";
+import { generateMessageIDV2, unixTimestampSeconds } from "../Utils/index.js";
+import {
+  getBinaryNodeChild,
+  getBinaryNodeChildren,
+  getBinaryNodeChildString,
+  isLidUser,
+  isPnUser,
+  jidEncode,
+  jidNormalizedUser,
+} from "../WABinary/index.js";
+import { makeChatsSocket } from "./chats.js";
+import { groupMetadataCache as defaultCache } from "../Store/cache-factory.js";
+
+export const makeGroupsSocket = (config) => {
+  const cache = config.groupMetadataCache || defaultCache;
+
+  const sock = makeChatsSocket(config);
+  const { authState, ev, query, upsertMessage } = sock;
+  const groupQuery = async (jid, type, content) =>
+    query({
+      tag: "iq",
+      attrs: { type: type, xmlns: "w:g2", to: jid },
+      content: content,
+    });
+  const groupMetadata = async (jid, useCache = true) => {
+    if (useCache) {
+      const cached = cache.get(jid) || config.store?.groups[jid];
+      if (cached && Array.isArray(cached.participants)) {
+        return cached;
+      }
+    }
+    const result = await groupQuery(jid, "get", [
+      { tag: "query", attrs: { request: "interactive" } },
+    ]);
+    const metadata = extractGroupMetadata(result);
+    cache.set(jid, metadata);
+    if (config.store && config.store.groups) {
+      config.store.groups[jid] = metadata;
+    }
+    return metadata;
+  };
+  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 = Object.create(null);
+    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;
+        cache.set(meta.id, meta);
+        if (config.store && config.store.groups) {
+          config.store.groups[meta.id] = meta;
+        }
+        const lidMappings = meta.participants
+          .filter((p) => p.lid && p.pn)
+          .map((p) => ({ lid: p.lid, pn: p.pn }));
+        if (lidMappings.length > 0) {
+          await sock.signalRepository.lidMapping.storeLIDPNMappings(
+            lidMappings,
+          );
+        }
+      }
+    }
+    sock.ev.emit("groups.update", Object.values(data));
+    return data;
+  };
+  sock.ws.on("CB:ib,,dirty", async (node) => {
+    const { attrs } = getBinaryNodeChild(node, "dirty");
+    if (attrs.type !== "groups") {
+      return;
+    }
+    await groupFetchAllParticipating();
+    await sock.cleanDirtyBits("groups");
+  });
+  return {
+    ...sock,
+    extractGroupMetadata: extractGroupMetadata,
+    groupMetadata: groupMetadata,
+    groupCreate: async (subject, participants) => {
+      const key = generateMessageIDV2();
+      const result = await groupQuery("@g.us", "set", [
+        {
+          tag: "create",
+          attrs: { subject: subject, key: key },
+          content: participants.map((jid) => ({
+            tag: "participant",
+            attrs: { jid: jid },
+          })),
+        },
+      ]);
+      return extractGroupMetadata(result);
+    },
+    groupLeave: async (id) => {
+      await groupQuery("@g.us", "set", [
+        {
+          tag: "leave",
+          attrs: {},
+          content: [{ tag: "group", attrs: { id: id } }],
+        },
+      ]);
+    },
+    groupUpdateSubject: async (jid, subject) => {
+      await groupQuery(jid, "set", [
+        {
+          tag: "subject",
+          attrs: {},
+          content: Buffer.from(subject, "utf-8"),
+        },
+      ]);
+      cache.del(jid);
+      if (config.store && config.store.groups) {
+        delete config.store.groups[jid];
+      }
+    },
+    groupRequestParticipantsList: async (jid) => {
+      const result = await groupQuery(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);
+    },
+    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: 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 };
+      });
+    },
+    groupParticipantsUpdate: async (jid, participants, action) => {
+      // PN-First API: Translate PN to LID only if the group's addressingMode requires it
+      const metadata = await groupMetadata(jid);
+      if (
+        metadata &&
+        metadata.addressingMode === "lid" &&
+        sock.signalRepository &&
+        sock.signalRepository.lidMapping
+      ) {
+        const resolvedParticipants = [];
+        for (const p of participants) {
+          if (p.includes("@s.whatsapp.net")) {
+            try {
+              let lid = await sock.signalRepository.lidMapping.getLIDForPN(p);
+              if (!lid && sock.onWhatsApp) {
+                await sock.onWhatsApp(p);
+                lid = await sock.signalRepository.lidMapping.getLIDForPN(p);
+              }
+              resolvedParticipants.push(lid || p);
+            } catch (e) {
+              resolvedParticipants.push(p);
+            }
+          } else {
+            resolvedParticipants.push(p);
+          }
+        }
+        participants = resolvedParticipants;
+      }
+
+      const result = await groupQuery(jid, "set", [
+        {
+          tag: action,
+          attrs: {},
+          content: participants.map((jid) => ({
+            tag: "participant",
+            attrs: { jid: 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,
+        };
+      });
+    },
+    groupUpdateDescription: async (jid, description) => {
+      const metadata = await groupMetadata(jid);
+      const prev = metadata.descId ?? null;
+      await groupQuery(jid, "set", [
+        {
+          tag: "description",
+          attrs: {
+            ...(description
+              ? { id: generateMessageIDV2() }
+              : { delete: "true" }),
+            ...(prev ? { prev: prev } : {}),
+          },
+          content: description
+            ? [
+                {
+                  tag: "body",
+                  attrs: {},
+                  content: Buffer.from(description, "utf-8"),
+                },
+              ]
+            : undefined,
+        },
+      ]);
+      cache.del(jid);
+      if (config.store && config.store.groups) {
+        delete config.store.groups[jid];
+      }
+    },
+    groupInviteCode: async (jid) => {
+      const result = await groupQuery(jid, "get", [
+        { tag: "invite", attrs: {} },
+      ]);
+      const inviteNode = getBinaryNodeChild(result, "invite");
+      return inviteNode?.attrs.code;
+    },
+    groupRevokeInvite: async (jid) => {
+      const result = await groupQuery(jid, "set", [
+        { tag: "invite", attrs: {} },
+      ]);
+      const inviteNode = getBinaryNodeChild(result, "invite");
+      return inviteNode?.attrs.code;
+    },
+    groupAcceptInvite: async (code) => {
+      const results = await groupQuery("@g.us", "set", [
+        { tag: "invite", attrs: { code: code } },
+      ]);
+      const result = getBinaryNodeChild(results, "group");
+      return result?.attrs.jid;
+    },
+    /**
+     * revoke a v4 invite for someone
+     * @param groupJid group jid
+     * @param invitedJid jid of person you invited
+     * @returns true if successful
+     */
+    groupRevokeInviteV4: async (groupJid, invitedJid) => {
+      const result = await groupQuery(groupJid, "set", [
+        {
+          tag: "revoke",
+          attrs: {},
+          content: [{ tag: "participant", attrs: { jid: invitedJid } }],
+        },
+      ]);
+      return !!result;
+    },
+    /**
+     * accept a GroupInviteMessage
+     * @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
+     */
+    groupAcceptInviteV4: ev.createBufferedFunction(
+      async (key, inviteMessage) => {
+        key = typeof key === "string" ? { remoteJid: key } : key;
+        const results = await groupQuery(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: key,
+              update: {
+                message: { groupInviteMessage: inviteMessage },
+              },
+            },
+          ]);
+        }
+        // generate the group add message
+        await upsertMessage(
+          {
+            key: {
+              remoteJid: inviteMessage.groupJid,
+              id: generateMessageIDV2(sock.user?.id),
+              fromMe: false,
+              participant: key.remoteJid,
+            },
+            messageStubType: WAMessageStubType.GROUP_PARTICIPANT_ADD,
+            messageStubParameters: [JSON.stringify(authState.creds.me)],
+            participant: key.remoteJid,
+            messageTimestamp: unixTimestampSeconds(),
+          },
+          "notify",
+        );
+        return results.attrs.from;
+      },
+    ),
+    groupGetInviteInfo: async (code) => {
+      const results = await groupQuery("@g.us", "get", [
+        { tag: "invite", attrs: { code: code } },
+      ]);
+      return extractGroupMetadata(results);
+    },
+    groupToggleEphemeral: async (jid, ephemeralExpiration) => {
+      const content = ephemeralExpiration
+        ? {
+            tag: "ephemeral",
+            attrs: { expiration: ephemeralExpiration.toString() },
+          }
+        : { tag: "not_ephemeral", attrs: {} };
+      await groupQuery(jid, "set", [content]);
+    },
+    groupSettingUpdate: async (jid, setting) => {
+      await groupQuery(jid, "set", [{ tag: setting, attrs: {} }]);
+    },
+    groupMemberAddMode: async (jid, mode) => {
+      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 } }],
+        },
+      ]);
+      cache.del(jid);
+      if (config.store && config.store.groups) {
+        delete config.store.groups[jid];
+      }
+    },
+    groupFetchAllParticipating: groupFetchAllParticipating,
+    groupQuery: groupQuery,
+  };
+};
+export const extractGroupMetadata = (result) => {
+  const group = getBinaryNodeChild(result, "group");
+  const descChild = getBinaryNodeChild(group, "description");
+  let desc;
+  let descId;
+  let descOwner;
+  let descOwnerPn;
+  let descTime;
+  if (descChild) {
+    desc = getBinaryNodeChildString(descChild, "body");
+    descOwner = descChild.attrs.participant
+      ? jidNormalizedUser(descChild.attrs.participant)
+      : undefined;
+    descOwnerPn = descChild.attrs.participant_pn
+      ? jidNormalizedUser(descChild.attrs.participant_pn)
+      : undefined;
+    descTime = +descChild.attrs.t;
+    descId = descChild.attrs.id;
+  }
+  const groupId = group.attrs.id.includes("@")
+    ? group.attrs.id
+    : jidEncode(group.attrs.id, "g.us");
+  const eph = getBinaryNodeChild(group, "ephemeral")?.attrs.expiration;
+  const memberAddMode =
+    getBinaryNodeChildString(group, "member_add_mode") === "all_member_add";
+  const metadata = {
+    id: groupId,
+    notify: group.attrs.notify,
+    addressingMode:
+      group.attrs.addressing_mode === "lid"
+        ? WAMessageAddressingMode.LID
+        : WAMessageAddressingMode.PN,
+    subject: group.attrs.subject,
+    subjectOwner: group.attrs.s_o,
+    subjectOwnerPn: group.attrs.s_o_pn,
+    subjectTime: +group.attrs.s_t,
+    size: group.attrs.size
+      ? +group.attrs.size
+      : getBinaryNodeChildren(group, "participant").length,
+    creation: +group.attrs.creation,
+    owner: group.attrs.creator
+      ? jidNormalizedUser(group.attrs.creator)
+      : undefined,
+    ownerPn: group.attrs.creator_pn
+      ? jidNormalizedUser(group.attrs.creator_pn)
+      : undefined,
+    owner_country_code: group.attrs.creator_country_code,
+    desc: desc,
+    descId: descId,
+    descOwner: descOwner,
+    descOwnerPn: descOwnerPn,
+    descTime: descTime,
+    linkedParent:
+      getBinaryNodeChild(group, "linked_parent")?.attrs.jid || undefined,
+    restrict: !!getBinaryNodeChild(group, "locked"),
+    announce: !!getBinaryNodeChild(group, "announcement"),
+    isCommunity: !!getBinaryNodeChild(group, "parent"),
+    isCommunityAnnounce: !!getBinaryNodeChild(group, "default_sub_group"),
+    joinApprovalMode: !!getBinaryNodeChild(group, "membership_approval_mode"),
+    memberAddMode: memberAddMode,
+    participants: getBinaryNodeChildren(group, "participant").map(
+      ({ attrs }) => {
+        const id = jidNormalizedUser(attrs.jid);
+        const phone = attrs.phone_number
+          ? jidNormalizedUser(attrs.phone_number)
+          : isPnUser(id)
+            ? id
+            : undefined;
+        const lid = attrs.lid
+          ? jidNormalizedUser(attrs.lid)
+          : isLidUser(id)
+            ? id
+            : undefined;
+        return {
+          id: phone,
+          lid: lid,
+          jid: phone,
+          admin: attrs.type || false,
+        };
+      },
+    ),
+    ephemeralDuration: eph ? +eph : undefined,
+  };
+  return metadata;
+};

package/lib/Socket/index.js

@@ -0,0 +1,14 @@
+import { DEFAULT_CONNECTION_CONFIG } from "../Defaults/index.js";
+import { makeCommunitiesSocket } from "./communities.js";
+const makeWASocket = (config) => {
+  delete config.userDevicesCache;
+  const newConfig = { ...DEFAULT_CONNECTION_CONFIG, ...config };
+  let sock;
+  const originalGetMessage = newConfig.getMessage;
+  newConfig.getMessage = async function (key) {
+    return originalGetMessage.call(sock || this, key);
+  };
+  sock = makeCommunitiesSocket(newConfig);
+  return sock;
+};
+export default makeWASocket;