From a8b6798dfe4939dc8c36cec6b36a4261477fb087 Mon Sep 17 00:00:00 2001
From: Tyler Murphy <tylermurphy534@gmail.com>
Date: Wed, 15 Feb 2023 00:47:55 -0500
Subject: fix root db call

---
 src/types/extract.rs | 16 ++++++++++++++--
 1 file changed, 14 insertions(+), 2 deletions(-)

(limited to 'src/types')

diff --git a/src/types/extract.rs b/src/types/extract.rs
index f05215f..a76eac4 100644
--- a/src/types/extract.rs
+++ b/src/types/extract.rs
@@ -9,7 +9,7 @@ use axum::{
     extract::{ConnectInfo, FromRequest, FromRequestParts},
     http::{header::USER_AGENT, request::Parts, Request},
     response::Response,
-    BoxError, RequestExt,
+    BoxError, RequestExt, middleware::Next,
 };
 use bytes::Bytes;
 use image::{io::Reader, DynamicImage, ImageFormat};
@@ -100,7 +100,7 @@ where
         };
 
         let Some(db) = parts.extensions.get::<DatabaseExtention>() else {
-            return Err(ResponseCode::Forbidden.text("Could not connect to database"))
+            return Err(ResponseCode::InternalServerError.text("Could not connect to database"))
         };
 
         let db = db.0.lock().await;
@@ -288,6 +288,18 @@ where
     }
 }
 
+pub async fn connect<B>(mut req: Request<B>, next: Next<B>) -> Response
+where
+    B: Send,
+{
+    if let Ok(db) = database::Database::connect() {
+        let ex = DatabaseExtention(Mutex::new(db));
+        req.extensions_mut().insert(ex);
+    }
+
+    next.run(req).await
+}
+
 async fn read_body<S, B>(mut req: Request<B>, state: &S) -> Result<Vec<u8>>
 where
     B: HttpBody + Sync + Send + 'static,
-- 
cgit v1.2.3-freya