Haha-Yes/commands/utility/download.js

const { Command } = require('discord-akairo');
const fs = require('fs');
const youtubedl = require('youtube-dl');
const hbjs = require('handbrake-js');
const os = require('os');
const filetype = require('file-type');

class DownloadCommand extends Command {
	constructor() {
		super('download', {
			aliases: ['download', 'dl'],
			category: 'utility',
			clientPermissions: ['SEND_MESSAGES', 'EMBED_LINKS', 'ATTACH_FILES'],
			args: [
				{
					id: 'link',
					type: 'string',
					prompt: {
						start: 'Send the link of which video you want to download',
					}
				},
				{
					id: 'caption',
					type: 'string',
					match: 'rest'
				},
				{
					id: 'spoiler',
					match: 'flag',
					flag: ['--spoil', '--spoiler', '-s']
				}
			],
			description: {
				content: 'Download videos from different website from the link you provided, use "-s" to make the vid a spoiler',
				usage: '[link] [caption]',
				examples: ['https://www.youtube.com/watch?v=6n3pFFPSlW4 Look at this funny gnome']
			}
		});
	}

	async exec(message, args) {
		if (args.caption == null) args.caption = '';
		
		let link = args.link;
		let fileName;

		if (args.spoiler) {
			fileName = `SPOILER_${message.id}_video`;
		} else {
			fileName = `${message.id}_video`;
		}


		const Embed = this.client.util.embed()
			.setColor(message.member ? message.member.displayHexColor : 'NAVY')
			.setAuthor(`Downloaded by ${message.author.username}`, message.author.displayAvatarURL(), link)
			.setDescription(args.caption)
			.setFooter(`You can get the original video by clicking on the "downloaded by ${message.author.username}" message!`);


		let compressEmbed = this.client.util.embed()
			.setColor(message.member ? message.member.displayHexColor : 'NAVY')
			.setTitle('This one will need compression!')
			.setDescription('Starting compression now!')
			.setFooter('Want it to go faster? Donate to the dev with the donate command, so i can get a better server and do it faster!');

		if (link.includes('http') || link.includes('www')) {
			let loadingmsg = await message.channel.send('Downloading <a:loadingmin:527579785212329984>');

			if (fs.existsSync(`${os.tmpdir()}/${fileName}`)) {
				fs.unlink(`${os.tmpdir()}/${fileName}`, (err) => {
					if (err);
				});
			}
			return youtubedl.exec(link, ['-o', `${os.tmpdir()}/${fileName}`], {}, async function(err) {
				if (err) {
					console.error(err);
					loadingmsg.delete();
					return message.channel.send('An error has occured, I can\'t download from the link you provided.');
				}

				let ext = 'mp4';

				if (fs.existsSync(`${os.tmpdir()}/${fileName}`)) {
					ext = await filetype.fromFile(`${os.tmpdir()}/${fileName}`);
					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()}/${fileName}`, `${os.tmpdir()}/${fileName}.${ext}`);
				} else if (fs.existsSync(`${os.tmpdir()}/${fileName}.mkv`)) { // If it can't find the video assume it got merged and end with mkv
					fs.renameSync(`${os.tmpdir()}/${fileName}.mkv`, `${os.tmpdir()}/${fileName}.mp4`); // Discord play mkv just fine but it need to end with mp4
				} else if (fs.existsSync(`${os.tmpdir()}/${fileName}.webm`)) { // Same as mkv
					ext = 'webm';
				}

				let file = fs.statSync(`${os.tmpdir()}/${fileName}.${ext}`);
				let fileSize = file.size / 1000000.0;

				//Compress vid if bigger than 8MB
				if (fileSize > 8) {
					let compressmsg = await message.channel.send(compressEmbed);
					loadingmsg.delete();

					const options = {
						input: `${os.tmpdir()}/${fileName}.${ext}`,
						output: `${os.tmpdir()}/${fileName}compressed.${ext}`,
						preset: 'General/Gmail Small 10 Minutes 288p30'
					};
	
					let handbrake = hbjs.spawn(options);
					handbrake.on('error', err => {
						console.error(err);
						compressmsg.delete();
						return message.channel.send('An error has occured while compressing the video');
					});


					let percentComplete;
					let eta;

					handbrake.on('progress', progress => {
						percentComplete = progress.percentComplete;
						eta = progress.eta;
						console.log(`Percent complete: ${progress.percentComplete}, ETA: ${progress.eta}`);
					});

					// Every 5 seconds update the compress message with the %
					let editmsg = setInterval(() => {
						compressEmbed.setDescription(`Ready in ${eta == '' ? 'soon enough' : eta}. ${percentComplete}% complete.`);
						compressmsg.edit(compressEmbed);
					}, 5000);

					handbrake.on('end', async function () {
						clearInterval(editmsg);
						file = fs.statSync(`${os.tmpdir()}/${fileName}compressed.${ext}`);
						fileSize = file.size / 1000000.0;

						compressmsg.delete();
						message.delete();

						if (fileSize > 8) {
							return message.channel.send('File too big!');
						}

						return message.channel.send({embed: Embed, files: [`${os.tmpdir()}/${fileName}compressed.${ext}`]})
							.catch(err => {
								console.error(err);
								return message.channel.send('File too big');		
							})
							.then(() => {
								// Delete file after it have been sent
								fs.unlinkSync(`${os.tmpdir()}/${fileName}.${ext}`);
							});			
					});
				} else {
					loadingmsg.delete();
					message.delete();

					message.channel.send({embed: Embed, files: [`${os.tmpdir()}/${fileName}.${ext}`]})
						.catch(err => {
							console.error(err);
							message.channel.send('File too big');	
						})
						.then(() => {
							// Delete file after it have been sent
							fs.unlinkSync(`${os.tmpdir()}/${fileName}.${ext}`);
						});
				}
			});
		}
		return message.channel.send('You need to input a valid link');
	}
}

module.exports = DownloadCommand;