feat: Send additional info on rate limit

src/main.rs

@@ -35,7 +35,7 @@ async fn main() {
     let app = routes::mount(Router::new())
         .route("/ws", axum::routing::get(gateway::upgrade))
         .layer(middleware::from_fn(auth::handle))
-        .layer(middleware::from_fn(ratelimit::handle!(50, 1000 * 60)))
+        .layer(middleware::from_fn(ratelimit::handle!(3, 1000 * 60)))
         .layer(cors::handle())
         .fallback((|| async { StatusCode::NOT_FOUND }).into_service());
 

src/utils/error.rs

@@ -1,7 +1,7 @@
 use crate::middlewares::ratelimit::RateLimitInfo;
 use axum::{
     extract::rejection::JsonRejection,
-    http::StatusCode,
+    http::{HeaderMap, StatusCode},
     response::{IntoResponse, Json, Response},
 };
 use quick_error::quick_error;
@@ -85,16 +85,24 @@ impl IntoResponse for Error {
         let status = match self {
             Error::RateLimited { .. } => StatusCode::TOO_MANY_REQUESTS,
             Error::InvalidToken => StatusCode::UNAUTHORIZED,
+            Error::InvalidBody => StatusCode::UNPROCESSABLE_ENTITY,
             _ => StatusCode::BAD_REQUEST,
         };
 
+        let mut headers = HeaderMap::new();
         let mut body = serde_json::json!({ "type": self });
         let msg = self.to_string();
 
         if msg.contains(' ') {
             body["message"] = serde_json::json!(msg);
         }
 
-        (status, Json(body)).into_response()
+        if let Error::RateLimited(info) = self {
+            headers.insert("X-RateLimit-Remaining", info.remaining.into());
+            headers.insert("X-RateLimit-Limit", info.limit.into());
+            headers.insert("Retry-After", info.retry_after.into());
+        }
+
+        (status, headers, Json(body)).into_response()
     }
 }