decord/src/update/tasks/webhook.ts

import { basename } from "node:path";
import type { Canvas } from "@napi-rs/canvas";
import {
	type APIMessageTopLevelComponent,
	ComponentType,
	MessageFlags,
	type RESTPostAPIWebhookWithTokenJSONBody,
	RouteBases,
	Routes,
} from "discord-api-types/v10";
import { drawSections } from "../../canvas";
import { makeSections } from "../../canvas/factory";
import { type CodeDiff, type Diff, type Differs, DiffType } from "../../types";
import { cuteVersion, maxDiffChanges, version } from "../shared";
import { assert, formatBytes, sortEntries } from "../utils";

function fileBase(path: string, other?: string) {
	if (!other) return `... ${basename(path)}`;
	const paths = path.split("/"),
		others = other.split("/");

	let offset = 0;
	while (offset < Math.min(paths.length, others.length) - 1 && paths[offset] === others[offset]) {
		offset++;
	}

	return `... ${paths.slice(offset).join("/")}`;
}

function makeFooter(size: number, item: string) {
	return `**${size.toLocaleString("en-US")}** ${item} change${size !== 1 ? "s" : ""} total`;
}

function getThreshold(lines: string[]) {
	return lines.some((x) => x.includes("\n")) ? Math.floor(maxDiffChanges / 2) : maxDiffChanges;
}

function cap(lines: string[], item: string) {
	const threshold = getThreshold(lines);
	if (lines.length > threshold)
		return [...lines.slice(0, threshold), `(+${(lines.length - threshold).toLocaleString("en-US")} ${item})`];
	return lines;
}

interface FormattedDiff {
	embed: string[];
	file: string;
	capped: boolean;
}

function formatDiff(diffs: Map<string, Diff | CodeDiff>): FormattedDiff {
	const entries = [...diffs.entries()];

	const sections = {
		Added: sortEntries(entries)
			.map(
				([name, diff]) =>
					diff.type === DiffType.Added &&
					("size" in diff
						? `+ ${fileBase(name)} (${formatBytes(diff.size)})`
						: `+ ${name}: ${diff.label || diff.source}`),
			)
			.filter((x) => typeof x === "string"),
		Changed: sortEntries(entries)
			.map(
				([name, diff]) =>
					diff.type === DiffType.Changed &&
					`- ${name}: ${diff.oldLabel || diff.oldSource}\n+ ${name}: ${diff.label || diff.source}`,
			)
			.filter((x) => typeof x === "string"),
		Renamed: sortEntries(entries)
			.map(
				([name, diff]) =>
					diff.type === DiffType.Renamed &&
					("size" in diff
						? `- ${fileBase(diff.oldName, name)}\n+ ${fileBase(name, diff.oldName)}`
						: `- ${diff.oldName}\n+ ${name}`),
			)
			.filter((x) => typeof x === "string"),
		Removed: sortEntries(entries)
			.map(
				([name, diff]) =>
					diff.type === DiffType.Removed &&
					("size" in diff
						? `- ${fileBase(name)} (${formatBytes(diff.size)})`
						: `- ${name}: ${diff.label || diff.source}`),
			)
			.filter((x) => typeof x === "string"),
	};

	const text = Object.entries(sections).filter(([, lines]) => lines.length);
	return {
		embed: text.map(
			([title, lines]) => `### ${title}\n\`\`\`diff\n${cap(lines, title.toLowerCase()).join("\n")}\`\`\``,
		),
		file: text.map(([title, lines]) => `# ${title}\n\n${lines.join("\n")}`).join("\n\n"),
		capped: text.some(([, lines]) => lines.length > getThreshold(lines)),
	};
}

interface WebhookEmbed {
	title: string;
	body: FormattedDiff;
	image?: Canvas;
	footer: string;
	key: string;
}

async function sendWebhook(webhook: string, role: string, embeds: WebhookEmbed[], color?: number) {
	const body = new FormData();
	const images = new Map<number, string>(),
		diffs = new Map<number, string>();

	let fi = 0;
	for (let i = 0; i < embeds.length; i++) {
		const embed = embeds[i];
		if (embed.image) {
			const key = fi++;
			const filename = `${embed.key}.png`;
			images.set(i, filename);

			body.append(`files[${key}]`, new Blob([embed.image.toBuffer("image/png")], { type: "image/png" }), filename);
		}
		if (embed.body.capped) {
			const key = fi++;
			const filename = `${embed.key}-changes.diff`;
			diffs.set(i, filename);

			body.append(`files[${key}]`, new Blob([embed.body.file], { type: "text/plain" }), filename);
		}
	}

	body.append(
		"payload_json",
		JSON.stringify({
			flags: MessageFlags.IsComponentsV2,
			components: [
				{
					type: ComponentType.TextDisplay,
					content: `<@&${role}>`,
				},
				...embeds.map(
					({ title, body, image, footer }, i) =>
						({
							type: ComponentType.Container,
							accent_color: color,
							components: [
								{
									type: ComponentType.TextDisplay,
									content: `## ${title}\n-# ${version} (${cuteVersion})`,
								},
								...body.embed.map((content) => ({
									type: ComponentType.TextDisplay,
									content,
								})),
								body.capped && {
									type: ComponentType.File,
									file: {
										url: `attachment://${diffs.get(i)}`,
									},
								},
								body.capped && {
									type: ComponentType.Separator,
								},
								image && {
									type: ComponentType.MediaGallery,
									items: [
										{
											media: {
												url: `attachment://${images.get(i)}`,
											},
										},
									],
								},
								{
									type: ComponentType.TextDisplay,
									content: footer,
								},
							].filter((x) => !!x),
						}) as APIMessageTopLevelComponent,
				),
			],
			allowed_mentions: process.env.NODE_ENV === "test" ? { parse: [] } : { roles: [role] },
		} as RESTPostAPIWebhookWithTokenJSONBody),
	);

	const url = new URL(webhook);
	url.searchParams.set("wait", "true");
	url.searchParams.set("with_components", "true");

	const res = await fetch(url, {
		method: "POST",
		body,
	});

	const data = await res.text();
	if (!res.ok)
		throw new Error(`Failed to send webhook message with embeds ${embeds.map((x) => x.title).join(", ")}: ${data}`);

	const discordToken = process.env.bot_token;
	if (!discordToken) return;

	// this can fail quietly
	try {
		const message = JSON.parse(data);

		const route = `${RouteBases.api}/${Routes.channelMessageCrosspost(message.channel_id, message.id)}`;
		await fetch(route, {
			method: "POST",
			headers: {
				Authorization: discordToken,
			},
		});
	} catch {}
}

export async function webhook(diffs: Differs) {
	if (diffs.raw?.size || diffs.semantic?.size)
		await sendWebhook(
			assert(process.env.color_webhook, "Missing color webhook env"),
			"1227327297795657850",
			[
				diffs.raw?.size && {
					title: "Raw colors",
					body: formatDiff(diffs.raw),
					image: drawSections(await makeSections(diffs.raw)),
					footer: makeFooter(diffs.raw.size, "raw color"),
					key: "raw-colors",
				},
				diffs.semantic?.size && {
					title: "Semantic colors",
					body: formatDiff(diffs.semantic),
					image: drawSections(await makeSections(diffs.semantic)),
					footer: makeFooter(diffs.semantic.size, "semantic color"),
					key: "semantic-colors",
				},
			].filter((x) => !!x),
			0xd8ef9e,
		);

	if (diffs.icons?.size)
		await sendWebhook(
			assert(process.env.icons_webhook, "Missing icons webhook env"),
			"1227327765079003217",
			[
				{
					title: "Icons",
					body: formatDiff(diffs.icons),
					image: drawSections(await makeSections(diffs.icons, true)),
					footer: makeFooter(diffs.icons.size, "icon"),
					key: "icons",
				},
			],
			0xf4b8f7,
		);

	if (diffs.code?.size)
		await sendWebhook(
			assert(process.env.code_webhook, "Missing code webhook env"),
			"1233861867059941387",
			[
				{
					title: "Code",
					body: formatDiff(diffs.code),
					footer: makeFooter(diffs.code.size, "code"),
					key: "code",
				},
			],
			0x86faf3,
		);
}