Upload server.js

server.js

@@ -422,6 +422,36 @@ app.all('/r', async (req, res) => {
       signal: controller.signal
     })
 
+    const isJson = (upstream.headers.get('content-type') || '').includes('application/json');
+
+    if (isJson) {
+      const buffer = await upstream.arrayBuffer();
+      let text = Buffer.from(buffer).toString('utf8');
+
+      // Replace wss:// urls to go through our proxy
+      const serverBase = getSelfBase(req);
+      const wsBase = serverBase.replace(/^http/, 'ws');
+      
+      text = text.replace(/wss:\/\/[^"'\s]+/g, (match) => {
+        return wsBase + '/ws?url=' + encodeURIComponent(match);
+      });
+
+      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;
+        }
+        const v = upstream.headers.get(h);
+        if (v) res.setHeader(h, v);
+      }
+      res.setHeader('access-control-allow-origin', '*');
+      
+      return res.send(outBuf);
+    }
+
     res.status(upstream.status)
     for (const h of RES_HEADERS_PASS) {
       if (h === 'content-length') {
@@ -450,6 +480,65 @@ app.all('/r', async (req, res) => {
 app.options('*', (req, res) => res.sendStatus(204))
 
 const port = Number(process.env.PORT || 7860)
-app.listen(port, () => {
+const server = app.listen(port, () => {
   console.log('lampa-es5-proxy on :' + port)
 })
+
+server.on('upgrade', (req, clientSocket, head) => {
+  try {
+    const urlObj = new URL(req.url, 'http://localhost');
+    if (urlObj.pathname === '/ws') {
+      let targetUrlStr = urlObj.searchParams.get('url');
+      if (!targetUrlStr) {
+        clientSocket.destroy();
+        return;
+      }
+      
+      const targetUrl = new URL(targetUrlStr);
+      const options = {
+        port: targetUrl.port || (targetUrl.protocol === 'wss:' ? 443 : 80),
+        host: targetUrl.hostname,
+        headers: req.headers,
+        path: targetUrl.pathname + targetUrl.search
+      };
+      
+      // Clean up headers
+      options.headers.host = targetUrl.hostname;
+      delete options.headers['x-forwarded-for'];
+      delete options.headers['x-forwarded-proto'];
+      delete options.headers['x-forwarded-host'];
+      
+      const proto = targetUrl.protocol === 'wss:' ? require('https') : require('http');
+      const proxyReq = proto.request(options);
+      
+      proxyReq.on('upgrade', (proxyRes, proxySocket, proxyHead) => {
+        let headers = 'HTTP/1.1 101 Switching Protocols\r\n';
+        for (let i = 0; i < proxyRes.rawHeaders.length; i += 2) {
+          headers += proxyRes.rawHeaders[i] + ': ' + proxyRes.rawHeaders[i+1] + '\r\n';
+        }
+        headers += '\r\n';
+        clientSocket.write(headers);
+        
+        if (proxyHead && proxyHead.length) clientSocket.write(proxyHead);
+        
+        proxySocket.pipe(clientSocket);
+        clientSocket.pipe(proxySocket);
+      });
+      
+      proxyReq.on('error', (err) => {
+        clientSocket.destroy();
+      });
+      
+      clientSocket.on('error', () => {
+        proxyReq.destroy();
+      });
+      
+      // Send the initial upgrade request
+      proxyReq.end();
+    } else {
+      clientSocket.destroy();
+    }
+  } catch(e) {
+    clientSocket.destroy();
+  }
+})