238 lines
7.5 KiB
JavaScript
238 lines
7.5 KiB
JavaScript
const express = require('express');
|
|
const router = express.Router();
|
|
const db = require('../db');
|
|
const fs = require('fs');
|
|
const rangeParser = require('range-parser');
|
|
const path = require('path');
|
|
const serverError = require('../utils/serverError');
|
|
const videoManager = require('../src/video/videoManager');
|
|
const measureManager = require('../src/measure/measureManager');
|
|
const dbTester = require('../test/tester');
|
|
|
|
router.get('/videos', (req, res) => {
|
|
const query = 'SELECT * FROM public.videos';
|
|
db.query(query, (err, results) => {
|
|
if (err) {
|
|
serverError.sendError('Erreur lors de la récupération des vidéos:', res, err, 500);
|
|
}
|
|
res.json(results.rows);
|
|
});
|
|
});
|
|
|
|
router.get('/videos/:id', (req, res) => {
|
|
const videoId = req.params.id;
|
|
if (!videoId || isNaN(videoId)) {
|
|
return res.status(400).json({ error: 'Invalid video ID' });
|
|
}
|
|
const query = 'SELECT * FROM public.videos WHERE id = $1';
|
|
db.query(query, [videoId], (err, results) => {
|
|
if (err) {
|
|
serverError.sendError('Erreur lors de la récupération de la vidéo:', res, err, 500);
|
|
}
|
|
res.json(results.rows);
|
|
});
|
|
});
|
|
|
|
router.post('/videos', async (req, res) => {
|
|
const { project_id, measurement_ids, name, resolution, duration } = req.body;
|
|
console.log('Creating video:', req.body);
|
|
if (!project_id || !measurement_ids || !name || !resolution || !duration) {
|
|
return res.status(400).json({ error: 'Tous les champs sont requis.' });
|
|
}
|
|
|
|
console.log('Creating video with measurements:', measurement_ids);
|
|
|
|
try {
|
|
const videoId = await videoManager.createVideoProject(project_id, measurement_ids, name, resolution, duration);
|
|
|
|
// Start rendering the video immediately after creation
|
|
const result = await db.query(
|
|
'SELECT measurement_ids, project_id, duration FROM public.videos WHERE id = $1',
|
|
[videoId]
|
|
);
|
|
|
|
if (result.rows.length === 0) {
|
|
return res.status(404).json({ error: 'Vidéo non trouvée' });
|
|
}
|
|
|
|
const { duration: videoDuration, measurement_ids: videoMeasurementIds, project_id: videoProjectId } = result.rows[0];
|
|
const pathList = await measureManager.getPathList(videoMeasurementIds, videoProjectId);
|
|
|
|
// Start background processing
|
|
videoManager.createVideoWithList(videoProjectId, pathList, videoDuration, videoId)
|
|
.then(videoFile => {
|
|
console.log('Rendu vidéo terminé:', videoFile);
|
|
return videoManager.updateVideoFile(videoId, videoFile);
|
|
})
|
|
.catch(error => {
|
|
console.error('Échec du rendu vidéo:', error);
|
|
});
|
|
|
|
// Immediate response
|
|
res.json({
|
|
message: 'Vidéo créée avec succès et le rendu a démarré',
|
|
videoId,
|
|
status: 'processing',
|
|
check_url: `/videos/status/${videoId}`
|
|
});
|
|
|
|
} catch (err) {
|
|
console.error('Erreur lors de la création de la vidéo:', err);
|
|
res.status(500).json({ error: 'Erreur lors de la création de la vidéo' });
|
|
}
|
|
});
|
|
|
|
|
|
router.delete('/videos/:id', (req, res) => {
|
|
const videoId = req.params.id;
|
|
if (!videoId || isNaN(videoId)) {
|
|
return res.status(400).json({ error: 'Invalid video ID' });
|
|
}
|
|
|
|
const query = 'SELECT video_file FROM public.videos WHERE id = $1';
|
|
db.query(query, [videoId], (err, results) => {
|
|
if (err) {
|
|
return serverError.sendError('Error getting video:', res, err, 500);
|
|
}
|
|
if (results.rows.length === 0) {
|
|
return res.status(404).json({ error: 'Video not found' });
|
|
}
|
|
|
|
const videoFile = results.rows[0].video_file;
|
|
fs.unlink(videoFile, (err) => {
|
|
if (err) {
|
|
console.error('Error deleting video file:', err);
|
|
return res.status(500).json({ error: 'Error deleting video file' });
|
|
}
|
|
|
|
videoManager.deleteVideoProject(videoId).then(() => {
|
|
res.json({ message: 'Vidéo supprimée avec succès' });
|
|
}).catch(err => {
|
|
console.error('Erreur lors de la suppression de la vidéo:', err);
|
|
res.status(500).json({ error: 'Erreur lors de la suppression de la vidéo' });
|
|
});
|
|
});
|
|
});
|
|
});
|
|
|
|
router.get('/videos/file/:video_id', (req, res) => {
|
|
const videoId = req.params.video_id;
|
|
const query = 'SELECT video_file, status FROM public.videos WHERE id = $1';
|
|
|
|
db.query(query, [videoId], (err, results) => {
|
|
if (err) {
|
|
console.error('Error getting video:', err);
|
|
return serveFallbackVideo(res);
|
|
}
|
|
if (results.rows.length === 0) {
|
|
console.error('Video not found');
|
|
return serveFallbackVideo(res);
|
|
}
|
|
|
|
const video = results.rows[0];
|
|
if (video.status === 0) {
|
|
return res.status(400).json({ error: 'Video not yet produced' });
|
|
}
|
|
|
|
let videoPath = video.video_file;
|
|
|
|
if (!videoPath) {
|
|
console.error('Video file path is null or undefined');
|
|
videoPath = dbTester.getCatVideo();
|
|
}
|
|
|
|
// Check if the video file exists
|
|
fs.access(videoPath, fs.constants.F_OK, (err) => {
|
|
if (err) {
|
|
console.error('Video file not found:', err);
|
|
videoPath = dbTester.getCatVideo();
|
|
}
|
|
|
|
const stat = fs.statSync(videoPath);
|
|
const fileSize = stat.size;
|
|
const range = req.headers.range;
|
|
|
|
if (range) {
|
|
const parts = rangeParser(fileSize, range);
|
|
const start = parts[0].start;
|
|
const end = parts[0].end;
|
|
const chunksize = (end - start) + 1;
|
|
const file = fs.createReadStream(videoPath, { start, end });
|
|
const head = {
|
|
'Content-Range': `bytes ${start}-${end}/${fileSize}`,
|
|
'Accept-Ranges': 'bytes',
|
|
'Content-Length': chunksize,
|
|
'Content-Type': 'video/mp4',
|
|
};
|
|
res.writeHead(206, head);
|
|
file.pipe(res);
|
|
} else {
|
|
const head = {
|
|
'Content-Length': fileSize,
|
|
'Content-Type': 'video/mp4',
|
|
};
|
|
res.writeHead(200, head);
|
|
fs.createReadStream(videoPath).pipe(res);
|
|
}
|
|
});
|
|
});
|
|
});
|
|
|
|
function serveFallbackVideo() {
|
|
return dbTester.getCatVideo();
|
|
}
|
|
|
|
router.get('/videos/progress/:video_id', async (req, res) => {
|
|
try {
|
|
const result = await db.query(`
|
|
SELECT
|
|
progress,
|
|
EXTRACT(EPOCH FROM (NOW() - started_at)) as elapsed,
|
|
eta,
|
|
status
|
|
FROM public.videos
|
|
WHERE id = $1
|
|
`, [req.params.video_id]);
|
|
|
|
if (result.rows.length === 0) {
|
|
return res.status(404).json({ error: 'Vidéo non trouvée' });
|
|
}
|
|
|
|
const video = result.rows[0];
|
|
res.json({
|
|
progress: video.progress,
|
|
elapsed: video.elapsed,
|
|
eta: video.eta,
|
|
status: this.getStatusLabel(video.status)
|
|
});
|
|
|
|
} catch (error) {
|
|
console.error(error);
|
|
res.status(500).json({ error: 'Erreur de récupération' });
|
|
}
|
|
});
|
|
|
|
function getStatusLabel(status) {
|
|
const statusMap = {
|
|
0: 'En attente',
|
|
1: 'Terminé',
|
|
2: 'Échec',
|
|
3: 'En cours'
|
|
};
|
|
return statusMap[status] || 'Inconnu';
|
|
}
|
|
|
|
|
|
router.get('/cat', (_, res) => {
|
|
const videoPath = dbTester.getCatVideo();
|
|
fs.access(videoPath, fs.constants.F_OK, (err) => {
|
|
if (err) {
|
|
console.error('Video not found:', err);
|
|
return res.status(404).json({ error: 'Video not found' });
|
|
}
|
|
res.download(videoPath);
|
|
});
|
|
});
|
|
|
|
module.exports = router;
|