backend/utils/ytdlp.js

const wget = require('wget-improved')
const DOMPurify = require('isomorphic-dompurify')
const metadata = require('./metadata.js')
const hr = require('@tsmx/human-readable')

const ffmpeg = require('fluent-ffmpeg')
const ffmpegStatic = require('ffmpeg-static')
const path = require('node:path')
const fs = require('node:fs')

ffmpeg.setFfmpegPath(ffmpegStatic)

async function downloadVideo(url, ws, id) {
    return new Promise(async (resolve, reject) => {
        let quality = '720'
        const video = await metadata.getVideoMetadata(id)
        if (video.error) {
            return resolve({
                message: `Failed to request Youtube with error ${video.error}. Please retry...`,
                fail: true
            })
        }
        if (video.basic_info.duration >= 900) quality = '360' // 15 minutes

        const downloadJson = await metadata.getVideoDownload(url, quality)
        if (downloadJson.status == 'error') {
            return resolve({
                message: `Failed to request Youtube with error ${downloadJson.text}. Please retry...`,
                fail: true
            })
        }

        let size = ''
        let startTime = Date.now()
        let prevBytes = 0
        let speed = 0
        const alreadyPrecentages = []

        const target = Array.isArray(downloadJson.url) ? downloadJson.url[0] : downloadJson.url
        const download = wget.download(target, `./videos/${id}.mp4`, {
            proxy: {
                protocol: 'http',
                host: 'gluetun',
                port: '8888',
                headers: {
                    'user-agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/114.0.0.0 Safari/537.36',
                    accept: '*/*',
                    origin: 'https://www.youtube.com',
                    referer: 'https://www.youtube.com',
                    DNT: '?1'
                }
            }
        })
        
        download.on('start', fileSize => {
            size = fileSize
            if (ws) ws.send(`DATA - Download has started in ${quality}p`)
        })

        download.on('progress', progress => {
            if (alreadyPrecentages.includes((progress*100).toFixed(0))) return 
            alreadyPrecentages.push((progress*100).toFixed(0))

            const currentTime = Date.now()
            const elapsedTime = (currentTime - startTime) / 1000
            const currentBytes = progress * size
            const bytesDownloaded = currentBytes - prevBytes
            speed = bytesDownloaded / elapsedTime 
            prevBytes = currentBytes

            const speedInMBps = speed / 1048576 
            const remainingBytes = size - currentBytes
            const remainingTime = remainingBytes / speed 

            if (ws) ws.send(`DATA - [download] ${(progress*100).toFixed(2)}% of ${hr.fromBytes(size)} at ${speedInMBps.toFixed(2)} MB/s ETA ${secondsToTime(remainingTime.toFixed(0))}`)
        })

        download.on('error', err => {
            if (ws) ws.send(`DATA - ${DOMPurify.sanitize(err)}`)
        })

        download.on('end', output => {
            if (output !== 'Finished writing to disk') {
                return resolve({
                    fail: true
                })
            }

            if (Array.isArray(downloadJson.url)) { // this means video is separate
                ws.send(`DATA - Downloading the video has finished, downloading audio.`);
                
                const audioDownload = wget.download(downloadJson.url[1], `./videos/${id}_audio.mp4`, {
                    proxy: {
                        protocol: 'http',
                        host: 'gluetun',
                        port: '8888',
                        headers: {
                            'user-agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/114.0.0.0 Safari/537.36',
                            accept: '*/*',
                            origin: 'https://www.youtube.com',
                            referer: 'https://www.youtube.com',
                            DNT: '?1'
                        }
                    }
                })
                audioDownload.on('end', async (audioOutput) => {
                    if (audioOutput !== 'Finished writing to disk') {
                        return resolve({ fail: true });
                    }

                    fs.renameSync(`./videos/${id}.mp4`, `./videos/${id}_video.mp4`);
                    try {
                        await mergeIt(`./videos/${id}_audio.mp4`, `./videos/${id}_video.mp4`, `./videos/${id}.mp4`);
                        ws.send(`DATA - Merging succeeded.`)

                        fs.rmSync(`./videos/${id}_audio.mp4`)
                        fs.rmSync(`./videos/${id}_video.mp4`)

                        return resolve({ fail: false });
                    } catch (error) {
                        console.error('Merging error:', error);
                        return resolve({ fail: true });
                    }
                });

                audioDownload.on('error', (err) => {
                    console.error('Audio download error:', err);
                    return resolve({ fail: true });
                });
            } else {
                ws.send(`DATA - Download has finished`)
                return resolve({
                    fail: false
                })
            }
        })
    })
}

function mergeIt(audioPath, videoPath, outputPath) {
    return new Promise((resolve, reject) => {
        ffmpeg()
            .addInput(videoPath)
            .addInput(audioPath)
            .outputOptions('-c:v copy') 
            .outputOptions('-c:a aac') 
            .output(outputPath)
            .on('end', () => {
                resolve('Merging finished!');
            })
            .on('error', (err) => {
                reject(new Error('An error occurred: ' + err.message));
            })
            .run();
    });
}

function secondsToTime(seconds) {
    const minutes = Math.floor(seconds / 60);
    const remainingSeconds = seconds % 60;
    const formattedSeconds = remainingSeconds < 10 ? '0' + remainingSeconds : remainingSeconds;
    return `${minutes}:${formattedSeconds}`;
}

module.exports = { downloadVideo }