app.js

const express = require('express');
const ytdl = require('ytdl-core');
const path = require('path');
const sanitize = require('sanitize-filename');
const ffmpeg = require('fluent-ffmpeg');
const ffmpegPath = require('@ffmpeg-installer/ffmpeg').path;

ffmpeg.setFfmpegPath(ffmpegPath);

const app = express();
const port = 3000;

app.set('view engine', 'ejs');
app.set('views', path.join(__dirname, 'views'));

app.use(express.static(path.join(__dirname, 'public')));

app.get('/', (req, res) => {
    res.render('index');
});

app.get('/search', async (req, res) => {
    const videoURL = req.query.url;
    if (!videoURL) {
        return res.status(400).send('URL is required');
    }

    try {
        const info = await ytdl.getInfo(videoURL);
        const formats = ytdl.filterFormats(info.formats, format => format.hasVideo);
        const thumbnail = info.videoDetails.thumbnails[info.videoDetails.thumbnails.length - 1].url;
        const title = info.videoDetails.title;
        res.render('results', { url: videoURL, formats: formats, thumbnail: thumbnail, title: title });
    } catch (error) {
        console.error('Error fetching video information:', error);
        res.status(500).send('Failed to fetch video information');
    }
});

app.get('/stream', async (req, res) => {
    const videoURL = req.query.url;
    const quality = req.query.quality;
    if (!videoURL || !quality) {
        return res.status(400).send('URL and quality are required');
    }

    try {
        const videoInfo = await ytdl.getInfo(videoURL);
        const format = ytdl.chooseFormat(videoInfo.formats, { quality: quality });

        if (!format) {
            return res.status(404).send('No such format found');
        }

        console.log(`Streaming video in format with quality: ${quality}`);

        res.header('Content-Type', 'video/mp4');

        ytdl(videoURL, {
            format: format,
            requestOptions: {
                headers: {
                    'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.124 Safari/537.36'
                }
            }
        })
        .on('error', (err) => {
            console.error('Error streaming video:', err);
            res.status(500).send('Failed to stream video');
        })
        .pipe(res);
    } catch (error) {
        console.error('Error fetching video info:', error);
        res.status(500).send('Failed to stream video');
    }
});

app.get('/download', async (req, res) => {
    const videoURL = req.query.url;
    const quality = req.query.quality;
    let title = decodeURIComponent(req.query.title || 'video');
    
    if (!videoURL || !quality) {
        console.error('Missing URL or quality');
        return res.status(400).send('URL and quality are required');
    }

    try {
        console.log(`Fetching video info for URL: ${videoURL}`);
        const videoInfo = await ytdl.getInfo(videoURL);
        console.log('Video info fetched successfully');

        console.log(`Choosing format with quality: ${quality}`);
        const format = ytdl.chooseFormat(videoInfo.formats, { quality: quality });

        if (!format) {
            console.error('No matching format found');
            return res.status(404).send('No such format found');
        }

        console.log('Format chosen successfully');

        title = sanitize(title);
        title = title.replace(/[^\x20-\x7E]/g, '');

        res.header('Content-Disposition', `attachment; filename="${title}.mp4"`);

        ytdl(videoURL, {
            format: format,
            requestOptions: {
                headers: {
                    'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.124 Safari/537.36'
                }
            }
        })
        .on('error', (err) => {
            console.error('Error downloading video:', err);
            res.status(500).send('Failed to download video');
        })
        .pipe(res);
    } catch (error) {
        console.error('Error fetching video info:', error);
        res.status(500).send('Failed to download video');
    }
});

app.listen(port, () => {
    console.log(`Server running at http://localhost:${port}`);
});