@badzz88/baileys 8.4.7 → 8.5.0

package/lib/Socket/Client/websocket.js

@@ -1,54 +1,62 @@
-import WebSocket from 'ws';
-import { DEFAULT_ORIGIN } from '../../Defaults/index.js';
-import { AbstractSocketClient } from './types.js';
-export class WebSocketClient extends AbstractSocketClient {
-    constructor() {
-        super(...arguments);
-        this.socket = null;
-    }
-    get isOpen() {
-        return this.socket?.readyState === WebSocket.OPEN;
-    }
-    get isClosed() {
-        return this.socket === null || this.socket?.readyState === WebSocket.CLOSED;
-    }
-    get isClosing() {
-        return this.socket === null || this.socket?.readyState === WebSocket.CLOSING;
-    }
-    get isConnecting() {
-        return this.socket?.readyState === WebSocket.CONNECTING;
-    }
-    connect() {
-        if (this.socket) {
-            return;
-        }
-        this.socket = new WebSocket(this.url, {
-            origin: DEFAULT_ORIGIN,
-            headers: this.config.options?.headers,
-            handshakeTimeout: this.config.connectTimeoutMs,
-            timeout: this.config.connectTimeoutMs,
-            agent: this.config.agent
-        });
-        this.socket.setMaxListeners(0);
-        const events = ['close', 'error', 'upgrade', 'message', 'open', 'ping', 'pong', 'unexpected-response'];
-        for (const event of events) {
-            this.socket?.on(event, (...args) => this.emit(event, ...args));
-        }
-    }
-    async close() {
-        if (!this.socket) {
-            return;
-        }
-        const closePromise = new Promise(resolve => {
-            this.socket?.once('close', resolve);
-        });
-        this.socket.close();
-        await closePromise;
-        this.socket = null;
-    }
-    send(str, cb) {
-        this.socket?.send(str, cb);
-        return Boolean(this.socket);
-    }
+'use strict'
+var __importDefault =
+	(this && this.__importDefault) ||
+	function (mod) {
+		return mod && mod.__esModule ? mod : { default: mod }
+	}
+Object.defineProperty(exports, '__esModule', { value: true })
+exports.WebSocketClient = void 0
+const ws_1 = __importDefault(require('ws'))
+const Defaults_1 = require('../../Defaults')
+const types_1 = require('./types')
+class WebSocketClient extends types_1.AbstractSocketClient {
+	constructor() {
+		super(...arguments)
+		this.socket = null
+	}
+	get isOpen() {
+		return this.socket?.readyState === ws_1.default.OPEN
+	}
+	get isClosed() {
+		return this.socket === null || this.socket?.readyState === ws_1.default.CLOSED
+	}
+	get isClosing() {
+		return this.socket === null || this.socket?.readyState === ws_1.default.CLOSING
+	}
+	get isConnecting() {
+		return this.socket?.readyState === ws_1.default.CONNECTING
+	}
+	connect() {
+		if (this.socket) {
+			return
+		}
+		this.socket = new ws_1.default(this.url, {
+			origin: Defaults_1.DEFAULT_ORIGIN,
+			headers: this.config.options?.headers,
+			handshakeTimeout: this.config.connectTimeoutMs,
+			timeout: this.config.connectTimeoutMs,
+			agent: this.config.agent
+		})
+		this.socket.setMaxListeners(0)
+		const events = ['close', 'error', 'upgrade', 'message', 'open', 'ping', 'pong', 'unexpected-response']
+		for (const event of events) {
+			this.socket?.on(event, (...args) => this.emit(event, ...args))
+		}
+	}
+	async close() {
+		if (!this.socket) {
+			return
+		}
+		const closePromise = new Promise(resolve => {
+			this.socket?.once('close', resolve)
+		})
+		this.socket.close()
+		await closePromise
+		this.socket = null
+	}
+	send(str, cb) {
+		this.socket?.send(str, cb)
+		return Boolean(this.socket)
+	}
 }
-//# sourceMappingURL=websocket.js.map
+exports.WebSocketClient = WebSocketClient

package/lib/Socket/aigroups.js

@@ -0,0 +1,240 @@
+'use strict'
+Object.defineProperty(exports, '__esModule', { value: true })
+exports.extractAIGroupMetadata = exports.makeAIGroupsSocket = void 0
+const Types_1 = require('../Types')
+const Utils_1 = require('../Utils')
+const WABinary_1 = require('../WABinary')
+const groups_1 = require('./groups')
+
+const makeAIGroupsSocket = config => {
+	const sock = (0, groups_1.makeGroupsSocket)(config)
+	const { ev, query } = sock
+
+	/** Query helper for AI groups (w:gp2 namespace) */
+	const aiGroupQuery = async (jid, type, content) =>
+		query({
+			tag: 'iq',
+			attrs: {
+				type,
+				xmlns: 'w:g2',
+				to: jid
+			},
+			content
+		})
+
+	const aiGroupMetadata = async jid => {
+		const result = await aiGroupQuery(jid, 'get', [{ tag: 'query', attrs: { request: 'interactive' } }])
+		return (0, exports.extractAIGroupMetadata)(result)
+	}
+
+	// Handle incoming w:gp2 notifications (create, promote, remove, add)
+	sock.ws.on('CB:notification,w:gp2', async node => {
+		const { attrs, content } = node
+		if (!Array.isArray(content) || content.length === 0) return
+
+		const inner = content[0]
+		const tag = inner.tag
+		const groupId =
+			typeof attrs.from === 'string'
+				? attrs.from
+				: attrs.from?.$1?.user
+					? (0, WABinary_1.jidEncode)(attrs.from.$1.user, 'g.us')
+					: undefined
+
+		if (!groupId) return
+
+		if (tag === 'create') {
+			// New AI group created — emit groups.upsert
+			try {
+				const meta = await aiGroupMetadata(groupId)
+				ev.emit('groups.upsert', [meta])
+			} catch {
+				// metadata fetch may fail; emit minimal info
+				ev.emit('groups.upsert', [{ id: groupId }])
+			}
+		} else if (tag === 'promote' || tag === 'demote' || tag === 'remove' || tag === 'add') {
+			const participants = (0, WABinary_1.getBinaryNodeChildren)(inner, 'participant')
+				.map(p => {
+					const jid = p.attrs.jid
+					if (typeof jid === 'string') return jid
+					if (jid?.$1) {
+						return (0, WABinary_1.jidEncode)(jid.$1.user, jid.$1.server || 's.whatsapp.net')
+					}
+					return undefined
+				})
+				.filter(Boolean)
+
+			ev.emit('group-participants.update', {
+				id: groupId,
+				participants,
+				action: tag
+			})
+		} else if (tag === 'subject') {
+			ev.emit('groups.update', [{ id: groupId, subject: inner.attrs?.subject }])
+		}
+
+		await sock.sendMessageAck(node)
+	})
+
+	return {
+		...sock,
+		aiGroupMetadata,
+		aiGroupCreate: async (subject, participants = [], options = {}) => {
+			if (!Array.isArray(participants)) participants = []
+			const key = (0, Utils_1.generateMessageIDV2)()
+			const {
+				ephemeralExpiration = 86400,
+				memberAddMode = 'all_member_add',
+				memberShareGroupHistoryMode = 'all_member_share',
+				memberLinkMode = 'all_member_link'
+			} = options
+			const result = await aiGroupQuery('@g.us', 'set', [
+				{
+					tag: 'create',
+					attrs: {
+						subject,
+						key
+					},
+					content: participants.map(jid => ({
+						tag: 'participant',
+						attrs: { jid }
+					}))
+				}
+			])
+			return (0, exports.extractAIGroupMetadata)(result)
+		},
+		aiGroupAddBot: async (jid, botUser = '867051314767696') => {
+			const result = await aiGroupQuery(jid, 'set', [
+				{
+					tag: 'add',
+					attrs: {},
+					content: [
+						{
+							tag: 'participant',
+							attrs: { jid: `${botUser}@bot` }
+						}
+					]
+				}
+			])
+			const node = (0, WABinary_1.getBinaryNodeChild)(result, 'add')
+			const participantsAffected = (0, WABinary_1.getBinaryNodeChildren)(node, 'participant')
+			return participantsAffected.map(p => ({
+				status: p.attrs.error || '200',
+				jid: p.attrs.jid
+			}))
+		},
+		aiGroupLeave: async id => {
+			await aiGroupQuery('@g.us', 'set', [
+				{
+					tag: 'leave',
+					attrs: {},
+					content: [{ tag: 'group', attrs: { id } }]
+				}
+			])
+		},
+		aiGroupParticipantsUpdate: async (jid, participants, action) => {
+			const result = await aiGroupQuery(jid, 'set', [
+				{
+					tag: action,
+					attrs: {},
+					content: participants.map(jid => ({
+						tag: 'participant',
+						attrs: { jid }
+					}))
+				}
+			])
+			const node = (0, WABinary_1.getBinaryNodeChild)(result, action)
+			const participantsAffected = (0, WABinary_1.getBinaryNodeChildren)(node, 'participant')
+			return participantsAffected.map(p => ({
+				status: p.attrs.error || '200',
+				jid: p.attrs.jid,
+				content: p
+			}))
+		},
+		aiGroupUpdateSubject: async (jid, subject) => {
+			await aiGroupQuery(jid, 'set', [
+				{
+					tag: 'subject',
+					attrs: {},
+					content: Buffer.from(subject, 'utf-8')
+				}
+			])
+		},
+		aiGroupInviteCode: async jid => {
+			const result = await aiGroupQuery(jid, 'get', [{ tag: 'invite', attrs: {} }])
+			const inviteNode = (0, WABinary_1.getBinaryNodeChild)(result, 'invite')
+			return inviteNode?.attrs.code
+		},
+		aiGroupRevokeInvite: async jid => {
+			const result = await aiGroupQuery(jid, 'set', [{ tag: 'invite', attrs: {} }])
+			const inviteNode = (0, WABinary_1.getBinaryNodeChild)(result, 'invite')
+			return inviteNode?.attrs.code
+		},
+		aiGroupAcceptInvite: async code => {
+			const results = await aiGroupQuery('@g.us', 'set', [{ tag: 'invite', attrs: { code } }])
+			const result = (0, WABinary_1.getBinaryNodeChild)(results, 'group')
+			return result?.attrs.jid
+		},
+		aiGroupSettingUpdate: async (jid, setting) => {
+			await aiGroupQuery(jid, 'set', [{ tag: setting, attrs: {} }])
+		},
+		aiGroupToggleEphemeral: async (jid, ephemeralExpiration) => {
+			const content = ephemeralExpiration
+				? { tag: 'ephemeral', attrs: { expiration: ephemeralExpiration.toString() } }
+				: { tag: 'not_ephemeral', attrs: {} }
+			await aiGroupQuery(jid, 'set', [content])
+		}
+	}
+}
+exports.makeAIGroupsSocket = makeAIGroupsSocket
+
+const extractAIGroupMetadata = result => {
+	// IQ response nests group inside create node: result → create → group
+	const createNode = (0, WABinary_1.getBinaryNodeChild)(result, 'create')
+	const group =
+		(0, WABinary_1.getBinaryNodeChild)(createNode || result, 'group') ||
+		(0, WABinary_1.getBinaryNodeChild)(result, 'group')
+	const descChild = (0, WABinary_1.getBinaryNodeChild)(group, 'description')
+	let desc, descId, descOwner, descOwnerPn, descTime
+	if (descChild) {
+		desc = (0, WABinary_1.getBinaryNodeChildString)(descChild, 'body')
+		descOwner = descChild.attrs.participant ? (0, WABinary_1.jidNormalizedUser)(descChild.attrs.participant) : undefined
+		descOwnerPn = descChild.attrs.participant_pn
+			? (0, WABinary_1.jidNormalizedUser)(descChild.attrs.participant_pn)
+			: undefined
+		descTime = +descChild.attrs.t
+		descId = descChild.attrs.id
+	}
+	const groupId = group.attrs.id.includes('@') ? group.attrs.id : (0, WABinary_1.jidEncode)(group.attrs.id, 'g.us')
+	const eph = (0, WABinary_1.getBinaryNodeChild)(group, 'ephemeral')?.attrs.expiration
+	const metadata = {
+		id: groupId,
+		subject: group.attrs.subject,
+		subjectTime: +group.attrs.s_t,
+		creation: +group.attrs.creation,
+		owner: group.attrs.creator ? (0, WABinary_1.jidNormalizedUser)(group.attrs.creator) : undefined,
+		ownerPn: group.attrs.creator_pn ? (0, WABinary_1.jidNormalizedUser)(group.attrs.creator_pn) : undefined,
+		owner_country_code: group.attrs.creator_country_code,
+		size: group.attrs.size ? +group.attrs.size : (0, WABinary_1.getBinaryNodeChildren)(group, 'participant').length,
+		desc,
+		descId,
+		descOwner,
+		descOwnerPn,
+		descTime,
+		isAIGroup: true,
+		addressingMode:
+			group.attrs.addressing_mode === 'lid' ? Types_1.WAMessageAddressingMode.LID : Types_1.WAMessageAddressingMode.PN,
+		participants: (0, WABinary_1.getBinaryNodeChildren)(group, 'participant').map(({ attrs }) => ({
+			id: attrs.jid,
+			phoneNumber:
+				(0, WABinary_1.isLidUser)(attrs.jid) && (0, WABinary_1.isPnUser)(attrs.phone_number)
+					? attrs.phone_number
+					: undefined,
+			lid: (0, WABinary_1.isPnUser)(attrs.jid) && (0, WABinary_1.isLidUser)(attrs.lid) ? attrs.lid : undefined,
+			admin: attrs.type || null
+		})),
+		ephemeralDuration: eph ? +eph : undefined
+	}
+	return metadata
+}
+exports.extractAIGroupMetadata = extractAIGroupMetadata