Update index.js

index.js

@@ -5,66 +5,68 @@ require('dotenv').config();
 const app = express();
 const eachfilesize = {};
 
-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('Error retrieving file size:', err);
-    return 0;
-  }
+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) => {
-  const shared = req.query.shared;
-  const filename = req.query.filename;
+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'] || ''
+        };
+        
+        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'] = 'https://embed.giize.com';
 
-  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 (filename) {
+                headers_dict['Content-Disposition'] = `attachment; filename="${filename}"`;
+            } else {
+                const filenameFromUrl = shared.split('/').pop();
+                headers_dict['Content-Disposition'] = `attachment; filename="${filenameFromUrl}"`;
+            }
 
-      const newResponse = new Response(originalResponse.body, { headers: originalResponse.headers });
-      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');
+            delete headers_dict['set-cookie'];
 
-      res.writeHead(newResponse.status, newResponse.headers);
-      newResponse.body.pipe(res);
-    } catch (err) {
-      console.error('Error processing request:', err);
-      res.status(500).send(`Internal Server Error: ${err.message}`);
+            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 {
-    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}`);
 });