joan/Haha-Yes
1
0
Fork 0
forked from Supositware/Haha-Yes
Code Pull requests Activity

Make it work with other file format than mp4

Browse source
This commit is contained in:
loicbersier 2020-02-24 16:23:13 +01:00
parent 946a4f70cc
commit ec071d8025
1 changed files with 17 additions and 11 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

28
commands/utility/download.js
View file

@ -3,6 +3,7 @@ const fs = require('fs');
const youtubedl = require('youtube-dl'); const youtubedl = require('youtube-dl');
const hbjs = require('handbrake-js'); const hbjs = require('handbrake-js');
const os = require('os'); const os = require('os');
const filetype = require('file-type');
class DownloadCommand extends Command { class DownloadCommand extends Command {
constructor() { constructor() {
@ -60,19 +61,24 @@ class DownloadCommand extends Command {
if (link.includes('http') || link.includes('www')) { if (link.includes('http') || link.includes('www')) {
let loadingmsg = await message.channel.send('Downloading <a:loadingmin:527579785212329984>'); let loadingmsg = await message.channel.send('Downloading <a:loadingmin:527579785212329984>');
if (fs.existsSync(`${os.tmpdir()}/${fileName}.mp4`)) { if (fs.existsSync(`${os.tmpdir()}/${fileName}`)) {
fs.unlink(`${os.tmpdir()}/${fileName}.mp4`, (err) => { fs.unlink(`${os.tmpdir()}/${fileName}`, (err) => {
if (err); if (err);
}); });
} }
return youtubedl.exec(link, ['--format=mp4', '-o', `${os.tmpdir()}/${fileName}.mp4`], {}, async function(err) { return youtubedl.exec(link, ['-o', `${os.tmpdir()}/${fileName}`], {}, async function(err) {
if (err) { if (err) {
console.error(err); console.error(err);
loadingmsg.delete(); loadingmsg.delete();
return message.channel.send('An error has occured, I can\'t download from the link you provided.'); return message.channel.send('An error has occured, I can\'t download from the link you provided.');
} }
let file = fs.statSync(`${os.tmpdir()}/${fileName}.mp4`); let ext = await filetype.fromFile(`${os.tmpdir()}/${fileName}`);
ext = ext.ext; // This look stupid but hey, it work
fs.renameSync(`${os.tmpdir()}/${fileName}`, `${os.tmpdir()}/${fileName}.${ext}`);
let file = fs.statSync(`${os.tmpdir()}/${fileName}.${ext}`);
let fileSize = file.size / 1000000.0; let fileSize = file.size / 1000000.0;
//Compress vid if bigger than 8MB //Compress vid if bigger than 8MB
@ -81,8 +87,8 @@ class DownloadCommand extends Command {
loadingmsg.delete(); loadingmsg.delete();
const options = { const options = {
input: `${os.tmpdir()}/${fileName}.mp4`, input: `${os.tmpdir()}/${fileName}.${ext}`,
output: `${os.tmpdir()}/${fileName}compressed.mp4`, output: `${os.tmpdir()}/${fileName}compressed.${ext}`,
preset: 'General/Gmail Small 10 Minutes 288p30' preset: 'General/Gmail Small 10 Minutes 288p30'
}; };
@ -110,7 +116,7 @@ class DownloadCommand extends Command {
handbrake.on('end', async function () { handbrake.on('end', async function () {
clearInterval(editmsg); clearInterval(editmsg);
file = fs.statSync(`${os.tmpdir()}/${fileName}compressed.mp4`); file = fs.statSync(`${os.tmpdir()}/${fileName}compressed.${ext}`);
fileSize = file.size / 1000000.0; fileSize = file.size / 1000000.0;
compressmsg.delete(); compressmsg.delete();
@ -120,28 +126,28 @@ class DownloadCommand extends Command {
return message.channel.send('File too big!'); return message.channel.send('File too big!');
} }
return message.channel.send({embed: Embed, files: [`${os.tmpdir()}/${fileName}compressed.mp4`]}) return message.channel.send({embed: Embed, files: [`${os.tmpdir()}/${fileName}compressed.${ext}`]})
.catch(err => { .catch(err => {
console.error(err); console.error(err);
return message.channel.send('File too big'); return message.channel.send('File too big');
}) })
.then(() => { .then(() => {
// Delete file after it have been sent // Delete file after it have been sent
fs.unlinkSync(`${os.tmpdir()}/${fileName}.mp4`); fs.unlinkSync(`${os.tmpdir()}/${fileName}.${ext}`);
}); });
}); });
} else { } else {
loadingmsg.delete(); loadingmsg.delete();
message.delete(); message.delete();
message.channel.send({embed: Embed, files: [`${os.tmpdir()}/${fileName}.mp4`]}) message.channel.send({embed: Embed, files: [`${os.tmpdir()}/${fileName}.${ext}`]})
.catch(err => { .catch(err => {
console.error(err); console.error(err);
message.channel.send('File too big'); message.channel.send('File too big');
}) })
.then(() => { .then(() => {
// Delete file after it have been sent // Delete file after it have been sent
fs.unlinkSync(`${os.tmpdir()}/${fileName}.mp4`); fs.unlinkSync(`${os.tmpdir()}/${fileName}.${ext}`);
}); });
} }
}); });