This commit is contained in:
loicbersier 2020-02-28 00:31:08 +01:00
parent 4b8297282b
commit 69cd99d3d6

View file

@ -3,6 +3,8 @@ const util = require('util');
const exec = util.promisify(require('child_process').exec); const exec = util.promisify(require('child_process').exec);
const youtubedl = require('youtube-dl'); const youtubedl = require('youtube-dl');
const os = require('os'); const os = require('os');
const filetype = require('file-type');
const fs = require('fs');
class midifyCommand extends Command { class midifyCommand extends Command {
constructor() { constructor() {
@ -37,7 +39,7 @@ class midifyCommand extends Command {
url = Attachment[0].url; url = Attachment[0].url;
} }
let input = `${os.tmpdir()}/${message.id}.mp4`; let input = `${os.tmpdir()}/${message.id}`;
let input2 = `${os.tmpdir()}/${message.id}.wav`; let input2 = `${os.tmpdir()}/${message.id}.wav`;
let output = `${os.tmpdir()}/${message.id}.mid`; let output = `${os.tmpdir()}/${message.id}.mid`;
let output2 = `${os.tmpdir()}/${message.id}.mp3`; let output2 = `${os.tmpdir()}/${message.id}.mp3`;
@ -46,12 +48,26 @@ class midifyCommand extends Command {
let loadingmsg = await message.channel.send('Processing (this can take some time) <a:loadingmin:527579785212329984>'); let loadingmsg = await message.channel.send('Processing (this can take some time) <a:loadingmin:527579785212329984>');
if (url) { if (url) {
return youtubedl.exec(url, ['--format=mp4', '-o', input], {}, function(err) { return youtubedl.exec(url, ['-o', input], {}, 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.');
} else { }
let ext = 'mp4';
if (fs.existsSync(`${os.tmpdir()}/${input}`)) {
ext = await filetype.fromFile(`${os.tmpdir()}/${input}`);
ext = ext.ext; // This look stupid but hey, it work
if (ext == '3gp') ext = 'mp4'; // Change 3gp file extension to mp4 so discord show the video ( and to stop people from complaining )
fs.renameSync(`${os.tmpdir()}/${input}`, `${os.tmpdir()}/${input}.${ext}`);
} else if (fs.existsSync(`${os.tmpdir()}/${input}.mkv`)) { // If it can't find the video assume it got merged and end with mkv
fs.renameSync(`${os.tmpdir()}/${input}.mkv`, `${os.tmpdir()}/${input}.mp4`); // Discord play mkv just fine but it need to end with mp4
}
input = `${os.tmpdir()}/${message.id}.${ext}`;
// Convert to wav // Convert to wav
exec(`ffmpeg -i ${input} ${input2}`) exec(`ffmpeg -i ${input} ${input2}`)
.then(() => { .then(() => {
@ -61,7 +77,7 @@ class midifyCommand extends Command {
console.error(err); console.error(err);
return message.channel.send('Oh no! an error has occured during the conversion, are you sure it is a valid file?'); return message.channel.send('Oh no! an error has occured during the conversion, are you sure it is a valid file?');
}); });
}
}); });
} else { } else {
return message.channel.send('You need a valid video link!'); return message.channel.send('You need a valid video link!');