Update proxy.js

proxy.js

@@ -1,40 +1,97 @@
-const http = require('http');
-
-const TARGET_PORT = 11470;
-const LISTEN_PORT = 7860;
-
-// Create a proxy server
-const proxy = http.createServer((clientReq, clientRes) => {
-  console.log(`Proxying request to: ${clientReq.url}`);
-
-  // Configure the request options
-  const options = {
-    hostname: 'localhost',
-    port: TARGET_PORT,
-    path: clientReq.url,
-    method: clientReq.method,
-    headers: clientReq.headers
-  };
-
-  // Create the proxied request
-  const proxyReq = http.request(options, (proxyRes) => {
-    clientRes.writeHead(proxyRes.statusCode, proxyRes.headers);
-    proxyRes.pipe(clientRes, { end: true });
-  });
-
-  // Handle errors
-  proxyReq.on('error', (err) => {
-    console.error('Proxy error:', err);
-    clientRes.writeHead(500);
-    clientRes.end('Proxy error connecting to Stremio server');
-  });
-
-  // Pipe the client request to the proxy request
-  clientReq.pipe(proxyReq, { end: true });
-});
-
-// Start the proxy server
-proxy.listen(LISTEN_PORT, () => {
-  console.log(`Proxy server running on port ${LISTEN_PORT}`);
-  console.log(`Forwarding to Stremio server on port ${TARGET_PORT}`);
-}); 
+const http = require('http');
+
+const TARGET_PORT = 11470;
+const LISTEN_PORT = 7860;
+
+console.log(`===== Stremio Proxy Initializing =====`);
+console.log(`${new Date().toISOString()}: Proxy will forward from port ${LISTEN_PORT} to ${TARGET_PORT}`);
+
+// Function to check if Stremio server is running
+function checkStremioServer() {
+  return new Promise((resolve) => {
+    const req = http.request({
+      hostname: 'localhost',
+      port: TARGET_PORT,
+      path: '/',
+      method: 'HEAD',
+      timeout: 2000
+    }, (res) => {
+      console.log(`${new Date().toISOString()}: Stremio server check status: ${res.statusCode}`);
+      resolve(true);
+    });
+    
+    req.on('error', (err) => {
+      console.log(`${new Date().toISOString()}: Stremio server check failed: ${err.message}`);
+      resolve(false);
+    });
+    
+    req.on('timeout', () => {
+      console.log(`${new Date().toISOString()}: Stremio server check timed out`);
+      req.destroy();
+      resolve(false);
+    });
+    
+    req.end();
+  });
+}
+
+// Create a proxy server
+const proxy = http.createServer((clientReq, clientRes) => {
+  const clientIP = clientReq.headers['x-forwarded-for'] || clientReq.socket.remoteAddress;
+  console.log(`${new Date().toISOString()}: Proxying request from ${clientIP} to: ${clientReq.url}`);
+
+  // Configure the request options
+  const options = {
+    hostname: 'localhost',
+    port: TARGET_PORT,
+    path: clientReq.url,
+    method: clientReq.method,
+    headers: {...clientReq.headers, host: `localhost:${TARGET_PORT}`}
+  };
+
+  // Create the proxied request
+  const proxyReq = http.request(options, (proxyRes) => {
+    console.log(`${new Date().toISOString()}: Got response from Stremio server: ${proxyRes.statusCode}`);
+    clientRes.writeHead(proxyRes.statusCode, proxyRes.headers);
+    proxyRes.pipe(clientRes, { end: true });
+  });
+
+  // Handle errors
+  proxyReq.on('error', (err) => {
+    console.error(`${new Date().toISOString()}: Proxy error: ${err.message}`);
+    clientRes.writeHead(503);
+    clientRes.end(`Service Unavailable: Stremio server not responding. Please check logs and try again later.`);
+  });
+
+  // Add request timeout
+  proxyReq.on('timeout', () => {
+    console.error(`${new Date().toISOString()}: Request to Stremio server timed out`);
+    proxyReq.destroy();
+  });
+  
+  // Pipe the client request to the proxy request
+  clientReq.pipe(proxyReq, { end: true });
+});
+
+// Add error handling for the proxy server
+proxy.on('error', (err) => {
+  console.error(`${new Date().toISOString()}: Proxy server error: ${err.message}`);
+});
+
+// Start health check then start the proxy server
+(async () => {
+  console.log(`${new Date().toISOString()}: Performing initial health check on Stremio server...`);
+  const isStremioRunning = await checkStremioServer();
+  
+  if (isStremioRunning) {
+    console.log(`${new Date().toISOString()}: Stremio server is responding, starting proxy`);
+  } else {
+    console.log(`${new Date().toISOString()}: Stremio server is not responding, but starting proxy anyway`);
+  }
+  
+  proxy.listen(LISTEN_PORT, () => {
+    console.log(`${new Date().toISOString()}: Proxy server running on port ${LISTEN_PORT}`);
+    console.log(`${new Date().toISOString()}: Forwarding to Stremio server on port ${TARGET_PORT}`);
+    console.log(`===== Stremio Proxy Started Successfully =====`);
+  });
+})();