mirror of
https://github.com/Vendicated/Vencord.git
synced 2024-09-21 07:00:34 +00:00
This keeps leading to issues due to circular imports. Import from specific files instead, index just reexports
182 lines
7.3 KiB
TypeScript
182 lines
7.3 KiB
TypeScript
/*
|
|
* Vencord, a modification for Discord's desktop app
|
|
* Copyright (c) 2022 Vendicated and contributors
|
|
*
|
|
* This program is free software: you can redistribute it and/or modify
|
|
* it under the terms of the GNU General Public License as published by
|
|
* the Free Software Foundation, either version 3 of the License, or
|
|
* (at your option) any later version.
|
|
*
|
|
* This program is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with this program. If not, see <https://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
import { ApplicationCommandInputType, ApplicationCommandOptionType, Argument, CommandContext, findOption } from "../api/Commands";
|
|
import { Devs } from "../utils/constants";
|
|
import { getGifEncoder } from "../utils/dependencies";
|
|
import { lazyWebpack, makeLazy } from "../utils/misc";
|
|
import definePlugin from "../utils/types";
|
|
import { filters } from "../webpack";
|
|
|
|
const DRAFT_TYPE = 0;
|
|
const DEFAULT_DELAY = 20;
|
|
const DEFAULT_RESOLUTION = 128;
|
|
const FRAMES = 10;
|
|
|
|
const getFrames = makeLazy(() => 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);
|
|
};
|
|
img.onerror = (event, _source, _lineno, _colno, err) => reject(err || event);
|
|
img.crossOrigin = "Anonymous";
|
|
img.src = url;
|
|
});
|
|
}
|
|
|
|
async function resolveImage(options: Argument[], ctx: CommandContext, noServerPfp: boolean): Promise<File | string | null> {
|
|
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);
|
|
return user.getAvatarURL(noServerPfp ? void 0 : ctx.guild?.id, 2048).replace(/\?size=\d+$/, "?size=2048");
|
|
} 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
|
|
},
|
|
{
|
|
name: "no-server-pfp",
|
|
description: "Use the normal avatar instead of the server specific one when using the 'user' option",
|
|
type: ApplicationCommandOptionType.BOOLEAN
|
|
}
|
|
],
|
|
execute: async (opts, cmdCtx) => {
|
|
const { GIFEncoder, quantize, applyPalette } = await getGifEncoder();
|
|
const frames = await getFrames();
|
|
|
|
const noServerPfp = findOption(opts, "no-server-pfp", false);
|
|
try {
|
|
var url = await resolveImage(opts, cmdCtx, noServerPfp);
|
|
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));
|
|
},
|
|
},
|
|
]
|
|
});
|