@whiskeysockets/baileys
Version:
A WebSockets library for interacting with WhatsApp Web
253 lines (252 loc) • 8.71 kB
JavaScript
;
var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
if (k2 === undefined) k2 = k;
var desc = Object.getOwnPropertyDescriptor(m, k);
if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
desc = { enumerable: true, get: function() { return m[k]; } };
}
Object.defineProperty(o, k2, desc);
}) : (function(o, m, k, k2) {
if (k2 === undefined) k2 = k;
o[k2] = m[k];
}));
var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
Object.defineProperty(o, "default", { enumerable: true, value: v });
}) : function(o, v) {
o["default"] = v;
});
var __importStar = (this && this.__importStar) || (function () {
var ownKeys = function(o) {
ownKeys = Object.getOwnPropertyNames || function (o) {
var ar = [];
for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
return ar;
};
return ownKeys(o);
};
return function (mod) {
if (mod && mod.__esModule) return mod;
var result = {};
if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
__setModuleDefault(result, mod);
return result;
};
})();
Object.defineProperty(exports, "__esModule", { value: true });
exports.encodeBinaryNode = void 0;
const constants = __importStar(require("./constants"));
const jid_utils_1 = require("./jid-utils");
const encodeBinaryNode = (node, opts = constants, buffer = [0]) => {
const encoded = encodeBinaryNodeInner(node, opts, buffer);
return Buffer.from(encoded);
};
exports.encodeBinaryNode = encodeBinaryNode;
const encodeBinaryNodeInner = ({ tag, attrs, content }, opts, buffer) => {
const { TAGS, TOKEN_MAP } = opts;
const pushByte = (value) => buffer.push(value & 0xff);
const pushInt = (value, n, littleEndian = false) => {
for (let i = 0; i < n; i++) {
const curShift = littleEndian ? i : n - 1 - i;
buffer.push((value >> (curShift * 8)) & 0xff);
}
};
const pushBytes = (bytes) => {
for (const b of bytes) {
buffer.push(b);
}
};
const pushInt16 = (value) => {
pushBytes([(value >> 8) & 0xff, value & 0xff]);
};
const pushInt20 = (value) => (pushBytes([(value >> 16) & 0x0f, (value >> 8) & 0xff, value & 0xff]));
const writeByteLength = (length) => {
if (length >= 4294967296) {
throw new Error('string too large to encode: ' + length);
}
if (length >= 1 << 20) {
pushByte(TAGS.BINARY_32);
pushInt(length, 4); // 32 bit integer
}
else if (length >= 256) {
pushByte(TAGS.BINARY_20);
pushInt20(length);
}
else {
pushByte(TAGS.BINARY_8);
pushByte(length);
}
};
const writeStringRaw = (str) => {
const bytes = Buffer.from(str, 'utf-8');
writeByteLength(bytes.length);
pushBytes(bytes);
};
const writeJid = ({ domainType, device, user, server }) => {
if (typeof device !== 'undefined') {
pushByte(TAGS.AD_JID);
pushByte(domainType || 0);
pushByte(device || 0);
writeString(user);
}
else {
pushByte(TAGS.JID_PAIR);
if (user.length) {
writeString(user);
}
else {
pushByte(TAGS.LIST_EMPTY);
}
writeString(server);
}
};
const packNibble = (char) => {
switch (char) {
case '-':
return 10;
case '.':
return 11;
case '\0':
return 15;
default:
if (char >= '0' && char <= '9') {
return char.charCodeAt(0) - '0'.charCodeAt(0);
}
throw new Error(`invalid byte for nibble "${char}"`);
}
};
const packHex = (char) => {
if (char >= '0' && char <= '9') {
return char.charCodeAt(0) - '0'.charCodeAt(0);
}
if (char >= 'A' && char <= 'F') {
return 10 + char.charCodeAt(0) - 'A'.charCodeAt(0);
}
if (char >= 'a' && char <= 'f') {
return 10 + char.charCodeAt(0) - 'a'.charCodeAt(0);
}
if (char === '\0') {
return 15;
}
throw new Error(`Invalid hex char "${char}"`);
};
const writePackedBytes = (str, type) => {
if (str.length > TAGS.PACKED_MAX) {
throw new Error('Too many bytes to pack');
}
pushByte(type === 'nibble' ? TAGS.NIBBLE_8 : TAGS.HEX_8);
let roundedLength = Math.ceil(str.length / 2.0);
if (str.length % 2 !== 0) {
roundedLength |= 128;
}
pushByte(roundedLength);
const packFunction = type === 'nibble' ? packNibble : packHex;
const packBytePair = (v1, v2) => {
const result = (packFunction(v1) << 4) | packFunction(v2);
return result;
};
const strLengthHalf = Math.floor(str.length / 2);
for (let i = 0; i < strLengthHalf; i++) {
pushByte(packBytePair(str[2 * i], str[2 * i + 1]));
}
if (str.length % 2 !== 0) {
pushByte(packBytePair(str[str.length - 1], '\x00'));
}
};
const isNibble = (str) => {
if (!str || str.length > TAGS.PACKED_MAX) {
return false;
}
for (const char of str) {
const isInNibbleRange = char >= '0' && char <= '9';
if (!isInNibbleRange && char !== '-' && char !== '.') {
return false;
}
}
return true;
};
const isHex = (str) => {
if (!str || str.length > TAGS.PACKED_MAX) {
return false;
}
for (const char of str) {
const isInNibbleRange = char >= '0' && char <= '9';
if (!isInNibbleRange && !(char >= 'A' && char <= 'F')) {
return false;
}
}
return true;
};
const writeString = (str) => {
if (str === undefined || str === null) {
pushByte(TAGS.LIST_EMPTY);
return;
}
const tokenIndex = TOKEN_MAP[str];
if (tokenIndex) {
if (typeof tokenIndex.dict === 'number') {
pushByte(TAGS.DICTIONARY_0 + tokenIndex.dict);
}
pushByte(tokenIndex.index);
}
else if (isNibble(str)) {
writePackedBytes(str, 'nibble');
}
else if (isHex(str)) {
writePackedBytes(str, 'hex');
}
else if (str) {
const decodedJid = (0, jid_utils_1.jidDecode)(str);
if (decodedJid) {
writeJid(decodedJid);
}
else {
writeStringRaw(str);
}
}
};
const writeListStart = (listSize) => {
if (listSize === 0) {
pushByte(TAGS.LIST_EMPTY);
}
else if (listSize < 256) {
pushBytes([TAGS.LIST_8, listSize]);
}
else {
pushByte(TAGS.LIST_16);
pushInt16(listSize);
}
};
if (!tag) {
throw new Error('Invalid node: tag cannot be undefined');
}
const validAttributes = Object.keys(attrs || {}).filter(k => (typeof attrs[k] !== 'undefined' && attrs[k] !== null));
writeListStart(2 * validAttributes.length + 1 + (typeof content !== 'undefined' ? 1 : 0));
writeString(tag);
for (const key of validAttributes) {
if (typeof attrs[key] === 'string') {
writeString(key);
writeString(attrs[key]);
}
}
if (typeof content === 'string') {
writeString(content);
}
else if (Buffer.isBuffer(content) || content instanceof Uint8Array) {
writeByteLength(content.length);
pushBytes(content);
}
else if (Array.isArray(content)) {
const validContent = content.filter(item => item && (item.tag || Buffer.isBuffer(item) || item instanceof Uint8Array || typeof item === 'string'));
writeListStart(validContent.length);
for (const item of validContent) {
encodeBinaryNodeInner(item, opts, buffer);
}
}
else if (typeof content === 'undefined') {
// do nothing
}
else {
throw new Error(`invalid children for header "${tag}": ${content} (${typeof content})`);
}
return buffer;
};