mas fixes

This commit is contained in:
The Ghost of FOSS' Past 2024-10-28 11:32:05 -05:00
parent 54ca4ea3d9
commit 7871a67084

View file

@ -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}`);
});