@alannxd/baileys 6.1.4 → 6.1.7

package/lib/Socket/username.js

@@ -1,147 +1,236 @@
 import { executeWMexQuery } from './mex.js'
 import { USyncQuery, USyncUser } from '../WAUSync/index.js'
 import { makeNewsletterSocket } from './newsletter.js'
-
-export const USERNAME_QUERY_IDS = {
-	CHECK: '26124072630599520',
-	CHECK_MULTI: '27134626522840290',
-	SET: '27108705368767936',
-	GET: '32618050064506056',
-	GET_RECOMMENDATIONS: '26077456248616956',
-	PIN_SET: '25529696019976770'
+import crypto from 'crypto'
+/**
+ * w:mex query IDs for username operations.
+ *
+ * Source: whatsapp-android-mex_client_persist_ids.json (WA 2.26.26.4 APK assets)
+ * Cross-checked: live Frida capture on real device (wa-logger-2.26.26.4.js, 2026-06-30)
+ */
+const USERNAME_QUERY_IDS = {
+  CHECK: '26124072630599518', // UsernameCheck
+  CHECK_MULTI: '27134626522840286', // UsernameCheckMulti
+  SET: '27108705368767936', // UsernameSet
+  GET: '32618050064506055', // UsernameGet
+  GET_RECOMMENDATIONS: '26077456248616957', // UsernameGetRecommendationsQuery
+  PIN_SET: '25529696019976770' // UsernamePinSet
 }
 
-export const USERNAME_CHECK_RESULT = {
-	SUCCESS: 'SUCCESS',
-	INVALID: 'INVALID'
+/**
+ * Source enum values from EnumC141106Vn (UsernameCheck result)
+ * and EnumC141056Vi (rejection reasons) confirmed via C164057Wg.java
+ */
+const USERNAME_CHECK_RESULT = {
+  SUCCESS: 'SUCCESS',
+  INVALID: 'INVALID'
 }
 
-export const USERNAME_SOURCE = {
-	FB: 'FB',
-	IG: 'IG',
-	USER_INPUT: 'USER_INPUT',
-	SUGGESTION: 'SUGGESTION'
+const USERNAME_SOURCE = {
+  FB: 'FB',
+  IG: 'IG',
+  USER_INPUT: 'USER_INPUT',
+  SUGGESTION: 'SUGGESTION'
 }
 
 export const makeUsernameSocket = config => {
-	const sock = makeNewsletterSocket(config)
-	const { query, generateMessageTag, executeUSyncQuery } = sock
-
-	const mexQuery = (variables, queryId, dataPath) =>
-		executeWMexQuery(variables, queryId, dataPath, query, generateMessageTag)
-
-	const checkUsername = async (username, includeSuggestions = true) => {
-		if (!USERNAME_QUERY_IDS.CHECK) {
-			throw new Error('Username CHECK query_id not configured — capture a live WA session to obtain it')
-	}
-		const data = await mexQuery(
-			{ username, include_suggestions: includeSuggestions },
-			USERNAME_QUERY_IDS.CHECK,
-			'xwa2_username_check'
-	)
-		if (data?.result === USERNAME_CHECK_RESULT.SUCCESS) {
-			return { available: true, username }
-	}
-		return {
-			available: false,
-			username,
-			suggestions: data?.suggestions?? [],
-			rejectionReasons: data?.rejection_reasons?? [],
-			suggestionsEligible: data?.suggestions_eligible?? true
-    	}
-	}
-
-	const setUsername = async (username, options = {}) => {
-		if (!USERNAME_QUERY_IDS.SET) {
-			throw new Error('Username SET query_id not configured — capture a live WA session to obtain it')
-	}
-		const { source = USERNAME_SOURCE.USER_INPUT, sessionId, pin } = options
-		const variables = {
-			username,
-			reserved: false,
-			source,
-			...(sessionId? { session_id: sessionId } : {}),
-			...(pin? { pin } : {})
-	}
-		return mexQuery(variables, USERNAME_QUERY_IDS.SET, 'xwa2_username_set')
-	}
-
-	const deleteUsername = async () => {
-		if (!USERNAME_QUERY_IDS.SET) {
-			throw new Error('Username SET query_id not configured — capture a live WA session to obtain it')
-	}
-		return mexQuery({ username: null }, USERNAME_QUERY_IDS.SET, 'xwa2_username_delete')
-	}
-
-	const getMyUsername = async () => {
-		if (!USERNAME_QUERY_IDS.GET) {
-			throw new Error('Username GET query_id not configured — capture a live WA session to obtain it')
-	}
-		const data = await mexQuery({}, USERNAME_QUERY_IDS.GET, 'xwa2_username_get')
-		return data?.username?? null
-	}
-
-	const setUsernamePin = async pin => {
-		if (!USERNAME_QUERY_IDS.PIN_SET) {
-			throw new Error('Username PIN_SET query_id not configured — capture a live WA session to obtain it')
-	}
-		return mexQuery({ pin }, USERNAME_QUERY_IDS.PIN_SET, 'xwa2_username_pin_set')
-	}
-
-	const findUserByUsername = async (username, pin) => {
-		const usyncQuery = new USyncQuery().withContactProtocol()
-		const user = new USyncUser().withUsername(username)
-		if (pin) user.withUsernameKey(pin)
-	usyncQuery.withUser(user)
-		const result = await executeUSyncQuery(usyncQuery)
-		if (!result?.list?.length) return null
-		const entry = result.list[0]
-		return {
-			jid: entry.id,
-			contact: entry.contact?? false
-	}
-	}
-
-
-	const fetchContactUsernames = async (...jids) => {
-		const usyncQuery = new USyncQuery().withUsernameProtocol()
-		for (const jid of jids) {
-			usyncQuery.withUser(new USyncUser().withId(jid))
-	}
-		const result = await executeUSyncQuery(usyncQuery)
-		return result?.list?? []
-	}
-
-	const checkUsernameMulti = async usernames => {
-		const data = await mexQuery(
-			{ usernames },
-			USERNAME_QUERY_IDS.CHECK_MULTI,
-			'xwa2_username_check_multi'
-	)
-		return data
-	}
-
-
-
-	const getUsernameRecommendations = async (source = null) => {
-		const variables = {}
-		if (source) variables.source = source
-		return mexQuery(variables, USERNAME_QUERY_IDS.GET_RECOMMENDATIONS, 'xwa2_username_get_recommendations')
-	}
-
-	return {
-	    ...sock,
-		checkUsername,
-		checkUsernameMulti,
-		setUsername,
-		deleteUsername,
-		getMyUsername,
-		getUsernameRecommendations,
-		setUsernamePin,
-		findUserByUsername,
-		fetchContactUsernames,
-		USERNAME_QUERY_IDS,
-		USERNAME_CHECK_RESULT,
-		USERNAME_SOURCE
-	}
+  const sock = makeNewsletterSocket(config)
+  const { query, generateMessageTag, executeUSyncQuery } = sock
+
+  const mexQuery = (variables, queryId, dataPath) =>
+    executeWMexQuery(variables, queryId, dataPath, query, generateMessageTag)
+
+  /**
+   * Check whether a username is available.
+   *
+   * @param {string} username - The @username to check (without @)
+   * @param {boolean} includeSuggestions - Request alternative suggestions when taken
+   * @returns {object}
+   * On success: { available: true, username }
+   * On taken: { available: false, username, suggestions: string[], rejectionReasons: string[], suggestionsEligible: boolean }
+   * On rate-limit: throws Boom with statusCode 429
+   *
+   * Confirmed fields from C164057Wg.java:
+   * data.xwa2_username_check.result → 'SUCCESS' | 'INVALID'
+   * data.xwa2_username_check.suggestions
+   * data.xwa2_username_check.rejection_reasons
+   * data.xwa2_username_check.suggestions_eligible
+   */
+  const checkUsername = async (username, includeSuggestions = true, sessionId) => {
+    // session_id ties this check to the subsequent set step — caller can pass one in
+    // so both use the same UUID (confirmed from live WA 2.26.26.4 capture).
+    const session_id = sessionId || crypto.randomUUID()
+    const data = await mexQuery(
+      {
+        username,
+        include_suggestions: includeSuggestions,
+        session_id,
+        source: USERNAME_SOURCE.USER_INPUT
+      },
+      USERNAME_QUERY_IDS.CHECK,
+      'xwa2_username_check'
+    )
+    if (data?.includes(USERNAME_CHECK_RESULT.SUCCESS) || data?.result === USERNAME_CHECK_RESULT.SUCCESS) {
+      return { available: true, username, session_id }
+    }
+    return {
+      available: false,
+      username,
+      session_id,
+      suggestions: data?.suggestions?? [],
+      rejectionReasons: data?.rejection_reasons?? [],
+      suggestionsEligible: data?.suggestions_eligible?? true
+    }
+  }
+
+  /**
+   * Set your own username.
+   *
+   * @param {string} username - The username to set (without @)
+   * @param {object} options
+   * @param {string} [options.source] - 'USER_INPUT' | 'FB' | 'IG' | 'SUGGESTION'
+   * @param {string} [options.sessionId] - Optional session tracking ID
+   * @param {string} [options.pin] - Optional PIN for protected usernames
+   *
+   * Confirmed variables from C1568872p.java A00():
+   * username, reserved (bool), session_id, source, pin
+   */
+  const setUsername = async (username, options = {}) => {
+    const { source = USERNAME_SOURCE.USER_INPUT, sessionId, pin } = options
+    // session_id links the check and set steps — generate one if caller doesn't supply it.
+    // reserved:true confirmed from live WA 2.26.26.4 capture (was wrongly false before).
+    const session_id = sessionId || crypto.randomUUID()
+    const variables = {
+      username,
+      reserved: true,
+      source,
+      session_id,
+     ...(pin? { pin } : {})
+    }
+    return mexQuery(variables, USERNAME_QUERY_IDS.SET, 'xwa2_username_set')
+  }
+  
+  const checkAndSetUsername = async (username) => {
+    const r = await checkUsername(username);
+    if (r.available) {
+      return await setUsername(username)
+    }
+    return r
+  }
+
+  /**
+   * Delete (unset) your own username.
+   *
+   * Confirmed from C1568872p.java:
+   * str4 = str == null? "delete" : "set"
+   * → sending username=null triggers the delete path on the server.
+   */
+  const deleteUsername = async () => {
+    return mexQuery({ username: null }, USERNAME_QUERY_IDS.SET, 'xwa2_username_delete')
+  }
+
+  /**
+   * Get your own current username.
+   *
+   * Confirmed from C1568872p.java A02():
+   * AbstractC41851rT.A0L(AbstractC130045pa.A0T(), C1363664w.class, "UsernameGet", false)
+   */
+  const getMyUsername = async () => {
+    const data = await mexQuery({}, USERNAME_QUERY_IDS.GET, 'xwa2_username_get')
+    return data?.username?? null
+  }
+
+  /**
+   * Set or delete the PIN that protects your username.
+   *
+   * @param {string|null} pin - New PIN, or null to delete the PIN
+   *
+   * Confirmed from MexUsernamePinProtocolApi.java:
+   * operation "UsernamePinSet", variable "pin"
+   * pin=null triggers the "delete" path on the server.
+   */
+  const setUsernamePin = async pin => {
+    // pin=null → delete PIN (server interprets empty/null variables as delete)
+    const variables = pin!= null? { pin } : {}
+    return mexQuery(variables, USERNAME_QUERY_IDS.PIN_SET, 'xwa2_username_pin_set')
+  }
+
+  /**
+   * Look up a contact by their @username via USync.
+   *
+   * Confirmed via USyncContactProtocol.getUserElement():
+   * { tag: 'contact', attrs: { username, pin? } }
+   *
+   * @param {string} username - The username to look up (without @)
+   * @param {string} [pin] - Optional PIN if the username is PIN-protected
+   * @returns {{ jid, lid?, contact: boolean }|null}
+   */
+  const findUserByUsername = async (username, pin) => {
+    const usyncQuery = new USyncQuery().withContactProtocol()
+    const user = new USyncUser().withUsername(username)
+    if (pin) user.withUsernameKey(pin)
+    usyncQuery.withUser(user)
+    const result = await executeUSyncQuery(usyncQuery)
+    if (!result?.list?.length) return null
+    const entry = result.list[0]
+    return {
+      jid: entry.id,
+      contact: entry.contact?? false
+    }
+  }
+
+  /**
+   * Fetch the username of one or more contacts by their JID.
+   * Uses USync with the username protocol.
+   *
+   * @param {...string} jids - One or more JIDs
+   * @returns {Array<{ id, username: string|null }>}
+   */
+  const fetchContactUsernames = async (...jids) => {
+    const usyncQuery = new USyncQuery().withUsernameProtocol()
+    for (const jid of jids) {
+      usyncQuery.withUser(new USyncUser().withId(jid))
+    }
+    const result = await executeUSyncQuery(usyncQuery)
+    return result?.list?? []
+  }
+
+  /**
+   * Check multiple usernames for availability at once.
+   * @param {string[]} usernames - Array of usernames (without @)
+   */
+  const checkUsernameMulti = async usernames => {
+    const data = await mexQuery(
+      { usernames },
+      USERNAME_QUERY_IDS.CHECK_MULTI,
+      'xwa2_username_check_multi'
+    )
+    return data
+  }
+
+  /**
+   * Fetch username recommendations for the current user.
+   * @param {string} [source] - Source hint: 'FB' | 'IG' | 'USER_INPUT'
+   */
+  const getUsernameRecommendations = async (source = null) => {
+    const variables = {}
+    if (source) variables.source = source
+    return mexQuery(variables, USERNAME_QUERY_IDS.GET_RECOMMENDATIONS, 'xwa2_username_get_recommendations')
+  }
+
+  return {
+   ...sock,
+    checkUsername,
+    checkUsernameMulti,
+    setUsername,
+    deleteUsername,
+    getMyUsername,
+    getUsernameRecommendations,
+    setUsernamePin,
+    findUserByUsername,
+    fetchContactUsernames,
+    USERNAME_QUERY_IDS,
+    USERNAME_CHECK_RESULT,
+    USERNAME_SOURCE
+  }
 }

package/lib/Store/make-in-memory-store.js

@@ -1,3 +1,4 @@
+import fs from "fs";
 import * as WAProto_1 from "../../WAProto/index.js";
 import * as Defaults_1 from "../Defaults/index.js";
 import { LabelAssociationType } from "../Types/LabelAssociation.js";
@@ -181,6 +182,48 @@ export default function makeInMemoryStore(config) {
     const loadMessage = async (jid, id) => {
         return messages[jid]?.get(id)
     }
+
+    /**
+     * Persist the whole store to a JSON file on disk (synchronous, atomic write).
+     * Writes to a temp file first then renames, so a crash mid-write can't
+     * corrupt the existing store file.
+     */
+    const writeToFile = (path) => {
+        try {
+            const tmpPath = `${path}.tmp`;
+            fs.writeFileSync(tmpPath, JSON.stringify(toJSON()));
+            fs.renameSync(tmpPath, path);
+        } catch (err) {
+            logger.error({ err }, "failed to write store to file");
+        }
+    };
+
+    /**
+     * Load store state from a JSON file on disk, if it exists.
+     * Safe to call even if the file was never created yet.
+     */
+    const readFromFile = (path) => {
+        if (!fs.existsSync(path)) return;
+        try {
+            const jsonStr = fs.readFileSync(path, { encoding: "utf-8" });
+            const json = JSON.parse(jsonStr);
+            fromJSON(json);
+        } catch (err) {
+            logger.error({ err }, "failed to read store from file");
+        }
+    };
+
+    /**
+     * Convenience helper: load once from `path` (if present), then keep
+     * saving to the same file every `intervalMs` (default 10s).
+     * Returns a function to stop the auto-save interval.
+     */
+    const writeToFileInterval = (path, intervalMs = 10_000) => {
+        readFromFile(path);
+        const interval = setInterval(() => writeToFile(path), intervalMs);
+        return () => clearInterval(interval);
+    };
+
     return {
         chats,
         contacts,
@@ -193,6 +236,9 @@ export default function makeInMemoryStore(config) {
         bind,
         toJSON,
         fromJSON,
-        loadMessage
+        loadMessage,
+        writeToFile,
+        readFromFile,
+        writeToFileInterval
     };
 }

package/lib/Types/Mex.js

@@ -16,12 +16,48 @@ export var XWAPaths;
     XWAPaths["xwa2_newsletter_delete_v2"] = "xwa2_newsletter_delete_v2";
     XWAPaths["xwa2_fetch_account_reachout_timelock"] = "xwa2_fetch_account_reachout_timelock";
     XWAPaths["xwa2_message_capping_info"] = "xwa2_message_capping_info";
+    XWAPaths["xwa2_newsletter_subscribed"] = "xwa2_newsletter_subscribed";
+    XWAPaths["xwa2_newsletter_update"] = "xwa2_newsletter_update";
+    XWAPaths["xwa2_newsletter_admin"] = "xwa2_newsletter_admin";
+    XWAPaths["xwa2_newsletter_admin_invite_create"] = "xwa2_newsletter_admin_invite_create";
+    XWAPaths["xwa2_newsletter_admin_invite_revoke"] = "xwa2_newsletter_admin_invite_revoke";
+    XWAPaths["xwa2_newsletter_admin_invite_accept"] = "xwa2_newsletter_admin_invite_accept";
+    XWAPaths["xwa2_newsletter_admin_profile_update"] = "xwa2_newsletter_admin_profile_update";
+    XWAPaths["xwa2_newsletter_admin_insights"] = "xwa2_newsletter_admin_insights";
+    XWAPaths["xwa2_newsletter_following"] = "xwa2_newsletter_following";
+    XWAPaths["xwa2_newsletter_create_verified"] = "xwa2_newsletter_create_verified";
+    XWAPaths["xwa2_newsletter_enforcements"] = "xwa2_newsletter_enforcements";
+    XWAPaths["xwa2_newsletter_user_reports"] = "xwa2_newsletter_user_reports";
+    XWAPaths["xwa2_newsletter_create_report_appeal"] = "xwa2_newsletter_create_report_appeal";
+    XWAPaths["xwa2_newsletter_link_preview_check"] = "xwa2_newsletter_link_preview_check";
+    XWAPaths["xwa2_newsletter_update_verification"] = "xwa2_newsletter_update_verification";
+    XWAPaths["xwa2_newsletter_label_paid_partnership"] = "xwa2_newsletter_label_paid_partnership";
+    XWAPaths["xwa2_newsletter_log_exposures"] = "xwa2_newsletter_log_exposures";
+    XWAPaths["xwa2_newsletter_update_user_setting"] = "xwa2_newsletter_update_user_setting";
+    XWAPaths["xwa2_newsletter_ranking_features"] = "xwa2_newsletter_ranking_features";
+    XWAPaths["xwa2_newsletter_block_user"] = "xwa2_newsletter_block_user";
+    XWAPaths["xwa2_newsletters_directory_list"] = "xwa2_newsletters_directory_list";
+    XWAPaths["xwa2_newsletters_directory_search"] = "xwa2_newsletters_directory_search";
+    XWAPaths["xwa2_newsletters_directory_category_preview"] = "xwa2_newsletters_directory_category_preview";
+    XWAPaths["xwa2_newsletters_search"] = "xwa2_newsletters_search";
+    XWAPaths["xwa2_newsletters_recommended"] = "xwa2_newsletters_recommended";
+    XWAPaths["xwa2_newsletters_similar"] = "xwa2_newsletters_similar";
+    XWAPaths["xwa2_newsletters_poll_voter_list"] = "xwa2_newsletters_poll_voter_list";
+    XWAPaths["xwa2_newsletters_reaction_sender_list"] = "xwa2_newsletters_reaction_sender_list";
+    XWAPaths["xwa2_wamo_afs_age_collection"] = "xwa2_wamo_afs_age_collection";
+    XWAPaths["xwa2_wamo_asset_collection"] = "xwa2_wamo_asset_collection";
+    XWAPaths["xwa2_wamo_fetch_adhoc_notice_by_id"] = "xwa2_wamo_fetch_adhoc_notice_by_id";
+    XWAPaths["xwa2_wamo_fetch_identity_token"] = "xwa2_wamo_fetch_identity_token";
+    XWAPaths["xwa2_wamo_sub_get_compliance_info"] = "xwa2_wamo_sub_get_compliance_info";
+    XWAPaths["xwa2_wamo_user_id_version"] = "xwa2_wamo_user_id_version";
+    XWAPaths["xwa2_wamo_set_user_id_version"] = "xwa2_wamo_set_user_id_version";
 })(XWAPaths || (XWAPaths = {}));
+
 export var QueryIds;
 (function (QueryIds) {
     QueryIds["CREATE"] = "8823471724422422";
-    QueryIds["UPDATE_METADATA"] = "24250201037901610";
-    QueryIds["METADATA"] = "6563316087068696";
+    QueryIds["METADATA"] = "26099060629688646";
+    QueryIds["UPDATE_METADATA"] = "25976052865358570";
     QueryIds["SUBSCRIBERS"] = "9783111038412085";
     QueryIds["FOLLOW"] = "24404358912487870";
     QueryIds["UNFOLLOW"] = "9767147403369991";
@@ -33,5 +69,43 @@ export var QueryIds;
     QueryIds["DELETE"] = "30062808666639665";
     QueryIds["REACHOUT_TIMELOCK"] = "23983697327930364";
     QueryIds["MESSAGE_CAPPING_INFO"] = "24503548349331633";
+    QueryIds["FETCH_SUBSCRIBE"] = "6388546374527196";
+    QueryIds["LEAVE"] = "9784043202933196";
+    QueryIds["ADMIN_INVITE"] = "7148599488585196";
+    QueryIds["ADMIN_INVITE_REVOKE"] = "6345824195482235";
+    QueryIds["ADMIN_INVITE_ACCEPT"] = "6345824195482236";
+    QueryIds["ADMIN_METADATA"] = "6895563180530100";
+    QueryIds["ADMIN_PROFILE_UPDATE"] = "6895563180530101";
+    QueryIds["INSIGHTS"] = "6243647912368353";
+    QueryIds["FOLLOWING_LIST"] = "6388546374527197";
+    QueryIds["CREATE_VERIFIED"] = "8823471724422423";
+    QueryIds["ENFORCEMENTS"] = "7130823597031707";
+    QueryIds["USER_REPORTS"] = "9783111038412086";
+    QueryIds["CREATE_REPORT_APPEAL"] = "6551828931592904";
+    QueryIds["LINK_PREVIEW_CHECK"] = "24404358912487871";
+    QueryIds["UPDATE_VERIFICATION"] = "25976052865358571";
+    QueryIds["LABEL_PAID_PARTNERSHIP"] = "7341777602580934";
+    QueryIds["LOG_EXPOSURES"] = "9864994326891138";
+    QueryIds["UPDATE_USER_SETTING"] = "29766401636284407";
+    QueryIds["RANKING_FEATURES"] = "7130823597031708";
+    QueryIds["BLOCK_USER"] = "6345824195482237";
+    QueryIds["DIRECTORY_LIST"] = "6388546374527198";
+    QueryIds["DIRECTORY_SEARCH"] = "6388546374527199";
+    QueryIds["DIRECTORY_CATEGORY_PREVIEW"] = "6388546374527200";
+    QueryIds["SEARCH"] = "26099060629688647";
+    QueryIds["RECOMMENDED"] = "26099060629688648";
+    QueryIds["SIMILAR"] = "26099060629688649";
+    QueryIds["POLL_VOTER_LIST"] = "9783111038412087";
+    QueryIds["REACTION_SENDERS_LIST"] = "9783111038412088";
+    QueryIds["WAMO_ENABLE_SUB"] = "7148599488585197";
+    QueryIds["WAMO_DISABLE_SUB"] = "7148599488585198";
+    QueryIds["WAMO_CHANGE_SUB"] = "7148599488585199";
+    QueryIds["WAMO_AFS_AGE_COLLECTION"] = "6895563180530102";
+    QueryIds["WAMO_ASSET_COLLECTION"] = "6895563180530103";
+    QueryIds["WAMO_FETCH_ADHOC_NOTICE"] = "6895563180530104";
+    QueryIds["WAMO_FETCH_IDENTITY_TOKEN"] = "6895563180530105";
+    QueryIds["WAMO_SUB_COMPLIANCE_INFO"] = "6895563180530106";
+    QueryIds["WAMO_USER_ID_VERSION"] = "6895563180530107";
+    QueryIds["WAMO_SET_USER_ID_VERSION"] = "6895563180530108";
 })(QueryIds || (QueryIds = {}));
 //# sourceMappingURL=Mex.js.map

package/lib/Utils/index.js

@@ -11,6 +11,7 @@ export * from './chat-utils.js';
 export * from './lt-hash.js';
 export * from './auth-utils.js';
 export * from './use-multi-file-auth-state.js';
+export * from './use-sqlite-auth-state.js';
 export * from './link-preview.js';
 export * from './event-buffer.js';
 export * from './process-message.js';

package/lib/Utils/messages-media.js

@@ -778,6 +778,88 @@ export const decryptMediaRetryData = ({ ciphertext, iv }, mediaKey, msgId) => {
     const plaintext = aesDecryptGCM(ciphertext, retryKey, iv, Buffer.from(msgId));
     return proto.MediaRetryNotification.decode(plaintext);
 };
+const CRC_TABLE = (() => {
+    const table = new Uint32Array(256);
+    for (let n = 0; n < 256; n++) {
+        let c = n;
+        for (let k = 0; k < 8; k++) {
+            c = c & 1 ? (0xedb88320 ^ (c >>> 1)) : (c >>> 1);
+        }
+        table[n] = c >>> 0;
+    }
+    return table;
+})();
+export const crc32 = (buf) => {
+    let crc = 0xffffffff;
+    for (let i = 0; i < buf.length; i++) {
+        crc = CRC_TABLE[(crc ^ buf[i]) & 0xff] ^ (crc >>> 8);
+    }
+    return (crc ^ 0xffffffff) >>> 0;
+};
+export const buildZipArchive = (files) => {
+    const localChunks = [];
+    const centralChunks = [];
+    let offset = 0;
+    for (const { name, data } of files) {
+        const nameBuf = Buffer.from(name, 'utf8');
+        const crc = crc32(data);
+        const dosTime = 0;
+        const dosDate = 0x21;
+        const localHeader = Buffer.alloc(30);
+        localHeader.writeUInt32LE(0x04034b50, 0);
+        localHeader.writeUInt16LE(20, 4);
+        localHeader.writeUInt16LE(0, 6);
+        localHeader.writeUInt16LE(0, 8);
+        localHeader.writeUInt16LE(dosTime, 10);
+        localHeader.writeUInt16LE(dosDate, 12);
+        localHeader.writeUInt32LE(crc, 14);
+        localHeader.writeUInt32LE(data.length, 18);
+        localHeader.writeUInt32LE(data.length, 22);
+        localHeader.writeUInt16LE(nameBuf.length, 26);
+        localHeader.writeUInt16LE(0, 28);
+        localChunks.push(localHeader, nameBuf, data);
+        const centralHeader = Buffer.alloc(46);
+        centralHeader.writeUInt32LE(0x02014b50, 0);
+        centralHeader.writeUInt16LE(20, 4);
+        centralHeader.writeUInt16LE(20, 6);
+        centralHeader.writeUInt16LE(0, 8);
+        centralHeader.writeUInt16LE(0, 10);
+        centralHeader.writeUInt16LE(dosTime, 12);
+        centralHeader.writeUInt16LE(dosDate, 14);
+        centralHeader.writeUInt32LE(crc, 16);
+        centralHeader.writeUInt32LE(data.length, 20);
+        centralHeader.writeUInt32LE(data.length, 24);
+        centralHeader.writeUInt16LE(nameBuf.length, 28);
+        centralHeader.writeUInt16LE(0, 30);
+        centralHeader.writeUInt16LE(0, 32);
+        centralHeader.writeUInt16LE(0, 34);
+        centralHeader.writeUInt16LE(0, 36);
+        centralHeader.writeUInt32LE(0, 38);
+        centralHeader.writeUInt32LE(offset, 42);
+        centralChunks.push(centralHeader, nameBuf);
+        offset += localHeader.length + nameBuf.length + data.length;
+    }
+    const centralDirStart = offset;
+    const centralDirBuf = Buffer.concat(centralChunks);
+    const eocd = Buffer.alloc(22);
+    eocd.writeUInt32LE(0x06054b50, 0);
+    eocd.writeUInt16LE(0, 4);
+    eocd.writeUInt16LE(0, 6);
+    eocd.writeUInt16LE(files.length, 8);
+    eocd.writeUInt16LE(files.length, 10);
+    eocd.writeUInt32LE(centralDirBuf.length, 12);
+    eocd.writeUInt32LE(centralDirStart, 16);
+    eocd.writeUInt16LE(0, 20);
+    return Buffer.concat([...localChunks, centralDirBuf, eocd]);
+};
+export const resolveStickerBuffer = async (item, opts) => {
+    if (Buffer.isBuffer(item)) {
+        return item;
+    }
+    const media = typeof item === 'string' ? { url: item } : item;
+    const { stream } = await getStream(media, opts);
+    return toBuffer(stream);
+};
 export const getStatusCodeForMediaRetry = (code) => MEDIA_RETRY_STATUS_MAP[code];
 const MEDIA_RETRY_STATUS_MAP = {
     [proto.MediaRetryNotification.ResultType.SUCCESS]: 200,