1 files changed, 6 insertions, 3 deletions
diff --git a/src/ui/routes/mod.rs b/src/ui/routes/mod.rs
index 80dc1e5..328eb73 100644
--- a/src/ui/routes/mod.rs
+++ b/src/ui/routes/mod.rs
@@ -1,6 +1,6 @@
-use axum::{Router, middleware, routing::get};
+use axum::{Router, response::Redirect, routing::get};
 
-use crate::{app::App, ui::middleware::setup_required};
+use crate::app::App;
 
 mod ch;
 mod get;
@@ -21,7 +21,10 @@ pub fn router(app: &App) -> Router<App> {
             .route("/login", get(login::get::handler))
             .route("/ch/{channel}", get(ch::channel::get::handler))
             .route("/invite/{invite}", get(invite::invite::get::handler))
-            .route_layer(middleware::from_fn_with_state(app.clone(), setup_required)),
+            .route_layer(crate::setup::Required::with_fallback(
+                app.clone(),
+                Redirect::to("/setup"),
+            )),
     ]
     .into_iter()
     .fold(Router::default(), Router::merge)