@bellaxchuu/flowred 0.8.6

package/lib/Utils/link-preview.js

@@ -0,0 +1,143 @@
+'use strict'
+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.getUrlInfo = void 0
+const messages_1 = require('./messages')
+const messages_media_1 = require('./messages-media')
+const THUMBNAIL_WIDTH_PX = 192
+/** Fetches an image and generates a thumbnail for it */
+const getCompressedJpegThumbnail = async (url, { thumbnailWidth, fetchOpts }) => {
+	const stream = await (0, messages_media_1.getHttpStream)(url, fetchOpts)
+	const result = await (0, messages_media_1.extractImageThumb)(stream, thumbnailWidth)
+	return result
+}
+/**
+ * Given a piece of text, checks for any URL present, generates link preview for the same and returns it
+ * Return undefined if the fetch failed or no URL was found
+ * @param text first matched URL in text
+ * @returns the URL info required to generate link preview
+ */
+const getUrlInfo = async (
+	text,
+	opts = {
+		thumbnailWidth: THUMBNAIL_WIDTH_PX,
+		fetchOpts: { timeout: 3000 }
+	}
+) => {
+	try {
+		// retries
+		const retries = 0
+		const maxRetry = 5
+		const { getLinkPreview } = await Promise.resolve().then(() => __importStar(require('link-preview-js')))
+		let previewLink = text
+		if (!text.startsWith('https://') && !text.startsWith('http://')) {
+			previewLink = 'https://' + previewLink
+		}
+		const info = await getLinkPreview(previewLink, {
+			...opts.fetchOpts,
+			followRedirects: 'follow',
+			handleRedirects: (baseURL, forwardedURL) => {
+				const urlObj = new URL(baseURL)
+				const forwardedURLObj = new URL(forwardedURL)
+				if (retries >= maxRetry) {
+					return false
+				}
+				if (
+					forwardedURLObj.hostname === urlObj.hostname ||
+					forwardedURLObj.hostname === 'www.' + urlObj.hostname ||
+					'www.' + forwardedURLObj.hostname === urlObj.hostname
+				) {
+					retries + 1
+					return true
+				} else {
+					return false
+				}
+			},
+			headers: opts.fetchOpts?.headers
+		})
+		if (info && 'title' in info && info.title) {
+			const [image] = info.images
+			const urlInfo = {
+				'canonical-url': info.url,
+				'matched-text': text,
+				title: info.title,
+				description: info.description,
+				originalThumbnailUrl: image
+			}
+			if (opts.uploadImage) {
+				const { imageMessage } = await (0, messages_1.prepareWAMessageMedia)(
+					{ image: { url: image } },
+					{
+						upload: opts.uploadImage,
+						mediaTypeOverride: 'thumbnail-link',
+						options: opts.fetchOpts
+					}
+				)
+				urlInfo.jpegThumbnail = imageMessage?.jpegThumbnail ? Buffer.from(imageMessage.jpegThumbnail) : undefined
+				urlInfo.highQualityThumbnail = imageMessage || undefined
+			} else {
+				try {
+					urlInfo.jpegThumbnail = image ? (await getCompressedJpegThumbnail(image, opts)).buffer : undefined
+				} catch (error) {
+					opts.logger?.debug({ err: error.stack, url: previewLink }, 'error in generating thumbnail')
+				}
+			}
+			return urlInfo
+		}
+	} catch (error) {
+		if (!error.message.includes('receive a valid')) {
+			throw error
+		}
+	}
+}
+exports.getUrlInfo = getUrlInfo

package/lib/Utils/logger.js

@@ -0,0 +1,9 @@
+'use strict'
+var __importDefault =
+	(this && this.__importDefault) ||
+	function (mod) {
+		return mod && mod.__esModule ? mod : { default: mod }
+	}
+Object.defineProperty(exports, '__esModule', { value: true })
+const pino_1 = __importDefault(require('pino'))
+exports.default = (0, pino_1.default)({ timestamp: () => `,"time":"${new Date().toJSON()}"` })

package/lib/Utils/lt-hash.js

@@ -0,0 +1,39 @@
+'use strict'
+Object.defineProperty(exports, '__esModule', { value: true })
+exports.LT_HASH_ANTI_TAMPERING = void 0
+const crypto_1 = require('./crypto')
+
+/**
+ * LT Hash ("lthash16") is a summation-based homomorphic hash algorithm that
+ * maintains the integrity of a piece of data over a series of mutations. You
+ * can add/remove mutations and it'll return a hash equal to if the same
+ * series of mutations was made sequentially.
+ *
+ * WhatsApp expands each element to 128 bytes via HKDF (info = "WhatsApp Patch
+ * Integrity"), treats that as 64 little-endian uint16 lanes, and combines
+ * elements by adding/subtracting lane-wise mod 2^16 -- this is a pure HKDF +
+ * integer-arithmetic algorithm, no elliptic-curve math involved. Pure-JS,
+ * built on the already-validated `hkdf` implementation.
+ */
+const LTHASH_INFO = 'WhatsApp Patch Integrity'
+const LTHASH_SIZE = 128 // bytes = 64 uint16 lanes
+
+const expand = element => (0, crypto_1.hkdf)(element, LTHASH_SIZE, { info: LTHASH_INFO })
+
+const applyDelta = (result, element, sign) => {
+	const expanded = expand(element)
+	for (let i = 0; i < LTHASH_SIZE; i += 2) {
+		const lane = result.readUInt16LE(i)
+		const delta = expanded.readUInt16LE(i)
+		result.writeUInt16LE((lane + sign * delta) & 0xffff, i)
+	}
+}
+
+exports.LT_HASH_ANTI_TAMPERING = {
+	subtractThenAdd: (base, subtract = [], add = []) => {
+		const result = Buffer.from(base)
+		for (const element of subtract) applyDelta(result, element, -1)
+		for (const element of add) applyDelta(result, element, 1)
+		return result
+	}
+}

package/lib/Utils/make-mutex.js

@@ -0,0 +1,36 @@
+'use strict'
+Object.defineProperty(exports, '__esModule', { value: true })
+exports.makeKeyedMutex = exports.makeMutex = void 0
+const async_mutex_1 = require('async-mutex')
+const makeMutex = () => {
+	const mutex = new async_mutex_1.Mutex()
+	return {
+		mutex(code) {
+			return mutex.runExclusive(code)
+		}
+	}
+}
+exports.makeMutex = makeMutex
+const makeKeyedMutex = () => {
+	const map = new Map()
+	return {
+		async mutex(key, task) {
+			let entry = map.get(key)
+			if (!entry) {
+				entry = { mutex: new async_mutex_1.Mutex(), refCount: 0 }
+				map.set(key, entry)
+			}
+			entry.refCount++
+			try {
+				return await entry.mutex.runExclusive(task)
+			} finally {
+				entry.refCount--
+				// only delete it if this is still the current entry
+				if (entry.refCount === 0 && map.get(key) === entry) {
+					map.delete(key)
+				}
+			}
+		}
+	}
+}
+exports.makeKeyedMutex = makeKeyedMutex

package/lib/Utils/message-composer.js

@@ -0,0 +1,479 @@
+'use strict'
+
+const { proto } = require('../../WAProto/index.js')
+const { generateMessageIDV2 } = require('./generics.js')
+
+const JS_KEYWORDS = new Set([
+	'import',
+	'export',
+	'from',
+	'default',
+	'as',
+	'const',
+	'let',
+	'var',
+	'function',
+	'class',
+	'extends',
+	'new',
+	'return',
+	'if',
+	'else',
+	'for',
+	'while',
+	'do',
+	'switch',
+	'case',
+	'break',
+	'continue',
+	'try',
+	'catch',
+	'finally',
+	'throw',
+	'async',
+	'await',
+	'yield',
+	'typeof',
+	'instanceof',
+	'in',
+	'of',
+	'delete',
+	'void',
+	'true',
+	'false',
+	'null',
+	'undefined',
+	'NaN',
+	'Infinity',
+	'this',
+	'super',
+	'static',
+	'get',
+	'set',
+	'debugger',
+	'with'
+])
+
+const PYTHON_KEYWORDS = new Set([
+	'import',
+	'from',
+	'as',
+	'def',
+	'class',
+	'return',
+	'if',
+	'elif',
+	'else',
+	'for',
+	'while',
+	'break',
+	'continue',
+	'try',
+	'except',
+	'finally',
+	'raise',
+	'with',
+	'yield',
+	'lambda',
+	'pass',
+	'del',
+	'global',
+	'nonlocal',
+	'assert',
+	'True',
+	'False',
+	'None',
+	'and',
+	'or',
+	'not',
+	'in',
+	'is',
+	'async',
+	'await',
+	'self',
+	'print'
+])
+
+const LANGUAGE_KEYWORDS = {
+	javascript: JS_KEYWORDS,
+	typescript: JS_KEYWORDS,
+	js: JS_KEYWORDS,
+	ts: JS_KEYWORDS,
+	python: PYTHON_KEYWORDS,
+	py: PYTHON_KEYWORDS
+}
+
+var CodeHighlightType
+;(function (CodeHighlightType) {
+	CodeHighlightType[(CodeHighlightType['DEFAULT'] = 0)] = 'DEFAULT'
+	CodeHighlightType[(CodeHighlightType['KEYWORD'] = 1)] = 'KEYWORD'
+	CodeHighlightType[(CodeHighlightType['METHOD'] = 2)] = 'METHOD'
+	CodeHighlightType[(CodeHighlightType['STRING'] = 3)] = 'STRING'
+	CodeHighlightType[(CodeHighlightType['NUMBER'] = 4)] = 'NUMBER'
+	CodeHighlightType[(CodeHighlightType['COMMENT'] = 5)] = 'COMMENT'
+})(CodeHighlightType || (CodeHighlightType = {}))
+
+var RichSubMessageType
+;(function (RichSubMessageType) {
+	RichSubMessageType[(RichSubMessageType['UNKNOWN'] = 0)] = 'UNKNOWN'
+	RichSubMessageType[(RichSubMessageType['GRID_IMAGE'] = 1)] = 'GRID_IMAGE'
+	RichSubMessageType[(RichSubMessageType['TEXT'] = 2)] = 'TEXT'
+	RichSubMessageType[(RichSubMessageType['INLINE_IMAGE'] = 3)] = 'INLINE_IMAGE'
+	RichSubMessageType[(RichSubMessageType['TABLE'] = 4)] = 'TABLE'
+	RichSubMessageType[(RichSubMessageType['CODE'] = 5)] = 'CODE'
+	RichSubMessageType[(RichSubMessageType['DYNAMIC'] = 6)] = 'DYNAMIC'
+	RichSubMessageType[(RichSubMessageType['MAP'] = 7)] = 'MAP'
+	RichSubMessageType[(RichSubMessageType['LATEX'] = 8)] = 'LATEX'
+	RichSubMessageType[(RichSubMessageType['CONTENT_ITEMS'] = 9)] = 'CONTENT_ITEMS'
+})(RichSubMessageType || (RichSubMessageType = {}))
+
+const tokenizeCode = (codeStr, language = 'javascript') => {
+	const keywords = LANGUAGE_KEYWORDS[language] || JS_KEYWORDS
+	const blocks = []
+	const lines = codeStr.split('\n')
+	for (let li = 0; li < lines.length; li++) {
+		const line = lines[li]
+		const isLast = li === lines.length - 1
+		const nl = isLast ? '' : '\n'
+		if (!line.trim()) {
+			blocks.push({
+				highlightType: CodeHighlightType.DEFAULT,
+				codeContent: line + nl
+			})
+			continue
+		}
+		if (line.trim().startsWith('//') || line.trim().startsWith('#')) {
+			blocks.push({
+				highlightType: CodeHighlightType.COMMENT,
+				codeContent: line + nl
+			})
+			continue
+		}
+		const regex =
+			/(\/\/.*$|#.*$)|(["'`](?:[^"'`\\]|\\.)*["'`])|(\b\d+(?:\.\d+)?\b)|(\b[a-zA-Z_$][\w$]*\b)|([^\s\w$"'`]+)|(\s+)/g
+		let match
+		const tokens = []
+		while ((match = regex.exec(line)) !== null) {
+			const val = match[0]
+			if (match[1]) {
+				tokens.push({
+					highlightType: CodeHighlightType.COMMENT,
+					codeContent: val
+				})
+			} else if (match[2]) {
+				tokens.push({
+					highlightType: CodeHighlightType.STRING,
+					codeContent: val
+				})
+			} else if (match[3]) {
+				tokens.push({
+					highlightType: CodeHighlightType.NUMBER,
+					codeContent: val
+				})
+			} else if (match[4]) {
+				if (keywords.has(val)) {
+					tokens.push({
+						highlightType: CodeHighlightType.KEYWORD,
+						codeContent: val
+					})
+				} else {
+					const after = line.slice(regex.lastIndex).trimStart()
+					if (after.startsWith('(')) {
+						tokens.push({
+							highlightType: CodeHighlightType.METHOD,
+							codeContent: val
+						})
+					} else {
+						tokens.push({
+							highlightType: CodeHighlightType.DEFAULT,
+							codeContent: val
+						})
+					}
+				}
+			} else {
+				tokens.push({
+					highlightType: CodeHighlightType.DEFAULT,
+					codeContent: val
+				})
+			}
+		}
+		if (tokens.length === 0) {
+			blocks.push({
+				highlightType: CodeHighlightType.DEFAULT,
+				codeContent: line + nl
+			})
+			continue
+		}
+		const merged = []
+		for (const t of tokens) {
+			const prev = merged.length > 0 ? merged[merged.length - 1] : undefined
+			if (prev && prev.highlightType === t.highlightType) {
+				prev.codeContent += t.codeContent
+			} else {
+				merged.push({ ...t })
+			}
+		}
+		if (merged.length > 0) {
+			merged[merged.length - 1].codeContent += nl
+		}
+		blocks.push(...merged)
+	}
+	return blocks
+}
+
+const buildRichContextInfo = (quoted, options) => {
+	const ctxInfo = {
+		forwardingScore: 1,
+		isForwarded: true,
+		forwardedAiBotMessageInfo: { botJid: options.botJid ? options.botJid : '867051314767696@bot' },
+		forwardOrigin: 4,
+		...(options.mentions ? { mentionedJid: options.mentions } : {})
+	}
+	if (quoted?.key) {
+		ctxInfo.stanzaId = quoted.key.id
+		ctxInfo.participant = quoted.key.participant || quoted.sender || quoted.key.remoteJid
+		ctxInfo.quotedMessage = quoted.message
+	}
+	return ctxInfo
+}
+
+const buildBotForwardedMessage = (submessages, contextInfo, unifiedResponse) => {
+	const richResponse = {
+		messageType: 1,
+		submessages,
+		contextInfo
+	}
+	if (unifiedResponse) {
+		richResponse.unifiedResponse = unifiedResponse
+	}
+	return {
+		botForwardedMessage: {
+			message: {
+				richResponseMessage: richResponse
+			}
+		}
+	}
+}
+
+const generateTableContent = (title, headers, rows, quoted, options = {}) => {
+	const { footer, headerText } = options
+	const tableRows = [{ items: headers, isHeading: true }, ...rows.map(row => ({ items: row.map(String) }))]
+	const submessages = []
+	if (headerText) {
+		submessages.push({ messageType: 2, messageText: headerText })
+	}
+	submessages.push({
+		messageType: 4,
+		tableMetadata: { title, rows: tableRows }
+	})
+	if (footer) {
+		submessages.push({ messageType: 2, messageText: footer })
+	}
+	const ctxInfo = buildRichContextInfo(quoted)
+	return {
+		message: buildBotForwardedMessage(submessages, ctxInfo),
+		messageId: generateMessageIDV2()
+	}
+}
+
+const generateListContent = (title, items, quoted, options = {}) => {
+	const { footer, headerText } = options
+	const tableRows = items.map(item => ({
+		items: Array.isArray(item) ? item.map(String) : [String(item)]
+	}))
+	const submessages = []
+	if (headerText) {
+		submessages.push({ messageType: 2, messageText: headerText })
+	}
+	submessages.push({
+		messageType: 4,
+		tableMetadata: { title, rows: tableRows }
+	})
+	if (footer) {
+		submessages.push({ messageType: 2, messageText: footer })
+	}
+	const ctxInfo = buildRichContextInfo(quoted)
+	return {
+		message: buildBotForwardedMessage(submessages, ctxInfo),
+		messageId: generateMessageIDV2()
+	}
+}
+
+const generateCodeBlockContent = (code, quoted, options = {}) => {
+	const { title, footer, language = 'javascript' } = options
+	const submessages = []
+	if (title) {
+		submessages.push({ messageType: 2, messageText: title })
+	}
+	submessages.push({
+		messageType: 5,
+		codeMetadata: {
+			codeLanguage: language,
+			codeBlocks: tokenizeCode(code, language)
+		}
+	})
+	if (footer) {
+		submessages.push({ messageType: 2, messageText: footer })
+	}
+	const ctxInfo = buildRichContextInfo(quoted)
+	return {
+		message: buildBotForwardedMessage(submessages, ctxInfo),
+		messageId: generateMessageIDV2()
+	}
+}
+
+const generateLatexContent = (quoted, options) => {
+	const { text, expressions, headerText, footer } = options
+	const submessages = []
+	if (headerText) {
+		submessages.push({ messageType: 2, messageText: headerText })
+	}
+	const latexExpressions = expressions.map(expr => {
+		const entry = {
+			latexExpression: expr.latexExpression,
+			url: expr.url,
+			width: expr.width,
+			height: expr.height
+		}
+		if (expr.fontHeight !== undefined) entry.fontHeight = expr.fontHeight
+		if (expr.imageTopPadding !== undefined) entry.imageTopPadding = expr.imageTopPadding
+		if (expr.imageLeadingPadding !== undefined) entry.imageLeadingPadding = expr.imageLeadingPadding
+		if (expr.imageBottomPadding !== undefined) entry.imageBottomPadding = expr.imageBottomPadding
+		if (expr.imageTrailingPadding !== undefined) entry.imageTrailingPadding = expr.imageTrailingPadding
+		return entry
+	})
+	submessages.push({
+		messageType: 8,
+		latexMetadata: {
+			text: text || '',
+			expressions: latexExpressions
+		}
+	})
+	if (footer) {
+		submessages.push({ messageType: 2, messageText: footer })
+	}
+	const ctxInfo = buildRichContextInfo(quoted)
+	return {
+		message: buildBotForwardedMessage(submessages, ctxInfo),
+		messageId: generateMessageIDV2()
+	}
+}
+
+const generateLatexImageContent = async (quoted, options, uploadFn, renderLatexToPng) => {
+	const { text, expressions, headerText, footer } = options
+	const submessages = []
+	if (headerText) {
+		submessages.push({ messageType: 2, messageText: headerText })
+	}
+	const latexExpressions = await Promise.all(
+		expressions.map(async expr => {
+			const { buffer, width, height } = await renderLatexToPng(expr.latexExpression)
+			const uploadResult = await uploadFn(buffer, 'image')
+			const imageUrl = uploadResult.url || uploadResult.directPath
+			return {
+				latexExpression: expr.latexExpression,
+				url: imageUrl,
+				width,
+				height
+			}
+		})
+	)
+	submessages.push({
+		messageType: 8,
+		latexMetadata: {
+			text: text || '',
+			expressions: latexExpressions
+		}
+	})
+	if (footer) {
+		submessages.push({ messageType: 2, messageText: footer })
+	}
+	const ctxInfo = buildRichContextInfo(quoted)
+	return {
+		message: buildBotForwardedMessage(submessages, ctxInfo),
+		messageId: generateMessageIDV2()
+	}
+}
+
+const generateLatexInlineImageContent = async (quoted, options, uploadFn, renderLatexToPng) => {
+	const { text, expressions, headerText, footer } = options
+	const submessages = []
+	if (headerText) {
+		submessages.push({ messageType: 2, messageText: headerText })
+	}
+	if (text) {
+		submessages.push({ messageType: 2, messageText: text })
+	}
+	for (const expr of expressions) {
+		const { buffer, width, height } = await renderLatexToPng(expr.latexExpression)
+		const uploadResult = await uploadFn(buffer, 'image')
+		const imageUrl = uploadResult.url || uploadResult.directPath
+		submessages.push({
+			messageType: 3,
+			imageMetadata: {
+				imageUrl: {
+					imagePreviewUrl: imageUrl,
+					imageHighResUrl: imageUrl
+				},
+				imageText: expr.latexExpression,
+				alignment: 2
+			}
+		})
+	}
+	if (footer) {
+		submessages.push({ messageType: 2, messageText: footer })
+	}
+	const ctxInfo = buildRichContextInfo(quoted)
+	return {
+		message: buildBotForwardedMessage(submessages, ctxInfo),
+		messageId: generateMessageIDV2()
+	}
+}
+
+const captureUnifiedResponse = msg => {
+	const botFwd = msg?.botForwardedMessage?.message
+	if (!botFwd) return null
+	const rich = botFwd.richResponseMessage
+	if (!rich?.unifiedResponse?.data) return null
+	return {
+		unifiedResponse: { data: rich.unifiedResponse.data },
+		submessages: rich.submessages || [],
+		contextInfo: rich.contextInfo || {}
+	}
+}
+
+const generateUnifiedResponseContent = (quoted, captured) => {
+	const ctxInfo = buildRichContextInfo(quoted)
+	return {
+		message: buildBotForwardedMessage(captured.submessages, ctxInfo, captured.unifiedResponse),
+		messageId: generateMessageIDV2()
+	}
+}
+
+const generateRichMessageContent = (submessages, quoted, options) => {
+	const ctxInfo = buildRichContextInfo(quoted, options)
+	return {
+		message: buildBotForwardedMessage(submessages, ctxInfo),
+		messageId: generateMessageIDV2()
+	}
+}
+
+module.exports = {
+	JS_KEYWORDS,
+	PYTHON_KEYWORDS,
+	LANGUAGE_KEYWORDS,
+	CodeHighlightType,
+	RichSubMessageType,
+	tokenizeCode,
+	buildRichContextInfo,
+	buildBotForwardedMessage,
+	generateTableContent,
+	generateListContent,
+	generateCodeBlockContent,
+	generateLatexContent,
+	generateLatexImageContent,
+	generateLatexInlineImageContent,
+	captureUnifiedResponse,
+	generateUnifiedResponseContent,
+	generateRichMessageContent
+}