arn-browser 0.1.20 → 0.1.21

package/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "arn-browser",
-	"version": "0.1.20",
+	"version": "0.1.21",
 	"description": "A lightweight, browser autmation helper.",
 	"main": "src/index.js",
 	"types": "src/index.d.ts",

package/src/utility/proxy-utility/proxy-chain.js

@@ -365,10 +365,10 @@ export async function startProxyServer({
 		for (const [key, val] of Object.entries(stats)) {
 			formatted[key] = {
 				req: val.request,
-				srcTx: formatBytes(val.srcTx),
-				srcRx: formatBytes(val.srcRx),
-				trgTx: formatBytes(val.trgTx),
-				trgRx: formatBytes(val.trgRx),
+				sTx: formatBytes(val.srcTx),
+				sRx: formatBytes(val.srcRx),
+				tTx: formatBytes(val.trgTx),
+				tRx: formatBytes(val.trgRx),
 			};
 		}
 		return formatted;
@@ -382,10 +382,10 @@ export async function startProxyServer({
 				.reduce((acc, [host, hostData]) => {
 					acc[host] = {
 						req: hostData.req,
-						srcTx: formatBytes(hostData.srcTx),
-						srcRx: formatBytes(hostData.srcRx),
-						trgTx: formatBytes(hostData.trgTx),
-						trgRx: formatBytes(hostData.trgRx),
+						sTx: formatBytes(hostData.srcTx),
+						sRx: formatBytes(hostData.srcRx),
+						tTx: formatBytes(hostData.trgTx),
+						tRx: formatBytes(hostData.trgRx),
 					};
 					return acc;
 				}, {});
@@ -421,16 +421,34 @@ export async function startProxyServer({
 		isServerRunning: () => serverRunning,
 
 		closeServer: async () => {
-			// Close the server — triggers connectionClosed events which accumulate byte stats
+			// Close the server — triggers socket destroys
 			await server.close(true);
 			serverRunning = false;
 
-			// Log stats AFTER close so all connectionClosed events have fired
+			// Wait up to 2000ms (2 seconds) for all async 'connectionClosed' events to fire
+			// When sockets are destroyed, their 'close' events happen asynchronously
+			let maxWait = 200; // 200 * 10ms = 2000ms
+			while (Object.keys(connectionMap).length > 0 && maxWait > 0) {
+				await new Promise((resolve) => setTimeout(resolve, 10));
+				maxWait--;
+			}
+
+			// Log stats AFTER close so all connectionClosed events have accumulated bytes
 			if (proxy_stats) {
-				console.log("░░ Proxy Stats:", getProxyStatsFormatted());
+				console.log("\n░░ Proxy Stats:");
+				for (const [type, data] of Object.entries(getProxyStatsFormatted())) {
+					console.log(`  ${type}: { req: ${data.req}, sTx: '${data.sTx}', sRx: '${data.sRx}', tTx: '${data.tTx}', tRx: '${data.tRx}' }`);
+				}
 			}
 			if (host_stats) {
-				console.log("░░ Host Stats:", getHostStatsFormatted());
+				console.log("░░ Host Stats:");
+				for (const [type, hosts] of Object.entries(getHostStatsFormatted())) {
+					console.log(`  ${type}: {`);
+					for (const [host, data] of Object.entries(hosts)) {
+						console.log(`    '${host}': { req: ${data.req}, sTx: '${data.sTx}', sRx: '${data.sRx}', tTx: '${data.tTx}', tRx: '${data.tRx}' }`);
+					}
+					console.log(`  }`);
+				}
 			}
 
 			console.log("░░ Proxy server closed.");