@baileys-md/baileys/lib/WAUSync/USyncQuery.js

Baileys WhatsApp API

//========================================//
 

import { USyncContactProtocol, USyncDeviceProtocol, USyncDisappearingModeProtocol, USyncStatusProtocol } from "./Protocols/index.js";
import { USyncBotProfileProtocol } from "./Protocols/UsyncBotProfileProtocol.js";
import { USyncLIDProtocol } from "./Protocols/UsyncLIDProtocol.js";
import { getBinaryNodeChild } from "../WABinary/index.js";
import { USyncUser } from "./USyncUser.js";

export class USyncQuery {
  constructor() {
    this.protocols = [];
    this.users = [];
    this.context = "interactive";
    this.mode = "query";
  }
  withMode(mode) {
    this.mode = mode;
    return this;
  }
  withContext(context) {
    this.context = context;
    return this;
  }
  withUser(user) {
    this.users.push(user);
    return this;
  }
  parseUSyncQueryResult(result) {
    if (!result || result.attrs.type !== "result") {
      return;
    }
    const protocolMap = Object.fromEntries(this.protocols.map(protocol => {
      return [protocol.name, protocol.parser];
    }));
    const queryResult = {
      list: [],
      sideList: []
    };
    const usyncNode = getBinaryNodeChild(result, "usync");
    const listNode = usyncNode ? getBinaryNodeChild(usyncNode, "list") : undefined;
    if (listNode?.content && Array.isArray(listNode.content)) {
      queryResult.list = listNode.content.reduce((acc, node) => {
        const id = node?.attrs.jid;
        if (id) {
          const data = Array.isArray(node?.content)
            ? Object.fromEntries(node.content
              .map(content => {
              const protocol = content.tag;
              const parser = protocolMap[protocol];
              if (parser) {
                return [protocol, parser(content)];
              }
              else {
                return [protocol, null];
              }
            })
              .filter(([, b]) => b !== null))
            : {};
          acc.push({ ...data, id });
        }
        return acc;
      }, []);
    }
    return queryResult;
  }
  withDeviceProtocol() {
    this.protocols.push(new USyncDeviceProtocol());
    return this;
  }
  withContactProtocol() {
    this.protocols.push(new USyncContactProtocol());
    return this;
  }
  withStatusProtocol() {
    this.protocols.push(new USyncStatusProtocol());
    return this;
  }
  withDisappearingModeProtocol() {
    this.protocols.push(new USyncDisappearingModeProtocol());
    return this;
  }
  withBotProfileProtocol() {
    this.protocols.push(new USyncBotProfileProtocol());
    return this;
  }
  withLIDProtocol() {
    this.protocols.push(new USyncLIDProtocol());
    return this;
  }
}