Upload 3 files

package.json

@@ -7,11 +7,11 @@
     "start": "node server.js"
   },
   "dependencies": {
-    "express": "^4.18.2",
-    "torrent-stream": "^1.2.0",
     "cors": "^2.8.5",
-    "pump": "^3.0.0",
+    "express": "^4.18.2",
     "mime-types": "^2.1.35",
-    "webtorrent": "^2.1.32"
+    "pump": "^3.0.0",
+    "torrent-stream": "^1.2.1",
+    "webtorrent": "^2.5.0"
   }
 }

server.js

@@ -1,5 +1,5 @@
 const express = require('express');
-const torrentStream = require('torrent-stream');
+const WebTorrent = require('webtorrent');
 const cors = require('cors');
 const pump = require('pump');
 const mime = require('mime-types');
@@ -11,11 +11,21 @@ const app = express();
 app.use(cors());
 
 // В Hugging Face Spaces для хранения кэша можно использовать /tmp
-const tmpDir = '/tmp/torrent-stream';
+const tmpDir = '/tmp/webtorrent';
 if (!fs.existsSync(tmpDir)){
     fs.mkdirSync(tmpDir, { recursive: true });
 }
 
+// Создаем ОДИН инстанс клиента Webtorrent на весь сервер.
+// Это критически важно: Webtorrent делит порты DHT и соединения между торрентами,
+// что позволяет находить пиров гораздо быстрее, чем каждый раз создавать новый движок.
+const client = new WebTorrent({
+    maxConns: 500, // Максимум соединений для клиента
+    tracker: true,
+    dht: true,
+    webSeeds: true // Поддержка web-пиров (ускоряет скачивание)
+});
+
 // Кэш активных торрентов
 const activeTorrents = new Map();
 const initializingTorrents = new Map();
@@ -69,15 +79,14 @@ function getTorrent(magnet) {
             }
         }
 
-        // Добавляем популярные трекеры для ускорения поиска пиров, если их нет
+        // WebTorrent сам умеет парсить трекеры из magnet-ссылки,
+        // но мы добавим несколько надежных WebSocket-трекеров для WebRTC-пиров
+        let opts = { path: tmpDir };
+        
         if (typeof torrentInput === 'string' && torrentInput.startsWith('magnet:')) {
             const trackers = [
                 'udp://tracker.opentrackr.org:1337/announce',
-                'udp://open.demonii.com:1337/announce',
                 'udp://tracker.openbittorrent.com:80',
-                'udp://tracker.coppersurfer.tk:6969',
-                'udp://tracker.leechers-paradise.org:6969',
-                'udp://exodus.desync.com:6969',
                 'wss://tracker.btorrent.xyz',
                 'wss://tracker.openwebtorrent.com'
             ];
@@ -89,12 +98,33 @@ function getTorrent(magnet) {
             });
         }
 
-        let engine;
         try {
-            engine = torrentStream(torrentInput, {
-                connections: 400, // Увеличиваем лимит пиров для большей скорости
-                uploads: 2,       // Ограничиваем отдачу, чтобы не забивать исходящий канал сервера
-                path: tmpDir
+            client.add(torrentInput, opts, (torrent) => {
+                clearTimeout(timeout);
+                console.log('Torrent ready:', torrent.infoHash);
+                
+                // Таймер неактивности
+                torrent.lastActivity = Date.now();
+                torrent.activeStreams = 0;
+                
+                activeTorrents.set(magnet, torrent);
+                resolve(torrent);
+                
+                const checkActivity = setInterval(() => {
+                    if (torrent.activeStreams === 0 && (Date.now() - torrent.lastActivity > 5 * 60 * 1000)) {
+                        if (activeTorrents.has(magnet)) {
+                            torrent.destroy();
+                            activeTorrents.delete(magnet);
+                            console.log('Torrent destroyed due to inactivity (no active streams for 5m):', torrent.infoHash);
+                            clearInterval(checkActivity);
+                        }
+                    }
+                }, 60 * 1000);
+                
+                torrent.on('error', (err) => {
+                    console.error('Torrent engine error:', err);
+                    clearInterval(checkActivity);
+                });
             });
         } catch (err) {
             console.error('Failed to parse torrent identifier:', err.message);
@@ -107,44 +137,16 @@ function getTorrent(magnet) {
 
         // Устанавливаем таймаут на получение метаданных (60 секунд)
         const timeout = setTimeout(() => {
-            if (engine) {
-                engine.destroy();
+            // Пытаемся найти и удалить зависший торрент
+            const hashMatch = typeof magnet === 'string' ? magnet.match(/urn:btih:([a-zA-Z0-9]+)/i) : null;
+            if (hashMatch) {
+                const existing = client.get(hashMatch[1]);
+                if (existing) existing.destroy();
             }
-            activeTorrents.delete(magnet); // Очищаем кэш при таймауте
+            
+            activeTorrents.delete(magnet);
             reject(new Error('Timeout: failed to fetch torrent metadata within 60s. No peers found. (Убедитесь, что торрент живой и имеет раздающих)'));
         }, 60000);
-
-        if (engine) {
-            engine.on('ready', () => {
-                clearTimeout(timeout);
-                console.log('Torrent ready:', engine.infoHash);
-                activeTorrents.set(magnet, engine);
-                resolve(engine);
-                
-                // Таймер неактивности (уничтожаем если нет активных стримов более 5 минут)
-                engine.lastActivity = Date.now();
-                engine.activeStreams = 0;
-                
-                const checkActivity = setInterval(() => {
-                    if (engine.activeStreams === 0 && (Date.now() - engine.lastActivity > 5 * 60 * 1000)) {
-                        if (activeTorrents.has(magnet)) {
-                            engine.destroy();
-                            activeTorrents.delete(magnet);
-                            console.log('Torrent destroyed due to inactivity (no active streams for 5m):', engine.infoHash);
-                            clearInterval(checkActivity);
-                        }
-                    }
-                }, 60 * 1000);
-                
-                engine.on('destroyed', () => clearInterval(checkActivity));
-            });
-
-            engine.on('error', (err) => {
-                clearTimeout(timeout);
-                console.error('Torrent engine error:', err);
-                reject(err);
-            });
-        }
     });
 
     initializingTorrents.set(magnet, initPromise);
@@ -260,12 +262,12 @@ app.listen(PORT, () => {
 
 // Мониторинг скорости в логах Hugging Face
 setInterval(() => {
-    activeTorrents.forEach((engine, magnet) => {
-        if (engine.activeStreams > 0 && engine.swarm) {
-            const speed = (engine.swarm.downloadSpeed() / 1024 / 1024).toFixed(2);
-            const peers = engine.swarm.wires ? engine.swarm.wires.length : 0;
+    activeTorrents.forEach((torrent, magnet) => {
+        if (torrent.activeStreams > 0) {
+            const speed = (torrent.downloadSpeed / 1024 / 1024).toFixed(2);
+            const peers = torrent.numPeers;
             if (speed > 0 || peers > 0) {
-                console.log(`[Stream Active] Hash: ${engine.infoHash} | Speed: ${speed} MB/s | Peers: ${peers}`);
+                console.log(`[Stream Active] Hash: ${torrent.infoHash} | Speed: ${speed} MB/s | Peers: ${peers}`);
             }
         }
     });