stream / index.js
tranquan24's picture
Update index.js
7766270 verified
raw
history blame
2.91 kB
const express = require('express');
const request = require('request');
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(err);
return 0;
}
}
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 });
let headers_dict = {};
for (const [key, value] of Object.entries(originalResponse.headers)) {
headers_dict[key] = value;
}
headers_dict['X-Speed'] = 'Fast';
headers_dict['X-Bandwidth'] = 'High';
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: 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', '*');
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}`);
});