@skrillex1224/playwright-toolkit 3.0.7 → 3.0.9

package/dist/internals/proxy-meter.js

@@ -32,6 +32,25 @@ const state = {
     : null,
 };
 
+const formatFatalError = (error) => {
+  if (error instanceof Error) {
+    return error.stack || error.message || String(error);
+  }
+  try {
+    return JSON.stringify(error);
+  } catch {
+    return String(error);
+  }
+};
+
+const exitAfterFatal = (kind, error) => {
+  try {
+    console.error(`[proxy-meter] fatal ${kind}: ${formatFatalError(error)}`);
+  } catch {}
+  flushSnapshot();
+  process.exit(1);
+};
+
 const getHostBucket = (host) => {
   if (!host) return null;
   if (!state.hosts[host]) {
@@ -60,6 +79,21 @@ const safeError = (error) => {
   return message.length > 240 ? message.slice(0, 240) : message;
 };
 
+const sendBadGateway = (res, error = null) => {
+  if (!res || res.destroyed || res.writableEnded) return;
+
+  try {
+    if (!res.headersSent) {
+      res.writeHead(502);
+      res.end('Bad Gateway');
+      return;
+    }
+
+    // Headers are already committed; the only truthful error signal left is closing the stream.
+    res.destroy(error instanceof Error ? error : undefined);
+  } catch {}
+};
+
 const statusLabel = (statusCode, error) => {
   const code = Number(statusCode) || 0;
   if (error || code <= 0) return 'ERR';
@@ -349,8 +383,7 @@ const forwardHttp = (req, res) => {
 
   if (!target) {
     tracker.close({ statusCode: 0, error: 'invalid_target_url' });
-    res.writeHead(502);
-    res.end('Bad Gateway');
+    sendBadGateway(res);
     return;
   }
 
@@ -381,7 +414,19 @@ const forwardHttp = (req, res) => {
   let responseStatus = 0;
   const proxyReq = http.request(requestOptions, (proxyRes) => {
     responseStatus = Number(proxyRes.statusCode) || 0;
-    res.writeHead(responseStatus || 502, proxyRes.headers);
+    if (res.destroyed || res.writableEnded) {
+      tracker.close({ statusCode: responseStatus || 499, error: 'client_response_closed' });
+      proxyRes.resume();
+      return;
+    }
+    try {
+      res.writeHead(responseStatus || 502, proxyRes.headers);
+    } catch (error) {
+      tracker.close({ statusCode: responseStatus, error: safeError(error) });
+      proxyRes.resume();
+      sendBadGateway(res, error);
+      return;
+    }
     proxyRes.on('data', (chunk) => {
       const size = chunk?.length || 0;
       addTraffic(hostname, 'in', size);
@@ -412,8 +457,7 @@ const forwardHttp = (req, res) => {
 
   proxyReq.on('error', (error) => {
     tracker.close({ statusCode: responseStatus, error: safeError(error) });
-    res.writeHead(502);
-    res.end('Bad Gateway');
+    sendBadGateway(res, error);
   });
 };
 
@@ -545,5 +589,7 @@ const shutdown = () => {
   process.exit(0);
 };
 
+process.on('uncaughtException', (error) => exitAfterFatal('uncaughtException', error));
+process.on('unhandledRejection', (error) => exitAfterFatal('unhandledRejection', error));
 process.on('SIGINT', shutdown);
 process.on('SIGTERM', shutdown);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@skrillex1224/playwright-toolkit",
-  "version": "3.0.7",
+  "version": "3.0.9",
   "description": "一个在 Apify/Crawlee Actor 中启用实时截图视图的实用工具库。",
   "main": "dist/index.cjs",
   "module": "dist/index.js",