Upload server.js

server.js

@@ -444,6 +444,7 @@ app.all('/r', async (req, res) => {
     })
 
     const isJson = (upstream.headers.get('content-type') || '').includes('application/json');
+    const isM3u8 = (upstream.headers.get('content-type') || '').includes('mpegurl') || source.includes('.m3u8');
 
     if (isJson) {
       const buffer = await upstream.arrayBuffer();
@@ -481,6 +482,46 @@ app.all('/r', async (req, res) => {
       }
       res.setHeader('access-control-allow-origin', '*');
       
+      return res.end(outBuf);
+    } else if (isM3u8) {
+      const buffer = await upstream.arrayBuffer();
+      let text = Buffer.from(buffer).toString('utf8');
+      
+      const serverBase = getSelfBase(req);
+      const baseUrl = new URL(source);
+      
+      const lines = text.split('\n');
+      for (let i = 0; i < lines.length; i++) {
+        let line = lines[i].trim();
+        if (line && !line.startsWith('#')) {
+          try {
+            const absoluteUrl = new URL(line, baseUrl).toString();
+            lines[i] = serverBase + '/r?url=' + encodeURIComponent(absoluteUrl);
+          } catch(e) {}
+        } else if (line.startsWith('#EXT-X-STREAM-INF:') || line.startsWith('#EXT-X-MEDIA:') || line.startsWith('#EXT-X-I-FRAME-STREAM-INF:')) {
+          lines[i] = line.replace(/URI=["']([^"']+)["']/g, (match, uri) => {
+            try {
+              const absoluteUrl = new URL(uri, baseUrl).toString();
+              return `URI="${serverBase}/r?url=${encodeURIComponent(absoluteUrl)}"`;
+            } catch(e) {
+              return match;
+            }
+          });
+        }
+      }
+      text = lines.join('\n');
+      const outBuf = Buffer.from(text, 'utf8');
+      
+      res.status(upstream.status);
+      for (const h of RES_HEADERS_PASS) {
+        if (h === 'content-length') {
+          res.setHeader(h, outBuf.length);
+          continue;
+        }
+        let v = upstream.headers.get(h);
+        if (v) res.setHeader(h, v);
+      }
+      res.setHeader('access-control-allow-origin', '*');
       return res.end(outBuf);
     }