@sgintokic/baileys 0.0.2

package/lib/WAM/encode.js

@@ -0,0 +1,151 @@
+import { BinaryInfo } from "./BinaryInfo.js";
+import {
+  FLAG_BYTE,
+  FLAG_EVENT,
+  FLAG_EXTENDED,
+  FLAG_FIELD,
+  FLAG_GLOBAL,
+  WEB_EVENTS,
+  WEB_GLOBALS,
+} from "./constants.js";
+const getHeaderBitLength = (key) => (key < 256 ? 2 : 3);
+export const encodeWAM = (binaryInfo) => {
+  binaryInfo.buffer = [];
+  encodeWAMHeader(binaryInfo);
+  encodeEvents(binaryInfo);
+  const totalSize = binaryInfo.buffer
+    .map((a) => a.length)
+    .reduce((a, b) => a + b);
+  const buffer = Buffer.alloc(totalSize);
+  let offset = 0;
+  for (const buffer_ of binaryInfo.buffer) {
+    buffer_.copy(buffer, offset);
+    offset += buffer_.length;
+  }
+  return buffer;
+};
+function encodeWAMHeader(binaryInfo) {
+  const headerBuffer = Buffer.alloc(8); // starting buffer
+  headerBuffer.write("WAM", 0, "utf8");
+  headerBuffer.writeUInt8(binaryInfo.protocolVersion, 3);
+  headerBuffer.writeUInt8(1, 4); //random flag
+  headerBuffer.writeUInt16BE(binaryInfo.sequence, 5);
+  headerBuffer.writeUInt8(0, 7); // regular channel
+  binaryInfo.buffer.push(headerBuffer);
+}
+function encodeGlobalAttributes(binaryInfo, globals) {
+  for (const [key, _value] of Object.entries(globals)) {
+    const id = WEB_GLOBALS.find((a) => a?.name === key).id;
+    let value = _value;
+    if (typeof value === "boolean") {
+      value = value ? 1 : 0;
+    }
+    binaryInfo.buffer.push(serializeData(id, value, FLAG_GLOBAL));
+  }
+}
+function encodeEvents(binaryInfo) {
+  for (const [name, { props, globals }] of binaryInfo.events.map(
+    (a) => Object.entries(a)[0],
+  )) {
+    encodeGlobalAttributes(binaryInfo, globals);
+    const event = WEB_EVENTS.find((a) => a.name === name);
+    const props_ = Object.entries(props);
+    let extended = false;
+    for (const [, value] of props_) {
+      extended || (extended = value !== null);
+    }
+    const eventFlag = extended ? FLAG_EVENT : FLAG_EVENT | FLAG_EXTENDED;
+    binaryInfo.buffer.push(serializeData(event.id, -event.weight, eventFlag));
+    for (let i = 0; i < props_.length; i++) {
+      const [key, _value] = props_[i];
+      const id = event.props[key]?.[0];
+      extended = i < props_.length - 1;
+      let value = _value;
+      if (typeof value === "boolean") {
+        value = value ? 1 : 0;
+      }
+      const fieldFlag = extended ? FLAG_EVENT : FLAG_FIELD | FLAG_EXTENDED;
+      binaryInfo.buffer.push(serializeData(id, value, fieldFlag));
+    }
+  }
+}
+function serializeData(key, value, flag) {
+  const bufferLength = getHeaderBitLength(key);
+  let buffer;
+  let offset = 0;
+  if (value === null) {
+    if (flag === FLAG_GLOBAL) {
+      buffer = Buffer.alloc(bufferLength);
+      offset = serializeHeader(buffer, offset, key, flag);
+      return buffer;
+    }
+  } else if (typeof value === "number" && Number.isInteger(value)) {
+    // is number
+    if (value === 0 || value === 1) {
+      buffer = Buffer.alloc(bufferLength);
+      offset = serializeHeader(buffer, offset, key, flag | ((value + 1) << 4));
+      return buffer;
+    } else if (-128 <= value && value < 128) {
+      buffer = Buffer.alloc(bufferLength + 1);
+      offset = serializeHeader(buffer, offset, key, flag | (3 << 4));
+      buffer.writeInt8(value, offset);
+      return buffer;
+    } else if (-32768 <= value && value < 32768) {
+      buffer = Buffer.alloc(bufferLength + 2);
+      offset = serializeHeader(buffer, offset, key, flag | (4 << 4));
+      buffer.writeInt16LE(value, offset);
+      return buffer;
+    } else if (-2147483648 <= value && value < 2147483648) {
+      buffer = Buffer.alloc(bufferLength + 4);
+      offset = serializeHeader(buffer, offset, key, flag | (5 << 4));
+      buffer.writeInt32LE(value, offset);
+      return buffer;
+    } else {
+      buffer = Buffer.alloc(bufferLength + 8);
+      offset = serializeHeader(buffer, offset, key, flag | (7 << 4));
+      buffer.writeDoubleLE(value, offset);
+      return buffer;
+    }
+  } else if (typeof value === "number") {
+    // is float
+    buffer = Buffer.alloc(bufferLength + 8);
+    offset = serializeHeader(buffer, offset, key, flag | (7 << 4));
+    buffer.writeDoubleLE(value, offset);
+    return buffer;
+  } else if (typeof value === "string") {
+    // is string
+    const utf8Bytes = Buffer.byteLength(value, "utf8");
+    if (utf8Bytes < 256) {
+      buffer = Buffer.alloc(bufferLength + 1 + utf8Bytes);
+      offset = serializeHeader(buffer, offset, key, flag | (8 << 4));
+      buffer.writeUint8(utf8Bytes, offset++);
+    } else if (utf8Bytes < 65536) {
+      buffer = Buffer.alloc(bufferLength + 2 + utf8Bytes);
+      offset = serializeHeader(buffer, offset, key, flag | (9 << 4));
+      buffer.writeUInt16LE(utf8Bytes, offset);
+      offset += 2;
+    } else {
+      buffer = Buffer.alloc(bufferLength + 4 + utf8Bytes);
+      offset = serializeHeader(buffer, offset, key, flag | (10 << 4));
+      buffer.writeUInt32LE(utf8Bytes, offset);
+      offset += 4;
+    }
+    buffer.write(value, offset, "utf8");
+    return buffer;
+  }
+  throw "missing";
+}
+function serializeHeader(buffer, offset, key, flag) {
+  if (key < 256) {
+    buffer.writeUInt8(flag, offset);
+    offset += 1;
+    buffer.writeUInt8(key, offset);
+    offset += 1;
+  } else {
+    buffer.writeUInt8(flag | FLAG_BYTE, offset);
+    offset += 1;
+    buffer.writeUInt16LE(key, offset);
+    offset += 2;
+  }
+  return offset;
+}

package/lib/WAM/index.js

@@ -0,0 +1,3 @@
+export * from "./constants.js";
+export * from "./encode.js";
+export * from "./BinaryInfo.js";

package/lib/WAUSync/Protocols/USyncContactProtocol.js

@@ -0,0 +1,21 @@
+import { assertNodeErrorFree } from "../../WABinary/index.js";
+import { USyncUser } from "../USyncUser.js";
+export class USyncContactProtocol {
+  constructor() {
+    this.name = "contact";
+  }
+  getQueryElement() {
+    return { tag: "contact", attrs: {} };
+  }
+  getUserElement(user) {
+    //TODO: Implement type / username fields (not yet supported)
+    return { tag: "contact", attrs: {}, content: user.phone };
+  }
+  parser(node) {
+    if (node.tag === "contact") {
+      assertNodeErrorFree(node);
+      return node?.attrs?.type === "in";
+    }
+    return false;
+  }
+}

package/lib/WAUSync/Protocols/USyncDeviceProtocol.js

@@ -0,0 +1,50 @@
+import {
+  assertNodeErrorFree,
+  getBinaryNodeChild,
+} from "../../WABinary/index.js";
+export class USyncDeviceProtocol {
+  constructor() {
+    this.name = "devices";
+  }
+  getQueryElement() {
+    return { tag: "devices", attrs: { version: "2" } };
+  }
+  getUserElement() {
+    //TODO: Implement device phashing, ts and expectedTs
+    //TODO: if all are not present, return null <- current behavior
+    //TODO: otherwise return a node w tag 'devices' w those as attrs
+    return null;
+  }
+  parser(node) {
+    const deviceList = [];
+    let keyIndex = undefined;
+    if (node.tag === "devices") {
+      assertNodeErrorFree(node);
+      const deviceListNode = getBinaryNodeChild(node, "device-list");
+      const keyIndexNode = getBinaryNodeChild(node, "key-index-list");
+      if (Array.isArray(deviceListNode?.content)) {
+        for (const { tag, attrs } of deviceListNode.content) {
+          const id = +attrs.id;
+          const keyIndex = +attrs["key-index"];
+          if (tag === "device") {
+            deviceList.push({
+              id: id,
+              keyIndex: keyIndex,
+              isHosted: !!(attrs["is_hosted"] && attrs["is_hosted"] === "true"),
+            });
+          }
+        }
+      }
+      if (keyIndexNode?.tag === "key-index-list") {
+        keyIndex = {
+          timestamp: +keyIndexNode.attrs["ts"],
+          signedKeyIndex: keyIndexNode?.content,
+          expectedTimestamp: keyIndexNode.attrs["expected_ts"]
+            ? +keyIndexNode.attrs["expected_ts"]
+            : undefined,
+        };
+      }
+    }
+    return { deviceList: deviceList, keyIndex: keyIndex };
+  }
+}

package/lib/WAUSync/Protocols/USyncDisappearingModeProtocol.js

@@ -0,0 +1,20 @@
+import { assertNodeErrorFree } from "../../WABinary/index.js";
+export class USyncDisappearingModeProtocol {
+  constructor() {
+    this.name = "disappearing_mode";
+  }
+  getQueryElement() {
+    return { tag: "disappearing_mode", attrs: {} };
+  }
+  getUserElement() {
+    return null;
+  }
+  parser(node) {
+    if (node.tag === "disappearing_mode") {
+      assertNodeErrorFree(node);
+      const duration = +node?.attrs.duration;
+      const setAt = new Date(+(node?.attrs.t || 0) * 1e3);
+      return { duration: duration, setAt: setAt };
+    }
+  }
+}

package/lib/WAUSync/Protocols/USyncStatusProtocol.js

@@ -0,0 +1,29 @@
+import { assertNodeErrorFree } from "../../WABinary/index.js";
+export class USyncStatusProtocol {
+  constructor() {
+    this.name = "status";
+  }
+  getQueryElement() {
+    return { tag: "status", attrs: {} };
+  }
+  getUserElement() {
+    return null;
+  }
+  parser(node) {
+    if (node.tag === "status") {
+      assertNodeErrorFree(node);
+      let status = node?.content?.toString() ?? null;
+      const setAt = new Date(+(node?.attrs.t || 0) * 1e3);
+      if (!status) {
+        if (node.attrs?.code && +node.attrs.code === 401) {
+          status = "";
+        } else {
+          status = null;
+        }
+      } else if (typeof status === "string" && status.length === 0) {
+        status = null;
+      }
+      return { status: status, setAt: setAt };
+    }
+  }
+}

package/lib/WAUSync/Protocols/UsyncBotProfileProtocol.js

@@ -0,0 +1,59 @@
+import {
+  getBinaryNodeChild,
+  getBinaryNodeChildren,
+  getBinaryNodeChildString,
+} from "../../WABinary/index.js";
+import { USyncUser } from "../USyncUser.js";
+export class USyncBotProfileProtocol {
+  constructor() {
+    this.name = "bot";
+  }
+  getQueryElement() {
+    return {
+      tag: "bot",
+      attrs: {},
+      content: [{ tag: "profile", attrs: { v: "1" } }],
+    };
+  }
+  getUserElement(user) {
+    return {
+      tag: "bot",
+      attrs: {},
+      content: [{ tag: "profile", attrs: { persona_id: user.personaId } }],
+    };
+  }
+  parser(node) {
+    const botNode = getBinaryNodeChild(node, "bot");
+    const profile = getBinaryNodeChild(botNode, "profile");
+    const commandsNode = getBinaryNodeChild(profile, "commands");
+    const promptsNode = getBinaryNodeChild(profile, "prompts");
+    const commands = [];
+    const prompts = [];
+    for (const command of getBinaryNodeChildren(commandsNode, "command")) {
+      commands.push({
+        name: getBinaryNodeChildString(command, "name"),
+        description: getBinaryNodeChildString(command, "description"),
+      });
+    }
+    for (const prompt of getBinaryNodeChildren(promptsNode, "prompt")) {
+      prompts.push(
+        `${getBinaryNodeChildString(prompt, "emoji")} ${getBinaryNodeChildString(prompt, "text")}`,
+      );
+    }
+    return {
+      isDefault: !!getBinaryNodeChild(profile, "default"),
+      jid: node.attrs.jid,
+      name: getBinaryNodeChildString(profile, "name"),
+      attributes: getBinaryNodeChildString(profile, "attributes"),
+      description: getBinaryNodeChildString(profile, "description"),
+      category: getBinaryNodeChildString(profile, "category"),
+      personaId: profile.attrs["persona_id"],
+      commandsDescription: getBinaryNodeChildString(
+        commandsNode,
+        "description",
+      ),
+      commands: commands,
+      prompts: prompts,
+    };
+  }
+}

package/lib/WAUSync/Protocols/UsyncLIDProtocol.js

@@ -0,0 +1,21 @@
+export class USyncLIDProtocol {
+  constructor() {
+    this.name = "lid";
+  }
+  getQueryElement() {
+    return { tag: "lid", attrs: {} };
+  }
+  getUserElement(user) {
+    if (user.lid) {
+      return { tag: "lid", attrs: { jid: user.lid } };
+    } else {
+      return null;
+    }
+  }
+  parser(node) {
+    if (node.tag === "lid") {
+      return node.attrs.val;
+    }
+    return null;
+  }
+}

package/lib/WAUSync/Protocols/index.js

@@ -0,0 +1,4 @@
+export * from "./USyncDeviceProtocol.js";
+export * from "./USyncContactProtocol.js";
+export * from "./USyncStatusProtocol.js";
+export * from "./USyncDisappearingModeProtocol.js";

package/lib/WAUSync/USyncQuery.js

@@ -0,0 +1,103 @@
+import { getBinaryNodeChild } from "../WABinary/index.js";
+import { USyncBotProfileProtocol } from "./Protocols/UsyncBotProfileProtocol.js";
+import { USyncLIDProtocol } from "./Protocols/UsyncLIDProtocol.js";
+import {
+  USyncContactProtocol,
+  USyncDeviceProtocol,
+  USyncDisappearingModeProtocol,
+  USyncStatusProtocol,
+} from "./Protocols/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 = {
+      // TODO: implement errors etc.
+      list: [],
+      sideList: [],
+    };
+    const usyncNode = getBinaryNodeChild(result, "usync");
+    //TODO: implement error backoff, refresh etc.
+    //TODO: see if there are any errors in the result node
+    //const resultNode = getBinaryNodeChild(usyncNode, 'result')
+    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: id });
+        }
+        return acc;
+      }, []);
+    }
+    //TODO: implement side list
+    //const sideListNode = getBinaryNodeChild(usyncNode, 'side_list')
+    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;
+  }
+}

package/lib/WAUSync/USyncUser.js

@@ -0,0 +1,22 @@
+export class USyncUser {
+  withId(id) {
+    this.id = id;
+    return this;
+  }
+  withLid(lid) {
+    this.lid = lid;
+    return this;
+  }
+  withPhone(phone) {
+    this.phone = phone;
+    return this;
+  }
+  withType(type) {
+    this.type = type;
+    return this;
+  }
+  withPersonaId(personaId) {
+    this.personaId = personaId;
+    return this;
+  }
+}

package/lib/WAUSync/index.js

@@ -0,0 +1,3 @@
+export * from "./Protocols/index.js";
+export * from "./USyncQuery.js";
+export * from "./USyncUser.js";

package/lib/index.js

@@ -0,0 +1,11 @@
+import makeWASocket from "./Socket/index.js";
+export * from "../WAProto/index.js";
+export * from "./Utils/index.js";
+export * from "./Types/index.js";
+export * from "./Defaults/index.js";
+export * from "./Store/index.js";
+export * from "./WABinary/index.js";
+export * from "./WAM/index.js";
+export * from "./WAUSync/index.js";
+export { makeWASocket };
+export default makeWASocket;

package/package.json

@@ -0,0 +1,58 @@
+{
+	"name": "@sgintokic/baileys",
+	"version": "0.0.2",
+	"description": "A simple fork of Baileys for WhatsApp automation",
+	"main": "lib/index.js",
+	"type": "module",
+	"scripts": {
+		"preinstall": "node ./engine-requirements.js"
+	},
+	"engines": {
+		"node": ">=20.0.0"
+	},
+	"repository": {
+		"type": "git",
+		"url": "git+https://github.com/GSK-F/baileys.git"
+	},
+	"files": ["lib/**/*", "WAProto/**/*", "engine-requirements.js"],
+	"keywords": ["automation", "baileys", "baileys-fork", "baileys-mod", "js-whatsapp", "multi-device", "whatsapp", "whatsapp-api", "whatsapp-bot", "whatsapp-chat", "whatsapp-group", "whatsapp-web"],
+	"homepage": "https://github.com/GSK-F/baileys",
+	"author": "Sakata Gintoki",
+	"license": "MIT",
+	"dependencies": {
+		"@adiwajshing/keyed-db": "^0.2.4",
+		"@cacheable/node-cache": "^1.4.0",
+		"@hapi/boom": "^9.1.3",
+		"async-mutex": "^0.5.0",
+		"fflate": "^0.8.2",
+		"libsignal": "github:WhiskeySockets/libsignal-wasm#master",
+		"lru-cache": "^11.2.6",
+		"music-metadata": "^11.7.0",
+		"p-queue": "^9.1.0",
+		"pino": "^9.6.0",
+		"protobufjs": "^7.5.4",
+		"whatsapp-rust-bridge": "0.5.3",
+		"ws": "^8.19.0"
+	},
+	"peerDependencies": {
+		"@napi-rs/image": "~1.12.0",
+		"audio-decode": "^2.2.3",
+		"jimp": "*",
+		"link-preview-js": "^3.0.0",
+		"sharp": "*"
+	},
+	"peerDependenciesMeta": {
+		"@napi-rs/image": {
+			"optional": true
+		},
+		"audio-decode": {
+			"optional": true
+		},
+		"jimp": {
+			"optional": true
+		},
+		"link-preview-js": {
+			"optional": true
+		}
+	}
+}