95 lines
3.8 KiB
JavaScript
95 lines
3.8 KiB
JavaScript
const fetch = require("node-fetch");
|
|
const AbortController = require("abort-controller");
|
|
const fileType = require("file-type");
|
|
const { promisify } = require("util");
|
|
const writeFile = promisify(require("fs").writeFile);
|
|
const execPromise = promisify(require("child_process").exec);
|
|
const urlRegex = /(?:\w+:)?\/\/(\S+)/;
|
|
|
|
// this checks if the file is, in fact, an image
|
|
const typeCheck = async (image, image2, gifv = false) => {
|
|
// download the file to a buffer
|
|
const controller = new AbortController();
|
|
const timeout = setTimeout(() => {
|
|
controller.abort();
|
|
}, 15000);
|
|
try {
|
|
const imageRequest = await fetch(image, { signal: controller.signal });
|
|
const imageBuffer = await imageRequest.buffer();
|
|
if (imageBuffer.size >= 25 * 1024 * 1024) return;
|
|
// get the file type
|
|
const imageType = await fileType.fromBuffer(imageBuffer);
|
|
// check if the file is a jpeg, png, or webp
|
|
const formats = ["image/jpeg", "image/png", "image/webp", "image/gif"];
|
|
if (gifv) formats.push("video/mp4");
|
|
if (imageType && formats.includes(imageType.mime)) {
|
|
// if it is, then return the url with the file type
|
|
const path = `/tmp/${Math.random().toString(36).substring(2, 15)}.${imageType.ext}`;
|
|
await writeFile(path, imageBuffer);
|
|
const payload = {
|
|
data: imageBuffer,
|
|
type: imageType.ext !== "mp4" ? (imageType.ext === "jpg" ? "jpeg" : imageType.ext) : "gif",
|
|
path: path,
|
|
url: image2
|
|
};
|
|
if (gifv) payload.delay = (await execPromise(`ffprobe -v 0 -of csv=p=0 -select_streams v:0 -show_entries stream=r_frame_rate ${path}`)).stdout.replace("\n", "");
|
|
return payload;
|
|
} else {
|
|
// if not, then return false
|
|
return false;
|
|
}
|
|
} catch (error) {
|
|
if (error.name === "AbortError") {
|
|
throw Error("Timed out");
|
|
} else {
|
|
throw error;
|
|
}
|
|
} finally {
|
|
clearTimeout(timeout);
|
|
}
|
|
};
|
|
|
|
// this checks for the latest message containing an image and returns the url of the image
|
|
module.exports = async (cmdMessage) => {
|
|
// we start by getting the messages
|
|
const messages = await cmdMessage.channel.getMessages();
|
|
// iterate over each message
|
|
for (const message of messages) {
|
|
// check the attachments first
|
|
if (message.embeds.length !== 0) {
|
|
// embeds can have 2 possible entries with images, we check the thumbnail first
|
|
if (message.embeds[0].type === "gifv") {
|
|
const type = await typeCheck(message.embeds[0].video.url, message.embeds[0].video.url, true);
|
|
if (type === false) continue;
|
|
return type;
|
|
} else if (message.embeds[0].thumbnail) {
|
|
const type = await typeCheck(message.embeds[0].thumbnail.proxy_url, message.embeds[0].thumbnail.url);
|
|
if (type === false) continue;
|
|
return type;
|
|
// if there isn't a thumbnail check the image area
|
|
} else if (message.embeds[0].image) {
|
|
const type = await typeCheck(message.embeds[0].image.proxy_url, message.embeds[0].image.url);
|
|
if (type === false) continue;
|
|
return type;
|
|
}
|
|
} else if (message.attachments.length !== 0) {
|
|
// get type of file
|
|
const type = await typeCheck(message.attachments[0].proxy_url, message.attachments[0].url);
|
|
// move to the next message if the file isn't an image
|
|
if (type === false) continue;
|
|
// if the file is an image then return it
|
|
return type;
|
|
// if there's nothing in the attachments check the urls in the message if there are any
|
|
} else if (urlRegex.test(message.content)) {
|
|
// get url
|
|
const url = message.content.match(urlRegex);
|
|
// get type of file
|
|
const type = await typeCheck(url[0], url[0]);
|
|
// move to the next message if the file isn't an image
|
|
if (type === false) continue;
|
|
// if the file is an image then return it
|
|
return type;
|
|
// if there's no urls then check the embeds
|
|
}
|
|
}
|
|
};
|