Update server.js

server.js

@@ -1,128 +1,195 @@
-const express = require('express');
-const { createProxyMiddleware } = require('http-proxy-middleware');
-const path = require('path');
-const crypto = require('crypto');
-const fs = require('fs');
-
-const app = express();
-const PORT = process.env.PORT || 8080;
-
-// 中间件：解析请求体
-app.use(express.json());
-app.use(express.urlencoded({ extended: true }));
-
-// 中间件：处理HTML文件中的环境变量注入
-app.use((req, res, next) => {
-  if (req.path.endsWith('.html') || req.path === '/' || req.path.endsWith('/')) {
-    const filePath = req.path === '/' || req.path.endsWith('/') 
-      ? path.join(__dirname, 'index.html')
-      : path.join(__dirname, req.path);
-    
-    if (fs.existsSync(filePath)) {
-      let content = fs.readFileSync(filePath, 'utf8');
-      
-      // 替换密码占位符
-      const password = process.env.PASSWORD || '';
-      let passwordHash = '';
-      
-      if (password) {
-        const hash = crypto.createHash('sha256');
-        hash.update(password);
-        passwordHash = hash.digest('hex');
-      }
-      
-      content = content.replace(
-        'window.__ENV__.PASSWORD = "{{PASSWORD}}";',
-        `window.__ENV__.PASSWORD = "${passwordHash}"; // SHA-256 hash`
-      );
-      
-      res.setHeader('Content-Type', 'text/html');
-      return res.send(content);
-    }
-  }
-  next();
-});
-
-// 创建代理中间件函数
-function createDynamicProxy(req, res, next) {
-  // 从URL参数获取目标URL
-  const targetUrl = decodeURIComponent(req.params.url);
-  
-  if (!targetUrl || !targetUrl.match(/^https?:\/\/.+/i)) {
-    return res.status(400).json({
-      success: false,
-      error: '无效的目标URL'
-    });
-  }
-  
-  // 提取主机和协议
-  try {
-    const urlObj = new URL(targetUrl);
-    const target = `${urlObj.protocol}//${urlObj.host}`;
-    const pathToProxy = urlObj.pathname + urlObj.search;
-    
-    // 创建代理
-    const proxy = createProxyMiddleware({
-      target,
-      changeOrigin: true,
-      pathRewrite: () => pathToProxy,
-      onProxyReq: (proxyReq, req, res) => {
-        // 设置请求头
-        proxyReq.setHeader('User-Agent', 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/122.0.0.0 Safari/537.36');
-        proxyReq.setHeader('Accept', req.headers.accept || '*/*');
-        proxyReq.setHeader('Accept-Language', req.headers['accept-language'] || 'zh-CN,zh;q=0.9');
-        proxyReq.setHeader('Referer', req.headers.referer || target);
-      },
-      onProxyRes: (proxyRes, req, res) => {
-        // 设置CORS头
-        proxyRes.headers['Access-Control-Allow-Origin'] = '*';
-        proxyRes.headers['Access-Control-Allow-Methods'] = 'GET, HEAD, OPTIONS';
-        proxyRes.headers['Access-Control-Allow-Headers'] = '*';
-        
-        // 设置缓存策略
-        proxyRes.headers['Cache-Control'] = 'public, max-age=86400';
-        
-        // 移除可能导致问题的头部
-        delete proxyRes.headers['content-length'];
-        delete proxyRes.headers['content-encoding'];
-      }
-    });
-    
-    proxy(req, res, next);
-  } catch (error) {
-    console.error(`代理错误: ${error.message}`);
-    return res.status(500).json({
-      success: false,
-      error: `代理请求失败: ${error.message}`
-    });
-  }
-}
-
-// 设置代理路由
-app.use('/proxy/:url(*)', createDynamicProxy);
-
-// OPTIONS请求处理
-app.options('/proxy/:url(*)', (req, res) => {
-  res.setHeader('Access-Control-Allow-Origin', '*');
-  res.setHeader('Access-Control-Allow-Methods', 'GET, HEAD, OPTIONS');
-  res.setHeader('Access-Control-Allow-Headers', '*');
-  res.setHeader('Access-Control-Max-Age', '86400');
-  res.status(204).end();
-});
-
-// 静态文件服务 - 所有其他请求
-app.use(express.static(path.join(__dirname), {
-  maxAge: '1d'
-}));
-
-// 错误处理中间件
-app.use((err, req, res, next) => {
-  console.error(`服务器错误: ${err.stack}`);
-  res.status(500).send('服务器内部错误');
-});
-
-// 启动服务器
-app.listen(PORT, () => {
-  console.log(`LibreTV 服务器已启动，运行在 http://localhost:${PORT}`);
-  console.log(`代理服务可通过 http://localhost:${PORT}/proxy/{URL} 访问`);
+const express = require('express');
+const http = require('http');
+const https = require('https');
+const zlib = require('zlib');
+const path = require('path');
+const crypto = require('crypto');
+const fs = require('fs');
+
+const app = express();
+const PORT = process.env.PORT || 8080;
+
+// 中间件：解析请求体
+app.use(express.json());
+app.use(express.urlencoded({ extended: true }));
+
+// 中间件：处理HTML文件中的环境变量注入
+app.use((req, res, next) => {
+  if (req.path.endsWith('.html') || req.path === '/' || req.path.endsWith('/')) {
+    const filePath = req.path === '/' || req.path.endsWith('/') 
+      ? path.join(__dirname, 'index.html')
+      : path.join(__dirname, req.path);
+    
+    if (fs.existsSync(filePath)) {
+      let content = fs.readFileSync(filePath, 'utf8');
+      
+      // 替换密码占位符
+      const password = process.env.PASSWORD || '';
+      let passwordHash = '';
+      
+      if (password) {
+        const hash = crypto.createHash('sha256');
+        hash.update(password);
+        passwordHash = hash.digest('hex');
+      }
+      
+      content = content.replace(
+        'window.__ENV__.PASSWORD = "{{PASSWORD}}";',
+        `window.__ENV__.PASSWORD = "${passwordHash}"; // SHA-256 hash`
+      );
+      
+      res.setHeader('Content-Type', 'text/html');
+      return res.send(content);
+    }
+  }
+  next();
+});
+
+// 代理处理函数
+app.use('/proxy/:url(*)', async (req, res) => {
+  try {
+    // 获取目标 URL
+    const targetUrl = decodeURIComponent(req.params.url);
+    if (!targetUrl || !targetUrl.match(/^https?:\/\/.+/i)) {
+      return res.status(400).json({ error: '无效的目标URL' });
+    }
+    
+    console.log(`处理代理请求: ${targetUrl}`);
+    
+    // 选择合适的协议
+    const protocol = targetUrl.startsWith('https') ? https : http;
+    
+    // 设置请求选项
+    const options = {
+      headers: {
+        'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/122.0.0.0 Safari/537.36',
+        'Accept': req.headers.accept || '*/*',
+        'Accept-Language': req.headers['accept-language'] || 'zh-CN,zh;q=0.9',
+        'Accept-Encoding': 'gzip, deflate',
+        'Referer': req.headers.referer || new URL(targetUrl).origin
+      },
+      timeout: 10000 // 10秒超时
+    };
+    
+    // 创建请求
+    const proxyReq = protocol.get(targetUrl, options, (proxyRes) => {
+      // 设置CORS头
+      res.setHeader('Access-Control-Allow-Origin', '*');
+      res.setHeader('Access-Control-Allow-Methods', 'GET, HEAD, OPTIONS');
+      res.setHeader('Access-Control-Allow-Headers', '*');
+      
+      // 设置缓存头
+      res.setHeader('Cache-Control', 'public, max-age=86400');
+      
+      // 如果是JSON内容，确保设置正确的编码
+      if (proxyRes.headers['content-type'] && 
+          proxyRes.headers['content-type'].includes('application/json')) {
+        res.setHeader('Content-Type', 'application/json; charset=utf-8');
+      } else {
+        // 复制其他响应头
+        for (const [key, value] of Object.entries(proxyRes.headers)) {
+          if (!['content-length', 'transfer-encoding', 'content-encoding'].includes(key.toLowerCase())) {
+            res.setHeader(key, value);
+          }
+        }
+      }
+      
+      // 设置状态码
+      res.status(proxyRes.statusCode);
+      
+      // 处理内容编码
+      let responseStream = proxyRes;
+      const contentEncoding = proxyRes.headers['content-encoding']?.toLowerCase();
+      
+      if (contentEncoding === 'gzip') {
+        responseStream = proxyRes.pipe(zlib.createGunzip());
+      } else if (contentEncoding === 'deflate') {
+        responseStream = proxyRes.pipe(zlib.createInflate());
+      }
+      
+      // 收集响应数据
+      const chunks = [];
+      responseStream.on('data', (chunk) => chunks.push(chunk));
+      
+      responseStream.on('end', () => {
+        try {
+          const buffer = Buffer.concat(chunks);
+          
+          // 检查是否是M3U8内容
+          const isM3u8 = (proxyRes.headers['content-type'] && 
+                         (proxyRes.headers['content-type'].includes('application/vnd.apple.mpegurl') || 
+                          proxyRes.headers['content-type'].includes('application/x-mpegurl'))) ||
+                         buffer.toString('utf8', 0, 7) === '#EXTM3U';
+          
+          if (isM3u8) {
+            // 针对M3U8内容的特殊处理
+            const content = buffer.toString('utf8');
+            // 这里可以添加M3U8处理逻辑，如果需要的话
+            res.setHeader('Content-Type', 'application/vnd.apple.mpegurl');
+            res.end(content);
+            return;
+          }
+          
+          // 二进制内容，如视频片段，直接传递
+          if (proxyRes.headers['content-type'] && 
+              (proxyRes.headers['content-type'].includes('video/') || 
+               proxyRes.headers['content-type'].includes('audio/') ||
+               proxyRes.headers['content-type'].includes('application/octet-stream'))) {
+            res.end(buffer);
+            return;
+          }
+          
+          // 文本内容（包括JSON）
+          const body = buffer.toString('utf8');
+          res.end(body);
+        } catch (error) {
+          console.error('处理响应数据错误:', error);
+          res.status(502).json({ error: '处理API响应失败' });
+        }
+      });
+    });
+    
+    // 处理请求错误
+    proxyReq.on('error', (err) => {
+      console.error(`代理请求错误: ${err.message} (${targetUrl})`);
+      res.status(502).json({ error: `代理请求失败: ${err.message}` });
+    });
+    
+    // 处���超时
+    proxyReq.on('timeout', () => {
+      proxyReq.destroy();
+      console.error(`代理请求超时: ${targetUrl}`);
+      res.status(504).json({ error: '代理请求超时' });
+    });
+    
+  } catch (error) {
+    console.error(`代理处理错误: ${error.message}`);
+    res.status(500).json({ error: `代理服务器错误: ${error.message}` });
+  }
+});
+
+// 处理OPTIONS请求
+app.options('/proxy/:url(*)', (req, res) => {
+  res.setHeader('Access-Control-Allow-Origin', '*');
+  res.setHeader('Access-Control-Allow-Methods', 'GET, HEAD, OPTIONS');
+  res.setHeader('Access-Control-Allow-Headers', '*');
+  res.setHeader('Access-Control-Max-Age', '86400');
+  res.status(204).end();
+});
+
+// 静态文件服务 - 所有其他请求
+app.use(express.static(path.join(__dirname), {
+  maxAge: '1d'
+}));
+
+// 错误处理中间件
+app.use((err, req, res, next) => {
+  console.error(`服务器错误: ${err.stack}`);
+  res.status(500).send('服务器内部错误');
+});
+
+// 启动服务器
+app.listen(PORT, () => {
+  console.log(`LibreTV 服务器已启动，运行在 http://localhost:${PORT}`);
+  console.log(`代理服务可通过 http://localhost:${PORT}/proxy/{URL} 访问`);
 });