Update server.js

server.js

@@ -5,8 +5,9 @@ const path = require("path");
 const https = require("https");
 const ffmpeg = require("fluent-ffmpeg");
 const ffmpegInstaller = require("@ffmpeg-installer/ffmpeg");
+const crypto = require("crypto"); // For generating random filenames
 
-// Set FFmpeg path from npm package
+// Set FFmpeg path
 ffmpeg.setFfmpegPath(ffmpegInstaller.path);
 
 const app = express();
@@ -18,26 +19,17 @@ if (!fs.existsSync(DOWNLOAD_DIR)) {
     fs.mkdirSync(DOWNLOAD_DIR, { recursive: true });
 }
 
-// Axios instance to ignore SSL certificate verification
+// Axios instance to ignore SSL 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 generate a unique filename
+const generateUniqueFilename = (extension = "") => {
+    return crypto.randomBytes(10).toString("hex") + extension;
 };
 
-// Function to delete file after a delay (default: 10 minutes)
+// Function to delete a file after a delay (default: 10 minutes)
 const scheduleFileDeletion = (filePath, delay = 600000) => {
     setTimeout(() => {
         fs.unlink(filePath, (err) => {
@@ -48,7 +40,7 @@ const scheduleFileDeletion = (filePath, delay = 600000) => {
     }, delay);
 };
 
-// Function to convert MKV to MP4 using fluent-ffmpeg
+// Function to convert MKV to MP4
 const convertToMp4 = (inputPath, outputPath) => {
     return new Promise((resolve, reject) => {
         ffmpeg(inputPath)
@@ -86,11 +78,12 @@ app.get("/download", async (req, res) => {
             }
         });
 
-        const originalFilename = getFileName(fileUrl, response.headers);
+        // Generate a unique filename
+        const isMkv = fileUrl.toLowerCase().endsWith(".mkv");
+        const originalFilename = generateUniqueFilename(isMkv ? ".mkv" : ".mp4");
         const originalFilePath = path.join(DOWNLOAD_DIR, originalFilename);
 
-        const isMkv = originalFilename.toLowerCase().endsWith(".mkv");
-        const mp4Filename = isMkv ? originalFilename.replace(/\.mkv$/, ".mp4") : originalFilename;
+        const mp4Filename = originalFilename.replace(".mkv", ".mp4");
         const mp4FilePath = path.join(DOWNLOAD_DIR, mp4Filename);
 
         const writer = fs.createWriteStream(originalFilePath);
@@ -127,7 +120,7 @@ app.get("/download", async (req, res) => {
     }
 });
 
-// Serve files from downloads directory (publicly accessible)
+// Serve files from downloads directory
 app.use("/files", express.static(DOWNLOAD_DIR));
 
 // Start the server