From c360d1d773bbfa9c9cfec97a202cc2b29b141010 Mon Sep 17 00:00:00 2001
From: Fuwn <contact@fuwn.me>
Date: Sat, 15 Apr 2023 02:00:45 +0000
Subject: refactor(hooks): implement call for hooks

---
 examples/callbacks.rs | 20 ++++++++++++--------
 1 file changed, 12 insertions(+), 8 deletions(-)

(limited to 'examples/callbacks.rs')

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
 }
-- 
cgit v1.2.3