435 lines
14 KiB
TypeScript
435 lines
14 KiB
TypeScript
import axios from "axios";
|
|
import sharp from "sharp";
|
|
import twemoji from "twemoji";
|
|
import emojiRegex from "emoji-regex";
|
|
|
|
import {createCanvas, GlobalFonts, type Image as CanvasImage, loadImage, SKRSContext2D} from "@napi-rs/canvas";
|
|
import {Message, PhotoSize} from "typescript-telegram-bot-api";
|
|
import {ChatCommand} from "../base/chat-command";
|
|
import {bot, botUser} from "../index";
|
|
import {
|
|
getChatAvatar,
|
|
getFileUrl,
|
|
getUserAvatar,
|
|
logError,
|
|
makeDarkGradientBgFancy,
|
|
oldReplyToMessage,
|
|
oldSendMessage
|
|
} from "../util/utils";
|
|
import {Requirements} from "../base/requirements";
|
|
import {Requirement} from "../base/requirement";
|
|
|
|
try {
|
|
GlobalFonts.registerFromPath("./assets/Inter_18pt-ExtraThin.ttf", "InterExtraThin");
|
|
GlobalFonts.registerFromPath("./assets/Inter_18pt-Thin.ttf", "InterThin");
|
|
GlobalFonts.registerFromPath("./assets/Inter_18pt-Light.ttf", "InterLight");
|
|
GlobalFonts.registerFromPath("./assets/Inter_18pt-Regular.ttf", "Inter");
|
|
GlobalFonts.registerFromPath("./assets/Inter_18pt-Medium.ttf", "InterMedium");
|
|
GlobalFonts.registerFromPath("./assets/Inter_18pt-SemiBold.ttf", "InterSemiBold");
|
|
GlobalFonts.registerFromPath("./assets/Inter_18pt-Bold.ttf", "InterBold");
|
|
GlobalFonts.registerFromPath("./assets/Inter_18pt-ExtraBold.ttf", "InterExtraBold");
|
|
GlobalFonts.registerFromPath("./assets/Inter_18pt-Black.ttf", "InterBlack");
|
|
} catch (e) {
|
|
console.error(e);
|
|
}
|
|
|
|
export class Quote extends ChatCommand {
|
|
command = ["cit", "citation", "q", "quote"];
|
|
argsMode = "none" as const;
|
|
|
|
title = "/quote";
|
|
description = "Make quote from text (or quote)";
|
|
|
|
requirements = Requirements.Build(Requirement.REPLY);
|
|
|
|
async execute(msg: Message): Promise<void> {
|
|
const chatId = msg.chat.id;
|
|
const reply = msg.reply_to_message;
|
|
|
|
if (!reply) {
|
|
await oldReplyToMessage(msg, "Сделай /quote реплаем на сообщение 🙂").catch(logError);
|
|
return;
|
|
}
|
|
|
|
try {
|
|
const quoteRaw = (msg.quote?.text ?? reply.text ?? reply.caption ?? "").trim();
|
|
if (quoteRaw.length === 0) {
|
|
await oldReplyToMessage(msg, "Не нашёл в сообщении текста 😢").catch(logError);
|
|
return;
|
|
}
|
|
|
|
let quote = quoteRaw.length ? quoteRaw : "…";
|
|
if (quote.length > 2500) quote = quote.slice(0, 2497) + "…";
|
|
|
|
const png = await renderQuoteCard(quote, reply);
|
|
await bot.sendPhoto({
|
|
chat_id: chatId,
|
|
photo: png,
|
|
reply_parameters: {
|
|
message_id: msg.message_id,
|
|
},
|
|
}).catch(logError);
|
|
} catch (e) {
|
|
console.error(e);
|
|
await oldSendMessage(msg, "Не смог собрать цитату 😢").catch(logError);
|
|
}
|
|
}
|
|
}
|
|
|
|
// ===== Emoji cache & helpers =====
|
|
|
|
const emojiCache = new Map<string, CanvasImage>();
|
|
|
|
function twemojiUrl(emoji: string) {
|
|
const code = twemoji.convert.toCodePoint(emoji);
|
|
return `https://cdnjs.cloudflare.com/ajax/libs/twemoji/14.0.2/72x72/${code}.png`;
|
|
}
|
|
|
|
async function loadEmoji(emoji: string): Promise<CanvasImage> {
|
|
const url = twemojiUrl(emoji);
|
|
const cached = emojiCache.get(url);
|
|
if (cached) return cached;
|
|
|
|
const res = await axios.get<ArrayBuffer>(url, {responseType: "arraybuffer"});
|
|
const img = await loadImage(Buffer.from(res.data));
|
|
emojiCache.set(url, img);
|
|
return img;
|
|
}
|
|
|
|
type Segment = { type: "text"; v: string } | { type: "emoji"; v: string };
|
|
|
|
function splitSegments(text: string): Segment[] {
|
|
const re = emojiRegex();
|
|
const out: Segment[] = [];
|
|
let last = 0;
|
|
|
|
for (const m of text.matchAll(re)) {
|
|
const i = m.index ?? 0;
|
|
if (i > last) out.push({type: "text", v: text.slice(last, i)});
|
|
out.push({type: "emoji", v: m[0]});
|
|
last = i + m[0].length;
|
|
}
|
|
if (last < text.length) out.push({type: "text", v: text.slice(last)});
|
|
return out;
|
|
}
|
|
|
|
function measure(ctx: SKRSContext2D, s: string) {
|
|
return ctx.measureText(s).width;
|
|
}
|
|
|
|
function wrapSegments(ctx: SKRSContext2D, segments: Segment[], maxW: number, emojiW: number) {
|
|
const lines: { segments: Segment[]; width: number }[] = [];
|
|
let cur: Segment[] = [];
|
|
let w = 0;
|
|
|
|
const push = () => {
|
|
lines.push({segments: cur, width: w});
|
|
cur = [];
|
|
w = 0;
|
|
};
|
|
|
|
const add = (seg: Segment, segW: number) => {
|
|
if (cur.length && w + segW > maxW) push();
|
|
cur.push(seg);
|
|
w += segW;
|
|
};
|
|
|
|
for (const seg of segments) {
|
|
if (seg.type === "emoji") {
|
|
add(seg, emojiW);
|
|
continue;
|
|
}
|
|
|
|
// переносы/пробелы
|
|
const parts = seg.v.split(/(\s+)/);
|
|
for (const p of parts) {
|
|
if (!p) continue;
|
|
|
|
const sub = p.split("\n");
|
|
for (let si = 0; si < sub.length; si++) {
|
|
const chunk = sub[si];
|
|
if (chunk) add({type: "text", v: chunk}, measure(ctx, chunk));
|
|
if (si !== sub.length - 1) push();
|
|
}
|
|
}
|
|
}
|
|
|
|
if (cur.length) push();
|
|
return lines;
|
|
}
|
|
|
|
function lineWidth(ctx: SKRSContext2D, segments: Segment[], fontSize: number) {
|
|
const emojiSize = Math.round(fontSize * 1.05);
|
|
let w = 0;
|
|
for (const s of segments) {
|
|
w += s.type === "emoji" ? emojiSize : ctx.measureText(s.v).width;
|
|
}
|
|
return w;
|
|
}
|
|
|
|
function addEllipsisToFit(ctx: SKRSContext2D, segments: Segment[], maxW: number, fontSize: number): Segment[] {
|
|
const emojiSize = Math.round(fontSize * 1.05);
|
|
const ell: Segment = {type: "text", v: "…"};
|
|
const ellW = ctx.measureText("…").width;
|
|
|
|
const out = segments.map((s) => ({...s})) as Segment[];
|
|
|
|
const widthOf = (arr: Segment[]) => {
|
|
let w = 0;
|
|
for (const s of arr) w += s.type === "emoji" ? emojiSize : ctx.measureText(s.v).width;
|
|
return w;
|
|
};
|
|
|
|
while (out.length && widthOf(out) + ellW > maxW) {
|
|
const last = out[out.length - 1];
|
|
if (last.type === "emoji") {
|
|
out.pop();
|
|
continue;
|
|
}
|
|
if (last.v.length <= 1) {
|
|
out.pop();
|
|
continue;
|
|
}
|
|
last.v = last.v.slice(0, -1);
|
|
}
|
|
|
|
return [...out, ell];
|
|
}
|
|
|
|
async function drawLine(ctx: SKRSContext2D, line: Segment[], x: number, baselineY: number, fontSize: number) {
|
|
const emojiSize = Math.round(fontSize * 1.05);
|
|
let cx = x;
|
|
|
|
for (const seg of line) {
|
|
if (seg.type === "text") {
|
|
ctx.fillText(seg.v, cx, baselineY);
|
|
cx += measure(ctx, seg.v);
|
|
} else {
|
|
const img = await loadEmoji(seg.v);
|
|
const y = baselineY - emojiSize + Math.round(fontSize * 0.2);
|
|
ctx.drawImage(img, cx, y, emojiSize, emojiSize);
|
|
cx += emojiSize;
|
|
}
|
|
}
|
|
}
|
|
|
|
type Fitted = {
|
|
fontSize: number;
|
|
lineH: number;
|
|
lines: { segments: Segment[]; width: number }[];
|
|
truncated: boolean;
|
|
};
|
|
|
|
function fitQuoteToBox(ctx: SKRSContext2D, quoteWithOpen: string, boxW: number, boxH: number): Fitted {
|
|
const MAX_FONT = 64;
|
|
const MIN_FONT = 18;
|
|
const endSuffix = " »";
|
|
|
|
const segments = splitSegments(quoteWithOpen);
|
|
|
|
for (let fontSize = MAX_FONT; fontSize >= MIN_FONT; fontSize -= 2) {
|
|
const emojiSize = Math.round(fontSize * 1.05);
|
|
ctx.font = `${fontSize}px Inter, sans-serif`;
|
|
|
|
const lines = wrapSegments(ctx, segments, boxW, emojiSize);
|
|
const lineH = Math.round(fontSize * 1.20);
|
|
const totalH = lines.length * lineH;
|
|
|
|
if (!lines.length) continue;
|
|
|
|
const endW = ctx.measureText(endSuffix).width;
|
|
const last = lines[lines.length - 1];
|
|
|
|
if (totalH <= boxH && last.width + endW <= boxW) {
|
|
last.segments = [...last.segments, {type: "text", v: endSuffix}];
|
|
last.width += endW;
|
|
|
|
return {fontSize: fontSize, lineH, lines, truncated: false};
|
|
}
|
|
}
|
|
|
|
const fontSize = MIN_FONT;
|
|
const emojiSize = Math.round(fontSize * 1.05);
|
|
ctx.font = `${fontSize}px Inter, sans-serif`;
|
|
|
|
const lineH = Math.round(fontSize * 1.20);
|
|
const maxLinesByHeight = Math.max(1, Math.floor(boxH / lineH));
|
|
|
|
let lines = wrapSegments(ctx, segments, boxW, emojiSize);
|
|
|
|
const endW = ctx.measureText(endSuffix).width;
|
|
|
|
if (lines.length > maxLinesByHeight) {
|
|
lines = lines.slice(0, maxLinesByHeight);
|
|
const last = lines[lines.length - 1];
|
|
|
|
if (last.width + endW > boxW) {
|
|
last.segments = addEllipsisToFit(ctx, last.segments, boxW - endW, fontSize);
|
|
last.width = lineWidth(ctx, last.segments, fontSize);
|
|
} else {
|
|
last.segments = addEllipsisToFit(ctx, last.segments, boxW - endW, fontSize);
|
|
last.width = lineWidth(ctx, last.segments, fontSize);
|
|
}
|
|
} else {
|
|
const last = lines[lines.length - 1];
|
|
if (last && last.width + endW > boxW) {
|
|
last.segments = addEllipsisToFit(ctx, last.segments, boxW - endW, fontSize);
|
|
last.width = lineWidth(ctx, last.segments, fontSize);
|
|
}
|
|
}
|
|
|
|
if (lines.length) {
|
|
const last = lines[lines.length - 1];
|
|
last.segments = [...last.segments, {type: "text", v: endSuffix}];
|
|
last.width += endW;
|
|
}
|
|
|
|
return {fontSize: fontSize, lineH, lines, truncated: true};
|
|
}
|
|
|
|
async function getBackground(
|
|
reply: Message,
|
|
W: number,
|
|
H: number,
|
|
author: QuoteAuthor,
|
|
isForwarded: boolean
|
|
): Promise<Buffer> {
|
|
let src: Buffer | null = null;
|
|
|
|
const photoArr = reply.photo as PhotoSize[] | undefined;
|
|
const msgPhoto = photoArr && photoArr.length ? photoArr[photoArr.length - 1] : undefined;
|
|
|
|
if (msgPhoto?.file_id) {
|
|
const url = await getFileUrl(msgPhoto.file_id);
|
|
const res = await axios.get<ArrayBuffer>(url, {responseType: "arraybuffer"});
|
|
src = Buffer.from(res.data);
|
|
} else {
|
|
if (author.userId) {
|
|
src = await getUserAvatar(author.userId);
|
|
} else if (author.chatId) {
|
|
src = await getChatAvatar(author.chatId);
|
|
} else if (!isForwarded && reply.from?.id) {
|
|
src = await getUserAvatar(reply.from.id);
|
|
}
|
|
}
|
|
|
|
if (!src) {
|
|
return makeDarkGradientBgFancy(W, H, `${reply.message_id}-${reply.date ?? ""}`);
|
|
// return sharp({create: {width: W, height: H, channels: 3, background: "#1f1f1f"}})
|
|
// .png()
|
|
// .toBuffer();
|
|
}
|
|
|
|
return sharp(src)
|
|
.resize(W, H, {fit: "cover"})
|
|
.blur(18)
|
|
.modulate({brightness: 0.75, saturation: 1.1})
|
|
.png()
|
|
.toBuffer();
|
|
}
|
|
|
|
async function renderQuoteCard(quote: string, reply: Message) {
|
|
const W = 1280;
|
|
const H = 720;
|
|
|
|
const author = getQuoteAuthor(reply);
|
|
const forwarded = !!reply.forward_origin;
|
|
const name = author.name;
|
|
const userTag = author.username ? `@${author.username}` : "";
|
|
|
|
const me = botUser;
|
|
const botTag = me.username ? `@${me.username}` : "@bot";
|
|
|
|
const date = new Date((reply.date ?? Math.floor(Date.now() / 1000)) * 1000);
|
|
const dd = String(date.getDate()).padStart(2, "0");
|
|
const mm = String(date.getMonth() + 1).padStart(2, "0");
|
|
const yyyy = String(date.getFullYear());
|
|
const dateStr = `${dd}.${mm}.${yyyy}`;
|
|
|
|
const bgBuf = await getBackground(reply, W, H, author, forwarded);
|
|
|
|
const canvas = createCanvas(W, H);
|
|
const c = canvas.getContext("2d");
|
|
|
|
const bgImg = await loadImage(bgBuf);
|
|
c.drawImage(bgImg, 0, 0, W, H);
|
|
|
|
c.fillStyle = "rgba(0,0,0,0.35)";
|
|
c.fillRect(0, 0, W, H);
|
|
|
|
const edgePad = 56;
|
|
|
|
const reservedBottom = 140;
|
|
|
|
const quoteBoxX = edgePad;
|
|
const quoteBoxW = W - edgePad * 2;
|
|
|
|
const quoteTop = 90;
|
|
const quoteBottom = H - reservedBottom;
|
|
const quoteH = quoteBottom - quoteTop;
|
|
|
|
c.fillStyle = "rgba(255,255,255,0.92)";
|
|
c.textBaseline = "alphabetic";
|
|
c.shadowColor = "rgba(0,0,0,0.55)";
|
|
c.shadowBlur = 10;
|
|
c.shadowOffsetY = 2;
|
|
|
|
const quoteForFit = `« ${quote}`;
|
|
const fitted = fitQuoteToBox(c, quoteForFit, quoteBoxW, quoteH);
|
|
|
|
c.font = `${fitted.fontSize}px InterSemiBold, sans-serif`;
|
|
|
|
const totalTextH = fitted.lines.length * fitted.lineH;
|
|
let y = quoteTop + (quoteH - totalTextH) / 2 + fitted.fontSize;
|
|
|
|
for (const ln of fitted.lines) {
|
|
const x = quoteBoxX + (quoteBoxW - ln.width) / 2;
|
|
await drawLine(c, ln.segments, x, y, fitted.fontSize);
|
|
y += fitted.lineH;
|
|
}
|
|
|
|
c.shadowBlur = 0;
|
|
c.shadowOffsetY = 0;
|
|
c.fillStyle = "rgba(255,255,255,0.70)";
|
|
c.font = "28px InterLight, Inter, sans-serif";
|
|
c.textAlign = "center";
|
|
c.fillText(userTag ? `${name} | ${userTag}` : name, W / 2, H - 86);
|
|
|
|
c.font = "22px InterMedium, sans-serif";
|
|
c.fillStyle = "rgba(255,255,255,0.45)";
|
|
c.textAlign = "left";
|
|
c.fillText(botTag, edgePad, H - 34);
|
|
c.textAlign = "right";
|
|
c.fillText(dateStr, W - edgePad, H - 34);
|
|
|
|
return canvas.toBuffer("image/png");
|
|
}
|
|
|
|
type QuoteAuthor = {
|
|
name: string;
|
|
username?: string;
|
|
userId?: number;
|
|
chatId?: number;
|
|
};
|
|
|
|
function getQuoteAuthor(reply: Message): QuoteAuthor {
|
|
const origin = reply.forward_origin;
|
|
if (origin) {
|
|
switch (origin.type) {
|
|
case "user": {
|
|
const u = origin.sender_user;
|
|
const name = [u.first_name, u.last_name].filter(Boolean).join(" ") || u.username || "Unknown";
|
|
return {name, username: u.username, userId: u.id};
|
|
}
|
|
case "hidden_user": {
|
|
const name = origin.sender_user_name || "Unknown";
|
|
return {name};
|
|
}
|
|
}
|
|
}
|
|
|
|
const u = reply.from!;
|
|
const name = [u.first_name, u.last_name].filter(Boolean).join(" ") || u.username || "Unknown";
|
|
return {name, username: u.username, userId: u.id};
|
|
}
|