diff --git a/index.js b/index.js index 305e81a..1955c01 100644 --- a/index.js +++ b/index.js @@ -44,56 +44,70 @@ app.get('/search', async (req, res) => { } }); +function getRandomHex() { + // Generate a random integer between 100 and 99999 + const randomNumber = Math.floor(Math.random() * (99999 - 100 + 1)) + 100; + + // Convert the number to a hexadecimal string and return it + return randomNumber.toString(16); +} + // Video download endpoint app.get('/download', (req, res) => { const videoUrl = req.query.url; if (!videoUrl) { + console.error('Error: Query parameter "url" is required.'); return res.status(400).json({ error: 'Query parameter "url" is required.' }); } // Set the output file path - var outputPath = path.join('/tmp/downloads', '%(title)s.%(ext)s'); + var outputPath = path.join('/tmp/downloads', `${getRandomHex}.mp4`); // Create /tmp/downloads directory if it doesn't exist - fs.mkdirSync(path.join('/tmp/downloads'), { recursive: true }); + try { + fs.mkdirSync(path.join('/tmp/downloads'), { recursive: true }); + } catch (err) { + console.error('Error creating directory:', err); + return res.status(500).json({ error: 'Failed to create download directory.', details: err.message }); + } // Use yt-dlp to download the video exec(`yt-dlp -o "${outputPath}" "${videoUrl}"`, (error, stdout, stderr) => { if (error) { + console.error('Error executing yt-dlp:', stderr); return res.status(500).json({ error: 'An error occurred while downloading the video.', details: stderr }); } - // Parse the output to get the file name - const filename = stdout.split('\n').find(line => line.includes('Destination'))?.split(' ')[1]; + console.log('yt-dlp output:', stdout); - if (!filename) { - return res.status(500).json({ error: 'Downloaded file not found.' }); + const downloadedFilePath = outputPath; + + console.log('Downloaded file path:', downloadedFilePath); + + if (!downloadedFilePath || !fs.existsSync(downloadedFilePath)) { + console.error('Error: Downloaded file not found or invalid path.'); + return res.status(500).json({ error: 'Downloaded file not found or invalid path.' }); } - // Full path of the downloaded file - const downloadedFilePath = outputPath + // Send the file + res.download(downloadedFilePath, (err) => { + if (err) { + console.error('Error sending the file:', err); + return res.status(500).json({ error: 'Failed to send the file.', details: err.message }); + } - // Check if the file exists and send it - if (fs.existsSync(downloadedFilePath)) { - res.download(downloadedFilePath, (err) => { + // Optionally, delete the file after sending it + fs.unlink(downloadedFilePath, (err) => { if (err) { - return res.status(500).json({ error: 'Failed to send the file.', details: err.message }); + console.error('Failed to delete the file:', err); } - - // Optionally, delete the file after sending it - fs.unlink(downloadedFilePath, (err) => { - if (err) { - console.error('Failed to delete the file:', err); - } - }); }); - } else { - res.status(404).json({ error: 'Downloaded file not found.' }); - } + }); }); }); + app.listen(port, () => { console.log(`Server running at http://localhost:${port}`); });