Refactor la fonction createVideoWithList pour simplifier la création de vidéos et mettre à jour le statut de la vidéo à "completed"
This commit is contained in:
@@ -28,9 +28,13 @@ async function deleteVideoProject(videoId) {
|
||||
}
|
||||
|
||||
async function createVideoWithList(projectId, pathList, duration, videoId) {
|
||||
const workdir = path.join(PROJECTS_DIR, 'storage', `${projectId}`);
|
||||
const tempFile = path.join(workdir, 'temp.txt');
|
||||
//pathList étant la liste des chemins déjà triés
|
||||
const tempFile = path.join('temp.txt');
|
||||
try {
|
||||
// Trouver tous les fichiers image pour le projet donné
|
||||
const workdir = path.join(PROJECTS_DIR, 'storage', `${projectId}`);
|
||||
const dir = path.join(PROJECTS_DIR, 'storage', `${projectId}`, 'images');
|
||||
console.log('dir:', dir);
|
||||
const images = pathList;
|
||||
console.log('images:', images);
|
||||
|
||||
@@ -41,72 +45,44 @@ async function createVideoWithList(projectId, pathList, duration, videoId) {
|
||||
return numA - numB;
|
||||
});
|
||||
|
||||
// Déterminer l'id de la première et de la dernière image utilisée
|
||||
// En déduire l'id de la première et dernière image utilisée
|
||||
const firstImageId = parseInt(path.basename(sortedImages[0]).match(/\d+/)[0], 10);
|
||||
const lastImageId = parseInt(path.basename(sortedImages[sortedImages.length - 1]).match(/\d+/)[0], 10);
|
||||
|
||||
console.log('firstImageId:', firstImageId);
|
||||
console.log('lastImageId:', lastImageId);
|
||||
|
||||
// Créer le fichier temporaire pour la liste des images
|
||||
// Créer un fichier temporaire pour la liste des images
|
||||
fs.writeFileSync(tempFile, sortedImages.map(image => `file '${image}'`).join('\n'));
|
||||
|
||||
const frameRate = Math.ceil(sortedImages.length / parseInt(duration));
|
||||
|
||||
// le fichier final prend cette forme : {projectId}_{firstImageId}_{lastImageId}-{timestamp}.mp4
|
||||
const timestamp = new Date().getTime();
|
||||
const outputVideo = path.join(workdir, `${projectId}_${firstImageId}_${lastImageId}-${timestamp}.mp4`);
|
||||
|
||||
// Préparer les arguments pour ffmpeg
|
||||
const ffmpegArgs = [
|
||||
'-r', frameRate.toString(),
|
||||
'-f', 'concat',
|
||||
'-safe', '0',
|
||||
'-i', tempFile,
|
||||
'-vsync', 'vfr',
|
||||
'-pix_fmt', 'yuv420p',
|
||||
outputVideo
|
||||
];
|
||||
// Commande ffmpeg pour créer la vidéo
|
||||
const ffmpegCommand = `ffmpeg -r ${frameRate} -f concat -safe 0 -i ${tempFile} -vsync vfr -pix_fmt yuv420p ${outputVideo}`;
|
||||
console.log('Running ffmpeg command:', ffmpegCommand);
|
||||
execSync(ffmpegCommand);
|
||||
console.log('Video created successfully:', outputVideo);
|
||||
|
||||
console.log('Lancement de ffmpeg avec les arguments:', ffmpegArgs);
|
||||
|
||||
// Lancer ffmpeg sans détachement
|
||||
const child = spawn('ffmpeg', ffmpegArgs, {
|
||||
cwd: workdir,
|
||||
detached: false,
|
||||
stdio: 'pipe'
|
||||
});
|
||||
|
||||
// Afficher les logs de ffmpeg
|
||||
child.stdout.on('data', (data) => {
|
||||
console.log(`ffmpeg stdout: ${data}`);
|
||||
});
|
||||
child.stderr.on('data', (data) => {
|
||||
console.error(`ffmpeg stderr: ${data}`);
|
||||
});
|
||||
|
||||
child.on('close', (code) => {
|
||||
console.log(`ffmpeg process exited with code ${code}`);
|
||||
// Vous pouvez ici gérer la suppression du fichier temporaire
|
||||
if (fs.existsSync(tempFile)) {
|
||||
fs.unlinkSync(tempFile);
|
||||
console.log('Temporary file deleted:', tempFile);
|
||||
}
|
||||
});
|
||||
|
||||
console.log('Video creation started:', outputVideo);
|
||||
|
||||
// Mettre à jour le statut de la vidéo dans la base de données
|
||||
const query = 'UPDATE public.videos SET status = 1 WHERE id = $1';
|
||||
db.query(query, [videoId], (err, results) => {
|
||||
if (err) {
|
||||
return serverError.sendError('Error updating video status:', res, err);
|
||||
}
|
||||
console.log('Video status updated:', results.rows ? results.rows[0] : 'Aucune ligne retournée');
|
||||
});
|
||||
// Mettre à jour le statut de la vidéo à "completed"
|
||||
const updateStatusQuery = 'UPDATE public.videos SET status = $2 WHERE id = $1 RETURNING *';
|
||||
const updateStatusValues = [videoId, 1]; // 1 pour le statut "completed"
|
||||
const updateStatusRes = await db.query(updateStatusQuery, updateStatusValues);
|
||||
console.log('Video status updated to completed:', updateStatusRes.rows[0]);
|
||||
|
||||
return outputVideo;
|
||||
} catch (error) {
|
||||
console.error('Error creating video:', error);
|
||||
serverError.sendError(error);
|
||||
} finally {
|
||||
// Supprimer le fichier temporaire
|
||||
if (fs.existsSync(tempFile)) {
|
||||
fs.unlinkSync(tempFile);
|
||||
console.log('Temporary file deleted:', tempFile);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
Reference in New Issue
Block a user