feat: add tracing for evaluate_script, ipc_handler and custom_protocols

.changes/tracing.md

@@ -0,0 +1,5 @@
+---
+"wry": patch
+---
+
+Added tracing spans for `evaluate_script`, `ipc_handler` and `custom_protocols`.

Cargo.toml

@@ -36,6 +36,7 @@ linux-headers = [ "webkit2gtk/v2_36" ]
 [dependencies]
 libc = "0.2"
 log = "0.4"
+tracing = "0.1"
 once_cell = "1"
 serde = { version = "1.0", features = [ "derive" ] }
 serde_json = "1.0"

src/webview/webkitgtk/mod.rs

@@ -89,6 +89,7 @@ impl InnerWebView {
 
     // Connect before registering as recommended by the docs
     manager.connect_script_message_received(None, move |_m, msg| {
+      let _span = tracing::info_span!("wry::ipc::handle").entered();
       if let Some(js) = msg.js_value() {
         if let Some(ipc_handler) = &ipc_handler {
           ipc_handler(&w, js.to_string());
@@ -401,7 +402,10 @@ impl InnerWebView {
       pending_scripts.push(js.into());
     } else {
       let cancellable: Option<&Cancellable> = None;
-      self.webview.run_javascript(js, cancellable, |_| ());
+      let span = SendEnteredSpan(tracing::debug_span!("wry::eval").entered());
+      self.webview.run_javascript(js, cancellable, move |_| {
+        drop(span);
+      });
     }
     Ok(())
   }
@@ -490,3 +494,8 @@ pub fn platform_webview_version() -> Result<String> {
   };
   Ok(format!("{}.{}.{}", major, minor, patch))
 }
+
+// SAFETY: only use this when you are sure the span will be dropped on the same thread it was entered
+struct SendEnteredSpan(tracing::span::EnteredSpan);
+
+unsafe impl Send for SendEnteredSpan {}

src/webview/webkitgtk/web_context.rs

@@ -287,9 +287,14 @@ where
     .register_uri_scheme_as_secure(name);
 
   context.register_uri_scheme(name, move |request| {
+    let span =
+      tracing::info_span!("wry::custom_protocol::handle", uri = tracing::field::Empty).entered();
+
     if let Some(uri) = request.uri() {
       let uri = uri.as_str();
 
+      span.record("uri", uri);
+
       // FIXME: Read the body (forms post)
       #[allow(unused_mut)]
       let mut http_request = Request::builder().uri(uri).method("GET");
@@ -325,7 +330,11 @@ where
         }
       };
 
-      match handler(&http_request) {
+      let res = {
+        let _span = tracing::info_span!("wry::custom_protocol::call_handler").entered();
+        handler(&http_request)
+      };
+      match res {
         Ok(http_response) => {
           let buffer = http_response.body();
           let input = gio::MemoryInputStream::from_bytes(&glib::Bytes::from(buffer));

src/webview/webview2/mod.rs

@@ -358,6 +358,7 @@ window.addEventListener('mousemove', (e) => window.chrome.webview.postMessage('_
             }
 
             if let Some(ipc_handler) = &ipc_handler {
+              let _span = tracing::info_span!("wry::ipc::handle").entered();
               ipc_handler(&window, js);
             }
           }
@@ -514,6 +515,9 @@ window.addEventListener('mousemove', (e) => window.chrome.webview.postMessage('_
         webview
           .add_WebResourceRequested(
             &WebResourceRequestedEventHandler::create(Box::new(move |_, args| {
+              let span =
+                tracing::info_span!("wry::custom_protocol::handle", uri = tracing::field::Empty)
+                  .entered();
               if let Some(args) = args {
                 let webview_request = args.Request()?;
                 let mut request = Request::builder();
@@ -570,6 +574,8 @@ window.addEventListener('mousemove', (e) => window.chrome.webview.postMessage('_
                 webview_request.Uri(&mut uri)?;
                 let uri = take_pwstr(uri);
 
+                span.record("uri", &uri);
+
                 if let Some(custom_protocol) = custom_protocols
                   .iter()
                   .find(|(name, _)| uri.starts_with(&format!("{scheme}://{name}.")))
@@ -589,7 +595,11 @@ window.addEventListener('mousemove', (e) => window.chrome.webview.postMessage('_
                     Err(_) => return Err(E_FAIL.into()),
                   };
 
-                  return match (custom_protocol.1)(&final_request) {
+                  let res = {
+                    let _span = tracing::info_span!("wry::custom_protocol::call_handler").entered();
+                    (custom_protocol.1)(&final_request)
+                  };
+                  return match res {
                     Ok(sent_response) => {
                       let content = sent_response.body();
                       let status_code = sent_response.status();
@@ -801,9 +811,13 @@ window.addEventListener('mousemove', (e) => window.chrome.webview.postMessage('_
 
   fn execute_script(webview: &ICoreWebView2, js: String) -> windows::core::Result<()> {
     unsafe {
+      let span = tracing::debug_span!("wry::eval").entered();
       webview.ExecuteScript(
         PCWSTR::from_raw(encode_wide(js).as_ptr()),
-        &ExecuteScriptCompletedHandler::create(Box::new(|_, _| (Ok(())))),
+        &ExecuteScriptCompletedHandler::create(Box::new(|_, _| {
+          drop(span);
+          Ok(())
+        })),
       )
     }
   }

src/webview/wkwebview/mod.rs

@@ -93,6 +93,8 @@ impl InnerWebView {
     extern "C" fn did_receive(this: &Object, _: Sel, _: id, msg: id) {
       // Safety: objc runtime calls are unsafe
       unsafe {
+        let _span = tracing::info_span!("wry::ipc::handle").entered();
+
         let function = this.get_ivar::<*mut c_void>("function");
         if !function.is_null() {
           let function =
@@ -111,6 +113,8 @@ impl InnerWebView {
     // Task handler for custom protocol
     extern "C" fn start_task(this: &Object, _: Sel, _webview: id, task: id) {
       unsafe {
+        let span = tracing::info_span!("wry::custom_protocol::handle", uri = tracing::field::Empty)
+          .entered();
         let function = this.get_ivar::<*mut c_void>("function");
         if !function.is_null() {
           let function = &mut *(*function
@@ -120,10 +124,13 @@ impl InnerWebView {
           let request: id = msg_send![task, request];
           let url: id = msg_send![request, URL];
 
-          let nsstring = {
+          let uri_nsstring = {
             let s: id = msg_send![url, absoluteString];
             NSString(s)
           };
+          let uri = uri_nsstring.to_str();
+
+          span.record("uri", uri);
 
           // Get request method (GET, POST, PUT etc...)
           let method = {
@@ -132,9 +139,7 @@ impl InnerWebView {
           };
 
           // Prepare our HttpRequest
-          let mut http_request = Request::builder()
-            .uri(nsstring.to_str())
-            .method(method.to_str());
+          let mut http_request = Request::builder().uri(uri).method(method.to_str());
 
           // Get body
           let mut sent_form_body = Vec::new();
@@ -158,6 +163,8 @@ impl InnerWebView {
             let _: () = msg_send![body_stream, close];
           }
 
+          tracing::debug!("done reading body");
+
           // Extract all headers fields
           let all_headers: id = msg_send![request, allHTTPHeaderFields];
 
@@ -179,7 +186,11 @@ impl InnerWebView {
           // send response
           match http_request.body(sent_form_body) {
             Ok(final_request) => {
-              if let Ok(sent_response) = function(&final_request) {
+              let res = {
+                let _span = tracing::info_span!("wry::custom_protocol::call_handler").entered();
+                function(&final_request)
+              };
+              if let Ok(sent_response) = res {
                 let content = sent_response.body();
                 // default: application/octet-stream, but should be provided by the client
                 let wanted_mime = sent_response.headers().get(CONTENT_TYPE);
@@ -852,7 +863,15 @@ r#"Object.defineProperty(window, 'ipc', {
     } else {
       // Safety: objc runtime calls are unsafe
       unsafe {
-        let _: id = msg_send![self.webview, evaluateJavaScript:NSString::new(js) completionHandler:null::<*const c_void>()];
+        let span = Mutex::new(Some(tracing::debug_span!("wry::eval").entered()));
+        let handler = block::ConcreteBlock::new(move |_object, _error| {
+          span.lock().unwrap().take();
+        });
+        // Put the block on the heap
+        let handler = handler.copy();
+        let completion_handler: &block::Block<(id, id), ()> = &handler;
+
+        let _: id = msg_send![self.webview, evaluateJavaScript:NSString::new(js) completionHandler:completion_handler];
       }
     }
     Ok(())