Update src/ProxyServer.js

src/ProxyServer.js

@@ -19,6 +19,8 @@ const logger = {
   error: (message) => console.error(chalk.red(`[ProxyServer] ${message}`)),
   warning: (message) => console.warn(chalk.yellow(`[ProxyServer] ${message}`)),
   success: (message) => console.log(chalk.green(`[ProxyServer] ${message}`)),
+  // 新增一个专门用于子进程输出的 logger
+  subprocess: (message) => console.log(chalk.gray(`[ProxyBinary] ${message}`)),
 };
 
 class ProxyServer {
@@ -26,57 +28,36 @@ class ProxyServer {
     this.proxyProcess = null;
     this.platform = process.env.PROXY_SERVER_PLATFORM || 'auto';
     this.port = process.env.PROXY_SERVER_PORT || 10655;
-    // ***【修改 1】*** 将日志路径指向 /tmp/logs，这是一个在容器中保证可写的目录
-    this.logPath = process.env.PROXY_SERVER_LOG_PATH || '/tmp/logs/proxy_server.log';
+    // ***【修改 1】*** 移除了 this.logPath 和 this.logStream
     this.enabled = process.env.ENABLE_PROXY_SERVER === 'true';
     this.proxyAuthToken = process.env.PROXY_AUTH_TOKEN || 'default_token';
-    this.logStream = null;
   }
 
-  // 获取当前系统平台
+  // 获取当前系统平台 (无变化)
   detectPlatform() {
     if (this.platform !== 'auto') {
       return this.platform;
     }
-
     const platform = os.platform();
     const arch = os.arch();
-
-    if (platform === 'win32') {
-      return 'windows';
-    } else if (platform === 'linux') {
-      // 这里的逻辑保持不变，因为 Docker 容器通常是 linux amd64
-      if (arch === 'arm64') {
-        return 'android'; // 或者 'linux_arm64' 如果你有这个二进制
-      } else {
-        return 'linux';
-      }
-    } else if (platform === 'android') {
-      return 'android';
-    } else {
-      logger.warning(`未知平台: ${platform}, ${arch}, 默认使用linux版本`);
-      return 'linux';
-    }
+    if (platform === 'win32') return 'windows';
+    if (platform === 'linux') return (arch === 'arm64' ? 'android' : 'linux');
+    if (platform === 'android') return 'android';
+    logger.warning(`未知平台: ${platform}, ${arch}, 默认使用linux版本`);
+    return 'linux';
   }
 
-  // 获取代理服务器可执行文件路径
+  // 获取代理服务器可执行文件路径 (无变化)
   getProxyServerPath() {
     const platform = this.detectPlatform();
-
-    // ***【修改 2】*** 在生产环境中(如Docker)，从 /tmp/proxy 获取二进制文件
-    // 在本地开发时，仍然从 src/proxy 获取
     const isProduction = process.env.NODE_ENV === 'production';
     const proxyDir = isProduction ? '/tmp/proxy' : join(__dirname, 'proxy');
-
     logger.info(`环境: ${isProduction ? 'Production' : 'Development'}, 代理二进制目录: ${proxyDir}`);
 
     switch (platform) {
-      case 'windows':
-        return join(proxyDir, 'chrome_proxy_server_windows_amd64.exe');
-      case 'linux':
-        return join(proxyDir, 'chrome_proxy_server_linux_amd64');
-      case 'android':
-        return join(proxyDir, 'chrome_proxy_server_android_arm64');
+      case 'windows': return join(proxyDir, 'chrome_proxy_server_windows_amd64.exe');
+      case 'linux': return join(proxyDir, 'chrome_proxy_server_linux_amd64');
+      case 'android': return join(proxyDir, 'chrome_proxy_server_android_arm64');
       default:
         logger.error(`不支持的平台: ${platform}`);
         return null;
@@ -89,123 +70,79 @@ class ProxyServer {
       logger.info('代理服务器未启用，跳过启动');
       return;
     }
-
     if (this.proxyProcess) {
       logger.warning('代理服务器已经在运行中');
       return;
     }
 
     const proxyServerPath = this.getProxyServerPath();
-    if (!proxyServerPath) {
-      logger.error('无法获取代理服务器路径');
+    if (!proxyServerPath || !fs.existsSync(proxyServerPath)) {
+      logger.error(`代理二进制文件不存在于路径: ${proxyServerPath || '未知'}`);
       return;
     }
-
-    // 检查文件是否存在，提供更详细的调试信息
-    if (!fs.existsSync(proxyServerPath)) {
-        logger.error(`代理二进制文件不存在于路径: ${proxyServerPath}`);
-        return;
-    }
-
+    
     try {
-      // 在生产环境中，Dockerfile 已经处理了权限问题，本地开发时可以尝试设置
-      if (process.env.NODE_ENV !== 'production' && this.detectPlatform() !== 'windows') {
-        try {
-          fs.chmodSync(proxyServerPath, 0o755);
-        } catch (err) {
-          logger.warning(`本地开发环境无法设置执行权限: ${err.message}`);
-        }
-      }
-
-      // 创建日志文件
-      this.logStream = fs.createWriteStream(this.logPath, { flags: 'a' });
+      // ***【修改 2】*** 移除了所有与文件日志相关的代码
       
       // 启动代理服务器进程
       this.proxyProcess = spawn(proxyServerPath, [
         '--port', this.port.toString(),
         '--token', this.proxyAuthToken
       ], {
-        stdio: ['ignore', 'pipe', 'pipe'], // 使用 pipe 而不是直接传递流
+        stdio: ['ignore', 'pipe', 'pipe'],
         detached: false
       });
 
-      // 将进程的输出重定向到日志文件
+      // ***【修改 3】*** 将子进程的输出直接打印到控制台
       if (this.proxyProcess.stdout) {
-        this.proxyProcess.stdout.pipe(this.logStream);
+        this.proxyProcess.stdout.on('data', (data) => {
+          logger.subprocess(data.toString().trim());
+        });
       }
-      
       if (this.proxyProcess.stderr) {
-        this.proxyProcess.stderr.pipe(this.logStream);
+        this.proxyProcess.stderr.on('data', (data) => {
+          logger.error(`[ProxyBinary-Error] ${data.toString().trim()}`);
+        });
       }
 
-      // 设置进程事件处理
       this.proxyProcess.on('error', (err) => {
         logger.error(`代理服务器启动失败: ${err.message}`);
         this.proxyProcess = null;
-        if (this.logStream) {
-          this.logStream.end();
-          this.logStream = null;
-        }
       });
-
       this.proxyProcess.on('exit', (code, signal) => {
         logger.info(`代理服务器已退出，退出码: ${code}, 信号: ${signal}`);
         this.proxyProcess = null;
-        if (this.logStream) {
-          this.logStream.end();
-          this.logStream = null;
-        }
       });
 
-      // 等待一段时间，确保服务器启动
       await new Promise(resolve => setTimeout(resolve, 1000));
       
       if (this.proxyProcess && this.proxyProcess.exitCode === null) {
-        logger.success(`代理服务器已启动，端口: ${this.port}, 日志文件: ${this.logPath}`);
+        logger.success(`代理服务器已启动，端口: ${this.port}`);
         return true;
       } else {
-        logger.error('代理服务器启动失败或立即退出。请检查日志。');
-        if (this.logStream) {
-          this.logStream.end();
-          this.logStream = null;
-        }
+        logger.error('代理服务器启动失败或立即退出。');
         return false;
       }
     } catch (error) {
       logger.error(`启动代理服务器时出错: ${error.message}`);
-      if (this.logStream) {
-        this.logStream.end();
-        this.logStream = null;
-      }
       return false;
     }
   }
 
-  // 停止代理服务器
+  // 停止代理服务器 (无变化)
   stop() {
-    if (!this.proxyProcess) {
-      //logger.info('代理服务器已关闭');
-      return;
-    }
-
+    if (!this.proxyProcess) return;
     try {
-      // 在Windows上使用taskkill确保子进程也被终止
       if (this.detectPlatform() === 'windows' && this.proxyProcess.pid) {
         spawn('taskkill', ['/pid', this.proxyProcess.pid, '/f', '/t']);
       } else {
-        // 在Linux/Android上使用kill信号
         this.proxyProcess.kill('SIGTERM');
       }
-
       logger.success('代理服务器已停止');
     } catch (error) {
       logger.error(`停止代理服务器时出错: ${error.message}`);
     } finally {
       this.proxyProcess = null;
-      if (this.logStream) {
-        this.logStream.end();
-        this.logStream = null;
-      }
     }
   }
 }