Update index.js

index.js

@@ -1,72 +1,121 @@
 import express from 'express';
-import fetch from 'node-fetch';
+import fetch, { Headers as FetchHeaders } from 'node-fetch';
 
 const app = express();
-// Увеличим лимит размера тела запроса, если необходимо (по умолчанию 100kb)
-// express.raw({type: '*/*', limit: '10mb'})
 app.use(express.raw({type: '*/*'}));
 
-// Hello World на главной
+app.options('*', (req, res) => {
+  const origin = req.headers.origin;
+
+  if (origin) {
+    res.setHeader('Access-Control-Allow-Origin', origin);
+    res.setHeader('Access-Control-Allow-Credentials', 'true');
+  } else {
+    res.setHeader('Access-Control-Allow-Origin', '*');
+  }
+
+  res.setHeader('Access-Control-Allow-Methods', 'GET, POST, PUT, DELETE, PATCH, OPTIONS, HEAD');
+  
+  const requestedHeaders = req.headers['access-control-request-headers'];
+  if (requestedHeaders) {
+    res.setHeader('Access-Control-Allow-Headers', requestedHeaders);
+  } else {
+    res.setHeader('Access-Control-Allow-Headers', 'Content-Type, Authorization, X-Requested-With, X-CSRF-Token, Accept, Origin');
+  }
+  
+  res.setHeader('Access-Control-Max-Age', '86400');
+
+  if (origin) {
+    res.setHeader('Vary', 'Origin');
+  }
+
+  res.status(204).end();
+});
+
+
 app.get('/', (req, res) => {
+  const origin = req.headers.origin;
+  if (origin) {
+    res.setHeader('Access-Control-Allow-Origin', origin);
+    res.setHeader('Vary', 'Origin');
+    if (req.method === 'OPTIONS') {
+        res.setHeader('Access-Control-Allow-Methods', 'GET, OPTIONS');
+        res.setHeader('Access-Control-Allow-Headers', 'Content-Type');
+        res.status(204).end();
+        return;
+    }
+  } else {
+    res.setHeader('Access-Control-Allow-Origin', '*');
+  }
   res.send('Hello World');
 });
 
-// Прокси для всех остальных путей
 app.all('*', async (req, res) => {
   if (req.url === '/') {
-    return; // Уже обработано app.get('/')
+    return;
   }
 
+  const clientRequestOrigin = req.headers.origin;
+
   try {
     const targetUrlString = req.url.substring(1);
 
     if (!targetUrlString) {
+      addCorsHeaders(res, clientRequestOrigin);
       res.status(400).send('Target URL is missing in the path.');
       return;
     }
 
-    // Проверка на валидность URL перед использованием
     let targetUrl;
     try {
         targetUrl = new URL(targetUrlString);
     } catch (e) {
+        addCorsHeaders(res, clientRequestOrigin);
         res.status(400).send(`Invalid target URL provided in path: ${targetUrlString}`);
         return;
     }
     
     const requestHeaders = {...req.headers};
-    delete requestHeaders['host']; // host будет установлен fetch на основе targetUrl
-    // content-length будет пересчитан fetch из req.body, если req.body - Buffer или строка
-    // req.body здесь - это Buffer из express.raw(), так что fetch установит Content-Length
+    delete requestHeaders['host'];
     delete requestHeaders['content-length'];
 
-
-    const response = await fetch(targetUrl.toString(), { // Используем targetUrl.toString()
+    const response = await fetch(targetUrl.toString(), {
       method: req.method,
       headers: requestHeaders,
       body: (req.method !== 'GET' && req.method !== 'HEAD') ? req.body : undefined,
       redirect: 'manual',
-      // ---- КЛЮЧЕВОЕ ИЗМЕНЕНИЕ ----
-      // Говорим node-fetch не раскодировать ответ автоматически.
-      // Прокси будет передавать сжатый поток как есть.
       compress: false 
-      // --------------------------
     });
 
-    // Копируем все заголовки из ответа целевого сервера
-    // Теперь, когда node-fetch не раскодирует, мы можем безопасно копировать Content-Encoding и Content-Length
-    Object.entries(response.headers.raw()).forEach(([key, value]) => {
-        // Некоторые "hop-by-hop" заголовки не должны копироваться,
-        // но для большинства это безопасно, особенно Connection, который Node.js обработает.
-        // Transfer-Encoding также будет управляться Node.js/Express при пайпинге.
-        if (value) {
-            res.setHeader(key, value.length === 1 ? value[0] : value);
+    response.headers.forEach((value, key) => {
+        const lowerKey = key.toLowerCase();
+        if (!lowerKey.startsWith('access-control-') && 
+            lowerKey !== 'strict-transport-security' &&
+            lowerKey !== 'content-security-policy' &&
+            lowerKey !== 'public-key-pins' &&
+            lowerKey !== 'transfer-encoding' &&
+            lowerKey !== 'connection' &&
+            lowerKey !== 'keep-alive' &&
+            lowerKey !== 'proxy-authenticate' &&
+            lowerKey !== 'proxy-authorization' &&
+            lowerKey !== 'te' &&
+            lowerKey !== 'trailers' &&
+            lowerKey !== 'upgrade'
+        ) {
+            res.setHeader(key, value);
         }
     });
     
+    addCorsHeaders(res, clientRequestOrigin);
+    const exposedHeaders = Array.from(response.headers.keys()).filter(key => 
+        !key.toLowerCase().startsWith('access-control-')
+    ).join(', ');
+    if (exposedHeaders) {
+        res.setHeader('Access-Control-Expose-Headers', exposedHeaders || '*');
+    }
+    
     res.status(response.status);
 
-    // Потоковая передача тела ответа
     if (response.body) {
         response.body.pipe(res);
     } else {
@@ -76,32 +125,46 @@ app.all('*', async (req, res) => {
   } catch (error) {
     console.error(`[${new Date().toISOString()}] Proxy error for ${req.method} ${req.url}:`, error);
     if (!res.headersSent) {
+      addCorsHeaders(res, clientRequestOrigin);
+      let statusCode = 500;
+      let message = 'Proxy error occurred.';
+
       if (error.code === 'ENOTFOUND') {
-          res.status(404).send(`Target host not found: ${req.url.substring(1)}`);
-      } else if (error.message && error.message.includes('Invalid URL')) { // Проверка может быть улучшена
-          res.status(400).send(`Invalid target URL in path: ${req.url.substring(1)}`);
+          statusCode = 404;
+          message = `Target host not found: ${req.url.substring(1)}`;
+      } else if (error.message && error.message.includes('Invalid URL')) {
+          statusCode = 400;
+          message = `Invalid target URL in path: ${req.url.substring(1)}`;
       } else if (error.code === 'ECONNREFUSED') {
-          res.status(502).send(`Bad Gateway: Could not connect to target server at ${req.url.substring(1)}`);
-      } else if (error.code === 'ERR_INVALID_URL') { // от new URL(targetUrlString)
-          res.status(400).send(`Invalid target URL format in path: ${req.url.substring(1)}`);
-      }
-      else {
-          res.status(500).send('Proxy error occurred.');
+          statusCode = 502;
+          message = `Bad Gateway: Could not connect to target server at ${req.url.substring(1)}`;
+      } else if (error.code === 'ERR_INVALID_URL') {
+          statusCode = 400;
+          message = `Invalid target URL format in path: ${req.url.substring(1)}`;
       }
+      res.status(statusCode).send(message);
     } else {
-      // Если заголовки уже отправлены, но ошибка произошла при стриминге тела,
-      // мы можем только завершить соединение или попытаться записать ошибку в лог.
-      // response.body.pipe(res) уже могло начать отправку.
       console.error(`[${new Date().toISOString()}] Proxy error after headers sent for ${req.method} ${req.url}.`);
-      res.end(); // Пытаемся корректно завершить ответ, если это возможно.
+      res.end();
     }
   }
 });
 
+function addCorsHeaders(res, clientRequestOrigin) {
+  if (clientRequestOrigin) {
+    res.setHeader('Access-Control-Allow-Origin', clientRequestOrigin);
+    res.setHeader('Access-Control-Allow-Credentials', 'true');
+    res.setHeader('Vary', 'Origin');
+  } else {
+    res.setHeader('Access-Control-Allow-Origin', '*');
+  }
+}
+
 const PORT = process.env.PORT || 7860;
 const HOST = '0.0.0.0';
 
 app.listen(PORT, HOST, () => {
-  console.log(`Server running on http://${HOST}:${PORT}`);
-  console.log(`Proxying requests. Example: http://${HOST === '0.0.0.0' ? 'localhost' : HOST}:${PORT}/https://www.google.com`);
+  console.log(`Server running on http://${HOST}:${PORT} and acting as a CORS proxy.`);
+  console.log(`Handles OPTIONS preflight requests.`);
+  console.log(`Example: http://${HOST === '0.0.0.0' ? 'localhost' : HOST}:${PORT}/https://www.google.com`);
 });