From ac58a612a3fe928793b77c592551fdd962b69064 Mon Sep 17 00:00:00 2001
From: Tyler Murphy <tylermurphy534@gmail.com>
Date: Sun, 29 Jan 2023 19:28:48 -0500
Subject: admin page

---
 src/types/extract.rs | 32 +++++++++++++++++++++++++++++++-
 src/types/post.rs    |  8 ++++++++
 src/types/session.rs |  8 ++++++++
 src/types/user.rs    |  8 ++++++++
 4 files changed, 55 insertions(+), 1 deletion(-)

(limited to 'src/types')

diff --git a/src/types/extract.rs b/src/types/extract.rs
index 4d92a3b..64a3e73 100644
--- a/src/types/extract.rs
+++ b/src/types/extract.rs
@@ -19,7 +19,7 @@ use crate::{
         http::{ResponseCode, Result},
         session::Session,
         user::User,
-    },
+    }, admin,
 };
 
 pub struct AuthorizedUser(pub User);
@@ -53,6 +53,36 @@ where
     }
 }
 
+pub struct AdminUser;
+
+#[async_trait]
+impl<S> FromRequestParts<S> for AdminUser
+where
+    S: Send + Sync,
+{
+    type Rejection = Response;
+
+    async fn from_request_parts(parts: &mut Parts, state: &S) -> Result<Self> {
+        let Ok(Some(cookies)) = Option::<TypedHeader<Cookie>>::from_request_parts(parts, state).await else {
+            return Err(ResponseCode::Forbidden.text("No cookies provided"))
+        };
+
+        let Some(secret) = cookies.get("admin") else {
+            return Err(ResponseCode::Forbidden.text("No admin secret provided"))
+        };
+
+        println!("{}", secret);
+
+        let check = admin::get_secret().await;
+
+        if check != secret {
+            return Err(ResponseCode::Unauthorized.text("Auth token invalid"))
+        }
+
+        Ok(Self)
+    }
+}
+
 pub struct Log;
 
 #[async_trait]
diff --git a/src/types/post.rs b/src/types/post.rs
index 90eada2..7397009 100644
--- a/src/types/post.rs
+++ b/src/types/post.rs
@@ -50,6 +50,14 @@ impl Post {
         Ok(posts)
     }
 
+    #[instrument()]
+    pub fn reterieve_all() -> Result<Vec<Self>> {
+        let Ok(posts) = database::posts::get_all_posts() else {
+            return Err(ResponseCode::InternalServerError.text("Failed to fetch posts"))
+        };
+        Ok(posts)
+    }
+
     #[instrument()]
     pub fn new(user_id: u64, content: String) -> Result<Self> {
         let Ok(post) = database::posts::add_post(user_id, &content) else {
diff --git a/src/types/session.rs b/src/types/session.rs
index e704ac7..a9073aa 100644
--- a/src/types/session.rs
+++ b/src/types/session.rs
@@ -21,6 +21,14 @@ impl Session {
         Ok(session)
     }
 
+    #[instrument()]
+    pub fn reterieve_all() -> Result<Vec<Self>> {
+        let Ok(sessions) = database::sessions::get_all_sessions() else {
+            return Err(ResponseCode::InternalServerError.text("Failed to fetch sessions"))
+        };
+        Ok(sessions)
+    }
+
     #[instrument()]
     pub fn new(user_id: u64) -> Result<Self> {
         let token: String = rand::thread_rng()
diff --git a/src/types/user.rs b/src/types/user.rs
index fcfbe91..2bffa52 100644
--- a/src/types/user.rs
+++ b/src/types/user.rs
@@ -68,6 +68,14 @@ impl User {
         Ok(user)
     }
 
+    #[instrument()]
+    pub fn reterieve_all() -> Result<Vec<Self>> {
+        let Ok(users) = database::users::get_all_users() else {
+            return Err(ResponseCode::InternalServerError.text("Failed to fetch users"))
+        };
+        Ok(users)
+    }
+
     #[instrument()]
     pub fn new(request: RegistrationRequet) -> Result<Self> {
         if Self::from_email(&request.email).is_ok() {
-- 
cgit v1.2.3-freya