DZ-Musicbot/commands/play.js

const { addToQueue, playNextInQueue } = require('../utils/queueManager');
const { v4: uuidv4 } = require('uuid');
const path = require('path');
const { EmbedBuilder } = require('discord.js');
const fs = require('fs');
const { exec, execSync } = require('child_process');

const MAX_RETRIES = 3;
const RETRY_DELAY = 1000;

async function spawnFFmpegProcess(args, callback, retries = 0) {
    const ffmpegProcess = spawn('ffmpeg', args);

    ffmpegProcess.on('close', (code) => {
        if (code === 0) {
            callback(null);
        } else if (retries < MAX_RETRIES) {
            console.warn(`FFmpeg process failed, retrying (${retries + 1}/${MAX_RETRIES})...`);
            setTimeout(() => {
                spawnFFmpegProcess(args, callback, retries + 1);
            }, RETRY_DELAY);
        } else {
            callback(new Error(`FFmpeg process failed after ${MAX_RETRIES} retries.`));
        }
    });

    ffmpegProcess.on('error', (err) => {
        if (retries < MAX_RETRIES) {
            console.warn(`FFmpeg process error, retrying (${retries + 1}/${MAX_RETRIES})...`, err);
            setTimeout(() => {
                spawnFFmpegProcess(args, callback, retries + 1);
            }, RETRY_DELAY);
        } else {
            callback(err);
        }
    });
}

module.exports = {
    name: 'play',
    description: 'Play a song from YouTube, a URL, or an uploaded media file',
    aliases: ['p'],
    async execute(message, args) {
        const fetch = await import('node-fetch').then(module => module.default);
        const limit = (await import('p-limit')).default(3);
        const searchQuery = args.join(' ');
        const voiceChannel = message.member.voice.channel;

        if (!voiceChannel) {
            return message.reply('You need to be in a voice channel to play music!');
        }

        const isPlaylist = searchQuery.includes("list=");
        let loadingMessage;

        const getDuration = (filePath) => {
            try {
                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 minutes = Math.floor(durationSeconds / 60);
                const seconds = Math.floor(durationSeconds % 60);
                return `${minutes}:${seconds.toString().padStart(2, '0')}`;
            } catch (error) {
                console.error('Error getting track duration:', error);
                return 'Unknown';
            }
        };

        try {
            if (message.attachments.size > 0) {
                const attachment = message.attachments.first();
                const attachmentName = attachment.name.toLowerCase();

                if (attachmentName.endsWith('.mp3')) {
                    const title = attachment.name;
                    const tempFilePath = path.join(__dirname, '../utils/tmp', attachment.name);

                    const response = await fetch(attachment.url);
                    const buffer = await response.buffer();
                    fs.writeFileSync(tempFilePath, buffer);

                    const duration = getDuration(tempFilePath);

                    const embed = new EmbedBuilder()
                        .setColor('#0099ff')
                        .setTitle('Added To Queue')
                        .setDescription(`**${title}** (${duration})`)
                        .setFooter({ text: `Requested by ${message.author.username}`, iconURL: message.author.displayAvatarURL() })
                        .setTimestamp();

                    message.channel.send({ embeds: [embed] });

                    addToQueue(message.guild.id, tempFilePath, title, voiceChannel, null, message.author.username, message.author.displayAvatarURL(), null);
                    playNextInQueue(message.guild.id);
                    return;
                } else if (attachmentName.endsWith('.mp4') || attachmentName.endsWith('.webm') || attachmentName.endsWith('.mov')) {
                    const title = attachment.name;
                    const convertedFileName = `${uuidv4()}.mp3`;
                    const tempFilePath = path.join(__dirname, '../utils/tmp', convertedFileName);

                    const response = await fetch(attachment.url);
                    const buffer = await response.buffer();
                    const tempVideoPath = path.join(__dirname, '../utils/tmp', attachment.name);
                    fs.writeFileSync(tempVideoPath, buffer);

                    spawnFFmpegProcess(['-i', tempVideoPath, '-f', 'mp3', '-ab', '192000', '-vn', tempFilePath], (err) => {
                        if (err) {
                            message.reply('Failed to convert the video file.');
                            return;
                        }

                        fs.unlinkSync(tempVideoPath);

                        const duration = getDuration(tempFilePath);

                        const embed = new EmbedBuilder()
                            .setColor('#0099ff')
                            .setTitle('Added To Queue')
                            .setDescription(`**${title}** (${duration})`)
                            .setFooter({ text: `Requested by ${message.author.username}`, iconURL: message.author.displayAvatarURL() })
                            .setTimestamp();

                        message.channel.send({ embeds: [embed] });

                        addToQueue(message.guild.id, tempFilePath, title, voiceChannel, null, message.author.username, message.author.displayAvatarURL(), null);
                        playNextInQueue(message.guild.id);
                    });

                    return;
                } else {
                    return message.reply('Only MP3, MP4, WEBM, and MOV files are supported for uploads.');
                }
            }

            if (isValidURL(searchQuery)) {
                if (searchQuery.endsWith('.mp3')) {
                    const title = path.basename(searchQuery.split('?')[0]);
                    const tempFilePath = path.join(__dirname, '../utils/tmp', `${uuidv4()}_${title}`);

                    const response = await fetch(searchQuery);
                    if (!response.ok) throw new Error('Failed to download MP3 file.');
                    const buffer = await response.buffer();
                    fs.writeFileSync(tempFilePath, buffer);

                    const duration = getDuration(tempFilePath);

                    const embed = new EmbedBuilder()
                        .setColor('#0099ff')
                        .setTitle('Added To Queue')
                        .setDescription(`**${title}** (${duration})`)
                        .setFooter({ text: `Requested by ${message.author.username}`, iconURL: message.author.displayAvatarURL() })
                        .setTimestamp();

                    message.channel.send({ embeds: [embed] });

                    addToQueue(message.guild.id, tempFilePath, title, voiceChannel, null, message.author.username, message.author.displayAvatarURL(), null);
                    playNextInQueue(message.guild.id);
                    return;
                } else if (searchQuery.includes("cdn.discordapp.com")) {
                    const title = path.basename(searchQuery.split('?')[0]);
                    const isVideo = searchQuery.endsWith('.mp4') || searchQuery.endsWith('.webm') || searchQuery.endsWith('.mov');
                    const fileExtension = isVideo ? 'mp3' : 'original';
                    const convertedFileName = `${uuidv4()}.${fileExtension}`;
                    const tempFilePath = path.join(__dirname, '../utils/tmp', convertedFileName);

                    const response = await fetch(searchQuery);
                    if (!response.ok) throw new Error('Failed to download media file from Discord.');
                    const buffer = await response.buffer();

                    if (isVideo) {
                        const tempVideoPath = path.join(__dirname, '../utils/tmp', path.basename(searchQuery.split('?')[0]));
                        fs.writeFileSync(tempVideoPath, buffer);

                        spawnFFmpegProcess(['-i', tempVideoPath, '-f', 'mp3', '-ab', '192000', '-vn', tempFilePath], (err) => {
                            if (err) {
                                message.reply('Failed to convert the video file.');
                                return;
                            }

                            fs.unlinkSync(tempVideoPath);

                            const duration = getDuration(tempFilePath);

                            const embed = new EmbedBuilder()
                                .setColor('#0099ff')
                                .setTitle('Added To Queue')
                                .setDescription(`**${title}** (${duration})`)
                                .setFooter({ text: `Requested by ${message.author.username}`, iconURL: message.author.displayAvatarURL() })
                                .setTimestamp();

                            message.channel.send({ embeds: [embed] });

                            addToQueue(message.guild.id, tempFilePath, title, voiceChannel, null, message.author.username, message.author.displayAvatarURL(), null);
                            playNextInQueue(message.guild.id);
                        });

                    } else {
                        fs.writeFileSync(tempFilePath, buffer);

                        const duration = getDuration(tempFilePath);

                        const embed = new EmbedBuilder()
                            .setColor('#0099ff')
                            .setTitle('Added To Queue')
                            .setDescription(`**${title}** (${duration})`)
                            .setFooter({ text: `Requested by ${message.author.username}`, iconURL: message.author.displayAvatarURL() })
                            .setTimestamp();

                        message.channel.send({ embeds: [embed] });

                        addToQueue(message.guild.id, tempFilePath, title, voiceChannel, null, message.author.username, message.author.displayAvatarURL(), null);
                        playNextInQueue(message.guild.id);
                    }
                    return;
                } else if (isPlaylist) {
                    loadingMessage = await message.channel.send(`Loading playlist...`);

                    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...`);

                        // Process the playlist with concurrency control using p-limit
                        const addToQueuePromises = playlist.map((video) => {
                            const videoUrl = `https://www.youtube.com/watch?v=${video.id}`;
                            return limit(() => addVideoToQueue(videoUrl, message, voiceChannel, false));
                        });

                        await Promise.all(addToQueuePromises);  // Wait for all promises to complete with concurrency limit
                        playNextInQueue(message.guild.id);
                    });
                } else {
                    exec(`yt-dlp --cookies ${path.join(__dirname, '../cookies.txt')} --print-json --skip-download "${searchQuery}"`, async (error, stdout) => {
                        if (error) {
                            message.reply('Failed to retrieve video info.');
                            return;
                        }

                        const info = JSON.parse(stdout);
                        const title = info.title || "Unknown Title";
                        const thumbnailUrl = info.thumbnail || null;

                        loadingMessage = await message.channel.send(`**Loading...** ${title}`);

                        const tempFilePath = path.join(__dirname, '../utils/tmp', `${uuidv4()}.mp3`);

                        exec(`yt-dlp --cookies ${path.join(__dirname, '../cookies.txt')} --format bestaudio --output "${tempFilePath}" "${searchQuery}"`, async (error) => {
                            if (error) {
                                message.reply('Failed to download audio file.');
                                return;
                            }

                            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] });

                            addToQueue(message.guild.id, tempFilePath, title, voiceChannel, searchQuery, message.author.username, message.author.displayAvatarURL(), thumbnailUrl);
                            playNextInQueue(message.guild.id);
                        });
                    });
                }
            } else {
                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) => {
                    if (error) {
                        message.reply('Failed to search for video.');
                        return;
                    }

                    const info = JSON.parse(stdout);
                    const videoUrl = info.entries[0].webpage_url;
                    await addVideoToQueue(videoUrl, message, voiceChannel, true);

                    if (loadingMessage) await loadingMessage.delete();
                });
            }

        } catch (error) {
            console.error('Error:', error);
            if (loadingMessage) await loadingMessage.delete();
            message.reply('An error occurred while trying to play the music.');
        }
    },
};

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) {
    try {
        new URL(string);
        return true;
    } catch (_) {
        return false;
    }
}