Haha-Yes/commands/utility/download.js

206 lines
6.5 KiB
JavaScript
Raw Normal View History

import { SlashCommandBuilder, EmbedBuilder, ActionRowBuilder, SelectMenuBuilder } from 'discord.js';
2022-06-17 01:25:05 +02:00
import { exec } from 'node:child_process';
import fs from 'node:fs';
import os from 'node:os';
2022-06-17 07:31:58 +02:00
import utils from '../../utils/videos.js';
2022-06-17 01:25:05 +02:00
let client;
let cleanUp;
2022-06-17 01:25:05 +02:00
export default {
data: new SlashCommandBuilder()
.setName('download')
.setDescription('Download a video.')
.addStringOption(option =>
option.setName('url')
.setDescription('url of the video you want to download.')
.setRequired(true))
.addBooleanOption(option =>
2022-08-22 20:23:27 +02:00
option.setName('format')
.setDescription('Choose the quality of the video.')
2022-08-22 20:23:27 +02:00
.setRequired(false))
.addBooleanOption(option =>
option.setName('compress')
.setDescription('Compress the video?')
.setRequired(false)),
category: 'utility',
2022-08-30 23:29:57 +02:00
alias: ['dl'],
async execute(interaction, args, c) {
client = c;
2022-09-01 01:43:59 +02:00
const url = args.url;
const format = args.format;
interaction.doCompress = args.compress;
if (interaction.cleanUp) {
cleanUp = interaction.cleanUp;
}
await interaction.deferReply({ ephemeral: false });
if (interaction.isMessage) {
interaction.delete();
}
if (!await utils.stringIsAValidurl(url)) {
console.error(`Not a url!!! ${url}`);
return interaction.editReply({ content: '❌ This does not look like a valid url!', ephemeral: true });
2022-06-16 23:11:01 +02:00
}
if (format) {
let qualitys = await new Promise((resolve, reject) => {
2022-07-06 18:52:44 +02:00
exec(`./bin/yt-dlp "${url}" --print "%()j"`, (err, stdout, stderr) => {
if (err) {
reject(stderr);
}
if (stderr) {
console.error(stderr);
}
resolve(stdout);
});
});
qualitys = JSON.parse(qualitys);
const options = [];
qualitys.formats.forEach(f => {
options.push({
2022-07-06 18:52:44 +02:00
label: f.resolution ? f.resolution : 'Unknown format',
2022-06-16 14:28:49 +02:00
description: `${f.format} V: ${f.vcodec} A: ${f.acodec}`,
value: f.format_id,
});
});
2022-07-06 18:52:44 +02:00
if (options.length < 2) {
await interaction.deleteReply();
return interaction.followUp({ content: '❌ There is no other quality option for this video!', ephemeral: true });
}
if (options.length > 25) {
// Reverse so the higher quality formats are first
options.reverse();
while (options.length > 25) {
// Remove the lower quality formats
options.pop();
}
// Reverse again so the lower quality appears first
options.reverse();
}
const row = new ActionRowBuilder()
.addComponents(
new SelectMenuBuilder()
.setCustomId('downloadQuality')
.setPlaceholder('Nothing selected')
2022-06-16 14:36:38 +02:00
.setMinValues(1)
.setMaxValues(2)
.addOptions(options),
);
await interaction.deleteReply();
await interaction.followUp({ content: 'Which quality do you want?', ephemeral: true, components: [row] });
client.once('interactionCreate', async (interactionMenu) => {
if (!interactionMenu.isSelectMenu()) return;
if (interactionMenu.customId === 'downloadQuality') {
await interactionMenu.deferReply({ ephemeral: false });
2022-08-22 20:23:27 +02:00
download(url, interactionMenu, interaction);
}
});
return;
}
download(url, interaction);
},
};
2022-08-22 20:23:27 +02:00
async function download(url, interaction, originalInteraction) {
let format = 'bestvideo*+bestaudio/best';
const Embed = new EmbedBuilder()
.setColor(interaction.member ? interaction.member.displayHexColor : 'NAVY')
2022-06-20 17:57:10 +02:00
.setAuthor({ name: `Downloaded by ${interaction.user.tag}`, iconURL: interaction.user.displayAvatarURL(), url: url })
2022-06-20 07:57:32 +02:00
.setFooter({ text: `You can get the original video by clicking on the "Downloaded by ${interaction.user.tag}" message!` });
if (interaction.customId === 'downloadQuality') {
format = interaction.values[0];
2022-06-16 14:36:38 +02:00
if (interaction.values[1]) format += '+' + interaction.values[1];
}
2022-06-17 01:25:05 +02:00
utils.downloadVideo(url, interaction.id, format)
.then(async () => {
const file = fs.readdirSync(os.tmpdir()).filter(fn => fn.startsWith(interaction.id));
const output = `${os.tmpdir()}/${file}`;
const fileStat = fs.statSync(output);
const fileSize = fileStat.size / 1000000.0;
2022-08-22 20:23:27 +02:00
const compressInteraction = originalInteraction ? originalInteraction : interaction;
if (compressInteraction.doCompress) {
2022-08-22 20:23:27 +02:00
const presets = [ 'Discord Tiny 5 Minutes 240p30', 'Discord Small 2 Minutes 360p30', 'Discord Nitro Small 10-20 Minutes 480p30', 'Discord Nitro Medium 5-10 Minutes 720p30', 'Discord Nitro Large 3-6 Minutes 1080p30' ];
const options = [];
presets.forEach(p => {
options.push({
label: p,
value: p,
});
});
const row = new ActionRowBuilder()
2022-08-22 20:23:27 +02:00
.addComponents(
new SelectMenuBuilder()
2022-08-22 20:23:27 +02:00
.setCustomId('preset')
.setPlaceholder('Nothing selected')
.addOptions(options),
);
await interaction.deleteReply();
await interaction.followUp({ content: 'Which compression preset do you want?', ephemeral: true, components: [row] });
client.once('interactionCreate', async (interactionMenu) => {
2022-08-22 20:23:27 +02:00
if (!interactionMenu.isSelectMenu()) return;
if (interactionMenu.customId === 'preset') {
await interactionMenu.deferReply({ ephemeral: false });
compress(file, interactionMenu, Embed);
cleanUp();
2022-08-22 20:23:27 +02:00
}
});
return;
}
if (fileSize > 100) {
await interaction.deleteReply();
await interaction.followUp('Uh oh! The video you tried to download is too big!', { ephemeral: true });
}
else if (fileSize > 8) {
const fileurl = await utils.upload(output)
.catch(err => {
console.error(err);
});
await interaction.editReply({ content: 'File was bigger than 8 mb. It has been uploaded to an external site.', embeds: [Embed], ephemeral: false });
await interaction.followUp({ content: fileurl, ephemeral: false });
}
else {
await interaction.editReply({ embeds: [Embed], files: [output], ephemeral: false });
}
cleanUp();
})
.catch(async err => {
console.error(err);
await interaction.deleteReply();
await interaction.followUp({ content: 'Uh oh! An error has occured!', ephemeral: true });
});
return;
}
2022-08-22 20:23:27 +02:00
async function compress(input, interaction, embed) {
const output = `compressed${input}.mp4`;
utils.compressVideo(`${os.tmpdir()}/${input}`, output, interaction.values[0])
.then(async () => {
const fileStat = fs.statSync(`${os.tmpdir()}/${output}`);
const fileSize = fileStat.size / 1000000.0;
if (fileSize > 8) {
await interaction.editReply({ content: 'File was bigger than 8 mb. but due to the compression it is not being uploaded externally.', ephemeral: true });
}
else {
await interaction.editReply({ embeds: [embed], files: [`${os.tmpdir()}/${output}`], ephemeral: false });
}
});
}