Add styling, user request

2 files changed, 17 insertions, 1 deletions

diff --git a/src/main.rs b/src/main.rs
index d63159a..99ad195 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -20,7 +20,12 @@ fn main() {
     rocket::ignite()
         .mount(
             "/",
-            routes![serve::token, serve::get_tracks, serve::match_users],
+            routes![
+                serve::token,
+                serve::get_tracks,
+                serve::match_users,
+                serve::get_users
+            ],
         )
         .launch();
 }
diff --git a/src/serve.rs b/src/serve.rs
index 08f3f8b..ade84ca 100644
--- a/src/serve.rs
+++ b/src/serve.rs
@@ -220,3 +220,14 @@ pub fn match_users(name1: String, name2: String) -> Result<String, status::NotFo
     }
     Ok(songs)
 }
+
+#[get("/user")]
+pub fn get_users() -> Result<String, status::NotFound<String>> {
+    let mut client = crate::CLIENT.lock().unwrap();
+    let mut users = String::new();
+    for row in client.query("SELECT user_name FROM suser", &[]).unwrap() {
+        let user: String = row.get(0);
+        users = format!("{}{}\n", users, user);
+    }
+    Ok(users)
+}