@kyyinfinite/baileys 1.0.0 → 2.1.0

package/lib/Kyyinfinite/Foad.js

@@ -0,0 +1,542 @@
+import crypto from 'crypto';
+import { generateWAMessageFromContent } from '../Utils/messages.js';
+import { generateMessageIDV2 } from '../Utils/generics.js';
+import { BaseBuilder } from './BaseBuilder.js';
+import { Toolkit } from './Toolkit.js';
+import { AIRich } from './AIRich.js';
+
+class Foad extends BaseBuilder {
+	#client;
+
+	constructor(client, { dynamic = true } = {}) {
+		super();
+
+		if (!client) {
+			throw new Error('Socket is required');
+		}
+
+		this.#client = client;
+
+		this._dynamic = !!dynamic;
+		this._messageText = '';
+		this._previewTitle = 'Preview';
+
+		this._tabs = [];
+
+		this._responseId = crypto.randomUUID();
+		this._botResponseId = crypto.randomUUID();
+
+		this._lastMessageKey = null;
+	}
+
+	/**
+	 * Set response ID
+	 */
+	setResponseId(id) {
+		if (typeof id !== 'string') {
+			throw new TypeError('Response id must be a string');
+		}
+
+		this._responseId = id;
+
+		return this;
+	}
+
+	/**
+	 * Generate new response ID
+	 */
+	refreshResponseId() {
+		this._responseId = crypto.randomUUID();
+
+		return this;
+	}
+
+	/**
+	 * Set bot response ID
+	 */
+	setBotResponseId(id) {
+		if (typeof id !== 'string') {
+			throw new TypeError('Bot response id must be a string');
+		}
+
+		this._botResponseId = id;
+
+		return this;
+	}
+
+	/**
+	 * Generate new bot response ID
+	 */
+	refreshBotResponseId() {
+		this._botResponseId = crypto.randomUUID();
+
+		return this;
+	}
+
+	/**
+	 * Add title / message text
+	 */
+	addTitle(text) {
+		if (typeof text !== 'string') {
+			throw new TypeError('Title must be a string');
+		}
+
+		this._messageText = text;
+
+		return this;
+	}
+
+	/**
+	 * Set FOAD preview title
+	 */
+	setPreviewTitle(title) {
+		if (typeof title !== 'string') {
+			throw new TypeError('Preview title must be a string');
+		}
+
+		this._previewTitle = title;
+
+		return this;
+	}
+
+	/**
+	 * Add HTML widget
+	 *
+	 * This creates:
+	 *
+	 * tabs[]
+	 *   └── sections[]
+	 *       └── GenAIaeacdsnwHtmlPrimitive
+	 */
+	addHtml(
+		html,
+		{
+			title = 'HTML',
+			id,
+			url = 'https://nixel.dev',
+			trustedSources = ['nixel.dev'],
+			stepEntries = []
+		} = {}
+	) {
+		if (typeof html !== 'string') {
+			throw new TypeError('HTML must be a string');
+		}
+
+		if (typeof title !== 'string') {
+			throw new TypeError('Tab title must be a string');
+		}
+
+		if (typeof url !== 'string') {
+			throw new TypeError('URL must be a string');
+		}
+
+		if (!Array.isArray(trustedSources)) {
+			throw new TypeError('trustedSources must be an array');
+		}
+
+		if (!Array.isArray(stepEntries)) {
+			throw new TypeError('stepEntries must be an array');
+		}
+
+		const tabId = id ?? `tab_${this._tabs.length}`;
+
+		if (typeof tabId !== 'string' || !tabId) {
+			throw new TypeError('Tab id must be a non-empty string');
+		}
+
+		if (this._tabs.some(tab => tab.id === tabId)) {
+			throw new Error(`Tab id "${tabId}" already exists`);
+		}
+
+		this._tabs.push({
+			id: tabId,
+
+			tab_header: title,
+
+			sections: [
+				{
+					__typename: 'GenAIUnifiedResponseSection',
+
+					view_model: {
+						__typename: 'GenAISingleLayoutViewModel',
+
+						primitive: {
+							__typename: 'GenAIaeacdsnwHtmlPrimitive',
+
+							payload: html,
+
+							url,
+
+							trusted_sources: trustedSources
+						}
+					}
+				}
+			],
+
+			step_entries: stepEntries
+		});
+
+		return this;
+	}
+
+	/**
+	 * Add raw FOAD tab
+	 *
+	 * Useful when you want full control over
+	 * sections / primitive structure.
+	 */
+	addTab(tab = {}) {
+		if (
+			!tab ||
+			typeof tab !== 'object' ||
+			Array.isArray(tab)
+		) {
+			throw new TypeError('Tab must be a plain object');
+		}
+
+		const {
+			id,
+			tab_header = 'HTML',
+			sections,
+			step_entries = []
+		} = tab;
+
+		if (typeof id !== 'string' || !id) {
+			throw new TypeError('Tab id must be a non-empty string');
+		}
+
+		if (typeof tab_header !== 'string') {
+			throw new TypeError('Tab header must be a string');
+		}
+
+		if (!Array.isArray(sections) || !sections.length) {
+			throw new TypeError(
+				'Tab sections must be a non-empty array'
+			);
+		}
+
+		if (!Array.isArray(step_entries)) {
+			throw new TypeError(
+				'step_entries must be an array'
+			);
+		}
+
+		if (this._tabs.some(item => item.id === id)) {
+			throw new Error(
+				`Tab id "${id}" already exists`
+			);
+		}
+
+		this._tabs.push({
+			id,
+
+			tab_header,
+
+			sections: structuredClone(sections),
+
+			step_entries: structuredClone(step_entries)
+		});
+
+		return this;
+	}
+
+	/**
+	 * Remove all tabs
+	 */
+	clearTabs() {
+		this._tabs = [];
+
+		return this;
+	}
+
+	/**
+	 * Get tabs
+	 */
+	get tabs() {
+		return structuredClone(this._tabs);
+	}
+
+	/**
+	 * Build decoded FOAD response object
+	 *
+	 * Structure:
+	 *
+	 * {
+	 *   response_id,
+	 *   sections,
+	 *   embedded_screens: [
+	 *     {
+	 *       title,
+	 *       content: [
+	 *         {
+	 *           __typename:
+	 *             'FOAIDNixelButtonSheets',
+	 *           tabs: [...]
+	 *         }
+	 *       ]
+	 *     }
+	 *   ]
+	 * }
+	 */
+	buildResponse() {
+		if (!this._tabs.length) {
+			throw new Error(
+				'Foad requires at least one HTML tab'
+			);
+		}
+
+		if (this._dynamic) {
+			this.refreshResponseId();
+			this.refreshBotResponseId();
+		}
+
+		return {
+			response_id: this._responseId,
+
+			sections: [
+				{
+					view_model: {
+						primitive: {
+							text: this._messageText,
+
+							__typename:
+								'GenAIMarkdownTextUXPrimitive'
+						},
+
+						__typename:
+							'GenAISingleLayoutViewModel'
+					}
+				}
+			],
+
+			embedded_screens: [
+				{
+					title: this._previewTitle,
+
+					content: [
+						{
+							__typename:
+								'FOAIDNixelButtonSheets',
+
+							tabs: structuredClone(
+								this._tabs
+							)
+						}
+					]
+				}
+			]
+		};
+	}
+
+	/**
+	 * Build WhatsApp message
+	 */
+	async build(
+		jid,
+		{
+			messageId,
+			...options
+		} = {}
+	) {
+		const responseData =
+			this.buildResponse();
+
+		const dataBase64 =
+			Buffer
+				.from(
+					JSON.stringify(responseData)
+				)
+				.toString('base64');
+
+		return generateWAMessageFromContent(
+			jid,
+
+			{
+				messageContextInfo: {
+					deviceListMetadata: {},
+
+					deviceListMetadataVersion: 2,
+
+					botMetadata: {
+						messageDisclaimerText:
+							this._title,
+
+						botResponseId:
+							this._botResponseId,
+
+						verificationMetadata:
+							AIRich.generateVerificationMetadata()
+					}
+				},
+
+				...this._extraPayload,
+
+				botForwardedMessage: {
+					message: {
+						richResponseMessage: {
+							messageType: 1,
+
+							submessages:
+								this._messageText
+									? [
+											{
+												messageType: 2,
+
+												messageText:
+													this._messageText
+											}
+										]
+									: [],
+
+							unifiedResponse: {
+								data: dataBase64
+							},
+
+							contextInfo: {
+								forwardingScore: 1,
+
+								isForwarded: true,
+
+								forwardedAiBotMessageInfo: {
+									botJid:
+										'867051314767696@bot'
+								},
+
+								forwardOrigin: 4
+							}
+						}
+					}
+				}
+			},
+
+			{
+				messageId:
+					messageId ||
+					generateMessageIDV2(),
+
+				...options
+			}
+		);
+	}
+
+	/**
+	 * Send FOAD
+	 */
+	async send(
+		jid,
+		{
+			messageId,
+			...options
+		} = {}
+	) {
+		const msg = await this.build(
+			jid,
+			{
+				messageId,
+				...options
+			}
+		);
+
+		await this.#client.relayMessage(
+			msg.key.remoteJid,
+			msg.message,
+			{
+				messageId: msg.key.id,
+
+				...options
+			}
+		);
+
+		this._lastMessageKey = msg.key;
+
+		return msg;
+	}
+
+	/**
+	 * Edit previously sent FOAD
+	 */
+	async sendEdit(
+		jid,
+		id,
+		{
+			msg,
+			messageId,
+			...options
+		} = {}
+	) {
+		jid =
+			jid ??
+			this._lastMessageKey?.remoteJid;
+
+		id =
+			id ??
+			this._lastMessageKey?.id;
+
+		if (!jid) {
+			throw new Error('JID is required');
+		}
+
+		if (!id) {
+			throw new Error(
+				'Message id is required'
+			);
+		}
+
+		const source =
+			msg ??
+			(
+				await this.build(
+					jid,
+					options
+				)
+			).message;
+
+		const editedMessage =
+			await generateWAMessageFromContent(
+				jid,
+
+				{
+					botForwardedMessage: {
+						message: {
+							protocolMessage: {
+								key: {
+									remoteJid: jid,
+
+									fromMe: true,
+
+									id
+								},
+
+								type: 14,
+
+								editedMessage:
+									source
+							}
+						}
+					}
+				},
+
+				{
+					messageId:
+						messageId ||
+						generateMessageIDV2(),
+
+					...options
+				}
+			);
+
+		await this.#client.relayMessage(
+			jid,
+			editedMessage.message,
+			{
+				messageId:
+					editedMessage.key.id
+			}
+		);
+
+		return editedMessage;
+	}
+}
+
+
+export { Foad };