From e2a38dda22cbe81ba435f56133f1ca9796635aa6 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Fl=C3=A1vio=20Caetano?= <flavio@vieiracaetano.com>
Date: Mon, 5 Mar 2018 18:36:35 -0300
Subject: [PATCH] Feature: debug logging

---
 ReCaptcha/Assets/recaptcha.html               | 24 ++++++++++++++-----
 ReCaptcha/Classes/ReCaptchaDecoder.swift      |  7 ++++++
 .../Classes/ReCaptchaWebViewManager.swift     |  5 ++++
 3 files changed, 30 insertions(+), 6 deletions(-)
diff --git a/ReCaptcha/Assets/recaptcha.html b/ReCaptcha/Assets/recaptcha.html
index e8223d1..37f5026 100644
--- a/ReCaptcha/Assets/recaptcha.html
+++ b/ReCaptcha/Assets/recaptcha.html
@@ -2,6 +2,19 @@
 <head>
   <meta name="viewport" content="width=device-width" />
   <script type="text/javascript">
+  var post = function(value) {
+    window.webkit.messageHandlers.recaptcha.postMessage(value);
+  };
+
+  console.log = function(message) {
+      post({log: message});
+  };
+
+  var showReCaptcha = function() {
+      console.log("showReCaptcha");
+      post({action: "showReCaptcha"});
+  };
+  
   var observeDOM = function(element, completion) {
     new MutationObserver(function(mutations) {
       mutations.forEach(function(mutationRecord) {
@@ -20,20 +33,19 @@
     }
 
     // Listens to changes on the div element that presents the ReCaptcha challenge
-    observeDOM(document.getElementsByTagName("div")[3], function() {
-      console.log({action: "showReCaptcha"});
-      window.webkit.messageHandlers.recaptcha.postMessage({action: "showReCaptcha"});
-    });
+    observeDOM(document.getElementsByTagName("div")[3], showReCaptcha);
 
+    console.log("executing");
     grecaptcha.execute();
-  }
+  };
 
   var onSubmit = function(token) {
     console.log(token);
-    window.webkit.messageHandlers.recaptcha.postMessage({token: token});
+    post({token: token});
   };
 
   var onloadCallback = function() {
+    console.log("did load");
     grecaptcha.render('submit', {
       'sitekey' : '${apiKey}',
       'callback' : onSubmit,
diff --git a/ReCaptcha/Classes/ReCaptchaDecoder.swift b/ReCaptcha/Classes/ReCaptchaDecoder.swift
index bfc679d..bb205a9 100644
--- a/ReCaptcha/Classes/ReCaptchaDecoder.swift
+++ b/ReCaptcha/Classes/ReCaptchaDecoder.swift
@@ -27,6 +27,9 @@ internal class ReCaptchaDecoder: NSObject {
 
         /// Did finish loading resources
         case didLoad
+
+        /// Logs a string onto the console
+        case log(String)
     }
 
     /// The closure that receives messages
@@ -100,6 +103,10 @@ fileprivate extension ReCaptchaDecoder.Result {
             }
         }
 
+        if let message = response["log"] as? String {
+            return .log(message)
+        }
+
         return .error(.wrongMessageFormat)
     }
 }
diff --git a/ReCaptcha/Classes/ReCaptchaWebViewManager.swift b/ReCaptcha/Classes/ReCaptchaWebViewManager.swift
index 93db2ed..089a0ee 100644
--- a/ReCaptcha/Classes/ReCaptchaWebViewManager.swift
+++ b/ReCaptcha/Classes/ReCaptchaWebViewManager.swift
@@ -257,6 +257,11 @@ fileprivate extension ReCaptchaWebViewManager {
         case .didLoad:
             // For testing purposes
             webviewDelegate.execute()
+
+        case .log(let message):
+            #if DEBUG
+            print("[JS LOG]:", message)
+            #endif
         }
     }
 
