From a6d1adf657ff8ef5fb1103fd34f9a26a47c94b97 Mon Sep 17 00:00:00 2001
From: loicbersier <loic.bersier1@gmail.com>
Date: Mon, 25 Nov 2019 23:41:44 +0100
Subject: [PATCH] Check for file size
---
commands/general/tweet.js | 9 +++++++++
1 file changed, 9 insertions(+)
diff --git a/commands/general/tweet.js b/commands/general/tweet.js
index 9fdcdf7..4e658ae 100644
--- a/commands/general/tweet.js
+++ b/commands/general/tweet.js
@@ -98,6 +98,15 @@ class tweetCommand extends Command {
const dest = fs.createWriteStream(`${os.tmpdir()}/${Attachment[0].name}`);
res.body.pipe(dest);
dest.on('finish', () => {
+ let file = fs.statSync(`${os.tmpdir()}/${Attachment[0].name}`);
+ let fileSize = file.size / 1000000.0;
+
+ if ((Attachment[0].name.endsWith('.jpg') || Attachment[0].name.endsWith('.png')) && fileSize > 5) {
+ return message.channel.send('Images can\'t be larger than 5 MB!');
+ } else if (Attachment[0].name.endsWith('.gif') && fileSize > 15) {
+ return message.channel.send('Gifs can\'t be larger than 15 MB!');
+ }
+
let b64Image = fs.readFileSync(`${os.tmpdir()}/${Attachment[0].name}`, { encoding: 'base64'});
T.post('media/upload', { media_data: b64Image }, function (err, data) {
if (err) {