You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
153 lines
4.1 KiB
153 lines
4.1 KiB
2 years ago
|
import renderLottie from "puppeteer-lottie";
|
||
|
import * as fs from "fs/promises";
|
||
|
import { spawn } from "node:child_process";
|
||
|
|
||
|
const API_URL = "https://discord.com/api/v9/sticker-packs";
|
||
|
const STICKER_SIZE = 160;
|
||
|
|
||
|
const LOTTIE_BASE_URL = "https://discord.com/stickers/{id}.json";
|
||
|
const PNG_BASE_URL = "https://media.discordapp.net/stickers/{id}.png";
|
||
|
|
||
|
interface StickerPacks {
|
||
|
sticker_packs: StickerPack[];
|
||
|
}
|
||
|
|
||
|
interface StickerPack {
|
||
|
id: string;
|
||
|
stickers: Sticker[];
|
||
|
name: string;
|
||
|
sku_id: string;
|
||
|
cover_sticker_id: string;
|
||
|
description: string;
|
||
|
banner_asset_id?: string;
|
||
|
}
|
||
|
|
||
|
interface Sticker {
|
||
|
id: string;
|
||
|
name: string;
|
||
|
tags: string;
|
||
|
type: number;
|
||
|
format_type: StickerFormat;
|
||
|
description: string;
|
||
|
asset: string;
|
||
|
pack_id: string;
|
||
|
sort_value: number;
|
||
|
}
|
||
|
|
||
|
enum StickerFormat {
|
||
|
STATIC = 1,
|
||
|
APNG = 2,
|
||
|
LOTTIE = 3,
|
||
|
}
|
||
|
|
||
|
async function findFilesWithPrefix(prefix: string, path: string): Promise<string[]> {
|
||
|
const files = await fs.readdir(path);
|
||
|
const filteredFiles = files.filter((file) => file.startsWith(prefix));
|
||
|
return filteredFiles;
|
||
|
}
|
||
|
|
||
|
const stickerPacks: StickerPacks = await (await fetch(API_URL)).json();
|
||
|
|
||
|
await fs.mkdir("output");
|
||
|
await fs.mkdir("temp");
|
||
|
|
||
|
for (const pack of stickerPacks.sticker_packs) {
|
||
|
for (const sticker of pack.stickers) {
|
||
|
switch (sticker.format_type) {
|
||
|
case StickerFormat.STATIC:
|
||
|
{
|
||
|
const stickerFile = await (
|
||
|
await fetch(PNG_BASE_URL.replace("{id}", sticker.id))
|
||
|
).blob();
|
||
|
await fs.writeFile(
|
||
|
`output/${sticker.id}.png`,
|
||
|
Buffer.from(await stickerFile.arrayBuffer())
|
||
|
);
|
||
|
}
|
||
|
break;
|
||
|
|
||
|
case StickerFormat.APNG:
|
||
|
{
|
||
|
const stickerFile = await (
|
||
|
await fetch(PNG_BASE_URL.replace("{id}", sticker.id))
|
||
|
).blob();
|
||
|
await fs.writeFile(
|
||
|
`temp/${sticker.id}.apng`,
|
||
|
Buffer.from(await stickerFile.arrayBuffer())
|
||
|
);
|
||
|
|
||
|
const ffmpeg = spawn("ffmpeg", [
|
||
|
"-v",
|
||
|
"error",
|
||
|
"-i",
|
||
|
`temp/${sticker.id}.apng`,
|
||
|
`temp/${sticker.id}-%8d.png`,
|
||
|
]);
|
||
|
ffmpeg.stdout.on("data", (data) => {
|
||
|
console.log(`stdout: ${data}`);
|
||
|
});
|
||
|
ffmpeg.stderr.on("data", (data) => {
|
||
|
console.error(`stderr: ${data}`);
|
||
|
});
|
||
|
await new Promise((resolve) => ffmpeg.on("exit", resolve));
|
||
|
|
||
|
let framerate = "";
|
||
|
|
||
|
const ffprobe = spawn("ffprobe", [
|
||
|
"-v",
|
||
|
"error",
|
||
|
"-select_streams",
|
||
|
"v",
|
||
|
"-of",
|
||
|
"default=noprint_wrappers=1:nokey=1",
|
||
|
"-show_entries",
|
||
|
"stream=r_frame_rate",
|
||
|
`temp/${sticker.id}.apng`
|
||
|
]);
|
||
|
ffprobe.stdout.on("data", (data) => {
|
||
|
console.log(`stdout: ${data}`);
|
||
|
framerate += data;
|
||
|
});
|
||
|
ffprobe.stderr.on("data", (data) => {
|
||
|
console.error(`stderr: ${data}`);
|
||
|
});
|
||
|
await new Promise((resolve) => ffprobe.on("exit", resolve));
|
||
|
|
||
|
framerate = eval(framerate);
|
||
|
|
||
|
const gifski = spawn("gifski", [
|
||
|
"-o",
|
||
|
`output/${sticker.id}.gif`,
|
||
|
"--fps",
|
||
|
framerate,
|
||
|
...(await findFilesWithPrefix(`${sticker.id}-`, "temp")).map(e => "temp/" + e)
|
||
|
]);
|
||
|
gifski.stdout.on("data", (data) => {
|
||
|
console.log(`stdout: ${data}`);
|
||
|
});
|
||
|
gifski.stderr.on("data", (data) => {
|
||
|
console.error(`stderr: ${data}`);
|
||
|
});
|
||
|
await new Promise((resolve) => gifski.on("exit", resolve));
|
||
|
}
|
||
|
break;
|
||
|
|
||
|
case StickerFormat.LOTTIE:
|
||
|
{
|
||
|
const stickerFile = await (
|
||
|
await fetch(LOTTIE_BASE_URL.replace("{id}", sticker.id))
|
||
|
).json();
|
||
|
await renderLottie({
|
||
|
animationData: stickerFile,
|
||
|
output: `output/${sticker.id}.gif`,
|
||
|
width: STICKER_SIZE,
|
||
|
height: STICKER_SIZE,
|
||
|
});
|
||
|
}
|
||
|
break;
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
await fs.rm("temp", { recursive: true });
|