152 lines
4.3 KiB
JavaScript
152 lines
4.3 KiB
JavaScript
|
const { SlashCommandBuilder } = require('@discordjs/builders');
|
||
|
const { MessageEmbed, MessageActionRow, MessageSelectMenu } = require('discord.js');
|
||
|
const { exec } = require('node:child_process');
|
||
|
const fs = require('node:fs');
|
||
|
const os = require('node:os');
|
||
|
|
||
|
module.exports = {
|
||
|
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 =>
|
||
|
option.setName('advanced')
|
||
|
.setDescription('Choose the quality of the video.')
|
||
|
.setRequired(false)),
|
||
|
|
||
|
async execute(interaction) {
|
||
|
await interaction.deferReply({ ephemeral: false });
|
||
|
const url = interaction.options.getString('url');
|
||
|
|
||
|
if (interaction.options.getBoolean('advanced')) {
|
||
|
let qualitys = await new Promise((resolve, reject) => {
|
||
|
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({
|
||
|
label: f.resolution,
|
||
|
description: f.format,
|
||
|
value: f.format_id,
|
||
|
});
|
||
|
});
|
||
|
|
||
|
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 MessageActionRow()
|
||
|
.addComponents(
|
||
|
new MessageSelectMenu()
|
||
|
.setCustomId('downloadQuality')
|
||
|
.setPlaceholder('Nothing selected')
|
||
|
.addOptions(options),
|
||
|
);
|
||
|
|
||
|
await interaction.deleteReply();
|
||
|
await interaction.followUp({ content: 'Which quality do you want?', ephemeral: true, components: [row] });
|
||
|
|
||
|
interaction.client.on('interactionCreate', async (interactionMenu) => {
|
||
|
if (!interactionMenu.isSelectMenu()) return;
|
||
|
if (interactionMenu.customId === 'downloadQuality') {
|
||
|
await interactionMenu.deferReply({ ephemeral: false });
|
||
|
download(url, interactionMenu);
|
||
|
}
|
||
|
});
|
||
|
return;
|
||
|
}
|
||
|
download(url, interaction);
|
||
|
},
|
||
|
};
|
||
|
|
||
|
async function download(url, interaction) {
|
||
|
let format = 'bestvideo*+bestaudio/best';
|
||
|
const Embed = new MessageEmbed()
|
||
|
.setColor(interaction.member ? interaction.member.displayHexColor : 'NAVY')
|
||
|
.setAuthor(`Downloaded by ${interaction.member.displayName}`, interaction.member.displayAvatarURL(), url)
|
||
|
.setDescription(url);
|
||
|
|
||
|
if (interaction.customId === 'downloadQuality') {
|
||
|
format = interaction.values[0];
|
||
|
}
|
||
|
|
||
|
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;
|
||
|
|
||
|
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 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 });
|
||
|
}
|
||
|
})
|
||
|
.catch(async err => {
|
||
|
console.error(err);
|
||
|
await interaction.deleteReply();
|
||
|
await interaction.followUp({ content: 'Uh oh! An error has occured!', ephemeral: true });
|
||
|
});
|
||
|
}
|
||
|
|
||
|
async function downloadVideo(url, output, format) {
|
||
|
await new Promise((resolve, reject) => {
|
||
|
exec(`./bin/yt-dlp -f ${format} ${url} -o "${os.tmpdir()}/${output}.%(ext)s" --force-overwrites`, (err, stdout, stderr) => {
|
||
|
if (err) {
|
||
|
reject(stderr);
|
||
|
}
|
||
|
if (stderr) {
|
||
|
console.error(stderr);
|
||
|
}
|
||
|
resolve(stdout);
|
||
|
});
|
||
|
});
|
||
|
}
|
||
|
|
||
|
async function upload(file) {
|
||
|
return await new Promise((resolve, reject) => {
|
||
|
exec(`./bin/upload.sh ${file}`, (err, stdout, stderr) => {
|
||
|
if (err) {
|
||
|
reject(stderr);
|
||
|
}
|
||
|
if (stderr) {
|
||
|
console.error(stderr);
|
||
|
}
|
||
|
resolve(stdout);
|
||
|
});
|
||
|
});
|
||
|
}
|