Update index.js

index.js

@@ -1,8 +1,8 @@
 import express from 'express';
 import multer from 'multer';
 import { spawn } from 'child_process';
-import { writeFile, unlink } from 'fs/promises';
-import { createReadStream, watch } from 'fs';
+import { writeFile, unlink, readFile } from 'fs/promises';
+import { createReadStream } from 'fs';
 import path from 'path';
 import { v4 as uuidv4 } from 'uuid';
 import fetch from 'node-fetch';
@@ -29,7 +29,7 @@ const downloadFile = async (url) => {
     return Buffer.from(arrayBuffer);
 };
 
-// --- "УМНЫЙ" ЭНДПОИНТ С НАСТОЯЩИМ LIVE-СТРИМИНГОМ ДЛЯ FFMPEG ---
+// --- "УМНЫЙ" ЭНДПОИНТ ДЛЯ ПОТОКОВОЙ ОБРАБОТКИ ---
 app.post('/api/run/stream', upload.single('file'), async (req, res) => {
     try {
         const { command, args: argsJson, file_url } = req.body;
@@ -53,61 +53,45 @@ app.post('/api/run/stream', upload.single('file'), async (req, res) => {
             return res.status(400).json({ error: 'A file must be provided via "file" or "file_url".' });
         }
         
-        // --- СПЕЦИАЛЬНАЯ ЛОГИКА ДЛЯ FFMPEG (LIVE STREAMING) ---
+        // --- СПЕЦИАЛЬНАЯ ЛОГИКА ДЛЯ FFMPEG ---
         if (command === 'ffmpeg') {
             const inputFilePath = path.join(TEMP_DIR, `${uuidv4()}-input`);
             const outputFilePath = path.join(TEMP_DIR, `${uuidv4()}-output`);
             
-            await writeFile(inputFilePath, inputBuffer);
-
-            const processedArgs = args.map(arg => 
-                arg.replace('{INPUT_FILE}', inputFilePath)
-                   .replace('{OUTPUT_FILE}', outputFilePath)
-            );
-
-            const ffmpegProcess = spawn(command, processedArgs);
-
-            // Функция для очистки
-            const cleanup = () => {
-                ffmpegProcess.kill('SIGKILL'); // Убеждаемся, что процесс завершен
-                unlink(inputFilePath).catch(() => {});
-                unlink(outputFilePath).catch(() => {});
-            };
-
-            // Как только ffmpeg создаст выходной файл, начинаем его стримить
-            const watcher = watch(TEMP_DIR, (eventType, filename) => {
-                if (filename === path.basename(outputFilePath)) {
-                    res.setHeader('Content-Type', 'application/octet-stream');
-                    const outputStream = createReadStream(outputFilePath);
-                    outputStream.pipe(res);
-                    // Прекращаем следить за файлом после того, как он найден
-                    watcher.close();
-                }
-            });
-
-            // Если пользователь закроет соединение, убиваем ffmpeg и чистим файлы
-            res.on('close', () => {
-                cleanup();
-            });
-
-            ffmpegProcess.on('close', (code) => {
-                watcher.close(); // На всякий случай
-                // Если ffmpeg завершился с ошибкой до того, как мы начали стримить
-                if (!res.headersSent) {
-                    res.status(500).json({ error: 'ffmpeg failed before output was generated.', code: code });
-                }
-                // Файлы будут удалены после завершения стрима (если он начался) или сразу
-                 setTimeout(cleanup, 100); // Небольшая задержка для завершения потока
-            });
-
-            ffmpegProcess.on('error', (err) => {
-                watcher.close();
-                if (!res.headersSent) {
-                     res.status(500).json({ error: 'Failed to start ffmpeg.', message: err.message });
-                }
-                 cleanup();
-            });
-
+            try {
+                await writeFile(inputFilePath, inputBuffer);
+
+                const processedArgs = args.map(arg => 
+                    arg.replace('{INPUT_FILE}', inputFilePath)
+                       .replace('{OUTPUT_FILE}', outputFilePath)
+                );
+
+                const process = spawn(command, processedArgs);
+                let stderrChunks = [];
+                process.stderr.on('data', (data) => stderrChunks.push(data));
+
+                process.on('close', async (code) => {
+                    if (code === 0) {
+                        try {
+                            const outputBuffer = await readFile(outputFilePath);
+                            res.setHeader('Content-Type', 'application/octet-stream');
+                            res.send(outputBuffer);
+                        } catch (readError) {
+                            res.status(500).json({ error: "Command succeeded, but failed to read output file.", message: readError.message });
+                        }
+                    } else {
+                        const stderr = Buffer.concat(stderrChunks).toString('utf8');
+                        res.status(500).json({ error: 'Command execution failed.', code: code, stderr: stderr });
+                    }
+                    // Очистка
+                    await unlink(inputFilePath).catch(()=>{});
+                    await unlink(outputFilePath).catch(()=>{});
+                });
+            } catch (execError) {
+                 res.status(500).json({ error: "Failed during ffmpeg file operations.", message: execError.message });
+                 await unlink(inputFilePath).catch(()=>{});
+                 await unlink(outputFilePath).catch(()=>{});
+            }
         // --- ОБЫЧНАЯ ЛОГИКА ДЛЯ ДРУГИХ КОМАНД ---
         } else {
             const process = spawn(command, args);
@@ -137,6 +121,7 @@ app.post('/api/run/stream', upload.single('file'), async (req, res) => {
 });
 
 
+// --- СИСТЕМА АСИНХРОННЫХ ЗАДАЧ (ОСТАЕТСЯ КАК АЛЬТЕРНАТИВА) ---
 // ... (остальной код для /api/task/* и /api/download/* без изменений) ...
 
 const executeTask = async (taskId) => {
@@ -252,4 +237,3 @@ app.get('/api/download/:fileId', (req, res) => {
 
 app.listen(PORT, () => { startCleanupJob(); });
 
-