windmark / examples / stateful_module.rs

//! `cargo run --example stateful_module --features response-macros`

use windmark::{context::HookContext, router::Router};

#[derive(Default)]
struct Clicker {
  clicks: usize,
}

impl windmark::module::Module for Clicker {
  fn on_attach(&mut self, _router: &mut Router) {
    println!("module 'clicker' has been attached!");
  }

  fn on_pre_route(&mut self, context: HookContext) {
    self.clicks += 1;

    println!(
      "module 'clicker' has been called before the route '{}' with {} clicks!",
      context.url.path(),
      self.clicks,
    );
  }

  fn on_post_route(&mut self, context: HookContext) {
    println!(
      "module 'clicker' clicker has been called after the route '{}' with {} \
       clicks!",
      context.url.path(),
      self.clicks,
    );
  }
}

#[windmark::main]
async fn main() -> Result<(), Box<dyn std::error::Error>> {
  let mut router = Router::new();

  router.set_private_key_file("windmark_private.pem");
  router.set_certificate_file("windmark_public.pem");
  #[cfg(feature = "logger")]
  {
    router.enable_default_logger(true);
  }
  router.attach(Clicker::default());
  router.mount("/", windmark::success!("Hello!"));

  router.run().await
}