From a2c89301d5c63f573ac6680f9c11acda69043a78 Mon Sep 17 00:00:00 2001
From: Tyler Murphy <=>
Date: Wed, 26 Jul 2023 01:04:39 -0400
Subject: start dms

---
 src/api/mod.rs | 33 ++++++---------------------------
 1 file changed, 6 insertions(+), 27 deletions(-)

(limited to 'src/api/mod.rs')

diff --git a/src/api/mod.rs b/src/api/mod.rs
index 8b631c8..0c01ea0 100644
--- a/src/api/mod.rs
+++ b/src/api/mod.rs
@@ -1,32 +1,20 @@
 use crate::types::extract::{RouterURI, self};
-use axum::{
-    error_handling::HandleErrorLayer,
-    BoxError, Extension, Router, middleware,
-};
-use tower::ServiceBuilder;
-use tower_governor::{
-    errors::display_error, governor::GovernorConfigBuilder, key_extractor::SmartIpKeyExtractor,
-    GovernorLayer,
-};
 
+pub mod chat;
 pub mod admin;
 pub mod auth;
 pub mod posts;
 pub mod users;
 
 pub use auth::RegistrationRequet;
+use axum::{Extension, Router, middleware};
 
 pub fn router() -> Router {
-    let governor_conf = Box::new(
-        GovernorConfigBuilder::default()
-            .burst_size(15)
-            .per_second(1)
-            .key_extractor(SmartIpKeyExtractor)
-            .finish()
-            .expect("Failed to create rate limiter"),
-    );
-
     Router::new()
+        .nest(
+            "/chat",
+            chat::router().layer(Extension(RouterURI("/api/chat"))),
+        )
         .nest(
             "/admin",
             admin::router().layer(Extension(RouterURI("/api/admin"))),
@@ -43,14 +31,5 @@ pub fn router() -> Router {
             "/posts",
             posts::router().layer(Extension(RouterURI("/api/posts"))),
         )
-        .layer(
-            ServiceBuilder::new()
-                .layer(HandleErrorLayer::new(|e: BoxError| async move {
-                    display_error(e)
-                }))
-                .layer(GovernorLayer {
-                    config: Box::leak(governor_conf),
-                }),
-        )
         .layer(middleware::from_fn(extract::connect))
 }
-- 
cgit v1.2.3-freya