Playlist support

This commit is contained in:
Wizzard 2024-09-08 12:35:31 -04:00
parent f15a527de1
commit 56178c08cd
1 changed files with 80 additions and 96 deletions

View File

@ -3,7 +3,7 @@ const { v4: uuidv4 } = require('uuid');
const path = require('path'); const path = require('path');
const { EmbedBuilder } = require('discord.js'); const { EmbedBuilder } = require('discord.js');
const fs = require('fs'); const fs = require('fs');
const { exec, spawn, execSync } = require('child_process'); const { exec, execSync } = require('child_process');
const MAX_RETRIES = 3; const MAX_RETRIES = 3;
const RETRY_DELAY = 1000; const RETRY_DELAY = 1000;
@ -45,31 +45,27 @@ module.exports = {
const searchQuery = args.join(' '); const searchQuery = args.join(' ');
const voiceChannel = message.member.voice.channel; const voiceChannel = message.member.voice.channel;
console.log(`Received command: play ${searchQuery}`);
console.log(`Voice channel: ${voiceChannel ? voiceChannel.name : 'None'}`);
if (!voiceChannel) { if (!voiceChannel) {
console.error('User is not in a voice channel.');
return message.reply('You need to be in a voice channel to play music!'); return message.reply('You need to be in a voice channel to play music!');
} }
let title, tempFilePath, videoUrl = null, thumbnailUrl = null; const isPlaylist = searchQuery.includes("list=");
let loadingMessage; let loadingMessage;
try { const getDuration = (filePath) => {
const getDuration = (filePath) => { try {
try { const output = execSync(`ffprobe -v error -show_entries format=duration -of default=noprint_wrappers=1:nokey=1 "${filePath}"`).toString().trim();
const output = execSync(`ffprobe -v error -show_entries format=duration -of default=noprint_wrappers=1:nokey=1 "${filePath}"`).toString().trim(); const durationSeconds = parseFloat(output);
const durationSeconds = parseFloat(output); const minutes = Math.floor(durationSeconds / 60);
const minutes = Math.floor(durationSeconds / 60); const seconds = Math.floor(durationSeconds % 60);
const seconds = Math.floor(durationSeconds % 60); return `${minutes}:${seconds.toString().padStart(2, '0')}`;
return `${minutes}:${seconds.toString().padStart(2, '0')}`; } catch (error) {
} catch (error) { console.error('Error getting track duration:', error);
console.error('Error getting track duration:', error); return 'Unknown';
return 'Unknown'; }
} };
};
try {
if (message.attachments.size > 0) { if (message.attachments.size > 0) {
const attachment = message.attachments.first(); const attachment = message.attachments.first();
const attachmentName = attachment.name.toLowerCase(); const attachmentName = attachment.name.toLowerCase();
@ -78,15 +74,10 @@ module.exports = {
title = attachment.name; title = attachment.name;
tempFilePath = path.join(__dirname, '../utils/tmp', attachment.name); tempFilePath = path.join(__dirname, '../utils/tmp', attachment.name);
console.log(`Attachment received: ${title}`);
console.log(`Downloading attachment to: ${tempFilePath}`);
const response = await fetch(attachment.url); const response = await fetch(attachment.url);
const buffer = await response.buffer(); const buffer = await response.buffer();
fs.writeFileSync(tempFilePath, buffer); fs.writeFileSync(tempFilePath, buffer);
console.log(`Downloaded and saved: ${tempFilePath}`);
const duration = getDuration(tempFilePath); const duration = getDuration(tempFilePath);
const embed = new EmbedBuilder() const embed = new EmbedBuilder()
@ -98,7 +89,6 @@ module.exports = {
message.channel.send({ embeds: [embed] }); message.channel.send({ embeds: [embed] });
console.log('Adding to queue and attempting to play.');
addToQueue(message.guild.id, tempFilePath, title, voiceChannel, null, message.author.username, message.author.displayAvatarURL(), null); addToQueue(message.guild.id, tempFilePath, title, voiceChannel, null, message.author.username, message.author.displayAvatarURL(), null);
playNextInQueue(message.guild.id); playNextInQueue(message.guild.id);
return; return;
@ -107,9 +97,6 @@ module.exports = {
const convertedFileName = `${uuidv4()}.mp3`; const convertedFileName = `${uuidv4()}.mp3`;
tempFilePath = path.join(__dirname, '../utils/tmp', convertedFileName); tempFilePath = path.join(__dirname, '../utils/tmp', convertedFileName);
console.log(`Attachment received: ${title}`);
console.log(`Converting and downloading attachment to: ${tempFilePath}`);
const response = await fetch(attachment.url); const response = await fetch(attachment.url);
const buffer = await response.buffer(); const buffer = await response.buffer();
const tempVideoPath = path.join(__dirname, '../utils/tmp', attachment.name); const tempVideoPath = path.join(__dirname, '../utils/tmp', attachment.name);
@ -117,12 +104,10 @@ module.exports = {
spawnFFmpegProcess(['-i', tempVideoPath, '-f', 'mp3', '-ab', '192000', '-vn', tempFilePath], (err) => { spawnFFmpegProcess(['-i', tempVideoPath, '-f', 'mp3', '-ab', '192000', '-vn', tempFilePath], (err) => {
if (err) { if (err) {
console.error('Error converting file:', err);
message.reply('Failed to convert the video file.'); message.reply('Failed to convert the video file.');
return; return;
} }
console.log(`Converted and saved: ${tempFilePath}`);
fs.unlinkSync(tempVideoPath); fs.unlinkSync(tempVideoPath);
const duration = getDuration(tempFilePath); const duration = getDuration(tempFilePath);
@ -136,14 +121,12 @@ module.exports = {
message.channel.send({ embeds: [embed] }); message.channel.send({ embeds: [embed] });
console.log('Adding to queue and attempting to play.');
addToQueue(message.guild.id, tempFilePath, title, voiceChannel, null, message.author.username, message.author.displayAvatarURL(), null); addToQueue(message.guild.id, tempFilePath, title, voiceChannel, null, message.author.username, message.author.displayAvatarURL(), null);
playNextInQueue(message.guild.id); playNextInQueue(message.guild.id);
}); });
return; return;
} else { } else {
console.error('Attachment is not a supported media file.');
return message.reply('Only MP3, MP4, WEBM, and MOV files are supported for uploads.'); return message.reply('Only MP3, MP4, WEBM, and MOV files are supported for uploads.');
} }
} }
@ -153,16 +136,11 @@ module.exports = {
title = path.basename(searchQuery.split('?')[0]); title = path.basename(searchQuery.split('?')[0]);
tempFilePath = path.join(__dirname, '../utils/tmp', `${uuidv4()}_${title}`); tempFilePath = path.join(__dirname, '../utils/tmp', `${uuidv4()}_${title}`);
console.log(`MP3 link received: ${searchQuery}`);
console.log(`Downloading MP3 to: ${tempFilePath}`);
const response = await fetch(searchQuery); const response = await fetch(searchQuery);
if (!response.ok) throw new Error('Failed to download MP3 file.'); if (!response.ok) throw new Error('Failed to download MP3 file.');
const buffer = await response.buffer(); const buffer = await response.buffer();
fs.writeFileSync(tempFilePath, buffer); fs.writeFileSync(tempFilePath, buffer);
console.log(`Downloaded and saved: ${tempFilePath}`);
const duration = getDuration(tempFilePath); const duration = getDuration(tempFilePath);
const embed = new EmbedBuilder() const embed = new EmbedBuilder()
@ -174,7 +152,6 @@ module.exports = {
message.channel.send({ embeds: [embed] }); message.channel.send({ embeds: [embed] });
console.log('Adding to queue and attempting to play.');
addToQueue(message.guild.id, tempFilePath, title, voiceChannel, null, message.author.username, message.author.displayAvatarURL(), null); addToQueue(message.guild.id, tempFilePath, title, voiceChannel, null, message.author.username, message.author.displayAvatarURL(), null);
playNextInQueue(message.guild.id); playNextInQueue(message.guild.id);
return; return;
@ -185,9 +162,6 @@ module.exports = {
const convertedFileName = `${uuidv4()}.${fileExtension}`; const convertedFileName = `${uuidv4()}.${fileExtension}`;
tempFilePath = path.join(__dirname, '../utils/tmp', convertedFileName); tempFilePath = path.join(__dirname, '../utils/tmp', convertedFileName);
console.log(`Discord media link received: ${searchQuery}`);
console.log(`Downloading media from Discord to: ${tempFilePath}`);
const response = await fetch(searchQuery); const response = await fetch(searchQuery);
if (!response.ok) throw new Error('Failed to download media file from Discord.'); if (!response.ok) throw new Error('Failed to download media file from Discord.');
const buffer = await response.buffer(); const buffer = await response.buffer();
@ -198,12 +172,10 @@ module.exports = {
spawnFFmpegProcess(['-i', tempVideoPath, '-f', 'mp3', '-ab', '192000', '-vn', tempFilePath], (err) => { spawnFFmpegProcess(['-i', tempVideoPath, '-f', 'mp3', '-ab', '192000', '-vn', tempFilePath], (err) => {
if (err) { if (err) {
console.error('Error converting file:', err);
message.reply('Failed to convert the video file.'); message.reply('Failed to convert the video file.');
return; return;
} }
console.log(`Converted and saved: ${tempFilePath}`);
fs.unlinkSync(tempVideoPath); fs.unlinkSync(tempVideoPath);
const duration = getDuration(tempFilePath); const duration = getDuration(tempFilePath);
@ -217,14 +189,12 @@ module.exports = {
message.channel.send({ embeds: [embed] }); message.channel.send({ embeds: [embed] });
console.log('Adding to queue and attempting to play.');
addToQueue(message.guild.id, tempFilePath, title, voiceChannel, null, message.author.username, message.author.displayAvatarURL(), null); addToQueue(message.guild.id, tempFilePath, title, voiceChannel, null, message.author.username, message.author.displayAvatarURL(), null);
playNextInQueue(message.guild.id); playNextInQueue(message.guild.id);
}); });
} else { } else {
fs.writeFileSync(tempFilePath, buffer); fs.writeFileSync(tempFilePath, buffer);
console.log(`Downloaded and saved: ${tempFilePath}`);
const duration = getDuration(tempFilePath); const duration = getDuration(tempFilePath);
@ -237,18 +207,34 @@ module.exports = {
message.channel.send({ embeds: [embed] }); message.channel.send({ embeds: [embed] });
console.log('Adding to queue and attempting to play.');
addToQueue(message.guild.id, tempFilePath, title, voiceChannel, null, message.author.username, message.author.displayAvatarURL(), null); addToQueue(message.guild.id, tempFilePath, title, voiceChannel, null, message.author.username, message.author.displayAvatarURL(), null);
playNextInQueue(message.guild.id); playNextInQueue(message.guild.id);
} }
return; return;
} else { } else if (isPlaylist) {
console.log(`YouTube link received: ${searchQuery}`); loadingMessage = await message.channel.send(`Loading playlist...`);
videoUrl = searchQuery;
exec(`yt-dlp --cookies ${path.join(__dirname, '../cookies.txt')} --print-json --skip-download ${searchQuery}`, async (error, stdout, stderr) => { exec(`yt-dlp --cookies ${path.join(__dirname, '../cookies.txt')} --flat-playlist --print-json "${searchQuery}"`, async (error, stdout) => {
if (error) {
message.reply('Failed to retrieve playlist info.');
return;
}
const playlist = stdout.split('\n').filter(Boolean).map(line => JSON.parse(line));
if (loadingMessage) await loadingMessage.delete();
message.channel.send(`Adding ${playlist.length} songs to the queue...`);
for (const video of playlist) {
const videoUrl = `https://www.youtube.com/watch?v=${video.id}`;
await addVideoToQueue(videoUrl, message, voiceChannel, false);
}
playNextInQueue(message.guild.id);
});
} else {
exec(`yt-dlp --cookies ${path.join(__dirname, '../cookies.txt')} --print-json --skip-download "${searchQuery}"`, async (error, stdout) => {
if (error) { if (error) {
console.error(`Error getting video info: ${error}`);
message.reply('Failed to retrieve video info.'); message.reply('Failed to retrieve video info.');
return; return;
} }
@ -264,15 +250,12 @@ module.exports = {
tempFilePath = path.join(__dirname, '../utils/tmp', `${uuidv4()}.mp3`); tempFilePath = path.join(__dirname, '../utils/tmp', `${uuidv4()}.mp3`);
exec(`yt-dlp --cookies ${path.join(__dirname, '../cookies.txt')} --format bestaudio --output "${tempFilePath}" ${searchQuery}`, async (error, stdout, stderr) => { exec(`yt-dlp --cookies ${path.join(__dirname, '../cookies.txt')} --format bestaudio --output "${tempFilePath}" "${searchQuery}"`, async (error, stdout, stderr) => {
if (error) { if (error) {
console.error(`Error downloading file: ${error}`);
message.reply('Failed to download audio file.'); message.reply('Failed to download audio file.');
return; return;
} }
console.log(`Downloaded and saved: ${tempFilePath}`);
const duration = getDuration(tempFilePath); const duration = getDuration(tempFilePath);
if (loadingMessage) { if (loadingMessage) {
@ -289,60 +272,24 @@ module.exports = {
message.channel.send({ embeds: [embed] }); message.channel.send({ embeds: [embed] });
console.log('Adding to queue and attempting to play.'); addToQueue(message.guild.id, tempFilePath, title, voiceChannel, searchQuery, message.author.username, message.author.displayAvatarURL(), thumbnailUrl);
addToQueue(message.guild.id, tempFilePath, title, voiceChannel, videoUrl, message.author.username, message.author.displayAvatarURL(), thumbnailUrl);
playNextInQueue(message.guild.id); playNextInQueue(message.guild.id);
}); });
}); });
} }
} else { } else {
loadingMessage = await message.channel.send(`Searching for: **${searchQuery}**...`); loadingMessage = await message.channel.send(`Searching for: **${searchQuery}**...`);
exec(`yt-dlp --cookies ${path.join(__dirname, '../cookies.txt')} --dump-single-json "ytsearch:${searchQuery}"`, async (error, stdout) => {
console.log(`Performing YouTube search: ${searchQuery}`);
exec(`yt-dlp --cookies ${path.join(__dirname, '../cookies.txt')} --dump-single-json "ytsearch:${searchQuery}"`, (error, stdout, stderr) => {
if (error) { if (error) {
console.error(`Error searching: ${error}`);
message.reply('Failed to search for video.'); message.reply('Failed to search for video.');
return; return;
} }
const info = JSON.parse(stdout); const info = JSON.parse(stdout);
const url = info.entries[0].webpage_url; const videoUrl = info.entries[0].webpage_url;
title = info.entries[0].title; await addVideoToQueue(videoUrl, message, voiceChannel, true);
thumbnailUrl = info.entries[0].thumbnail;
tempFilePath = path.join(__dirname, '../utils/tmp', `${uuidv4()}.mp3`); if (loadingMessage) await loadingMessage.delete();
console.log(`Downloading file to: ${tempFilePath}`);
exec(`yt-dlp --cookies ${path.join(__dirname, '../cookies.txt')} --format bestaudio --output "${tempFilePath}" ${url}`, async (error, stdout, stderr) => {
if (error) {
console.error(`Error downloading file: ${error}`);
message.reply('Failed to download audio file.');
return;
}
console.log(`Downloaded and saved: ${tempFilePath}`);
const duration = getDuration(tempFilePath);
if (loadingMessage) {
await loadingMessage.delete();
}
const embed = new EmbedBuilder()
.setColor('#0099ff')
.setTitle('Added To Queue')
.setDescription(`**${title}** (${duration})`)
.setThumbnail(thumbnailUrl)
.setFooter({ text: `Requested by ${message.author.username}`, iconURL: message.author.displayAvatarURL() })
.setTimestamp();
message.channel.send({ embeds: [embed] });
console.log('Adding to queue and attempting to play.');
addToQueue(message.guild.id, tempFilePath, title, voiceChannel, url, message.author.username, message.author.displayAvatarURL(), thumbnailUrl);
playNextInQueue(message.guild.id);
});
}); });
} }
@ -354,6 +301,43 @@ module.exports = {
}, },
}; };
async function addVideoToQueue(videoUrl, message, voiceChannel, sendEmbed = true) {
const tempFilePath = path.join(__dirname, '../utils/tmp', `${uuidv4()}.mp3`);
exec(`yt-dlp --cookies ${path.join(__dirname, '../cookies.txt')} --print-json --skip-download "${videoUrl}"`, async (error, stdout) => {
if (error) {
message.reply('Failed to retrieve video info.');
return;
}
const info = JSON.parse(stdout);
const title = info.title;
const thumbnailUrl = info.thumbnail;
exec(`yt-dlp --cookies ${path.join(__dirname, '../cookies.txt')} --format bestaudio --output "${tempFilePath}" "${videoUrl}"`, async (error) => {
if (error) {
message.reply('Failed to download audio file.');
return;
}
if (sendEmbed) {
const embed = new EmbedBuilder()
.setColor('#0099ff')
.setTitle('Added To Queue')
.setDescription(`**${title}**`)
.setThumbnail(thumbnailUrl)
.setFooter({ text: `Requested by ${message.author.username}`, iconURL: message.author.displayAvatarURL() })
.setTimestamp();
message.channel.send({ embeds: [embed] });
}
addToQueue(message.guild.id, tempFilePath, title, voiceChannel, videoUrl, message.author.username, message.author.displayAvatarURL(), thumbnailUrl);
playNextInQueue(message.guild.id);
});
});
}
function isValidURL(string) { function isValidURL(string) {
try { try {
new URL(string); new URL(string);