From 8ee5b8b4365a806bb626cc688b43ea3004b4aa95 Mon Sep 17 00:00:00 2001
From: Yury Semikhatsky <yurys@chromium.org>
Date: Fri, 23 Apr 2021 14:25:32 -0700
Subject: [PATCH] feat: support channel logging (#410)

---
 .../java/com/microsoft/playwright/impl/Connection.java | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

diff --git a/playwright/src/main/java/com/microsoft/playwright/impl/Connection.java b/playwright/src/main/java/com/microsoft/playwright/impl/Connection.java
index e1990a34..5a60cd98 100644
--- a/playwright/src/main/java/com/microsoft/playwright/impl/Connection.java
+++ b/playwright/src/main/java/com/microsoft/playwright/impl/Connection.java
@@ -67,6 +67,7 @@ public class Connection {
   private int lastId = 0;
   private final Path srcDir;
   private final Map<Integer, WaitableResult<JsonElement>> callbacks = new HashMap<>();
+  private final boolean isLogging = "pw:channel".equals(System.getenv("DEBUG"));
 
   class Root extends ChannelOwner {
     Root(Connection connection) {
@@ -153,7 +154,11 @@ public class Connection {
       metadata.add("stack", currentStackTrace());
       message.add("metadata", metadata);
     }
-    transport.send(gson().toJson(message));
+    String messageString = gson().toJson(message);
+    if (isLogging) {
+      System.err.println("SEND ► " + messageString);
+    }
+    transport.send(messageString);
     return result;
   }
 
@@ -184,6 +189,9 @@ public class Connection {
     if (messageString == null) {
       return;
     }
+    if (isLogging) {
+      System.err.println("◀ RECV " + messageString);
+    }
     Gson gson = gson();
     Message message = gson.fromJson(messageString, Message.class);
     dispatch(message);