middleware/app: Add axum equivalent

Author: Turbo87

src/middleware/app.rs

@@ -1,4 +1,8 @@
 use super::prelude::*;
+use axum::extract::State;
+use axum::middleware::Next;
+use axum::response::Response;
+use http::Request;
 
 use crate::app::AppState;
 
@@ -20,6 +24,17 @@ impl Middleware for AppMiddleware {
     }
 }
 
+/// `axum` middleware that injects the `AppState` instance into the `Request` extensions.
+pub async fn add_app_state_extension<B>(
+    State(app_state): State<AppState>,
+    mut request: Request<B>,
+    next: Next<B>,
+) -> Response {
+    request.extensions_mut().insert(app_state);
+
+    next.run(request).await
+}
+
 /// Adds an `app()` method to the `Request` type returning the global `App` instance
 pub trait RequestApp {
     fn app(&self) -> &AppState;