diff --git a/src/server/main.rs b/src/server/main.rs index c057aa2..fc20010 100644 --- a/src/server/main.rs +++ b/src/server/main.rs @@ -1,7 +1,7 @@ use dioxus::{fullstack::axum::Router, server::axum::Extension}; use crate::App; -use crate::server::{config, database, key_val_store}; +use crate::server::{auth::build_auth_layer, config, database, key_val_store}; use crate::util::error::{Contextualize, Error, Result}; pub fn main() -> Result { @@ -24,11 +24,15 @@ async fn router_setup() -> Result { .await .err_context("Failed database setup")?; - let _key_val_pool = key_val_store::setup(&config.key_val_store.connection_uri()) + let key_val_pool = key_val_store::setup(&config.key_val_store.connection_uri()) .await .err_context("Failed key-value store setup")?; - let router = dioxus::server::router(App).layer(Extension(db_pool)); + let auth_layer = build_auth_layer(db_pool.clone(), key_val_pool); + + let router = dioxus::server::router(App) + .layer(Extension(db_pool)) + .layer(auth_layer); tracing::info!("Setup complete, returning Router..."); Ok(router)