Vencord/src/plugins/petpet.ts

169 lines
6.8 KiB
TypeScript
Raw Normal View History

2022-10-12 03:35:34 +00:00
import { ApplicationCommandOptionType, findOption, ApplicationCommandInputType, Argument, CommandContext } from "../api/Commands";
import { Devs } from "../utils/constants";
import definePlugin from "../utils/types";
2022-10-12 20:22:37 +00:00
import { lazy, lazyWebpack } from "../utils/misc";
2022-10-12 03:35:34 +00:00
import { filters } from "../webpack";
const DRAFT_TYPE = 0;
const DEFAULT_DELAY = 20;
const DEFAULT_RESOLUTION = 128;
const FRAMES = 10;
// https://github.com/mattdesl/gifenc
// this lib is way better than gif.js and all other libs, they're all so terrible but this one is nice
// @ts-ignore ts mad
const getGifEncoder = lazy(() => import("https://unpkg.com/gifenc@1.0.3/dist/gifenc.esm.js"));
const getFrames = lazy(() => Promise.all(
Array.from(
{ length: FRAMES },
(_, i) => loadImage(`https://raw.githubusercontent.com/VenPlugs/petpet/main/frames/pet${i}.gif`)
))
);
const fetchUser = lazyWebpack(filters.byCode(".USER("));
const promptToUpload = lazyWebpack(filters.byCode("UPLOAD_FILE_LIMIT_ERROR"));
const UploadStore = lazyWebpack(filters.byProps(["getUploads"]));
function loadImage(source: File | string) {
const isFile = source instanceof File;
const url = isFile ? URL.createObjectURL(source) : source;
return new Promise<HTMLImageElement>((resolve, reject) => {
const img = new Image();
img.onload = () => {
if (isFile)
URL.revokeObjectURL(url);
resolve(img);
};
2022-10-12 20:22:37 +00:00
img.onerror = (event, _source, _lineno, _colno, err) => reject(err || event);
2022-10-12 03:35:34 +00:00
img.crossOrigin = "Anonymous";
img.src = url;
});
}
2022-10-12 03:51:09 +00:00
async function resolveImage(options: Argument[], ctx: CommandContext, noServerPfp: boolean): Promise<File | string | null> {
2022-10-12 03:35:34 +00:00
for (const opt of options) {
switch (opt.name) {
case "image":
const upload = UploadStore.getUploads(ctx.channel.id, DRAFT_TYPE)[0];
if (upload) {
if (!upload.isImage) throw "Upload is not an image";
return upload.item.file;
}
break;
case "url":
return opt.value;
case "user":
try {
const user = await fetchUser(opt.value);
2022-10-12 03:51:09 +00:00
return user.getAvatarURL(noServerPfp ? void 0 : ctx.guild?.id, 2048).replace(/\?size=\d+$/, "?size=2048");
2022-10-12 03:35:34 +00:00
} catch (err) {
console.error("[petpet] Failed to fetch user\n", err);
throw "Failed to fetch user. Check the console for more info.";
}
}
}
return null;
}
export default definePlugin({
name: "petpet",
description: "headpet a cutie",
authors: [Devs.Ven],
dependencies: ["CommandsAPI"],
commands: [
{
inputType: ApplicationCommandInputType.BUILT_IN,
name: "petpet",
description: "Create a petpet gif. You can only specify one of the image options",
options: [
{
name: "delay",
description: "The delay between each frame. Defaults to 20.",
type: ApplicationCommandOptionType.INTEGER
},
{
name: "resolution",
description: "Resolution for the gif. Defaults to 120. If you enter an insane number and it freezes Discord that's your fault.",
type: ApplicationCommandOptionType.INTEGER
},
{
name: "image",
description: "Image attachment to use",
type: ApplicationCommandOptionType.ATTACHMENT
},
{
name: "url",
description: "URL to fetch image from",
type: ApplicationCommandOptionType.STRING
},
{
name: "user",
description: "User whose avatar to use as image",
type: ApplicationCommandOptionType.USER
2022-10-12 03:51:09 +00:00
},
{
name: "no-server-pfp",
description: "Use the normal avatar instead of the server specific one when using the 'user' option",
type: ApplicationCommandOptionType.BOOLEAN
2022-10-12 03:35:34 +00:00
}
],
2022-10-12 20:22:37 +00:00
execute: async (opts, cmdCtx) => {
2022-10-12 03:35:34 +00:00
const { GIFEncoder, quantize, applyPalette } = await getGifEncoder();
const frames = await getFrames();
2022-10-12 03:51:09 +00:00
const noServerPfp = findOption(opts, "no-server-pfp", false);
2022-10-12 03:35:34 +00:00
try {
2022-10-12 03:51:09 +00:00
var url = await resolveImage(opts, cmdCtx, noServerPfp);
2022-10-12 03:35:34 +00:00
if (!url) throw "No Image specified!";
} catch (err) {
// Todo make this send a clyde message once that PR is done
console.log(err);
return;
}
const avatar = await loadImage(url);
const delay = findOption(opts, "delay", DEFAULT_DELAY);
const resolution = findOption(opts, "resolution", DEFAULT_RESOLUTION);
const gif = new GIFEncoder();
const canvas = document.createElement("canvas");
canvas.width = canvas.height = resolution;
const ctx = canvas.getContext("2d")!;
for (let i = 0; i < FRAMES; i++) {
ctx.clearRect(0, 0, canvas.width, canvas.height);
const j = i < FRAMES / 2 ? i : FRAMES - i;
const width = 0.8 + j * 0.02;
const height = 0.8 - j * 0.05;
const offsetX = (1 - width) * 0.5 + 0.1;
const offsetY = 1 - height - 0.08;
ctx.drawImage(avatar, offsetX * resolution, offsetY * resolution, width * resolution, height * resolution);
ctx.drawImage(frames[i], 0, 0, resolution, resolution);
const { data } = ctx.getImageData(0, 0, resolution, resolution);
const palette = quantize(data, 256);
const index = applyPalette(data, palette);
gif.writeFrame(index, resolution, resolution, {
transparent: true,
palette,
delay,
});
}
gif.finish();
const file = new File([gif.bytesView()], "petpet.gif", { type: "image/gif" });
// Immediately after the command finishes, Discord clears all input, including pending attachments.
// Thus, setImmediate is needed to make this execute after Discord cleared the input
setImmediate(() => promptToUpload([file], cmdCtx.channel, DRAFT_TYPE));
2022-10-12 20:22:37 +00:00
},
2022-10-12 03:35:34 +00:00
},
]
});