trim whitespace on usernames

1 files changed, 7 insertions, 1 deletions

diff --git a/src/main.rs b/src/main.rs
index 151ab30..7edc01c 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -88,6 +88,12 @@ async fn new_user(data: Json<UserForm>, db: &State<Mutex<DB>>) -> (Status, Resul
     let mut db = db.lock().await;
     let id;
 
+    let name = data.name.to_string();
+    let name = name.trim();
+    if name.is_empty() {
+        return (Status::BadRequest, Err("Invalid Username".to_string().into()))
+    }
+
     if data.id == uid::EMPTY_UID {
         match db.uid_generator.new_uid() {
             // Gotten a new UID to make a new user with
@@ -113,7 +119,7 @@ async fn new_user(data: Json<UserForm>, db: &State<Mutex<DB>>) -> (Status, Resul
     match id {
         None => return (Status::InternalServerError, Err("".to_string().into())),
         Some(n) => {
-            match db.new_user(data.name.clone(), data.password.clone(), n).await {
+            match db.new_user(name.into(), data.password.clone(), n).await {
                 // User has been created
                 Ok(u) => return (Status::Created, Ok(Json(u.into()))),
                 // Could not create user