refactor(hooks): implement call for hooks

1 files changed, 12 insertions, 8 deletions

diff --git a/examples/callbacks.rs b/examples/callbacks.rs
index b161a03..d10dab9 100644
--- a/examples/callbacks.rs
+++ b/examples/callbacks.rs
@@ -18,27 +18,31 @@
 
 //! `cargo run --example callbacks`
 
+use windmark::context::HookContext;
+
 #[windmark::main]
 async fn main() -> Result<(), Box<dyn std::error::Error>> {
   windmark::Router::new()
     .set_private_key_file("windmark_private.pem")
     .set_certificate_file("windmark_public.pem")
     .mount("/", windmark::success!("Hello!"))
-    .set_pre_route_callback(|context| {
+    .set_pre_route_callback(|context: HookContext| {
       println!(
         "accepted connection from {} to {}",
         context.peer_address.unwrap().ip(),
         context.url.to_string()
       )
     })
-    .set_post_route_callback(|context, content| {
-      content.content = content.content.replace("Hello", "Hi");
+    .set_post_route_callback(
+      |context: HookContext, content: &mut windmark::Response| {
+        content.content = content.content.replace("Hello", "Hi");
 
-      println!(
-        "closed connection from {}",
-        context.peer_address.unwrap().ip()
-      )
-    })
+        println!(
+          "closed connection from {}",
+          context.peer_address.unwrap().ip()
+        )
+      },
+    )
     .run()
     .await
 }