mirror of
https://github.com/FranP-code/Baileys.git
synced 2025-10-13 00:32:22 +00:00
feat: customize link preview image width
This commit is contained in:
@@ -37,6 +37,7 @@ const BASE_CONNECTION_CONFIG: CommonSocketConfig<any> = {
|
||||
export const DEFAULT_CONNECTION_CONFIG: SocketConfig = {
|
||||
...BASE_CONNECTION_CONFIG,
|
||||
waWebSocketUrl: 'wss://web.whatsapp.com/ws/chat',
|
||||
linkPreviewImageThumbnailWidth: 192,
|
||||
getMessage: async() => undefined
|
||||
}
|
||||
|
||||
|
||||
@@ -9,7 +9,7 @@ import { BinaryNode, BinaryNodeAttributes, getBinaryNodeChild, getBinaryNodeChil
|
||||
import { makeGroupsSocket } from './groups'
|
||||
|
||||
export const makeMessagesSocket = (config: SocketConfig) => {
|
||||
const { logger } = config
|
||||
const { logger, linkPreviewImageThumbnailWidth } = config
|
||||
const sock = makeGroupsSocket(config)
|
||||
const {
|
||||
ev,
|
||||
@@ -485,7 +485,7 @@ export const makeMessagesSocket = (config: SocketConfig) => {
|
||||
logger,
|
||||
userJid,
|
||||
// multi-device does not have this yet
|
||||
getUrlInfo: getUrlInfo,
|
||||
getUrlInfo: text => getUrlInfo(text, { thumbnailWidth: linkPreviewImageThumbnailWidth }),
|
||||
upload: waUploadToServer,
|
||||
mediaCache: config.mediaCache,
|
||||
...options,
|
||||
|
||||
@@ -19,6 +19,8 @@ export type SocketConfig = CommonSocketConfig<AuthenticationState> & {
|
||||
userDevicesCache?: NodeCache
|
||||
/** map to store the retry counts for failed messages */
|
||||
msgRetryCounterMap?: { [msgId: string]: number }
|
||||
/** width for link preview images */
|
||||
linkPreviewImageThumbnailWidth: number
|
||||
/**
|
||||
* fetch a message from your store
|
||||
* implement this so that messages failed to send (solves the "this message can take a while" issue) can be retried
|
||||
|
||||
@@ -1,22 +1,29 @@
|
||||
import { WAUrlInfo } from '../Types'
|
||||
import { extractImageThumb, getHttpStream } from './messages-media'
|
||||
|
||||
const THUMBNAIL_WIDTH_PX = 128
|
||||
const THUMBNAIL_WIDTH_PX = 192
|
||||
|
||||
/** Fetches an image and generates a thumbnail for it */
|
||||
const getCompressedJpegThumbnail = async(url: string) => {
|
||||
const getCompressedJpegThumbnail = async(url: string, thumbnailWidth: number) => {
|
||||
const stream = await getHttpStream(url)
|
||||
const result = await extractImageThumb(stream, THUMBNAIL_WIDTH_PX)
|
||||
const result = await extractImageThumb(stream, thumbnailWidth)
|
||||
return result
|
||||
}
|
||||
|
||||
export type URLGenerationOptions = {
|
||||
thumbnailWidth: number
|
||||
}
|
||||
|
||||
/**
|
||||
* 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 the text containing URL
|
||||
* @returns the URL info required to generate link preview
|
||||
*/
|
||||
export const getUrlInfo = async(text: string): Promise<WAUrlInfo | undefined> => {
|
||||
export const getUrlInfo = async(
|
||||
text: string,
|
||||
opts: URLGenerationOptions = { thumbnailWidth: THUMBNAIL_WIDTH_PX }
|
||||
): Promise<WAUrlInfo | undefined> => {
|
||||
try {
|
||||
const { getLinkPreview } = await import('link-preview-js')
|
||||
|
||||
@@ -24,7 +31,9 @@ export const getUrlInfo = async(text: string): Promise<WAUrlInfo | undefined> =>
|
||||
if(info && 'title' in info) {
|
||||
const [image] = info.images
|
||||
|
||||
const jpegThumbnail = image ? await getCompressedJpegThumbnail(image) : undefined
|
||||
const jpegThumbnail = image
|
||||
? await getCompressedJpegThumbnail(image, opts.thumbnailWidth)
|
||||
: undefined
|
||||
|
||||
return {
|
||||
'canonical-url': info.url,
|
||||
|
||||
Reference in New Issue
Block a user