From fb282218757d606fbd90353eed37a140f4fe2972 Mon Sep 17 00:00:00 2001
From: AINDUSTRIES <contact.aindustries@gmail.com>
Date: Fri, 4 Oct 2024 17:17:46 +0200
Subject: [PATCH] added check for token generation (look if exists) (better)

---
 src/main.rs | 5 +----
 1 file changed, 1 insertion(+), 4 deletions(-)

diff --git a/src/main.rs b/src/main.rs
index fea44d9..8c1f1e6 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -581,13 +581,10 @@ async fn register(
                 .unwrap()
                 .to_string();
             let mut token = Alphanumeric.sample_string(&mut OsRng, 256);
-            while match sqlx::query!(r#"SELECT id FROM users WHERE token=?1"#, token)
+            while let Ok(Some(_)) = sqlx::query!(r#"SELECT id FROM users WHERE token=?1"#, token)
                 .fetch_optional(&mut *conn)
                 .await
             {
-                Ok(Some(user)) => true,
-                _ => false,
-            } {
                 token = Alphanumeric.sample_string(&mut OsRng, 256);
             }
             let result = sqlx::query!(r#"INSERT INTO users ( username, saltyhash, permissions, token) VALUES ( ?1, ?2, ?3, ?4 )"#, login.username, hash, 0, token).execute(&mut *conn).await;