quote-api / utils /quote-generate.js
rippanteq7's picture
Upload folder using huggingface_hub
0dcf096 verified
const fs = require('fs')
const { createCanvas, registerFont } = require('canvas')
const EmojiDbLib = require('emoji-db')
const { loadImage } = require('canvas')
const loadImageFromUrl = require('./image-load-url')
const sharp = require('sharp')
const Jimp = require('jimp')
const smartcrop = require('smartcrop-sharp')
const runes = require('runes')
const zlib = require('zlib')
const { Telegram } = require('telegraf')
const emojiDb = new EmojiDbLib({ useDefaultDb: true })
function loadFont () {
console.log('font load start')
const fontsDir = 'assets/fonts/'
fs.readdir(fontsDir, (_err, files) => {
files.forEach((file) => {
try {
registerFont(`${fontsDir}${file}`, { family: file.replace(/\.[^/.]+$/, '') })
} catch (error) {
console.error(`${fontsDir}${file} not font file`)
}
})
})
console.log('font load end')
}
loadFont()
const emojiImageByBrand = require('./emoji-image')
const LRU = require('lru-cache')
const avatarCache = new LRU({
max: 20,
maxAge: 1000 * 60 * 5
})
// write a nodejs function that accepts 2 colors. the first is the background color and the second is the text color. as a result, the first color should come out brighter or darker depending on the contrast. for example, if the first text is dark, then make the second brighter and return it. you need to change not the background color, but the text color
// here are all the possible colors that will be passed as the second argument. the first color can be any
class ColorContrast {
constructor() {
this.brightnessThreshold = 175; // A threshold to determine when a color is considered bright or dark
}
getBrightness(color) {
// Calculates the brightness of a color using the formula from the WCAG 2.0
// See: https://www.w3.org/TR/WCAG20-TECHS/G18.html#G18-tests
const [r, g, b] = this.hexToRgb(color);
return (r * 299 + g * 587 + b * 114) / 1000;
}
hexToRgb(hex) {
// Converts a hex color string to an RGB array
const r = parseInt(hex.substring(1, 3), 16);
const g = parseInt(hex.substring(3, 5), 16);
const b = parseInt(hex.substring(5, 7), 16);
return [r, g, b];
}
rgbToHex([r, g, b]) {
// Converts an RGB array to a hex color string
return `#${r.toString(16).padStart(2, '0')}${g.toString(16).padStart(2, '0')}${b.toString(16).padStart(2, '0')}`;
}
adjustBrightness(color, amount) {
// Adjusts the brightness of a color by a specified amount
const [r, g, b] = this.hexToRgb(color);
const newR = Math.max(0, Math.min(255, r + amount));
const newG = Math.max(0, Math.min(255, g + amount));
const newB = Math.max(0, Math.min(255, b + amount));
return this.rgbToHex([newR, newG, newB]);
}
getContrastRatio(background, foreground) {
// Calculates the contrast ratio between two colors using the formula from the WCAG 2.0
// See: https://www.w3.org/TR/WCAG20-TECHS/G18.html#G18-tests
const brightness1 = this.getBrightness(background);
const brightness2 = this.getBrightness(foreground);
const lightest = Math.max(brightness1, brightness2);
const darkest = Math.min(brightness1, brightness2);
return (lightest + 0.05) / (darkest + 0.05);
}
adjustContrast(background, foreground) {
// Adjusts the brightness of the foreground color to meet the minimum contrast ratio
// with the background color
const contrastRatio = this.getContrastRatio(background, foreground);
const brightnessDiff = this.getBrightness(background) - this.getBrightness(foreground);
if (contrastRatio >= 4.5) {
return foreground; // The contrast ratio is already sufficient
} else if (brightnessDiff >= 0) {
// The background is brighter than the foreground
const amount = Math.ceil((this.brightnessThreshold - this.getBrightness(foreground)) / 2);
return this.adjustBrightness(foreground, amount);
} else {
// The background is darker than the foreground
const amount = Math.ceil((this.getBrightness(foreground) - this.brightnessThreshold) / 2);
return this.adjustBrightness(foreground, -amount);
}
}
}
class QuoteGenerate {
constructor (botToken) {
this.telegram = new Telegram(botToken)
}
async avatarImageLatters (letters, color) {
const size = 500
const canvas = createCanvas(size, size)
const context = canvas.getContext('2d')
const gradient = context.createLinearGradient(0, 0, canvas.width, canvas.height)
gradient.addColorStop(0, color[0])
gradient.addColorStop(1, color[1])
context.fillStyle = gradient
context.fillRect(0, 0, canvas.width, canvas.height)
const drawLetters = await this.drawMultilineText(
letters,
null,
size / 2,
'#FFF',
0,
size,
size * 5,
size * 5
)
context.drawImage(drawLetters, (canvas.width - drawLetters.width) / 2, (canvas.height - drawLetters.height) / 1.5)
return canvas.toBuffer()
}
async downloadAvatarImage (user) {
let avatarImage
let nameLatters
if (user.first_name && user.last_name) nameLatters = runes(user.first_name)[0] + (runes(user.last_name || '')[0])
else {
let name = user.first_name || user.name || user.title
name = name.toUpperCase()
const nameWord = name.split(' ')
if (nameWord.length > 1) nameLatters = runes(nameWord[0])[0] + runes(nameWord.splice(-1)[0])[0]
else nameLatters = runes(nameWord[0])[0]
}
const cacheKey = user.id
const avatarImageCache = avatarCache.get(cacheKey)
const avatarColorArray = [
[ '#FF885E', '#FF516A' ], // red
[ '#FFCD6A', '#FFA85C' ], // orange
[ '#E0A2F3', '#D669ED' ], // purple
[ '#A0DE7E', '#54CB68' ], // green
[ '#53EDD6', '#28C9B7' ], // sea
[ '#72D5FD', '#2A9EF1' ], // blue
[ '#FFA8A8', '#FF719A' ] // pink
]
const nameIndex = Math.abs(user.id) % 7
const avatarColor = avatarColorArray[nameIndex]
if (avatarImageCache) {
avatarImage = avatarImageCache
} else if (user.photo && user.photo.url) {
avatarImage = await loadImage(user.photo.url)
} else {
try {
let userPhoto, userPhotoUrl
if (user.photo && user.photo.big_file_id) userPhotoUrl = await this.telegram.getFileLink(user.photo.big_file_id).catch(console.error)
if (!userPhotoUrl) {
const getChat = await this.telegram.getChat(user.id).catch(console.error)
if (getChat && getChat.photo && getChat.photo.big_file_id) userPhoto = getChat.photo.big_file_id
if (userPhoto) userPhotoUrl = await this.telegram.getFileLink(userPhoto)
else if (user.username) userPhotoUrl = `https://telega.one/i/userpic/320/${user.username}.jpg`
else avatarImage = await loadImage(await this.avatarImageLatters(nameLatters, avatarColor))
}
if (userPhotoUrl) avatarImage = await loadImage(userPhotoUrl)
avatarCache.set(cacheKey, avatarImage)
} catch (error) {
avatarImage = await loadImage(await this.avatarImageLatters(nameLatters, avatarColor))
}
}
return avatarImage
}
ungzip (input, options) {
return new Promise((resolve, reject) => {
zlib.gunzip(input, options, (error, result) => {
if (!error) resolve(result)
else reject(Error(error))
})
})
}
async downloadMediaImage (media, mediaSize, type = 'id', crop = true) {
let mediaUrl
if (type === 'id') mediaUrl = await this.telegram.getFileLink(media).catch(console.error)
else mediaUrl = media
const load = await loadImageFromUrl(mediaUrl)
if (crop || mediaUrl.match(/.webp/)) {
const imageSharp = sharp(load)
const imageMetadata = await imageSharp.metadata()
const sharpPng = await imageSharp.png({ lossless: true, force: true }).toBuffer()
let croppedImage
if (imageMetadata.format === 'webp') {
const jimpImage = await Jimp.read(sharpPng)
croppedImage = await jimpImage.autocrop(false).getBufferAsync(Jimp.MIME_PNG)
} else {
const smartcropResult = await smartcrop.crop(sharpPng, { width: mediaSize, height: imageMetadata.height })
const crop = smartcropResult.topCrop
croppedImage = imageSharp.extract({ width: crop.width, height: crop.height, left: crop.x, top: crop.y })
croppedImage = await imageSharp.png({ lossless: true, force: true }).toBuffer()
}
return loadImage(croppedImage)
} else {
return loadImage(load)
}
}
hexToRgb (hex) {
return hex.replace(/^#?([a-f\d])([a-f\d])([a-f\d])$/i
, (m, r, g, b) => '#' + r + r + g + g + b + b)
.substring(1).match(/.{2}/g)
.map(x => parseInt(x, 16))
}
// https://codepen.io/andreaswik/pen/YjJqpK
lightOrDark (color) {
let r, g, b
// Check the format of the color, HEX or RGB?
if (color.match(/^rgb/)) {
// If HEX --> store the red, green, blue values in separate variables
color = color.match(/^rgba?\((\d+),\s*(\d+),\s*(\d+)(?:,\s*(\d+(?:\.\d+)?))?\)$/)
r = color[1]
g = color[2]
b = color[3]
} else {
// If RGB --> Convert it to HEX: http://gist.github.com/983661
color = +('0x' + color.slice(1).replace(
color.length < 5 && /./g, '$&$&'
)
)
r = color >> 16
g = color >> 8 & 255
b = color & 255
}
// HSP (Highly Sensitive Poo) equation from http://alienryderflex.com/hsp.html
const hsp = Math.sqrt(
0.299 * (r * r) +
0.587 * (g * g) +
0.114 * (b * b)
)
// Using the HSP value, determine whether the color is light or dark
if (hsp > 127.5) {
return 'light'
} else {
return 'dark'
}
}
async drawMultilineText (text, entities, fontSize, fontColor, textX, textY, maxWidth, maxHeight, emojiBrand = 'apple') {
if (maxWidth > 10000) maxWidth = 10000
if (maxHeight > 10000) maxHeight = 10000
const emojiImageJson = emojiImageByBrand[emojiBrand]
let fallbackEmojiBrand = 'apple'
if (emojiBrand === 'blob') fallbackEmojiBrand = 'google'
const fallbackEmojiImageJson = emojiImageByBrand[fallbackEmojiBrand]
const canvas = createCanvas(maxWidth + fontSize, maxHeight + fontSize)
const canvasCtx = canvas.getContext('2d')
// text = text.slice(0, 4096)
text = text.replace(/і/g, 'i') // замена украинской буквы і на английскую, так как она отсутствует в шрифтах Noto
const chars = text.split('')
const lineHeight = 4 * (fontSize * 0.3)
const styledChar = []
const emojis = emojiDb.searchFromText({ input: text, fixCodePoints: true })
for (let charIndex = 0; charIndex < chars.length; charIndex++) {
const char = chars[charIndex]
styledChar[charIndex] = {
char,
style: []
}
if (entities && typeof entities === 'string') styledChar[charIndex].style.push(entities)
}
if (entities && typeof entities === 'object') {
for (let entityIndex = 0; entityIndex < entities.length; entityIndex++) {
const entity = entities[entityIndex]
const style = []
if (['pre', 'code', 'pre_code'].includes(entity.type)) {
style.push('monospace')
} else if (
['mention', 'text_mention', 'hashtag', 'email', 'phone_number', 'bot_command', 'url', 'text_link']
.includes(entity.type)
) {
style.push('mention')
} else {
style.push(entity.type)
}
if (entity.type === 'custom_emoji') {
styledChar[entity.offset].customEmojiId = entity.custom_emoji_id
}
for (let charIndex = entity.offset; charIndex < entity.offset + entity.length; charIndex++) {
styledChar[charIndex].style = styledChar[charIndex].style.concat(style)
}
}
}
for (let emojiIndex = 0; emojiIndex < emojis.length; emojiIndex++) {
const emoji = emojis[emojiIndex]
for (let charIndex = emoji.offset; charIndex < emoji.offset + emoji.length; charIndex++) {
styledChar[charIndex].emoji = {
index: emojiIndex,
code: emoji.found
}
}
}
const styledWords = []
let stringNum = 0
const breakMatch = /<br>|\n|\r/
const spaceMatch = /[\f\n\r\t\v\u0020\u1680\u2000-\u200a\u2028\u2029\u205f\u3000]/
const CJKMatch = /[\u1100-\u11ff\u2e80-\u2eff\u3000-\u303f\u3040-\u309f\u30a0-\u30ff\u3100-\u312f\u3130-\u318f\u3190-\u319f\u31a0-\u31bf\u31c0-\u31ef\u31f0-\u31ff\u3200-\u32ff\u3300-\u33ff\u3400-\u4dbf\u4e00-\u9fff\uac00-\ud7af\uf900-\ufaff]/
const RTLMatch = /[\u0591-\u07FF\u200F\u202B\u202E\uFB1D-\uFDFD\uFE70-\uFEFC]/
for (let index = 0; index < styledChar.length; index++) {
const charStyle = styledChar[index]
const lastChar = styledChar[index - 1]
if (
lastChar && (
(
(charStyle.emoji && !lastChar.emoji) ||
(!charStyle.emoji && lastChar.emoji) ||
(charStyle.emoji && lastChar.emoji && charStyle.emoji.index !== lastChar.emoji.index)
) ||
(
(charStyle.char.match(breakMatch)) ||
(charStyle.char.match(spaceMatch) && !lastChar.char.match(spaceMatch)) ||
(lastChar.char.match(spaceMatch) && !charStyle.char.match(spaceMatch)) ||
(charStyle.style && lastChar.style && charStyle.style.toString() !== lastChar.style.toString())
) || (
charStyle.char.match(CJKMatch) ||
lastChar.char.match(CJKMatch)
)
)
) {
stringNum++
}
if (!styledWords[stringNum]) {
styledWords[stringNum] = {
word: charStyle.char
}
if (charStyle.style) styledWords[stringNum].style = charStyle.style
if (charStyle.emoji) styledWords[stringNum].emoji = charStyle.emoji
if (charStyle.customEmojiId) styledWords[stringNum].customEmojiId = charStyle.customEmojiId
} else styledWords[stringNum].word += charStyle.char
}
let lineX = textX
let lineY = textY
let textWidth = 0
// load custom emoji
const customEmojiIds = []
for (let index = 0; index < styledWords.length; index++) {
const word = styledWords[index]
if (word.customEmojiId) {
customEmojiIds.push(word.customEmojiId)
}
}
const getCustomEmojiStickers = await this.telegram.callApi('getCustomEmojiStickers', {
custom_emoji_ids: customEmojiIds
}).catch(() => {})
const customEmojiStickers = {}
const loadCustomEmojiStickerPromises = []
if (getCustomEmojiStickers) {
for (let index = 0; index < getCustomEmojiStickers.length; index++) {
const sticker = getCustomEmojiStickers[index]
loadCustomEmojiStickerPromises.push((async () => {
const getFileLink = await this.telegram.getFileLink(sticker.thumb.file_id).catch(() => {})
if (getFileLink) {
const load = await loadImageFromUrl(getFileLink).catch(() => {})
const imageSharp = sharp(load)
const sharpPng = await imageSharp.png({ lossless: true, force: true }).toBuffer()
customEmojiStickers[sticker.custom_emoji_id] = await loadImage(sharpPng).catch(() => {})
}
})())
}
await Promise.all(loadCustomEmojiStickerPromises).catch(() => {})
}
let breakWrite = false
let lineDirection = styledWords[0].word.match(RTLMatch) ? 'rtl' : 'ltr'
for (let index = 0; index < styledWords.length; index++) {
const styledWord = styledWords[index]
let emojiImage
if (styledWord.emoji) {
if (styledWord.customEmojiId && customEmojiStickers[styledWord.customEmojiId]) {
emojiImage = customEmojiStickers[styledWord.customEmojiId]
} else {
const emojiImageBase = emojiImageJson[styledWord.emoji.code]
if (emojiImageBase) {
emojiImage = await loadImage(
Buffer.from(emojiImageBase, 'base64')
).catch(() => {})
}
if (!emojiImage) {
emojiImage = await loadImage(
Buffer.from(fallbackEmojiImageJson[styledWord.emoji.code], 'base64')
).catch(() => {})
}
}
}
let fontType = ''
let fontName = 'NotoSans'
let fillStyle = fontColor
if (styledWord.style.includes('bold')) {
fontType += 'bold '
}
if (styledWord.style.includes('italic')) {
fontType += 'italic '
}
if (styledWord.style.includes('monospace')) {
fontName = 'NotoSansMono'
fillStyle = '#5887a7'
}
if (styledWord.style.includes('mention')) {
fillStyle = '#6ab7ec'
}
if (styledWord.style.includes('spoiler')) {
const rbaColor = this.hexToRgb(this.normalizeColor(fontColor))
fillStyle = `rgba(${rbaColor[0]}, ${rbaColor[1]}, ${rbaColor[2]}, 0.15)`
}
// else {
// canvasCtx.font = `${fontSize}px OpenSans`
// canvasCtx.fillStyle = fontColor
// }
canvasCtx.font = `${fontType} ${fontSize}px ${fontName}`
canvasCtx.fillStyle = fillStyle
if (canvasCtx.measureText(styledWord.word).width > maxWidth - fontSize * 3) {
while (canvasCtx.measureText(styledWord.word).width > maxWidth - fontSize * 3) {
styledWord.word = styledWord.word.substr(0, styledWord.word.length - 1)
if (styledWord.word.length <= 0) break
}
styledWord.word += '…'
}
let lineWidth
const wordlWidth = canvasCtx.measureText(styledWord.word).width
if (styledWord.emoji) lineWidth = lineX + fontSize
else lineWidth = lineX + wordlWidth
if (styledWord.word.match(breakMatch) || (lineWidth > maxWidth - fontSize * 2 && wordlWidth < maxWidth)) {
if (styledWord.word.match(spaceMatch) && !styledWord.word.match(breakMatch)) styledWord.word = ''
if ((styledWord.word.match(spaceMatch) || !styledWord.word.match(breakMatch)) && lineY + lineHeight > maxHeight) {
while (lineWidth > maxWidth - fontSize * 2) {
styledWord.word = styledWord.word.substr(0, styledWord.word.length - 1)
lineWidth = lineX + canvasCtx.measureText(styledWord.word).width
if (styledWord.word.length <= 0) break
}
styledWord.word += '…'
lineWidth = lineX + canvasCtx.measureText(styledWord.word).width
breakWrite = true
} else {
if (styledWord.emoji) lineWidth = textX + fontSize + (fontSize * 0.2)
else lineWidth = textX + canvasCtx.measureText(styledWord.word).width
lineX = textX
lineY += lineHeight
if (index < styledWords.length - 1) {
let nextLineDirection = styledWords[index+1].word.match(RTLMatch) ? 'rtl' : 'ltr'
if (lineDirection != nextLineDirection) textWidth = maxWidth - fontSize * 2
lineDirection = nextLineDirection
}
}
}
if (styledWord.emoji) lineWidth += (fontSize * 0.2)
if (lineWidth > textWidth) textWidth = lineWidth
if (textWidth > maxWidth) textWidth = maxWidth
let wordX = (lineDirection == 'rtl') ? maxWidth-lineX-wordlWidth-fontSize * 2 : lineX
if (emojiImage) {
canvasCtx.drawImage(emojiImage, wordX, lineY - fontSize + (fontSize * 0.15), fontSize + (fontSize * 0.22), fontSize + (fontSize * 0.22))
} else {
canvasCtx.fillText(styledWord.word, wordX, lineY)
if (styledWord.style.includes('strikethrough')) canvasCtx.fillRect(wordX, lineY - fontSize / 2.8, canvasCtx.measureText(styledWord.word).width, fontSize * 0.1)
if (styledWord.style.includes('underline')) canvasCtx.fillRect(wordX, lineY + 2, canvasCtx.measureText(styledWord.word).width, fontSize * 0.1)
}
lineX = lineWidth
if (breakWrite) break
}
const canvasResize = createCanvas(textWidth, lineY + fontSize)
const canvasResizeCtx = canvasResize.getContext('2d')
let dx = (lineDirection == 'rtl') ? textWidth - maxWidth + fontSize * 2 : 0
canvasResizeCtx.drawImage(canvas, dx, 0)
return canvasResize
}
// https://stackoverflow.com/a/3368118
drawRoundRect (color, w, h, r) {
const x = 0
const y = 0
const canvas = createCanvas(w, h)
const canvasCtx = canvas.getContext('2d')
canvasCtx.fillStyle = color
if (w < 2 * r) r = w / 2
if (h < 2 * r) r = h / 2
canvasCtx.beginPath()
canvasCtx.moveTo(x + r, y)
canvasCtx.arcTo(x + w, y, x + w, y + h, r)
canvasCtx.arcTo(x + w, y + h, x, y + h, r)
canvasCtx.arcTo(x, y + h, x, y, r)
canvasCtx.arcTo(x, y, x + w, y, r)
canvasCtx.closePath()
canvasCtx.fill()
return canvas
}
drawGradientRoundRect (colorOne, colorTwo, w, h, r) {
const x = 0
const y = 0
const canvas = createCanvas(w, h)
const canvasCtx = canvas.getContext('2d')
const gradient = canvasCtx.createLinearGradient(0, 0, w, h)
gradient.addColorStop(0, colorOne)
gradient.addColorStop(1, colorTwo)
canvasCtx.fillStyle = gradient
if (w < 2 * r) r = w / 2
if (h < 2 * r) r = h / 2
canvasCtx.beginPath()
canvasCtx.moveTo(x + r, y)
canvasCtx.arcTo(x + w, y, x + w, y + h, r)
canvasCtx.arcTo(x + w, y + h, x, y + h, r)
canvasCtx.arcTo(x, y + h, x, y, r)
canvasCtx.arcTo(x, y, x + w, y, r)
canvasCtx.closePath()
canvasCtx.fill()
return canvas
}
colorLuminance (hex, lum) {
hex = String(hex).replace(/[^0-9a-f]/gi, '')
if (hex.length < 6) {
hex = hex[0] + hex[0] + hex[1] + hex[1] + hex[2] + hex[2]
}
lum = lum || 0
// convert to decimal and change luminosity
let rgb = '#'
let c
let i
for (i = 0; i < 3; i++) {
c = parseInt(hex.substr(i * 2, 2), 16)
c = Math.round(Math.min(Math.max(0, c + (c * lum)), 255)).toString(16)
rgb += ('00' + c).substr(c.length)
}
return rgb
}
roundImage (image, r) {
const w = image.width
const h = image.height
const canvas = createCanvas(w, h)
const canvasCtx = canvas.getContext('2d')
const x = 0
const y = 0
if (w < 2 * r) r = w / 2
if (h < 2 * r) r = h / 2
canvasCtx.beginPath()
canvasCtx.moveTo(x + r, y)
canvasCtx.arcTo(x + w, y, x + w, y + h, r)
canvasCtx.arcTo(x + w, y + h, x, y + h, r)
canvasCtx.arcTo(x, y + h, x, y, r)
canvasCtx.arcTo(x, y, x + w, y, r)
canvasCtx.clip()
canvasCtx.closePath()
canvasCtx.restore()
canvasCtx.drawImage(image, x, y)
return canvas
}
deawReplyLine (lineWidth, height, color) {
const canvas = createCanvas(20, height)
const context = canvas.getContext('2d')
context.beginPath()
context.moveTo(10, 0)
context.lineTo(10, height)
context.lineWidth = lineWidth
context.strokeStyle = color
context.stroke()
return canvas
}
async drawAvatar (user) {
const avatarImage = await this.downloadAvatarImage(user)
if (avatarImage) {
const avatarSize = avatarImage.naturalHeight
const canvas = createCanvas(avatarSize, avatarSize)
const canvasCtx = canvas.getContext('2d')
const avatarX = 0
const avatarY = 0
canvasCtx.beginPath()
canvasCtx.arc(avatarX + avatarSize / 2, avatarY + avatarSize / 2, avatarSize / 2, 0, Math.PI * 2, true)
canvasCtx.clip()
canvasCtx.closePath()
canvasCtx.restore()
canvasCtx.drawImage(avatarImage, avatarX, avatarY, avatarSize, avatarSize)
return canvas
}
}
drawLineSegment (ctx, x, y, width, isEven) {
ctx.lineWidth = 35 // how thick the line is
ctx.strokeStyle = '#aec6cf' // what color our line is
ctx.beginPath()
y = isEven ? y : -y
ctx.moveTo(x, 0)
ctx.lineTo(x, y)
ctx.arc(x + width / 2, y, width / 2, Math.PI, 0, isEven)
ctx.lineTo(x + width, 0)
ctx.stroke()
}
drawWaveform (data) {
const normalizedData = data.map(i => i / 32)
const canvas = createCanvas(4500, 500)
const padding = 50
canvas.height = (canvas.height + padding * 2)
const ctx = canvas.getContext('2d')
ctx.translate(0, canvas.height / 2 + padding)
// draw the line segments
const width = canvas.width / normalizedData.length
for (let i = 0; i < normalizedData.length; i++) {
const x = width * i
let height = normalizedData[i] * canvas.height - padding
if (height < 0) {
height = 0
} else if (height > canvas.height / 2) {
height = height > canvas.height / 2
}
this.drawLineSegment(ctx, x, height, width, (i + 1) % 2)
}
return canvas
}
async drawQuote (scale = 1, backgroundColorOne, backgroundColorTwo, avatar, replyName, replyNameColor, replyText, name, text, media, mediaType, maxMediaSize) {
const avatarPosX = 0 * scale
const avatarPosY = 5 * scale
const avatarSize = 50 * scale
const blockPosX = avatarSize + 10 * scale
const blockPosY = 0
const indent = 14 * scale
if (mediaType === 'sticker') name = undefined
let width = 0
if (name) width = name.width
if (text && width < text.width + indent) width = text.width + indent
if (name && width < name.width + indent) width = name.width + indent
if (replyName) {
if (width < replyName.width) width = replyName.width + indent * 2
if (width < replyText.width) width = replyText.width + indent * 2
}
let height = indent
if (text) height += text.height
else height += indent
if (name) {
height = name.height
if (text) height = text.height + name.height
else height += indent
}
width += blockPosX + indent
height += blockPosY
let namePosX = blockPosX + indent
let namePosY = indent
if (!name) {
namePosX = 0
namePosY = -indent
}
const textPosX = blockPosX + indent
let textPosY = indent
if (name) {
textPosY = name.height + indent * 0.25
height += indent * 0.25
}
let replyPosX = 0
let replyNamePosY = 0
let replyTextPosY = 0
if (replyName) {
replyPosX = textPosX + indent
const replyNameHeight = replyName.height
const replyTextHeight = replyText.height * 0.5
replyNamePosY = namePosY + replyNameHeight
replyTextPosY = replyNamePosY + replyTextHeight
textPosY += replyNameHeight + replyTextHeight + (indent / 4)
height += replyNameHeight + replyTextHeight + (indent / 4)
}
let mediaPosX = 0
let mediaPosY = 0
let mediaWidth, mediaHeight
if (media) {
mediaWidth = media.width * (maxMediaSize / media.height)
mediaHeight = maxMediaSize
if (mediaWidth >= maxMediaSize) {
mediaWidth = maxMediaSize
mediaHeight = media.height * (maxMediaSize / media.width)
}
if (!text || text.width <= mediaWidth || mediaWidth > (width - blockPosX)) {
width = mediaWidth + indent * 6
}
height += mediaHeight
if (!text) height += indent
if (name) {
mediaPosX = namePosX
mediaPosY = name.height + 5 * scale
} else {
mediaPosX = blockPosX + indent
mediaPosY = indent
}
if (replyName) mediaPosY += replyNamePosY + indent / 2
textPosY = mediaPosY + mediaHeight + 5 * scale
}
if (mediaType === 'sticker' && (name || replyName)) {
mediaPosY += indent * 4
height += indent * 2
}
const canvas = createCanvas(width, height)
const canvasCtx = canvas.getContext('2d')
let rectWidth = width - blockPosX
let rectHeight = height
const rectPosX = blockPosX
const rectPosY = blockPosY
const rectRoundRadius = 25 * scale
let rect
if (mediaType === 'sticker' && (name || replyName)) {
rectHeight -= mediaHeight + indent * 2
}
if (mediaType !== 'sticker' || name || replyName) {
if (backgroundColorOne === backgroundColorTwo) {
rect = this.drawRoundRect(backgroundColorOne, rectWidth, rectHeight, rectRoundRadius)
} else {
rect = this.drawGradientRoundRect(backgroundColorOne, backgroundColorTwo, rectWidth, rectHeight, rectRoundRadius)
}
}
if (avatar) canvasCtx.drawImage(avatar, avatarPosX, avatarPosY, avatarSize, avatarSize)
if (rect) canvasCtx.drawImage(rect, rectPosX, rectPosY)
if (name) canvasCtx.drawImage(name, namePosX, namePosY)
if (text) canvasCtx.drawImage(text, textPosX, textPosY)
if (media) canvasCtx.drawImage(this.roundImage(media, 5 * scale), mediaPosX, mediaPosY, mediaWidth, mediaHeight)
if (replyName) {
canvasCtx.drawImage(this.deawReplyLine(3 * scale, replyName.height + replyText.height * 0.4, replyNameColor), textPosX - 3, replyNamePosY)
canvasCtx.drawImage(replyName, replyPosX, replyNamePosY)
canvasCtx.drawImage(replyText, replyPosX, replyTextPosY)
}
return canvas
}
normalizeColor (color) {
const canvas = createCanvas(0, 0)
const canvasCtx = canvas.getContext('2d')
canvasCtx.fillStyle = color
color = canvasCtx.fillStyle
return color
}
async generate (backgroundColorOne, backgroundColorTwo, message, width = 512, height = 512, scale = 2, emojiBrand = 'apple') {
if (!scale) scale = 2
if (scale > 20) scale = 20
width *= scale
height *= scale
// check background style color black/light
const backStyle = this.lightOrDark(backgroundColorOne)
// historyPeer1NameFg: #c03d33; // red
// historyPeer2NameFg: #4fad2d; // green
// historyPeer3NameFg: #d09306; // yellow
// historyPeer4NameFg: #168acd; // blue
// historyPeer5NameFg: #8544d6; // purple
// historyPeer6NameFg: #cd4073; // pink
// historyPeer7NameFg: #2996ad; // sea
// historyPeer8NameFg: #ce671b; // orange
// { 0, 7, 4, 1, 6, 3, 5 }
// const nameColor = [
// '#c03d33', // red
// '#ce671b', // orange
// '#8544d6', // purple
// '#4fad2d', // green
// '#2996ad', // sea
// '#168acd', // blue
// '#cd4073' // pink
// ]
const nameColorLight = [
'#FC5C51', // red
'#FA790F', // orange
'#895DD5', // purple
'#0FB297', // green
'#0FC9D6', // sea
'#3CA5EC', // blue
'#D54FAF' // pink
]
const nameColorDark = [
'#FF8E86', // red
'#FFA357', // orange
'#B18FFF', // purple
'#4DD6BF', // green
'#45E8D1', // sea
'#7AC9FF', // blue
'#FF7FD5' // pink
]
// user name color
let nameIndex = 1
if (message.from.id) nameIndex = Math.abs(message.from.id) % 7
const nameColorArray = backStyle === 'light' ? nameColorLight : nameColorDark
let nameColor = nameColorArray[nameIndex]
const colorContrast = new ColorContrast()
// change name color based on background color by contrast
const contrast = colorContrast.getContrastRatio(this.colorLuminance(backgroundColorOne, 0.55), nameColor)
if (contrast > 90 || contrast < 30) {
nameColor = colorContrast.adjustContrast(this.colorLuminance(backgroundColorTwo, 0.55), nameColor)
}
const nameSize = 22 * scale
let nameCanvas
if (message?.from?.name) {
let name = message.from.name
const nameEntities = [
{
type: 'bold',
offset: 0,
length: name.length
}
]
if (message.from.emoji_status) {
name += ' 🤡'
nameEntities.push({
type: 'custom_emoji',
offset: name.length - 2,
length: 2,
custom_emoji_id: message.from.emoji_status
})
}
nameCanvas = await this.drawMultilineText(
name,
nameEntities,
nameSize,
nameColor,
0,
nameSize,
width,
nameSize,
emojiBrand
)
}
let fontSize = 24 * scale
let textColor = '#fff'
if (backStyle === 'light') textColor = '#000'
let textCanvas
if (message.text) {
textCanvas = await this.drawMultilineText(
message.text,
message.entities,
fontSize,
textColor,
0,
fontSize,
width,
height - fontSize,
emojiBrand
)
}
let avatarCanvas
if (message.avatar) avatarCanvas = await this.drawAvatar(message.from)
let replyName, replyNameColor, replyText
if (message.replyMessage && message.replyMessage.name && message.replyMessage.text) {
const replyNameIndex = Math.abs(message.replyMessage.chatId) % 7
replyNameColor = nameColorArray[replyNameIndex]
const replyNameFontSize = 16 * scale
if (message.replyMessage.name) {
replyName = await this.drawMultilineText(
message.replyMessage.name,
'bold',
replyNameFontSize,
replyNameColor,
0,
replyNameFontSize,
width * 0.9,
replyNameFontSize,
emojiBrand
)
}
let textColor = '#fff'
if (backStyle === 'light') textColor = '#000'
const replyTextFontSize = 21 * scale
replyText = await this.drawMultilineText(
message.replyMessage.text,
message.replyMessage.entities,
replyTextFontSize,
textColor,
0,
replyTextFontSize,
width * 0.9,
replyTextFontSize,
emojiBrand
)
}
let mediaCanvas, mediaType, maxMediaSize
if (message.media) {
let media, type
let crop = false
if (message.mediaCrop) crop = true
if (message.media.url) {
type = 'url'
media = message.media.url
} else {
type = 'id'
if (message.media.length > 1) {
if (crop) media = message.media[1]
else media = message.media.pop()
} else media = message.media[0]
}
maxMediaSize = width / 3 * scale
if (message.text && maxMediaSize < textCanvas.width) maxMediaSize = textCanvas.width
if (media.is_animated) {
media = media.thumb
maxMediaSize = maxMediaSize / 2
}
mediaCanvas = await this.downloadMediaImage(media, maxMediaSize, type, crop)
mediaType = message.mediaType
}
if (message.voice) {
mediaCanvas = this.drawWaveform(message.voice.waveform)
maxMediaSize = width / 3 * scale
}
const quote = this.drawQuote(
scale,
backgroundColorOne, backgroundColorTwo,
avatarCanvas,
replyName, replyNameColor, replyText,
nameCanvas, textCanvas,
mediaCanvas, mediaType, maxMediaSize
)
return quote
}
}
module.exports = QuoteGenerate