Update index.js

index.js

@@ -5,68 +5,82 @@ require('dotenv').config();
 const app = express();
 const eachfilesize = {};
 
-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;
+async function getFileSize(link) {
+  try {
+    const res = await request.head(link);
+    const contentLength = parseInt(res.headers['content-length'] || 0.1);
+    eachfilesize[link] = contentLength;
+    return contentLength;
+  } catch (err) {
+    console.error(err);
+    return 0;
+  }
 }
 
-app.get('/', (req, res) => {
-    const shared = req.query.shared;
-    const filename = req.query.filename;
+app.get('/', async (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'] || ''
+    };
+    
+    try {
+      const originalResponse = await request.get({ url: originalUrl, headers });
 
-    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'] || ''
-        };
-        
-        request.get({
-            url: originalUrl,
-            headers: headers
-        }).on('response', (originalResponse) => {
-            const links = [originalUrl];
-            const fileSize = getFileSize(links);
+      let headers_dict = {};
+      for (const [key, value] of Object.entries(originalResponse.headers)) {
+        headers_dict[key] = value;
+      }
 
-            let headers_dict = originalResponse.headers;
-            headers_dict['X-Speed'] = 'Fast';
-            headers_dict['X-Bandwidth'] = 'High';
-            headers_dict['Access-Control-Allow-Origin'] = '*';
+      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}"`;
-            }
+      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 newResponse = new Response(originalResponse.body, { headers: headers_dict });
+      newResponse.headers.delete('Date');
+      newResponse.headers.delete('Etag');
+      newResponse.headers.delete('Last-Modified');
+      newResponse.headers.delete('Cf-Cache-Status');
+      newResponse.headers.delete('Content-Disposition');
+      newResponse.headers.delete('Nel');
+      newResponse.headers.delete('Cf-Ray');
+      newResponse.headers.delete('Report-To');
+      newResponse.headers.delete('X-Amz-Cf-Id');
+      newResponse.headers.delete('X-Amz-Cf-Pop');
+      newResponse.headers.delete('X-Amz-Server-Side-Encryption');
+      newResponse.headers.delete('X-Amz-Storage-Class');
+      newResponse.headers.delete('X-Cache');
+      newResponse.headers.set('Cache-Control', 'private, max-age=21188');
+      newResponse.headers.set('Connection', 'close');
+      newResponse.headers.set('Server', 'gvs 1.0');
+      newResponse.headers.set('Vary', 'Origin');
+      newResponse.headers.set('X-Content-Type-Options', 'nosniff');
+      newResponse.headers.set('Access-Control-Allow-Origin', 'https://tpcloud.rf.gd');
 
-            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');
+      res.writeHead(newResponse.status, newResponse.headers);
+      newResponse.body.pipe(res);
+    } catch (err) {
+      console.error(err);
+      res.status(500).send('Internal Server Error');
     }
+  } else {
+    res.status(400).send('Missing shared parameter');
+  }
 });
 
 const port = 7860;
 app.listen(port, '0.0.0.0', () => {
-    console.log(`Server is running on port ${port}`);
+  console.log(`Server is running on port ${port}`);
 });