Update server.js

server.js

@@ -1,9 +1,9 @@
-// File: app.js
-
+// Import required modules
 import express from "express";
 import bodyParser from "body-parser";
 import { ytmp4 } from "ruhend-scraper";
 import yts from "youtube-yts";
+import axios from "axios"; // Add axios for making HTTP requests
 
 const app = express();
 const PORT = 7860;
@@ -22,6 +22,18 @@ const isUrl = (input) => {
   }
 };
 
+// Function to get the size of a video file from its URL
+const getVideoSize = async (url) => {
+  try {
+    const response = await axios.head(url);
+    const contentLength = response.headers["content-length"];
+    return contentLength ? parseInt(contentLength, 10) : null;
+  } catch (error) {
+    console.error("Failed to fetch video size:", error.message);
+    return null;
+  }
+};
+
 // GET endpoint for downloading YouTube videos
 app.get("/download", async (req, res) => {
   const { input } = req.query; // Extract single input parameter
@@ -48,6 +60,9 @@ app.get("/download", async (req, res) => {
       videoData = await ytmp4(firstResult.url);
     }
 
+    // Get video size
+    const videoSize = await getVideoSize(videoData.video);
+
     // Respond with video details
     const { title, video, author, description, duration, views, upload, thumbnail } = videoData;
     res.json({
@@ -59,6 +74,7 @@ app.get("/download", async (req, res) => {
       views,
       uploadDate: upload,
       thumbnail,
+      size: videoSize ? `${(videoSize / 1048576).toFixed(2)} MB` : "Unknown", // Convert bytes to MB
     });
   } catch (error) {
     console.error(error);