@vanzxysenpai/baileys 0.0.0-stage → 0.0.2

package/lib/Utils/companion-reg-client-utils.d.ts

@@ -0,0 +1,4 @@
+export const CompanionWebClientType: any;
+export function getCompanionWebClientType([os, browserName]: [any, any]): any;
+export function getCompanionPlatformId(browser: any): any;
+export function buildPairingQRData(ref: any, noiseKeyB64: any, identityKeyB64: any, advB64: any, browser: any): string;

package/lib/Utils/companion-reg-client-utils.js

@@ -0,0 +1,39 @@
+export var CompanionWebClientType;
+(function (CompanionWebClientType) {
+    CompanionWebClientType[CompanionWebClientType["UNKNOWN"] = 0] = "UNKNOWN";
+    CompanionWebClientType[CompanionWebClientType["CHROME"] = 1] = "CHROME";
+    CompanionWebClientType[CompanionWebClientType["EDGE"] = 2] = "EDGE";
+    CompanionWebClientType[CompanionWebClientType["FIREFOX"] = 3] = "FIREFOX";
+    CompanionWebClientType[CompanionWebClientType["IE"] = 4] = "IE";
+    CompanionWebClientType[CompanionWebClientType["OPERA"] = 5] = "OPERA";
+    CompanionWebClientType[CompanionWebClientType["SAFARI"] = 6] = "SAFARI";
+    CompanionWebClientType[CompanionWebClientType["ELECTRON"] = 7] = "ELECTRON";
+    CompanionWebClientType[CompanionWebClientType["UWP"] = 8] = "UWP";
+    CompanionWebClientType[CompanionWebClientType["OTHER_WEB_CLIENT"] = 9] = "OTHER_WEB_CLIENT";
+})(CompanionWebClientType || (CompanionWebClientType = {}));
+const BROWSER_TO_COMPANION_WEB_CLIENT = {
+    Chrome: CompanionWebClientType.CHROME,
+    Edge: CompanionWebClientType.EDGE,
+    Firefox: CompanionWebClientType.FIREFOX,
+    IE: CompanionWebClientType.IE,
+    Opera: CompanionWebClientType.OPERA,
+    Safari: CompanionWebClientType.SAFARI
+};
+export const getCompanionWebClientType = ([os, browserName]) => {
+    if (browserName === 'Desktop') {
+        return os === 'Windows' ? CompanionWebClientType.UWP : CompanionWebClientType.ELECTRON;
+    }
+    return BROWSER_TO_COMPANION_WEB_CLIENT[browserName] || CompanionWebClientType.OTHER_WEB_CLIENT;
+};
+export const getCompanionPlatformId = (browser) => {
+    return getCompanionWebClientType(browser).toString();
+};
+export const buildPairingQRData = (ref, noiseKeyB64, identityKeyB64, advB64, browser) => {
+    return 'https://wa.me/settings/linked_devices#' + [
+        ref,
+        noiseKeyB64,
+        identityKeyB64,
+        advB64,
+        getCompanionPlatformId(browser)
+    ].join(',');
+};

package/lib/Utils/connection-watchdog.d.ts

@@ -0,0 +1,10 @@
+/**
+ * @param {import('../Types/index.js').WASocket} sock
+ * @param {{ idleThresholdMs?: number, checkIntervalMs?: number, probe?: 'ping' | 'presence' }} [opts]
+ * @returns {() => void} stop function
+ */
+export function setupConnectionWatchdog(sock: any, opts1?: { [key: string]: any; 
+    idleThresholdMs?: number;
+    checkIntervalMs?: number;
+    probe?: "ping" | "presence";
+}): () => void;

package/lib/Utils/connection-watchdog.js

@@ -0,0 +1,74 @@
+// Vanz@Port (v2.1.14) --- adapted from @lpzeravk/baileys 1.0.0 (lib/Socket/connection-watchdog.js, MIT), rewritten:
+//  - default probe is the same `w:p` ping iq the built-in keep-alive uses, NOT `sendPresenceUpdate('available')`
+//    (that flips the linked device to "online" every time it fires, which is visible to contacts and can suppress
+//    phone notifications). Pass `{ probe: 'presence' }` for the original behaviour;
+//  - the timer is unref()'d and stops itself when the connection closes;
+//  - `sock.end()` is always given an Error and awaited defensively.
+//
+// Zombie-socket detector: after `idleThresholdMs` with no incoming message it probes the socket, and ends it if the
+// WebSocket is closed or the probe fails. That emits the normal `connection.update` close — your existing reconnect
+// logic takes over; this does NOT recreate the socket.
+//
+// Note: Baileys' own keep-alive (`keepAliveIntervalMs`, 15s here) already ends the connection when the server goes
+// silent, so this mostly matters if you raised that interval or disabled it. Opt-in, default off.
+import { S_WHATSAPP_NET } from '../WABinary/index.js';
+const DEFAULT_IDLE_THRESHOLD_MS = 600000;
+const DEFAULT_CHECK_INTERVAL_MS = 120000;
+/**
+ * @param {import('../Types/index.js').WASocket} sock
+ * @param {{ idleThresholdMs?: number, checkIntervalMs?: number, probe?: 'ping' | 'presence' }} [opts]
+ * @returns {() => void} stop function
+ */
+export function setupConnectionWatchdog(sock, { idleThresholdMs = DEFAULT_IDLE_THRESHOLD_MS, checkIntervalMs = DEFAULT_CHECK_INTERVAL_MS, probe = 'ping' } = {}) {
+    let lastActivity = Date.now();
+    let checking = false;
+    let stopped = false;
+    const onActivity = () => {
+        lastActivity = Date.now();
+    };
+    const stop = () => {
+        if (stopped) return;
+        stopped = true;
+        clearInterval(timer);
+        sock.ev.off('messages.upsert', onActivity);
+        sock.ev.off('connection.update', onConnectionUpdate);
+    };
+    const onConnectionUpdate = ({ connection }) => {
+        if (connection === 'open') onActivity();
+        if (connection === 'close') stop();
+    };
+    const killSocket = (reason) => {
+        sock.logger?.warn?.(`[watchdog] ${reason} — ending socket to force a reconnect`);
+        Promise.resolve(sock.end(new Error(`connection-watchdog: ${reason}`))).catch(() => { });
+    };
+    const timer = setInterval(async () => {
+        if (checking || stopped) return;
+        const idleMs = Date.now() - lastActivity;
+        if (idleMs < idleThresholdMs) return;
+        checking = true;
+        try {
+            if (!sock.ws?.isOpen) {
+                killSocket(`websocket not open after ${Math.floor(idleMs / 1000)}s idle`);
+                return;
+            }
+            if (probe === 'presence') {
+                await sock.sendPresenceUpdate('available');
+            } else {
+                await sock.query({
+                    tag: 'iq',
+                    attrs: { id: sock.generateMessageTag(), to: S_WHATSAPP_NET, type: 'get', xmlns: 'w:p' },
+                    content: [{ tag: 'ping', attrs: {} }]
+                });
+            }
+            lastActivity = Date.now();
+        } catch (e) {
+            killSocket(`${probe} probe failed (${e?.message ?? e})`);
+        } finally {
+            checking = false;
+        }
+    }, checkIntervalMs);
+    timer.unref?.();
+    sock.ev.on('messages.upsert', onActivity);
+    sock.ev.on('connection.update', onConnectionUpdate);
+    return stop;
+}

package/lib/Utils/crypto.d.ts

@@ -0,0 +1,41 @@
+/**
+ * encrypt AES 256 GCM;
+ * where the tag tag is suffixed to the ciphertext
+ * */
+export function aesEncryptGCM(plaintext: any, key: any, iv: any, additionalData: any): any;
+/**
+ * decrypt AES 256 GCM;
+ * where the auth tag is suffixed to the ciphertext
+ * */
+export function aesDecryptGCM(ciphertext: any, key: any, iv: any, additionalData: any): any;
+export function aesEncryptCTR(plaintext: any, key: any, iv: any): any;
+export function aesDecryptCTR(ciphertext: any, key: any, iv: any): any;
+/** decrypt AES 256 CBC; where the IV is prefixed to the buffer */
+export function aesDecrypt(buffer: any, key: any): any;
+/** decrypt AES 256 CBC */
+export function aesDecryptWithIV(buffer: any, key: any, IV: any): any;
+export function aesEncrypt(buffer: any, key: any): any;
+export function aesEncrypWithIV(buffer: any, key: any, IV: any): any;
+export function hmacSign(buffer: any, key: any, variant?: string): any;
+export function sha256(buffer: any): any;
+export function derivePairingCodeKey(pairingCode: any, salt: any): Promise<any>;
+export function generateSignalPubKey(pubKey: any): any;
+export namespace Curve {
+    function generateKeyPair(): {
+        private: any;
+        public: any;
+    };
+    function sharedKey(privateKey: any, publicKey: any): any;
+    function sign(privateKey: any, buf: any): any;
+    function verify(pubKey: any, message: any, signature: any): boolean;
+}
+export function signedKeyPair(identityKeyPair: any, keyId: any): {
+    keyPair: {
+        private: any;
+        public: any;
+    };
+    signature: any;
+    keyId: any;
+};
+export let md5: any;
+export let hkdf: any;

package/lib/Utils/crypto.js

@@ -0,0 +1,145 @@
+import { createCipheriv, createDecipheriv, createHash, createHmac, randomBytes } from 'crypto';
+import * as curve from 'libsignal/src/curve.js';
+import { KEY_BUNDLE_TYPE } from '../Defaults/index.js';
+
+let md5, hkdf;
+try {
+    const rustBridge = await import('whatsapp-rust-bridge');
+    md5 = rustBridge.md5;
+    hkdf = rustBridge.hkdf;
+} catch {
+    // Pure JS fallback for md5
+    md5 = (buf) => createHash('md5').update(buf).digest();
+    // Pure JS fallback for hkdf (HKDF-SHA256)
+    hkdf = (key, length, { salt, info } = {}) => {
+        const hashLen = 32; // SHA-256
+        const prk = salt
+            ? createHmac('sha256', salt).update(key).digest()
+            : createHmac('sha256', Buffer.alloc(hashLen)).update(key).digest();
+        const infoBuffer = info ? (typeof info === 'string' ? Buffer.from(info) : Buffer.from(info)) : Buffer.alloc(0);
+        const blocks = Math.ceil(length / hashLen);
+        let prev = Buffer.alloc(0);
+        const output = [];
+        for (let i = 1; i <= blocks; i++) {
+            prev = createHmac('sha256', prk)
+                .update(Buffer.concat([prev, infoBuffer, Buffer.from([i])]))
+                .digest();
+            output.push(prev);
+        }
+        return Buffer.concat(output).subarray(0, length);
+    };
+}
+export { md5, hkdf };
+// insure browser & node compatibility
+const { subtle } = globalThis.crypto;
+/** prefix version byte to the pub keys, required for some curve crypto functions */
+export const generateSignalPubKey = (pubKey) => pubKey.length === 33 ? pubKey : Buffer.concat([KEY_BUNDLE_TYPE, pubKey]);
+export const Curve = {
+    generateKeyPair: () => {
+        const { pubKey, privKey } = curve.generateKeyPair();
+        return {
+            private: Buffer.from(privKey),
+            // remove version byte
+            public: Buffer.from(pubKey.slice(1))
+        };
+    },
+    sharedKey: (privateKey, publicKey) => {
+        const shared = curve.calculateAgreement(generateSignalPubKey(publicKey), privateKey);
+        return Buffer.from(shared);
+    },
+    sign: (privateKey, buf) => curve.calculateSignature(privateKey, buf),
+    verify: (pubKey, message, signature) => {
+        try {
+            curve.verifySignature(generateSignalPubKey(pubKey), message, signature);
+            return true;
+        }
+        catch (error) {
+            return false;
+        }
+    }
+};
+export const signedKeyPair = (identityKeyPair, keyId) => {
+    const preKey = Curve.generateKeyPair();
+    const pubKey = generateSignalPubKey(preKey.public);
+    const signature = Curve.sign(identityKeyPair.private, pubKey);
+    return { keyPair: preKey, signature, keyId };
+};
+const GCM_TAG_LENGTH = 128 >> 3;
+/**
+ * encrypt AES 256 GCM;
+ * where the tag tag is suffixed to the ciphertext
+ * */
+export function aesEncryptGCM(plaintext, key, iv, additionalData) {
+    const cipher = createCipheriv('aes-256-gcm', key, iv);
+    cipher.setAAD(additionalData);
+    return Buffer.concat([cipher.update(plaintext), cipher.final(), cipher.getAuthTag()]);
+}
+/**
+ * decrypt AES 256 GCM;
+ * where the auth tag is suffixed to the ciphertext
+ * */
+export function aesDecryptGCM(ciphertext, key, iv, additionalData) {
+    const decipher = createDecipheriv('aes-256-gcm', key, iv);
+    // decrypt additional adata
+    const enc = ciphertext.slice(0, ciphertext.length - GCM_TAG_LENGTH);
+    const tag = ciphertext.slice(ciphertext.length - GCM_TAG_LENGTH);
+    // set additional data
+    decipher.setAAD(additionalData);
+    decipher.setAuthTag(tag);
+    return Buffer.concat([decipher.update(enc), decipher.final()]);
+}
+export function aesEncryptCTR(plaintext, key, iv) {
+    const cipher = createCipheriv('aes-256-ctr', key, iv);
+    return Buffer.concat([cipher.update(plaintext), cipher.final()]);
+}
+export function aesDecryptCTR(ciphertext, key, iv) {
+    const decipher = createDecipheriv('aes-256-ctr', key, iv);
+    return Buffer.concat([decipher.update(ciphertext), decipher.final()]);
+}
+/** decrypt AES 256 CBC; where the IV is prefixed to the buffer */
+export function aesDecrypt(buffer, key) {
+    return aesDecryptWithIV(buffer.subarray(16), key, buffer.subarray(0, 16));
+}
+/** decrypt AES 256 CBC */
+export function aesDecryptWithIV(buffer, key, IV) {
+    const aes = createDecipheriv('aes-256-cbc', key, IV);
+    return Buffer.concat([aes.update(buffer), aes.final()]);
+}
+// encrypt AES 256 CBC; where a random IV is prefixed to the buffer
+export function aesEncrypt(buffer, key) {
+    const IV = randomBytes(16);
+    const aes = createCipheriv('aes-256-cbc', key, IV);
+    return Buffer.concat([IV, aes.update(buffer), aes.final()]); // prefix IV to the buffer
+}
+// encrypt AES 256 CBC with a given IV
+export function aesEncrypWithIV(buffer, key, IV) {
+    const aes = createCipheriv('aes-256-cbc', key, IV);
+    return Buffer.concat([aes.update(buffer), aes.final()]); // prefix IV to the buffer
+}
+// sign HMAC using SHA 256
+export function hmacSign(buffer, key, variant = 'sha256') {
+    return createHmac(variant, key).update(buffer).digest();
+}
+export function sha256(buffer) {
+    return createHash('sha256').update(buffer).digest();
+}
+export async function derivePairingCodeKey(pairingCode, salt) {
+    // Convert inputs to formats Web Crypto API can work with
+    const encoder = new TextEncoder();
+    const pairingCodeBuffer = encoder.encode(pairingCode);
+    const saltBuffer = new Uint8Array(salt instanceof Uint8Array ? salt : new Uint8Array(salt));
+    // Import the pairing code as key material
+    const keyMaterial = await subtle.importKey('raw', pairingCodeBuffer, { name: 'PBKDF2' }, false, [
+        'deriveBits'
+    ]);
+    // Derive bits using PBKDF2 with the same parameters
+    // 2 << 16 = 131,072 iterations
+    const derivedBits = await subtle.deriveBits({
+        name: 'PBKDF2',
+        salt: saltBuffer,
+        iterations: 2 << 16,
+        hash: 'SHA-256'
+    }, keyMaterial, 32 * 8 // 32 bytes * 8 = 256 bits
+    );
+    return Buffer.from(derivedBits);
+}

package/lib/Utils/decode-wa-message.d.ts

@@ -0,0 +1,83 @@
+/**
+ * Decode the received node as a message.
+ * @note this will only parse the message, not decrypt it
+ */
+export function decodeMessageNode(stanza: any, meId: any, meLid: any): {
+    fullMessage: {
+        key: {
+            server_id?: any;
+            remoteJid: any;
+            remoteJidAlt: any;
+            remoteJidUsername: any;
+            fromMe: boolean;
+            id: any;
+            participant: any;
+            participantAlt: any;
+            participantUsername: any;
+            addressingMode: any;
+        };
+        category: any;
+        messageTimestamp: number;
+        pushName: any;
+        broadcast: any;
+    };
+    author: any;
+    sender: any;
+};
+export function getDecryptionJid(sender: any, repository: any): Promise<any>;
+export const NO_MESSAGE_FOUND_ERROR_TEXT: "Message absent from node";
+export const MISSING_KEYS_ERROR_TEXT: "Key used already or never filled";
+export const ACCOUNT_RESTRICTED_TEXT: "Your account has been restricted";
+export namespace DECRYPTION_RETRY_CONFIG {
+    let maxRetries: number;
+    let baseDelayMs: number;
+    let sessionRecordErrors: string[];
+}
+export namespace NACK_REASONS {
+    let SenderReachoutTimelocked: number;
+    let ParsingError: number;
+    let UnrecognizedStanza: number;
+    let UnrecognizedStanzaClass: number;
+    let UnrecognizedStanzaType: number;
+    let InvalidProtobuf: number;
+    let InvalidHostedCompanionStanza: number;
+    let MissingMessageSecret: number;
+    let SignalErrorOldCounter: number;
+    let MessageDeletedOnPeer: number;
+    let UnhandledError: number;
+    let UnsupportedAdminRevoke: number;
+    let UnsupportedLIDGroup: number;
+    let DBOperationFailed: number;
+}
+export namespace SERVER_ERROR_CODES {
+    let MessageAccountRestriction: string;
+    let SmaxInvalid: string;
+}
+export function extractAddressingContext(stanza: any): {
+    addressingMode: any;
+    senderAlt: any;
+    recipientAlt: any;
+};
+export function decryptMessageNode(stanza: any, meId: any, meLid: any, repository: any, logger: any): {
+    fullMessage: {
+        key: {
+            server_id?: any;
+            remoteJid: any;
+            remoteJidAlt: any;
+            remoteJidUsername: any;
+            fromMe: boolean;
+            id: any;
+            participant: any;
+            participantAlt: any;
+            participantUsername: any;
+            addressingMode: any;
+        };
+        category: any;
+        messageTimestamp: number;
+        pushName: any;
+        broadcast: any;
+    };
+    category: any;
+    author: any;
+    decrypt(): Promise<void>;
+};