Update index.js

index.js

@@ -2,62 +2,71 @@ const express = require('express');
 const request = require('request');
 require('dotenv').config();
 
-
 const app = express();
-const PORT = 7860;
-
 const eachfilesize = {};
 
-const REQUEST_URL = process.env.REQUEST_URL;
-const TOKEN = process.env.TOKEN;
-
-async function getFileSize(link) {
-  return new Promise((resolve, reject) => {
-    request.head(link, (err, res, body) => {
-      if (err) {
-        reject(err);
-      } else {
-        const contentLength = (parseInt(res.headers['content-length']) || 0.1).toString();
-        eachfilesize[link] = contentLength;
-        resolve(contentLength);
-      }
-    });
-  });
+function getFileSize(links) {
+    let fileSize = 0;
+    for (const link of links) {
+        request.head(link, (err, res) => {
+            if (err) {
+                console.error(err);
+                return;
+            }
+            const contentLength = parseInt(res.headers['content-length'] || 0.1);
+            fileSize += contentLength;
+            eachfilesize[link] = contentLength;
+        });
+    }
+    return fileSize;
 }
 
-app.get('*', async (req, res) => {
-  if (req.method !== 'GET') {
-    return res.status(405).send('Method Not Allowed');
-  }
-
-  const filename = req.params[0];
-
-  if (filename) {
-    const originalUrl = REQUEST_URL ? `${REQUEST_URL}${filename}` : `https://huggingface.co/tranquan24/video/resolve/main/${filename}`;
-    const headers = {
-      'Authorization': `Bearer ${TOKEN}`,
-      'Range': req.headers.range || ''
-    };
-
-    try {
-      request.get({ url: originalUrl, headers: headers }, (err, response, body) => {
-        if (err) {
-          console.error(err);
-          return res.status(500).send('500');
-        }
+app.get('/', (req, res) => {
+    const shared = req.query.shared;
+    const filename = req.query.filename;
+
+    if (shared) {
+        const originalUrl = `https://huggingface.co/tranquan24/video/resolve/main/${shared}`;
+        const token = process.env.TOKEN;
+        const headers = {
+            'Authorization': `Bearer ${token}`,
+            'Range': req.headers['range'] || ''
+        };
         
-        res.set(response.headers);
-        res.status(response.statusCode).send(body);
-      });
-    } catch (err) {
-      console.error(err);
-      return res.status(500).send('500');
+        request.get({
+            url: originalUrl,
+            headers: headers
+        }).on('response', (originalResponse) => {
+            const links = [originalUrl];
+            const fileSize = getFileSize(links);
+
+            let headers_dict = originalResponse.headers;
+            headers_dict['X-Speed'] = 'Fast';
+            headers_dict['X-Bandwidth'] = 'High';
+            headers_dict['Access-Control-Allow-Origin'] = '*';
+
+            if (filename) {
+                headers_dict['Content-Disposition'] = `attachment; filename="${filename}"`;
+            } else {
+                const filenameFromUrl = shared.split('/').pop();
+                headers_dict['Content-Disposition'] = `attachment; filename="${filenameFromUrl}"`;
+            }
+
+            delete headers_dict['set-cookie'];
+
+            const { statusCode, headers } = originalResponse;
+            res.writeHead(statusCode, headers);
+            originalResponse.pipe(res);
+        }).on('error', (err) => {
+            console.error(err);
+            res.status(500).send('Internal Server Error');
+        });
+    } else {
+        res.status(400).send('Missing shared parameter');
     }
-  } else {
-    return res.status(400).send('400');
-  }
 });
 
-app.listen(PORT, '0.0.0.0', () => {
-  console.log(`Server is running on http://0.0.0.0:${PORT}`);
+const port = 7860;
+app.listen(port, '0.0.0.0', () => {
+    console.log(`Server is running on port ${port}`);
 });