Update server.js

server.js

@@ -2,7 +2,7 @@ const express = require("express");
 const axios = require("axios");
 const fs = require("fs");
 const path = require("path");
-const progress = require("progress-stream");
+const https = require("https");
 
 const app = express();
 const PORT = 7860;
@@ -10,18 +10,38 @@ const DOWNLOAD_DIR = path.join(__dirname, "downloads");
 
 // Ensure the downloads directory exists
 if (!fs.existsSync(DOWNLOAD_DIR)) {
-    fs.mkdirSync(DOWNLOAD_DIR);
+    fs.mkdirSync(DOWNLOAD_DIR, { recursive: true });
 }
 
-// Function to format bytes into human-readable size
-function formatBytes(bytes, decimals = 2) {
-    if (bytes === 0) return "0 Bytes";
-    const k = 1024;
-    const dm = decimals < 0 ? 0 : decimals;
-    const sizes = ["Bytes", "KB", "MB", "GB", "TB"];
-    const i = Math.floor(Math.log(bytes) / Math.log(k));
-    return parseFloat((bytes / Math.pow(k, i)).toFixed(dm)) + " " + sizes[i];
-}
+// Axios instance to ignore SSL certificate verification
+const axiosInstance = axios.create({
+    httpsAgent: new https.Agent({ rejectUnauthorized: false })
+});
+
+// Function to extract filename from URL or headers
+const getFileName = (fileUrl, headers) => {
+    let filename = path.basename(new URL(fileUrl).pathname);
+
+    if (headers["content-disposition"]) {
+        const match = headers["content-disposition"].match(/filename="(.+)"/);
+        if (match) {
+            filename = match[1];
+        }
+    }
+    
+    return filename.replace(/[<>:"/\\|?*]+/g, ""); // Remove invalid characters
+};
+
+// Function to delete file after a delay
+const scheduleFileDeletion = (filePath, delay = 600000) => { // 10 minutes
+    setTimeout(() => {
+        fs.unlink(filePath, (err) => {
+            if (!err) {
+                console.log(`Deleted file: ${filePath}`);
+            }
+        });
+    }, delay);
+};
 
 // API Route to download and serve .mkv files
 app.get("/download", async (req, res) => {
@@ -31,55 +51,31 @@ app.get("/download", async (req, res) => {
     }
 
     try {
-        // Get filename from URL
-        const filename = path.basename(new URL(fileUrl).pathname);
-        const filePath = path.join(DOWNLOAD_DIR, filename);
+        console.log(`Downloading: ${fileUrl}`);
 
-        // Start the download request
-        const response = await axios({
+        // Request file with forced download handling
+        const response = await axiosInstance({
             url: fileUrl,
             method: "GET",
-            responseType: "stream"
-        });
-
-        // Get file size from headers
-        const totalSize = parseInt(response.headers["content-length"], 10);
-        console.log(`Downloading: ${filename} (${formatBytes(totalSize)})`);
-
-        // Progress tracking
-        const progressStream = progress({
-            length: totalSize,
-            time: 1000 // Update every second
+            responseType: "stream",
+            headers: {
+                "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0.0.0 Safari/537.36",
+                "Accept": "*/*"
+            }
         });
 
-        progressStream.on("progress", (progress) => {
-            const speed = formatBytes(progress.speed) + "/s";
-            const downloaded = formatBytes(progress.transferred);
-            const percent = progress.percentage.toFixed(2);
-            const timeLeft = progress.eta ? progress.eta.toFixed(2) + "s" : "Calculating...";
-            console.log(`Progress: ${percent}% | Speed: ${speed} | Downloaded: ${downloaded} | Time Left: ${timeLeft}`);
-        });
+        // Extract correct filename
+        const filename = getFileName(fileUrl, response.headers);
+        const filePath = path.join(DOWNLOAD_DIR, filename);
 
-        // Pipe the download stream through progress tracking and into a file
+        // Pipe response stream to file
         const writer = fs.createWriteStream(filePath);
-        response.data.pipe(progressStream).pipe(writer);
+        response.data.pipe(writer);
 
         writer.on("finish", () => {
-            const fileUrl = `http://localhost:${PORT}/files/${filename}`;
-            console.log(`Download complete: ${fileUrl}`);
-
-            // Schedule file deletion after 10 minutes (600,000 ms)
-            setTimeout(() => {
-                fs.unlink(filePath, (err) => {
-                    if (err) {
-                        console.error(`Error deleting file ${filename}:`, err);
-                    } else {
-                        console.log(`File ${filename} deleted after 10 minutes.`);
-                    }
-                });
-            }, 600000); // 10 minutes
-
-            return res.json({ message: "Download complete", fileUrl });
+            const servedUrl = `http://localhost:${PORT}/files/${filename}`;
+            scheduleFileDeletion(filePath); // Auto-delete after 10 minutes
+            return res.json({ message: "Download complete", fileUrl: servedUrl });
         });
 
         writer.on("error", (err) => {
@@ -88,7 +84,7 @@ app.get("/download", async (req, res) => {
         });
 
     } catch (error) {
-        console.error("Download error:", error);
+        console.error("Download error:", error.message);
         return res.status(500).json({ error: "Failed to download file" });
     }
 });