megacommit

backend/src/main.rs

@@ -18,6 +18,7 @@ use crate::messages::ChatBroadcaster;
 pub mod auth;
 pub mod cdn;
 pub mod db;
+pub mod handlers;
 pub mod llm;
 pub mod messages;
 
@@ -71,9 +72,11 @@ fn rocket() -> Rocket<Build> {
             routes![
                 favicon,
                 messages::chat_page,
+                messages::chat_page_preview,
                 auth::signup_page,
                 auth::login_page,
                 auth::mfa_page,
+                auth::invite_page,
             ],
         )
         .mount(
@@ -87,6 +90,16 @@ fn rocket() -> Rocket<Build> {
                 auth::signup,
                 auth::login,
                 auth::get_totp,
+                auth::confirm_totp,
+                auth::generate_invite,
+            ],
+        )
+        .register(
+            "/",
+            catchers![
+                handlers::handle_404,
+                handlers::handle_401,
+                handlers::handle_default
             ],
         )
 }