Update server.js

server.js

@@ -3,7 +3,7 @@ const axios = require("axios");
 const fs = require("fs");
 const path = require("path");
 const https = require("https");
-const crypto = require("crypto"); // For generating random filenames
+const crypto = require("crypto");
 
 const app = express();
 const PORT = 7860;
@@ -17,7 +17,7 @@ if (!fs.existsSync(DOWNLOAD_DIR)) {
 // Axios instance to ignore SSL verification
 const axiosInstance = axios.create({
     httpsAgent: new https.Agent({ rejectUnauthorized: false }),
-    timeout: 300000 // 5 minutes timeout
+    timeout: 300000, // 5 minutes timeout
 });
 
 // Function to generate a unique request ID
@@ -44,6 +44,56 @@ const scheduleFileDeletion = (filePath, delay = 600000) => {
     }, delay);
 };
 
+// **Download Function with Integrity Check**
+const downloadFile = async (fileUrl, filePath, requestId) => {
+    let retries = 3; // Retry up to 3 times
+
+    while (retries > 0) {
+        try {
+            console.log(`⬇️ [${requestId}] Downloading: ${fileUrl}`);
+
+            const response = await axiosInstance({
+                url: fileUrl,
+                method: "GET",
+                responseType: "stream",
+                headers: { "User-Agent": "Mozilla/5.0", "Accept": "*/*" }
+            });
+
+            const totalSize = response.headers["content-length"]; // Get file size
+            console.log(`📏 [${requestId}] Expected size: ${totalSize} bytes`);
+
+            const writer = fs.createWriteStream(filePath);
+            response.data.pipe(writer);
+
+            await new Promise((resolve, reject) => {
+                writer.on("finish", resolve);
+                writer.on("error", reject);
+            });
+
+            // Verify file size
+            const fileSize = fs.statSync(filePath).size;
+            if (totalSize && fileSize < totalSize * 0.9) { // Allow minor variance
+                console.error(`⚠️ [${requestId}] Download incomplete: ${fileSize} bytes (expected ${totalSize} bytes)`);
+                fs.unlinkSync(filePath);
+                throw new Error("Incomplete download, retrying...");
+            }
+
+            console.log(`✅ [${requestId}] Download complete: ${filePath}`);
+            return true;
+
+        } catch (error) {
+            console.error(`❌ [${requestId}] Download failed: ${error.message}`);
+            retries--;
+
+            if (retries === 0) {
+                console.error(`⛔ [${requestId}] All retries failed, skipping.`);
+            }
+        }
+    }
+
+    return false;
+};
+
 // **API Route - Responds Immediately, Processes in Background**
 app.get("/download", async (req, res) => {
     const fileUrl = req.query.url;
@@ -70,31 +120,9 @@ app.get("/download", async (req, res) => {
 
     // **Background Processing (Download)**
     (async () => {
-        try {
-            console.log(`⬇️ [${requestId}] Downloading: ${fileUrl}`);
-
-            const response = await axiosInstance({
-                url: fileUrl,
-                method: "GET",
-                responseType: "stream",
-                headers: { "User-Agent": "Mozilla/5.0", "Accept": "*/*" }
-            });
-
-            const writer = fs.createWriteStream(filePath);
-            response.data.pipe(writer);
-
-            await new Promise((resolve, reject) => {
-                writer.on("finish", resolve);
-                writer.on("error", reject);
-            });
-
-            console.log(`✅ [${requestId}] Download complete: ${filePath}`);
-
-            // Schedule deletion
+        const success = await downloadFile(fileUrl, filePath, requestId);
+        if (success) {
             scheduleFileDeletion(filePath);
-
-        } catch (error) {
-            console.error(`❌ [${requestId}] Processing error:`, error.message);
         }
     })();
 });