Haha-Yes/commands/utility/download.js

133 lines
4.3 KiB
JavaScript

import { SlashCommandBuilder } from '@discordjs/builders';
import { MessageEmbed, MessageActionRow, MessageSelectMenu } from 'discord.js';
import { exec } from 'node:child_process';
import fs from 'node:fs';
import os from 'node:os';
import utils from '../../utils/videos.js';
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 =>
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');
const urlRE = new RegExp('([a-zA-Z0-9]+://)?([a-zA-Z0-9_]+:[a-zA-Z0-9_]+@)?([a-zA-Z0-9.-]+\\.[A-Za-z]{2,4})(:[0-9]+)?([^ ])+');
if (!url.match(urlRE)) {
return interaction.editReply({ content: '❌ This does not look like a valid URL!', ephemeral: true });
}
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} V: ${f.vcodec} A: ${f.acodec}`,
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')
.setMinValues(1)
.setMaxValues(2)
.addOptions(options),
);
await interaction.deleteReply();
await interaction.followUp({ content: 'Which quality do you want?', ephemeral: true, components: [row] });
interaction.client.once('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)
.setFooter(`You can get the original video by clicking on the "Downloaded by ${interaction.member.displayName}" message!`);
if (interaction.customId === 'downloadQuality') {
format = interaction.values[0];
if (interaction.values[1]) format += '+' + interaction.values[1];
}
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;
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 });
}
})
.catch(async err => {
console.error(err);
await interaction.deleteReply();
await interaction.followUp({ content: 'Uh oh! An error has occured!', ephemeral: true });
});
return;
}