Upload 4 files

Dockerfile

@@ -1,5 +1,4 @@
 # Используем официальный образ Node.js 18.
-# ESM-модули и fetch API стабильны в этой версии.
 FROM node:18-slim
 
 # Устанавливаем рабочую директорию внутри контейнера
@@ -9,20 +8,15 @@ WORKDIR /usr/src/app
 COPY package*.json ./
 
 # Устанавливаем зависимости проекта
-# --only=production гарантирует, что установятся только производственные зависимости
 RUN npm install --only=production
 
 # Копируем остальной код приложения в рабочую директорию
 COPY . .
 
-# ИСПРАВЛЕНИЕ: Строка ниже удалена, т.к. приложение само создает нужную директорию
-# RUN mkdir -p /tmp/huggingface-runner
-
 # Открываем порт, на котором будет работать приложение (стандартный для Spaces - 7860)
 EXPOSE 7860
 
 # Указываем команду для запуска приложения при старте контейнера
-# Используем массив для корректной обработки сигналов
 CMD [ "node", "index.js" ]
 
 

cleanup.js

@@ -1,18 +1,11 @@
 import cron from 'node-cron';
 import fs from 'fs/promises';
 import path from 'path';
-import { fileURLToPath } from 'url';
-
-// ИСПРАВЛЕНИЕ: Получаем путь к директории аналогично index.js
-const __filename = fileURLToPath(import.meta.url);
-const __dirname = path.dirname(__filename);
-const TEMP_DIR = path.join(__dirname, 'tmp');
 
+// ИСПРАВЛЕНИЕ: Используем глобальную временную директорию /tmp.
+const TEMP_DIR = '/tmp';
 const MAX_AGE_MS = 24 * 60 * 60 * 1000; // 24 часа в миллисекундах
 
-/**
- * Удаляет старые файлы из временной директории.
- */
 async function cleanupOldFiles() {
     console.log('Запуск задачи по очистке старых файлов...');
     try {
@@ -20,24 +13,28 @@ async function cleanupOldFiles() {
         const now = Date.now();
 
         for (const file of files) {
-            const filePath = path.join(TEMP_DIR, file);
-            try {
-                const stats = await fs.stat(filePath);
-                const fileAge = now - stats.mtime.getTime();
-
-                if (fileAge > MAX_AGE_MS) {
-                    await fs.unlink(filePath);
-                    console.log(`Удален старый файл: ${filePath}`);
-                }
-            } catch (statError) {
-                // Если файл был удален между readdir и stat, просто игнорируем ошибку
-                if (statError.code !== 'ENOENT') {
-                    console.error(`Не удалось получить информацию о файле ${filePath}:`, statError);
+            // Очищаем только файлы, созданные нашим приложением (у них UUID в имени)
+            // Это мера предосторожности, чтобы не удалить чужие файлы в /tmp
+            if (/\b[0-9a-f]{8}\b-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{4}-\b[0-9a-f]{12}\b/i.test(file)) {
+                 const filePath = path.join(TEMP_DIR, file);
+                try {
+                    const stats = await fs.stat(filePath);
+                    const fileAge = now - stats.mtime.getTime();
+
+                    if (fileAge > MAX_AGE_MS) {
+                        await fs.unlink(filePath);
+                        console.log(`Удален старый файл: ${filePath}`);
+                    }
+                } catch (statError) {
+                    if (statError.code !== 'ENOENT') {
+                        console.error(`Не удалось получить информацию о файле ${filePath}:`, statError);
+                    }
                 }
             }
         }
     } catch (readDirError) {
         if (readDirError.code === 'ENOENT') {
+            // Этого не должно случиться с /tmp, но на всякий случай.
             console.log('Временная директория не существует, очистка не требуется.');
         } else {
             console.error('Ошибка при чтении временной директории:', readDirError);
@@ -46,15 +43,9 @@ async function cleanupOldFiles() {
     console.log('Очистка завершена.');
 }
 
-/**
- * Запускает периодическую задачу (cron job) для очистки.
- * Задача будет выполняться каждый час.
- */
 export function startCleanupJob() {
-    // Запускаем сразу при старте, а затем каждый час
     cleanupOldFiles(); 
-    
-    cron.schedule('0 * * * *', cleanupOldFiles); // '0 * * * *' - каждую 0-ю минуту каждого часа
+    cron.schedule('0 * * * *', cleanupOldFiles);
     console.log('Задача по очистке запланирована на запуск каждый час.');
 }
 

index.js

@@ -1,50 +1,30 @@
 import express from 'express';
 import multer from 'multer';
 import { spawn } from 'child_process';
-import { writeFile, unlink, createReadStream, mkdirSync } from 'fs';
-import { fileURLToPath } from 'url';
-import path, { dirname } from 'path';
+import { writeFile, unlink, createReadStream } from 'fs';
+import path from 'path';
 import { v4 as uuidv4 } from 'uuid';
 import fetch from 'node-fetch';
 import { startCleanupJob } from './cleanup.js';
 
 // --- НАСТРОЙКА ---
-const __filename = fileURLToPath(import.meta.url);
-const __dirname = dirname(__filename);
-
-// ИСПРАВЛЕНИЕ: Создаем временную директорию внутри проекта, чтобы избежать проблем с правами.
-const TEMP_DIR = path.join(__dirname, 'tmp');
-
-// Создаем директорию синхронно при запуске, если она не существует.
-try {
-    mkdirSync(TEMP_DIR, { recursive: true });
-    console.log(`Временная директория готова для использования: ${TEMP_DIR}`);
-} catch (e) {
-    console.error(`Критическая ошибка: не удалось создать временную директорию ${TEMP_DIR}.`, e);
-    process.exit(1); // Завершаем работу, если не можем создать папку.
-}
-
+// ИСПРАВЛЕНИЕ: Используем глобальную временную директорию /tmp, которая всегда доступна для записи.
+const TEMP_DIR = '/tmp'; 
 
 const app = express();
-const PORT = process.env.PORT || 7860; // Hugging Face Spaces использует порт 7860
+const PORT = process.env.PORT || 7860;
+
+console.log(`Используется временная директория: ${TEMP_DIR}`);
 
 // --- MIDDLEWARE ---
 app.use(express.json());
 app.use(express.urlencoded({ extended: true }));
 
-// Настройка Multer для загрузки файлов в память
 const storage = multer.memoryStorage();
 const upload = multer({ storage: storage });
 
 // --- ЛОГИКА ---
 
-/**
- * Функция для выполнения внешней команды.
- * @param {string} command - Команда для выполнения (например, 'ffmpeg').
- * @param {Array<string>} args - Аргументы для команды.
- * @param {Buffer | null} inputBuffer - Входные данные для stdin.
- * @returns {Promise<{stdout: Buffer, stderr: string}>} - Промис с результатом.
- */
 const executeCommand = (command, args, inputBuffer = null) => {
     return new Promise((resolve, reject) => {
         const process = spawn(command, args);
@@ -76,12 +56,6 @@ const executeCommand = (command, args, inputBuffer = null) => {
     });
 };
 
-
-/**
- * Скачивает файл по URL.
- * @param {string} url - URL файла.
- * @returns {Promise<Buffer>} - Промис с буфером файла.
- */
 const downloadFile = async (url) => {
     const response = await fetch(url);
     if (!response.ok) {
@@ -97,22 +71,12 @@ app.get('/', (req, res) => {
     res.send('Сервер удаленного выполнения команд готов к работе!');
 });
 
-
-/**
- * МАРШРУТ 1: Потоковая обработка (без сохранения)
- * Принимает файл (загрузка или URL) и команду, возвращает результат потоком.
- * POST /api/run/stream
- * Body: { command: "ваша команда", args: ["аргумент1", "аргумент2"], file_url?: "URL файла" }
- * или multipart/form-data с полями command, args и file.
- */
 app.post('/api/run/stream', upload.single('file'), async (req, res) => {
     try {
         const { command, args: argsJson, file_url } = req.body;
         const file = req.file;
 
-        if (!command) {
-            return res.status(400).send({ error: 'Параметр "command" обязателен.' });
-        }
+        if (!command) return res.status(400).send({ error: 'Параметр "command" обязателен.' });
         
         let args;
         try {
@@ -134,7 +98,6 @@ app.post('/api/run/stream', upload.single('file'), async (req, res) => {
         
         console.log(`Stderr для ${command}: ${stderr}`);
         
-        // Отправляем результат как бинарные данные
         res.setHeader('Content-Type', 'application/octet-stream');
         res.send(stdout);
 
@@ -148,22 +111,14 @@ app.post('/api/run/stream', upload.single('file'), async (req, res) => {
     }
 });
 
-
-/**
- * МАРШРУТ 2: Обработка с сохранением файла
- * Сохраняет файл, выполняет команду, сохраняет результат и отдает ссылку.
- * POST /api/run/file
- */
 app.post('/api/run/file', upload.single('file'), async (req, res) => {
-    const tempFiles = []; // Массив для отслеживания созданных файлов
+    const tempFiles = [];
 
     try {
         const { command, args: argsJson, file_url, output_filename } = req.body;
         const file = req.file;
 
-        if (!command) {
-            return res.status(400).send({ error: 'Параметр "command" обязателен.' });
-        }
+        if (!command) return res.status(400).send({ error: 'Параметр "command" обязателен.' });
         
         let args;
         try {
@@ -180,9 +135,11 @@ app.post('/api/run/file', upload.single('file'), async (req, res) => {
         } else {
             return res.status(400).send({ error: 'Необходимо предоставить файл через "file" или "file_url".' });
         }
-
-        // Заменяем плейсхолдеры в аргументах
-        const inputFilePath = path.join(TEMP_DIR, `${uuidv4()}-${file?.originalname || 'input'}`);
+        
+        // Получаем имя файла из загрузки, или используем 'input' как запасной вариант
+        const originalName = file?.originalname?.replace(/[^a-zA-Z0-9._-]/g, '') || 'input';
+        
+        const inputFilePath = path.join(TEMP_DIR, `${uuidv4()}-${originalName}`);
         const outputFilePath = path.join(TEMP_DIR, `${uuidv4()}-${output_filename || 'output'}`);
         
         tempFiles.push(inputFilePath, outputFilePath);
@@ -192,7 +149,6 @@ app.post('/api/run/file', upload.single('file'), async (req, res) => {
                .replace('{OUTPUT_FILE}', outputFilePath)
         );
 
-        // Сохраняем входной файл
         await new Promise((resolve, reject) => {
             writeFile(inputFilePath, inputBuffer, (err) => {
                 if (err) {
@@ -203,12 +159,9 @@ app.post('/api/run/file', upload.single('file'), async (req, res) => {
             });
         });
 
-        // Выполняем команду
-        // Здесь stdin не используется, так как команда работает с файлами на диске
         const { stderr } = await executeCommand(command, processedArgs);
         console.log(`Stderr для ${command}: ${stderr}`);
         
-        // Отправляем ссылку на скачивание файла
         const fileId = path.basename(outputFilePath);
         res.status(200).json({
             message: 'Команда выполнена успешно.',
@@ -219,10 +172,12 @@ app.post('/api/run/file', upload.single('file'), async (req, res) => {
     } catch (error) {
         console.error('Ошибка в /api/run/file:', error);
         
-        // Удаляем временные файлы в случае ошибки
         for (const filePath of tempFiles) {
             unlink(filePath, (err) => {
-                if (err) console.error(`Не удалось удалить временный файл ${filePath}:`, err);
+                if (err) {
+                     // Не выводим ошибку, если файла просто нет (уже удален или не был создан)
+                    if(err.code !== 'ENOENT') console.error(`Не удалось удалить временный файл ${filePath}:`, err);
+                }
             });
         }
         
@@ -234,26 +189,20 @@ app.post('/api/run/file', upload.single('file'), async (req, res) => {
     }
 });
 
-
-/**
- * МАРШРУТ 3: Скачивание файла по ID
- * GET /api/download/:fileId
- */
 app.get('/api/download/:fileId', (req, res) => {
     const { fileId } = req.params;
-    // Важно: проверяем, что fileId не содержит ".." для предотвращения выхода из директории
     if (fileId.includes('..')) {
         return res.status(400).send('Неверный ID файла.');
     }
 
     const filePath = path.join(TEMP_DIR, fileId);
 
-    // Проверяем, существует ли файл, и отправляем его
     const stream = createReadStream(filePath);
     stream.on('error', (err) => {
         if (err.code === 'ENOENT') {
             res.status(404).send('Файл не найден или был удален.');
         } else {
+            console.error(`Ошибка чтения файла ${filePath}:`, err);
             res.status(500).send('Ошибка сервера.');
         }
     });
@@ -262,12 +211,10 @@ app.get('/api/download/:fileId', (req, res) => {
     stream.pipe(res);
 });
 
-
 // --- ЗАПУСК СЕРВЕРА И ОЧИСТКИ ---
 
 app.listen(PORT, () => {
     console.log(`Сервер запущен на порту ${PORT}`);
-    // Запускаем задачу по очистке старых файлов
     startCleanupJob();
 });