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/database/mod.rs      |  9 +++++++++
 src/database/posts.rs    | 12 ++++++++++++
 src/database/sessions.rs | 14 ++++++++++++++
 src/database/users.rs    | 12 ++++++++++++
 4 files changed, 47 insertions(+)

(limited to 'src/database')

diff --git a/src/database/mod.rs b/src/database/mod.rs
index d48f352..b24c1e1 100644
--- a/src/database/mod.rs
+++ b/src/database/mod.rs
@@ -1,3 +1,5 @@
+use tracing::instrument;
+
 pub mod posts;
 pub mod sessions;
 pub mod users;
@@ -12,3 +14,10 @@ pub fn init() -> Result<(), rusqlite::Error> {
     sessions::init()?;
     Ok(())
 }
+
+#[instrument()]
+pub fn query(query: String) -> Result<usize, rusqlite::Error> {
+    tracing::trace!("Running custom query");
+    let conn = connect()?;
+    conn.execute(&query, [])
+}
\ No newline at end of file
diff --git a/src/database/posts.rs b/src/database/posts.rs
index 58470f0..3f2fc58 100644
--- a/src/database/posts.rs
+++ b/src/database/posts.rs
@@ -77,6 +77,18 @@ pub fn get_post_page(page: u64) -> Result<Vec<Post>, rusqlite::Error> {
     Ok(row.into_iter().flatten().collect())
 }
 
+#[instrument()]
+pub fn get_all_posts() -> Result<Vec<Post>, rusqlite::Error> {
+    tracing::trace!("Retrieving posts page");
+    let conn = database::connect()?;
+    let mut stmt = conn.prepare("SELECT * FROM posts ORDER BY post_id")?;
+    let row = stmt.query_map([], |row| {
+        let row = post_from_row(row)?;
+        Ok(row)
+    })?;
+    Ok(row.into_iter().flatten().collect())
+}
+
 #[instrument()]
 pub fn get_users_posts(user_id: u64) -> Result<Vec<Post>, rusqlite::Error> {
     tracing::trace!("Retrieving users posts");
diff --git a/src/database/sessions.rs b/src/database/sessions.rs
index 8d4ca73..9adccd4 100644
--- a/src/database/sessions.rs
+++ b/src/database/sessions.rs
@@ -32,6 +32,20 @@ pub fn get_session(token: &str) -> Result<Option<Session>, rusqlite::Error> {
     Ok(row)
 }
 
+#[instrument()]
+pub fn get_all_sessions() -> Result<Vec<Session>, rusqlite::Error> {
+    tracing::trace!("Retrieving session");
+    let conn = database::connect()?;
+    let mut stmt = conn.prepare("SELECT * FROM sessions")?;
+    let row = stmt.query_map([], |row| {
+        Ok(Session {
+            user_id: row.get(0)?,
+            token: row.get(1)?,
+        })
+    })?;
+    Ok(row.into_iter().flatten().collect())
+}
+
 #[instrument()]
 pub fn set_session(user_id: u64, token: &str) -> Result<(), Box<dyn std::error::Error>> {
     tracing::trace!("Setting new session");
diff --git a/src/database/users.rs b/src/database/users.rs
index 05a3a57..7f8e407 100644
--- a/src/database/users.rs
+++ b/src/database/users.rs
@@ -117,6 +117,18 @@ pub fn get_user_page(page: u64, hide_password: bool) -> Result<Vec<User>, rusqli
     Ok(row.into_iter().flatten().collect())
 }
 
+#[instrument()]
+pub fn get_all_users() -> Result<Vec<User>, rusqlite::Error> {
+    tracing::trace!("Retrieving user page");
+    let conn = database::connect()?;
+    let mut stmt = conn.prepare("SELECT * FROM users ORDER BY user_id")?;
+    let row = stmt.query_map([], |row| {
+        let row = user_from_row(row, false)?;
+        Ok(row)
+    })?;
+    Ok(row.into_iter().flatten().collect())
+}
+
 #[instrument()]
 pub fn add_user(request: RegistrationRequet) -> Result<User, rusqlite::Error> {
     tracing::trace!("Adding new user");
-- 
cgit v1.2.3-freya